digitalkin 0.3.4.dev0__tar.gz → 0.3.4.dev1__tar.gz

{digitalkin-0.3.4.dev0 → digitalkin-0.3.4.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.4.dev0
+Version: 0.3.4.dev1
 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.4.dev0 → digitalkin-0.3.4.dev1}/pyproject.toml

@@ -34,7 +34,7 @@
         "grpcio-status==1.78.0",
         "pydantic==2.12.5",
     ]
-    version = "0.3.4.dev0"
+    version = "0.3.4.dev1"
 
     [project.optional-dependencies]
         profiling = [

{digitalkin-0.3.4.dev0 → digitalkin-0.3.4.dev1}/src/digitalkin/__version__.py

@@ -5,4 +5,4 @@ from importlib.metadata import PackageNotFoundError, version
 try:
     __version__ = version("digitalkin")
 except PackageNotFoundError:
-    __version__ = "0.3.4.dev0"
+    __version__ = "0.3.4.dev1"

{digitalkin-0.3.4.dev0 → digitalkin-0.3.4.dev1}/src/digitalkin/models/module/__init__.py

@@ -12,7 +12,6 @@ from digitalkin.models.module.tool_cache import (
     ToolCache,
     ToolDefinition,
     ToolModuleInfo,
-    ToolParameter,
 )
 from digitalkin.models.module.tool_reference import (
     ToolReference,
@@ -38,7 +37,6 @@ __all__ = [
     "ToolCache",
     "ToolDefinition",
     "ToolModuleInfo",
-    "ToolParameter",
     "ToolReference",
     "ToolSelection",
     "UtilityProtocol",

{digitalkin-0.3.4.dev0 → digitalkin-0.3.4.dev1}/src/digitalkin/models/module/module_context.py

@@ -9,7 +9,7 @@ from zoneinfo import ZoneInfo
 
 from digitalkin.logger import logger
 from digitalkin.models.module.request_metadata import RequestMetadata
-from digitalkin.models.module.tool_cache import ToolCache, ToolDefinition, ToolModuleInfo, ToolParameter
+from digitalkin.models.module.tool_cache import ToolCache, ToolDefinition, ToolModuleInfo
 from digitalkin.services.agent.agent_strategy import AgentStrategy
 from digitalkin.services.communication.communication_strategy import CommunicationStrategy
 from digitalkin.services.cost.cost_strategy import CostStrategy
@@ -225,29 +225,13 @@ class ModuleContext:
                     "toolkit_name": tool_module_info.tool_name or "undefined",
                     "name": tool_module_info.slug + "__" + tool_def.name,
                     "description": tool_def.description + cost_description,
-                    "parameters": ModuleContext._build_parameters_schema(tool_def.parameters),
+                    "parameters": tool_def.parameters_schema,
                 },
                 "cost_info": cost_info,
             }
             for tool_def in tool_module_info.tools
         ]
 
-    @staticmethod
-    def _build_parameters_schema(params: list[ToolParameter]) -> dict[str, Any]:
-        """Convert ToolParameter list to JSON Schema.
-
-        Args:
-            params: List of tool parameters.
-
-        Returns:
-            JSON Schema object with properties and required fields.
-        """
-        return {
-            "type": "object",
-            "properties": {p.name: {"type": p.type, "description": p.description or ""} for p in params},
-            "required": [p.name for p in params if p.required],
-        }
-
     @staticmethod
     def _build_cost_info(cost_config: dict[str, Any]) -> dict[str, Any]:
         """Build cost information structure for tool metadata.

{digitalkin-0.3.4.dev0 → digitalkin-0.3.4.dev1}/src/digitalkin/models/module/tool_cache.py

@@ -7,6 +7,7 @@ from pydantic import BaseModel, Field
 
 from digitalkin.logger import logger
 from digitalkin.models.services.registry import ModuleInfo
+from digitalkin.utils.llm_ready_schema import inline_refs
 
 
 class SelectedTool(BaseModel):
@@ -22,40 +23,30 @@ if TYPE_CHECKING:
     from digitalkin.services.communication import CommunicationStrategy
 
 
-class ToolParameter(BaseModel):
-    """Definition of a single tool parameter.
-
-    Attributes:
-        name: Parameter name.
-        type: JSON Schema type (string, integer, number, boolean, array, object).
-        description: Parameter description for the LLM.
-        required: Whether this parameter is required.
-        enum: Optional list of allowed values.
-        items: Optional schema for array item types.
-        properties: Optional schema for object properties.
-    """
-
-    name: str
-    type: str
-    description: str
-    required: bool = True
-    enum: list[str] | None = None
-    items: dict[str, Any] | None = None
-    properties: dict[str, Any] | None = None
-
-
 class ToolDefinition(BaseModel):
-    """Complete definition of an LLM tool with grouped parameters.
+    """Complete definition of an LLM tool with resolved JSON Schema parameters.
 
     Attributes:
         name: Tool name (from protocol const or trigger class name).
         description: Tool description (from trigger docstring).
-        parameters: List of parameter definitions.
+        parameters_schema: JSON Schema object describing the tool's parameters.
     """
 
     name: str
     description: str
-    parameters: list[ToolParameter] = Field(default_factory=list)
+    parameters_schema: dict[str, Any] = Field(
+        default_factory=lambda: {"type": "object", "properties": {}, "required": []}
+    )
+
+    @property
+    def parameter_names(self) -> set[str]:
+        """Return the set of parameter names from the schema."""
+        return set[str](self.parameters_schema.get("properties", {}).keys())
+
+    @property
+    def parameter_count(self) -> int:
+        """Return the number of parameters in the schema."""
+        return len(self.parameters_schema.get("properties", {}))
 
 
 class ToolModuleInfo(ModuleInfo):
@@ -192,14 +183,44 @@ async def module_info_to_tool_module_info(
     )
 
 
+def _build_parameters_from_schema(def_schema: dict[str, Any]) -> dict[str, Any]:
+    """Build parameters_schema directly from an inlined JSON Schema.
+
+    Extracts tool parameters from the trigger's JSON Schema, skipping
+    internal fields (``protocol``, ``created_at``).
+
+    Args:
+        def_schema: JSON Schema for the trigger with all ``$ref`` already inlined.
+
+    Returns:
+        JSON Schema dict with properties and required fields.
+    """
+    properties = def_schema.get("properties", {})
+    required_fields = set[Any](def_schema.get("required", []))
+    param_properties: dict[str, Any] = {}
+    required_list: list[str] = []
+
+    for prop_name, prop_info in properties.items():
+        if prop_name in {"protocol", "created_at"}:
+            continue
+        param_properties[prop_name] = dict[Any, Any](prop_info.items())
+        if prop_name in required_fields:
+            required_list.append(prop_name)
+
+    return {"type": "object", "properties": param_properties, "required": required_list}
+
+
 def _extract_tools_from_schema(schema: dict[str, Any]) -> list[ToolDefinition]:
     """Extract tool definitions from a discriminated union input schema.
 
+    Inlines ``$ref`` references and extracts parameters directly from the
+    JSON Schema — no intermediate Python model reconstruction needed.
+
     Args:
         schema: JSON schema with $defs containing protocol-based types.
 
     Returns:
-        List of ToolDefinition with parameters grouped by trigger.
+        List of ToolDefinition with parameters_schema per trigger.
     """
     tools: list[ToolDefinition] = []
     defs = schema.get("$defs", {})
@@ -220,33 +241,18 @@ def _extract_tools_from_schema(schema: dict[str, Any]) -> list[ToolDefinition]:
         if "const" not in protocol_prop:
             continue
 
-        # Extract tool-level info from trigger
         tool_name = protocol_prop.get("const", def_name)
         tool_description = def_schema.get("description", "")
 
-        required_fields = set[Any](def_schema.get("required", []))
-        parameters: list[ToolParameter] = []
-
-        for prop_name, prop_info in properties.items():
-            if prop_name in {"protocol", "created_at"}:
-                continue
-
-            param = ToolParameter(
-                name=prop_name,
-                type=prop_info.get("type", "string"),
-                description=prop_info.get("description", ""),
-                required=prop_name in required_fields,
-                enum=prop_info.get("enum"),
-                items=prop_info.get("items"),
-                properties=prop_info.get("properties"),
-            )
-            parameters.append(param)
+        # Inline $ref references so properties are self-contained
+        inlined = inline_refs({**def_schema, "$defs": defs})
+        parameters_schema = _build_parameters_from_schema(inlined)
 
         tools.append(
             ToolDefinition(
                 name=tool_name,
                 description=tool_description,
-                parameters=parameters,
+                parameters_schema=parameters_schema,
             )
         )
 

{digitalkin-0.3.4.dev0 → digitalkin-0.3.4.dev1}/src/digitalkin/modules/_base_module.py

@@ -626,12 +626,18 @@ class BaseModule(  # Module SDK base class requires many public methods # noqa:
                         await handler.flush_file_history(self.context)
             except Exception:
                 logger.warning("Failed to flush handler history during stop", exc_info=True)
-            await self.context.callbacks.send_message(
-                DataModel(
-                    root=EndOfStreamOutput(),
-                    annotations={"role": BaseRole.SYSTEM},
+            try:
+                await self.context.callbacks.send_message(
+                    DataModel[EndOfStreamOutput](
+                        root=EndOfStreamOutput(),
+                        annotations={"role": BaseRole.SYSTEM},
+                    )
+                )
+            except AttributeError:
+                logger.warning(
+                    "send_message callback not set, skipping end-of-stream"
+                    " (expected for start_config_setup which does not register send_message)"
                 )
-            )
             self._status = ModuleStatus.STOPPED
             logger.debug("Module %s cleaned", self.name)
         except Exception:

{digitalkin-0.3.4.dev0 → digitalkin-0.3.4.dev1}/src/digitalkin.egg-info/PKG-INFO

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