Upload app.py

app.py

@@ -232,18 +232,21 @@ def handle_api_error(error, attempt):
         switch_api_key()
         return 0, None
 
-    elif isinstance(error, BlockedPromptException):
-        reason = error.block_reason
-        if reason == generation_types.BlockReason.SAFETY:
-            logger.warning(f"提示因安全原因被阻止：{error.safety_ratings}")
+    elif isinstance(error, generation_types.BlockedPromptException):
+        reason = error.prompt_feedback.block_reason
+        if reason == generation_types.PromptFeedback.BlockReason.SAFETY:
+            logger.warning(f"提示因安全原因被阻止：{error.prompt_feedback.safety_ratings}")
             return 2, None
-        elif reason == generation_types.BlockReason.BLOCKLIST:
+        elif reason == generation_types.PromptFeedback.BlockReason.BLOCKLIST:
             logger.warning(f"提示因包含阻止列表中的术语而被阻止")
             return 2, None
-        elif reason == generation_types.BlockReason.PROHIBITED_CONTENT:
+        elif reason == generation_types.PromptFeedback.BlockReason.PROHIBITED_CONTENT:
             logger.warning(f"提示因包含禁止内容而被阻止")
             return 2, None
-        elif reason == generation_types.BlockReason.OTHER:
+        elif reason == generation_types.PromptFeedback.BlockReason.OTHER:
+            logger.warning(f"提示因未知原因被阻止")
+            return 2, None
+        elif reason == generation_types.PromptFeedback.BlockReason.BLOCK_REASON_UNSPECIFIED:
             logger.warning(f"提示因未知原因被阻止")
             return 2, None
         else: