digitalkin 0.3.2.dev7__py3-none-any.whl → 0.3.2.dev10__py3-none-any.whl

digitalkin/services/registry/__init__.py

@@ -2,7 +2,6 @@
 
 from digitalkin.models.services.registry import (
     ModuleInfo,
-    ModuleStatusInfo,
     RegistryModuleStatus,
     RegistryModuleType,
 )
@@ -12,6 +11,7 @@ from digitalkin.services.registry.exceptions import (
     RegistryServiceError,
 )
 from digitalkin.services.registry.grpc_registry import GrpcRegistry
+from digitalkin.services.registry.registry_models import ModuleStatusInfo
 from digitalkin.services.registry.registry_strategy import RegistryStrategy
 
 __all__ = [

digitalkin/services/registry/default_registry.py

@@ -4,11 +4,11 @@ from typing import ClassVar
 
 from digitalkin.models.services.registry import (
     ModuleInfo,
-    ModuleStatusInfo,
     RegistryModuleStatus,
     RegistryModuleType,
 )
 from digitalkin.services.registry.exceptions import RegistryModuleNotFoundError
+from digitalkin.services.registry.registry_models import ModuleStatusInfo
 from digitalkin.services.registry.registry_strategy import RegistryStrategy
 
 

digitalkin/services/registry/grpc_registry.py

@@ -20,7 +20,6 @@ from digitalkin.logger import logger
 from digitalkin.models.grpc_servers.models import ClientConfig
 from digitalkin.models.services.registry import (
     ModuleInfo,
-    ModuleStatusInfo,
     RegistryModuleStatus,
     RegistryModuleType,
 )
@@ -28,6 +27,7 @@ from digitalkin.services.registry.exceptions import (
     RegistryModuleNotFoundError,
     RegistryServiceError,
 )
+from digitalkin.services.registry.registry_models import ModuleStatusInfo
 from digitalkin.services.registry.registry_strategy import RegistryStrategy
 
 

digitalkin/services/registry/registry_models.py

@@ -3,37 +3,9 @@
 This module contains Pydantic models for registry service data structures.
 """
 
-from enum import IntEnum
-
 from pydantic import BaseModel
 
-
-class RegistryModuleStatus(IntEnum):
-    """Module status in the registry.
-
-    Maps to proto ModuleStatus enum values.
-    """
-
-    UNKNOWN = 0
-    READY = 1
-    ACTIVE = 2
-    OFFLINE = 3
-
-
-class ModuleInfo(BaseModel):
-    """Complete module information from registry.
-
-    Maps to proto ModuleDescriptor message.
-    """
-
-    module_id: str
-    module_type: str
-    address: str
-    port: int
-    version: str
-    name: str = ""
-    documentation: str | None = None
-    status: RegistryModuleStatus | None = None
+from digitalkin.models.services.registry import RegistryModuleStatus
 
 
 class ModuleStatusInfo(BaseModel):

digitalkin/services/registry/registry_strategy.py

@@ -5,10 +5,10 @@ from typing import Any
 
 from digitalkin.models.services.registry import (
     ModuleInfo,
-    ModuleStatusInfo,
     RegistryModuleStatus,
 )
 from digitalkin.services.base_strategy import BaseStrategy
+from digitalkin.services.registry.registry_models import ModuleStatusInfo
 
 
 class RegistryStrategy(BaseStrategy, ABC):

digitalkin/utils/schema_splitter.py

@@ -0,0 +1,207 @@
+"""Schema splitter for react-jsonschema-form."""
+
+from typing import Any
+
+
+class SchemaSplitter:
+    """Splits a combined JSON schema into jsonschema and uischema for react-jsonschema-form."""
+
+    @classmethod
+    def split(cls, combined_schema: dict[str, Any]) -> tuple[dict[str, Any], dict[str, Any]]:
+        """Split schema into (jsonschema, uischema).
+
+        Args:
+            combined_schema: Combined JSON schema with ui:* properties.
+
+        Returns:
+            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():
+                if isinstance(def_value, dict):
+                    defs_ui[def_name] = {}
+                    cls._extract_ui_properties(def_value, defs_ui[def_name])
+
+        json_schema: dict[str, Any] = {}
+        ui_schema: dict[str, Any] = {}
+        cls._process_object(combined_schema, json_schema, ui_schema, defs_ui)
+        return json_schema, ui_schema
+
+    @classmethod
+    def _extract_ui_properties(cls, source: dict[str, Any], ui_target: dict[str, Any]) -> None:  # noqa: C901
+        """Extract ui:* properties from source into ui_target recursively.
+
+        Args:
+            source: Source dict to extract from.
+            ui_target: Target dict for ui properties.
+        """
+        for key, value in source.items():
+            if key.startswith("ui:"):
+                ui_target[key] = value
+            elif key == "properties" and isinstance(value, dict):
+                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)
+                        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)
+                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)
+
+    @classmethod
+    def _process_object(  # noqa: C901, PLR0912
+        cls,
+        source: dict[str, Any],
+        json_target: dict[str, Any],
+        ui_target: dict[str, Any],
+        defs_ui: dict[str, dict[str, Any]],
+    ) -> None:
+        """Process an object, splitting json and ui properties.
+
+        Args:
+            source: Source object to process.
+            json_target: Target dict for json schema.
+            ui_target: Target dict for ui schema.
+            defs_ui: Pre-extracted UI properties from $defs.
+        """
+        for key, value in source.items():
+            if key.startswith("ui:"):
+                ui_target[key] = value
+            elif key == "properties" and isinstance(value, dict):
+                json_target["properties"] = {}
+                for prop_name, prop_value in value.items():
+                    if isinstance(prop_value, dict):
+                        json_target["properties"][prop_name] = {}
+                        prop_ui: dict[str, Any] = {}
+                        cls._process_property(prop_value, json_target["properties"][prop_name], prop_ui, defs_ui)
+                        if prop_ui:
+                            ui_target[prop_name] = prop_ui
+                    else:
+                        json_target["properties"][prop_name] = prop_value
+            elif key == "$defs" and isinstance(value, dict):
+                json_target["$defs"] = {}
+                for def_name, def_value in value.items():
+                    if isinstance(def_value, dict):
+                        json_target["$defs"][def_name] = {}
+                        cls._strip_ui_properties(def_value, json_target["$defs"][def_name])
+                    else:
+                        json_target["$defs"][def_name] = def_value
+            elif key == "items" and isinstance(value, dict):
+                json_target["items"] = {}
+                items_ui: dict[str, Any] = {}
+                cls._process_property(value, json_target["items"], items_ui, defs_ui)
+                if items_ui:
+                    ui_target["items"] = items_ui
+            elif key == "allOf" and isinstance(value, list):
+                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)
+                    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])
+            else:
+                json_target[key] = value
+
+    @classmethod
+    def _process_property(  # noqa: C901, PLR0912
+        cls,
+        source: dict[str, Any],
+        json_target: dict[str, Any],
+        ui_target: dict[str, Any],
+        defs_ui: dict[str, dict[str, Any]],
+    ) -> None:
+        """Process a property, resolving $ref for UI properties.
+
+        Args:
+            source: Source property dict.
+            json_target: Target dict for json schema.
+            ui_target: Target dict for ui schema.
+            defs_ui: Pre-extracted UI properties from $defs.
+        """
+        if "$ref" in source:
+            ref_path = source["$ref"]
+            if ref_path.startswith("#/$defs/"):
+                def_name = ref_path[8:]
+                if def_name in defs_ui:
+                    ui_target.update(defs_ui[def_name])
+
+        for key, value in source.items():
+            if key.startswith("ui:"):
+                ui_target[key] = value
+            elif key == "properties" and isinstance(value, dict):
+                json_target["properties"] = {}
+                for prop_name, prop_value in value.items():
+                    if isinstance(prop_value, dict):
+                        json_target["properties"][prop_name] = {}
+                        prop_ui: dict[str, Any] = {}
+                        cls._process_property(prop_value, json_target["properties"][prop_name], prop_ui, defs_ui)
+                        if prop_ui:
+                            ui_target[prop_name] = prop_ui
+                    else:
+                        json_target["properties"][prop_name] = prop_value
+            elif key == "items" and isinstance(value, dict):
+                json_target["items"] = {}
+                items_ui: dict[str, Any] = {}
+                cls._process_property(value, json_target["items"], items_ui, defs_ui)
+                if items_ui:
+                    ui_target["items"] = items_ui
+            else:
+                json_target[key] = value
+
+    @classmethod
+    def _strip_ui_properties(cls, source: dict[str, Any], json_target: dict[str, Any]) -> None:  # noqa: C901, PLR0912
+        """Copy source to json_target, stripping ui:* properties.
+
+        Args:
+            source: Source dict.
+            json_target: Target dict without ui:* properties.
+        """
+        for key, value in source.items():
+            if key.startswith("ui:"):
+                continue
+            if key == "properties" and isinstance(value, dict):
+                json_target["properties"] = {}
+                for prop_name, prop_value in value.items():
+                    if isinstance(prop_value, dict):
+                        json_target["properties"][prop_name] = {}
+                        cls._strip_ui_properties(prop_value, json_target["properties"][prop_name])
+                    else:
+                        json_target["properties"][prop_name] = prop_value
+            elif key == "$defs" and isinstance(value, dict):
+                json_target["$defs"] = {}
+                for def_name, def_value in value.items():
+                    if isinstance(def_value, dict):
+                        json_target["$defs"][def_name] = {}
+                        cls._strip_ui_properties(def_value, json_target["$defs"][def_name])
+                    else:
+                        json_target["$defs"][def_name] = def_value
+            elif key == "items" and isinstance(value, dict):
+                json_target["items"] = {}
+                cls._strip_ui_properties(value, json_target["items"])
+            elif key == "allOf" and isinstance(value, list):
+                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)
+                    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])
+            else:
+                json_target[key] = value

