opensource/MetaGPT
3
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-10-13 16:46:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #801 from pkdog/fix-zhipu

Browse Source 
Refactor get_choice_delta_text for safer dict access while using ZHIPU API
This commit is contained in:
geekan
2024-01-31 23:35:03 +08:00
committed by GitHub
parent ee0801a116 a68c3442bc
commit d35527b0ba
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
metagpt/provider/base_llm.py
+1 -1
View File
@@ -91,7 +91,7 @@ class BaseLLM(ABC):
def get_choice_delta_text(self, rsp: dict) -> str:
"""Required to provide the first text of stream choice"""
return rsp.get("choices")[0]["delta"]["content"]
return rsp.get("choices", [{}])[0].get("delta", {}).get("content", "")
def get_choice_function(self, rsp: dict) -> dict:
"""Required to provide the first function of choice