indexify 0.2.14__py3-none-any.whl → 0.2.16__py3-none-any.whl

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/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/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.dist-info → indexify-0.2.16.dist-info}/METADATA

@@ -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.dist-info → indexify-0.2.16.dist-info}/RECORD

@@ -17,17 +17,17 @@ indexify/functions_sdk/data_objects.py,sha256=CQZMzYiV7l6dyzFkYquWQHqdte6JnC7XA3
 indexify/functions_sdk/graph.py,sha256=3hUZjdCesiAHJ9y1_heQsEFO-WHROR4kRYxBtVnb7Y4,11937
 indexify/functions_sdk/graph_definition.py,sha256=EJfC0MdKEbFF1CBaU0htrveSlcAQJCH96DLSNfZ02V4,1178
 indexify/functions_sdk/graph_validation.py,sha256=XLHiC9PAtZungJLysU3hIUOPNDkO5TXUDZ_jiZ0H4hg,2508
-indexify/functions_sdk/image.py,sha256=Rsol3LfE15Kca78ruPfDrDPqCkeZPxZx0ULYrFFS5ig,905
-indexify/functions_sdk/indexify_functions.py,sha256=QnumxE0Ln_ryyyjLMopjCGZQUGSX7N25PRpisIdEEXI,9600
+indexify/functions_sdk/image.py,sha256=0uM2VQXqB45ymnFyZVJIiyTGUYHWAHDZ1ORu-hWrofk,1338
+indexify/functions_sdk/indexify_functions.py,sha256=2KZRKdyLSzsrVrWfJGQp_b8grffDKSt_ripgiISTz8Q,9660
 indexify/functions_sdk/local_cache.py,sha256=cNWF67zbhbTJe3g86hyLBy3Rqzs6dNvp2SjLazGZWvw,1348
 indexify/functions_sdk/object_serializer.py,sha256=Zz4GobW3ZamBBtFDF76QxU3TP6oJNdWnhsfKd0OUFoc,1660
 indexify/functions_sdk/pipeline.py,sha256=7hDatRK-SCHYvttf2Vj5YFyiJEVU0OOXEZBOIQenSmk,847
-indexify/http_client.py,sha256=EBzqwT3r90W4w0ADG8oRy8tCh47BHuMPB-vVh-bjUO8,13640
+indexify/http_client.py,sha256=x92EbFD987pIxuBVVQZ-eN9nSL-oI2vpfv6OSKuBGgs,13589
 indexify/remote_graph.py,sha256=ILg6IY6EFgyvnfz1DSzicBZhqvPkg2-UUwgI6lxp6sA,3094
 indexify/remote_pipeline.py,sha256=FW7IAv3r24OOpiqlprw3kuFrpdkqi6Ic4_tT26FThjA,761
 indexify/settings.py,sha256=LSaWZ0ADIVmUv6o6dHWRC3-Ry5uLbCw2sBSg1e_U7UM,99
-indexify-0.2.14.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.2.14.dist-info/METADATA,sha256=zMLk24iy5hXk8BGlvSxkAMHenl2JNxSeXQuENZLNsuY,6199
-indexify-0.2.14.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-indexify-0.2.14.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
-indexify-0.2.14.dist-info/RECORD,,
+indexify-0.2.16.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+indexify-0.2.16.dist-info/METADATA,sha256=kvomzkLgJZoVQYfukNvExhEMmzwJBMwOBTq6oPnEy9A,6199
+indexify-0.2.16.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+indexify-0.2.16.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
+indexify-0.2.16.dist-info/RECORD,,