digitalkin 0.3.0__py3-none-any.whl → 0.3.0.dev1__py3-none-any.whl

digitalkin/__version__.py

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

digitalkin/modules/job_manager/single_job_manager.py

@@ -242,7 +242,7 @@ class SingleJobManager(BaseJobManager, TaskManager, Generic[InputModelT, OutputM
         Raises:
             Exception: If an error occurs while stopping the module.
         """
-        logger.critical(f"STOP {job_id=} | {self.tasks_sessions.keys()}")
+        logger.info(f"STOP required for {job_id=}")
 
         async with self._lock:
             session = self.tasks_sessions.get(job_id)

digitalkin/modules/job_manager/surrealdb_repository.py

@@ -75,7 +75,10 @@ class SurrealDBConnection(Generic[TSurreal]):
             timeout: Timeout for database operations
         """
         self.timeout = timeout
-        self.url = f"{os.getenv('SURREALDB_URL', 'ws://localhost')}:{os.getenv('SURREALDB_PORT', '8000')}/rpc"
+        base_url = os.getenv("SURREALDB_URL", "ws://localhost").strip()
+        port = (os.getenv("SURREALDB_PORT") or "").strip()
+        self.url = f"{base_url}{f':{port}' if port else ''}/rpc"
+
         self.username = os.getenv("SURREALDB_USERNAME", "root")
         self.password = os.getenv("SURREALDB_PASSWORD", "root")
         self.namespace = os.getenv("SURREALDB_NAMESPACE", "test")

digitalkin/modules/job_manager/task_manager.py

@@ -123,12 +123,10 @@ class TaskManager:
                 completed = next(iter(done))
                 await completed
 
-                logger.critical(f"{completed=} | {main_task=} | {hb_task=} | {sig_task=}")
-
                 if completed is main_task:
                     session.status = TaskStatus.COMPLETED
                 elif completed is sig_task or (completed is hb_task and sig_task.done()):
-                    logger.critical(f"{sig_task=}")
+                    logger.debug(f"Task cancelled due to signal {sig_task=}")
                     session.status = TaskStatus.CANCELLED
                 elif completed is hb_task:
                     session.status = TaskStatus.FAILED

digitalkin/modules/job_manager/task_session.py

@@ -90,7 +90,6 @@ class TaskSession:
         if self.heartbeat_record_id is None:
             try:
                 success = await self.db.create("heartbeats", heartbeat.model_dump())
-                logger.critical(f"{success=} | {'code' not in success}")
                 if "code" not in success:
                     self.heartbeat_record_id = success.get("id")  # type: ignore
                     self._last_heartbeat = heartbeat.timestamp
@@ -138,7 +137,7 @@ class TaskSession:
 
     async def generate_heartbeats(self) -> None:
         """Periodic heartbeat generator with cancellation support."""
-        logger.critical("Heartbeat started")
+        logger.debug("Heartbeat started")
         while not self.cancelled:
             logger.debug(f"Heartbeat tick for task: '{self.task_id}' | {self.cancelled=}")
             success = await self.send_heartbeat()
@@ -167,7 +166,7 @@ class TaskSession:
         live_id, live_signals = await self.db.start_live("tasks")
         try:
             async for signal in live_signals:
-                logger.critical("Signal received for task '%s': %s", self.task_id, signal)
+                logger.debug("Signal received for task '%s': %s", self.task_id, signal)
                 if self.cancelled:
                     break
 
@@ -199,7 +198,7 @@ class TaskSession:
 
     async def _handle_cancel(self) -> None:
         """Idempotent cancellation with acknowledgment."""
-        logger.critical("Handle cancel called")
+        logger.debug("Handle cancel called")
         if self.is_cancelled.is_set():
             logger.debug(
                 "Cancel signal ignored - task already cancelled: '%s'", self.task_id, extra={"task_id": self.task_id}

digitalkin/services/filesystem/default_filesystem.py

@@ -198,8 +198,6 @@ class DefaultFilesystem(FilesystemStrategy):
             end_idx = start_idx + list_size
             paginated_files = filtered_files[start_idx:end_idx]
 
-            logger.critical(f"{filters=} | {paginated_files=}")
-
             if include_content:
                 for file in paginated_files:
                     file.content = Path(file.storage_uri).read_bytes()

{digitalkin-0.3.0.dist-info → digitalkin-0.3.0.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.0
+Version: 0.3.0.dev1
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -456,7 +456,7 @@ 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.11.10
+Requires-Dist: pydantic>=2.12.1
 Requires-Dist: surrealdb>=1.0.6
 Provides-Extra: taskiq
 Requires-Dist: rstream>=0.31.0; extra == "taskiq"
@@ -469,7 +469,7 @@ Requires-Dist: ruff>=0.13.3; 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: bump2version>=1.0.1; 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"
@@ -481,7 +481,8 @@ 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: hypothesis>=6.140.3; extra == "tests"
+Requires-Dist: pytest-html==4.1.1; extra == "tests"
+Requires-Dist: pytest-json-report==1.5.0; extra == "tests"
 Dynamic: license-file
 
 # DigitalKin Python SDK

{digitalkin-0.3.0.dist-info → digitalkin-0.3.0.dev1.dist-info}/RECORD

@@ -7,7 +7,7 @@ base_server/mock/__init__.py,sha256=YZFT-F1l_TpvJYuIPX-7kTeE1CfOjhx9YmNRXVoi-jQ,
 base_server/mock/mock_pb2.py,sha256=sETakcS3PAAm4E-hTCV1jIVaQTPEAIoVVHupB8Z_k7Y,1843
 base_server/mock/mock_pb2_grpc.py,sha256=BbOT70H6q3laKgkHfOx1QdfmCS_HxCY4wCOX84YAdG4,3180
 digitalkin/__init__.py,sha256=7LLBAba0th-3SGqcpqFO-lopWdUkVLKzLZiMtB-mW3M,162
-digitalkin/__version__.py,sha256=1cAQQLeheVwVxIu7yYP-6PyMVzQpLtDNXsgt9VoOgTQ,190
+digitalkin/__version__.py,sha256=TouYEd0hGF2qSimjErQ5O3uouXtxL5suh2P6BxRWN2c,195
 digitalkin/logger.py,sha256=8ze_tjt2G6mDTuQcsf7-UTXWP3UHZ7LZVSs_iqF4rX4,4685
 digitalkin/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 digitalkin/grpc_servers/__init__.py,sha256=0cJBlwipSmFdXkyH3T0i6OJ1WpAtNsZgYX7JaSnkbtg,804
@@ -46,10 +46,10 @@ digitalkin/modules/tool_module.py,sha256=rLJfdYMWddiUgPLBDGG_UcHnRCHEeaqX6Fdp4DO
 digitalkin/modules/trigger_handler.py,sha256=qPNMi-8NHqscOxciHeaXtpwjXApT3YzjMF23zQAjaZY,1770
 digitalkin/modules/job_manager/base_job_manager.py,sha256=iDv9caispLmM7mqPgG_g7oaACMvNW7i7MTzBeSlmNp0,6355
 digitalkin/modules/job_manager/job_manager_models.py,sha256=onHy-DfInLZQveniMIWIKwTKSQjojz500JvHB54x93c,1129
-digitalkin/modules/job_manager/single_job_manager.py,sha256=Fw64gGRdQuTLZO-HChBTbBlZvOPyNaeb_p-HhcQpGJE,11177
-digitalkin/modules/job_manager/surrealdb_repository.py,sha256=3Kaz-PqfaMzSpX2RmbgbiO2pQcLVQ1tcspdCMPm7VTI,7712
-digitalkin/modules/job_manager/task_manager.py,sha256=VpRoscXkNf8LBWw3jCE6MwPvtUb2Q8V_dGi059fVyVw,14402
-digitalkin/modules/job_manager/task_session.py,sha256=rDKWgXf0sq7gBd6KODLPYjov3GIIyHAmFxAKTI-n1J0,9655
+digitalkin/modules/job_manager/single_job_manager.py,sha256=rJ6ppgXQkLu2X4NF2oicWf3rutYN4Qtfey3EM7ONfKM,11155
+digitalkin/modules/job_manager/surrealdb_repository.py,sha256=qGDDEEV80vKVUe4r2yvTqTNfKrQoiZjeFok01rn_eaE,7798
+digitalkin/modules/job_manager/task_manager.py,sha256=Z_luiGdHg6jnNSslf9Dl8m9w6PIBtgrOHZXfyikBVwY,14336
+digitalkin/modules/job_manager/task_session.py,sha256=KpBoOH2ND7UQauRok7pGsNfybH-rQDIwvdtg07ZFz40,9573
 digitalkin/modules/job_manager/taskiq_broker.py,sha256=4q3U03SNlb1ypup0VOa73KHLKJ9peuZUPCR5eFLMNAk,7498
 digitalkin/modules/job_manager/taskiq_job_manager.py,sha256=NHSGsqqKCWQQqx0FqM3c4tf011vFlqX1npHkcv0kEOU,10249
 digitalkin/services/__init__.py,sha256=LqGk_5DJy8Bzz62ajIq9jCeYNKQUIgtSCpafZk15FLc,910
@@ -64,7 +64,7 @@ digitalkin/services/cost/cost_strategy.py,sha256=MpPX33P_S5b2by6F4zT-rcyeRuh2V4N
 digitalkin/services/cost/default_cost.py,sha256=XE7kNFde8NmbulU9m1lc3mi-vHFkbaJf0XHUc0D4UHE,3945
 digitalkin/services/cost/grpc_cost.py,sha256=cGtb0atPXSEEOrNIWee-o3ScfNRSAFXJGDu0vcWT6zg,6295
 digitalkin/services/filesystem/__init__.py,sha256=BhwMl_BUvM0d65fmglkp0SVwn3RfYiUOKJgIMnOCaGM,381
-digitalkin/services/filesystem/default_filesystem.py,sha256=qfC4jorfo86fBBnth-4ft13VuaGdvIHgWOVurCykXIk,15182
+digitalkin/services/filesystem/default_filesystem.py,sha256=WQbU-Bsi9r-28VqhKbrplce3otzjSKS-5iqKEpGWdQU,15117
 digitalkin/services/filesystem/filesystem_strategy.py,sha256=zibVLvX_IBQ-kgh-KYzHdszDeiHFPEAZszu_k99x1GQ,9487
 digitalkin/services/filesystem/grpc_filesystem.py,sha256=ilxTFjelmf8_bVSs3xLlsyWFHVlSJf27c4j8H7R1Rkw,12987
 digitalkin/services/identity/__init__.py,sha256=InkeyLgFYYwItx8mePA8HpfacOMWZwwuc0G4pWtKq9s,270
@@ -89,14 +89,14 @@ digitalkin/utils/arg_parser.py,sha256=nvjI1pKDY1HfS0oGcMQPtdTQcggXLtpxXMbnMxNEKR
 digitalkin/utils/development_mode_action.py,sha256=TqRuAF_A7bDD4twRB4PnZcRoNeaiAnEdxM5kvy4aoaA,1511
 digitalkin/utils/llm_ready_schema.py,sha256=JjMug_lrQllqFoanaC091VgOqwAd-_YzcpqFlS7p778,2375
 digitalkin/utils/package_discover.py,sha256=3e9-6Vf3yAAv2VkHHVK4QVqHJBxQqg3d8uuDTsXph24,13471
-digitalkin-0.3.0.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
+digitalkin-0.3.0.dev1.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
 modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modules/cpu_intensive_module.py,sha256=ejB9XPnFfA0uCuFUQbM3fy5UYfqqAlF36rv_P5Ri8ho,8363
 modules/minimal_llm_module.py,sha256=Ijld__ZnhzfLwpXD1XVkLZ7jyKZKyOFZczOpiPttJZc,11216
 modules/text_transform_module.py,sha256=bwPSnEUthZQyfLwcTLo52iAxItAoknkLh8Y3m5aywaY,7251
 services/filesystem_module.py,sha256=71Mcja8jCQqiqFHPdsIXplFIHTvgkxRhp0TRXuCfgkk,7430
 services/storage_module.py,sha256=ybTMqmvGaTrR8PqJ4FU0cwxaDjT36TskVrGoetTGmno,6955
-digitalkin-0.3.0.dist-info/METADATA,sha256=_6KDoMRgjmgnAXmPLUVpNY1EcyN86SE5c_zvtNo-K8w,30588
-digitalkin-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-digitalkin-0.3.0.dist-info/top_level.txt,sha256=gcjqlyrZuLjIyxrOIavCQM_olpr6ND5kPKkZd2j0xGo,40
-digitalkin-0.3.0.dist-info/RECORD,,
+digitalkin-0.3.0.dev1.dist-info/METADATA,sha256=1slDz-MhzogLJEErQ7zxlqIo5IneECx-NxLzGOJdXUg,30653
+digitalkin-0.3.0.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+digitalkin-0.3.0.dev1.dist-info/top_level.txt,sha256=gcjqlyrZuLjIyxrOIavCQM_olpr6ND5kPKkZd2j0xGo,40
+digitalkin-0.3.0.dev1.dist-info/RECORD,,