Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -54,7 +54,17 @@ class AITools():
             client = AITools._get_client()
             
             file = client.files.upload(file=file_path)
-    
+
+            ###
+            while True:
+                media_file = client.files.get(name=file.name)
+                if media_file.state == "ACTIVE":
+                    break
+                elif media_file.state == "FAILED":
+                    raise RuntimeError("Media file processing failed")
+                time.sleep(1)            
+            ###
+
             response = client.models.generate_content(
                 model=model,
                 contents=[file, question],