Update tools.py

tools.py

@@ -89,13 +89,9 @@ class get_hotel_information(ToolBase):
     @classmethod
     def invoke(cls, input: Dict) -> str: 
         return """### **Nou Vall de Núria – Brief Description**  
-
 Nestled in the stunning **Vall de Núria** in the Pyrenees, **Nou Vall de Núria** offers a perfect blend of comfort and adventure. Guests can enjoy breathtaking mountain views, premium accommodations, and excellent facilities, including an outdoor pool, gym, and sauna.  
-
 The hotel features **two dining options**, serving traditional Catalan cuisine and refreshing drinks. Accommodations range from **cozy standard rooms to luxurious suites and fully equipped apartments**, catering to couples, families, and groups.  
-
 For an unforgettable stay, guests can choose from **special packages**, including family-friendly stays, romantic getaways, ski adventures, and relaxation retreats. Outdoor enthusiasts can explore **hiking trails, ski slopes, and fishing spots** in the surrounding natural beauty.  
-
 Whether for relaxation or adventure, **Nou Vall de Núria** promises a unique and memorable experience.""" 
 
 @tool_register
@@ -167,8 +163,23 @@ class check_room_availability(ToolBase):
         instance = cls(**input)
         room_type = instance.room_type
         check_in_date = instance.check_in_date
-        check_in_date = instance.check_in_date
+        check_out_date = instance.check_out_date
         guests = instance.guests
+
+        missing = []
+        if not room_type:
+            missing.append("room_type")
+        if not check_in_date:
+            missing.append("check_in_date")
+        if not check_out_date:
+            missing.append("check_out_date")
+        if not guests:
+            missing.append("guests")
+
+        if len(missing):
+            value = ", ".join(missing)
+            return f"Unable to check the room availability. The following required arguments are missing:{value}." 
+        
         rooms = [room for room in json_data["accomodations"] if room_type in room["type"]]
         if len(rooms) == 0:
             return f"There is no room exists with room type {room_type}"
@@ -197,10 +208,25 @@ class make_reservation(ToolBase):
         instance = cls(**input)
         room_type = instance.room_type
         check_in_date = instance.check_in_date
-        check_in_date = instance.check_in_date
+        check_out_date = instance.check_out_date
         guests = instance.guests
         user_id = instance.user_id
-        
+
+        missing = []
+        if not room_type:
+            missing.append("room_type")
+        if not check_in_date:
+            missing.append("check_in_date")
+        if not check_out_date:
+            missing.append("check_out_date")
+        if not guests:
+            missing.append("guests")
+        if not user_id:
+            missing.append("user_id")
+
+        if len(missing):
+            value = ", ".join(missing)
+            return f"Unable to complete the reservation. The following required arguments are missing:{value}."     
 
         rooms = [room for room in json_data["accomodations"] if room_type in room["type"]]
         if len(rooms) == 0:
@@ -222,7 +248,7 @@ class make_reservation(ToolBase):
             "room_number": room["room_number"],
             "room_type": room_type,
             "check_in_date": check_in_date,
-            "check_in_date": check_in_date,
+            "check_out_date": check_out_date,
             "guests": guests,
             "reservation_id": rand,
             "user_id": user_id,
@@ -249,8 +275,18 @@ class cancel_reservation(ToolBase):
         reservation_id = instance.reservation_id
         user_id = instance.user_id
 
+        missing = []
+        if not reservation_id:
+            missing.append("reservation_id")
+        if not user_id:
+            missing.append("user_id")
+
+        if len(missing):
+            value = ", ".join(missing)
+            return f"Unable to cancel the reservation. The following required arguments are missing:{value}."     
+
         if reservation_id not in reservations:
-            return f"The is no reservations with the id: {reservation_id}"
+            return f"There is no reservations with the id: {reservation_id}"
         
         if reservations["reservation_id"]["user_id"] != user_id:
             return "The user id is wrong, please provide same user id that was used make to reservation."
@@ -283,8 +319,22 @@ class modify_reservation(ToolBase):
         user_id = instance.user_id
         reservation_id = instance.reservation_id
 
+        missing = []
+        if not reservation_id:
+            missing.append("reservation_id")
+        if not user_id:
+            missing.append("user_id")
+
+        if len(missing):
+            value = ", ".join(missing)
+            return f"Unable to modify the reservation. The following required arguments are missing:{value}."     
+
+        if not (new_room_type or new_check_in_date or new_check_out_date or guests):
+            return "Unable to modify the reservation. One of the following arguments must be passed: new_room_type, new_check_in_date, new_check_out_date, guests."     
+
+
         if reservation_id not in reservations:
-            return f"The is no reservations with the id: {reservation_id}"
+            return f"There is no reservations with the id: {reservation_id}"
         
         if reservations["reservation_id"]["user_id"] != user_id:
             return "The user id is wrong, please provide same user id that was used make to reservation."
@@ -330,12 +380,22 @@ class get_reservation_details(ToolBase):
         user_id = instance.user_id
         reservation_id = instance.reservation_id
 
+        missing = []
+        if not reservation_id:
+            missing.append("reservation_id")
+        if not user_id:
+            missing.append("user_id")
+
+        if len(missing):
+            value = ", ".join(missing)
+            return f"Unable to get the details. The following required arguments are missing:{value}."     
+
         if reservation_id not in reservations:
-            return f"The is no reservations with the id: {reservation_id}"
+            return f"There is no reservations with the id: {reservation_id}"
         
         if reservations["reservation_id"]["user_id"] != user_id:
             return "The user id is wrong, please provide same user id that was used make to reservation."
         
         tmp_data = copy.deepcopy(reservations["reservation_id"])
         tmp_data.pop("user_id")
-        return json.dumps(tmp_data)
+        return json.dumps(tmp_data)