digitalkin 1.0.2.dev1__tar.gz → 1.0.2.dev2__tar.gz

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 1.0.2.dev1
+Version: 1.0.2.dev2
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -453,7 +453,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ag-ui-protocol>=0.1.18
-Requires-Dist: agentic-mesh-protocol==1.0.1.dev3
+Requires-Dist: agentic-mesh-protocol==1.0.1.dev4
 Requires-Dist: anyio>=4.13.0
 Requires-Dist: grpcio-health-checking==1.82.1
 Requires-Dist: grpcio-reflection==1.82.1

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/pyproject.toml

@@ -28,7 +28,7 @@
 
     dependencies = [
         "ag-ui-protocol>=0.1.18",
-        "agentic-mesh-protocol==1.0.1.dev3",
+        "agentic-mesh-protocol==1.0.1.dev4",
         "anyio>=4.13.0",
         "grpcio-health-checking==1.82.1",
         "grpcio-reflection==1.82.1",
@@ -37,7 +37,7 @@
         "pydantic>=2.12.4",
         "redis[hiredis]>=7.4.0,<9",
     ]
-    version = "1.0.2.dev1"
+    version = "1.0.2.dev2"
 
     [project.optional-dependencies]
         agno = [ "agno>=2.6" ]

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/__version__.py

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

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/community/agno/toolkits/registry.py

@@ -37,7 +37,7 @@ class RegistryTools(DkToolkit):
         self._registry = registry
         super().__init__(
             name="registry_tools",
-            tools=[self.search_setups, self.search_modules],
+            tools=[self.search_setups, self.search_modules, self.get_service_setup],
             context=context,
         )
 
@@ -49,10 +49,10 @@ class RegistryTools(DkToolkit):
             kind: The requested kind filter.
 
         Returns:
-            Error message when ``kind`` is neither 'tool' nor 'kin', else None.
+            Error message when ``kind`` is not 'tool', 'kin' or 'service', else None.
         """
-        if kind is not None and kind not in {"tool", "kin"}:
-            return f"invalid kind '{kind}'; use 'tool' or 'kin'"
+        if kind is not None and kind not in {"tool", "kin", "service"}:
+            return f"invalid kind '{kind}'; use 'tool', 'kin' or 'service'"
         return None
 
     def _clamp(self, limit: int) -> int:
@@ -67,16 +67,16 @@ class RegistryTools(DkToolkit):
         return min(max(limit, 1), self._MAX_RESULTS)
 
     async def search_setups(
-        self, query: str | None = None, kind: Literal["tool", "kin"] | None = None, limit: int = 10
+        self, query: str | None = None, kind: Literal["tool", "kin", "service"] | None = None, limit: int = 10
     ) -> str:
-        """Search ready-to-use setups (configured agent/tool instances you can actually invoke).
+        """Search ready-to-use setups (configured agent/tool/service instances you can actually invoke).
 
         A setup is an installed, configured instance of a module — the thing you can
-        call. Use this to discover which tools or agents (kins) are available.
+        call. Use this to discover which tools, agents (kins) or services are available.
 
         Args:
             query: Free text matched against setup name and documentation. Omit to list all.
-            kind: Optional filter: "tool" (invocable tools) or "kin" (agents).
+            kind: Optional filter: "tool" (invocable tools), "kin" (agents) or "service".
             limit: Max results (default 10, max 25).
 
         Returns:
@@ -90,7 +90,13 @@ class RegistryTools(DkToolkit):
         try:
             setups = await self._registry.search_setups(
                 query=query,
-                module_types=[RegistryModuleType.ARCHETYPE if kind == "kin" else RegistryModuleType.TOOL_MODULE]
+                module_types=[
+                    {
+                        "tool": RegistryModuleType.TOOL_MODULE,
+                        "kin": RegistryModuleType.ARCHETYPE,
+                        "service": RegistryModuleType.SERVICE,
+                    }[kind]
+                ]
                 if kind
                 else None,
                 statuses=[RegistrySetupStatus.READY, RegistrySetupStatus.CONFIGURATION_SUCCEEDED],
@@ -122,8 +128,33 @@ class RegistryTools(DkToolkit):
             tool="search_setups",
         )
 
+    async def get_service_setup(self, setup_id: str) -> str:
+        """Fetch a service's configuration content (a JSON document) by setup id.
+
+        Use after discovering a service via ``search_setups`` and the user accepted it:
+        pass the proposed setup's ``setup_id`` to read the service content. Always
+        returns the latest version.
+
+        Args:
+            setup_id: The service setup id (from a ``search_setups`` result).
+
+        Returns:
+            The canonical envelope; ``output`` = the service configuration JSON object.
+        """
+        try:
+            content = await self._registry.get_service_setup(setup_id)
+        except PermissionDeniedError:
+            return self._fail("permission denied: get_service_setup", tool="get_service_setup")
+        except RegistryServiceError as error:
+            logger.warning("RegistryTools: service setup fetch failed: %s", error)
+            return self._fail(self._UNAVAILABLE, tool="get_service_setup")
+
+        if content is None:
+            return self._fail(f"service setup '{setup_id}' not found or has no content", tool="get_service_setup")
+        return self._ok(content, tool="get_service_setup")
+
     async def search_modules(
-        self, query: str | None = None, kind: Literal["tool", "kin"] | None = None, limit: int = 10
+        self, query: str | None = None, kind: Literal["tool", "kin", "service"] | None = None, limit: int = 10
     ) -> str:
         """Search the module catalog (module TYPES, not configured instances).
 
