Update services/agent_crewai.py

services/agent_crewai.py

@@ -276,38 +276,30 @@ Current session file: {session_file_path}
 Execution plan: {plan_json}
 """
     
-    # Initialize LLM explicitly using LiteLLM wrapper
-    # This works even if google-generativeai is not installed
+    # Initialize Gemini LLM using google.generativeai directly
+    # This bypasses CrewAI's string parser which requires google-genai extra
     try:
-        from litellm import completion
-        from langchain.llms.base import LLM
-        from typing import Any, List, Optional
+        import google.generativeai as genai
         
-        class LiteLLMWrapper(LLM):
-            """Wrapper for LiteLLM to use with CrewAI"""
-            model_name: str = "gemini/gemini-2.0-flash"
-            
-            @property
-            def _llm_type(self) -> str:
-                return "litellm"
-            
-            def _call(
-                self,
-                prompt: str,
-                stop: Optional[List[str]] = None,
-                 **kwargs: Any,
-            ) -> str:
-                response = completion(
-                    model=self.model_name,
-                    messages=[{"role": "user", "content": prompt}],
-                    api_key=os.getenv("GOOGLE_API_KEY") or os.getenv("GEMINI_API_KEY"),
-                )
-                return response.choices[0].message.content
+        # Configure Gemini
+        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")
+        
+    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"
         
-        llm = LiteLLMWrapper()
     except Exception as e:
-        print(f"Warning: Could not initialize LiteLLM wrapper: {e}")
-        print("Falling back to string-based LLM (may require google-generativeai)")
+        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")
     
     agent = Agent(