sort keys

PrefectHQ · Jan 22, 2025 · 5fdc225 · 5fdc225
1 parent a4e1be7
commit 5fdc225
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/marvin/utilities/jsonschema.py b/src/marvin/utilities/jsonschema.py
@@ -33,6 +33,7 @@
 """
 
 import hashlib
+import json
 import re
 from copy import deepcopy
 from dataclasses import MISSING, field, make_dataclass
@@ -58,7 +59,6 @@
     StringConstraints,
     model_validator,
 )
-from pydantic_core import to_json
 from typing_extensions import NotRequired, TypedDict
 
 __all__ = ["jsonschema_to_type", "JSONSchema"]
@@ -142,7 +142,7 @@ class Name:
 
 def hash_schema(schema: Mapping[str, Any]) -> str:
     """Generate a deterministic hash for schema caching."""
-    return hashlib.sha256(to_json(schema)).hexdigest()
+    return hashlib.sha256(json.dumps(schema, sort_keys=True).encode()).hexdigest()
 
 
 def resolve_ref(ref: str, schemas: Mapping[str, Any]) -> Mapping[str, Any]: