Update config.py

config.py

@@ -2,31 +2,23 @@ import os
 from dotenv import load_dotenv
 from typing import Dict, Any
 
-# Load environment variables from .env file
 load_dotenv()
 
-# Groq API Key (prefer loading from environment variables for security)
-GROQ_API_KEY = os.getenv("GROQ_API_KEY", "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")
+# Groq Configuration
+GROQ_MODEL = os.getenv("GROQ_MODEL", "deepseek-ai/deepseek-r1-distill-llama-70b")
 
-# AI Provider Configuration
-AI_PROVIDER = "groq"  # We're now using Groq exclusively
-AI_MODEL = os.getenv("AI_MODEL", "deepseek-ai/deepseek-r1-distill-llama-70b")  # Default model
-AI_API_KEY = GROQ_API_KEY  # Use the Groq API key
-
-# Model Settings (Groq-specific settings)
+# Model Settings
 MODEL_SETTINGS = {
-    "max_tokens": 4000,  # Maximum tokens for the response
-    "temperature": 0.7,  # Controls randomness (0.0 = deterministic, 1.0 = creative)
-    "top_p": 1.0,  # Controls diversity via nucleus sampling
-    "frequency_penalty": 0.0,  # Penalizes repeated tokens
-    "presence_penalty": 0.0,  # Penalizes new tokens
+    "max_tokens": 4000,
+    "temperature": 0.7,
+    "top_p": 1.0,
 }
 
 # Language Settings
 DEFAULT_LANGUAGE = 'ar'  # Arabic by default
-SUPPORTED_LANGUAGES = ['ar', 'en']  # Supported languages for the application
+SUPPORTED_LANGUAGES = ['ar', 'en']
 
-# UAE Legal Resources (Domains for legal research)
+# UAE Legal Resources
 UAE_LEGAL_DOMAINS = [
     'https://elaws.moj.gov.ae',
     'https://www.mohre.gov.ae',
@@ -35,7 +27,7 @@ UAE_LEGAL_DOMAINS = [
     'https://www.dc.gov.ae'
 ]
 
-# Legal Categories (Mapping of legal categories to their Arabic names)
+# Legal Categories
 LEGAL_CATEGORIES = {
     'civil': 'القانون المدني',
     'criminal': 'القانون الجنائي',
@@ -46,13 +38,16 @@ LEGAL_CATEGORIES = {
 }
 
 def get_ai_config() -> Dict[str, Any]:
-    """
-    Get AI configuration based on environment variables.
-    Returns a dictionary containing the provider, model, API key, and model settings.
-    """
+    """Get Groq configuration based on environment variables"""
+    api_key = os.getenv("GROQ_API_KEY")
+    if not api_key:
+        raise ValueError("GROQ_API_KEY must be set in environment variables")
+    
     return {
-        "provider": AI_PROVIDER,
-        "model": AI_MODEL,
-        "api_key": AI_API_KEY,
-        **MODEL_SETTINGS  # Include all model settings in the config
+        "model": GROQ_MODEL,
+        "api_key": api_key,
+        "temperature": MODEL_SETTINGS["temperature"],
+        "max_tokens": MODEL_SETTINGS["max_tokens"],
+        "top_p": MODEL_SETTINGS["top_p"],
+        "llm_type": "groq"
     }