digitalkin 0.3.0rc0__py3-none-any.whl → 0.3.0rc1__py3-none-any.whl

digitalkin/modules/_base_module.py

@@ -50,13 +50,7 @@ class BaseModule(  # noqa: PLR0904
     services_config_params: ClassVar[dict[str, dict[str, Any | None] | None]]
     services_config: ServicesConfig
 
-    # runtime params
-    job_id: str
-    mission_id: str
-    setup_id: str
-    setup_version_id: str
-
-    def _init_strategies(self) -> dict[str, Any]:
+    def _init_strategies(self, mission_id: str, setup_id: str, setup_version_id: str) -> dict[str, Any]:
         """Initialize the services configuration.
 
         Returns:
@@ -73,9 +67,9 @@ class BaseModule(  # noqa: PLR0904
         return {
             service_name: self.services_config.init_strategy(
                 service_name,
-                self.mission_id,
-                self.setup_id,
-                self.setup_version_id,
+                mission_id,
+                setup_id,
+                setup_version_id,
             )
             for service_name in self.services_config.valid_strategy_names()
         }
@@ -88,23 +82,19 @@ class BaseModule(  # noqa: PLR0904
         setup_version_id: str,
     ) -> None:
         """Initialize the module."""
-        self.job_id: str = job_id
-        self.mission_id: str = mission_id
-        # Setup reference needed for the overall Kin scope as the filesystem context
-        self.setup_id: str = setup_id
-        # SetupVersion reference needed for the precise Kin scope as the cost
-        self.setup_version_id: str = setup_version_id
         self._status = ModuleStatus.CREATED
 
         # Initialize minimum context
         self.context = ModuleContext(
             # Initialize services configuration
-            **self._init_strategies(),
+            **self._init_strategies(mission_id, setup_id, setup_version_id),
             session={
+                "setup_id": setup_id,
                 "mission_id": mission_id,
                 "setup_version_id": setup_version_id,
                 "job_id": job_id,
             },
+            callbacks={"logger": logger},
         )
 
     @property
@@ -335,7 +325,6 @@ class BaseModule(  # noqa: PLR0904
         Args:
             input_data (InputModelT): The input data to be processed by the module.
             setup_data (SetupModelT): The setup or configuration data required for the module.
-            callback (Callable[[OutputModelT], Coroutine[Any, Any, None]]): callback to be invoked to stream any result.
 
         Raises:
             ValueError: If no handler for the protocol is found.
@@ -368,51 +357,15 @@ class BaseModule(  # noqa: PLR0904
             asyncio.CancelledError: If the module is cancelled
         """
         try:
-            logger.info(
-                "Starting module %s",
-                self.name,
-                extra={
-                    "mission_id": self.mission_id,
-                    "setup_id": self.setup_id,
-                    "setup_version_id": self.setup_version_id,
-                    "job_id": self.job_id,
-                },
-            )
+            logger.info("Starting module %s", self.name, extra=self.context.session.current_ids())
             await self.run(input_data, setup_data)
-            logger.info(
-                "Module %s finished",
-                self.name,
-                extra={
-                    "mission_id": self.mission_id,
-                    "setup_id": self.setup_id,
-                    "setup_version_id": self.setup_version_id,
-                    "job_id": self.job_id,
-                },
-            )
+            logger.info("Module %s finished", self.name, extra=self.context.session.current_ids())
         except asyncio.CancelledError:
             self._status = ModuleStatus.CANCELLED
-            logger.error(
-                "Module %s cancelled",
-                self.name,
-                extra={
-                    "mission_id": self.mission_id,
-                    "setup_id": self.setup_id,
-                    "setup_version_id": self.setup_version_id,
-                    "job_id": self.job_id,
-                },
-            )
+            logger.error("Module %s cancelled", self.name, extra=self.context.session.current_ids())
         except Exception:
             self._status = ModuleStatus.FAILED
-            logger.exception(
-                "Error inside module %s",
-                self.name,
-                extra={
-                    "mission_id": self.mission_id,
-                    "setup_id": self.setup_id,
-                    "setup_version_id": self.setup_version_id,
-                    "job_id": self.job_id,
-                },
-            )
+            logger.exception("Error inside module %s", self.name, extra=self.context.session.current_ids())
         else:
             self._status = ModuleStatus.STOPPING
 
@@ -425,9 +378,8 @@ class BaseModule(  # noqa: PLR0904
     ) -> None:
         """Start the module."""
         try:
-            self.context.callbacks.logger = logger
             self.context.callbacks.send_message = callback
-            logger.info(f"Inititalize module {self.job_id}")
+            logger.info(f"Inititalize module {self.context.session.job_id}")
             await self.initialize(self.context, setup_data)
         except Exception as e:
             self._status = ModuleStatus.FAILED
@@ -448,7 +400,7 @@ class BaseModule(  # noqa: PLR0904
         try:
             logger.debug("Init the discovered input handlers.")
             self.triggers_discoverer.init_handlers(self.context)
-            logger.debug(f"Run lifecycle {self.job_id}")
+            logger.debug(f"Run lifecycle {self.context.session.job_id}")
             await self._run_lifecycle(input_data, setup_data)
         except Exception:
             self._status = ModuleStatus.FAILED
@@ -458,7 +410,7 @@ class BaseModule(  # noqa: PLR0904
 
     async def stop(self) -> None:
         """Stop the module."""
-        logger.info("Stopping module %s | job_id=%s", self.name, self.job_id)
+        logger.info("Stopping module %s | job_id=%s", self.name, self.context.session.job_id)
         try:
             self._status = ModuleStatus.STOPPING
             logger.debug("Module %s stopped", self.name)
@@ -477,15 +429,7 @@ class BaseModule(  # noqa: PLR0904
     ) -> None:
         """Start the module."""
         try:
-            logger.info(
-                "Run Config Setup lifecycle",
-                extra={
-                    "mission_id": self.mission_id,
-                    "setup_id": self.setup_id,
-                    "setup_version_id": self.setup_version_id,
-                    "job_id": self.job_id,
-                },
-            )
+            logger.info("Run Config Setup lifecycle", extra=self.context.session.current_ids())
             self._status = ModuleStatus.RUNNING
             self.context.callbacks.set_config_setup = callback
             content = await self.run_config_setup(self.context, config_setup_data)
@@ -497,12 +441,4 @@ class BaseModule(  # noqa: PLR0904
         except Exception:
             logger.error("Error during module lifecyle")
             self._status = ModuleStatus.FAILED
-            logger.exception(
-                "Error during module lifecyle",
-                extra={
-                    "mission_id": self.mission_id,
-                    "setup_id": self.setup_id,
-                    "setup_version_id": self.setup_version_id,
-                    "job_id": self.job_id,
-                },
-            )
+            logger.exception("Error during module lifecyle", extra=self.context.session.current_ids())

digitalkin/services/cost/grpc_cost.py

@@ -9,8 +9,8 @@ from google.protobuf import json_format
 
 from digitalkin.grpc_servers.utils.exceptions import ServerError
 from digitalkin.grpc_servers.utils.grpc_client_wrapper import GrpcClientWrapper
-from digitalkin.grpc_servers.utils.models import ClientConfig
 from digitalkin.logger import logger
+from digitalkin.models.grpc_servers.models import ClientConfig
 from digitalkin.services.cost.cost_strategy import (
     CostConfig,
     CostData,

digitalkin/services/filesystem/grpc_filesystem.py

@@ -10,8 +10,8 @@ from google.protobuf.json_format import MessageToDict
 
 from digitalkin.grpc_servers.utils.exceptions import ServerError
 from digitalkin.grpc_servers.utils.grpc_client_wrapper import GrpcClientWrapper
-from digitalkin.grpc_servers.utils.models import ClientConfig
 from digitalkin.logger import logger
+from digitalkin.models.grpc_servers.models import ClientConfig
 from digitalkin.services.filesystem.filesystem_strategy import (
     FileFilter,
     FilesystemRecord,

digitalkin/services/setup/grpc_setup.py

@@ -15,8 +15,8 @@ from pydantic import ValidationError
 
 from digitalkin.grpc_servers.utils.exceptions import ServerError
 from digitalkin.grpc_servers.utils.grpc_client_wrapper import GrpcClientWrapper
-from digitalkin.grpc_servers.utils.models import ClientConfig
 from digitalkin.logger import logger
+from digitalkin.models.grpc_servers.models import ClientConfig
 from digitalkin.services.setup.setup_strategy import SetupData, SetupServiceError, SetupStrategy, SetupVersionData
 
 

digitalkin/services/storage/grpc_storage.py

@@ -6,8 +6,8 @@ from google.protobuf.struct_pb2 import Struct
 from pydantic import BaseModel
 
 from digitalkin.grpc_servers.utils.grpc_client_wrapper import GrpcClientWrapper
-from digitalkin.grpc_servers.utils.models import ClientConfig
 from digitalkin.logger import logger
+from digitalkin.models.grpc_servers.models import ClientConfig
 from digitalkin.services.storage.storage_strategy import (
     DataType,
     StorageRecord,

digitalkin/utils/arg_parser.py

@@ -55,7 +55,7 @@ class ArgParser:
     """
 
     class HelpAction(_HelpAction):
-        """."""
+        """Custom HelpAction to display subparsers helps too."""
 
         def __call__(
             self,

digitalkin/utils/development_mode_action.py

@@ -13,7 +13,7 @@ logger.setLevel(logging.INFO)
 
 
 class DevelopmentModeMappingAction(Action):
-    """."""
+    """ArgParse Action to map an environment variable to a ServicesMode enum."""
 
     def __init__(
         self,
@@ -22,7 +22,7 @@ class DevelopmentModeMappingAction(Action):
         default: str | None = None,
         **kwargs: dict[str, Any],
     ) -> None:
-        """."""
+        """Initialize the DevelopmentModeMappingAction."""
         default = ServicesMode(os.environ.get(env_var, default))
 
         if required and default:

digitalkin/utils/package_discover.py

@@ -1,4 +1,4 @@
-"""."""
+"""Secure module discovery and import utility for trigger handlers."""
 
 import importlib
 import importlib.util
@@ -276,7 +276,6 @@ class ModuleDiscoverer:
         Args:
             packages: List of package names to scan.
             file_pattern: Glob pattern for matching modules.
-            safe_mode: If True, blocks modules with forbidden names.
             max_file_size: Limit for module file sizes in bytes.
         """
         self.packages = packages

{digitalkin-0.3.0rc0.dist-info → digitalkin-0.3.0rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.3.0rc0
+Version: 0.3.0rc1
 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.0rc0.dist-info → digitalkin-0.3.0rc1.dist-info}/RECORD

@@ -7,51 +7,56 @@ 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=55TYewS29nMe6l0Eb8NmybBJQFv1sBUGs3u4DD7HAh0,194
+digitalkin/__version__.py,sha256=R-V9kMfZiYqI-Y5MJxvvyFY9miiXM4kkiJl2gmUyr1E,194
 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
-digitalkin/grpc_servers/_base_server.py,sha256=32yrMCFAFE7FmdDL1WbuUlDuxRkaUJ_P7Dc-3_b40qs,18678
-digitalkin/grpc_servers/module_server.py,sha256=bi1XreL5zaPxbaDfsFN7-qtEbo0sKFNeyyLJjWdYMck,10212
-digitalkin/grpc_servers/module_servicer.py,sha256=Df2mQYrGngGYHC6fPV3Othc5hCR2g8IuLxJx9IhzDAc,19345
-digitalkin/grpc_servers/registry_server.py,sha256=StY18DKYoPKQIU1SIzgito6D4_QA1aMVddZ8O2WGlHY,2223
+digitalkin/core/__init__.py,sha256=FJRcJ-B1Viyn-38L8XpOpZ8KOnf1I7PCDOAmKXLQhqc,71
+digitalkin/core/job_manager/__init__.py,sha256=gGtgQpE6vbBHxAj1SYMbcpj45Q6x8IcsqnyQPfyZZ-8,25
+digitalkin/core/job_manager/base_job_manager.py,sha256=tkm-nT_SuctG4cSzMbvo25zwS8WSaMf-dF8MWnXnO6I,6441
+digitalkin/core/job_manager/single_job_manager.py,sha256=7wswR1hJ-yM6nLPi-7ibyvdMlG3I4pKKdW8TVsGFGWw,11078
+digitalkin/core/job_manager/taskiq_broker.py,sha256=qnN45i688z81WnEbpeoaiT9R60udnLmvCCW-XLm10y8,7461
+digitalkin/core/job_manager/taskiq_job_manager.py,sha256=UsqBsSYm88ibgRabSVPzK-9o2iXtGPsRlUO58M_PqVA,10147
+digitalkin/core/task_manager/__init__.py,sha256=k9i-qIoee_1yXogyQolaVFDUQBIZU3ENbYKtjrCNmTQ,31
+digitalkin/core/task_manager/surrealdb_repository.py,sha256=bmKaUoLTGfF-26zG_lX8mD-0dL3eTNr-BerUkSH7T2k,7739
+digitalkin/core/task_manager/task_manager.py,sha256=7wzilf2C4PvGXJVzc2sE_ZL9YrVoQ44E8kRcioP1OOw,16378
+digitalkin/core/task_manager/task_session.py,sha256=tjXm0gZOYxrnyzr07x6DtB6vZRl2U7BDBNEitSJwxLs,10525
+digitalkin/grpc_servers/__init__.py,sha256=ZIRMJ1Lcas8yQ106GCup6hn2UBOsx1sNk8ap0lpEDnY,72
+digitalkin/grpc_servers/_base_server.py,sha256=ZVeCDwI7w7fFbPTXPkeJb_SOuLfd2T7za3T4oCu2UWY,18680
+digitalkin/grpc_servers/module_server.py,sha256=kXvPOfJ9LT9oRd5oPj-ZpYB9EYY_4RDlRpcyGzWkilc,10213
+digitalkin/grpc_servers/module_servicer.py,sha256=Rr1Cq2zMaiu9oRAdjgmIlfLa_QDFzWPDamzzhE-mgzU,19365
+digitalkin/grpc_servers/registry_server.py,sha256=7jNzGiacr6XOm4EdHN7tHhaA_J0q70y38Xxbnlt8WxQ,2224
 digitalkin/grpc_servers/registry_servicer.py,sha256=dqsKGHZ0LnaIvGt4ipaAuigd37sbJBndT4MAT029GsY,16471
-digitalkin/grpc_servers/utils/exceptions.py,sha256=SyOgvjggaUECYmSiqy8KJLHwHVt5IClSTxslHM-IZzI,931
-digitalkin/grpc_servers/utils/factory.py,sha256=jm6rFjiqmtSv7BIHNAOxsG9xXtSvWpx9TfzSQiX97MQ,5899
-digitalkin/grpc_servers/utils/grpc_client_wrapper.py,sha256=Vtt05KdePSFHq8FZUJ-bJo8l_d5vu7gdOv5UBtsgScg,2752
-digitalkin/grpc_servers/utils/models.py,sha256=ZQ1Gk4zmLRX_ueuwFOwXgtyG9OEPSzFa_91I_CXZmZ0,8957
-digitalkin/grpc_servers/utils/types.py,sha256=rQ78s4nAet2jy-NIDj_PUWriT0kuGHr_w6ELjmjgBao,539
+digitalkin/grpc_servers/utils/__init__.py,sha256=ZnAIb_F8z4NhtPypqkdmzgRSzolKnJTk3oZx5GfWH5Y,38
+digitalkin/grpc_servers/utils/exceptions.py,sha256=LtaDtlqXCeT6iqApogs4pbtezotOVeg4fhnFzGBvFsY,692
+digitalkin/grpc_servers/utils/grpc_client_wrapper.py,sha256=a4xeulweovugI-IUVBrSZ-CswZHrBdeh9r-yzP8dl1w,2753
 digitalkin/mixins/__init__.py,sha256=d6ljaoyJZJT9XxOrXZG5FVNvbLURb3_CZrkp4GPZWYM,590
 digitalkin/mixins/base_mixin.py,sha256=uLkg6MbDtVc9DysjdfNIGKahxQLnnjuL3DYpuyNLbk8,486
 digitalkin/mixins/callback_mixin.py,sha256=90nHm9-pbKT14GAy3CB3fsBtpYu5IH0woOQdNLM2e_Y,836
-digitalkin/mixins/chat_history_mixin.py,sha256=zkjvL6TOJzVTYro_BmBLO0S5eHsReI1tbqTbbMkcGWA,4106
+digitalkin/mixins/chat_history_mixin.py,sha256=SqCmnh6Ybed3TQoQbvUjbzwKs_evvfW_lr4j_Uncg0A,4231
 digitalkin/mixins/cost_mixin.py,sha256=nFqhLsRHdXAt3GOH6qI8sqX9KW3leTPW3gaclensBVw,2274
-digitalkin/mixins/file_history_mixin.py,sha256=4rDpm-TVcD9ihZGPuqr73y1_4FnUzBlWUSoje-pQIIU,3759
-digitalkin/mixins/filesystem_mixin.py,sha256=myCUmocdumG7XSN1acnn4hw1e1QbkngbnBtQtrVZQHU,1604
-digitalkin/mixins/logger_mixin.py,sha256=FdVIlPXOQ7eo8DUlMmsygO9L8bis-Jhj_zw83RiE6wo,2150
+digitalkin/mixins/file_history_mixin.py,sha256=5MgxglccVe11M7YnoeuXbJYXNEwVZdiGd5Fl8sytcTM,3504
+digitalkin/mixins/filesystem_mixin.py,sha256=Q1sEY_dKZVi2Sa21fypwpsz9h7uQ7ggVqzpuVgheF_o,1510
+digitalkin/mixins/logger_mixin.py,sha256=l-SK3qACIzRfyHgj7KhzvW3ZhmVa-W1JB9tmppgpqg4,1842
 digitalkin/mixins/storage_mixin.py,sha256=ptZ4a2bydIa48q0V9e395vWHTu7yw4A6rI4jwKY6gwI,2392
 digitalkin/models/__init__.py,sha256=hDHtUfswaNh8wo4NZaBItg9JqC0uNSRqXArNWSrGynY,163
+digitalkin/models/core/__init__.py,sha256=jOMDmPX0uSfGA9zUi0u_kOvYJ46VdIssoIhVYvNSeew,19
+digitalkin/models/core/job_manager_models.py,sha256=RCrW20HCTQ7l2D1JtNgEKefZpMzoN-uAXK8bPorIqqA,1120
+digitalkin/models/core/task_monitor.py,sha256=JBMzqaKLrsxb0QZYM1BlvCxxkbqD_YKgsWob7bsmBeE,1626
+digitalkin/models/grpc_servers/__init__.py,sha256=0tA71nPSXgRrh9DoLvx-TSwZXdYIRUEItoadpTL1cTo,42
+digitalkin/models/grpc_servers/models.py,sha256=ZQ1Gk4zmLRX_ueuwFOwXgtyG9OEPSzFa_91I_CXZmZ0,8957
+digitalkin/models/grpc_servers/types.py,sha256=rQ78s4nAet2jy-NIDj_PUWriT0kuGHr_w6ELjmjgBao,539
 digitalkin/models/module/__init__.py,sha256=nFPHp0JcAZq9ISkcSu1y2zeNyXH3V4j10oMjHzMBYBU,599
 digitalkin/models/module/module.py,sha256=k0W8vfJJFth8XdDzkHm32SyTuSf3h2qF0hSrxAfGF1s,956
-digitalkin/models/module/module_context.py,sha256=nhZ_Pw9lWcq5B2PPy-Rf3N0S7MXz9Ehh5Nb-4tEOmmM,3764
+digitalkin/models/module/module_context.py,sha256=0sKIvLYt0KHR4oF0MUQ4RQZDBrPjmOSaIHNMcl7vUmI,4649
 digitalkin/models/module/module_types.py,sha256=kkOst1OWeQVzLdf0P8JBlZqf2LLWj14-bQHyYvwUN8Y,3562
-digitalkin/models/module/task_monitor.py,sha256=sEzpfmTg1cXC1ieD72YL3zwkSE6D5CIHWRi2CgK3GeQ,1288
 digitalkin/models/services/__init__.py,sha256=jhfVw6egq0OcHmos_fypH9XFehbHTBw09wluVFVFEyw,226
-digitalkin/models/services/cost.py,sha256=QTEuFD6xz62nob0z4ksE-INJWcZ-iFiuNW5mvXhpFes,1599
+digitalkin/models/services/cost.py,sha256=9PXvd5RrIk9vCrRjcUGQ9ZyAokEbwLg4s0RfnE-aLP4,1616
 digitalkin/models/services/storage.py,sha256=wp7F-AvTsU46ujGPcguqM5kUKRZx4399D4EGAAJt2zs,1143
 digitalkin/modules/__init__.py,sha256=VwVbKok81NGyPIBZgEj_SR-59G8tTlSb4eBJI9W6Vx4,281
-digitalkin/modules/_base_module.py,sha256=y_Pw2twqlAcsJlUUa-8QKdITSrZv6KpiwVw2X7mNads,18392
+digitalkin/modules/_base_module.py,sha256=UDLPbsVqlRdvT1weJ_Skzgv7R16TOyfSWeddg5ZqxzI,16513
 digitalkin/modules/archetype_module.py,sha256=lOe3yYufwfylZR_VGy1w-zqdqVaMI_JANfKkbH9eODE,471
 digitalkin/modules/tool_module.py,sha256=rLJfdYMWddiUgPLBDGG_UcHnRCHEeaqX6Fdp4DOUaCs,472
 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=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
 digitalkin/services/base_strategy.py,sha256=yA9KUJGRKuuaxA6l3GcMv8zKfWoIsW03UxJT80Yea2I,766
 digitalkin/services/services_config.py,sha256=JnyzZcG7OYBelwgn-wdVgY2n3yFTEkwLPHRZB8Tjw10,7468
@@ -62,11 +67,11 @@ digitalkin/services/agent/default_agent.py,sha256=4N_E_eQxJGOx1KVUUg5jNOje-3ncMx
 digitalkin/services/cost/__init__.py,sha256=sD_a5LrnLluASOC5m5vgIqjaco-MzZJd6XhillIBHr0,400
 digitalkin/services/cost/cost_strategy.py,sha256=MpPX33P_S5b2by6F4zT-rcyeRuh2V4NYPZe05VpDOGQ,2649
 digitalkin/services/cost/default_cost.py,sha256=XE7kNFde8NmbulU9m1lc3mi-vHFkbaJf0XHUc0D4UHE,3945
-digitalkin/services/cost/grpc_cost.py,sha256=cGtb0atPXSEEOrNIWee-o3ScfNRSAFXJGDu0vcWT6zg,6295
+digitalkin/services/cost/grpc_cost.py,sha256=wX6lrIAl92ErVe_4xjBg2z7y9Aoziv8aLVA_J3PKEHE,6296
 digitalkin/services/filesystem/__init__.py,sha256=BhwMl_BUvM0d65fmglkp0SVwn3RfYiUOKJgIMnOCaGM,381
 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/filesystem/grpc_filesystem.py,sha256=-gSA8Bmuea73DbBEG8x5b6-E0bQuT5WrK6xdSbOlH5M,12988
 digitalkin/services/identity/__init__.py,sha256=InkeyLgFYYwItx8mePA8HpfacOMWZwwuc0G4pWtKq9s,270
 digitalkin/services/identity/default_identity.py,sha256=Y2auZHrGSZTIN5D8HyjLvLcNbYFM1CNUE23x7p5VIGw,386
 digitalkin/services/identity/identity_strategy.py,sha256=skappBbds1_qa0Gr24FGrNX1N0_OYhYT1Lh7dUaAirE,429
@@ -75,28 +80,28 @@ digitalkin/services/registry/default_registry.py,sha256=VnWkF6nHpFxUKuUbZLPqzXqd
 digitalkin/services/registry/registry_strategy.py,sha256=uBXgZIv25jeXbeVO8vWvlNPxxNYu7_KiCw2PoE6AWr8,423
 digitalkin/services/setup/__init__.py,sha256=t6xcvEWqTbcRZstBFK9cESEqaZKvpW14VtYygxIqfYQ,65
 digitalkin/services/setup/default_setup.py,sha256=9VM3KwsuQcFQQ08RoOHWOE_-9BsRW0YGRtDWYTbQGdA,8246
-digitalkin/services/setup/grpc_setup.py,sha256=qjdED3HYZtpgeYYpiIwBWwPgB1NtoAuKmB5unNNRnVQ,12480
+digitalkin/services/setup/grpc_setup.py,sha256=PSfZhzPxln6fxAvFo1dL8CJBBSERPezdKp9zmkU425g,12481
 digitalkin/services/setup/setup_strategy.py,sha256=ZnJ_HwWCkHCPrqKekSD5L9y3p8wMwfjQ8sj2hLZq6go,4004
 digitalkin/services/snapshot/__init__.py,sha256=Uzlnzo0CYlSpVsdiI37hW7xQk8hu3YA1fOI6O6MSzB0,270
 digitalkin/services/snapshot/default_snapshot.py,sha256=Mb8QwWRsHh9I_tN0ln_ZiFa1QCZxOVWmuVLemQOTWpc,1058
 digitalkin/services/snapshot/snapshot_strategy.py,sha256=B1TU3V_k9A-OdqBkdyc41-ihnrW5Btcwd1KyQdHT46A,898
 digitalkin/services/storage/__init__.py,sha256=T-ocYLLphudkQgzvG47jBOm5GQsRFRIGA88y7Ur4akg,341
 digitalkin/services/storage/default_storage.py,sha256=D8e-UYUkb2GvDEHMWcN3EkcIKXWA8DrsaQsXVjoXAYQ,7975
-digitalkin/services/storage/grpc_storage.py,sha256=7RPeqkjG4JdW1lhjGpKx1qF5fp_LE-sST7fp5oT6uD0,7204
+digitalkin/services/storage/grpc_storage.py,sha256=ONilhj0-ySLqPUqWI-UfxRsSgu89zbyohcOXt5EcSdU,7205
 digitalkin/services/storage/storage_strategy.py,sha256=sERF5tIJnzpb1iNqTXic9xRkGaXMifo6kb709ubB-Yo,8848
 digitalkin/utils/__init__.py,sha256=sJnY-ZUgsjMfojAjONC1VN14mhgIDnzyOlGkw21rRnM,28
-digitalkin/utils/arg_parser.py,sha256=nvjI1pKDY1HfS0oGcMQPtdTQcggXLtpxXMbnMxNEKRU,3109
-digitalkin/utils/development_mode_action.py,sha256=TqRuAF_A7bDD4twRB4PnZcRoNeaiAnEdxM5kvy4aoaA,1511
+digitalkin/utils/arg_parser.py,sha256=wzscRlE1Qp1gGl-lAJlkkwnbU1O2oezj6BwK_BZFBIk,3158
+digitalkin/utils/development_mode_action.py,sha256=2hznh0ajW_4ZTysfoc0Y49161f_PQPATRgNk8NAn1_o,1623
 digitalkin/utils/llm_ready_schema.py,sha256=JjMug_lrQllqFoanaC091VgOqwAd-_YzcpqFlS7p778,2375
-digitalkin/utils/package_discover.py,sha256=3e9-6Vf3yAAv2VkHHVK4QVqHJBxQqg3d8uuDTsXph24,13471
-digitalkin-0.3.0rc0.dist-info/licenses/LICENSE,sha256=Ies4HFv2r2hzDRakJYxk3Y60uDFLiG-orIgeTpstnIo,20327
+digitalkin/utils/package_discover.py,sha256=sa6Zp5Kape1Zr4iYiNrnZxiHDnqM06ODk6yfWHom53w,13465
+digitalkin-0.3.0rc1.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.0rc0.dist-info/METADATA,sha256=HgycxD0XwLP9_iK2tgnjvEcR9zZlGvFYu7xamA-M9lI,30651
-digitalkin-0.3.0rc0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-digitalkin-0.3.0rc0.dist-info/top_level.txt,sha256=gcjqlyrZuLjIyxrOIavCQM_olpr6ND5kPKkZd2j0xGo,40
-digitalkin-0.3.0rc0.dist-info/RECORD,,
+digitalkin-0.3.0rc1.dist-info/METADATA,sha256=dpzbrevBvxDMjGm2fHdKxEI5V3y2WVzV-mpIwpqGA8g,30651
+digitalkin-0.3.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+digitalkin-0.3.0rc1.dist-info/top_level.txt,sha256=gcjqlyrZuLjIyxrOIavCQM_olpr6ND5kPKkZd2j0xGo,40
+digitalkin-0.3.0rc1.dist-info/RECORD,,

digitalkin/grpc_servers/utils/factory.py

@@ -1,180 +0,0 @@
-"""Factory functions for creating gRPC servers."""
-
-from pathlib import Path
-from typing import Any
-
-from digitalkin.grpc_servers.module_server import ModuleServer
-from digitalkin.grpc_servers.registry_server import RegistryServer
-from digitalkin.grpc_servers.utils.models import (
-    ModuleServerConfig,
-    RegistryServerConfig,
-    SecurityMode,
-    ServerCredentials,
-    ServerMode,
-)
-from digitalkin.modules._base_module import BaseModule
-
-
-def create_module_server(
-    module: type[BaseModule],
-    host: str = "0.0.0.0",  # noqa: S104
-    port: int = 50051,
-    max_workers: int = 10,
-    mode: str = "sync",
-    security: str = "insecure",
-    registry_address: str | None = None,
-    server_key_path: str | None = None,
-    server_cert_path: str | None = None,
-    root_cert_path: str | None = None,
-    server_options: list[tuple[str, Any]] | None = None,
-) -> ModuleServer:
-    """Create a new module server.
-
-    Args:
-        module: The module to serve.
-        host: The host address to bind to.
-        port: The port to listen on.
-        max_workers: Maximum number of workers for the thread pool (sync mode only).
-        mode: Server mode ("sync" or "async").
-        security: Security mode ("secure" or "insecure").
-        registry_address: Optional address of a registry server for auto-registration.
-        server_key_path: Path to server private key (required for secure mode).
-        server_cert_path: Path to server certificate (required for secure mode).
-        root_cert_path: Optional path to root certificate.
-        server_options: Additional server options.
-
-    Returns:
-        A configured ModuleServer instance.
-
-    Raises:
-        ValueError: If secure mode is requested but credentials are missing.
-    """
-    # Create configuration with credentials if needed
-    config = _create_server_config(
-        ModuleServerConfig,
-        host=host,
-        port=port,
-        max_workers=max_workers,
-        mode=mode,
-        security=security,
-        server_key_path=server_key_path,
-        server_cert_path=server_cert_path,
-        root_cert_path=root_cert_path,
-        server_options=server_options,
-        registry_address=registry_address,
-    )
-
-    # Create and return the server
-    return ModuleServer(module, config)
-
-
-def create_registry_server(
-    host: str = "0.0.0.0",  # noqa: S104
-    port: int = 50052,
-    max_workers: int = 10,
-    mode: str = "sync",
-    security: str = "insecure",
-    database_url: str | None = None,
-    server_key_path: str | None = None,
-    server_cert_path: str | None = None,
-    root_cert_path: str | None = None,
-    server_options: list[tuple[str, Any]] | None = None,
-) -> RegistryServer:
-    """Create a new registry server.
-
-    Args:
-        host: The host address to bind to.
-        port: The port to listen on.
-        max_workers: Maximum number of workers for the thread pool (sync mode only).
-        mode: Server mode ("sync" or "async").
-        security: Security mode ("secure" or "insecure").
-        database_url: Optional database URL for registry data storage.
-        server_key_path: Path to server private key (required for secure mode).
-        server_cert_path: Path to server certificate (required for secure mode).
-        root_cert_path: Optional path to root certificate.
-        server_options: Additional server options.
-
-    Returns:
-        A configured RegistryServer instance.
-
-    Raises:
-        ValueError: If secure mode is requested but credentials are missing.
-    """
-    # Create configuration with credentials if needed
-    config = _create_server_config(
-        RegistryServerConfig,
-        host=host,
-        port=port,
-        max_workers=max_workers,
-        mode=mode,
-        security=security,
-        server_key_path=server_key_path,
-        server_cert_path=server_cert_path,
-        root_cert_path=root_cert_path,
-        server_options=server_options,
-        database_url=database_url,
-    )
-
-    # Create and return the server
-    return RegistryServer(config)
-
-
-def _create_server_config(
-    config_class: Any,
-    host: str,
-    port: int,
-    max_workers: int,
-    mode: str,
-    security: str,
-    server_key_path: str | None,
-    server_cert_path: str | None,
-    root_cert_path: str | None,
-    server_options: list[tuple[str, Any]] | None,
-    **kwargs,
-) -> ModuleServerConfig | RegistryServerConfig:
-    """Create a server configuration with appropriate settings.
-
-    Args:
-        config_class: The configuration class to instantiate.
-        host: The host address.
-        port: The port number.
-        max_workers: Maximum number of workers.
-        mode: Server mode.
-        security: Security mode.
-        server_key_path: Path to server key.
-        server_cert_path: Path to server certificate.
-        root_cert_path: Path to root certificate.
-        server_options: Additional server options.
-        **kwargs: Additional configuration parameters.
-
-    Returns:
-        A server configuration instance.
-
-    Raises:
-        ValueError: If secure mode is requested but credentials are missing.
-    """
-    # Create basic config
-    config_params = {
-        "host": host,
-        "port": port,
-        "max_workers": max_workers,
-        "mode": ServerMode(mode),
-        "security": SecurityMode(security),
-        "server_options": server_options or [],
-        **kwargs,
-    }
-
-    # Add credentials if secure mode
-    if security == "secure":
-        if not server_key_path or not server_cert_path:
-            raise ValueError(
-                "Server key and certificate paths are required for secure mode"
-            )
-
-        config_params["credentials"] = ServerCredentials(
-            server_key_path=Path(server_key_path),
-            server_cert_path=Path(server_cert_path),
-            root_cert_path=Path(root_cert_path) if root_cert_path else None,
-        )
-
-    return config_class(**config_params)