fix: 兼容旧版本google-genai sdk

2025-04-16 00:34:08 +08:00
parent a0ce0cf18a
commit 1be5b4c7ff
1 changed files with 5 additions and 1 deletions
@@ -249,10 +249,14 @@ class ProviderGoogleGenAI(Provider):
            types.FinishReason.PROHIBITED_CONTENT,
            types.FinishReason.SPII,
            types.FinishReason.BLOCKLIST,
-            types.FinishReason.IMAGE_SAFETY,
        }:
            raise Exception("模型生成内容违反Gemini平台政策")

+        # 防止旧版本SDK不存在IMAGE_SAFETY
+        if hasattr(types.FinishReason, "IMAGE_SAFETY"):
+            if finish_reason == types.FinishReason.IMAGE_SAFETY:
+                raise Exception("模型生成内容违反Gemini平台政策")
+
        if not result_parts:
            logger.debug(result.candidates)
            raise Exception("API 返回的内容为空。")