🐛 fix: add type check for completion response from API to ensure correct handling

2025-02-25 11:46:17 +08:00
parent 9451dc3fd4
commit 98e8ecb8e2
1 changed files with 3 additions and 1 deletions
@@ -89,7 +89,9 @@ class ProviderOpenAIOfficial(Provider):
            extra_body=extra_body
        )

-        assert isinstance(completion, ChatCompletion)
+        if not isinstance(completion, ChatCompletion):
+            raise Exception(f"API 返回的 completion 类型错误：{type(completion)}: {completion}。")
+
        logger.debug(f"completion: {completion}")

        if len(completion.choices) == 0: