digitalkin 0.2.9__tar.gz → 0.2.11__tar.gz

{digitalkin-0.2.9 → digitalkin-0.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.2.9
+Version: 0.2.11
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International

{digitalkin-0.2.9 → digitalkin-0.2.11}/examples/modules/storage_module.py

@@ -62,7 +62,7 @@ class ExampleModule(ArchetypeModule[ExampleInput, ExampleOutput, ExampleSetup, E
 
     # Define services_config_params with default values
     services_config_strategies = {}
-    services_config_params = {"storage": {"config": {"example_outputs": ExampleOutput}}, "filesystem": {"config": {}}}
+    services_config_params = {"storage": {"config": {"example": ExampleOutput}}, "filesystem": {"config": {}}}
 
     def __init__(self, job_id: str, mission_id: str) -> None:
         """Initialize the example module.
@@ -172,7 +172,7 @@ async def test_module() -> None:
 def test_storage_directly() -> None:
     """Test the storage service directly."""
     # Initialize storage service
-    storage = ServicesConfig().storage(mission_id="test-mission", config={"test_table": ExampleStorage})
+    storage = ServicesConfig().storage(mission_id="test-mission", config={"example": ExampleStorage})
 
     # Create a test record
     storage.store("example", "test_table", {"test_key": "test_value"}, "OUTPUT")

{digitalkin-0.2.9 → digitalkin-0.2.11}/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.9"
+    version = "0.2.11"
 
     classifiers = [
         "Development Status :: 3 - Alpha",

{digitalkin-0.2.9 → digitalkin-0.2.11}/src/digitalkin/__version__.py

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

{digitalkin-0.2.9 → digitalkin-0.2.11}/src/digitalkin/services/storage/grpc_storage.py

@@ -37,7 +37,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
         dtype = DataType[raw["data_type"]]
         payload = raw.get("data", {})
 
-        validated = self._validate_data(rid, payload)
+        validated = self._validate_data(coll, payload)
         return StorageRecord(
             mission_id=mission,
             collection=coll,
@@ -114,10 +114,10 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
                 collection=collection,
                 record_id=record_id,
             )
-            resp = self.exec_grpc_query("ModifyRecord", req)
+            resp = self.exec_grpc_query("UpdateRecord", req)
             return self._build_record_from_proto(resp.stored_data)
         except Exception:
-            logger.exception("gRPC ModifyRecord failed for %s:%s", collection, record_id)
+            logger.exception("gRPC UpdateRecord failed for %s:%s", collection, record_id)
             return None
 
     def _remove(self, collection: str, record_id: str) -> bool:

{digitalkin-0.2.9 → digitalkin-0.2.11}/src/digitalkin/services/storage/storage_strategy.py

@@ -39,11 +39,11 @@ class StorageRecord(BaseModel):
 class StorageStrategy(BaseStrategy, ABC):
     """Define CRUD + list/remove-collection against a collection/record store."""
 
-    def _validate_data(self, record_id: str, data: dict[str, Any]) -> BaseModel:
+    def _validate_data(self, collection: str, data: dict[str, Any]) -> BaseModel:
         """Validate data against the model schema for the given key.
 
         Args:
-            record_id: The unique ID for the record
+            collection: The unique name for the record type
             data: The data to validate
 
         Returns:
@@ -52,15 +52,15 @@ class StorageStrategy(BaseStrategy, ABC):
         Raises:
             ValueError: If the key has no associated model or validation fails
         """
-        model_cls = self.config.get(record_id)
+        model_cls = self.config.get(collection)
         if not model_cls:
-            msg = f"No schema registered for collection '{record_id}'"
+            msg = f"No schema registered for collection '{collection}'"
             raise ValueError(msg)
 
         try:
             return model_cls.model_validate(data)
         except Exception as e:
-            msg = f"Validation failed for '{record_id}': {e!s}"
+            msg = f"Validation failed for '{collection}': {e!s}"
             raise ValueError(msg) from e
 
     def _create_storage_record(
@@ -200,7 +200,7 @@ class StorageStrategy(BaseStrategy, ABC):
             raise ValueError(msg)
         record_id = record_id or uuid4().hex
         data_type_enum = DataType[data_type]
-        validated_data = self._validate_data(record_id, {**data, "mission_id": self.mission_id})
+        validated_data = self._validate_data(collection, {**data, "mission_id": self.mission_id})
         record = self._create_storage_record(collection, record_id, validated_data, data_type_enum)
         return self._store(record)
 
@@ -227,7 +227,7 @@ class StorageStrategy(BaseStrategy, ABC):
         Returns:
             StorageRecord: The modified record
         """
-        validated_data = self._validate_data(record_id, data)
+        validated_data = self._validate_data(collection, data)
         return self._update(collection, record_id, validated_data)
 
     def remove(self, collection: str, record_id: str) -> bool:

{digitalkin-0.2.9 → digitalkin-0.2.11}/src/digitalkin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.2.9
+Version: 0.2.11
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International