Update agents.py

agents.py

@@ -1,22 +1,32 @@
-# agents.py
-
 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, GroqProvider  # Import GroqProvider here
+from typing import Dict, Any
 
-def get_base_llm_config():
+def get_base_llm_config() -> Dict[str, Any]:
     """Get LLM configuration based on current settings"""
     ai_config = get_ai_config()
-    provider = AIProviderFactory.create_provider(
-        ai_config["provider"],
-        api_key=ai_config["api_key"],
-        model=ai_config["model"]
-    )
-    return provider.get_config()
+    
+    return {
+        "config_list": [{
+            "model": ai_config["model"],
+            "api_key": ai_config["api_key"]
+        }],
+        "temperature": ai_config["temperature"],
+        "max_tokens": ai_config["max_tokens"],
+        "request_timeout": 600,
+        "seed": 42,
+        "config": {
+            "use_lite_llm": True,
+            "lite_llm_model_name": ai_config["model"],
+            "api_key": ai_config["api_key"]
+        },
+        "llm_type": "groq"
+    }
 
 def create_judge_agent():
+    llm_config = get_base_llm_config()
     return Agent(
         role='قاضي قانوني إماراتي',
         goal='تقديم أحكام وتفسيرات قانونية دقيقة بناءً على القانون الإماراتي',
@@ -28,11 +38,12 @@ def create_judge_agent():
         """,
         verbose=True,
         allow_delegation=False,
-        llm_config=get_base_llm_config(),
+        llm_config=llm_config,
         tools=create_uae_legal_tools()
     )
 
 def create_advocate_agent():
+    llm_config = get_base_llm_config()
     return Agent(
         role='محامي إماراتي',
         goal='تقديم التمثيل القانوني والمشورة المتخصصة بناءً على القانون الإماراتي',
@@ -43,11 +54,12 @@ def create_advocate_agent():
         """,
         verbose=True,
         allow_delegation=False,
-        llm_config=get_base_llm_config(),
+        llm_config=llm_config,
         tools=create_uae_legal_tools()
     )
 
 def create_consultant_agent():
+    llm_config = get_base_llm_config()
     return Agent(
         role='مستشار قضائي إماراتي',
         goal='تقديم الاستشارات والتوجيه القانوني المتخصص في القانون الإماراتي',
@@ -58,6 +70,6 @@ def create_consultant_agent():
         """,
         verbose=True,
         allow_delegation=False,
-        llm_config=get_base_llm_config(),
+        llm_config=llm_config,
         tools=create_uae_legal_tools()
     )