Spaces:
Running
Running
Update RAG_tool.py
Browse files- RAG_tool.py +1 -1
RAG_tool.py
CHANGED
|
@@ -50,7 +50,7 @@ class BuscadorGuiasClinicas(BaseTool):
|
|
| 50 |
)
|
| 51 |
|
| 52 |
db = Chroma(persist_directory="./chroma_db", embedding_function=embedding_function)
|
| 53 |
-
results_raw = db.similarity_search(query, k=
|
| 54 |
|
| 55 |
if not results_raw:
|
| 56 |
return "No se encontró información relevante en las guías para esta consulta."
|
|
|
|
| 50 |
)
|
| 51 |
|
| 52 |
db = Chroma(persist_directory="./chroma_db", embedding_function=embedding_function)
|
| 53 |
+
results_raw = db.similarity_search(query, k=5)
|
| 54 |
|
| 55 |
if not results_raw:
|
| 56 |
return "No se encontró información relevante en las guías para esta consulta."
|