indexify 0.2.14__tar.gz → 0.2.16__tar.gz

{indexify-0.2.14 → indexify-0.2.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: indexify
-Version: 0.2.14
+Version: 0.2.16
 Summary: Python Client for Indexify
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0

{indexify-0.2.14 → indexify-0.2.16}/indexify/functions_sdk/image.py

@@ -1,11 +1,20 @@
+def python_version_to_image(python_version):
+    if python_version.startswith("3.9"):
+        return "python:3.9.20-bookworm"
+    elif python_version.startswith("3.10"):
+        return "python:3.10.15-bookworm"
+    elif python_version.startswith("3.11"):
+        return "python:3.11.10-bookworm"
+    else:
+        raise ValueError(f"unsupported Python version: {python_version}")
+
+
 class Image:
-    def __init__(self):
+    def __init__(self, python="3.10"):
         self._image_name = None
-
         self._tag = "latest"
-
-        self._base_image = "python:3.11.10-slim-bookworm"
-
+        self._base_image = python_version_to_image(python)
+        self._python_version = python
         self._run_strs = []
 
     def name(self, image_name):

{indexify-0.2.14 → indexify-0.2.16}/indexify/functions_sdk/indexify_functions.py

@@ -79,6 +79,10 @@ class IndexifyFunction(ABC):
 
         return partial(self.run, **kwargs)
 
+    @classmethod
+    def deserialize_output(cls, output: IndexifyData) -> Any:
+        serializer = get_serializer(cls.payload_encoder)
+        return serializer.deserialize(output.payload)
 
 class IndexifyRouter(ABC):
     name: str = ""
@@ -214,6 +218,8 @@ class IndexifyFunctionWrapper:
             args.append(input)
 
         extracted_data = self.indexify_function.run(*args, **kwargs)
+        if extracted_data is None:
+            return []
 
         return extracted_data if isinstance(extracted_data, list) else [extracted_data]
 
@@ -230,7 +236,7 @@ class IndexifyFunctionWrapper:
             acc = self.indexify_function.accumulate.model_validate(
                 self.indexify_function.accumulate()
             )
-        outputs: List[Any] = self.run_fn(input, acc=acc)
+        outputs: Optional[List[Any]] = self.run_fn(input, acc=acc)
         return [
             IndexifyData(payload=serializer.serialize(output)) for output in outputs
         ]
@@ -265,7 +271,3 @@ class IndexifyFunctionWrapper:
                 payload = list(payload.values())[0]
             return arg_type.model_validate(payload)
         return payload
-
-    def deserialize_fn_output(self, output: IndexifyData) -> Any:
-        serializer = get_serializer(self.indexify_function.payload_encoder)
-        return serializer.deserialize(output.payload)

{indexify-0.2.14 → indexify-0.2.16}/indexify/http_client.py

@@ -13,7 +13,7 @@ from rich import print
 from indexify.error import ApiException
 from indexify.functions_sdk.data_objects import IndexifyData
 from indexify.functions_sdk.graph import ComputeGraphMetadata, Graph
-from indexify.functions_sdk.indexify_functions import IndexifyFunctionWrapper
+from indexify.functions_sdk.indexify_functions import IndexifyFunction
 from indexify.settings import DEFAULT_SERVICE_URL, DEFAULT_SERVICE_URL_HTTPS
 
 
@@ -71,7 +71,7 @@ class IndexifyClient:
         self._service_url = service_url
         self._timeout = kwargs.get("timeout")
         self._graphs: Dict[str, Graph] = {}
-        self._fns = {}
+        self._fns: Dict[str, IndexifyFunction] = {}
 
     def _request(self, method: str, **kwargs) -> httpx.Response:
         try:
@@ -181,14 +181,12 @@ class IndexifyClient:
         response = self._get(f"namespaces/{self.namespace}/compute_graphs/{name}")
         return ComputeGraphMetadata(**response.json())
 
-    def load_fn_wrapper(self, name: str, fn_name: str) -> IndexifyFunctionWrapper:
+    def load_fn(self, name: str, fn_name: str) -> IndexifyFunction:
         response = self._get(
             f"internal/namespaces/{self.namespace}/compute_graphs/{name}/code"
         )
         pickled_functions_by_name = cloudpickle.loads(response.content)
-        return IndexifyFunctionWrapper(
-            cloudpickle.loads(pickled_functions_by_name[fn_name])
-        )
+        return cloudpickle.loads(pickled_functions_by_name[fn_name])
 
     def namespaces(self) -> List[str]:
         response = self._get(f"namespaces")
@@ -323,7 +321,7 @@ class IndexifyClient:
         """
         fn_key = f"{graph}/{fn_name}"
         if fn_key not in self._fns:
-            self._fns[fn_key] = self.load_fn_wrapper(graph, fn_name)
+            self._fns[fn_key] = self.load_fn(graph, fn_name)
         response = self._get(
             f"namespaces/{self.namespace}/compute_graphs/{graph}/invocations/{invocation_id}/outputs",
         )
@@ -335,7 +333,7 @@ class IndexifyClient:
                 indexify_data = self._download_output(
                     self.namespace, graph, invocation_id, fn_name, output.id
                 )
-                output = self._fns[fn_key].deserialize_fn_output(indexify_data)
+                output = self._fns[fn_key].deserialize_output(indexify_data)
                 outputs.append(output)
         return outputs
 

{indexify-0.2.14 → indexify-0.2.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "indexify"
-version = "0.2.14"
+version = "0.2.16"
 description = "Python Client for Indexify"
 authors = ["Tensorlake Inc. <support@tensorlake.ai>"]
 license = "Apache 2.0"