Spaces:
Sleeping
Sleeping
cdupland
commited on
Commit
·
ae06536
1
Parent(s):
22c6fdf
CHange k parameter ford common namespace retiever
Browse files
rag.py
CHANGED
@@ -131,7 +131,7 @@ def get_retreive_answer(enterprise_id,prompt,index,common_id):
|
|
131 |
|
132 |
retriever_commun = vector_store_commun.as_retriever(
|
133 |
search_type="similarity_score_threshold",
|
134 |
-
search_kwargs={"k":
|
135 |
)
|
136 |
|
137 |
commun_context = retriever_commun.invoke(prompt)
|
|
|
131 |
|
132 |
retriever_commun = vector_store_commun.as_retriever(
|
133 |
search_type="similarity_score_threshold",
|
134 |
+
search_kwargs={"k": 5, "score_threshold": 0.1},
|
135 |
)
|
136 |
|
137 |
commun_context = retriever_commun.invoke(prompt)
|