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

indexify/functions_sdk/graph.py

@@ -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,
@@ -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/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.31.dist-info → indexify-0.2.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: indexify
-Version: 0.2.31
+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.31.dist-info → indexify-0.2.32.dist-info}/RECORD

@@ -16,7 +16,7 @@ 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=_XsHfkQaxTnBNVcDlq4KsnyZ0HQ_DZOAuwygDzpsCNo,11805
+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
@@ -25,11 +25,11 @@ indexify/functions_sdk/local_cache.py,sha256=cNWF67zbhbTJe3g86hyLBy3Rqzs6dNvp2Sj
 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.31.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.2.31.dist-info/METADATA,sha256=Jrl1EVAJoI1uYMYKaA1pofIao2iV8FhfytvxEHXZvCE,6202
-indexify-0.2.31.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-indexify-0.2.31.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
-indexify-0.2.31.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,,