@@ -133,7 +164,7 @@ class RegistryTools(DkToolkit):
 
         Args:
             query: Free text matched against module names. Omit to list all.
-            kind: Optional filter: "tool" or "kin" (agents/archetypes).
+            kind: Optional filter: "tool", "kin" (agents/archetypes) or "service".
             limit: Max results (default 10, max 25).
 
         Returns:
@@ -148,6 +179,8 @@ class RegistryTools(DkToolkit):
             pending = self._registry.search_tools(name=query, limit=cap)
         elif kind == "kin":
             pending = self._registry.search_kins(name=query, limit=cap)
+        elif kind == "service":
+            pending = self._registry.search_services(name=query, limit=cap)
         else:
             pending = self._registry.search(name=query, limit=cap)
         try:

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/grpc_servers/module_servicer.py

@@ -137,6 +137,8 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
 
     def invalidate_tool_cache(self) -> None:
         """Clear tool cache. Next request re-resolves tool definitions."""
+        if self._tool_cache_by_setup:
+            logger.info("tool cache invalidated, dropped setups: %s", list(self._tool_cache_by_setup))
         self._tool_cache_by_setup.clear()
 
     def get_tool_cache(self, setup_id: str) -> Any | None:
@@ -154,6 +156,7 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
         value, expires_at = entry
         if time.monotonic() >= expires_at:
             self._tool_cache_by_setup.pop(setup_id, None)
+            logger.debug("tool cache expired for setup '%s'", setup_id)
             return None
         return value
 
@@ -167,10 +170,14 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
         if len(self._tool_cache_by_setup) >= get_module_servicer_settings().setup_cache_max:
             oldest_key = next(iter(self._tool_cache_by_setup))
             del self._tool_cache_by_setup[oldest_key]
