Spaces:
Sleeping
Sleeping
Update ver3_streamlit.py
Browse files- 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=
|
| 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)
|