indexify 0.2.46__tar.gz → 0.2.48__tar.gz

{indexify-0.2.46 → indexify-0.2.48}/PKG-INFO

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

{indexify-0.2.46 → indexify-0.2.48}/indexify/executor/function_executor/single_task_runner.py

@@ -80,8 +80,8 @@ class SingleTaskRunner:
             return function_executor
         except Exception as e:
             self._logger.error(
-                "failed to initialize function executor",
-                exc_info=e,
+                f"failed to initialize function executor: {e.details()}",
+                # exc_info=e.details(),
             )
             await function_executor.destroy()
             raise

{indexify-0.2.46 → indexify-0.2.48}/indexify/executor/task_runner.py

@@ -33,8 +33,8 @@ class TaskRunner:
             return await self._run(task_input, logger)
         except Exception as e:
             logger.error(
-                "failed running the task",
-                exc_info=e,
+                f"failed running the task: {e.details()}",
+                # exc_info=e.debug_error_string(),
             )
             return TaskOutput.internal_error(task_input.task)
 

{indexify-0.2.46 → indexify-0.2.48}/indexify/function_executor/function_executor_service.py

@@ -56,7 +56,13 @@ class FunctionExecutorService(FunctionExecutorServicer):
         # share the model's file descriptor between all tasks or download function configuration
         # only once.
         graph_serializer = get_serializer(request.graph.content_type)
-        graph = graph_serializer.deserialize(request.graph.bytes)
+
+        try:
+            graph = graph_serializer.deserialize(request.graph.bytes)
+        except Exception as e:
+            self._logger.error(f"Caught exception {e}")
+            return InitializeResponse(success=False)
+
         self._function = graph_serializer.deserialize(graph[request.function_name])
 
         self._logger = self._logger.bind(

{indexify-0.2.46 → indexify-0.2.48}/indexify/functions_sdk/image.py

@@ -7,7 +7,7 @@ import pathlib
 import sys
 import tarfile
 from io import BytesIO
-from typing import List, Optional
+from typing import Dict, List, Optional
 
 import docker
 import docker.api.build
@@ -32,12 +32,13 @@ HASH_BUFF_SIZE = 1024**2
 
 class BuildOp(BaseModel):
     op_type: str
+    options: Dict[str, str] = {}
     args: List[str]
 
     def hash(self, hash):
         match self.op_type:
-            case "RUN":
-                hash.update("RUN".encode())
+            case "RUN" | "ADD":
+                hash.update(self.op_type.encode())
                 for a in self.args:
                     hash.update(a.encode())
 
@@ -57,8 +58,10 @@ class BuildOp(BaseModel):
 
     def render(self):
         match self.op_type:
-            case "RUN":
-                return f"RUN {''.join(self.args)}"
+            case "RUN" | "ADD":
+                options = [f"--{k}={v}" for k, v in self.options.items()]
+                return f"{self.op_type} {' '.join(options)} {' '.join(self.args)}"
+
             case "COPY":
                 return f"COPY {self.args[0]} {self.args[1]}"
             case _:
@@ -76,6 +79,7 @@ class Build(BaseModel):
     image_hash: str
     status: str | None
     result: str | None
+    error_message: str | None = None  # Only provided when result is "failed"
 
     created_at: datetime.datetime | None
     started_at: datetime.datetime | None = None
@@ -105,12 +109,20 @@ class Image:
         self._base_image = base_image
         return self
 
-    def run(self, run_str):
-        self._build_ops.append(BuildOp(op_type="RUN", args=[run_str]))
+    def add(self, source: str, dest: str, **kwargs):
+        self._build_ops.append(
+            BuildOp(op_type="ADD", args=[source, dest], options=kwargs)
+        )
         return self
 
-    def copy(self, source: str, dest: str):
-        self._build_ops.append(BuildOp(op_type="COPY", args=[source, dest]))
+    def run(self, run_str, **kwargs):
+        self._build_ops.append(BuildOp(op_type="RUN", args=[run_str], options=kwargs))
+        return self
+
+    def copy(self, source: str, dest: str, **kwargs):
+        self._build_ops.append(
+            BuildOp(op_type="COPY", args=[source, dest], options=kwargs)
+        )
         return self
 
     def to_image_information(self):

{indexify-0.2.46 → indexify-0.2.48}/indexify/functions_sdk/indexify_functions.py

@@ -267,6 +267,8 @@ class IndexifyFunctionWrapper:
             # with json encoding which won't deserialize in tuple.
             if isinstance(input, tuple) or isinstance(input, list):
                 args += input
+            elif isinstance(input, dict):
+                kwargs.update(input)
             else:
                 args.append(input)
             extracted_data = self.indexify_function._call_run(*args, **kwargs)
@@ -292,6 +294,8 @@ class IndexifyFunctionWrapper:
         # with json encoding which won't deserialize in tuple.
         if isinstance(input, tuple) or isinstance(input, list):
             args += input
+        elif isinstance(input, dict):
+            kwargs.update(input)
         else:
             args.append(input)
 

{indexify-0.2.46 → indexify-0.2.48}/pyproject.toml

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