ankush13r commited on
Commit
963adbe
·
verified ·
1 Parent(s): a4b79b1

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +11 -2
tools.py CHANGED
@@ -145,8 +145,17 @@ class get_documents(ToolBase):
145
 
146
  # return "We are currently working on it. You can't use this tool right now—please try again later. Thank you for your patience!"
147
  return vector_store.get_context(query)
148
-
149
- # @tool_register
 
 
 
 
 
 
 
 
 
150
  class hotel_facilities(ToolBase):
151
  """Provides a list of available general facilities at the hotel, which could include amenities like a spa, pool, gym, conference rooms, etc."""
152
 
 
145
 
146
  # return "We are currently working on it. You can't use this tool right now—please try again later. Thank you for your patience!"
147
  return vector_store.get_context(query)
148
+
149
+
150
+ @tool_register
151
+ class packs(ToolBase):
152
+ """Provides a list of available activity pack at the hotel."""
153
+
154
+ @classmethod
155
+ def invoke(cls, input: Dict) -> str:
156
+ return json_data["packs"]
157
+
158
+ @tool_register
159
  class hotel_facilities(ToolBase):
160
  """Provides a list of available general facilities at the hotel, which could include amenities like a spa, pool, gym, conference rooms, etc."""
161