digitalkin 0.2.19__tar.gz → 0.2.21__tar.gz

{digitalkin-0.2.19 → digitalkin-0.2.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.2.19
+Version: 0.2.21
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -462,9 +462,9 @@ Requires-Dist: typos>=1.34.0; extra == "dev"
 Requires-Dist: ruff>=0.12.5; extra == "dev"
 Requires-Dist: mypy>=1.17.0; extra == "dev"
 Requires-Dist: pyright>=1.1.403; extra == "dev"
-Requires-Dist: pre-commit>=4.2.0; extra == "dev"
+Requires-Dist: pre-commit>=4.3.0; extra == "dev"
 Requires-Dist: bump2version>=1.0.1; extra == "dev"
-Requires-Dist: build>=1.2.2; extra == "dev"
+Requires-Dist: build>=1.3.0; extra == "dev"
 Requires-Dist: twine>=6.1.0; extra == "dev"
 Requires-Dist: cryptography>=45.0.5; extra == "dev"
 Provides-Extra: examples

{digitalkin-0.2.19 → digitalkin-0.2.21}/pyproject.toml

@@ -12,7 +12,7 @@
 
     keywords = [ "digitalkin", "kin", "agent", "gprc", "sdk" ]
     # Version of the package automatically updated by bump2version (that is why it is separated)
-    version = "0.2.19"
+    version = "0.2.21"
 
     classifiers = [
         "Development Status :: 3 - Alpha",
@@ -42,9 +42,9 @@
             "ruff>=0.12.5",
             "mypy>=1.17.0",
             "pyright>=1.1.403",
-            "pre-commit>=4.2.0",
+            "pre-commit>=4.3.0",
             "bump2version>=1.0.1",
-            "build>=1.2.2",
+            "build>=1.3.0",
             "twine>=6.1.0",
             "cryptography>=45.0.5",
         ]

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/__version__.py

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

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/grpc_servers/module_servicer.py

@@ -100,6 +100,9 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
             ServicerError: if the setup data is not returned or job creation fails.
         """
         logger.info("ConfigSetupVersion called for module: '%s'", self.module_class.__name__)
+        logger.info(
+            "Context : %s, setup_version: %s, mission_id: %s", context, request.setup_version, request.mission_id
+        )
         # Process the module input
         # TODO: Secret should be used here as well
         setup_version = request.setup_version
@@ -121,6 +124,7 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
         job_id = await self.job_manager.create_config_setup_instance_job(
             config_setup_data,
             request.mission_id,
+            setup_version.setup_id,
             setup_version.id,
         )
 
@@ -156,6 +160,7 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
             ServicerError: the necessary query didn't work.
         """
         logger.info("StartModule called for module: '%s'", self.module_class.__name__)
+        logger.info("Context : %s, setup_id: %s, mission_id: %s", context, request.setup_id, request.mission_id)
         # Process the module input
         # TODO: Check failure of input data format
         input_data = self.module_class.create_input_model(dict(request.input.items()))
@@ -177,6 +182,7 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
             input_data,
             setup_data,
             mission_id=request.mission_id,
+            setup_id=setup_data_class.current_setup_version.setup_id,
             setup_version_id=setup_data_class.current_setup_version.id,
         )
 

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/modules/_base_module.py

