chore: retrieving only best match

#8
by laverdes - opened
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