Update utils/response_manager.py

utils/response_manager.py

@@ -3,43 +3,29 @@ import openai
 import logging
 from typing import Optional
 
-
 class ResponseManager:
     """
     This class initializes the OpenAI client and provides methods to create responses, 
     maintain conversation history, and handle user queries.
-    Attributes:
-        DEFAULT_META_PROMPT_FILE (str): Default path to the meta prompt file.
-        DEFAULT_MODEL (str): Default OpenAI model to use.
-        DEFAULT_TEMPERATURE (float): Default temperature for response generation.
-        DEFAULT_MAX_OUTPUT_TOKENS (int): Default maximum number of output tokens.
-        DEFAULT_MAX_NUM_RESULTS (int): Default maximum number of search results.
-    Methods:
-        __init__(vector_store_id: Optional[str], api_key: Optional[str], meta_prompt_file: Optional[str]):
-            Initializes the ResponseManager with a vector store ID, API key, and meta prompt file.
-        _load_meta_prompt(meta_prompt_file: str) -> str:
-            Loads the meta prompt from the specified file.
-        create_response(query: str, model: Optional[str], temperature: Optional[float], 
-                        max_output_tokens: Optional[int], max_num_results: Optional[int]) -> str:
-            Creates a response to a user query using the OpenAI API.
-        conversation(query: str, model: Optional[str], temperature: Optional[float], 
-                     max_output_tokens: Optional[int], max_num_results: Optional[int], 
-            Handles chatbot interaction and maintains conversation history.
     """
 
-    DEFAULT_META_PROMPT_FILE = 'config/meta_prompt.txt'
-    DEFAULT_MODEL = "gpt-4o-mini"
-    DEFAULT_TEMPERATURE = 0
-    DEFAULT_MAX_OUTPUT_TOKENS = 800
-    DEFAULT_MAX_NUM_RESULTS = 15
-
- 
-    def __init__(self, vector_store_id: Optional[str] = None, api_key: Optional[str] = None, meta_prompt_file: Optional[str] = None):
+    def __init__(self, 
+                 vector_store_id: Optional[str] = None, 
+                 api_key: Optional[str] = None, 
+                 meta_prompt_file: Optional[str] = None,
+                 model: str = "gpt-4o-mini",
+                 temperature: float = 0,
+                 max_output_tokens: int = 800,
+                 max_num_results: int = 15):
         """
-        Initialize the ResponseManager with a vector store ID, API key, and meta prompt file.
+        Initialize the ResponseManager with optional parameters for configuration.
         :param vector_store_id: The ID of the vector store to use for file search.
         :param api_key: The OpenAI API key for authentication.
         :param meta_prompt_file: Path to the meta prompt file (default: 'config/meta_prompt.txt').
+        :param model: The OpenAI model to use (default: 'gpt-4o-mini').
+        :param temperature: The temperature for response generation (default: 0).
+        :param max_output_tokens: The maximum number of output tokens (default: 800).
+        :param max_num_results: The maximum number of search results to return (default: 15).
         """
         # Load vector_store_id and api_key from environment variables if not provided
         self.vector_store_id = vector_store_id or os.getenv('VECTOR_STORE_ID')
@@ -62,6 +48,12 @@ class ResponseManager:
         # Load the meta prompt from the specified file
         self.meta_prompt = self._load_meta_prompt(self.meta_prompt_file)
 
+        # Set default parameters for response generation
+        self.model = model
+        self.temperature = temperature
+        self.max_output_tokens = max_output_tokens
+        self.max_num_results = max_num_results
+
     def _load_meta_prompt(self, meta_prompt_file: str) -> str:
         """
         Load the meta prompt from the specified file.
@@ -101,16 +93,16 @@ class ResponseManager:
         try:
             logging.info("Sending request to OpenAI API...")
             response = self.client.responses.create(
-                model=self.DEFAULT_MODEL,
+                model=self.model,
                 previous_response_id=self.previous_response_id,
                 input=input_data,
                 tools=[{
                     "type": "file_search",
                     "vector_store_ids": [self.vector_store_id],
-                    "max_num_results": self.DEFAULT_MAX_NUM_RESULTS
+                    "max_num_results": self.max_num_results
                 }],
-                temperature=self.DEFAULT_TEMPERATURE,
-                max_output_tokens=self.DEFAULT_MAX_OUTPUT_TOKENS
+                temperature=self.temperature,
+                max_output_tokens=self.max_output_tokens
             )
             self.previous_response_id = response.id
             logging.info("Response received successfully.")
@@ -121,6 +113,4 @@ class ResponseManager:
             logging.error(f"An error occurred while generating a response: {e}")
             error_message = "Sorry, I couldn't generate a response at this time. Please try again later."
             input_data.append({"role": "assistant", "content": error_message})
-            return history + input_data
-
-       
+            return history + input_data