dlaima commited on
Commit
81faefe
·
verified ·
1 Parent(s): bd35e0b

Update retriever.py

Browse files
Files changed (1) hide show
  1. retriever.py +2 -1
retriever.py CHANGED
@@ -64,7 +64,8 @@ class GuestInfoRetrieverTool(Tool):
64
  results = self.retriever.get_relevant_documents(query)
65
  if results:
66
  responses = []
67
- for doc in results[:3]:
 
68
  content = doc.page_content
69
  starter = self._generate_conversation_starter(doc)
70
  responses.append(f"{content}\n\n{starter}")
 
64
  results = self.retriever.get_relevant_documents(query)
65
  if results:
66
  responses = []
67
+ #for doc in results[:3]:
68
+ for doc in results[:10]:
69
  content = doc.page_content
70
  starter = self._generate_conversation_starter(doc)
71
  responses.append(f"{content}\n\n{starter}")