digitalkin 0.2.5__tar.gz → 0.2.6__tar.gz

{digitalkin-0.2.5 → digitalkin-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.2.5
+Version: 0.2.6
 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.5 → digitalkin-0.2.6}/examples/modules/minimal_llm_module.py

@@ -8,7 +8,7 @@ import grpc
 import openai
 from pydantic import BaseModel
 
-from digitalkin.grpc_servers.utils.models import SecurityMode, ServerConfig, ServerMode
+from digitalkin.grpc_servers.utils.models import SecurityMode, ClientConfig, ServerMode
 from digitalkin.modules._base_module import BaseModule
 from digitalkin.services.setup.setup_strategy import SetupData
 
@@ -45,12 +45,11 @@ class OpenAIToolSecret(BaseModel):
     """Secret model defining module configuration parameters."""
 
 
-server_config = ServerConfig(
+client_config = ClientConfig(
     host="[::]",
     port=50151,
     mode=ServerMode.ASYNC,
     security=SecurityMode.INSECURE,
-    max_workers=10,
     credentials=None,
 )
 
@@ -81,11 +80,11 @@ class OpenAIToolModule(BaseModule[OpenAIToolInput, OpenAIToolOutput, OpenAIToolS
     services_config_params = {
         "storage": {
             "config": {"setups": OpenAIToolSetup},
-            "server_config": server_config,
+            "client_config": client_config,
         },
         "filesystem": {
             "config": {},
-            "server_config": server_config,
+            "client_config": client_config,
         },
     }
 

{digitalkin-0.2.5 → digitalkin-0.2.6}/examples/modules/text_transform_module.py

@@ -6,7 +6,7 @@ from typing import Any, ClassVar
 
 from pydantic import BaseModel
 
-from digitalkin.grpc_servers.utils.models import SecurityMode, ServerConfig, ServerMode
+from digitalkin.grpc_servers.utils.models import SecurityMode, ClientConfig, ServerMode
 from digitalkin.modules._base_module import BaseModule
 from digitalkin.services.setup.setup_strategy import SetupData
 from digitalkin.services.storage.storage_strategy import DataType, StorageRecord
@@ -54,12 +54,11 @@ class TextTransformStorage(BaseModel):
     ended: bool = False
 
 
-server_config = ServerConfig(
+client_config = ClientConfig(
     host="[::]",
     port=50151,
     mode=ServerMode.ASYNC,
     security=SecurityMode.INSECURE,
-    max_workers=10,
     credentials=None,
 )
 
@@ -91,8 +90,12 @@ class TextTransformModule(BaseModule[TextTransformInput, TextTransformOutput, Te
     services_config_params = {
         "storage": {
             "config": {"monitor": TextTransformStorage, "setups": TextTransformStorage},
-            "server_config": server_config,
-        }
+            "client_config": client_config,
+        },
+        "filesystem": {
+            "config": {},
+            "client_config": client_config,
+        },
     }
 
     async def initialize(self, setup_data: SetupData) -> None:

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

{digitalkin-0.2.5 → digitalkin-0.2.6}/src/digitalkin/__version__.py

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

{digitalkin-0.2.5 → digitalkin-0.2.6}/src/digitalkin/services/filesystem/grpc_filesystem.py

@@ -10,7 +10,7 @@ from digitalkin_proto.digitalkin.filesystem.v2.filesystem_pb2 import FileType as
 
 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 ServerConfig
+from digitalkin.grpc_servers.utils.models import ClientConfig
 from digitalkin.services.filesystem.filesystem_strategy import (
     FilesystemData,
     FilesystemServiceError,
@@ -28,7 +28,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
         self,
         mission_id: str,
         config: dict[str, str],
-        server_config: ServerConfig,
+        client_config: ClientConfig,
         **kwargs,  # noqa: ANN003, ARG002
     ) -> None:
         """Initialize the default filesystem strategy.
@@ -36,11 +36,11 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
         Args:
             mission_id: The ID of the mission this strategy is associated with
             config: A dictionary mapping names to Pydantic model classes
-            server_config: The server configuration object
+            client_config: The client configuration object
             kwargs: other optional arguments to pass to the parent class constructor
         """
         super().__init__(mission_id, config)
-        channel = self._init_channel(server_config)
+        channel = self._init_channel(client_config)
         self.stub = filesystem_service_pb2_grpc.FilesystemServiceStub(channel)
         logger.info("Channel client 'Filesystem' initialized succesfully")
 

{digitalkin-0.2.5 → digitalkin-0.2.6}/src/digitalkin/services/storage/grpc_storage.py

@@ -8,8 +8,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 ServerConfig
-from digitalkin.services.storage.storage_strategy import StorageRecord, StorageServiceError, StorageStrategy
+from digitalkin.grpc_servers.utils.models import ClientConfig
+from digitalkin.services.storage.storage_strategy import DataType, StorageRecord, StorageServiceError, StorageStrategy
 
 logger = logging.getLogger(__name__)
 
@@ -21,13 +21,13 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
         self,
         mission_id: str,
         config: dict[str, type[BaseModel]],
-        server_config: ServerConfig,
+        client_config: ClientConfig,
         **kwargs,  # noqa: ANN003, ARG002
     ) -> None:
         """Initialize the storage."""
         super().__init__(mission_id=mission_id, config=config)
 
-        channel = self._init_channel(server_config)
+        channel = self._init_channel(client_config)
         self.stub = storage_service_pb2_grpc.StorageServiceStub(channel)
         logger.info("Channel client 'storage' initialized succesfully")
 
@@ -98,7 +98,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
 
             request = data_pb2.ModifyRecordRequest(data=data_struct, mission_id=self.mission_id, name=name)
             response: data_pb2.ModifyRecordResponse = self.exec_grpc_query("ModifyRecord", request)
-            return StorageRecord(**json_format.MessageToDict(response.stored_data))
+            return self._build_record_from_proto(response.stored_data, name)
         except Exception:
             msg = f"Error while modifing record {name}"
             logger.exception(msg)
@@ -121,3 +121,36 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
             logger.exception(msg)
             return False
         return True
+
+    def _build_record_from_proto(self, stored_data: data_pb2.StorageRecord, default_name: str) -> StorageRecord:
+        """Helper to construire un StorageRecord complet à partir du message gRPC.
+
+        Args:
+            stored_data: Le message gRPC contenant les données stockées.
+            default_name: Le nom par défaut à utiliser si le nom n'est pas présent dans les données.
+
+        Returns:
+            StorageRecord: Un objet StorageRecord construit à partir des données stockées.
+        """
+        # Converti en dict, avec tous les champs même s'ils sont absents
+        raw = json_format.MessageToDict(
+            stored_data,
+            preserving_proto_field_name=True,
+            always_print_fields_with_no_presence=True,
+        )
+        # On récupère ou on complète les champs obligatoires
+        name = raw.get("name", default_name)
+        dtype = raw.get("data_type", DataType.OUTPUT.name)
+        payload = raw.get("data", {})
+
+        # Valide le modèle pydantic pour le champ `data`
+        validated = self._validate_data(name, payload)
+
+        return StorageRecord(
+            mission_id=self.mission_id,
+            name=name,
+            data_type=DataType[dtype],
+            data=validated,
+            creation_date=raw.get("creation_date"),
+            update_date=raw.get("update_date"),
+        )

{digitalkin-0.2.5 → digitalkin-0.2.6}/src/digitalkin.egg-info/PKG-INFO

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