digitalkin 0.3.2.dev20__py3-none-any.whl → 0.3.2.dev21__py3-none-any.whl

digitalkin/__version__.py

@@ -5,4 +5,4 @@ from importlib.metadata import PackageNotFoundError, version
 try:
     __version__ = version("digitalkin")
 except PackageNotFoundError:
-    __version__ = "0.3.2.dev20"
+    __version__ = "0.3.2.dev21"

digitalkin/utils/schema_splitter.py

@@ -17,24 +17,31 @@ class SchemaSplitter:
             Tuple of (jsonschema, uischema).
         """
         defs_ui: dict[str, dict[str, Any]] = {}
-        if "$defs" in combined_schema:
-            for def_name, def_value in combined_schema["$defs"].items():
+        schema_defs = combined_schema.get("$defs", {})
+        if schema_defs:
+            for def_name, def_value in schema_defs.items():
                 if isinstance(def_value, dict):
                     defs_ui[def_name] = {}
-                    cls._extract_ui_properties(def_value, defs_ui[def_name])
+                    cls._extract_ui_properties(def_value, defs_ui[def_name], schema_defs)
 
         json_schema: dict[str, Any] = {}
         ui_schema: dict[str, Any] = {}
-        cls._process_object(combined_schema, json_schema, ui_schema, defs_ui)
+        cls._process_object(combined_schema, json_schema, ui_schema, defs_ui, schema_defs)
         return json_schema, ui_schema
 
     @classmethod
-    def _extract_ui_properties(cls, source: dict[str, Any], ui_target: dict[str, Any]) -> None:  # noqa: C901
+    def _extract_ui_properties(
+        cls,
+        source: dict[str, Any],
+        ui_target: dict[str, Any],
+        defs: dict[str, Any] | None = None,
+    ) -> None:
         """Extract ui:* properties from source into ui_target recursively.
 
         Args:
             source: Source dict to extract from.
             ui_target: Target dict for ui properties.
+            defs: The $defs dictionary to resolve $ref references.
         """
         for key, value in source.items():
             if key.startswith("ui:"):
@@ -43,20 +50,26 @@ class SchemaSplitter:
                 for prop_name, prop_value in value.items():
                     if isinstance(prop_value, dict):
                         prop_ui: dict[str, Any] = {}
-                        cls._extract_ui_properties(prop_value, prop_ui)
+                        cls._extract_ui_properties(prop_value, prop_ui, defs)
                         if prop_ui:
                             ui_target[prop_name] = prop_ui
             elif key == "items" and isinstance(value, dict):
                 items_ui: dict[str, Any] = {}
-                cls._extract_ui_properties(value, items_ui)
+                cls._extract_ui_properties(value, items_ui, defs)
                 if items_ui:
                     ui_target["items"] = items_ui
             elif key == "allOf" and isinstance(value, list):
                 for item in value:
                     if isinstance(item, dict):
-                        cls._extract_ui_properties(item, ui_target)
+                        cls._extract_ui_properties(item, ui_target, defs)
             elif key in {"if", "then", "else"} and isinstance(value, dict):
-                cls._extract_ui_properties(value, ui_target)
+                cls._extract_ui_properties(value, ui_target, defs)
+            elif key == "$ref" and isinstance(value, str) and defs is not None:
+                # Resolve $ref and extract UI properties from the referenced definition
+                if value.startswith("#/$defs/"):
+                    def_name = value[8:]
+                    if def_name in defs:
+                        cls._extract_ui_properties(defs[def_name], ui_target, defs)
 
     @classmethod
     def _process_object(  # noqa: C901, PLR0912
@@ -65,6 +78,7 @@ class SchemaSplitter:
         json_target: dict[str, Any],
         ui_target: dict[str, Any],
         defs_ui: dict[str, dict[str, Any]],
+        schema_defs: dict[str, Any] | None = None,
     ) -> None:
         """Process an object, splitting json and ui properties.
 
@@ -73,6 +87,7 @@ class SchemaSplitter:
             json_target: Target dict for json schema.
             ui_target: Target dict for ui schema.
             defs_ui: Pre-extracted UI properties from $defs.
