diego2317 commited on
Commit
03f443f
·
verified ·
1 Parent(s): feb4cfa

Update utils/response_manager.py

Browse files
Files changed (1) hide show
  1. utils/response_manager.py +2 -2
utils/response_manager.py CHANGED
@@ -29,12 +29,12 @@ class ResponseManager:
29
  :param max_num_results: The maximum number of search results to return (default: 15).
30
  """
31
  # Load vector_store_id and api_key from environment variables if not provided
32
- self.vector_store_id = vector_store_id or os.getenv('VECTOR_STORE_ID')
33
  if not self.vector_store_id:
34
  logging.error("VECTOR_STORE_ID is not provided or set in the environment.")
35
  raise ValueError("VECTOR_STORE_ID is required.")
36
 
37
- self.api_key = api_key or os.getenv('OPENAI_API_KEY')
38
  if not self.api_key:
39
  logging.error("OPENAI_API_KEY is not provided or set in the environment.")
40
  raise ValueError("OPENAI_API_KEY is required.")
 
29
  :param max_num_results: The maximum number of search results to return (default: 15).
30
  """
31
  # Load vector_store_id and api_key from environment variables if not provided
32
+ self.vector_store_id = vector_store_id or os.getenv('USS_DOCUMENTATION_VECTOR_STORE')
33
  if not self.vector_store_id:
34
  logging.error("VECTOR_STORE_ID is not provided or set in the environment.")
35
  raise ValueError("VECTOR_STORE_ID is required.")
36
 
37
+ self.api_key = api_key or os.getenv('SETTING_UP_API_KEY')
38
  if not self.api_key:
39
  logging.error("OPENAI_API_KEY is not provided or set in the environment.")
40
  raise ValueError("OPENAI_API_KEY is required.")