Reverting to version before hyperparameter changes

app.py

@@ -14,27 +14,26 @@ from langchain.vectorstores import Chroma
 openai_api_key = os.getenv("openai_token")
 embedding = OpenAIEmbeddings(openai_api_key=openai_api_key)
 
-@st.cache_resource
-def get_vectordb():
-    embedding = OpenAIEmbeddings(openai_api_key=os.getenv("openai_token"))
-    return Chroma(persist_directory="./chroma_db", embedding_function=embedding)
-
-vectordb = get_vectordb()
-
-# # Setup vector database
-# persist_directory = './chroma_db'
-# vectordb = Chroma(persist_directory=persist_directory, embedding_function=embedding)
+# Setup vector database
+persist_directory = './chroma_db'
+vectordb = Chroma(persist_directory=persist_directory, embedding_function=embedding)
 
 llm_name = "gpt-3.5-turbo"
 
-llm = ChatOpenAI(model_name=llm_name, temperature=0.7,
+llm = ChatOpenAI(model_name=llm_name, temperature=0,
                  openai_api_key=openai_api_key)
 
 qa_chain = RetrievalQA.from_chain_type(
     llm,
-    retriever=vectordb.as_retriever(search_kwargs={"k": 5})
+    retriever=vectordb.as_retriever()
 )
 
+question = "production is broken how do I fix it?"
+
+result = qa_chain({"query": question})
+
+print(result['result'])
+
 
 # Streamed response emulator
 def response_generator(prompt):
@@ -45,7 +44,7 @@ def response_generator(prompt):
         time.sleep(0.05)
 
 
-st.title("Technical Support Chatbot")
+st.title("Simple chat")
 
 # Initialize chat history
 if "messages" not in st.session_state:
@@ -57,7 +56,7 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 # Accept user input
-if prompt := st.chat_input("Enter your question here"):
+if prompt := st.chat_input("What is up?"):
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
     # Display user message in chat message container
@@ -69,4 +68,3 @@ if prompt := st.chat_input("Enter your question here"):
         response = st.write_stream(response_generator(prompt))
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
-