digitalkin 0.3.1.dev1__tar.gz → 0.3.2a2__tar.gz

{digitalkin-0.3.1.dev1 → digitalkin-0.3.2a2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.1.dev1
+Version: 0.3.2a2
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -434,35 +434,35 @@ License: Attribution-NonCommercial-ShareAlike 4.0 International
         Creative Commons may be contacted at creativecommons.org.
         https://creativecommons.org/licenses/by-nc-sa/4.0/
         
-Project-URL: Homepage, https://github.com/DigitalKin-ai/digitalkin
 Project-URL: Documentation, https://github.com/DigitalKin-ai/digitalkin
+Project-URL: Homepage, https://github.com/DigitalKin-ai/digitalkin
 Project-URL: Issues, https://github.com/DigitalKin-ai/digitalkin/issues
-Keywords: digitalkin,kin,agent,gprc,sdk
+Keywords: agent,digitalkin,gprc,kin,sdk
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
+Classifier: License :: Other/Proprietary License
 Classifier: Operating System :: OS Independent
-Classifier: Topic :: Software Development :: Libraries
-Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Python
+Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: digitalkin-proto==0.2.0.dev4
+Requires-Dist: agentic-mesh-protocol==0.2.1.dev1
 Requires-Dist: grpcio-health-checking>=1.76.0
 Requires-Dist: grpcio-reflection>=1.76.0
 Requires-Dist: grpcio-status>=1.76.0
-Requires-Dist: pydantic>=2.12.4
-Requires-Dist: surrealdb>=1.0.6
+Requires-Dist: pydantic>=2.12.5
+Requires-Dist: surrealdb>=1.0.7
 Provides-Extra: taskiq
-Requires-Dist: rstream>=0.40.0; extra == "taskiq"
+Requires-Dist: rstream>=0.40.1; extra == "taskiq"
 Requires-Dist: taskiq-aio-pika>=0.5.0; extra == "taskiq"
-Requires-Dist: taskiq-redis>=1.1.2; extra == "taskiq"
-Requires-Dist: taskiq[reload]>=0.11.20; extra == "taskiq"
+Requires-Dist: taskiq-redis>=1.2.0; extra == "taskiq"
+Requires-Dist: taskiq[reload]>=0.12.1; extra == "taskiq"
 Dynamic: license-file
 
 # DigitalKin Python SDK

{digitalkin-0.3.1.dev1 → digitalkin-0.3.2a2}/examples/base_server/server_async_insecure.py

@@ -9,8 +9,9 @@ from pathlib import Path
 # Add parent directory to path to enable imports
 sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent))
 
-from digitalkin.grpc_servers._base_server import BaseServer
 from digitalkin.grpc_servers.utils.models import SecurityMode, ServerConfig, ServerMode
+
+from digitalkin.grpc_servers._base_server import BaseServer
 from examples.base_server.mock.mock_pb2 import DESCRIPTOR, HelloReply  # type: ignore
 from examples.base_server.mock.mock_pb2_grpc import (
     Greeter,
@@ -30,7 +31,7 @@ class AsyncGreeterImpl(Greeter):
 
     async def SayHello(self, request, context):  # noqa: N802
         """Asynchronous implementation of SayHello method."""
-        logger.info(f"Received request object: {request}")
+        logger.info("Received request object: %s", request)
         logger.info(f"Request attributes: {vars(request)}")
         logger.info(f"Received request with name: {request.name}")
 
@@ -40,7 +41,7 @@ class AsyncGreeterImpl(Greeter):
             name = "unknown"
             # Check context metadata
             for key, value in context.invocation_metadata():
-                logger.info(f"Metadata: {key}={value}")
+                logger.info("Metadata: %s=%s", key, value)
                 if key.lower() == "name":
                     name = value
 
@@ -97,7 +98,7 @@ async def main_async() -> int:
         # as the KeyboardInterrupt usually breaks out of asyncio.run()
         logger.info("Server stopping due to keyboard interrupt...")
     except Exception as e:
-        logger.exception(f"Error running server: {e}")
+        logger.exception("Error running server: %s", e)
         return 1
     finally:
         # Clean up resources if server was started
@@ -116,7 +117,7 @@ def main():
         logger.info("Server stopped by keyboard interrupt")
         return 0  # Clean exit
     except Exception as e:
-        logger.exception(f"Fatal error: {e}")
+        logger.exception("Fatal error: %s", e)
         return 1
 
 

{digitalkin-0.3.1.dev1 → digitalkin-0.3.2a2}/examples/base_server/server_async_secure.py

@@ -9,13 +9,14 @@ from pathlib import Path
 # Add parent directory to path to enable imports
 sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent))
 
-from digitalkin.grpc_servers._base_server import BaseServer
 from digitalkin.grpc_servers.utils.models import (
     SecurityMode,
     ServerConfig,
     ServerCredentials,
     ServerMode,
 )
