|
@@ -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):
|
|
|
"""
|