luojunhui há 2 meses atrás
pai
commit
bfff1e0004
1 ficheiros alterados com 13 adições e 8 exclusões
  1. 13 8
      applications/api/google_ai_api.py

+ 13 - 8
applications/api/google_ai_api.py

@@ -3,6 +3,7 @@
 """
 
 from google import genai
+from google.genai.errors import ClientError
 from tqdm import tqdm
 
 
@@ -61,14 +62,18 @@ class GoogleAIAPI(object):
         prompt: 提示词
         video_file: <class 'google.genai.types.File'>
         """
-        response = self.client.models.generate_content(
-            model='gemini-1.5-flash',
-            contents=[
-                video_file,
-                prompt
-            ]
-        )
-        return response.text
+        try:
+            response = self.client.models.generate_content(
+                model='gemini-2.0-flash',
+                contents=[
+                    video_file,
+                    prompt
+                ]
+            )
+            return response.text
+        except ClientError as e:
+            print(e)
+            return None
 
     def delete_video(self, file_name: str):
         """