Ran linting

ahnlich/ai/Cargo.toml

@@ -38,7 +38,6 @@ strum = { version = "0.26", features = ["derive"] }
 log.workspace = true
 fallible_collections.workspace = true
 rayon.workspace = true
-fastembed = { version = "4.0.0", features = ["default"] }
 hf-hub = { version = "0.3", default-features = false }
 dirs = "5.0.1"
 ort = { version = "=2.0.0-rc.5", default-features = false }

sdk/ahnlich-client-py/ahnlich_client_py/internals/ai_query.py

@@ -1,9 +1,8 @@
 # pyre-strict
-import typing
 from dataclasses import dataclass
-
-from ahnlich_client_py.internals import bincode
+import typing
 from ahnlich_client_py.internals import serde_types as st
+from ahnlich_client_py.internals import bincode
 
 
 class AIModel:
@@ -55,6 +54,7 @@ class AIModel__ClipVitB32Image(AIModel):
     INDEX = 5  # type: int
     pass
 
+
 @dataclass(frozen=True)
 class AIModel__ClipVitB32Text(AIModel):
     INDEX = 6  # type: int
@@ -305,38 +305,6 @@ class Algorithm__KDTree(Algorithm):
 ]
 
 
-class ImageAction:
-    VARIANTS = []  # type: typing.Sequence[typing.Type[ImageAction]]
-
-    def bincode_serialize(self) -> bytes:
-        return bincode.serialize(self, ImageAction)
-
-    @staticmethod
-    def bincode_deserialize(input: bytes) -> "ImageAction":
-        v, buffer = bincode.deserialize(input, ImageAction)
-        if buffer:
-            raise st.DeserializationError("Some input bytes were not read")
-        return v
-
-
-@dataclass(frozen=True)
-class ImageAction__ResizeImage(ImageAction):
-    INDEX = 0  # type: int
-    pass
-
-
-@dataclass(frozen=True)
-class ImageAction__ErrorIfDimensionsMismatch(ImageAction):
-    INDEX = 1  # type: int
-    pass
-
-
-ImageAction.VARIANTS = [
-    ImageAction__ResizeImage,
-    ImageAction__ErrorIfDimensionsMismatch,
-]
-
-
 class MetadataValue:
     VARIANTS = []  # type: typing.Sequence[typing.Type[MetadataValue]]
 
@@ -498,20 +466,20 @@ def bincode_deserialize(input: bytes) -> "PreprocessAction":
 
 
 @dataclass(frozen=True)
-class PreprocessAction__RawString(PreprocessAction):
+class PreprocessAction__NoPreprocessing(PreprocessAction):
     INDEX = 0  # type: int
-    value: "StringAction"
+    pass
 
 
 @dataclass(frozen=True)
-class PreprocessAction__Image(PreprocessAction):
+class PreprocessAction__ModelPreprocessing(PreprocessAction):
     INDEX = 1  # type: int
-    value: "ImageAction"
+    pass
 
 
 PreprocessAction.VARIANTS = [
-    PreprocessAction__RawString,
-    PreprocessAction__Image,
+    PreprocessAction__NoPreprocessing,
+    PreprocessAction__ModelPreprocessing,
 ]
 
 
@@ -545,35 +513,3 @@ class StoreInput__Image(StoreInput):
     StoreInput__RawString,
     StoreInput__Image,
 ]
-
-
-class StringAction:
-    VARIANTS = []  # type: typing.Sequence[typing.Type[StringAction]]
-
-    def bincode_serialize(self) -> bytes:
-        return bincode.serialize(self, StringAction)
-
-    @staticmethod
-    def bincode_deserialize(input: bytes) -> "StringAction":
-        v, buffer = bincode.deserialize(input, StringAction)
-        if buffer:
-            raise st.DeserializationError("Some input bytes were not read")
-        return v
-
-
-@dataclass(frozen=True)
-class StringAction__TruncateIfTokensExceed(StringAction):
-    INDEX = 0  # type: int
-    pass
-
-
-@dataclass(frozen=True)
-class StringAction__ErrorIfTokensExceed(StringAction):
-    INDEX = 1  # type: int
-    pass
-
-
-StringAction.VARIANTS = [
-    StringAction__TruncateIfTokensExceed,
-    StringAction__ErrorIfTokensExceed,
-]

sdk/ahnlich-client-py/ahnlich_client_py/internals/ai_response.py

@@ -1,9 +1,8 @@
 # pyre-strict
-import typing
 from dataclasses import dataclass
-
-from ahnlich_client_py.internals import bincode
+import typing
 from ahnlich_client_py.internals import serde_types as st
+from ahnlich_client_py.internals import bincode
 
 
 class AIModel:
@@ -56,13 +55,20 @@ class AIModel__ClipVitB32Image(AIModel):
     pass
 
 
+@dataclass(frozen=True)
+class AIModel__ClipVitB32Text(AIModel):
+    INDEX = 6  # type: int
+    pass
+
+
 AIModel.VARIANTS = [
     AIModel__AllMiniLML6V2,
     AIModel__AllMiniLML12V2,
     AIModel__BGEBaseEnV15,
     AIModel__BGELargeEnV15,
     AIModel__Resnet50,
     AIModel__ClipVitB32Image,
+    AIModel__ClipVitB32Text,
 ]
 
 

sdk/ahnlich-client-py/ahnlich_client_py/internals/bincode/__init__.py

@@ -1,16 +1,16 @@
 # Copyright (c) Facebook, Inc. and its affiliates
 # SPDX-License-Identifier: MIT OR Apache-2.0
 
-import collections
 import dataclasses
+import collections
 import io
 import struct
 import typing
 from copy import copy
 from typing import get_type_hints
 
-from ahnlich_client_py.internals import serde_binary as sb
 from ahnlich_client_py.internals import serde_types as st
+from ahnlich_client_py.internals import serde_binary as sb
 
 # Maximum length in practice for sequences (e.g. in Java).
 MAX_LENGTH = (1 << 31) - 1

sdk/ahnlich-client-py/ahnlich_client_py/internals/db_query.py

@@ -1,9 +1,8 @@
 # pyre-strict
-import typing
 from dataclasses import dataclass
-
-from ahnlich_client_py.internals import bincode
+import typing
 from ahnlich_client_py.internals import serde_types as st
+from ahnlich_client_py.internals import bincode
 
 
 class Algorithm:

sdk/ahnlich-client-py/ahnlich_client_py/internals/db_response.py

@@ -1,9 +1,8 @@
 # pyre-strict
-import typing
 from dataclasses import dataclass
-
-from ahnlich_client_py.internals import bincode
+import typing
 from ahnlich_client_py.internals import serde_types as st
+from ahnlich_client_py.internals import bincode
 
 
 @dataclass(frozen=True)