digitalkin 0.3.0rc1__tar.gz → 0.3.0rc2__tar.gz

{digitalkin-0.3.0rc1 → digitalkin-0.3.0rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.0rc1
+Version: 0.3.0rc2
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -456,33 +456,13 @@ Requires-Dist: digitalkin-proto>=0.1.16
 Requires-Dist: grpcio-health-checking>=1.71.0
 Requires-Dist: grpcio-reflection>=1.71.0
 Requires-Dist: grpcio-status>=1.71.0
-Requires-Dist: pydantic>=2.12.1
+Requires-Dist: pydantic>=2.12.3
 Requires-Dist: surrealdb>=1.0.6
 Provides-Extra: taskiq
-Requires-Dist: rstream>=0.31.0; extra == "taskiq"
-Requires-Dist: taskiq-aio-pika>=0.4.3; extra == "taskiq"
+Requires-Dist: rstream>=0.40.0; extra == "taskiq"
+Requires-Dist: taskiq-aio-pika>=0.4.4; extra == "taskiq"
 Requires-Dist: taskiq-redis>=1.1.2; extra == "taskiq"
-Requires-Dist: taskiq[reload]>=0.11.18; extra == "taskiq"
-Provides-Extra: dev
-Requires-Dist: typos>=1.38.0; extra == "dev"
-Requires-Dist: ruff>=0.14.1; extra == "dev"
-Requires-Dist: mypy>=1.18.2; extra == "dev"
-Requires-Dist: pyright>=1.1.406; extra == "dev"
-Requires-Dist: pre-commit>=4.3.0; extra == "dev"
-Requires-Dist: bump-my-version>=1.2.4; extra == "dev"
-Requires-Dist: build>=1.3.0; extra == "dev"
-Requires-Dist: twine>=6.2.0; extra == "dev"
-Requires-Dist: cryptography>=46.0.2; extra == "dev"
-Provides-Extra: tests
-Requires-Dist: freezegun>=1.5.3; extra == "tests"
-Requires-Dist: hdrhistogram>=0.10.3; extra == "tests"
-Requires-Dist: grpcio-testing>=1.71.0; extra == "tests"
-Requires-Dist: psutil>=7.0.0; extra == "tests"
-Requires-Dist: pytest>=8.4.2; extra == "tests"
-Requires-Dist: pytest-asyncio>=1.2.0; extra == "tests"
-Requires-Dist: pytest-cov>=7.0.0; extra == "tests"
-Requires-Dist: pytest-html==4.1.1; extra == "tests"
-Requires-Dist: pytest-json-report==1.5.0; extra == "tests"
+Requires-Dist: taskiq[reload]>=0.11.19; extra == "taskiq"
 Dynamic: license-file
 
 # DigitalKin Python SDK

{digitalkin-0.3.0rc1 → digitalkin-0.3.0rc2}/pyproject.toml

@@ -12,7 +12,7 @@
 
     keywords = [ "digitalkin", "kin", "agent", "gprc", "sdk" ]
 
-    version = "0.3.0-rc1"
+    version = "0.3.0-rc2"
     classifiers = [
         "Development Status :: 3 - Alpha",
         "Intended Audience :: Developers",
@@ -32,39 +32,18 @@
         "grpcio-health-checking>=1.71.0",
         "grpcio-reflection>=1.71.0",
         "grpcio-status>=1.71.0",
-        "pydantic>=2.12.1",
+        "pydantic>=2.12.3",
         "surrealdb>=1.0.6",
     ]
 
     [project.optional-dependencies]
         taskiq = [
-            "rstream>=0.31.0",
-            "taskiq-aio-pika>=0.4.3",
+            "rstream>=0.40.0",
+            "taskiq-aio-pika>=0.4.4",
             "taskiq-redis>=1.1.2",
-            "taskiq[reload]>=0.11.18",
-        ]
-        dev = [
-            "typos>=1.38.0",
-            "ruff>=0.14.1",
-            "mypy>=1.18.2",
-            "pyright>=1.1.406",
-            "pre-commit>=4.3.0",
-            "bump-my-version>=1.2.4",
-            "build>=1.3.0",
-            "twine>=6.2.0",
-            "cryptography>=46.0.2",
-        ]
-        tests = [
-            "freezegun>=1.5.3",
-            "hdrhistogram>=0.10.3",
-            "grpcio-testing>=1.71.0",
-            "psutil>=7.0.0",
-            "pytest>=8.4.2",
-            "pytest-asyncio>=1.2.0",
-            "pytest-cov>=7.0.0",
-            "pytest-html==4.1.1",
-            "pytest-json-report==1.5.0",
+            "taskiq[reload]>=0.11.19",
         ]
+
     [project.urls]
         Homepage      = "https://github.com/DigitalKin-ai/digitalkin"
         Documentation = "https://github.com/DigitalKin-ai/digitalkin"
@@ -74,6 +53,55 @@
         name  = "DigitalKin.ai"
         email = "contact@digitalkin.ai"
 
+[dependency-groups]
+    dev = [
+        "typos>=1.38.0",
+        "ruff>=0.14.2",
+        "mypy>=1.18.2",
+        "pyright>=1.1.407",
+        "pre-commit>=4.3.0",
+        "bump-my-version>=1.2.4",
+        "build>=1.3.0",
+        "twine>=6.2.0",
+        "cryptography>=46.0.3",
+    ]
+    tests = [
+        "freezegun>=1.5.3",
+        "hdrhistogram>=0.10.3",
+        "grpcio-testing>=1.71.0",
+        "psutil>=7.1.2",
+        "pytest>=8.4.2",
+        "pytest-asyncio>=1.2.0",
+        "pytest-cov>=7.0.0",
+        "pytest-html==4.1.1",
+        "pytest-json-report==1.5.0",
+    ]
+    docs = [
+        "mike>=2.1.3",
+        "markdown-callouts>=0.4",
+        "markdown-exec>=1.8",
+        "mkdocs>=1.6",
+        "mkdocs-coverage>=2.0.0",
+        "mkdocs-llmstxt>=0.2",
+        "mkdocs-redirects>=1.2",
+        "mkdocstrings>=0.29",
+        "griffe-inherited-docstrings>=1.1.2",
+        "mkdocs-autorefs>=1.4.3",
+        "mkdocs-awesome-pages-plugin>=2.10.1",
+        "mkdocs-git-committers-plugin-2>=2.5.0",
+        "mkdocs-git-revision-date-localized-plugin>=1.4.7",
+        "mkdocs-glightbox>=0.5.2",
+        "mkdocs-include-markdown-plugin>=7.2.0",
+        "mkdocs-literate-nav>=0.6.2",
+        "mkdocs-material[imaging]>=9.6.22",
+        "mkdocs-minify-plugin>=0.8.0",
+        "mkdocs-section-index>=0.3.10",
+        "mkdocstrings>=0.30.1",
+        "mkdocstrings-python>=1.18.2",
+        "mkdocs-open-in-new-tab>=1.0.8",
+        "tomli>=2.3.0",
+]
+
 [tool.setuptools]
     package-dir = { "" = "src" }
 
@@ -114,6 +142,7 @@
         "generate_certificates.py",
         "examples/*",
         "tests/*",
+        "docs/*",
     ]
     preview = true
 