-        self._tool_cache_by_setup[setup_id] = (
-            value,
-            time.monotonic() + get_gateway_settings().queue.toolkit_cache_ttl_s,
-        )
+            logger.warning(
+                "tool cache full (%d), evicting setup '%s'",
+                get_module_servicer_settings().setup_cache_max,
+                oldest_key,
+            )
+        ttl_s = get_gateway_settings().queue.toolkit_cache_ttl_s
+        self._tool_cache_by_setup[setup_id] = (value, time.monotonic() + ttl_s)
+        logger.debug("tool cache set for setup '%s' (ttl %.0fs)", setup_id, ttl_s)
 
     async def get_or_build_tool_cache(
         self,
@@ -188,9 +195,11 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
         """
         cached = self.get_tool_cache(setup_id)
         if cached is not None:
+            logger.debug("tool cache hit for setup '%s'", setup_id)
             return cached
         inflight = self._tool_cache_inflight.get(setup_id)
         if inflight is not None:
+            logger.debug("tool cache build in flight for setup '%s', awaiting", setup_id)
             return await inflight
         loop = asyncio.get_event_loop()
         fut: asyncio.Future[Any] = loop.create_future()
@@ -205,6 +214,7 @@ class ModuleServicer(module_service_pb2_grpc.ModuleServiceServicer, ArgParser):
             # ``module_servicer.py:367``) clears the entry.
             if value is not None:
                 self.set_tool_cache(setup_id, value)
+                logger.info("tool cache built for setup '%s'", setup_id)
             fut.set_result(value)
         except Exception as exc:
             fut.set_exception(exc)

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/models/module/module_context.py

@@ -350,18 +350,43 @@ class ModuleContext:
         """
         setup = await self.registry.get_setup(setup_id)
         if setup is None or not setup.module_id:
+            logger.warning(
+                "resolve_tool: setup '%s' not found or has no module", setup_id, extra=self.session.current_ids()
+            )
             return None
         cached = self.tool_cache.entries.get(setup_id)
         if cached is not None:
+            logger.debug(
+                "resolve_tool: cache hit for setup '%s' (authz re-checked)", setup_id, extra=self.session.current_ids()
+            )
             return cached
         try:
             info = await self.registry.discover_by_id(setup.module_id)
         except RegistryModuleNotFoundError:
+            logger.warning(
+                "resolve_tool: module '%s' for setup '%s' not found in registry",
+                setup.module_id,
+                setup_id,
+                extra=self.session.current_ids(),
+            )
             return None
         if info is None:
+            logger.warning(
+                "resolve_tool: module '%s' for setup '%s' not found in registry",
+                setup.module_id,
+                setup_id,
+                extra=self.session.current_ids(),
+            )
             return None
         tool_info = await ToolModuleInfo.from_module_info(info, setup_id, setup.name, self.communication)
         self.tool_cache.add(tool_info)
+        logger.info(
+            "resolve_tool: resolved setup '%s' -> module '%s' (%d tools), cached",
+            setup_id,
+            setup.module_id,
+            len(tool_info.tools),
+            extra=self.session.current_ids(),
+        )
         return tool_info
 
     @staticmethod

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/models/services/registry.py

@@ -21,6 +21,7 @@ class RegistryModuleType(str, Enum):
     UNSPECIFIED = "unspecified"
     ARCHETYPE = "archetype"
     TOOL_MODULE = "tool_module"
+    SERVICE = "service"
 
 
 class ModuleInfo(BaseModel):

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/services/filesystem/grpc_filesystem.py

@@ -79,6 +79,29 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper, GrpcErrorHandlerMixi
             content=file.content,
         )
 
+    @staticmethod
+    def _context_enum(context: str) -> filesystem_pb2.ContextFile:
+        """Map a scope literal to the wire's context-kind enum.
+
+        Since dev4 the request carries only the kind; the concrete id is resolved
+        server-side from the x-mission-id / x-setup-id task metadata stamped by
+        ``RequestIdClientInterceptor``.
+
+        Args:
+            context: The scope literal ("mission" or "setup").
+
+        Returns:
+            The matching ``ContextFile`` enum value, ``CONTEXT_UNSPECIFIED`` otherwise.
+        """
+        # TODO(validate): remove after prod validation
+        # [VALIDATE CTXENUM] server resolves the concrete id from metadata
+        match context:
+            case "setup":
+                return filesystem_pb2.CONTEXT_SETUP
+            case "mission":
+                return filesystem_pb2.CONTEXT_MISSIONS
+        return filesystem_pb2.CONTEXT_UNSPECIFIED
+
     def _filter_to_proto(self, filters: FileFilter) -> filesystem_pb2.FileFilter:
         """Convert a FileFilter to a FileFilter proto message.
 
@@ -88,19 +111,13 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper, GrpcErrorHandlerMixi
         Returns:
             filesystem_pb2.FileFilter: The converted FileFilter proto message
         """
