🚧 test

Author: ff137

pydid/did_url.py

@@ -1,6 +1,6 @@
 """DID URL Object."""
 
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Callable, Dict, Optional
 from urllib.parse import parse_qsl, urlencode, urlparse
 
 from pydantic import GetCoreSchemaHandler, GetJsonSchemaHandler
@@ -44,10 +44,19 @@ def __init__(self, url: str):
 
     @classmethod
     def __get_pydantic_core_schema__(
-        cls, source_type: Any, handler: GetCoreSchemaHandler
-    ) -> CoreSchema:
-        """Update schema fields."""
-        return core_schema.no_info_after_validator_function(cls, handler(str))
+        cls,
+        _source_type: Any,
+        _handler: Callable[[Any], core_schema.CoreSchema],
+    ) -> core_schema.CoreSchema:
+
+        return core_schema.union_schema(
+            [
+                # check if it's an instance first before doing any further work
+                core_schema.is_instance_schema(str),
+                core_schema.no_info_plain_validator_function(cls.validate),
+            ],
+            serialization=core_schema.to_string_ser_schema(),
+        )
 
     @classmethod
     def __get_pydantic_json_schema__(