nkcong206 commited on
Commit
ad44ee4
·
1 Parent(s): d9671d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -375,7 +375,8 @@ if st.session_state.gemini_api:
375
  respone = st.session_state.rag.invoke(prompt)
376
  st.write(respone)
377
  else:
378
- respone = st.session_state.llm.invoke(prompt)
 
379
  st.write(respone)
380
 
381
- st.session_state.chat_history.append({"role": "assistant", "content": respone})
 
375
  respone = st.session_state.rag.invoke(prompt)
376
  st.write(respone)
377
  else:
378
+ ans = st.session_state.llm.invoke(prompt)
379
+ respone = ans.content
380
  st.write(respone)
381
 
382
+ st.session_state.chat_history.append({"role": "assistant", "content": respone})