-        context_id = "unknown"
-        match filters.context:
-            case "setup":
-                context_id = self.setup_id
-            case "mission":
-                context_id = self.mission_id
         return filesystem_pb2.FileFilter(
             **filters.model_dump(exclude={"file_types", "status", "context"}),
             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,
-            context=context_id,
+            context=self._context_enum(filters.context),
         )
 
     def __init__(
@@ -152,7 +169,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper, GrpcErrorHandlerMixi
                     metadata_struct.update(file.metadata)
                 upload_files.append(
                     filesystem_pb2.UploadFileData(
-                        context=self.mission_id,
+                        context=filesystem_pb2.CONTEXT_MISSIONS,
                         name=file.name,
                         file_type=self._file_type_to_enum(file.file_type),
                         content_type=file.content_type or "application/octet-stream",
@@ -188,15 +205,10 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper, GrpcErrorHandlerMixi
         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
         logger.debug("debug:get_file file_id=%s context=%s", file_id, context)
         async with self.handle_grpc_errors("GetFile", FilesystemServiceError):
             request = filesystem_pb2.GetFileRequest(
-                context=context_id,
+                context=self._context_enum(context),
                 file_id=file_id,
                 include_content=include_content,
             )
@@ -244,7 +256,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper, GrpcErrorHandlerMixi
         """
         async with self.handle_grpc_errors("UpdateFile", FilesystemServiceError):
             request = filesystem_pb2.UpdateFileRequest(
-                context=self.mission_id,
+                context=filesystem_pb2.CONTEXT_MISSIONS,
                 file_id=file_id,
                 content=content,
                 file_type=self._file_type_to_enum(file_type) if file_type else None,
@@ -279,7 +291,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper, GrpcErrorHandlerMixi
         logger.debug("debug:delete_files permanent=%s force=%s", permanent, force)
         async with self.handle_grpc_errors("DeleteFiles", FilesystemServiceError):
             request = filesystem_pb2.DeleteFilesRequest(
-                context=self.mission_id,
+                context=filesystem_pb2.CONTEXT_MISSIONS,
                 filters=self._filter_to_proto(filters),
                 permanent=permanent,
                 force=force,
@@ -309,14 +321,9 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper, GrpcErrorHandlerMixi
         Returns:
             tuple[list[FilesystemRecord], int]: List of files and total count
         """
-        match filters.context:
-            case "setup":
-                context_id = self.setup_id
-            case "mission":
-                context_id = self.mission_id
         async with self.handle_grpc_errors("GetFiles", FilesystemServiceError):
             request = filesystem_pb2.GetFilesRequest(
-                context=context_id,
+                context=self._context_enum(filters.context),
                 filters=self._filter_to_proto(filters),
                 include_content=include_content,
                 list_size=list_size,

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/services/registry/default_registry.py

@@ -63,7 +63,7 @@ class DefaultRegistry(RegistryStrategy):
 
         Args:
             name: Case-insensitive free text matched against module name AND documentation.
-            module_type: Filter by type (archetype, tool_module).
+            module_type: Filter by type (archetype, tool_module, service).
             limit: Max results (1-100).
             offset: Pagination offset.
 
@@ -221,7 +221,7 @@ class DefaultRegistry(RegistryStrategy):
             query: Case-insensitive free text matched against setup name AND documentation.
             setup_ids: Restrict to these setup ids.
             module_ids: Restrict to setups backed by these modules.
-            module_types: Filter by backing module type (tool_module, archetype).
+            module_types: Filter by backing module type (tool_module, archetype, service).
             statuses: Filter by setup status. None = no filter.
             visibilities: Filter by visibility.
             limit: Max results (1-100).

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/services/registry/grpc_registry.py

@@ -213,7 +213,7 @@ class GrpcRegistry(RegistryStrategy, GrpcClientWrapper, GrpcErrorHandlerMixin):
 
         Args:
             name: Case-insensitive free text matched against module name AND documentation.
-            module_type: Filter by type (archetype, tool_module).
+            module_type: Filter by type (archetype, tool_module, service).
             limit: Max results (1-100).
             offset: Pagination offset.
 
@@ -495,7 +495,7 @@ class GrpcRegistry(RegistryStrategy, GrpcClientWrapper, GrpcErrorHandlerMixin):
             query: Case-insensitive free text matched against setup name AND documentation.
             setup_ids: Restrict to these setup ids.
             module_ids: Restrict to setups backed by these modules.
-            module_types: Filter by backing module type (tool_module, archetype).
+            module_types: Filter by backing module type (tool_module, archetype, service).
             statuses: Filter by setup status. None = no filter; agent-facing callers
                 should pass READY/CONFIGURATION_SUCCEEDED for invocable setups.
             visibilities: Filter by visibility.

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/services/registry/registry_strategy.py

@@ -51,7 +51,7 @@ class RegistryStrategy(BaseStrategy, ABC):
 
         Args:
             name: Case-insensitive free text matched against module name AND documentation.
-            module_type: Filter by type (archetype, tool_module).
+            module_type: Filter by type (archetype, tool_module, service).
             limit: Max results (1-100).
             offset: Pagination offset.
 
@@ -107,6 +107,29 @@ class RegistryStrategy(BaseStrategy, ABC):
             offset=offset,
         )
 
+    async def search_services(
+        self,
+        name: str | None = None,
+        limit: int = 20,
+        offset: int = 0,
+    ) -> list[ModuleInfo]:
+        """Service registry view: modules of type SERVICE.
+
+        Args:
+            name: Case-insensitive free text matched against module name AND documentation.
+            limit: Max results (1-100).
+            offset: Pagination offset.
+
+        Returns:
+            List of matching service modules.
+        """
+        return await self.search(
+            name=name,
+            module_type=RegistryModuleType.SERVICE.value,
+            limit=limit,
+            offset=offset,
+        )
+
     @abstractmethod
     async def search_setups(
         self,
@@ -125,7 +148,7 @@ class RegistryStrategy(BaseStrategy, ABC):
             query: Case-insensitive free text matched against setup name AND documentation.
             setup_ids: Restrict to these setup ids.
             module_ids: Restrict to setups backed by these modules.
-            module_types: Filter by backing module type (tool_module, archetype).
+            module_types: Filter by backing module type (tool_module, archetype, service).
             statuses: Filter by setup status. None = no filter; agent-facing callers
                 should pass READY/CONFIGURATION_SUCCEEDED for invocable setups.
             visibilities: Filter by visibility.
@@ -190,6 +213,23 @@ class RegistryStrategy(BaseStrategy, ABC):
         """Get setup info."""
         ...
 
+    async def get_service_setup(self, setup_id: str) -> dict[str, Any] | None:
+        """Fetch a service setup's setup_version content JSON.
+
+        The id comes from chat-driven discovery (``search_setups`` + user acceptance),
+        not from configuration. Goes through ``get_setup`` on every call — the registry
+        stays the permission gate; nothing cached. Content always reflects the latest
+        setup version.
+
+        Args:
+            setup_id: The discovered service setup id.
+
+        Returns:
+            The setup_version content, or None when the setup is missing or has no content.
+        """
+        setup = await self.get_setup(setup_id)
+        return setup.config if setup else None
+
     async def wait_for_ready(self, timeout: float = 1.0) -> bool:  # noqa: PLR6301
         """Check if the registry backend is reachable.
 

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin/services/storage/grpc_storage.py

@@ -38,6 +38,25 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
         """
         return isinstance(error.__cause__, CircuitOpenError)
 
+    def _context_enum(self, context: str) -> data_pb2.ContextStorage:
+        """Map a resolved context id-string to the wire's context-kind enum.
+
+        Since dev4 the request carries only the kind; the concrete id is resolved
+        server-side from the x-mission-id / x-setup-id task metadata stamped by
+        ``RequestIdClientInterceptor``.
+
+        Args:
+            context: The resolved context id (``self.mission_id`` or ``self.setup_version_id``).
+
+        Returns:
+            ``CONTEXT_SETUP_VERSIONS`` for the setup-version scope, else ``CONTEXT_MISSIONS``.
+        """
+        # TODO(validate): remove after prod validation
+        # [VALIDATE CTXENUM] server resolves the concrete id (incl. setup->current version) from metadata
+        if context == self.setup_version_id or context.startswith("setup_versions:"):
+            return data_pb2.CONTEXT_SETUP_VERSIONS
+        return data_pb2.CONTEXT_MISSIONS
+
     def _build_record_from_proto(self, proto: data_pb2.StorageRecord) -> StorageRecord:
         """Convert a protobuf StorageRecord message into our Pydantic model.
 
@@ -113,7 +132,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
             data_struct.update(record.data.model_dump())
             req = data_pb2.StoreRecordRequest(
                 data=data_struct,
-                context=record.context,
+                context=self._context_enum(record.context),
                 collection=record.collection,
                 record_id=record.record_id,
                 data_type=record.data_type.name,
@@ -143,7 +162,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
         logger.debug("debug:_read context=%s collection=%s id=%s", context, collection, record_id)
         try:
             req = data_pb2.ReadRecordRequest(
-                context=context,
+                context=self._context_enum(context),
                 collection=collection,
                 record_id=record_id,
             )
@@ -187,7 +206,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
             struct.update(data.model_dump())
             req = data_pb2.UpdateRecordRequest(
                 data=struct,
-                context=context,
+                context=self._context_enum(context),
                 collection=collection,
                 record_id=record_id,
             )
@@ -216,7 +235,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
         logger.debug("debug:_remove context=%s collection=%s id=%s", context, collection, record_id)
         try:
             req = data_pb2.RemoveRecordRequest(
-                context=context,
+                context=self._context_enum(context),
                 collection=collection,
                 record_id=record_id,
             )
@@ -245,7 +264,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
         logger.debug("debug:_list context=%s collection=%s", context, collection)
         try:
             req = data_pb2.ListRecordsRequest(
-                context=context,
+                context=self._context_enum(context),
                 collection=collection,
             )
             resp = await self.exec_grpc_query("ListRecords", req)
@@ -273,7 +292,7 @@ class GrpcStorage(StorageStrategy, GrpcClientWrapper):
         """
         try:
             req = data_pb2.RemoveCollectionRequest(
-                context=context,
+                context=self._context_enum(context),
                 collection=collection,
             )
             await self.exec_grpc_query("RemoveCollection", req)

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 1.0.2.dev1
+Version: 1.0.2.dev2
 Summary: SDK to build kin used in DigitalKin
 Author-email: "DigitalKin.ai" <contact@digitalkin.ai>
 License: Attribution-NonCommercial-ShareAlike 4.0 International
@@ -453,7 +453,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: ag-ui-protocol>=0.1.18
-Requires-Dist: agentic-mesh-protocol==1.0.1.dev3
+Requires-Dist: agentic-mesh-protocol==1.0.1.dev4
 Requires-Dist: anyio>=4.13.0
 Requires-Dist: grpcio-health-checking==1.82.1
 Requires-Dist: grpcio-reflection==1.82.1

{digitalkin-1.0.2.dev1 → digitalkin-1.0.2.dev2}/src/digitalkin.egg-info/requires.txt

@@ -1,5 +1,5 @@
 ag-ui-protocol>=0.1.18
-agentic-mesh-protocol==1.0.1.dev3
+agentic-mesh-protocol==1.0.1.dev4
 anyio>=4.13.0
 grpcio-health-checking==1.82.1
 grpcio-reflection==1.82.1