CindyDelage commited on
Commit
347faa8
·
verified ·
1 Parent(s): c84f703

Update retriever.py

Browse files
Files changed (1) hide show
  1. retriever.py +2 -3
retriever.py CHANGED
@@ -18,13 +18,12 @@ class GuestInfoRetrieverTool(Tool):
18
  def __init__(self, docs):
19
  self.is_initialized = False
20
  self.retriever = BM25Retriever.from_documents(docs)
21
- self.subject = BM25Retriever.to_json(docs)
22
 
23
 
24
  def forward(self, query: str):
25
- results = self.subject.get_relevant_documents(query)
26
  if results:
27
- return "\n\n" .jion([results]) #.join([doc.page_content for doc in results[:3]])
28
  else:
29
  return "No matching guest information found."
30
 
 
18
  def __init__(self, docs):
19
  self.is_initialized = False
20
  self.retriever = BM25Retriever.from_documents(docs)
 
21
 
22
 
23
  def forward(self, query: str):
24
+ results = self.retriever.invoke("Description")#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