Fix handling of pydantic_core.Url and pydantic.AnyUrl when building logfire JSON schemas (#1130)

Author: dmontagu

logfire/_internal/json_encoder.py

@@ -213,10 +213,12 @@ def encoder_by_type() -> dict[type[Any], EncoderFunction]:
     }
     with contextlib.suppress(ModuleNotFoundError):
         import pydantic
+        import pydantic_core
 
         lookup.update(
             {
                 pydantic.AnyUrl: _to_str,
+                pydantic_core.Url: _to_str,
                 pydantic.NameEmail: _to_str,
                 pydantic.SecretBytes: _to_str,
                 pydantic.SecretStr: _to_str,

logfire/_internal/json_formatter.py

@@ -36,6 +36,7 @@ def __init__(self, *, indent: int):
             'timedelta': self._format_timedelta,
             'Enum': partial(self._write, '(', ')', True),
             'IPv4Address': partial(self._write, 'IPv4Address(', ')', True),
+            'AnyUrl': partial(self._write, 'AnyUrl(', ')', True),
             'Url': partial(self._write, 'Url(', ')', True),
             'IPv4Interface': partial(self._write, 'IPv4Interface(', ')', True),
             'IPv4Network': partial(self._write, 'IPv4Network(', ')', True),

logfire/_internal/json_types.py

@@ -40,6 +40,7 @@
     'Exception',
     # pydantic types
     'PydanticModel',
+    'AnyUrl',
     'Url',
     'NameEmail',
     'SecretBytes',