digitalkin 0.3.2.dev5__tar.gz → 0.3.2.dev6__tar.gz

{digitalkin-0.3.2.dev5 → digitalkin-0.3.2.dev6}/PKG-INFO

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

@@ -12,7 +12,7 @@
 
     keywords = [ "digitalkin", "kin", "agent", "gprc", "sdk" ]
 
-    version = "0.3.2.dev5"
+    version = "0.3.2.dev6"
     classifiers = [
         "Development Status :: 3 - Alpha",
         "Intended Audience :: Developers",

{digitalkin-0.3.2.dev5 → digitalkin-0.3.2.dev6}/src/digitalkin/__version__.py

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

{digitalkin-0.3.2.dev5 → digitalkin-0.3.2.dev6}/src/digitalkin/grpc_servers/module_server.py

@@ -174,7 +174,7 @@ class ModuleServer(BaseServer):
             "Attempting to register module with registry",
             extra={
                 "module_id": module_id,
-                "address": self.server_config.address,
+                "address": self.server_config.host,
                 "port": self.server_config.port,
                 "version": version,
                 "registry_address": self.server_config.registry_address,
@@ -183,7 +183,7 @@ class ModuleServer(BaseServer):
 
         result = self.registry.register(
             module_id=module_id,
-            address=self.server_config.address,
+            address=self.server_config.host,
             port=self.server_config.port,
             version=version,
         )
@@ -193,7 +193,7 @@ class ModuleServer(BaseServer):
                 "Module registered successfully",
                 extra={
                     "module_id": result.module_id,
-                    "address": self.server_config.address,
+                    "address": self.server_config.host,
                     "port": self.server_config.port,
                     "registry_address": self.server_config.registry_address,
                 },

{digitalkin-0.3.2.dev5 → digitalkin-0.3.2.dev6}/src/digitalkin/grpc_servers/utils/utility_schema_extender.py

@@ -67,7 +67,8 @@ class UtilitySchemaExtender:
         original_annotation = base_model.model_fields["root"].annotation
         original_types = cls._extract_union_types(original_annotation)
         extended_types = (*original_types, *cls._output_protocols)
-        extended_root = Annotated[extended_types, Field(discriminator="protocol")]  # type: ignore[valid-type]
+        union_type = Union[extended_types]  # type: ignore[valid-type] # noqa: UP007
+        extended_root = Annotated[union_type, Field(discriminator="protocol")]  # type: ignore[valid-type]
         return create_model(
             f"{base_model.__name__}Utilities",
             __base__=DataModel,
@@ -88,7 +89,8 @@ class UtilitySchemaExtender:
         original_annotation = base_model.model_fields["root"].annotation
         original_types = cls._extract_union_types(original_annotation)
         extended_types = (*original_types, *cls._input_protocols)
-        extended_root = Annotated[extended_types, Field(discriminator="protocol")]  # type: ignore[valid-type]
+        union_type = Union[extended_types]  # type: ignore[valid-type] # noqa: UP007
+        extended_root = Annotated[union_type, Field(discriminator="protocol")]  # type: ignore[valid-type]
         return create_model(
             f"{base_model.__name__}Utilities",
             __base__=DataModel,

{digitalkin-0.3.2.dev5 → digitalkin-0.3.2.dev6}/src/digitalkin/models/module/module_context.py

@@ -6,6 +6,7 @@ from types import SimpleNamespace
 from typing import TYPE_CHECKING, Any
 from zoneinfo import ZoneInfo
 
+from digitalkin.models.module.module_helpers import ModuleHelpers
 from digitalkin.models.module.tool_cache import ToolCache
 from digitalkin.services.agent.agent_strategy import AgentStrategy
 from digitalkin.services.communication.communication_strategy import CommunicationStrategy
@@ -100,7 +101,7 @@ class ModuleContext:
     session: Session
     callbacks: SimpleNamespace
     metadata: SimpleNamespace
-    helpers: SimpleNamespace
+    helpers: ModuleHelpers
     state: SimpleNamespace = SimpleNamespace()
     tool_cache: ToolCache
 
@@ -152,7 +153,7 @@ class ModuleContext:
 
         self.metadata = SimpleNamespace(**metadata)
         self.session = Session(**session)
-        self.helpers = SimpleNamespace(**helpers)
+        self.helpers = ModuleHelpers(context=self, **helpers)
         self.callbacks = SimpleNamespace(**callbacks)
         self.tool_cache = tool_cache or ToolCache()
 

digitalkin-0.3.2.dev6/src/digitalkin/models/module/module_helpers.py

@@ -0,0 +1,189 @@
+"""Module helpers for inter-module communication."""
+
+from collections.abc import AsyncGenerator, Callable, Coroutine
+from types import SimpleNamespace
+from typing import TYPE_CHECKING, Any
+
+from digitalkin.logger import logger
+
+if TYPE_CHECKING:
+    from digitalkin.models.module.module_context import ModuleContext
+
+
+class ModuleHelpers(SimpleNamespace):
+    """Helpers for module-to-module communication.
+
+    Extends SimpleNamespace to allow dynamic attribute assignment
+    while providing built-in helper methods.
+    """
+
+    def __init__(self, context: "ModuleContext", **kwargs: dict[str, Any]) -> None:
+        """Initialize helpers with context reference.
+
+        Args:
+            context: ModuleContext providing access to services.
+            **kwargs: Additional attributes to set on the namespace.
+        """
+        super().__init__(**kwargs)
+        self._context = context
+
+    async def call_module_by_id(
+        self,
+        module_id: str,
+        input_data: dict,
+        setup_id: str,
+        mission_id: str,
+        callback: Callable[[dict], Coroutine[Any, Any, None]] | None = None,
+    ) -> AsyncGenerator[dict, None]:
+        """Call a module by ID, discovering address/port from registry.
+
+        Args:
+            module_id: Module identifier to look up in registry
+            input_data: Input data as dictionary
+            setup_id: Setup configuration ID
+            mission_id: Mission context ID
+            callback: Optional callback for each response
+
+        Yields:
+            Streaming responses from module as dictionaries
+        """
+        module_info = self._context.registry.discover_by_id(module_id)
+
+        logger.debug(
+            "Calling module by ID",
+            extra={
+                "module_id": module_id,
+                "address": module_info.address,
+                "port": module_info.port,
+            },
+        )
+
+        async for response in self._context.communication.call_module(
+            module_address=module_info.address,
+            module_port=module_info.port,
+            input_data=input_data,
+            setup_id=setup_id,
+            mission_id=mission_id,
+            callback=callback,
+        ):
+            yield response
+
+    async def get_module_schemas_by_id(
+        self,
+        module_id: str,
+        *,
+        llm_format: bool = False,
+    ) -> dict[str, dict]:
+        """Get module schemas by ID, discovering address/port from registry.
+
+        Args:
+            module_id: Module identifier to look up in registry
+            llm_format: If True, return LLM-optimized schema format
+
+        Returns:
+            Dictionary containing schemas: {"input": ..., "output": ..., "setup": ..., "secret": ...}
+        """
+        module_info = self._context.registry.discover_by_id(module_id)
+
+        logger.debug(
+            "Getting module schemas by ID",
+            extra={
+                "module_id": module_id,
+                "address": module_info.address,
+                "port": module_info.port,
+            },
+        )
+
+        return await self._context.communication.get_module_schemas(
+            module_address=module_info.address,
+            module_port=module_info.port,
+            llm_format=llm_format,
+        )
+
+    async def create_openai_style_tool(self, module_id: str) -> dict[str, Any] | None:
+        """Create OpenAI-style function calling schema for a tool.
+
+        Uses tool cache (fast path) with registry fallback. Fetches the tool's
+        input schema and wraps it in OpenAI function calling format.
+
+        Args:
+            module_id: Module ID to look up (checks cache first, then registry)
+
+        Returns:
+            OpenAI-style tool schema if found:
+            {
+                "type": "function",
+                "function": {
+                    "name": "...",
+                    "description": "...",
+                    "parameters": {...}  # Input JSON Schema
+                }
+            }
+            None if tool not found.
+        """
+        module_info = self._context.tool_cache.check_and_get(module_id, self._context.registry)
+        if not module_info:
+            return None
+
+        schemas = await self._context.communication.get_module_schemas(
+            module_address=module_info.address,
+            module_port=module_info.port,
+            llm_format=True,
+        )
+
+        return {
+            "type": "function",
+            "function": {
+                "name": module_info.name or module_info.module_id,
+                "description": module_info.documentation or "",
+                "parameters": schemas["input"],
+            },
+        }
+
+    def create_tool_function(
+        self,
+        module_id: str,
+    ) -> Callable[..., AsyncGenerator[dict, None]] | None:
+        """Create async generator function for a tool.
+
+        Returns an async generator that calls the remote tool module via gRPC
+        and yields each response as it arrives until end_of_stream or gRPC ends.
+
+        Args:
+            module_id: Module ID to look up (checks cache first, then registry)
+
+        Returns:
+            Async generator function if tool found, None otherwise.
+            The function accepts **kwargs matching the tool's input schema
+            and yields dict responses.
+        """
+        module_info = self._context.tool_cache.check_and_get(module_id, self._context.registry)
+        if not module_info:
+            return None
+
+        # Capture references for closure
+        communication = self._context.communication
+        session = self._context.session
+        address = module_info.address
+        port = module_info.port
+
+        async def tool_function(**kwargs: Any) -> AsyncGenerator[dict, None]:  # noqa: ANN401
+            """Call remote tool module and yield responses.
+
+            Yields:
+                dict: Each response from the module until end_of_stream.
+            """
+            wrapped_input = {"root": kwargs}
+            async for response in communication.call_module(
+                module_address=address,
+                module_port=port,
+                input_data=wrapped_input,
+                setup_id=session.setup_id,
+                mission_id=session.mission_id,
+            ):
+                yield response
+
+        tool_function.__name__ = module_info.name or module_info.module_id
+        tool_function.__doc__ = module_info.documentation or ""
+
+        return tool_function

{digitalkin-0.3.2.dev5 → digitalkin-0.3.2.dev6}/src/digitalkin/services/communication/grpc_communication.py

@@ -187,6 +187,17 @@ class GrpcCommunication(CommunicationStrategy, GrpcClientWrapper):
                 # Convert protobuf Struct to dict
                 output_dict = json_format.MessageToDict(response.output)
 
+                # Check for end_of_stream signal
+                if output_dict.get("root", {}).get("protocol") == "end_of_stream":
+                    logger.debug(
+                        "End of stream received",
+                        extra={
+                            "module_address": module_address,
+                            "module_port": module_port,
+                        },
+                    )
+                    break
+
                 # Add job_id and success flag
                 response_dict = {
                     "success": response.success,

{digitalkin-0.3.2.dev5 → digitalkin-0.3.2.dev6}/src/digitalkin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.2.dev5
+Version: 0.3.2.dev6
 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.dev5 → digitalkin-0.3.2.dev6}/src/digitalkin.egg-info/SOURCES.txt

@@ -69,6 +69,7 @@ src/digitalkin/models/module/__init__.py
 src/digitalkin/models/module/base_types.py
 src/digitalkin/models/module/module.py
 src/digitalkin/models/module/module_context.py
+src/digitalkin/models/module/module_helpers.py
 src/digitalkin/models/module/module_types.py
 src/digitalkin/models/module/setup_types.py
 src/digitalkin/models/module/tool_cache.py