fix bug tools

tools.py

@@ -131,7 +131,7 @@ Whether for relaxation or adventure, **Nou Vall de Núria** promises a unique an
 
 
 @tool_register
-class get_documents(ToolBase):
+class get_hotal_general_information(ToolBase):
     """
     Retrieves general information about a region, its cities, activities, tourism, or surrounding areas based on query.
     """
@@ -220,7 +220,7 @@ class check_room_availability(ToolBase):
     
     @classmethod
     def invoke(cls, input: Dict) -> str: 
-        
+        input["room_type"] = input.get("room_type", None)
         room_type = input.get("room_type", None)
         reservation_start_date = input.get("reservation_start_date", None)
         reservation_end_date = input.get("reservation_end_date", None)
@@ -274,7 +274,7 @@ class make_reservation(ToolBase):
     
     @classmethod
     def invoke(cls, input: Dict) -> str:
-        
+        input["room_type"] = input.get("room_type", None)
         room_type = input.get("room_type", None)
         reservation_start_date = input.get("reservation_start_date", None)
         reservation_end_date = input.get("reservation_end_date", None)
@@ -369,18 +369,18 @@ class cancel_reservation(ToolBase):
 class modify_reservation(ToolBase):
     """
     Allows a user to modify an existing reservation by updating the check-in/check-out dates or changing the room type, subject to availability.
+    One of these parameter is mandatory: new_room_type, new_reservation_start_date, new_reservation_end_date, guests.
     """
 
-
+    reservation_id: int = Field(description="The unique identifier of the reservation to be modified.")
     new_room_type: str | None = Field(default=list(json_data["room_types"].keys()), description=f"The type of new room to be modified, if {None} same room will be modified.")
     new_reservation_start_date: str = Field(default=None, description="New check out date in format DD/MM/YYYY")
     new_reservation_end_date: str = Field(default=None, description="New check out date in format DD/MM/YYYY")
-    guests: int = Field(default=None, description="New number of guests for the reservation.")
-    reservation_id: int = Field(description="The unique identifier of the reservation to be modified.")
+    new_guests: int = Field(default=None, description="New number of guests for the reservation.")
 
     @classmethod
     def invoke(cls, input: Dict) -> str: 
-        
+        input["new_room_type"] = input.get("new_room_type", None)
         reservation_id = input.get("reservation_id", None)
 
         missing = []
@@ -391,26 +391,26 @@ class modify_reservation(ToolBase):
         new_room_type = instance.new_room_type
         new_reservation_start_date = instance.new_reservation_start_date
         new_reservation_end_date = instance.new_reservation_end_date
-        guests = instance.guests
+        new_guests = instance.new_guests
         reservation_id = instance.reservation_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_reservation_start_date or new_reservation_end_date or guests):
-            return "Unable to modify the reservation. One of the following arguments must be passed: new_room_type, new_reservation_start_date, new_reservation_end_date, guests."     
+    
+        if not (new_room_type or new_reservation_start_date or new_reservation_end_date or new_guests):
+            return "Unable to modify the reservation. One of the following arguments must be passed: new_room_type, new_reservation_start_date, new_reservation_end_date, new_guests."     
 
 
         if reservation_id not in reservations:
             return f"There is no reservations with the id: {reservation_id}"
         
-        if new_room_type or guests:
-            rooms = [room for room in json_data["room_types"] if new_room_type in room["type"]]
+        if new_room_type or new_guests:
+            rooms = [room for room in json_data["accomodations"]["rooms"] if new_room_type in room["type"]]
             if len(rooms) == 0:
                 return f"There is no room exists with room type {new_room_type}"
             
-            rooms = [room for room in rooms if guests <= room["number_of_guests"]]
+            rooms = [room for room in rooms if new_guests <= room["number_of_guests"]]
             if len(rooms) == 0:
                 max_guests = json_data["room_types"][new_room_type]["number_of_guests"]
                 return f"The number of guest is superior then the availibilty, maximum is {max_guests}"
@@ -421,7 +421,7 @@ class modify_reservation(ToolBase):
             room_number = reservations[reservation_id]["room_number"]
 
         
-        reservations[reservation_id]["guests"] = guests if guests else reservations[reservation_id]["guests"]
+        reservations[reservation_id]["guests"] = new_guests if new_guests else reservations[reservation_id]["guests"]
         reservations[reservation_id]["reservation_start_date"] = new_reservation_start_date if new_reservation_start_date else reservations[reservation_id]["reservation_start_date"]
         reservations[reservation_id]["reservation_end_date"] = new_reservation_end_date if new_reservation_end_date else reservations[reservation_id]["reservation_end_date"]
         reservations[reservation_id]["room_type"] = new_room_type if new_room_type else reservations[reservation_id]["room_type"]
@@ -430,8 +430,8 @@ class modify_reservation(ToolBase):
         return f"The reservation {reservation_id} is modified correctly: {json.dumps(tmp_data)}"
 
 @tool_register
-class reservation_details(ToolBase):
-    """Playing a specific playlist by its name."""
+class get_reservation_detail(ToolBase):
+    """Retrieve comprehensive details of a specific room reservation by providing its unique reservation ID."""
 
     reservation_id: int = Field(description="Id of the reservation")