Update helpers/ai_client.py

helpers/ai_client.py

@@ -10,17 +10,8 @@ logger = logging.getLogger(__name__)
 
 class AIClient:
     def __init__(self):
-        # Load environment variables
-        self.base_url = os.getenv("LLM_API_URL", "https://openrouter.ai/api/v1")
-        self.api_key = os.getenv("OPENROUTER_API_KEY")
-        self.site_url = os.getenv("SITE_URL", "")  # Your site URL
-        self.site_name = os.getenv("SITE_NAME", "")  # Your site name
         # Initialize OpenAI client
-        self.client = OpenAI(
-            base_url=self.base_url,
-            api_key=self.api_key
-        )
-
+        self.client = OpenAI(api_key=os.getenv("OPENROUTER_API_KEY"), base_url="https://openrouter.ai/api/v1")
     def chat(
         self,
         prompt: str,
@@ -29,20 +20,6 @@ class AIClient:
         conversation_id: str = "",
         user_id: str = "string"
     ) -> str:
-        """
-        Sends a prompt to the OpenRouter API and returns the response as text.
-        Args:
-            prompt (str): The user's input prompt.
-            system_message (str): Optional system message for the LLM.
-            model_id (str): The model ID to use (default: "openai/gpt-4o-mini").
-            conversation_id (str): Unique ID for the conversation.
-            user_id (str): Unique ID for the user.
-            api_key (str): API key for authentication.
-        Returns:
-            str: The text response from the LLM API.
-        Raises:
-            Exception: If the API request fails.
-        """
         try:
             messages = []
             
@@ -55,10 +32,6 @@ class AIClient:
 
             # Create completion
             completion = self.client.chat.completions.create(
-                extra_headers={
-                    "HTTP-Referer": self.site_url,  # Optional
-                    "X-Title": self.site_name,      # Optional
-                },
                 model=model_id,
                 messages=messages
             )