Spaces:
Sleeping
Sleeping
Update retriever.py
Browse files- 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.
|
26 |
if results:
|
27 |
-
return "\n\n" .
|
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 |
|