Upload 4 files

app.py

@@ -234,9 +234,11 @@ def handle_api_error(error, attempt):
 
     elif isinstance(error, generation_types.BlockedPromptException):
         try:
-            full_reason_str = error.args[0]  
-            if full_reason_str.startswith("block_reason:"):
-                block_reason_str = full_reason_str.split(":", 1)[1].strip()
+            full_reason_str = str(error.args[0])
+
+            if "block_reason:" in full_reason_str:
+                start_index = full_reason_str.find("block_reason:") + len("block_reason:")
+                block_reason_str = full_reason_str[start_index:].strip()
 
                 if block_reason_str == "SAFETY":
                     logger.warning(f"提示因安全原因被阻止")