Refactor get_choice_delta_text for safer dict access

geekan · Jan 28, 2024 · a68c344 · a68c344
1 parent ee0801a
commit a68c344
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metagpt/provider/base_llm.py b/metagpt/provider/base_llm.py
@@ -91,7 +91,7 @@ def get_choice_text(self, rsp: dict) -> str:
 
     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