Update app.py

app.py

@@ -46,22 +46,15 @@ class PDOKLocationInfo(BaseModel):
     centroide_rd: str
 
 @tool
-def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional[str] = None, street_name: Optional[str] = None, city: Optional[str] = None) -> Optional[PDOKLocationInfo]:
-    """Provides information about a Dutch address or postal code.
+def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional[str] = None, street_name: Optional[str] = None, city: Optional[str] = None) -> Union[PDOKLocationInfo, str]:
+    """[Original docstring remains the same]"""
     
-    Args:
-    postal_code: Postal code in the format '1234 AA'.
-    house_number: House number.
-    street_name: Street name.
-    city: City name.
-
-    A postalcode and housenumber or a streetname, housenumber and city are required.
-
-    Ask the user for additional input if you miss an element to fullfill the requirements of this tool, don't fill in these elements on your own.
-    
-    Returns:
-        A PDOKLocationInfo object containing the location information, or None if no results are found.
-    """
+    debug_info = []
+    debug_info.append(f"Input values:\n"
+                     f"  Postal code: {postal_code}\n"
+                     f"  House number: {house_number}\n"
+                     f"  Street name: {street_name}\n"
+                     f"  City: {city}")
     
     base_url = "https://api.pdok.nl/bzk/locatieserver/search/v3_1/free"
     headers = {"accept": "application/json"}
@@ -69,12 +62,14 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
     input_data = PDOKLocationSearchInput(postal_code=postal_code, house_number=house_number, street_name=street_name, city=city)
     query_string = input_data.construct_query()
 
+    debug_info.append(f"Constructed query string: {query_string}")
+
     if not query_string:
-        return None
+        return f"Error: Empty query string\nDebug info:\n" + "\n".join(debug_info)
 
     params = {
         "q": query_string,
-        "fl": "*", # Request all fields
+        "fl": "*",
         "fq": "type:(gemeente OR woonplaats OR weg OR postcode OR adres)",
         "df": "tekst",
         "bq": "type:provincie^1.5",
@@ -89,32 +84,32 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
         "wt": "json",
     }
     
+    debug_info.append(f"API request details:\n"
+                     f"  URL: {base_url}\n"
+                     f"  Params: {params}")
+    
     try:
         response = requests.get(base_url, params=params, headers=headers)
+        debug_info.append(f"Response status code: {response.status_code}\n"
+                         f"Response URL: {response.url}")
+        
         response.raise_for_status()
         data = response.json()
         
         docs = data.get("response", {}).get("docs", [])
+        debug_info.append(f"Number of docs found: {len(docs)}")
+        
         if not docs:
-            return None
+            return f"Error: No results found\nDebug info:\n" + "\n".join(debug_info)
 
         first_result = docs[0]
         location_info = PDOKLocationInfo(**first_result)
-
-        # Format the output in a more user-friendly way
-        print(f"Information about: {location_info.weergavenaam}")
-        print("-" * 30)
-        for field, value in location_info.dict().items():
-            print(f"  {field.capitalize()}: {value}")
-        
         return location_info
     
     except requests.exceptions.RequestException as e:
-        print(f"Error during API request: {e}")
-        return None
+        return f"Error during API request: {e}\nDebug info:\n" + "\n".join(debug_info)
     except (ValueError, KeyError) as e:
-        print(f"Error processing API response: {e}")
-        return None
+        return f"Error processing API response: {e}\nDebug info:\n" + "\n".join(debug_info)
 
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type