+            schema_defs: The original $defs dictionary to resolve $ref in _extract_ui_properties.
         """
         for key, value in source.items():
             if key.startswith("ui:"):
@@ -113,14 +128,14 @@ class SchemaSplitter:
                         cls._strip_ui_properties(item, item_json)
                         json_target["allOf"].append(item_json)
                         # Extract UI properties from allOf item
-                        cls._extract_ui_properties(item, ui_target)
+                        cls._extract_ui_properties(item, ui_target, schema_defs)
                     else:
                         json_target["allOf"].append(item)
             elif key in {"if", "then", "else"} and isinstance(value, dict):
                 json_target[key] = {}
                 cls._strip_ui_properties(value, json_target[key])
                 # Extract UI properties from conditional
-                cls._extract_ui_properties(value, ui_target)
+                cls._extract_ui_properties(value, ui_target, schema_defs)
             elif key == "hidden":
                 # Strip hidden key from json schema
                 continue

{digitalkin-0.3.2.dev20.dist-info → digitalkin-0.3.2.dev21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.2.dev20
+Version: 0.3.2.dev21
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International

{digitalkin-0.3.2.dev20.dist-info → digitalkin-0.3.2.dev21.dist-info}/RECORD

@@ -7,7 +7,7 @@ base_server/mock/__init__.py,sha256=YZFT-F1l_TpvJYuIPX-7kTeE1CfOjhx9YmNRXVoi-jQ,
 base_server/mock/mock_pb2.py,sha256=sETakcS3PAAm4E-hTCV1jIVaQTPEAIoVVHupB8Z_k7Y,1843
 base_server/mock/mock_pb2_grpc.py,sha256=BbOT70H6q3laKgkHfOx1QdfmCS_HxCY4wCOX84YAdG4,3180
 digitalkin/__init__.py,sha256=7LLBAba0th-3SGqcpqFO-lopWdUkVLKzLZiMtB-mW3M,162
-digitalkin/__version__.py,sha256=dN3fPmuqWC0wGuJWF7p4H8rHGbT_gGfHvNrei8I6TU8,196
+digitalkin/__version__.py,sha256=rOjHWGOhXZoGaCKIj8TeRUY1SA0ZmJjAbwE9acxZsuU,196
 digitalkin/logger.py,sha256=8ze_tjt2G6mDTuQcsf7-UTXWP3UHZ7LZVSs_iqF4rX4,4685
 digitalkin/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 digitalkin/core/__init__.py,sha256=FJRcJ-B1Viyn-38L8XpOpZ8KOnf1I7PCDOAmKXLQhqc,71
@@ -122,8 +122,8 @@ digitalkin/utils/development_mode_action.py,sha256=2hznh0ajW_4ZTysfoc0Y49161f_PQ
 digitalkin/utils/dynamic_schema.py,sha256=y5csxjuqVHjWDpnTUzxbcUuI_wou9-ibRVHQlBs_btY,15275
 digitalkin/utils/llm_ready_schema.py,sha256=JjMug_lrQllqFoanaC091VgOqwAd-_YzcpqFlS7p778,2375
 digitalkin/utils/package_discover.py,sha256=sa6Zp5Kape1Zr4iYiNrnZxiHDnqM06ODk6yfWHom53w,13465
-digitalkin/utils/schema_splitter.py,sha256=7KFiRUhFE9HsH_Z8p0POSbIhiJlgZEeSxL_7TDY0n1U,10374
-digitalkin-0.3.2.dev20.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
+digitalkin/utils/schema_splitter.py,sha256=6PeDrEBGEFyVEPgzGnwcCZuwxLobThmuPwJhGlRpU_0,11137
+digitalkin-0.3.2.dev21.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
 modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modules/archetype_with_tools_module.py,sha256=PXTS6IXmC_OjxTmVrL_pYVI0MKwXjD5I1UJO_2xa10Q,7632
 modules/cpu_intensive_module.py,sha256=GZlirQDZdYuXrI46sv1q4RNAHZjL4EptHVQTvgK9zz8,8363
@@ -138,7 +138,7 @@ monitoring/digitalkin_observability/prometheus.py,sha256=gDmM9ySaVwPAe7Yg84pLxmE
 monitoring/tests/test_metrics.py,sha256=ugnYfAwqBPO6zA8z4afKTlyBWECTivacYSN-URQCn2E,5856
 services/filesystem_module.py,sha256=U4dgqtuDadaXz8PJ1d_uQ_1EPncBqudAQCLUICF9yL4,7421
 services/storage_module.py,sha256=Wz2MzLvqs2D_bnBBgtnujYcAKK2V2KFMk8K21RoepSE,6972
-digitalkin-0.3.2.dev20.dist-info/METADATA,sha256=EosqD8N82EhYPNyjWKp-_n2az8875Qp8iRfdZph5uB4,29725
-digitalkin-0.3.2.dev20.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-digitalkin-0.3.2.dev20.dist-info/top_level.txt,sha256=AYVIesKrO0jnedQ-Muog9JBehG81WeTCNeOFoJgwsgE,51
-digitalkin-0.3.2.dev20.dist-info/RECORD,,
+digitalkin-0.3.2.dev21.dist-info/METADATA,sha256=TSn7ZjFiT9GYhIoAedzR7OzC2LT8_VvMS61vph1X6SA,29725
+digitalkin-0.3.2.dev21.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+digitalkin-0.3.2.dev21.dist-info/top_level.txt,sha256=AYVIesKrO0jnedQ-Muog9JBehG81WeTCNeOFoJgwsgE,51
+digitalkin-0.3.2.dev21.dist-info/RECORD,,