Tarun-intellentech commited on
Commit
9831972
·
verified ·
1 Parent(s): 0eba7fb

Update ver3_streamlit.py

Browse files
Files changed (1) hide show
  1. ver3_streamlit.py +1 -1
ver3_streamlit.py CHANGED
@@ -211,7 +211,7 @@ if "summary" in st.session_state:
211
  try:
212
  intent = intent_llm.invoke(f"Candidates: {st.session_state.name_list}\nQuestion: {prompt}")
213
  filt = {"candidate_name": {"$in": intent.specific_candidates}} if intent.specific_candidates else None
214
- docs = vs.similarity_search(prompt, k=5, filter=filt)
215
  ctx = "\n\n".join([d.page_content for d in docs])
216
 
217
  chat_llm = ChatMistralAI(model="mistral-large-latest", api_key=api_key)
 
211
  try:
212
  intent = intent_llm.invoke(f"Candidates: {st.session_state.name_list}\nQuestion: {prompt}")
213
  filt = {"candidate_name": {"$in": intent.specific_candidates}} if intent.specific_candidates else None
214
+ docs = vs.similarity_search(prompt, k=20, filter=filt)
215
  ctx = "\n\n".join([d.page_content for d in docs])
216
 
217
  chat_llm = ChatMistralAI(model="mistral-large-latest", api_key=api_key)