@@ -54,7 +54,6 @@ class BaseModule(  # noqa: PLR0904
     description: str
 
     setup_format: type[SetupModelT]
-
     input_format: type[InputModelT]
     output_format: type[OutputModelT]
     secret_format: type[SecretModelT]
@@ -77,21 +76,38 @@ class BaseModule(  # noqa: PLR0904
     snapshot: SnapshotStrategy
     storage: StorageStrategy
 
+    # runtime params
+    job_id: str
+    mission_id: str
+    setup_id: str
+    setup_version_id: str
+    _status: ModuleStatus
+    _task: asyncio.Task | None
+
     def _init_strategies(self) -> None:
         """Initialize the services configuration."""
         for service_name in self.services_config.valid_strategy_names():
-            service = self.services_config.init_strategy(service_name, self.mission_id, self.setup_version_id)
+            service = self.services_config.init_strategy(
+                service_name,
+                self.mission_id,
+                self.setup_id,
+                self.setup_version_id,
+            )
             setattr(self, service_name, service)
 
     def __init__(
         self,
         job_id: str,
         mission_id: str,
+        setup_id: str,
         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
         self._task: asyncio.Task | None = None

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/modules/job_manager/base_job_manager.py

@@ -87,6 +87,7 @@ class BaseJobManager(abc.ABC, Generic[InputModelT, SetupModelT]):
         input_data: InputModelT,
         setup_data: SetupModelT,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
     ) -> str:
         """Create and start a new job for the module's instance.
@@ -95,7 +96,8 @@ class BaseJobManager(abc.ABC, Generic[InputModelT, SetupModelT]):
             input_data: The input data required to start the job.
             setup_data: The setup configuration for the module.
             mission_id: The mission ID associated with the job.
-            setup_version_id: The setup ID associated with the module.
+            setup_id: The setup ID.
+            setup_version_id: The setup version ID associated with the module.
 
         Returns:
             str: The unique identifier (job ID) of the created job.
@@ -121,6 +123,7 @@ class BaseJobManager(abc.ABC, Generic[InputModelT, SetupModelT]):
         self,
         config_setup_data: SetupModelT,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
     ) -> str:
         """Create and start a new module job.
@@ -131,7 +134,8 @@ class BaseJobManager(abc.ABC, Generic[InputModelT, SetupModelT]):
         Args:
             config_setup_data: The input data required to start the job.
             mission_id: The mission ID associated with the job.
-            setup_version_id: The setup ID.
+            setup_id: The setup ID.
+            setup_version_id: The setup version ID.
 
         Returns:
             str: The unique identifier (job ID) of the created job.

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/modules/job_manager/single_job_manager.py

@@ -71,6 +71,7 @@ class SingleJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
         self,
         config_setup_data: SetupModelT,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
     ) -> str:
         """Create and start a new module setup configuration job.
@@ -82,6 +83,7 @@ class SingleJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
             config_setup_data: The input data required to start the job.
             setup_data: The setup configuration for the module.
             mission_id: The mission ID associated with the job.
+            setup_id: The setup ID associated with the module.
             setup_version_id: The setup ID.
 
         Returns:
@@ -92,7 +94,7 @@ class SingleJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
         """
         job_id = str(uuid.uuid4())
         # TODO: Ensure the job_id is unique.
-        module = self.module_class(job_id, mission_id=mission_id, setup_version_id=setup_version_id)
+        module = self.module_class(job_id, mission_id=mission_id, setup_id=setup_id, setup_version_id=setup_version_id)
         self.modules[job_id] = module
         self.queues[job_id] = asyncio.Queue()
 
@@ -176,6 +178,7 @@ class SingleJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
         input_data: InputModelT,
         setup_data: SetupModelT,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
     ) -> str:
         """Create and start a new module job.
@@ -187,7 +190,8 @@ class SingleJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
             input_data: The input data required to start the job.
             setup_data: The setup configuration for the module.
             mission_id: The mission ID associated with the job.
-            setup_version_id: The setup ID associated with the module.
+            setup_id: The setup ID associated with the module.
+            setup_version_id: The setup Version ID associated with the module.
 
         Returns:
             str: The unique identifier (job ID) of the created job.
@@ -197,7 +201,12 @@ class SingleJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
         """
         job_id = str(uuid.uuid4())
         # TODO: Ensure the job_id is unique.
-        module = self.module_class(job_id, mission_id=mission_id, setup_version_id=setup_version_id)
+        module = self.module_class(
+            job_id,
+            mission_id=mission_id,
+            setup_id=setup_id,
+            setup_version_id=setup_version_id,
+        )
         self.modules[job_id] = module
         self.queues[job_id] = asyncio.Queue()
         callback = await self.job_specific_callback(self.add_to_queue, job_id)

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/modules/job_manager/taskiq_broker.py

