Update agents.py

agents.py

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