MGC1991MF commited on
Commit
0a7299d
·
verified ·
1 Parent(s): 0d471e2

Update RAG_tool.py

Browse files
Files changed (1) hide show
  1. 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=1)
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."