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

langgenius · Oct 13, 2023 · f4be2b8 · f4be2b8
1 parent e83e239
commit f4be2b8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/api/core/third_party/langchain/llms/minimax_llm.py b/api/core/third_party/langchain/llms/minimax_llm.py
@@ -211,6 +211,16 @@ def _stream(
 
                 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 = {