@@ -232,11 +261,3 @@
     testpaths    = [ "tests" ]
     python_files = "test_*.py"
     asyncio_mode = "auto"
-
-[dependency-groups]
-    dev = [
-        "pytest>=8.4.2",
-        "pytest-cov>=7.0.0",
-        "pytest-html>=4.1.1",
-        "pytest-json-report>=1.5.0",
-    ]

{digitalkin-0.3.0rc1 → digitalkin-0.3.0rc2}/src/digitalkin/__version__.py

@@ -5,4 +5,4 @@ from importlib.metadata import PackageNotFoundError, version
 try:
     __version__ = version("digitalkin")
 except PackageNotFoundError:
-    __version__ = "0.3.0-rc1"
+    __version__ = "0.3.0-rc2"

{digitalkin-0.3.0rc1 → digitalkin-0.3.0rc2}/src/digitalkin/core/task_manager/task_manager.py

@@ -26,7 +26,7 @@ class TaskManager:
     max_concurrent_tasks: int
     _shutdown_event: asyncio.Event
 
-    def __init__(self, default_timeout: float = 10.0, max_concurrent_tasks: int = 100) -> None:
+    def __init__(self, default_timeout: float = 10.0, max_concurrent_tasks: int = 1000) -> None:
         """Defining task manager properties."""
         self.tasks = {}
         self.tasks_sessions = {}
