Update app.py

app.py

@@ -10,7 +10,7 @@ import glob
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import Chroma
 from langchain.text_splitter import TokenTextSplitter
-from langchain.llms import OpenAI
+#from langchain.llms import OpenAI
 from langchain.chat_models import ChatOpenAI
 #from langchain.chains import ChatVectorDBChain
 from langchain.chains import RetrievalQA
@@ -103,7 +103,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
 
 # completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=system_prompt + history[-context_length*2:] + [prompt_msg], temperature=temperature, max_tokens=max_tokens)
 
-        completion_chain = load_qa_chain(OpenAI(temperature=temperature, max_tokens=max_tokens, model_name="gpt-3.5-turbo"), chain_type="stuff" )
+        completion_chain = load_qa_chain(ChatOpenAI(temperature=temperature, max_tokens=max_tokens, model_name="gpt-3.5-turbo"), chain_type="stuff" )
         completion = RetrievalQA(combine_documents_chain=completion_chain, retriever=vectordb.as_retriever())
         query = str(system_prompt + history[-context_length*2:] +  [prompt_msg])
         completion = completion.run(query)