Update app.py

app.py

@@ -6,6 +6,7 @@ import os
 import langchain
 import chromadb
 import glob
+import logging 
 
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import Chroma
@@ -25,6 +26,7 @@ from langchain.chains.question_answering import load_qa_chain
 #    persist_directory="./embeddings" # Optional, defaults to .chromadb/ in the current directory
 #))
 
+logger=logging.getLogger() 
 
 def get_empty_state():
     return {"total_tokens": 0, "messages": []}
@@ -113,6 +115,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         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)
         query = str(system_prompt + history[-context_length*2:] +  [prompt_msg])
         completion = completion({"query": query})
+        logger.info(completion["result"]) 
 #        completion = completion({"question": query, "chat_history": history[-context_length*2:]})
 
 
@@ -121,8 +124,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         
         history.append(prompt_msg)
         #history.append(completion.choices[0].message.to_dict())
-        #history.append(completion["result"].choices[0].message.to_dict())
-        history.append("test")
+        history.append(completion["result"].choices[0].message.to_dict())
 
         state['total_tokens'] += completion['usage']['total_tokens']