ryanrwatkins commited on
Commit
d7c5520
1 Parent(s): 36b5d2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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.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)
 
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"])