totallysaber commited on
Commit
4da3ef0
·
verified ·
1 Parent(s): 1dac261

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def run_query(query, chat_history, k, temperature,openai_api_key):
11
  load_dotenv()
12
  persist_directory = 'db'
13
 
14
- db = Chroma(persist_directory=persist_directory, embedding_function=OpenAIEmbeddings()) # access db
15
  retriever = db.as_retriever(search_kwargs={"k": k}) # kwargs determines how many docs it uses
16
 
17
  llm = OpenAI(api_key=openai_api_key, max_tokens=1500, temperature=temperature) # api key self explanatory. max_tokens provides how long of a response
 
11
  load_dotenv()
12
  persist_directory = 'db'
13
 
14
+ db = Chroma(persist_directory=persist_directory, embedding_function=OpenAIEmbeddings(openai_api_key=openai_api_key)) # access db
15
  retriever = db.as_retriever(search_kwargs={"k": k}) # kwargs determines how many docs it uses
16
 
17
  llm = OpenAI(api_key=openai_api_key, max_tokens=1500, temperature=temperature) # api key self explanatory. max_tokens provides how long of a response