indexify 0.2.28__tar.gz → 0.2.29__tar.gz

{indexify-0.2.28 → indexify-0.2.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: indexify
-Version: 0.2.28
+Version: 0.2.29
 Summary: Python Client for Indexify
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0

{indexify-0.2.28 → indexify-0.2.29}/indexify/functions_sdk/graph_definition.py

@@ -49,3 +49,11 @@ class ComputeGraphMetadata(BaseModel):
 
     def get_input_payload_serializer(self):
         return get_serializer(self.start_node.compute_fn.encoder)
+
+    def get_input_encoder(self) -> str:
+        if self.start_node.compute_fn:
+            return self.start_node.compute_fn.encoder
+        elif self.start_node.dynamic_router:
+            return self.start_node.dynamic_router.encoder
+
+        raise ValueError("start node is not set on the graph")

{indexify-0.2.28 → indexify-0.2.29}/indexify/http_client.py

@@ -1,6 +1,6 @@
 import json
 import os
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional
 
 import cloudpickle
 import httpx
@@ -13,11 +13,7 @@ from indexify.error import ApiException, GraphStillProcessing
 from indexify.functions_sdk.data_objects import IndexifyData
 from indexify.functions_sdk.graph import ComputeGraphMetadata, Graph
 from indexify.functions_sdk.indexify_functions import IndexifyFunction
-from indexify.functions_sdk.object_serializer import (
-    CloudPickleSerializer,
-    JsonSerializer,
-    get_serializer,
-)
+from indexify.functions_sdk.object_serializer import get_serializer
 from indexify.settings import DEFAULT_SERVICE_URL
 
 
@@ -276,11 +272,10 @@ class IndexifyClient:
         self,
         graph: str,
         block_until_done: bool = False,
-        serializer: Union[
-            CloudPickleSerializer, JsonSerializer
-        ] = CloudPickleSerializer,
+        serializer: str = "cloudpickle",
         **kwargs,
     ) -> str:
+        serializer = get_serializer(serializer)
         ser_input = serializer.serialize(kwargs)
         params = {"block_until_finish": block_until_done}
         kwargs = {

{indexify-0.2.28 → indexify-0.2.29}/indexify/remote_graph.py

@@ -1,6 +1,6 @@
 from typing import Any, List, Optional
 
-from indexify.functions_sdk.graph import Graph
+from indexify.functions_sdk.graph import ComputeGraphMetadata, Graph
 
 from .http_client import IndexifyClient
 from .settings import DEFAULT_SERVICE_URL
@@ -30,6 +30,8 @@ class RemoteGraph:
         else:
             self._client = IndexifyClient(service_url=server_url)
 
+        self._graph_definition: ComputeGraphMetadata = self._client.graph(self._name)
+
     def run(self, block_until_done: bool = False, **kwargs) -> str:
         """
         Run the graph with the given inputs. The input is for the start function of the graph.
@@ -49,7 +51,7 @@ class RemoteGraph:
         return self._client.invoke_graph_with_object(
             self._name,
             block_until_done,
-            self.graph.definition().get_input_payload_serializer(),
+            self._graph_definition.get_input_encoder(),
             **kwargs
         )
 
@@ -81,7 +83,6 @@ class RemoteGraph:
         :param client: The IndexifyClient used to communicate with the server.
             Prefered over server_url.
         """
-        cls.graph = g
         if not client:
             client = IndexifyClient(service_url=server_url)
         client.register_compute_graph(g, additional_modules)

{indexify-0.2.28 → indexify-0.2.29}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "indexify"
-version = "0.2.28"
+version = "0.2.29"
 description = "Python Client for Indexify"
 authors = ["Tensorlake Inc. <support@tensorlake.ai>"]
 license = "Apache 2.0"