Update services/agent_crewai.py

services/agent_crewai.py

@@ -276,31 +276,31 @@ Current session file: {session_file_path}
 Execution plan: {plan_json}
 """
     
-    # Initialize Gemini LLM using google.generativeai directly
-    # This bypasses CrewAI's string parser which requires google-genai extra
+    # Use LiteLLM-compatible string format for Gemini
+    # CrewAI internally uses LiteLLM which requires "gemini/" prefix
+    # Now that google-generativeai is installed, this should work
     try:
         import google.generativeai as genai
         
-        # Configure Gemini
+        # Verify package is installed and configure it
         api_key = os.getenv("GOOGLE_API_KEY") or os.getenv("GEMINI_API_KEY")
         if api_key:
             genai.configure(api_key=api_key)
-        
-        # Create Gemini model instance
-        model = genai.GenerativeModel('gemini-2.0-flash-exp')
-        llm = model
-        print(f"✅ CrewAI using Google Gemini directly: gemini-2.0-flash-exp")
+            # Use LiteLLM string format (not the model object)
+            llm = "gemini/gemini-2.0-flash-exp"
+            print(f"✅ CrewAI using Gemini via LiteLLM: {llm}")
+        else:
+            print("⚠️  No Google API key found")
+            llm = "gemini/gemini-2.0-flash-exp"
         
     except ImportError as e:
-        print(f"Warning: google.generativeai not available: {e}")
-        print("Falling back to LiteLLM string format")
-        # Use litellm format as fallback
-        llm = "gemini/gemini-2.0-flash"
+        print(f"⚠️  google.generativeai not available: {e}")
+        print("Using LiteLLM format anyway")
+        llm = "gemini/gemini-2.0-flash-exp"
         
     except Exception as e:
-        print(f"Warning: Could not initialize Gemini: {e}")
-        print("Falling back to CrewAI string format")
-        llm = os.getenv("CREWAI_LLM", "gemini/gemini-2.0-flash")
+        print(f"⚠️  Error configuring Gemini: {e}")
+        llm = "gemini/gemini-2.0-flash-exp"
     
     agent = Agent(
         role="Document Processing Specialist",