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)
|