Spaces:
Sleeping
Sleeping
Update retriever.py
Browse files- retriever.py +2 -1
retriever.py
CHANGED
|
@@ -55,7 +55,8 @@ class GuestInfoRetrieverTool(Tool):
|
|
| 55 |
#])
|
| 56 |
|
| 57 |
def forward(self, query: str):
|
| 58 |
-
if
|
|
|
|
| 59 |
return "\n".join([doc.metadata.get("name", "Unknown") for doc in self.docs])
|
| 60 |
|
| 61 |
|
|
|
|
| 55 |
#])
|
| 56 |
|
| 57 |
def forward(self, query: str):
|
| 58 |
+
if any(keyword in query.lower() for keyword in ["list guests", "guest names", "list all guests", "show guests", "all guests", "everyone invited"]):
|
| 59 |
+
#if "list" in query.lower() and "guest" in query.lower() and "name" in query.lower():
|
| 60 |
return "\n".join([doc.metadata.get("name", "Unknown") for doc in self.docs])
|
| 61 |
|
| 62 |
|