digitalkin 0.3.2.dev27__tar.gz → 0.3.2.dev29__tar.gz

{digitalkin-0.3.2.dev27 → digitalkin-0.3.2.dev29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.2.dev27
+Version: 0.3.2.dev29
 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.dev27 → digitalkin-0.3.2.dev29}/pyproject.toml

@@ -25,7 +25,7 @@
         "Programming Language :: Python",
         "Topic :: Software Development :: Libraries",
     ]
-    version = "0.3.2.dev27"
+    version = "0.3.2.dev29"
 
     dependencies = [
         "agentic-mesh-protocol==0.2.1.dev0",

{digitalkin-0.3.2.dev27 → digitalkin-0.3.2.dev29}/src/digitalkin/__version__.py

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

{digitalkin-0.3.2.dev27 → digitalkin-0.3.2.dev29}/src/digitalkin/models/module/tool_reference.py

@@ -2,7 +2,7 @@
 
 from typing import Annotated
 
-from pydantic import AfterValidator, BaseModel, BeforeValidator, Field
+from pydantic import AfterValidator, BaseModel, BeforeValidator, Field, PlainSerializer
 from pydantic.annotated_handlers import GetJsonSchemaHandler
 from pydantic.json_schema import JsonSchemaValue
 from pydantic_core import CoreSchema
@@ -134,10 +134,19 @@ def tool_reference_input(
             raise ValueError(msg)
         return v
 
+    def serialize_to_list(v: ToolReference) -> list[str]:
+        """Serialize ToolReference as plain list for frontend compatibility.
+
+        Returns:
+            List of setup IDs.
+        """
+        return v.selected_tools
+
     return Annotated[  # type: ignore[return-value]
         ToolReference,
         BeforeValidator(convert_to_tool_reference),
         AfterValidator(validate_tools_count),
+        PlainSerializer(serialize_to_list, return_type=list[str]),
         _ToolReferenceInputSchema(
             setup_ids=setup_ids,
             module_ids=module_ids,

{digitalkin-0.3.2.dev27 → digitalkin-0.3.2.dev29}/src/digitalkin/utils/schema_splitter.py

@@ -58,7 +58,7 @@ class SchemaSplitter:
                 cls._extract_ui_properties(value, items_ui, defs)
                 if items_ui:
                     ui_target["items"] = items_ui
-            elif key == "allOf" and isinstance(value, list):
+            elif key in {"allOf", "oneOf", "anyOf"} and isinstance(value, list):
                 for item in value:
                     if isinstance(item, dict):
                         cls._extract_ui_properties(item, ui_target, defs)
@@ -71,7 +71,7 @@ class SchemaSplitter:
                     cls._extract_ui_properties(defs[def_name], ui_target, defs)
 
     @classmethod
-    def _process_object(  # noqa: C901, PLR0912
+    def _process_object(  # noqa: C901, PLR0912, PLR0915
         cls,
         source: dict[str, Any],
         json_target: dict[str, Any],
@@ -123,13 +123,24 @@ class SchemaSplitter:
                 json_target["allOf"] = []
                 for item in value:
                     if isinstance(item, dict):
-                        item_json: dict[str, Any] = {}
-                        cls._strip_ui_properties(item, item_json)
-                        json_target["allOf"].append(item_json)
+                        item_json_all_of: dict[str, Any] = {}
+                        cls._strip_ui_properties(item, item_json_all_of)
+                        json_target["allOf"].append(item_json_all_of)
                         # Extract UI properties from allOf item
                         cls._extract_ui_properties(item, ui_target, schema_defs)
                     else:
                         json_target["allOf"].append(item)
+            elif key in {"oneOf", "anyOf"} and isinstance(value, list):
+                json_target[key] = []
+                for item in value:
+                    if isinstance(item, dict):
+                        item_json_oneof_anyof: dict[str, Any] = {}
+                        cls._strip_ui_properties(item, item_json_oneof_anyof)
+                        json_target[key].append(item_json_oneof_anyof)
+                        # Extract UI properties from oneOf/anyOf item (resolves $ref)
+                        cls._extract_ui_properties(item, ui_target, schema_defs)
+                    else:
+                        json_target[key].append(item)
             elif key in {"if", "then", "else"} and isinstance(value, dict):
                 json_target[key] = {}
                 cls._strip_ui_properties(value, json_target[key])
@@ -187,6 +198,19 @@ class SchemaSplitter:
                 cls._process_property(value, json_target["items"], items_ui, defs_ui)
                 if items_ui:
                     ui_target["items"] = items_ui
+            elif key in {"oneOf", "anyOf"} and isinstance(value, list):
+                json_target[key] = []
+                for item in value:
+                    if isinstance(item, dict):
+                        item_json_oneof_anyof: dict[str, Any] = {}
+                        cls._strip_ui_properties(item, item_json_oneof_anyof)
+                        json_target[key].append(item_json_oneof_anyof)
+                        # Extract UI properties from $ref in oneOf/anyOf items
+                        ref_path = item.get("$ref", "")
+                        if ref_path.startswith("#/$defs/") and ref_path[8:] in defs_ui:
+                            ui_target.update(defs_ui[ref_path[8:]])
+                    else:
+                        json_target[key].append(item)
             elif key == "hidden":
                 # Strip hidden key from json schema
                 continue
@@ -230,11 +254,20 @@ class SchemaSplitter:
                 json_target["allOf"] = []
                 for item in value:
                     if isinstance(item, dict):
-                        item_json: dict[str, Any] = {}
-                        cls._strip_ui_properties(item, item_json)
-                        json_target["allOf"].append(item_json)
+                        item_json_all_of: dict[str, Any] = {}
+                        cls._strip_ui_properties(item, item_json_all_of)
+                        json_target["allOf"].append(item_json_all_of)
                     else:
                         json_target["allOf"].append(item)
+            elif key in {"oneOf", "anyOf"} and isinstance(value, list):
+                json_target[key] = []
+                for item in value:
+                    if isinstance(item, dict):
+                        item_json_oneof_anyof: dict[str, Any] = {}
+                        cls._strip_ui_properties(item, item_json_oneof_anyof)
+                        json_target[key].append(item_json_oneof_anyof)
+                    else:
+                        json_target[key].append(item)
             elif key in {"if", "then", "else"} and isinstance(value, dict):
                 json_target[key] = {}
                 cls._strip_ui_properties(value, json_target[key])

{digitalkin-0.3.2.dev27 → digitalkin-0.3.2.dev29}/src/digitalkin.egg-info/PKG-INFO

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