Spaces:
Runtime error
Runtime error
ryanrwatkins
commited on
Commit
Β·
d7c5520
1
Parent(s):
36b5d2d
Update app.py
Browse files
app.py
CHANGED
@@ -111,8 +111,9 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
|
|
111 |
#query = str(system_prompt + history[-context_length*2:] + [prompt_msg])
|
112 |
#completion = completion.run(query)
|
113 |
# from https://blog.devgenius.io/chat-with-document-s-using-openai-chatgpt-api-and-text-embedding-6a0ce3dc8bc8
|
114 |
-
|
115 |
-
completion = RetrievalQA
|
|
|
116 |
query = str(system_prompt + history[-context_length*2:] + [prompt_msg])
|
117 |
completion = completion({"query": query})
|
118 |
logger.info(completion["result"])
|
|
|
111 |
#query = str(system_prompt + history[-context_length*2:] + [prompt_msg])
|
112 |
#completion = completion.run(query)
|
113 |
# from https://blog.devgenius.io/chat-with-document-s-using-openai-chatgpt-api-and-text-embedding-6a0ce3dc8bc8
|
114 |
+
completion_chain = load_qa_chain(ChatOpenAI(temperature=temperature, max_tokens=max_tokens, model_name="gpt-3.5-turbo"), chain_type="stuff" )
|
115 |
+
completion = RetrievalQA(combine_documents_chain=completion_chain, retriever=vectordb.as_retriever(), return_source_documents=True)
|
116 |
+
#completion = RetrievalQA.from_chain_type(llm=ChatOpenAI(temperature=temperature, max_tokens=max_tokens, model_name="gpt-3.5-turbo"), chain_type="stuff", retriever=vectordb.as_retriever(), return_source_documents=True)
|
117 |
query = str(system_prompt + history[-context_length*2:] + [prompt_msg])
|
118 |
completion = completion({"query": query})
|
119 |
logger.info(completion["result"])
|