indexify 0.2.30__py3-none-any.whl → 0.2.32__py3-none-any.whl

indexify/executor/downloader.py

@@ -120,7 +120,7 @@ class Downloader:
         deserialized_content = serializer.deserialize(response.content)
 
         if reducer_url:
-            init_value = httpx.get(reducer_url)
+            init_value = self._client.get(reducer_url)
             try:
                 init_value.raise_for_status()
             except httpx.HTTPStatusError as e:

indexify/functions_sdk/graph.py

@@ -101,9 +101,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
@@ -210,6 +210,7 @@ class Graph:
         )
 
     def run(self, block_until_done: bool = False, **kwargs) -> str:
+        self.validate_graph()
         start_node = self.nodes[self._start_node]
         serializer = get_serializer(start_node.encoder)
         input = IndexifyData(
@@ -236,6 +237,35 @@ class Graph:
         self._run(input, outputs)
         return input.id
 
+    def validate_graph(self) -> None:
+        """
+        A method to validate that each node in the graph is
+        reachable from start node using BFS.
+        """
+        total_number_of_nodes = len(self.nodes)
+        queue = deque([self._start_node])
+        visited = {self._start_node}
+
+        while queue:
+            current_node_name = queue.popleft()
+            neighbours = (
+                self.edges[current_node_name]
+                if self.edges[current_node_name]
+                else self.routers[current_node_name]
+            )
+            for neighbour in neighbours:
+                if neighbour in visited:
+                    continue
+                else:
+                    visited.add(neighbour)
+                    queue.append(neighbour)
+
+        if total_number_of_nodes != len(visited):
+            # all the nodes are not reachable from the start_node.
+            raise Exception(
+                "Some nodes in the graph are not reachable from start node."
+            )
+
     def _run(
         self,
         initial_input: IndexifyData,
@@ -244,9 +274,9 @@ class Graph:
         queue = deque([(self._start_node, initial_input)])
         while queue:
             node_name, input = queue.popleft()
-            function_outputs: Union[
-                FunctionCallResult, RouterCallResult
-            ] = self._invoke_fn(node_name, input)
+            function_outputs: Union[FunctionCallResult, RouterCallResult] = (
+                self._invoke_fn(node_name, input)
+            )
             self._log_local_exec_tracebacks(function_outputs)
             if isinstance(function_outputs, RouterCallResult):
                 for edge in function_outputs.edges:
@@ -275,7 +305,7 @@ class Graph:
         self, node_name: str, input: IndexifyData
     ) -> Optional[Union[RouterCallResult, FunctionCallResult]]:
         node = self.nodes[node_name]
-        if node_name in self.routers:
+        if node_name in self.routers and len(self.routers[node_name]) > 0:
             result = IndexifyFunctionWrapper(node, self._local_graph_ctx).invoke_router(
                 node_name, input
             )

indexify/functions_sdk/indexify_functions.py

@@ -137,9 +137,11 @@ def indexify_router(
 
         attrs = {
             "name": name if name else fn.__name__,
-            "description": description
-            if description
-            else (fn.__doc__ or "").strip().replace("\n", ""),
+            "description": (
+                description
+                if description
+                else (fn.__doc__ or "").strip().replace("\n", "")
+            ),
             "image": image,
             "placement_constraints": placement_constraints,
             "encoder": encoder,
@@ -174,9 +176,11 @@ def indexify_function(
 
         attrs = {
             "name": name if name else fn.__name__,
-            "description": description
-            if description
-            else (fn.__doc__ or "").strip().replace("\n", ""),
+            "description": (
+                description
+                if description
+                else (fn.__doc__ or "").strip().replace("\n", "")
+            ),
             "image": image,
             "placement_constraints": placement_constraints,
             "accumulate": accumulate,
@@ -205,9 +209,9 @@ class IndexifyFunctionWrapper:
         indexify_function: Union[IndexifyFunction, IndexifyRouter],
         context: GraphInvocationContext,
     ):
-        self.indexify_function: Union[
-            IndexifyFunction, IndexifyRouter
-        ] = indexify_function()
+        self.indexify_function: Union[IndexifyFunction, IndexifyRouter] = (
+            indexify_function()
+        )
         self.indexify_function._ctx = context
 
     def get_output_model(self) -> Any:

indexify/remote_graph.py

@@ -90,6 +90,7 @@ class RemoteGraph:
         :param client: The IndexifyClient used to communicate with the server.
             Prefered over server_url.
         """
+        g.validate_graph()
         if not client:
             client = IndexifyClient(service_url=server_url)
         client.register_compute_graph(g, additional_modules)

{indexify-0.2.30.dist-info → indexify-0.2.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: indexify
-Version: 0.2.30
+Version: 0.2.32
 Summary: Python Client for Indexify
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0
@@ -20,7 +20,7 @@ Requires-Dist: httpx-sse (>=0.4.0,<0.5.0)
 Requires-Dist: httpx[http2] (>=0,<1)
 Requires-Dist: jsonpickle (>=4.0.0,<5.0.0)
 Requires-Dist: nanoid (>=2.0.0,<3.0.0)
-Requires-Dist: pydantic (>=2.9.2,<3.0.0)
+Requires-Dist: pydantic (==2.10.2)
 Requires-Dist: pyyaml (>=6,<7)
 Requires-Dist: rich (>=13.9.2,<14.0.0)
 Requires-Dist: typer (>=0.13.0,<0.14.0)

{indexify-0.2.30.dist-info → indexify-0.2.32.dist-info}/RECORD

@@ -7,7 +7,7 @@ indexify/data_loaders/url_loader.py,sha256=32SERljcq1Xsi4RdLz2dgyk2TER5pQPTtXl3g
 indexify/error.py,sha256=qAWr8R6AxPkjsxHSzXTc8zqYnNO_AjOqqYEPsQvF1Zs,238
 indexify/executor/agent.py,sha256=Zt-KU2yrOuFuHu8CiOg-QmAzkNU1idcv5UZ3YN2YSDU,18649
 indexify/executor/api_objects.py,sha256=mvmwGbK4paJNQGFvbtNHMPpiH_LpVhrlRnCcrqS6HOQ,859
-indexify/executor/downloader.py,sha256=_SQ-6-0Kj3Zg0Dqp_pMwTYxvBP3xvNUOREyDucCJp8M,4944
+indexify/executor/downloader.py,sha256=wwkTdKRrDBHwKQOb_3uUb8pVl1tvg4QzOcYFnGQwNtA,4951
 indexify/executor/executor_tasks.py,sha256=A0UIEZ5VaB6zSkFQG81UmTW0E57MTYhGlaXuAbRV8lQ,1884
 indexify/executor/function_worker.py,sha256=wRW2-X9dNI80KhwTD1vD-pcyetsVKVs6vVdg7L7JjcQ,6462
 indexify/executor/image_dependency_installer.py,sha256=ct8GmzgkaPi6NAblk68IJJWo5MecIUubELotmSrgoRQ,1759
@@ -16,20 +16,20 @@ indexify/executor/runtime_probes.py,sha256=mjw2_mGQ622wRT_39WPGGgPEZQTgtrf3-ICcU
 indexify/executor/task_reporter.py,sha256=4unHxLUHedKwIoqO3e5YdDJpUe_pJng-vHReoqcPNNU,5141
 indexify/executor/task_store.py,sha256=JlRlWwAm4YjFRkTNRx-6GsUcmOzcyvzb5Csa5XDpRTI,3982
 indexify/functions_sdk/data_objects.py,sha256=wXbUa9hjU6rsXmmk19vQ5Kixf3FsI59VBWPNmHasAX0,854
-indexify/functions_sdk/graph.py,sha256=TbHtIcAzRcEn3BWewVhNsUGMNfRduI1aSAvK6Vyx-fk,11801
+indexify/functions_sdk/graph.py,sha256=ewm-XUlqIXA0RSl1JxU91yT2XXK3WRqrD9_YSvlK_44,12884
 indexify/functions_sdk/graph_definition.py,sha256=UeC0PMchgZgo2T0Goog0SW1lIRdFf6ug4iHOS9ItRvI,1622
 indexify/functions_sdk/graph_validation.py,sha256=mN2Fcp91GIwFZEQP6z_qGqt4LkLM70SnI7AWBi4CmKQ,2509
 indexify/functions_sdk/image.py,sha256=QK0H6KxLWriB_z4M0kunKzzHdHxYLWL670RPYgYuf_8,1762
-indexify/functions_sdk/indexify_functions.py,sha256=EbNC5gZktRc-G9ribQrpMG2X_G-8cujsxTgot734iGQ,10340
+indexify/functions_sdk/indexify_functions.py,sha256=cPUN7Zlgkwl6YUOuhgKogPdoBakQuf-VspQvU3IDmfU,10422
 indexify/functions_sdk/local_cache.py,sha256=cNWF67zbhbTJe3g86hyLBy3Rqzs6dNvp2SjLazGZWvw,1348
 indexify/functions_sdk/object_serializer.py,sha256=pOgUOWbRNRix9uZT0aQn0LTCnJCeMNGO1nAE0jAybmg,1546
 indexify/functions_sdk/pipeline.py,sha256=KmxZE8eBFAQ4bbEcYURXXR26HSyoAT3O6iu9H38-OXE,974
 indexify/http_client.py,sha256=5d70A6j2vJBn3OfUVjrQvnPml07p7fxOXWBUAUfAkiU,15915
-indexify/remote_graph.py,sha256=CWfFxB0jlx8Zgp_xUjvjgrIwv-yNAfEbQwlXt3jfRNM,4976
+indexify/remote_graph.py,sha256=aox9NibZIU8YDiP92syerFSjiZc2jVPkF6g2kfqOvCA,5003
 indexify/remote_pipeline.py,sha256=oqx57rSPszNS3DToXO_nf-CKqkCZWptm1u_p3orV_gQ,790
 indexify/settings.py,sha256=Ny59mzYI4gbXoK8hjx66a_men6ndbd1J1zCTcKOoyzg,50
-indexify-0.2.30.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.2.30.dist-info/METADATA,sha256=iDiwkZ0gmVSy9jio0m6MdcuGCz9x8v92HdRpMDaXBU8,6202
-indexify-0.2.30.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-indexify-0.2.30.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
-indexify-0.2.30.dist-info/RECORD,,
+indexify-0.2.32.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+indexify-0.2.32.dist-info/METADATA,sha256=3-sKMgJvXaZByfEAUE6h2fBtymF3pw7MXhXvv9gwhnU,6196
+indexify-0.2.32.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+indexify-0.2.32.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
+indexify-0.2.32.dist-info/RECORD,,