feat: make Pydantic AnyUrls json serializable in requests

FUTURE_COPYBARA_INTEGRATE_REVIEW=#1558 from googleapis:release-please--branches--main a73a12f
PiperOrigin-RevId: 826167091

Author: speedstorm1

google/genai/_api_client.py

@@ -1112,11 +1112,19 @@ def _request_once(
         http_request.headers['x-goog-user-project'] = (
             self._credentials.quota_project_id
         )
-      data = json.dumps(http_request.data) if http_request.data else None
+      data = json.dumps(
+          _common.convert_uri_to_string(http_request.data)
+          if http_request.data
+          else None
+      )
     else:
       if http_request.data:
         if not isinstance(http_request.data, bytes):
-          data = json.dumps(http_request.data) if http_request.data else None
+          data = json.dumps(
+              _common.convert_uri_to_string(http_request.data)
+              if http_request.data
+              else None
+          )
         else:
           data = http_request.data
 
@@ -1180,11 +1188,11 @@ async def _async_request_once(
         http_request.headers['x-goog-user-project'] = (
             self._credentials.quota_project_id
         )
-      data = json.dumps(http_request.data) if http_request.data else None
+      data = json.dumps(_common.convert_uri_to_string(http_request.data) if http_request.data else None)
     else:
       if http_request.data:
         if not isinstance(http_request.data, bytes):
-          data = json.dumps(http_request.data) if http_request.data else None
+          data = json.dumps(_common.convert_uri_to_string(http_request.data) if http_request.data else None)
         else:
           data = http_request.data
 

google/genai/_common.py

@@ -776,3 +776,15 @@ def recursive_dict_update(
       target_dict[key] = value
     else:
       target_dict[key] = value
+
+
+def convert_uri_to_string(obj: Any) -> Any:
+  if isinstance(obj, dict):
+    return {
+        k: convert_uri_to_string(v) for k, v in obj.items()
+    }
+  elif isinstance(obj, list):
+    return [convert_uri_to_string(i) for i in obj]
+  elif type(obj).__name__ == 'AnyUrl':
+    return str(obj)
+  return obj

google/genai/tests/common/test_common.py

@@ -18,6 +18,7 @@
 
 from enum import Enum
 import inspect
+import json
 import logging
 import textwrap
 import typing
@@ -765,3 +766,52 @@ def test_move_value_by_path():
   }
 
   assert data == expected
+
+
+def test_convert_uri_to_string():
+  url = pydantic.AnyUrl("http://example.com")
+  data = {
+      "name": "test",
+      "url": url,
+      "nested_list": [1, {"url_in_list": url}],
+      "nested_dict": {
+          "url_in_dict": url,
+      },
+  }
+  expected = {
+      "name": "test",
+      "url": "http://example.com/",
+      "nested_list": [1, {"url_in_list": "http://example.com/"}],
+      "nested_dict": {
+          "url_in_dict": "http://example.com/",
+      },
+  }
+  result = _common.convert_uri_to_string(data)
+  assert result == expected
+
+
+def test_convert_uri_to_string_no_anyurl():
+  data = {"name": "test", "value": 1}
+  result = _common.convert_uri_to_string(data)
+  assert result == data
+
+
+def test_convert_uri_to_string_list_of_anyurl():
+  url = pydantic.AnyUrl("http://example.com")
+  data = [url, {"url": url}]
+  expected = ["http://example.com/", {"url": "http://example.com/"}]
+  result = _common.convert_uri_to_string(data)
+  assert result == expected
+
+
+def test_convert_uri_to_string_anyurl_direct():
+  url = pydantic.AnyUrl("http://example.com")
+  result = _common.convert_uri_to_string(url)
+  assert result == "http://example.com/"
+
+
+def test_json_dumps_fails_with_anyurl():
+  url = pydantic.AnyUrl("http://example.com")
+  data = {"url": url}
+  with pytest.raises(TypeError):
+    json.dumps(data)