Update app.py

app.py

@@ -35,22 +35,28 @@ class PDOKLocationSearchInput(BaseModel):
             return ""  # Will raise error in the tool itself
 
 @tool
-def pdok_location_search(input_data: PDOKLocationSearchInput) -> str:
-    """A tool that queries the PDOK API to find the province related to an adress in the Netherlands.
+def pdok_location_search(postal_code: Optional[str] = None, house_number: Optional[str] = None, street_name: Optional[str] = None, city: Optional[str] = None) -> str:
+    """A tool that queries the PDOK API to find location information in the Netherlands.
 
     Args:
-        input_data:  Input parameters, provide either 'postal_code' and 'house_number' OR 'street_name', 'city', and 'house_number'.
+        postal_code: Postal code in the format '1234 AA'.
+        house_number: House number.
+        street_name: Street name.
+        city: City name.
     """
     base_url = "https://api.pdok.nl/bzk/locatieserver/search/v3_1/free"
     headers = {"accept": "application/json"}
 
+    # Create an instance of the Pydantic model from the function arguments
+    input_data = PDOKLocationSearchInput(postal_code=postal_code, house_number=house_number, street_name=street_name, city=city)
+
     query_string = input_data.construct_query()
     if not query_string:
         return "Please provide either a postal code and house number, or a street name, city, and house number."
-    
+
     params = {
         "q": query_string,
-        "fl": "provincienaam",  # Only fetch provincienaam
+        "fl": "provincienaam",
         "fq": "type:(gemeente OR woonplaats OR weg OR postcode OR adres)",
         "df": "tekst",
         "bq": "type:provincie^1.5",
@@ -67,14 +73,13 @@ def pdok_location_search(input_data: PDOKLocationSearchInput) -> str:
 
     try:
         response = requests.get(base_url, params=params, headers=headers)
-        response.raise_for_status()  # Raise HTTPError for bad responses (4xx or 5xx)
+        response.raise_for_status()
         data = response.json()
 
         docs = data.get("response", {}).get("docs", [])
         if not docs:
             return "No results found for the given query."
 
-        # Extract and return the 'provincienaam' from the first result.
         first_result = docs[0]
         provincie_naam = first_result.get("provincienaam")
 
@@ -83,7 +88,6 @@ def pdok_location_search(input_data: PDOKLocationSearchInput) -> str:
         else:
             return "Provincienaam not found in the API response."
 
-
     except requests.exceptions.RequestException as e:
         return f"Error during API request: {e}"
     except (ValueError, KeyError) as e: