Update app.py

app.py

@@ -35,16 +35,19 @@ class PDOKLocationSearchInput(BaseModel):
 class PDOKLocationInfo(BaseModel):
     type: str
     weergavenaam: str
-    straatnaam: str
-    huisnummer: int
-    huisletter: Optional[str]
-    huisnummertoevoeging: Optional[str]
-    postcode: str
-    woonplaatsnaam: str
-    gemeentenaam: str
-    provincienaam: str
-    centroide_ll: str
-    centroide_rd: str
+    straatnaam: Optional[str] = None
+    huisnummer: Optional[int] = None
+    huisletter: Optional[str] = None
+    huisnummertoevoeging: Optional[str] = None
+    postcode: Optional[str] = None
+    woonplaatsnaam: Optional[str] = None
+    gemeentenaam: Optional[str] = None
+    provincienaam: Optional[str] = None
+    centroide_ll: Optional[str] = None
+    centroide_rd: Optional[str] = None
+
+    class Config:
+        extra = "allow"  # This will allow extra fields from the API response
 
 @tool
 def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional[str] = None, street_name: Optional[str] = None, city: Optional[str] = None) -> str:
@@ -95,14 +98,9 @@ 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}")
+        debug_info.append(f"Response status code: {response.status_code}")
         
         response.raise_for_status()
         data = response.json()
@@ -114,8 +112,9 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
             return f"Error: No results found\nDebug info:\n" + "\n".join(debug_info)
 
         first_result = docs[0]
+        debug_info.append(f"First result data: {first_result}")
+        
         location_info = PDOKLocationInfo(**first_result)
-        # Convert the PDOKLocationInfo object to a JSON string
         return location_info.model_dump_json()
     
     except requests.exceptions.RequestException as e:
@@ -123,7 +122,6 @@ def pdok_location_info(postal_code: Optional[str] = None, house_number: Optional
     except (ValueError, KeyError) as e:
         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
     #Keep this format for the description / args / args description but feel free to modify the tool