dlaima commited on
Commit
36f7f47
·
verified ·
1 Parent(s): e1847da

Update retriever.py

Browse files
Files changed (1) hide show
  1. retriever.py +2 -1
retriever.py CHANGED
@@ -55,7 +55,8 @@ class GuestInfoRetrieverTool(Tool):
55
  #])
56
 
57
  def forward(self, query: str):
58
- if "list" in query.lower() and "guest" in query.lower() and "name" in query.lower():
 
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