+
+from digitalkin.grpc_servers._base_server import BaseServer
 from examples.base_server.mock.mock_pb2 import DESCRIPTOR, HelloReply  # type: ignore
 from examples.base_server.mock.mock_pb2_grpc import (
     Greeter,
@@ -35,7 +36,7 @@ class AsyncGreeterImpl(Greeter):
 
     async def SayHello(self, request, context):  # noqa: N802
         """Asynchronous implementation of SayHello method."""
-        logger.info(f"Received request object: {request}")
+        logger.info("Received request object: %s", request)
         logger.info(f"Request attributes: {vars(request)}")
         logger.info(f"Received request with name: {request.name}")
 
@@ -45,7 +46,7 @@ class AsyncGreeterImpl(Greeter):
             name = "unknown"
             # Check context metadata
             for key, value in context.invocation_metadata():
-                logger.info(f"Metadata: {key}={value}")
+                logger.info("Metadata: %s=%s", key, value)
                 if key.lower() == "name":
                     name = value
 
@@ -115,7 +116,7 @@ async def main_async() -> int:
         # as the KeyboardInterrupt usually breaks out of asyncio.run()
         logger.info("Server stopping due to keyboard interrupt...")
     except Exception as e:
-        logger.exception(f"Error running server: {e}")
+        logger.exception("Error running server: %s", e)
         return 1
     finally:
         # Clean up resources if server was started
@@ -134,7 +135,7 @@ def main():
         logger.info("Server stopped by keyboard interrupt")
         return 0  # Clean exit
     except Exception as e:
-        logger.exception(f"Fatal error: {e}")
+        logger.exception("Fatal error: %s", e)
         return 1
 
 

{digitalkin-0.3.1.dev1 → digitalkin-0.3.2a2}/examples/base_server/server_sync_insecure.py

@@ -8,8 +8,9 @@ from pathlib import Path
 # Add parent directory to path to enable imports
 sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent))
 
-from digitalkin.grpc_servers._base_server import BaseServer
 from digitalkin.grpc_servers.utils.models import SecurityMode, ServerConfig, ServerMode
+
+from digitalkin.grpc_servers._base_server import BaseServer
 from examples.base_server.mock.mock_pb2 import DESCRIPTOR, HelloReply  # type: ignore
 from examples.base_server.mock.mock_pb2_grpc import (
     Greeter,
@@ -29,7 +30,7 @@ class SyncGreeterServicer(Greeter):
 
     def SayHello(self, request, context):  # noqa: N802
         """Implementation of SayHello method."""
-        logger.info(f"Received request object: {request}")
+        logger.info("Received request object: %s", request)
         logger.info(f"Request attributes: {vars(request)}")
         logger.info(f"Received request with name: {request.name}")
 
@@ -39,7 +40,7 @@ class SyncGreeterServicer(Greeter):
             name = "unknown"
             # Check context metadata
             for key, value in context.invocation_metadata():
-                logger.info(f"Metadata: {key}={value}")
+                logger.info("Metadata: %s=%s", key, value)
                 if key.lower() == "name":
                     name = value
 
@@ -92,7 +93,7 @@ def main() -> int:
             server.stop()
 
     except Exception as e:
-        logger.exception(f"Error running server: {e}")
+        logger.exception("Error running server: %s", e)
         return 1
 
     return 0

{digitalkin-0.3.1.dev1 → digitalkin-0.3.2a2}/examples/base_server/server_sync_secure.py

@@ -8,13 +8,14 @@ from pathlib import Path
 # Add parent directory to path to enable imports
 sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent))
 
-from digitalkin.grpc_servers._base_server import BaseServer
 from digitalkin.grpc_servers.utils.models import (
     SecurityMode,
     ServerConfig,
     ServerCredentials,
     ServerMode,
 )
+
+from digitalkin.grpc_servers._base_server import BaseServer
 from examples.base_server.mock.mock_pb2 import DESCRIPTOR, HelloReply  # type: ignore
 from examples.base_server.mock.mock_pb2_grpc import (
     Greeter,
@@ -34,7 +35,7 @@ class SyncGreeterServicer(Greeter):
 
     def SayHello(self, request, context):  # noqa: N802
         """Implementation of SayHello method."""
-        logger.info(f"Received request object: {request}")
+        logger.info("Received request object: %s", request)
         logger.info(f"Request attributes: {vars(request)}")
         logger.info(f"Received request with name: {request.name}")
 
@@ -44,7 +45,7 @@ class SyncGreeterServicer(Greeter):
             name = "unknown"
             # Check context metadata
             for key, value in context.invocation_metadata():
-                logger.info(f"Metadata: {key}={value}")
+                logger.info("Metadata: %s=%s", key, value)
                 if key.lower() == "name":
                     name = value
 
@@ -111,7 +112,7 @@ def main() -> int:
             server.stop()
 
     except Exception as e:
-        logger.exception(f"Error running server: {e}")
+        logger.exception("Error running server: %s", e)
         return 1
 
     return 0

digitalkin-0.3.2a2/examples/modules/archetype_with_tools_module.py

@@ -0,0 +1,232 @@
+"""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
+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(
+            module_ids=["search-tool-v1"],
+        ),
+        json_schema_extra={"config": True},
+    )
+
+    calculator_tool: ToolReference = Field(
+        default_factory=lambda: ToolReference(
+            tags=["math-calculator"],
+        ),
+        json_schema_extra={"config": True},
+    )
+
+    dynamic_tool: ToolReference = Field(
+        default_factory=lambda: ToolReference(
+            tags=["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)