indexify 0.2.30__tar.gz → 0.2.32__tar.gz

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

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "indexify"
-version = "0.2.30"
+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,12 +25,8 @@ 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"
-pytest = "^7.1.1"
-pytest-cov = "^4.0.0"
-pytest-runner = "^6.0.0"
-pytest-watch = "^4.2.0"
 parameterized = "^0.9.0"
 
 [build-system]