{digitalkin-0.3.2.dev7.dist-info → digitalkin-0.3.2.dev10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.2.dev7
+Version: 0.3.2.dev10
 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.dev7.dist-info → digitalkin-0.3.2.dev10.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=CcSqodNkINfd0uyPHVCwE5GCsEYQVveGr3ohUqS3x6A,195
+digitalkin/__version__.py,sha256=HFosF_nDLUB58s-K9x_90emrDL2e-FS3aqanQgLZcDM,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
@@ -28,12 +28,12 @@ digitalkin/core/task_manager/task_session.py,sha256=5jw21bT_SPXUzWE7tk6YG62EXqlR
 digitalkin/grpc_servers/__init__.py,sha256=ZIRMJ1Lcas8yQ106GCup6hn2UBOsx1sNk8ap0lpEDnY,72
 digitalkin/grpc_servers/_base_server.py,sha256=ZVeCDwI7w7fFbPTXPkeJb_SOuLfd2T7za3T4oCu2UWY,18680
 digitalkin/grpc_servers/module_server.py,sha256=Ec3izzV2YpdN8rGs_cX-iVulQ00FkLR5dBflHlQ8a6Y,7849
-digitalkin/grpc_servers/module_servicer.py,sha256=a5qVOd0beAwXlFAch9VfLNR8PdaYj_fF2O8rXcwInSU,21087
+digitalkin/grpc_servers/module_servicer.py,sha256=7GQOyAPYMxHVaJGplgDNiVoKr1oaAIL-zdZpyDpznTA,20530
 digitalkin/grpc_servers/utils/__init__.py,sha256=ZnAIb_F8z4NhtPypqkdmzgRSzolKnJTk3oZx5GfWH5Y,38
 digitalkin/grpc_servers/utils/exceptions.py,sha256=LtaDtlqXCeT6iqApogs4pbtezotOVeg4fhnFzGBvFsY,692
-digitalkin/grpc_servers/utils/grpc_client_wrapper.py,sha256=ElGvp6evY5q-EBmDVyQZaDJktfShtMsptMmq16jfgxA,3285
+digitalkin/grpc_servers/utils/grpc_client_wrapper.py,sha256=nGG8QdKnBH0UG9qbKrlPwIvcvPgW3osw7O3cImxisPE,3279
 digitalkin/grpc_servers/utils/grpc_error_handler.py,sha256=0wPEU4713_ZlgIilaeXJV2bi90tHwYO1myDrSLeenKk,1848
-digitalkin/grpc_servers/utils/utility_schema_extender.py,sha256=qTnGrRcze2qe_W61FOfcXGWRILNctidm7z_1GKlCM1E,3719
+digitalkin/grpc_servers/utils/utility_schema_extender.py,sha256=UCJR5YAKA_Wg5Q9feInf4AZW6AVtSQZYRZuMsaEgZVo,3775
 digitalkin/mixins/__init__.py,sha256=d6ljaoyJZJT9XxOrXZG5FVNvbLURb3_CZrkp4GPZWYM,590
 digitalkin/mixins/base_mixin.py,sha256=uLkg6MbDtVc9DysjdfNIGKahxQLnnjuL3DYpuyNLbk8,486
 digitalkin/mixins/callback_mixin.py,sha256=90nHm9-pbKT14GAy3CB3fsBtpYu5IH0woOQdNLM2e_Y,836
@@ -48,24 +48,23 @@ digitalkin/models/core/__init__.py,sha256=jOMDmPX0uSfGA9zUi0u_kOvYJ46VdIssoIhVYv
 digitalkin/models/core/job_manager_models.py,sha256=wvf2dzRzAu0-zzzAXQe6XTC36cNA10sXRLt2p_TFqjk,1003
 digitalkin/models/core/task_monitor.py,sha256=CW-jydSgXMV464W0pqfar0HpgqlSxqdujmC-f8p9EQc,2639
 digitalkin/models/grpc_servers/__init__.py,sha256=0tA71nPSXgRrh9DoLvx-TSwZXdYIRUEItoadpTL1cTo,42
-digitalkin/models/grpc_servers/models.py,sha256=unV1Wo0u3Efm7ddgYyYZZYUC_W6F0S5BQYH3xsOmXjw,8965
+digitalkin/models/grpc_servers/models.py,sha256=gRX94eL71a5mLIie-lCOwE7a0As_AuGduxPPzTHbAe4,13797
 digitalkin/models/grpc_servers/types.py,sha256=rQ78s4nAet2jy-NIDj_PUWriT0kuGHr_w6ELjmjgBao,539
 digitalkin/models/module/__init__.py,sha256=N55wan3rAUVPEGLIDjXoFM_-DYY_zxvbQOZHzNDfwoY,751
 digitalkin/models/module/base_types.py,sha256=oIylVNqo0idTFj4dRgCt7P19daNZ-AlvgCPpL9TJvto,1850
 digitalkin/models/module/module.py,sha256=k0W8vfJJFth8XdDzkHm32SyTuSf3h2qF0hSrxAfGF1s,956
-digitalkin/models/module/module_context.py,sha256=cjMP3cYPC4ZjTWXY3JhF_77PMT72ZglhHfq2I4JIl6I,6430
-digitalkin/models/module/module_helpers.py,sha256=n5lx25c5zRmuivcThC-HdumsjyWDv31xC6tI6NMLR7U,6438
+digitalkin/models/module/module_context.py,sha256=qpjyMYgTCyQS0lW2RhTKZsoQKwFpv6l08FIM0sqC6DQ,10326
 digitalkin/models/module/module_types.py,sha256=C9azCNBk76xMa-Mww8_6AiwQR8MLAsEyUOvBYxytovI,739
-digitalkin/models/module/setup_types.py,sha256=e9CG8x75Mkh7dDIQ66UWU99VcWmyk-JAqHCpUtcgyns,21323
-digitalkin/models/module/tool_cache.py,sha256=WFoPFtHLs1Q3jdeQZpu-ZycnJwhocyYyeAo9k0JXj5o,7434
-digitalkin/models/module/tool_reference.py,sha256=A4EewKMlfhdI2hW7F3kgPc6_UinUrN-hhOweM0KzSa0,4302
+digitalkin/models/module/setup_types.py,sha256=XMKyDm-7n_Za9tDBv1AkgXUowlUj46SMxWANa_yg_LU,18311
+digitalkin/models/module/tool_cache.py,sha256=RP3JwASV8dFUZEKudyALbu0_tq81lstuEc4QHMQpmvM,2073
+digitalkin/models/module/tool_reference.py,sha256=1AyNK-8McsSTEb62aaSou5AAXkKUOJqrFtJ5YWrFbmw,3512
 digitalkin/models/module/utility.py,sha256=gnbYfWpXGbomUI0fWf7T-Qm_VvT-LXDv1OuA9zObwVg,5589
 digitalkin/models/services/__init__.py,sha256=jhfVw6egq0OcHmos_fypH9XFehbHTBw09wluVFVFEyw,226
 digitalkin/models/services/cost.py,sha256=9PXvd5RrIk9vCrRjcUGQ9ZyAokEbwLg4s0RfnE-aLP4,1616
-digitalkin/models/services/registry.py,sha256=0qMAIi52ovofZ4XTV7UkI4i7sR5C3_VFLp-kw9AjQ24,829
+digitalkin/models/services/registry.py,sha256=hz_r03-K633XHu2fOb5HWsE59EPFusBBipy0Gz6OBvI,705
 digitalkin/models/services/storage.py,sha256=wp7F-AvTsU46ujGPcguqM5kUKRZx4399D4EGAAJt2zs,1143
 digitalkin/modules/__init__.py,sha256=vTQk8DWopxQSJ17BjE5dNhq247Rou55iQLJdBxoPUmo,296
-digitalkin/modules/_base_module.py,sha256=0bbiPOcLSiKqjrVYaWdmc72jbpaPa8dOzozEcQO0zMg,21285
+digitalkin/modules/_base_module.py,sha256=4-fS376f9ti3xSgM946yRccHcEGttdo2QYKaxdqk1kY,22451
 digitalkin/modules/archetype_module.py,sha256=XC9tl1Yr6QlbPn_x0eov6UUZwQgwW--BYPPMYVJH_NU,505
 digitalkin/modules/tool_module.py,sha256=GBis7bKCkvWFCYLRvaS9oZVmLBBve1w8BhVnKOU2sCc,506
 digitalkin/modules/trigger_handler.py,sha256=qPNMi-8NHqscOxciHeaXtpwjXApT3YzjMF23zQAjaZY,1770
@@ -95,12 +94,12 @@ digitalkin/services/filesystem/grpc_filesystem.py,sha256=9Cjp4ie-mJ2EMMZ8DjCqeV6
 digitalkin/services/identity/__init__.py,sha256=InkeyLgFYYwItx8mePA8HpfacOMWZwwuc0G4pWtKq9s,270
 digitalkin/services/identity/default_identity.py,sha256=Y2auZHrGSZTIN5D8HyjLvLcNbYFM1CNUE23x7p5VIGw,386
 digitalkin/services/identity/identity_strategy.py,sha256=skappBbds1_qa0Gr24FGrNX1N0_OYhYT1Lh7dUaAirE,429
-digitalkin/services/registry/__init__.py,sha256=4C3bJzOa7gv7fe3etxOg0653wDOwEt6NrA-pBcwFr2s,783
-digitalkin/services/registry/default_registry.py,sha256=b6Y0TVffC1_nYlPpJ0kcfn11Gj2s7I5II5oPypD6cIM,4206
+digitalkin/services/registry/__init__.py,sha256=WPGQM3U-QvMXhsaOy9BN0kVMU3QkPFwAMT3lGmTR-Ko,835
+digitalkin/services/registry/default_registry.py,sha256=tOqw9Ve9w_BzhqrZmHuUl5Ps-J_KTEwYg3tu1gNIHmw,4258
 digitalkin/services/registry/exceptions.py,sha256=tAcVXioCzDqfBvxB_P0uQpaK_LDLrFb0KpymROuqs-8,1371
-digitalkin/services/registry/grpc_registry.py,sha256=TOcwa0b0dkdQh7V8Jm7XajWiIL9MQ0ReXcAlm4SJxfQ,10770
-digitalkin/services/registry/registry_models.py,sha256=9ypU68K2IMIKwt7QAJERc0cvXJeGcfh8iIfo5sOim5M,805
-digitalkin/services/registry/registry_strategy.py,sha256=qeHLY_T8NEFRaeBt7QfaJxezA2TVvFR1ysK863pI5j0,2693
+digitalkin/services/registry/grpc_registry.py,sha256=2_He-I9I4SXdjYA9QYxFWQrr_xBao6LIYXsgiICkeHY,10822
+digitalkin/services/registry/registry_models.py,sha256=DJEwMJg5_BewpgHDtY8xIGWj9jA9H07iYgHLCv81giY,331
+digitalkin/services/registry/registry_strategy.py,sha256=N6oLxMwnYkM_XVW7WDxZoSfFafjUtsFDig3F8VQZj7Q,2745
 digitalkin/services/setup/__init__.py,sha256=t6xcvEWqTbcRZstBFK9cESEqaZKvpW14VtYygxIqfYQ,65
 digitalkin/services/setup/default_setup.py,sha256=zTG9Nvw3Tqy9qctSEnjWE4Mlqol321AUaTlIm6Jl1k0,8195
 digitalkin/services/setup/grpc_setup.py,sha256=aEAHiG0WSzNSQtJjn7xAGQwj70b0HrFO8GBKFXRx3aI,13925
@@ -122,15 +121,23 @@ digitalkin/utils/development_mode_action.py,sha256=2hznh0ajW_4ZTysfoc0Y49161f_PQ
 digitalkin/utils/dynamic_schema.py,sha256=5-B3dBGlCYYv6uRJkgudtc0ZpBOTYxl0yKedDGsteZQ,15184
 digitalkin/utils/llm_ready_schema.py,sha256=JjMug_lrQllqFoanaC091VgOqwAd-_YzcpqFlS7p778,2375
 digitalkin/utils/package_discover.py,sha256=sa6Zp5Kape1Zr4iYiNrnZxiHDnqM06ODk6yfWHom53w,13465
-digitalkin-0.3.2.dev7.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
+digitalkin/utils/schema_splitter.py,sha256=KMvYRHDHlwdhh_c6FJxkWvLStZo9Kbj-jd3pIGPZfxk,9317
+digitalkin-0.3.2.dev10.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
 modules/dynamic_setup_module.py,sha256=tKvUWZdlYZkfAgKR0mLuFcLiFGKpVgpsz10LeJ6B2QI,11410
 modules/minimal_llm_module.py,sha256=N9aIzZQI-miyH4AB4xTmGHpMvdSLnYyXNOD4Z3YFzis,11216
 modules/text_transform_module.py,sha256=MfhI_Ki1U6qk379ne6oazNEu4PhO4R3cRezEcr0nGPw,7251
+monitoring/digitalkin_observability/__init__.py,sha256=PXG4xbMbmiVNkIaVG899H9CHvcld7c0bSbaLHqKeDeo,1247
+monitoring/digitalkin_observability/http_server.py,sha256=kFbWFQvaqq2VVwwqDozV0pRPe7RkAXsa9xCWruMlOyk,4527
+monitoring/digitalkin_observability/interceptors.py,sha256=Sx6o0s8FSMfIGKVAtjVl7l-sIygP45od6eITl7DroBU,6219
+monitoring/digitalkin_observability/metrics.py,sha256=XhVRE8y6tbzJqxIzEHa5MT3I0k3hgpxjzZf_R6SyjbQ,7553
+monitoring/digitalkin_observability/prometheus.py,sha256=gDmM9ySaVwPAe7Yg84pLxmEKu0Cudezi8pK3RtqWs6g,5457
+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.dev7.dist-info/METADATA,sha256=NvyE2xOIuPXWTJBeqXd_ZMVI-EfKf6Ho-LZFTDVkIuc,29724
-digitalkin-0.3.2.dev7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-digitalkin-0.3.2.dev7.dist-info/top_level.txt,sha256=gcjqlyrZuLjIyxrOIavCQM_olpr6ND5kPKkZd2j0xGo,40
-digitalkin-0.3.2.dev7.dist-info/RECORD,,
+digitalkin-0.3.2.dev10.dist-info/METADATA,sha256=qExgKNDQPoWMU4v-k-tRwJyJ6on7nFhA83U4pqCgrwE,29725
+digitalkin-0.3.2.dev10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+digitalkin-0.3.2.dev10.dist-info/top_level.txt,sha256=AYVIesKrO0jnedQ-Muog9JBehG81WeTCNeOFoJgwsgE,51
+digitalkin-0.3.2.dev10.dist-info/RECORD,,

{digitalkin-0.3.2.dev7.dist-info → digitalkin-0.3.2.dev10.dist-info}/top_level.txt

@@ -1,4 +1,5 @@
 base_server
 digitalkin
 modules
+monitoring
 services

modules/archetype_with_tools_module.py

@@ -0,0 +1,244 @@
+"""Example archetype module with tool cache integration."""
+
+import logging
+from typing import Any, ClassVar, Literal
+
+from pydantic import BaseModel, Field
+
+from digitalkin.models.grpc_servers.models import ClientConfig, SecurityMode, ServerMode
+from digitalkin.models.module.module_context import ModuleContext
+from digitalkin.models.module.setup_types import SetupModel
+from digitalkin.models.module.tool_reference import (
+    ToolReference,
+    ToolReferenceConfig,
+    ToolSelectionMode,
+)
+from digitalkin.modules._base_module import BaseModule  # noqa: PLC2701
+from digitalkin.services.services_models import ServicesStrategy
+
+logging.basicConfig(
+    level=logging.DEBUG,
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+)
+logger = logging.getLogger(__name__)
+
+
+class MessageInputPayload(BaseModel):
+    """Message input payload."""
+
+    payload_type: Literal["message"] = "message"
+    user_prompt: str
+
+
+class ArchetypeInput(BaseModel):
+    """Archetype input."""
+
+    payload: MessageInputPayload = Field(discriminator="payload_type")
+
+
+class MessageOutputPayload(BaseModel):
+    """Message output payload."""
+
+    payload_type: Literal["message"] = "message"
+    response: str
+    tools_used: list[str] = Field(default_factory=list)
+
+
+class ArchetypeOutput(BaseModel):
+    """Archetype output."""
+
+    payload: MessageOutputPayload = Field(discriminator="payload_type")
+
+
+class ArchetypeSetup(SetupModel):
+    """Setup with tool references resolved during config setup."""
+
+    model_name: str = Field(
+        default="gpt-4",
+        json_schema_extra={"config": True},
+    )
+    temperature: float = Field(
+        default=0.7,
+        json_schema_extra={"config": True},
+    )
+
+    search_tool: ToolReference = Field(
+        default_factory=lambda: ToolReference(
+            config=ToolReferenceConfig(
+                mode=ToolSelectionMode.FIXED,
+                module_id="search-tool-v1",
+            )
+        ),
+        json_schema_extra={"config": True},
+    )
+
+    calculator_tool: ToolReference = Field(
+        default_factory=lambda: ToolReference(
+            config=ToolReferenceConfig(
+                mode=ToolSelectionMode.TAG,
+                tag="math-calculator",
+            )
+        ),
+        json_schema_extra={"config": True},
+    )
+
+    dynamic_tool: ToolReference = Field(
+        default_factory=lambda: ToolReference(
+            config=ToolReferenceConfig(
+                mode=ToolSelectionMode.DISCOVERABLE,
+            )
+        ),
+        json_schema_extra={"config": True},
+    )
+
+    system_prompt: str = Field(
+        default="You are a helpful assistant with access to tools.",
+        json_schema_extra={"hidden": True},
+    )
+
+
+class ArchetypeConfigSetup(BaseModel):
+    """Config setup model."""
+
+    additional_instructions: str | None = None
+
+
+class ArchetypeSecret(BaseModel):
+    """Secrets model."""
+
+
+client_config = ClientConfig(
+    host="[::]",
+    port=50152,
+    mode=ServerMode.ASYNC,
+    security=SecurityMode.INSECURE,
+    credentials=None,
+)
+
+
+class ArchetypeWithToolsModule(
+    BaseModule[
+        ArchetypeInput,
+        ArchetypeOutput,
+        ArchetypeSetup,
+        ArchetypeSecret,
+    ]
+):
+    """Archetype module demonstrating tool cache usage."""
+
+    name = "ArchetypeWithToolsModule"
+    description = "Archetype with tool cache integration"
+
+    config_setup_format = ArchetypeConfigSetup
+    input_format = ArchetypeInput
+    output_format = ArchetypeOutput
+    setup_format = ArchetypeSetup
+    secret_format = ArchetypeSecret
+
+    metadata: ClassVar[dict[str, Any]] = {
+        "name": "ArchetypeWithToolsModule",
+        "version": "1.0.0",
+        "tags": ["archetype", "tools"],
+    }
+
+    services_config_strategies: ClassVar[dict[str, ServicesStrategy | None]] = {}
+    services_config_params: ClassVar[dict[str, dict[str, Any | None] | None]] = {
+        "registry": {
+            "config": {},
+            "client_config": client_config,
+        },
+    }
+
+    async def run_config_setup(
+        self,
+        context: ModuleContext,  # noqa: ARG002
+        config_setup_data: ArchetypeSetup,
+    ) -> ArchetypeSetup:
+        """Custom config setup logic, runs in parallel with tool resolution.
+
+        Args:
+            context: Module context with services.
+            config_setup_data: Setup data being configured.
+
+        Returns:
+            Configured setup data.
+        """
+        logger.info("Running config setup for %s", self.name)
+        return config_setup_data
+
+    async def initialize(self, context: ModuleContext, setup_data: ArchetypeSetup) -> None:  # noqa: ARG002
+        """Initialize module.
+
+        Args:
+            context: Module context with services and tool cache.
+            setup_data: Setup data for the module.
+        """
+        logger.info("Initializing %s", self.name)
+        if context.tool_cache:
+            logger.info("Available tools: %s", context.tool_cache.list_tools())
+
+    async def run(
+        self,
+        input_data: ArchetypeInput,
+        setup_data: ArchetypeSetup,  # noqa: ARG002
+    ) -> None:
+        """Run module with tool cache lookups and call_module_by_id.
+
+        Args:
+            input_data: Input data to process.
+            setup_data: Setup configuration.
+        """
+        logger.info("Running %s", self.name)
+
+        tools_used: list[str] = []
+        tool_results: list[str] = []
+
+        # Get search tool from cache and call via call_module_by_id
+        search_info = self.context.tool_cache.get("search_tool")
+        if search_info:
+            tools_used.append(f"search:{search_info.module_id}")
+            async for response in self.context.call_module_by_id(
+                module_id=search_info.module_id,
+                input_data={"query": input_data.payload.user_prompt},
+                setup_id=self.context.session.setup_id,
+                mission_id=self.context.session.mission_id,
+            ):
+                tool_results.append(f"search_result: {response}")
+
+        # Get calculator tool from cache
+        calc_info = self.context.tool_cache.get("calculator_tool")
+        if calc_info:
+            tools_used.append(f"calculator:{calc_info.module_id}")
+            async for response in self.context.call_module_by_id(
+                module_id=calc_info.module_id,
+                input_data={"expression": "2 + 2"},
+                setup_id=self.context.session.setup_id,
+                mission_id=self.context.session.mission_id,
+            ):
+                tool_results.append(f"calc_result: {response}")
+
+        # Dynamic discovery via registry fallback for tools not in cache
+        dynamic_info = self.context.tool_cache.get(
+            "some_dynamic_tool",
+            registry=self.context.registry,
+        )
+        if dynamic_info:
+            tools_used.append(f"dynamic:{dynamic_info.module_id}")
+            async for response in self.context.call_module_by_id(
+                module_id=dynamic_info.module_id,
+                input_data={"prompt": input_data.payload.user_prompt},
+                setup_id=self.context.session.setup_id,
+                mission_id=self.context.session.mission_id,
+            ):
+                tool_results.append(f"dynamic_result: {response}")
+
+        response = MessageOutputPayload(
+            response=f"Processed: {input_data.payload.user_prompt} | Results: {len(tool_results)}",
+            tools_used=tools_used,
+        )
+
+        await self.context.callbacks.send_message(ArchetypeOutput(payload=response))
+
+    async def cleanup(self) -> None:
+        """Clean up resources."""
+        logger.info("Cleaning up %s", self.name)