Upload 4 files

app.py

@@ -91,7 +91,6 @@ class ResponseWrapper:
         self._text = self._extract_text()
         self._finish_reason = self._extract_finish_reason()
         if self.finish_reason != "STOP":
-            # 抛出错误
             raise BlockedPromptException(f"生成文本失败: {self.finish_reason}")
         self._prompt_token_count = self._extract_prompt_token_count()
         self._candidates_token_count = self._extract_candidates_token_count()
@@ -446,7 +445,7 @@ def chat_completions():
                 return 1, ResponseWrapper(response.json())
         except requests.exceptions.RequestException as e:
             return handle_api_error(e, attempt)
-        except StopCandidateException as e:
+        except (StopCandidateException, BlockedPromptException) as e:
             return handle_api_error(e, attempt)
 
     def generate_stream(response):