Seth0330 commited on
Commit
ac87b70
·
verified ·
1 Parent(s): e604e69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -164,7 +164,7 @@ qa_chain = RetrievalQA.from_chain_type(
164
  llm=llm,
165
  retriever=retriever,
166
  return_source_documents=True,
167
- chain_type_kwargs={"input_key": "query"}
168
  )
169
 
170
  # ---- Ingestion UI ----
@@ -211,7 +211,7 @@ def send_message():
211
  q = user_input
212
  st.session_state.messages.append({"role": "user", "content": user_input})
213
  with st.spinner("Thinking..."):
214
- result = qa_chain({"query": q})
215
  answer = result['result']
216
  st.session_state.messages.append({"role": "assistant", "content": answer})
217
  docs = result['source_documents']
 
164
  llm=llm,
165
  retriever=retriever,
166
  return_source_documents=True,
167
+ # chain_type_kwargs={"input_key": "query"} # <--- REMOVED
168
  )
169
 
170
  # ---- Ingestion UI ----
 
211
  q = user_input
212
  st.session_state.messages.append({"role": "user", "content": user_input})
213
  with st.spinner("Thinking..."):
214
+ result = qa_chain({"question": q}) # <---- FIXED: use 'question' as input key
215
  answer = result['result']
216
  st.session_state.messages.append({"role": "assistant", "content": answer})
217
  docs = result['source_documents']