indexify 0.2.13__py3-none-any.whl → 0.2.15__py3-none-any.whl

indexify/functions_sdk/graph.py

@@ -153,7 +153,7 @@ class Graph:
         start_node = self.nodes[self._start_node]
         start_node = FunctionMetadata(
             name=start_node.name,
-            fn_name=start_node.fn_name,
+            fn_name=start_node.name,
             description=start_node.description,
             reducer=start_node.accumulate is not None,
             image_name=start_node.image._image_name,
@@ -209,13 +209,15 @@ class Graph:
                 k: IndexifyData(payload=serializer.serialize(v))
             }
         self._results[input.id] = outputs
-        self._run(input, outputs)
+        enable_cache = kwargs.get('enable_cache', True)
+        self._run(input, outputs,enable_cache)
         return input.id
 
     def _run(
         self,
         initial_input: IndexifyData,
         outputs: Dict[str, List[bytes]],
+        enable_cache: bool
     ):
         accumulator_values = self._accumulator_values[initial_input.id]
         queue = deque([(self._start_node, initial_input)])
@@ -227,7 +229,7 @@ class Graph:
             cached_output_bytes: Optional[bytes] = self._cache.get(
                 self.name, node_name, input_bytes
             )
-            if cached_output_bytes is not None:
+            if cached_output_bytes is not None and enable_cache:
                 print(
                     f"ran {node_name}: num outputs: {len(cached_output_bytes)} (cache hit)"
                 )

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 = ""
@@ -265,7 +269,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:
@@ -79,7 +79,7 @@ class IndexifyClient:
             status_code = str(response.status_code)
             if status_code.startswith("4"):
                 raise ApiException(
-                    "status code: " + status_code + " request args: " + str(kwargs)
+                    "status code: " + status_code + " message: " + response.text
                 )
             if status_code.startswith("5"):
                 raise ApiException(response.text)
@@ -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.13.dist-info → indexify-0.2.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: indexify
-Version: 0.2.13
+Version: 0.2.15
 Summary: Python Client for Indexify
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0
@@ -13,7 +13,8 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: cloudpickle (>=3,<4)
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: cloudpickle (>=3.1.0,<4.0.0)
 Requires-Dist: docker (>=7.1.0,<8.0.0)
 Requires-Dist: httpx-sse (>=0.4.0,<0.5.0)
 Requires-Dist: httpx[http2] (>=0,<1)
@@ -21,7 +22,7 @@ Requires-Dist: msgpack (>=1.1.0,<2.0.0)
 Requires-Dist: nanoid (>=2.0.0,<3.0.0)
 Requires-Dist: pydantic (>=2.9.2,<3.0.0)
 Requires-Dist: pyyaml (>=6,<7)
-Requires-Dist: rich (>=13,<14)
+Requires-Dist: rich (>=13.9.2,<14.0.0)
 Requires-Dist: typer (>=0.12.5,<0.13.0)
 Project-URL: Repository, https://github.com/tensorlakeai/indexify
 Description-Content-Type: text/markdown

{indexify-0.2.13.dist-info → indexify-0.2.15.dist-info}/RECORD

@@ -14,20 +14,20 @@ indexify/executor/runtime_probes.py,sha256=L-nB4D2zBZ287xxPwErv5R_gxFOmTEvg61FMe
 indexify/executor/task_reporter.py,sha256=gnnse0v6rjjni8lNzeb-ZYq6iF2DgafKoT7dcGUZhQ4,3716
 indexify/executor/task_store.py,sha256=q8s2gImsFffWeXQR0mk1Xlo1Aj_2GfclNPjQ2EA_YBo,3984
 indexify/functions_sdk/data_objects.py,sha256=CQZMzYiV7l6dyzFkYquWQHqdte6JnC7XA3i2ZyvPvgQ,844
-indexify/functions_sdk/graph.py,sha256=t35z26jRCBgoz7fncdlAL7MVTOYjeJuoPpz-xI7WjBw,11827
+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/indexify_functions.py,sha256=KTEbTs-waqXTKC5e2f1MWxShr87nwJd3PgNEL3QMvIU,9593
 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=x9ABHWc_oQH8_fa5qwFi53D10m9ibhrjvfw_OkTJfhc,13643
+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.13.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.2.13.dist-info/METADATA,sha256=urBxY_dtcmMIfTd_UQvHq0efxq4-LHJahRGEOM0WYhE,6132
-indexify-0.2.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-indexify-0.2.13.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
-indexify-0.2.13.dist-info/RECORD,,
+indexify-0.2.15.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+indexify-0.2.15.dist-info/METADATA,sha256=6iDb6m_2f2tjoA9y4630Z6j23ymDLGmoNjkIIyV1kxA,6199
+indexify-0.2.15.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+indexify-0.2.15.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
+indexify-0.2.15.dist-info/RECORD,,

{indexify-0.2.13.dist-info → indexify-0.2.15.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 1.9.1
 Root-Is-Purelib: true
 Tag: py3-none-any