Spaces:
Paused
Paused
fix invoke 2 parameter error
Browse files
app.py
CHANGED
@@ -68,10 +68,13 @@ def get_response_from_model(prompt):
|
|
68 |
|
69 |
docs = st.session_state.knowledge_base.similarity_search(prompt)
|
70 |
print("docs:", docs) # 이까진 됐는데
|
71 |
-
# Chain의 invoke() 메소드 사용
|
72 |
-
response = st.session_state.qa_chain.invoke(
|
|
|
|
|
|
|
73 |
print("response:", response)
|
74 |
-
|
75 |
if "Helpful Answer:" in response:
|
76 |
response = response.split("Helpful Answer:")[1].strip()
|
77 |
|
|
|
68 |
|
69 |
docs = st.session_state.knowledge_base.similarity_search(prompt)
|
70 |
print("docs:", docs) # 이까진 됐는데
|
71 |
+
# Chain의 invoke() 메소드 사용 (input_documents로 전달)
|
72 |
+
response = st.session_state.qa_chain.invoke({
|
73 |
+
"input_documents": docs,
|
74 |
+
"question": prompt
|
75 |
+
})
|
76 |
print("response:", response)
|
77 |
+
|
78 |
if "Helpful Answer:" in response:
|
79 |
response = response.split("Helpful Answer:")[1].strip()
|
80 |
|