Update app.py
Browse files
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({"
|
| 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']
|