laverdes commited on
Commit
ea7eea9
·
verified ·
1 Parent(s): 8507438

chore: retrieving only best match

Browse files

I think it makes more sense to retrieve the information from one guest only per query: The best matching one. WDYT :)

Files changed (1) hide show
  1. retriever.py +1 -1
retriever.py CHANGED
@@ -23,7 +23,7 @@ class GuestInfoRetrieverTool(Tool):
23
  def forward(self, query: str):
24
  results = self.retriever.get_relevant_documents(query)
25
  if results:
26
- return "\n\n".join([doc.page_content for doc in results[:3]])
27
  else:
28
  return "No matching guest information found."
29
 
 
23
  def forward(self, query: str):
24
  results = self.retriever.get_relevant_documents(query)
25
  if results:
26
+ return results[0].page_content
27
  else:
28
  return "No matching guest information found."
29