@@ -67,6 +67,8 @@ class TaskManager:
         if task_id in self.tasks_sessions:
             await self.tasks_sessions[task_id].db.close()
         # Remove from collections
+        self.tasks.pop(task_id, None)
+        self.tasks_sessions.pop(task_id, None)
 
     async def _task_wrapper(  # noqa: C901, PLR0915
         self,
@@ -356,10 +358,11 @@ class TaskManager:
             return False
 
         await self.tasks_sessions[task_id].module.stop()
-        await self.cancel_task(mission_id, task_id)
+        await self.cancel_task(task_id, mission_id)
 
         logger.info("Cleaning up session for task: '%s'", task_id, extra={"mission_id": mission_id, "task_id": task_id})
         self.tasks_sessions.pop(task_id, None)
+        self.tasks.pop(task_id, None)
         return True
 
     async def pause_task(self, task_id: str, mission_id: str) -> bool:

{digitalkin-0.3.0rc1 → digitalkin-0.3.0rc2}/src/digitalkin/core/task_manager/task_session.py

@@ -42,13 +42,23 @@ class TaskSession:
         db: SurrealDBConnection,
         module: BaseModule,
         heartbeat_interval: datetime.timedelta = datetime.timedelta(seconds=2),
+        queue_maxsize: int = 1000,
     ) -> None:
-        """Initialize Task Session."""
+        """Initialize Task Session.
+
+        Args:
+            task_id: Unique task identifier
+            mission_id: Mission identifier
+            db: SurrealDB connection
+            module: Module instance
+            heartbeat_interval: Interval between heartbeats
+            queue_maxsize: Maximum size for the queue (0 = unlimited)
+        """
         self.db = db
         self.module = module
 
         self.status = TaskStatus.PENDING
-        self.queue: asyncio.Queue = asyncio.Queue()
+        self.queue: asyncio.Queue = asyncio.Queue(maxsize=queue_maxsize)
 
         self.task_id = task_id
         self.mission_id = mission_id

{digitalkin-0.3.0rc1 → digitalkin-0.3.0rc2}/src/digitalkin/models/grpc_servers/models.py

@@ -175,8 +175,8 @@ class ClientConfig(ChannelConfig):
     credentials: ClientCredentials | None = Field(None, description="Client credentials for secure mode")
     channel_options: list[tuple[str, Any]] = Field(
         default_factory=lambda: [
-            ("grpc.max_receive_message_length", 50 * 1024 * 1024),  # 50MB
-            ("grpc.max_send_message_length", 50 * 1024 * 1024),  # 50MB
+            ("grpc.max_receive_message_length", 100 * 1024 * 1024),  # 100MB
+            ("grpc.max_send_message_length", 100 * 1024 * 1024),  # 100MB
         ],
         description="Additional channel options",
     )
@@ -223,8 +223,8 @@ class ServerConfig(ChannelConfig):
     credentials: ServerCredentials | None = Field(None, description="Server credentials for secure mode")
     server_options: list[tuple[str, Any]] = Field(
         default_factory=lambda: [
-            ("grpc.max_receive_message_length", 50 * 1024 * 1024),  # 50MB
-            ("grpc.max_send_message_length", 50 * 1024 * 1024),  # 50MB
+            ("grpc.max_receive_message_length", 100 * 1024 * 1024),  # 100MB
+            ("grpc.max_send_message_length", 100 * 1024 * 1024),  # 100MB
         ],
         description="Additional server options",
     )

{digitalkin-0.3.0rc1 → digitalkin-0.3.0rc2}/src/digitalkin/models/module/module_types.py

