fix: tongyi multimodal response format error response process

langchain-ai · Dec 10, 2024 · b46152f · b46152f
1 parent f56728b
commit b46152f
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/libs/community/langchain_community/chat_models/tongyi.py b/libs/community/langchain_community/chat_models/tongyi.py
@@ -549,16 +549,17 @@ def _stream_completion_with_retry(**_kwargs: Any) -> Any:
                 ):
                     # inline fix response text logic
                     resp_copy = json.loads(json.dumps(resp))
-                    choice = resp_copy["output"]["choices"][0]
-                    message = choice["message"]
-                    if isinstance(message.get("content"), list):
-                        content_text = "".join(
-                            item.get("text", "")
-                            for item in message["content"]
-                            if isinstance(item, dict)
-                        )
-                        message["content"] = content_text
-                    resp = resp_copy
+                    if resp_copy.get("output") and resp_copy["output"].get("choices"):
+                        choice = resp_copy["output"]["choices"][0]
+                        message = choice["message"]
+                        if isinstance(message.get("content"), list):
+                            content_text = "".join(
+                                item.get("text", "")
+                                for item in message["content"]
+                                if isinstance(item, dict)
+                            )
+                            message["content"] = content_text
+                        resp = resp_copy
                     if prev_resp is None:
                         delta_resp = resp
                     else: