digitalkin 1.0.0.dev25__tar.gz → 1.0.2.dev0__tar.gz

{digitalkin-1.0.0.dev25 → digitalkin-1.0.2.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 1.0.0.dev25
+Version: 1.0.2.dev0
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -453,14 +453,16 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ag-ui-protocol>=0.1.18
-Requires-Dist: agentic-mesh-protocol==1.0.0a2
+Requires-Dist: agentic-mesh-protocol==1.0.1.dev2
 Requires-Dist: anyio>=4.13.0
-Requires-Dist: grpcio-health-checking==1.81.0
-Requires-Dist: grpcio-reflection==1.81.0
-Requires-Dist: grpcio-status==1.81.0
-Requires-Dist: pydantic>=2.12.4
+Requires-Dist: grpcio-health-checking==1.82.1
+Requires-Dist: grpcio-reflection==1.82.1
+Requires-Dist: grpcio-status==1.82.1
 Requires-Dist: pydantic-settings>=2.14.1
+Requires-Dist: pydantic>=2.12.4
 Requires-Dist: redis[hiredis]<9,>=7.4.0
+Provides-Extra: agno
+Requires-Dist: agno>=2.6; extra == "agno"
 Provides-Extra: performance
 Requires-Dist: uvloop>=0.21; extra == "performance"
 Provides-Extra: profiling

{digitalkin-1.0.0.dev25 → digitalkin-1.0.2.dev0}/pyproject.toml

@@ -28,18 +28,19 @@
 
     dependencies = [
         "ag-ui-protocol>=0.1.18",
-        "agentic-mesh-protocol==1.0.0a2",
+        "agentic-mesh-protocol==1.0.1.dev2",
         "anyio>=4.13.0",
-        "grpcio-health-checking==1.81.0",
-        "grpcio-reflection==1.81.0",
-        "grpcio-status==1.81.0",
-        "pydantic>=2.12.4",
+        "grpcio-health-checking==1.82.1",
+        "grpcio-reflection==1.82.1",
+        "grpcio-status==1.82.1",
         "pydantic-settings>=2.14.1",
+        "pydantic>=2.12.4",
         "redis[hiredis]>=7.4.0,<9",
     ]
-    version = "1.0.0.dev25"
+    version = "1.0.2.dev0"
 
     [project.optional-dependencies]
+        agno = [ "agno>=2.6" ]
         performance = [ "uvloop>=0.21" ]
         profiling = [
             "pyinstrument>=5.1.2",
@@ -56,6 +57,22 @@
         name  = "DigitalKin.ai"
 
 [dependency-groups]
+    tests = [
+        "fakeredis[lua]>=2.35.1",
+        "freezegun>=1.5.5",
+        "grpcio-testing>=1.81.0",
+        "hdrhistogram>=0.10.3",
+        "hypothesis>=6.152.8",
+        "objgraph>=3.6",
+        "psutil>=7.2.2",
+        "pytest-asyncio>=1.3.0",
+        "pytest-benchmark>=5.2.3",
+        "pytest-cov>=7.1.0",
+        "pytest-html>=4.2.0",
+        "pytest-json-report>=1.5.0",
+        "pytest-timeout>=2.4.0",
+        "pytest>=9.0.3",
+    ]
     dev = [
         "build>=1.5.0",
         "bump-my-version>=1.3.0",
@@ -67,7 +84,7 @@
         "twine>=6.2.0",
         "types-grpcio-health-checking>=1.0.0.20260518",
         "types-grpcio-reflection>=1.0.0.20260508",
-        "types-grpcio>=1.0.0.20260518",
+        "types-grpcio>=1.82.1.20260711",
         "types-protobuf>=7.34.1.20260518",
         "typos>=1.48.0",
     ]
@@ -95,22 +112,6 @@
         "mkdocstrings>=1.0.4",
         "tomli>=2.4.1",
     ]
-    tests = [
-        "fakeredis[lua]>=2.35.1",
-        "freezegun>=1.5.5",
-        "grpcio-testing>=1.81.0",
-        "hdrhistogram>=0.10.3",
-        "hypothesis>=6.152.8",
-        "objgraph>=3.6",
-        "psutil>=7.2.2",
-        "pytest-asyncio>=1.3.0",
-        "pytest-benchmark>=4.0",
-        "pytest-cov>=7.1.0",
-        "pytest-html>=4.2.0",
-        "pytest-json-report>=1.5.0",
-        "pytest-timeout>=2.4.0",
-        "pytest>=9.0.3",
-    ]
 [tool.setuptools]
     package-dir = { "" = "src" }
 
@@ -146,12 +147,12 @@
         "buck-out",
         "build",
         "dist",
-        "scripts",
         "docs/*",
         "examples/*",
         "factory.py",
         "generate_certificates.py",
         "node_modules",
+        "scripts",
         "tests/*",
         "venv",
     ]
@@ -268,10 +269,10 @@
         skip-magic-trailing-comma = false
 
 [tool.mypy]
-    exclude                = [ "examples", "tests", "scripts" ]
+    exclude                = [ "examples", "scripts", "tests" ]
     ignore_missing_imports = true
-    warn_unused_ignores    = true
     warn_redundant_casts   = true
+    warn_unused_ignores    = true
 
 [tool.pytest.ini_options]
 

{digitalkin-1.0.0.dev25 → digitalkin-1.0.2.dev0}/src/digitalkin/__version__.py

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

{digitalkin-1.0.0.dev25 → digitalkin-1.0.2.dev0}/src/digitalkin/community/agno/__init__.py

@@ -12,6 +12,19 @@ on top of the Agno agent framework. Exports:
   paused Agno run via the module's
   :class:`~digitalkin.services.storage.StorageStrategy` and resumes it
   when the front replies with a ``ToolMessage``.
+- :class:`ToolCallMetadata`, :class:`ToolOutputMetadata` — metadata models
+  for module tool calls made through the toolkit.
+
+``ModuleToolkit`` (Agno Toolkit wrapping a module's remote tools) requires the
+optional ``agno`` dependency at import time and is therefore NOT exported here;
+import it directly::
+
+    from digitalkin.community.agno.module_toolkit import ModuleToolkit
+
+Default agent toolkits (``ChatHistoryTools``, ``UserProfileTools``,
+``RegistryTools``, ``DefaultToolkits``) live in
+:mod:`digitalkin.community.agno.toolkits` — imported separately because they
+require the optional ``agno`` dependency at import time.
 """
 
 from digitalkin.community.agno.agno_adapter import AgnoStreamAdapter
@@ -23,7 +36,7 @@ from digitalkin.community.agno.hitl import (
     PausedRunRecord,
     PausedRunStore,
 )
-from digitalkin.community.agno.models import PauseInfo
+from digitalkin.community.agno.models import PauseInfo, ToolCallMetadata, ToolOutputMetadata
 
 __all__ = [
     "HITL_STORAGE_CONFIG",
@@ -34,4 +47,6 @@ __all__ = [
     "PauseInfo",
     "PausedRunRecord",
     "PausedRunStore",
+    "ToolCallMetadata",
+    "ToolOutputMetadata",
 ]

digitalkin-1.0.2.dev0/src/digitalkin/community/agno/models.py

@@ -0,0 +1,164 @@
+"""Models for the Agno community integration."""
+
+from typing import Any
+
+from ag_ui.core.types import Message as AgUiMessage
+from pydantic import BaseModel, Field
+
+
+class PauseInfo(BaseModel):
+    """Summary of a paused Agno run.
+
+    Returned by :meth:`AgnoHitlRunner.run` and related methods whenever
+    the run paused on one or more external tool calls. Callers typically
+    use it to emit the AG-UI awaiting-tool-result event to the front.
+
+    ``new_messages`` carries the AG-UI messages generated by Agno during
+    the paused run (user echoes, the assistant message with ``tool_calls``,
+    and any tool results emitted before the pause). It's provided because
+    Agno does not emit stream events from which the front can reconstruct
+    the assistant-with-tool-calls message — in particular, when the LLM
+    goes straight from reasoning to a frontend tool call without emitting
+    any text. Consumers typically push these messages to the front via a
+    :class:`~ag_ui.core.events.MessagesSnapshotEvent` so the client has an
+    authoritative view of the conversation.
+    """
+
+    thread_id: str
+    run_id: str
+    pending_tool_call_ids: list[str]
+    new_messages: list[AgUiMessage] = Field(default_factory=list)
+
+
+class ToolOutputMetadata(BaseModel):
+    """Cost and execution metadata reported by a tool module itself.
+
+    Enables accurate cost aggregation across modules.
+
+    Attributes:
+        response_time_ms: Time taken for the API call(s) in milliseconds.
+        api_calls_made: Number of API calls made by the tool.
+        cost_estimate_usd: Estimated cost in USD.
+        tavily_credits_used: Tavily API credits consumed (if applicable).
+        search_depth: Search depth used (basic, advanced, etc.).
+        include_raw_content: Whether raw content was requested.
+        queries_count: Number of queries executed.
+        results_returned: Number of results returned.
+        urls_processed: Number of URLs processed (for extract operations).
+    """
+
+    response_time_ms: float | None = Field(default=None, description="API response time in ms")
+    api_calls_made: int = Field(default=1, description="Number of API calls made")
+    cost_estimate_usd: float | None = Field(default=None, description="Estimated cost in USD")
+    tavily_credits_used: int | None = Field(default=None, description="Tavily credits consumed")
+    search_depth: str | None = Field(default=None, description="Search depth used")
+    include_raw_content: bool = Field(default=False, description="Whether raw content was requested")
+    queries_count: int = Field(default=1, description="Number of queries executed")
+    results_returned: int = Field(default=0, description="Number of results returned")
+    urls_processed: int = Field(default=0, description="Number of URLs processed")
+
+
+class ToolCallMetadata(BaseModel):
+    """Metadata from a module tool call including cost and timing information.
+
+    Attributes:
+        module_id: The SDK module ID that was called (e.g., "modules:tool_websearch").
+        success: Whether the tool call completed successfully.
+        duration_ms: Execution time in milliseconds.
+        cost_tracked: Whether cost metadata is available from the tool.
+        error: Error message if the call failed, None otherwise.
+        input_kwargs: The kwargs passed to the tool (for debugging).
+        output_summary: Brief summary of the output (truncated for logs).
+        tool_metadata: Metadata returned by the tool itself (cost, API calls, etc.).
+    """
+
+    module_id: str = Field(..., description="The SDK module ID that was called")
+    success: bool = Field(..., description="Whether the tool call succeeded")
+    duration_ms: float = Field(..., description="Execution time in milliseconds")
+    cost_tracked: bool = Field(default=False, description="Whether cost metadata is available from tool")
+    error: str | None = Field(default=None, description="Error message if failed")
+    input_kwargs: dict[str, Any] | None = Field(
+        default=None,
+        description="Input kwargs passed to the tool for debugging",
+    )
+    output_summary: str | None = Field(
+        default=None,
+        description="Brief summary of the output (truncated)",
+    )
+    tool_metadata: ToolOutputMetadata | None = Field(
+        default=None,
+        description="Metadata returned by the tool module (cost, API calls, etc.)",
+    )
+
+    def to_success_dict(self) -> dict[str, Any]:
+        """Convert metadata to dict for successful responses.
+
+        Returns:
+            Dictionary with success-relevant fields (error excluded).
+        """
+        return self.model_dump(exclude={"error"})
+
+    def to_error_dict(self) -> dict[str, Any]:
+        """Convert metadata to dict for error responses.
+
+        Returns:
+            Dictionary with error-relevant fields.
+        """
+        return self.model_dump(include={"module_id", "success", "duration_ms", "error"})
+
+    def to_log_dict(self) -> dict[str, Any]:
+        """Convert metadata to dict for logging.
+
+        Returns:
+            Dictionary suitable for log extra data, large kwargs truncated.
+        """
+        data: dict[str, Any] = {
+            "module_id": self.module_id,
+            "success": self.success,
+            "duration_ms": self.duration_ms,
+            "cost_tracked": self.cost_tracked,
+        }
+        if self.error:
+            data["error"] = self.error
+        if self.input_kwargs:
+            max_len = 100
+            data["input_kwargs"] = {
+                k: (str(v)[:max_len] + "..." if len(str(v)) > max_len else v) for k, v in self.input_kwargs.items()
+            }
+        if self.output_summary:
+            data["output_summary"] = self.output_summary
+        if self.tool_metadata:
+            data["tool_metadata"] = self.tool_metadata.model_dump(exclude_none=True)
+        return data
+
+    @staticmethod
+    def extract_tool_metadata(output: dict[str, Any]) -> "ToolOutputMetadata | None":
+        """Extract tool metadata from a tool's output if present.
+
+        Args:
+            output: The tool output dictionary.
+
+        Returns:
+            ToolOutputMetadata if metadata is present, None otherwise.
+        """
+        if not isinstance(output, dict):
+            return None
+
+        metadata = output.get("metadata")
+        if not metadata or not isinstance(metadata, dict):
+            return None
+
+        try:
+            return ToolOutputMetadata(
+                response_time_ms=metadata.get("response_time_ms"),
+                api_calls_made=metadata.get("api_calls_made", 1),
+                cost_estimate_usd=metadata.get("cost_estimate_usd"),
+                tavily_credits_used=metadata.get("tavily_credits_used"),
+                search_depth=metadata.get("search_depth"),
+                include_raw_content=metadata.get("include_raw_content", False),
+                queries_count=metadata.get("queries_count", 1),
+                results_returned=metadata.get("results_returned", 0),
+                urls_processed=metadata.get("urls_processed", 0),
+            )
+        except Exception:
+            return None