solved chatllm problems

services/agents/base_agent.py

@@ -1,6 +1,6 @@
 # services/agents/base_agent.py
 """
-Base class for all utility agents with logging, LiteLLM integration, and interface contract.
+Base class for all utility agents with logging and CrewAI integration.
 """
 import os
 import hashlib
@@ -11,7 +11,6 @@ from typing import Dict, Any, Callable, Optional
 from abc import ABC, abstractmethod
 
 from crewai import Agent, Task, Crew
-from litellm import completion
 
 # Configure logging
 logging.basicConfig(
@@ -67,28 +66,22 @@ class BaseUtilityAgent(ABC):
         )
     
     def _create_llm(self):
-        """Create LiteLLM instance for Gemini."""
-        # LiteLLM format for CrewAI
-        # CrewAI expects llm to be a callable or LLM instance
-        # We'll use a simple wrapper that calls litellm.completion
-        class LiteLLMWrapper:
-            def __init__(self, model: str):
-                self.model = model
-                self.api_key = os.getenv("GEMINI_API_KEY")
-                if not self.api_key:
-                    raise ValueError("GEMINI_API_KEY not found in environment")
-            
-            def __call__(self, messages, **kwargs):
-                """Call LiteLLM completion."""
-                response = completion(
-                    model=self.model,
-                    messages=messages,
-                    api_key=self.api_key,
-                    **kwargs
-                )
-                return response.choices[0].message.content
+        """Create LLM instance compatible with CrewAI ≥0.80.0"""
+        # CrewAI ≥0.80.0 has native Gemini support via google-generativeai
+        # We use the model string directly instead of a wrapper
+        # CrewAI will handle the LLM initialization internally
+        
+        # For CrewAI with Gemini, we can pass the model string directly
+        # The format is "gemini/<model-name>"
+        # CrewAI will use the GEMINI_API_KEY or GOOGLE_API_KEY from environment
+        
+        # Ensure API key is set
+        if not os.getenv("GEMINI_API_KEY") and not os.getenv("GOOGLE_API_KEY"):
+            raise ValueError("GEMINI_API_KEY or GOOGLE_API_KEY not found in environment")
         
-        return LiteLLMWrapper(self.model)
+        # Return the model string - CrewAI will handle it
+        # CrewAI ≥0.80.0 accepts model strings directly
+        return self.model
     
     def _hash_data(self, data: Any) -> str:
         """Create SHA256 hash of data for logging."""

services/agents/master_orchestrator.py

@@ -12,7 +12,6 @@ from typing import Dict, Any, List, Optional
 from datetime import datetime, timezone
 
 from crewai import Agent, Task, Crew
-from litellm import completion
 
 from services.agents.message_dispatcher import MessageDispatcher, AgentMessage
 
@@ -63,24 +62,18 @@ class MasterOrchestratorAgent:
         logger.info(f"MasterLLM Orchestrator initialized (delegation: ENABLED)")
     
     def _create_llm(self):
-        """Create LiteLLM wrapper for Gemini."""
-        class LiteLLMWrapper:
-            def __init__(self, model: str):
-                self.model = model
-                self.api_key = os.getenv("GEMINI_API_KEY")
-                if not self.api_key:
-                    raise ValueError("GEMINI_API_KEY not found in environment")
-            
-            def __call__(self, messages, **kwargs):
-                response = completion(
-                    model=self.model,
-                    messages=messages,
-                    api_key=self.api_key,
-                    **kwargs
-                )
-                return response.choices[0].message.content
+        """Create LLM instance compatible with CrewAI ≥0.80.0"""
+        # CrewAI ≥0.80.0 has native Gemini support via google-generativeai
+        # We use the model string directly instead of a wrapper
+        # CrewAI will handle the LLM initialization internally
+        
+        # Ensure API key is set
+        if not os.getenv("GEMINI_API_KEY") and not os.getenv("GOOGLE_API_KEY"):
+            raise ValueError("GEMINI_API_KEY or GOOGLE_API_KEY not found in environment")
         
-        return LiteLLMWrapper(self.model)
+        # Return the model string - CrewAI will handle it
+        # CrewAI ≥0.80.0 accepts model strings directly
+        return self.model
     
     def create_plan(self, description: str, context: Dict[str, Any]) -> Dict[str, Any]:
         """