ahuang11 sophiamyang commited on
Commit
44f03d1
·
1 Parent(s): 6189a5e

remove openai_api_key from functions (#1)

Browse files

- remove openai_api_key from functions (7219b3e3da6ff8de2e27f5e9a4a21c12a4035ae9)


Co-authored-by: Sophia Yang <[email protected]>

Files changed (1) hide show
  1. utils/ai.py +1 -2
utils/ai.py CHANGED
@@ -17,7 +17,6 @@ def retrieve(query, k=10, filters=None):
17
 
18
  Args:
19
  query (str): User query to pass in
20
- openai_api_key (str): openai api key. If not passed in, uses environment variable
21
  k (int, optional): number of results passed back. Defaults to 10.
22
  filters (dict, optional): Filters to apply to the query. You can filter based off
23
  any piece of metadata by passing in a dict of the format {metadata_name: filter_value}
@@ -40,7 +39,7 @@ def retrieve(query, k=10, filters=None):
40
  return requests.post(url, json=params, timeout=120).json()
41
 
42
 
43
- def retrieve_context(query, openai_api_key, k=10, filters=None):
44
  """Gets the context from our libraries vector db for a given query.
45
 
46
  Args:
 
17
 
18
  Args:
19
  query (str): User query to pass in
 
20
  k (int, optional): number of results passed back. Defaults to 10.
21
  filters (dict, optional): Filters to apply to the query. You can filter based off
22
  any piece of metadata by passing in a dict of the format {metadata_name: filter_value}
 
39
  return requests.post(url, json=params, timeout=120).json()
40
 
41
 
42
+ def retrieve_context(query, k=10, filters=None):
43
  """Gets the context from our libraries vector db for a given query.
44
 
45
  Args: