halyn commited on
Commit
9506c20
·
1 Parent(s): b1960ab

fix invoke 2 parameter error

Browse files
Files changed (1) hide show
  1. app.py +6 -3
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(input_documents=docs, question=prompt)
 
 
 
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