Update agents.py

agents.py

@@ -4,25 +4,17 @@ from crewai import Agent
 from langchain.tools import Tool
 from utils import create_uae_legal_tools
 from config import get_ai_config
-from ai_providers import AIProviderFactory
-from crewai.llms import Groq  # Add this import
+from ai_providers import AIProviderFactory, GroqProvider  # Import GroqProvider here
 
 def get_base_llm_config():
     """Get LLM configuration based on current settings"""
     ai_config = get_ai_config()
-    
-    # Create Groq LLM instance
-    llm = Groq(
-        groq_api_key=ai_config["api_key"],
-        model_name=ai_config["model"]
+    provider = AIProviderFactory.create_provider(
+        ai_config["provider"],
+        api_key=ai_config["api_key"],
+        model=ai_config["model"]
     )
-    
-    # Return configuration with explicit LLM instance
-    return {
-        "llm": llm,
-        "temperature": ai_config.get("temperature", 0.7),
-        "max_tokens": ai_config.get("max_tokens", 4000)
-    }
+    return provider.get_config()
 
 def create_judge_agent():
     return Agent(