indexify 0.2.18__tar.gz → 0.2.19__tar.gz

{indexify-0.2.18 → indexify-0.2.19}/PKG-INFO

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

{indexify-0.2.18 → indexify-0.2.19}/indexify/executor/agent.py

@@ -2,6 +2,7 @@ import asyncio
 import json
 import ssl
 from concurrent.futures.process import BrokenProcessPool
+from importlib.metadata import version
 from typing import Dict, List, Optional
 
 import httpx
@@ -115,9 +116,9 @@ class ExtractorAgent:
                 )
                 console.print(
                     Panel(
-                        f"Reporting outcome of task {task_outcome.task.id}\n"
+                        f"Reporting outcome of task: {task_outcome.task.id}, function: {task_outcome.task.compute_fn}\n"
                         f"Outcome: {style_outcome}\n"
-                        f"Outputs: {len(task_outcome.outputs or [])}  Router Output: {task_outcome.router_output}",
+                        f"Num Fn Outputs: {len(task_outcome.outputs or [])}  Router Output: {task_outcome.router_output}",
                         title="Task Completion",
                         border_style="info",
                     )
@@ -314,8 +315,10 @@ class ExtractorAgent:
                 # rewrite the image name
                 pass
 
+            executor_version = version("indexify")
             data = ExecutorMetadata(
                 id=self._executor_id,
+                executor_version=executor_version,
                 addr="",
                 image_name=runtime_probe.image_name,
                 labels=runtime_probe.labels,

{indexify-0.2.18 → indexify-0.2.19}/indexify/executor/api_objects.py

@@ -18,6 +18,7 @@ class Task(BaseModel):
 
 class ExecutorMetadata(BaseModel):
     id: str
+    executor_version: str
     addr: str
     image_name: str
     labels: Dict[str, Any]

{indexify-0.2.18 → indexify-0.2.19}/indexify/executor/task_reporter.py

@@ -28,11 +28,11 @@ class TaskReporter:
 
     def report_task_outcome(self, completed_task: CompletedTask):
         fn_outputs = []
-        print(
-            f"[bold]task-reporter[/bold] uploading output of size: {len(completed_task.outputs or [])}"
-        )
         for output in completed_task.outputs or []:
             output_bytes = MsgPackSerializer.serialize(output)
+            print(
+                f"[bold]task-reporter[/bold] uploading output of size: {len(output_bytes)} bytes"
+            )
             fn_outputs.append(
                 ("node_outputs", (nanoid.generate(), io.BytesIO(output_bytes)))
             )

{indexify-0.2.18 → indexify-0.2.19}/indexify/functions_sdk/graph.py

@@ -96,9 +96,9 @@ class Graph:
             return self
 
         if issubclass(indexify_fn, IndexifyFunction) and indexify_fn.accumulate:
-            self.accumulator_zero_values[
-                indexify_fn.name
-            ] = indexify_fn.accumulate().model_dump()
+            self.accumulator_zero_values[indexify_fn.name] = (
+                indexify_fn.accumulate().model_dump()
+            )
 
         self.nodes[indexify_fn.name] = indexify_fn
         return self

{indexify-0.2.18 → indexify-0.2.19}/indexify/functions_sdk/indexify_functions.py

@@ -206,9 +206,9 @@ class RouterCallResult(BaseModel):
 
 class IndexifyFunctionWrapper:
     def __init__(self, indexify_function: Union[IndexifyFunction, IndexifyRouter]):
-        self.indexify_function: Union[
-            IndexifyFunction, IndexifyRouter
-        ] = indexify_function()
+        self.indexify_function: Union[IndexifyFunction, IndexifyRouter] = (
+            indexify_function()
+        )
 
     def get_output_model(self) -> Any:
         if not isinstance(self.indexify_function, IndexifyFunction):

{indexify-0.2.18 → indexify-0.2.19}/pyproject.toml

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