indexify 0.2.15__tar.gz → 0.2.16__tar.gz

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

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

{indexify-0.2.15 → 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.15 → indexify-0.2.16}/indexify/functions_sdk/indexify_functions.py

@@ -218,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]
 
@@ -234,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
         ]

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

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