Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -131,16 +131,7 @@ class AITools():
     
             response = client.models.generate_content(
                 model=LLM_IMAGE_ANALYSIS,
-                contents=[
-                    types.Content(
-                        parts=[
-                            types.Part(
-                                file_data=types.FileData(file_uri=file.uri)
-                            ),
-                            types.Part(text=question.strip())
-                        ]
-                    )
-                ],
+                contents=[file, question],
                 config=types.GenerateContentConfig(
                     thinking_config=types.ThinkingConfig(
                         thinking_level=THINKING_LEVEL_TOOLS
@@ -173,16 +164,7 @@ class AITools():
             
             response = client.models.generate_content(
                 model=LLM_AUDIO_ANALYSIS,
-                contents=[
-                    types.Content(
-                        parts=[
-                            types.Part(
-                                file_data=types.FileData(file_uri=file.uri)
-                            ),
-                            types.Part(text=question.strip())
-                        ]
-                    )
-                ],
+                contents=[file, question],
                 config=types.GenerateContentConfig(
                     thinking_config=types.ThinkingConfig(
                         thinking_level=THINKING_LEVEL_TOOLS
@@ -215,16 +197,7 @@ class AITools():
             
             response = client.models.generate_content(
                 model=LLM_VIDEO_ANALYSIS,
-                contents=[
-                    types.Content(
-                        parts=[
-                            types.Part(
-                                file_data=types.FileData(file_uri=file.uri)
-                            ),
-                            types.Part(text=question.strip())
-                        ]
-                    )
-                ],
+                contents=[file, question],
                 config=types.GenerateContentConfig(
                     thinking_config=types.ThinkingConfig(
                         thinking_level=THINKING_LEVEL_TOOLS
@@ -367,16 +340,7 @@ class AITools():
     
             response = client.models.generate_content(
                 model=LLM_CODE_EXECUTION,
-                contents=[
-                    types.Content(
-                        parts=[
-                            types.Part(
-                                file_data=types.FileData(file_uri=file.uri)
-                            ),
-                            types.Part(text=question.strip())
-                        ]
-                    )
-                ],
+                contents=[file, question],
                 config=types.GenerateContentConfig(
                     tools=[types.Tool(code_execution=types.ToolCodeExecution)],
                     thinking_config=types.ThinkingConfig(
@@ -425,10 +389,13 @@ class AITools():
                     )
                 ]
             )
+
+            file = client.files.upload(file=file_path) # TODO
             
             response = client.models.generate_content(
                 model=LLM_IMAGE_TO_FEN,
-                contents=[content],
+                #contents=[content], # TODO
+                contents=[file, question],
                 config=types.GenerateContentConfig(
                     thinking_config=types.ThinkingConfig(
                         thinking_level=THINKING_LEVEL_TOOLS