indexify 0.2.31__tar.gz → 0.2.32__tar.gz

{indexify-0.2.31 → indexify-0.2.32}/PKG-INFO

@@ -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 → indexify-0.2.32}/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-0.2.31 → indexify-0.2.32}/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 → indexify-0.2.32}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "indexify"
-version = "0.2.31"
+version = "0.2.32"
 description = "Python Client for Indexify"
 authors = ["Tensorlake Inc. <support@tensorlake.ai>"]
 license = "Apache 2.0"
@@ -15,7 +15,7 @@ indexify-cli = "indexify.cli:app"
 python = "^3.9"
 httpx = { version = "^0", extras = ["http2"] }
 pyyaml = "^6"
-pydantic = "^2.9.2"
+pydantic = "2.10.2"
 cloudpickle = "^3.1.0"
 rich = "^13.9.2"
 nanoid = "^2.0.0"
@@ -25,7 +25,7 @@ httpx-sse = "^0.4.0"
 jsonpickle = "^4.0.0"
 
 [tool.poetry.dev-dependencies]
-black = "^22.3.0"
+black = "^24.10.0"
 pylint = "^2.4.0"
 parameterized = "^0.9.0"