Update tools.py

tools.py

@@ -203,11 +203,11 @@ class make_reservation(ToolBase):
     Creates a new reservation for the hotel by booking a room of the specified type for the desired dates, and associating the booking with a user.
     """
 
+    user_name: int = Field(description="The name of user who is doing reservation.")
     room_type: str = Field(default=list(json_data["room_types"].keys()), description="The type of room being reserved.")
     check_in_date: str = Field(description="The starting date of the reservation (e.g., \"2025-04-01\")")
     check_out_date: str = Field(description="The ending date of the reservation (e.g., \"2025-04-05\").")
     guests: int = Field(description="The number of guests for the reservation.")
-    user_id: int = Field(description="The identifier for the user making the reservation.")
     
     @classmethod
     def invoke(cls, input: Dict) -> str:
@@ -216,7 +216,7 @@ class make_reservation(ToolBase):
         check_in_date = input.get("check_in_date", None)
         check_out_date = input.get("check_out_date", None)
         guests = input.get("guests", None)
-        user_id = input.get("user_id", None)
+        user_name = input.get("user_name", None)
         
         missing = []
         if not room_type:
@@ -227,8 +227,8 @@ class make_reservation(ToolBase):
             missing.append("check_out_date")
         if not guests:
             missing.append("guests")
-        if not user_id:
-            missing.append("user_id")
+        if not user_name:
+            missing.append("user_name")
 
         if len(missing):
             value = ", ".join(missing)
@@ -240,7 +240,7 @@ class make_reservation(ToolBase):
         check_in_date = instance.check_in_date
         check_out_date = instance.check_out_date
         guests = instance.guests
-        user_id = instance.user_id
+        user_name = instance.user_name.lower()
 
         rooms = [room for room in json_data["accomodations"]["rooms"] if room_type in room["type"]]
         if len(rooms) == 0:
@@ -265,7 +265,7 @@ class make_reservation(ToolBase):
             "check_out_date": check_out_date,
             "guests": guests,
             "reservation_id": rand,
-            "user_id": user_id,
+            "user_name": user_name,
         }
 
         reservations[rand] = tmp_data
@@ -276,20 +276,16 @@ class make_reservation(ToolBase):
 class cancel_reservation(ToolBase):
     """Playing a specific playlist by its name."""
 
-    user_id: int = Field(description="The identifier for the user requesting the cancellation.")
     reservation_id: int = Field(description="The unique identifier of the reservation to be canceled.")
     
     @classmethod
     def invoke(cls, input: Dict) -> str: 
 
         reservation_id = input.get("reservation_id", None)
-        user_id = input.get("user_id", None)
 
         missing = []
         if not reservation_id:
             missing.append("reservation_id")
-        if not user_id:
-            missing.append("user_id")
 
         if len(missing):
             value = ", ".join(missing)
@@ -297,16 +293,12 @@ class cancel_reservation(ToolBase):
         
         instance = cls(**input)
         reservation_id = instance.reservation_id
-        user_id = instance.user_id
-
 
 
+        
         if reservation_id not in reservations:
             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."
-        
         reservations.pop(reservation_id)
         return f"The reservation {reservation_id} is cancled correctly"
 
@@ -321,27 +313,22 @@ class modify_reservation(ToolBase):
     new_check_in_date: str = Field(default=None, description="New check out date in format DD/MM/YYYY")
     new_check_out_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.")
-    user_id: int = Field(description="The identifier for the user requesting the modification.")
     reservation_id: int = Field(description="The unique identifier of the reservation to be modified.")
 
     @classmethod
     def invoke(cls, input: Dict) -> str: 
         
-        user_id = input.get("user_id", None)
         reservation_id = input.get("reservation_id", None)
 
         missing = []
         if not reservation_id:
             missing.append("reservation_id")
-        if not user_id:
-            missing.append("user_id")
 
         instance = cls(**input)
         new_room_type = instance.new_room_type
         new_check_in_date = instance.new_check_in_date
         new_check_out_date = instance.new_check_out_date
         guests = instance.guests
-        user_id = instance.user_id
         reservation_id = instance.reservation_id
 
         if len(missing):
@@ -355,9 +342,6 @@ class modify_reservation(ToolBase):
         if reservation_id not in reservations:
             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."
-        
         if new_room_type or guests:
             rooms = [room for room in json_data["restaurants"] if new_room_type in room["type"]]
             if len(rooms) == 0:
@@ -380,43 +364,32 @@ class modify_reservation(ToolBase):
         reservations["reservation_id"]["room_type"] = new_room_type if new_room_type else reservations["reservation_id"]["room_type"]
         reservations["reservation_id"]["room_number"] = room_number
 
-        tmp_data = copy.deepcopy(reservations["reservation_id"])
-        tmp_data.pop("user_id")
-
-        return f"The reservation {reservation_id} is modified correctly: {json.dumps(tmp_data)}"
+        return f"The reservation {reservation_id} is modified correctly: {json.dumps(reservations["reservation_id"])}"
 
 @tool_register
 class reservation_details(ToolBase):
     """Playing a specific playlist by its name."""
 
-    user_id: int = Field(description="Id of user, could be passport or national Identity number")
     reservation_id: int = Field(description="Id of the reservation")
 
     @classmethod
     def invoke(cls, input: Dict) -> str: 
-        user_id = input.get("user_id", None)
         reservation_id = input.get("reservation_id", None)
     
         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}."     
 
         instance = cls(**input)
-        user_id = instance.user_id
         reservation_id = instance.reservation_id
 
         if reservation_id not in reservations:
             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)