@@ -24,7 +24,11 @@ class DataTrigger(BaseModel):
     """
 
     protocol: ClassVar[str]
-    created_at: str = datetime.now(tz=timezone.utc).isoformat()
+    created_at: str = Field(
+        default_factory=lambda: datetime.now(tz=timezone.utc).isoformat(),
+        title="Created At",
+        description="Timestamp when the payload was created.",
+    )
 
 
 DataTriggerT = TypeVar("DataTriggerT", bound=DataTrigger)

{digitalkin-0.3.0rc1 → digitalkin-0.3.0rc2}/src/digitalkin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.0rc1
+Version: 0.3.0rc2
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -456,33 +456,13 @@ Requires-Dist: digitalkin-proto>=0.1.16
 Requires-Dist: grpcio-health-checking>=1.71.0
 Requires-Dist: grpcio-reflection>=1.71.0
 Requires-Dist: grpcio-status>=1.71.0
-Requires-Dist: pydantic>=2.12.1
+Requires-Dist: pydantic>=2.12.3
 Requires-Dist: surrealdb>=1.0.6
 Provides-Extra: taskiq
-Requires-Dist: rstream>=0.31.0; extra == "taskiq"
-Requires-Dist: taskiq-aio-pika>=0.4.3; extra == "taskiq"
+Requires-Dist: rstream>=0.40.0; extra == "taskiq"
+Requires-Dist: taskiq-aio-pika>=0.4.4; extra == "taskiq"
 Requires-Dist: taskiq-redis>=1.1.2; extra == "taskiq"
-Requires-Dist: taskiq[reload]>=0.11.18; extra == "taskiq"
-Provides-Extra: dev
-Requires-Dist: typos>=1.38.0; extra == "dev"
-Requires-Dist: ruff>=0.14.1; extra == "dev"
-Requires-Dist: mypy>=1.18.2; extra == "dev"
-Requires-Dist: pyright>=1.1.406; extra == "dev"
-Requires-Dist: pre-commit>=4.3.0; extra == "dev"
-Requires-Dist: bump-my-version>=1.2.4; extra == "dev"
-Requires-Dist: build>=1.3.0; extra == "dev"
-Requires-Dist: twine>=6.2.0; extra == "dev"
-Requires-Dist: cryptography>=46.0.2; extra == "dev"
-Provides-Extra: tests
-Requires-Dist: freezegun>=1.5.3; extra == "tests"
-Requires-Dist: hdrhistogram>=0.10.3; extra == "tests"
-Requires-Dist: grpcio-testing>=1.71.0; extra == "tests"
-Requires-Dist: psutil>=7.0.0; extra == "tests"
-Requires-Dist: pytest>=8.4.2; extra == "tests"
-Requires-Dist: pytest-asyncio>=1.2.0; extra == "tests"
-Requires-Dist: pytest-cov>=7.0.0; extra == "tests"
-Requires-Dist: pytest-html==4.1.1; extra == "tests"
-Requires-Dist: pytest-json-report==1.5.0; extra == "tests"
+Requires-Dist: taskiq[reload]>=0.11.19; extra == "taskiq"
 Dynamic: license-file
 
 # DigitalKin Python SDK

digitalkin-0.3.0rc2/src/digitalkin.egg-info/requires.txt

@@ -0,0 +1,12 @@
+digitalkin-proto>=0.1.16
+grpcio-health-checking>=1.71.0
+grpcio-reflection>=1.71.0
+grpcio-status>=1.71.0
+pydantic>=2.12.3
+surrealdb>=1.0.6
+
+[taskiq]
+rstream>=0.40.0
+taskiq-aio-pika>=0.4.4
+taskiq-redis>=1.1.2
+taskiq[reload]>=0.11.19

digitalkin-0.3.0rc1/src/digitalkin.egg-info/requires.txt

@@ -1,34 +0,0 @@
-digitalkin-proto>=0.1.16
-grpcio-health-checking>=1.71.0
-grpcio-reflection>=1.71.0
-grpcio-status>=1.71.0
-pydantic>=2.12.1
-surrealdb>=1.0.6
-
-[dev]
-typos>=1.38.0
-ruff>=0.14.1
-mypy>=1.18.2
-pyright>=1.1.406
-pre-commit>=4.3.0
-bump-my-version>=1.2.4
-build>=1.3.0
-twine>=6.2.0
-cryptography>=46.0.2
-
-[taskiq]
-rstream>=0.31.0
-taskiq-aio-pika>=0.4.3
-taskiq-redis>=1.1.2
-taskiq[reload]>=0.11.18
-
-[tests]
-freezegun>=1.5.3
-hdrhistogram>=0.10.3
-grpcio-testing>=1.71.0
-psutil>=7.0.0
-pytest>=8.4.2
-pytest-asyncio>=1.2.0
-pytest-cov>=7.0.0
-pytest-html==4.1.1
-pytest-json-report==1.5.0