digitalkin 0.2.15__tar.gz → 0.2.16__tar.gz

{digitalkin-0.2.15 → digitalkin-0.2.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: digitalkin
-Version: 0.2.15
+Version: 0.2.16
 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.15 → digitalkin-0.2.16}/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.15"
+    version = "0.2.16"
 
     classifiers = [
         "Development Status :: 3 - Alpha",

{digitalkin-0.2.15 → digitalkin-0.2.16}/src/digitalkin/__version__.py

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

{digitalkin-0.2.15 → digitalkin-0.2.16}/src/digitalkin/services/filesystem/default_filesystem.py

@@ -5,7 +5,7 @@ import os
 import tempfile
 import uuid
 from pathlib import Path
-from typing import Any
+from typing import Any, Literal
 
 from digitalkin.logger import logger
 from digitalkin.services.filesystem.filesystem_strategy import (
@@ -259,7 +259,17 @@ class DefaultFilesystem(FilesystemStrategy):
         self,
         file_id: str,
         content: bytes | None = None,
-        file_type: str | None = None,
+        file_type: Literal[
+            "UNSPECIFIED",
+            "DOCUMENT",
+            "IMAGE",
+            "VIDEO",
+            "AUDIO",
+            "ARCHIVE",
+            "CODE",
+            "OTHER",
+        ]
+        | None = None,
         content_type: str | None = None,
         metadata: dict[str, Any] | None = None,
         new_name: str | None = None,

{digitalkin-0.2.15 → digitalkin-0.2.16}/src/digitalkin/services/filesystem/filesystem_strategy.py

@@ -2,7 +2,7 @@
 
 from abc import ABC, abstractmethod
 from datetime import datetime
-from typing import Any
+from typing import Any, Literal
 
 from pydantic import BaseModel, Field
 
@@ -34,7 +34,21 @@ class FileFilter(BaseModel):
 
     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: list[str] | None = Field(default=None, description="Filter by file types")
+    file_types: (
+        list[
+            Literal[
+                "UNSPECIFIED",
+                "DOCUMENT",
+                "IMAGE",
+                "AUDIO",
+                "VIDEO",
+                "ARCHIVE",
+                "CODE",
+                "OTHER",
+            ]
+        ]
+        | None
+    ) = Field(default=None, description="Filter by file types")
     created_after: datetime | None = Field(default=None, description="Filter files created after this timestamp")
     created_before: datetime | None = Field(default=None, description="Filter files created before this timestamp")
     updated_after: datetime | None = Field(default=None, description="Filter files updated after this timestamp")
@@ -52,7 +66,16 @@ class UploadFileData(BaseModel):
 
     content: bytes = Field(description="The content of the file")
     name: str = Field(description="The name of the file")
-    file_type: str = Field(description="The type of the file")
+    file_type: Literal[
+        "UNSPECIFIED",
+        "DOCUMENT",
+        "IMAGE",
+        "AUDIO",
+        "VIDEO",
+        "ARCHIVE",
+        "CODE",
+        "OTHER",
+    ] = Field(description="The type of the file")
     content_type: str | None = Field(default=None, description="The content type of the file")
     metadata: dict[str, Any] | None = Field(default=None, description="The metadata of the file")
     replace_if_exists: bool = Field(default=False, description="Whether to replace the file if it already exists")
@@ -153,7 +176,17 @@ class FilesystemStrategy(BaseStrategy, ABC):
         self,
         file_id: str,
         content: bytes | None = None,
-        file_type: str | None = None,
+        file_type: Literal[
+            "UNSPECIFIED",
+            "DOCUMENT",
+            "IMAGE",
+            "VIDEO",
+            "AUDIO",
+            "ARCHIVE",
+            "CODE",
+            "OTHER",
+        ]
+        | None = None,
         content_type: str | None = None,
         metadata: dict[str, Any] | None = None,
         new_name: str | None = None,

{digitalkin-0.2.15 → digitalkin-0.2.16}/src/digitalkin/services/filesystem/grpc_filesystem.py

@@ -2,7 +2,7 @@
 
 from collections.abc import Generator
 from contextlib import contextmanager
-from typing import Any
+from typing import Any, Literal
 
 from digitalkin_proto.digitalkin.filesystem.v1 import filesystem_pb2, filesystem_service_pb2_grpc
 from google.protobuf import struct_pb2
@@ -103,12 +103,11 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
             status=self._file_status_to_enum(filters.status) if filters.status else None,
         )
 
-    def _file_proto_to_data(self, file: filesystem_pb2.File, content: bytes | None = None) -> FilesystemRecord:
+    def _file_proto_to_data(self, file: filesystem_pb2.File) -> FilesystemRecord:
         """Convert a File proto message to FilesystemRecord.
 
         Args:
             file: The File proto message to convert
-            content: The content of the file
 
         Returns:
             FilesystemRecord: The converted data
@@ -124,7 +123,7 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
             metadata=MessageToDict(file.metadata),
             storage_url=file.storage_url,
             status=filesystem_pb2.FileStatus.Name(file.status),
-            content=content,
+            content=file.content,
         )
 
     def __init__(
@@ -213,13 +212,23 @@ class GrpcFilesystem(FilesystemStrategy, GrpcClientWrapper):
 
             response: filesystem_pb2.GetFileResponse = self.exec_grpc_query("GetFile", request)
 
-            return self._file_proto_to_data(response.file, response.content)
+            return self._file_proto_to_data(response.file)
 
     def update_file(
         self,
         file_id: str,
         content: bytes | None = None,
-        file_type: str | None = None,
+        file_type: Literal[
+            "UNSPECIFIED",
+            "DOCUMENT",
+            "IMAGE",
+            "VIDEO",
+            "AUDIO",
+            "ARCHIVE",
+            "CODE",
+            "OTHER",
+        ]
+        | None = None,
         content_type: str | None = None,
         metadata: dict[str, Any] | None = None,
         new_name: str | None = None,

{digitalkin-0.2.15 → digitalkin-0.2.16}/src/digitalkin.egg-info/PKG-INFO

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