Update app.py

app.py

@@ -11,18 +11,17 @@ import requests
 from pydantic import BaseModel, Field, validator
 
 class PDOKLocationSearchInput(BaseModel):
-    """Input for the PDOK location search tool."""
     postal_code: Optional[str] = Field(None, description="Postal code in the format '1234 AA'.")
     house_number: Optional[str] = Field(None, description="House number.")
     street_name: Optional[str] = Field(None, description="Street name.")
     city: Optional[str] = Field(None, description="City name.")
 
-    @validator("postal_code")
+    @field_validator('postal_code')
     def validate_postal_code(cls, v):
         if v is not None and (len(v) != 7 or not v[0:4].isdigit() or v[4] != " " or not v[5:7].isalpha()):
             raise ValueError("Invalid postal code format. It must be '1234 AA'.")
         return v
-
+        
     def construct_query(self) -> str:
         """Constructs the query string based on provided inputs."""
         if self.postal_code and self.house_number:
@@ -160,7 +159,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, pdok_location_search], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, pdok_location_info], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,