@@ -132,6 +132,7 @@ async def send_message_to_stream(job_id: str, output_data: OutputModelT) -> None
 @TASKIQ_BROKER.task
 async def run_start_module(
     mission_id: str,
+    setup_id: str,
     setup_version_id: str,
     module_class: type[BaseModule],
     services_mode: ServicesMode,
@@ -143,6 +144,7 @@ async def run_start_module(
 
     Args:
         mission_id: str,
+        setup_id: The setup ID associated with the module.
         setup_version_id: The setup ID associated with the module.
         module_class: type[BaseModule],
         services_mode: ServicesMode,
@@ -161,7 +163,7 @@ async def run_start_module(
 
     job_id = context.message.task_id
     callback = await BaseJobManager.job_specific_callback(send_message_to_stream, job_id)
-    module = module_class(job_id, mission_id=mission_id, setup_version_id=setup_version_id)
+    module = module_class(job_id, mission_id=mission_id, setup_id=setup_id, setup_version_id=setup_version_id)
 
     await module.start(
         input_data,
@@ -176,6 +178,7 @@ async def run_start_module(
 @TASKIQ_BROKER.task
 async def run_config_module(
     mission_id: str,
+    setup_id: str,
     setup_version_id: str,
     module_class: type[BaseModule],
     services_mode: ServicesMode,
@@ -186,6 +189,7 @@ async def run_config_module(
 
     Args:
         mission_id: str,
+        setup_id: The setup ID associated with the module.
         setup_version_id: The setup ID associated with the module.
         module_class: type[BaseModule],
         services_mode: ServicesMode,
@@ -204,7 +208,7 @@ async def run_config_module(
 
     job_id = context.message.task_id
     callback = await BaseJobManager.job_specific_callback(send_message_to_stream, job_id)
-    module = module_class(job_id, mission_id=mission_id, setup_version_id=setup_version_id)
+    module = module_class(job_id, mission_id=mission_id, setup_id=setup_id, setup_version_id=setup_version_id)
 
     await module.start_config_setup(
         module_class.create_config_setup_model(config_setup_data),

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/modules/job_manager/taskiq_job_manager.py

@@ -136,6 +136,7 @@ class TaskiqJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
         self,
         config_setup_data: SetupModelT,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
     ) -> str:
         """Create and start a new module setup configuration job.
@@ -147,6 +148,7 @@ class TaskiqJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
             config_setup_data: The input data required to start the job.
             setup_data: The setup configuration for the module.
             mission_id: The mission ID associated with the job.
+            setup_id: The setup ID associated with the module.
             setup_version_id: The setup ID.
 
         Returns:
@@ -168,6 +170,7 @@ class TaskiqJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
 
         running_task: AsyncTaskiqTask[Any] = await task.kiq(
             mission_id,
+            setup_id,
             setup_version_id,
             self.module_class,
             self.services_mode,
@@ -221,6 +224,7 @@ class TaskiqJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
         input_data: InputModelT,
         setup_data: SetupModelT,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
     ) -> str:
         """Launches the module_task in Taskiq, returns the Taskiq task id as job_id.
@@ -229,6 +233,7 @@ class TaskiqJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
             input_data: Input data for the module
             setup_data: Setup data for the module
             mission_id: Mission ID for the module
+            setup_id: The setup ID associated with the module.
             setup_version_id: The setup ID associated with the module.
 
         Returns:
@@ -245,6 +250,7 @@ class TaskiqJobManager(BaseJobManager, Generic[InputModelT, SetupModelT]):
 
         running_task: AsyncTaskiqTask[Any] = await task.kiq(
             mission_id,
+            setup_id,
             setup_version_id,
             self.module_class,
             self.services_mode,

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/base_strategy.py

@@ -9,12 +9,14 @@ class BaseStrategy(ABC):
     This class defines the interface for all strategies.
     """
 
-    def __init__(self, mission_id: str, setup_version_id: str) -> None:
+    def __init__(self, mission_id: str, setup_id: str, setup_version_id: str) -> None:
         """Initialize the strategy.
 
         Args:
             mission_id: The ID of the mission this strategy is associated with
+            setup_id: The ID of the setup this strategy is associated with
             setup_version_id: The ID of the setup version this strategy is associated with
         """
         self.mission_id: str = mission_id
+        self.setup_id: str = setup_id
         self.setup_version_id: str = setup_version_id

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/cost/cost_strategy.py

@@ -57,15 +57,22 @@ class CostServiceError(Exception):
 class CostStrategy(BaseStrategy, ABC):
     """Abstract base class for cost strategies."""
 
-    def __init__(self, mission_id: str, setup_version_id: str, config: dict[str, CostConfig]) -> None:
+    def __init__(
+        self,
+        mission_id: str,
+        setup_id: str,
+        setup_version_id: str,
+        config: dict[str, CostConfig],
+    ) -> None:
         """Initialize the strategy.
 
         Args:
             mission_id: The ID of the mission this strategy is associated with
+            setup_id: The ID of the setup
             setup_version_id: The ID of the setup version this strategy is associated with
             config: Configuration dictionary for the strategy
         """
-        super().__init__(mission_id, setup_version_id)
+        super().__init__(mission_id, setup_id, setup_version_id)
         self.config = config
 
     @abstractmethod

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/cost/default_cost.py

@@ -15,15 +15,16 @@ from digitalkin.services.cost.cost_strategy import (
 class DefaultCost(CostStrategy):
     """Default cost strategy."""
 
-    def __init__(self, mission_id: str, setup_version_id: str, config: dict[str, CostConfig]) -> None:
+    def __init__(self, mission_id: str, setup_id: str, setup_version_id: str, config: dict[str, CostConfig]) -> None:
         """Initialize the strategy.
 
         Args:
             mission_id: The ID of the mission this strategy is associated with
+            setup_id: The ID of the setup
             setup_version_id: The ID of the setup version this strategy is associated with
             config: The configuration dictionary for the cost
         """
-        super().__init__(mission_id=mission_id, setup_version_id=setup_version_id, config=config)
+        super().__init__(mission_id=mission_id, setup_id=setup_id, setup_version_id=setup_version_id, config=config)
         self.db: dict[str, list[CostData]] = {}
 
     def add(

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/cost/grpc_cost.py

@@ -57,12 +57,13 @@ class GrpcCost(CostStrategy, GrpcClientWrapper):
     def __init__(
         self,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
         config: dict[str, CostConfig],
         client_config: ClientConfig,
     ) -> None:
         """Initialize the cost."""
-        super().__init__(mission_id=mission_id, setup_version_id=setup_version_id, config=config)
+        super().__init__(mission_id=mission_id, setup_id=setup_id, setup_version_id=setup_version_id, config=config)
         channel = self._init_channel(client_config)
         self.stub = cost_service_pb2_grpc.CostServiceStub(channel)
         logger.debug("Channel client 'Cost' initialized succesfully")

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/filesystem/default_filesystem.py

@@ -25,14 +25,15 @@ class DefaultFilesystem(FilesystemStrategy):
     Files are stored in a temporary directory with proper metadata tracking.
     """
 
-    def __init__(self, mission_id: str, setup_version_id: str) -> None:
+    def __init__(self, mission_id: str, setup_id: str, setup_version_id: str) -> None:
         """Initialize the default filesystem strategy.
 
         Args:
             mission_id: The ID of the mission this strategy is associated with
+            setup_id: The ID of the setup
             setup_version_id: The ID of the setup version this strategy is associated with
         """
-        super().__init__(mission_id, setup_version_id)
+        super().__init__(mission_id, setup_id, setup_version_id)
         self.temp_root: str = tempfile.mkdtemp()
         os.makedirs(self.temp_root, exist_ok=True)
         self.db: dict[str, FilesystemRecord] = {}
@@ -83,7 +84,6 @@ class DefaultFilesystem(FilesystemStrategy):
             if (not filters.names or f.name in filters.names)
             and (not filters.file_ids or f.id in filters.file_ids)
             and (not filters.file_types or f.file_type in filters.file_types)
-            and f.context == self.mission_id
             and (not filters.status or f.status == filters.status)
             and (not filters.content_type_prefix or f.content_type.startswith(filters.content_type_prefix))
             and (not filters.min_size_bytes or f.size_bytes >= filters.min_size_bytes)
@@ -118,7 +118,7 @@ class DefaultFilesystem(FilesystemStrategy):
         for file in files:
             try:
                 # Check if file with same name exists in the context
-                context_dir = self._get_context_temp_dir(self.mission_id)
+                context_dir = self._get_context_temp_dir(self.setup_id)
                 file_path = os.path.join(context_dir, file.name)
                 if os.path.exists(file_path) and not file.replace_if_exists:
                     msg = f"File with name {file.name} already exists."
@@ -129,7 +129,7 @@ class DefaultFilesystem(FilesystemStrategy):
                 storage_uri = str(Path(file_path).resolve())
                 file_data = FilesystemRecord(
                     id=str(uuid.uuid4()),
-                    context=self.mission_id,
+                    context=self.setup_id,
                     name=file.name,
                     file_type=file.file_type,
                     content_type=file.content_type or "application/octet-stream",
@@ -138,14 +138,13 @@ class DefaultFilesystem(FilesystemStrategy):
                     metadata=file.metadata,
                     storage_uri=storage_uri,
                     file_url=storage_uri,
-                    status=file.status if hasattr(file, "status") and file.status else "ACTIVE",
+                    status="ACTIVE",
                 )
 
                 self.db[file_data.id] = file_data
                 uploaded_files.append(file_data)
                 total_uploaded += 1
                 logger.debug("Uploaded file %s", file_data)
-
             except Exception as e:  # noqa: PERF203
                 logger.exception("Error uploading file %s: %s", file.name, e)
                 total_failed += 1
@@ -199,6 +198,8 @@ 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()
@@ -213,6 +214,7 @@ class DefaultFilesystem(FilesystemStrategy):
     def get_file(
         self,
         file_id: str,
+        context: Literal["mission", "setup"] = "mission",  # noqa: ARG002
         *,
         include_content: bool = False,
     ) -> FilesystemRecord:
@@ -224,6 +226,7 @@ class DefaultFilesystem(FilesystemStrategy):
 
         Args:
             file_id: The ID of the file to be retrieved
+            context: The context of the files (mission or setup)
             include_content: Whether to include file content in response
 
         Returns:
@@ -306,7 +309,7 @@ class DefaultFilesystem(FilesystemStrategy):
             raise FilesystemServiceError(msg)
 
         try:
-            context_dir = self._get_context_temp_dir(self.mission_id)
+            context_dir = self._get_context_temp_dir(self.setup_id)
             file_path = os.path.join(context_dir, file_id)
             existing_file = self.db[file_id]
 

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/filesystem/filesystem_strategy.py

@@ -33,6 +33,9 @@ class FilesystemRecord(BaseModel):
 class FileFilter(BaseModel):
     """Filter criteria for querying files."""
 
+    context: Literal["mission", "setup"] = Field(
+        default="mission", description="The context of the files (mission or setup)"
+    )
     names: list[str] | None = Field(default=None, description="Filter by file names (exact matches)")
     file_ids: list[str] | None = Field(default=None, description="Filter by file IDs")
     file_types: (
@@ -90,15 +93,22 @@ class FilesystemStrategy(BaseStrategy, ABC):
     filtering, and pagination.
     """
 
-    def __init__(self, mission_id: str, setup_version_id: str, config: dict[str, Any] | None = None) -> None:
+    def __init__(
+        self,
+        mission_id: str,
+        setup_id: str,
+        setup_version_id: str,
+        config: dict[str, Any] | None = None,
+    ) -> None:
         """Initialize the strategy.
 
         Args:
             mission_id: The ID of the mission this strategy is associated with
+            setup_id: The ID of the setup
             setup_version_id: The ID of the setup version this strategy is associated with
             config: Configuration for the filesystem strategy
         """
-        super().__init__(mission_id, setup_version_id)
+        super().__init__(mission_id, setup_id, setup_version_id)
         self.config = config
 
     @abstractmethod
@@ -123,9 +133,10 @@ class FilesystemStrategy(BaseStrategy, ABC):
     def get_file(
         self,
         file_id: str,
+        context: Literal["mission", "setup"] = "mission",
         *,
         include_content: bool = False,
-    ) -> tuple[FilesystemRecord, bytes | None]:
+    ) -> FilesystemRecord:
         """Get a specific file by ID or name.
 
         This method fetches detailed information about a single file,
@@ -134,6 +145,7 @@ class FilesystemStrategy(BaseStrategy, ABC):
 
         Args:
             file_id: The ID of the file to be retrieved
+            context: The context of the files (mission or setup)
             include_content: Whether to include file content in response
 
         Returns:

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/filesystem/grpc_filesystem.py

@@ -85,25 +85,8 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
         except AttributeError:
             return filesystem_pb2.FileStatus.FILE_STATUS_UNSPECIFIED
 
-    def _filter_to_proto(self, filters: FileFilter) -> filesystem_pb2.FileFilter:
-        """Convert a FileFilter to a FileFilter proto message.
-
-        Args:
-            filters: The FileFilter to convert
-
-        Returns:
-            filesystem_pb2.FileFilter: The converted FileFilter proto message
-        """
-        return filesystem_pb2.FileFilter(
-            context=self.mission_id,
-            **filters.model_dump(exclude={"file_types", "status"}),
-            file_types=[self._file_type_to_enum(file_type) for file_type in filters.file_types]
-            if filters.file_types
-            else None,
-            status=self._file_status_to_enum(filters.status) if filters.status else None,
-        )
-
-    def _file_proto_to_data(self, file: filesystem_pb2.File) -> FilesystemRecord:
+    @staticmethod
+    def _file_proto_to_data(file: filesystem_pb2.File) -> FilesystemRecord:
         """Convert a File proto message to FilesystemRecord.
 
         Args:
@@ -114,7 +97,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
         """
         return FilesystemRecord(
             id=file.file_id,
-            context=self.mission_id,
+            context=file.context,
             name=file.name,
             file_type=filesystem_pb2.FileType.Name(file.file_type),
             content_type=file.content_type,
@@ -127,9 +110,27 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
             content=file.content,
         )
 
+    def _filter_to_proto(self, filters: FileFilter) -> filesystem_pb2.FileFilter:
+        """Convert a FileFilter to a FileFilter proto message.
+
+        Args:
+            filters: The FileFilter to convert
+
+        Returns:
+            filesystem_pb2.FileFilter: The converted FileFilter proto message
+        """
+        return filesystem_pb2.FileFilter(
+            **filters.model_dump(exclude={"file_types", "status"}),
+            file_types=[self._file_type_to_enum(file_type) for file_type in filters.file_types]
+            if filters.file_types
+            else None,
+            status=self._file_status_to_enum(filters.status) if filters.status else None,
+        )
+
     def __init__(
         self,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
         client_config: ClientConfig,
         config: dict[str, Any] | None = None,
@@ -138,11 +139,12 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
 
         Args:
             mission_id: The ID of the mission this strategy is associated with
+            setup_id: The ID of the setup
             setup_version_id: The ID of the setup version this strategy is associated with
             client_config: Configuration for the gRPC client connection
             config: Configuration for the filesystem strategy
         """
-        super().__init__(mission_id, setup_version_id, config)
+        super().__init__(mission_id, setup_id, setup_version_id, config)
         self.service_name = "FilesystemService"
         channel = self._init_channel(client_config)
         self.stub = filesystem_service_pb2_grpc.FilesystemServiceStub(channel)
@@ -189,6 +191,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
     def get_file(
         self,
         file_id: str,
+        context: Literal["mission", "setup"] = "mission",
         *,
         include_content: bool = False,
     ) -> FilesystemRecord:
@@ -196,6 +199,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
 
         Args:
             file_id: The ID of the file to be retrieved
+            context: The context of the files (mission or setup)
             include_content: Whether to include file content in response
 
         Returns:
@@ -204,9 +208,14 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
         Raises:
             FilesystemServiceError: If there is an error retrieving the file
         """
+        match context:
+            case "setup":
+                context_id = self.setup_id
+            case "mission":
+                context_id = self.mission_id
         with GrpcFilesystem._handle_grpc_errors("GetFile"):
             request = filesystem_pb2.GetFileRequest(
-                context=self.mission_id,
+                context=context_id,
                 file_id=file_id,
                 include_content=include_content,
             )
@@ -320,7 +329,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
         """
         with GrpcFilesystem._handle_grpc_errors("GetFiles"):
             request = filesystem_pb2.GetFilesRequest(
-                context=self.mission_id,
+                context=filters.context,
                 filters=self._filter_to_proto(filters),
                 include_content=include_content,
                 list_size=list_size,

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/services_config.py

@@ -111,12 +111,13 @@ class ServicesConfig(BaseModel):
         """
         return getattr(self, f"_config_{name}", {})
 
-    def init_strategy(self, name: str, mission_id: str, setup_version_id: str) -> ServicesStrategy:
+    def init_strategy(self, name: str, mission_id: str, setup_id: str, setup_version_id: str) -> ServicesStrategy:
         """Initialize a specific strategy.
 
         Args:
             name: The name of the strategy to initialize
             mission_id: The ID of the mission this strategy is associated with
+            setup_id: The setup ID for the strategy
             setup_version_id: The setup version ID for the strategy
 
         Returns:
@@ -131,7 +132,7 @@ class ServicesConfig(BaseModel):
             raise ValueError(msg)
 
         # Instantiate the strategy with the mission ID, setup version ID, and configuration
-        return strategy_type(mission_id, setup_version_id, **self.get_strategy_config(name) or {})
+        return strategy_type(mission_id, setup_id, setup_version_id, **self.get_strategy_config(name) or {})
 
     @property
     def storage(self) -> type[StorageStrategy]:

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/storage/default_storage.py

@@ -215,13 +215,14 @@ class DefaultStorage(StorageStrategy):
     def __init__(
         self,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
         config: dict[str, type[BaseModel]],
         storage_file_path: str = "local_storage",
         **kwargs,  # noqa: ANN003, ARG002
     ) -> None:
         """Initialize the storage."""
-        super().__init__(mission_id=mission_id, setup_version_id=setup_version_id, config=config)
+        super().__init__(mission_id=mission_id, setup_id=setup_id, setup_version_id=setup_version_id, config=config)
         self.storage_file_path = f"{self.mission_id}_{storage_file_path}.json"
         self.storage_file = Path(self.storage_file_path)
         self.storage = self._load_from_file()

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/storage/grpc_storage.py

@@ -200,13 +200,14 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
     def __init__(
         self,
         mission_id: str,
+        setup_id: str,
         setup_version_id: str,
         config: dict[str, type[BaseModel]],
         client_config: ClientConfig,
         **kwargs,  # noqa: ANN003, ARG002
     ) -> None:
         """Initialize the storage."""
-        super().__init__(mission_id=mission_id, setup_version_id=setup_version_id, config=config)
+        super().__init__(mission_id=mission_id, setup_id=setup_id, setup_version_id=setup_version_id, config=config)
 
         channel = self._init_channel(client_config)
         self.stub = storage_service_pb2_grpc.StorageServiceStub(channel)

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin/services/storage/storage_strategy.py

@@ -163,15 +163,22 @@ class StorageStrategy(BaseStrategy, ABC):
             True if the deletion was successful, False otherwise
         """
 
-    def __init__(self, mission_id: str, setup_version_id: str, config: dict[str, type[BaseModel]]) -> None:
+    def __init__(
+        self,
+        mission_id: str,
+        setup_id: str,
+        setup_version_id: str,
+        config: dict[str, type[BaseModel]],
+    ) -> None:
         """Initialize the storage strategy.
 
         Args:
             mission_id: The ID of the mission this strategy is associated with
+            setup_id: The ID of the setup
             setup_version_id: The ID of the setup version
             config: A dictionary mapping names to Pydantic model classes
         """
-        super().__init__(mission_id, setup_version_id)
+        super().__init__(mission_id, setup_id, setup_version_id)
         # Schema configuration mapping keys to model classes
         self.config: dict[str, type[BaseModel]] = config
 

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.2.19
+Version: 0.2.21
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -462,9 +462,9 @@ Requires-Dist: typos>=1.34.0; extra == "dev"
 Requires-Dist: ruff>=0.12.5; extra == "dev"
 Requires-Dist: mypy>=1.17.0; extra == "dev"
 Requires-Dist: pyright>=1.1.403; extra == "dev"
-Requires-Dist: pre-commit>=4.2.0; extra == "dev"
+Requires-Dist: pre-commit>=4.3.0; extra == "dev"
 Requires-Dist: bump2version>=1.0.1; extra == "dev"
-Requires-Dist: build>=1.2.2; extra == "dev"
+Requires-Dist: build>=1.3.0; extra == "dev"
 Requires-Dist: twine>=6.1.0; extra == "dev"
 Requires-Dist: cryptography>=45.0.5; extra == "dev"
 Provides-Extra: examples

{digitalkin-0.2.19 → digitalkin-0.2.21}/src/digitalkin.egg-info/requires.txt

@@ -9,9 +9,9 @@ typos>=1.34.0
 ruff>=0.12.5
 mypy>=1.17.0
 pyright>=1.1.403
-pre-commit>=4.2.0
+pre-commit>=4.3.0
 bump2version>=1.0.1
-build>=1.2.2
+build>=1.3.0
 twine>=6.1.0
 cryptography>=45.0.5