Upload 4 files

app.py

@@ -1,5 +1,5 @@
 from flask import Flask, request, jsonify, Response, stream_with_context, render_template_string
-from google.generativeai.types import StopCandidateException, generation_types
+from google.generativeai.types import StopCandidateException, generation_types, BlockedPromptException
 from google.api_core.exceptions import InvalidArgument, ResourceExhausted, Aborted, InternalServerError, ServiceUnavailable, PermissionDenied
 import google.generativeai as genai
 import json
@@ -92,7 +92,7 @@ class ResponseWrapper:
         self._finish_reason = self._extract_finish_reason()
         if self.finish_reason != "STOP":
             # 抛出错误
-            raise StopCandidateException(f"生成文本失败: {self.finish_reason}")
+            raise BlockedPromptException(f"生成文本失败: {self.finish_reason}")
         self._prompt_token_count = self._extract_prompt_token_count()
         self._candidates_token_count = self._extract_candidates_token_count()
         self._total_token_count = self._extract_total_token_count()