Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|