ArturG9 commited on
Commit
ec375fc
1 Parent(s): 4bf5fc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -98,7 +98,7 @@ def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='m
98
  [
99
  ("system", contextualize_q_system_prompt),
100
  MessagesPlaceholder("chat_history"),
101
- ("human", "{question}"),
102
  ]
103
  )
104
 
@@ -147,7 +147,7 @@ def main():
147
  def handle_userinput(user_question,retriever):
148
  st.session_state.messages.append({"role": "user", "content": user_question})
149
  st.chat_message("user").write(user_question)
150
- docs = retriever.invoke({"question": user_question, "chat_history": st.session_state.messages})
151
 
152
  with st.sidebar:
153
  st.subheader("Your documents")
 
98
  [
99
  ("system", contextualize_q_system_prompt),
100
  MessagesPlaceholder("chat_history"),
101
+ ("human", "{input}"),
102
  ]
103
  )
104
 
 
147
  def handle_userinput(user_question,retriever):
148
  st.session_state.messages.append({"role": "user", "content": user_question})
149
  st.chat_message("user").write(user_question)
150
+ docs = retriever.invoke({"input": user_question, "chat_history": st.session_state.messages})
151
 
152
  with st.sidebar:
153
  st.subheader("Your documents")