Forráskód Böngészése

fix: raise error in minimax stream generate (#1336)

takatost 1 éve
szülő
commit
f4be2b8bcd
1 módosított fájl, 10 hozzáadás és 0 törlés
  1. 10 0
      api/core/third_party/langchain/llms/minimax_llm.py

+ 10 - 0
api/core/third_party/langchain/llms/minimax_llm.py

@@ -211,6 +211,16 @@ class MinimaxChatLLM(BaseChatModel):
 
                 token = token.lstrip("data:").strip()
                 data = json.loads(token)
+
+                if "base_resp" in data and data["base_resp"]["status_code"] > 0:
+                    raise ValueError(
+                        f"API {data['base_resp']['status_code']}"
+                        f" error: {data['base_resp']['status_msg']}"
+                    )
+
+                if not data['choices']:
+                    continue
+
                 content = data['choices'][0]['delta']
 
                 chunk_kwargs = {