indexify 0.2.28__py3-none-any.whl → 0.2.29__py3-none-any.whl

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/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/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.dist-info → indexify-0.2.29.dist-info}/METADATA

@@ -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.dist-info → indexify-0.2.29.dist-info}/RECORD

@@ -17,19 +17,19 @@ indexify/executor/task_reporter.py,sha256=Wr8caDtjHBHvK5C-wwghJ5TNxIK_moxjQokGRc
 indexify/executor/task_store.py,sha256=u48FdRKAh_KH7WOMQOArdOY5CawlyW5MJx8V0W79JM0,3951
 indexify/functions_sdk/data_objects.py,sha256=wXbUa9hjU6rsXmmk19vQ5Kixf3FsI59VBWPNmHasAX0,854
 indexify/functions_sdk/graph.py,sha256=TbHtIcAzRcEn3BWewVhNsUGMNfRduI1aSAvK6Vyx-fk,11801
-indexify/functions_sdk/graph_definition.py,sha256=5M2I1FaWxmDucdirnaCYR4F_-8WHPBxyNLqKl_tennk,1293
+indexify/functions_sdk/graph_definition.py,sha256=5jUzjfk4gxxQCmQeOlNS9VndZK7KQBsZzABcDpQKgrs,1594
 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/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=Gi67Z1_hGY9N-so_6YY-PobP6JparH5vWq1PDQp2SEw,15924
-indexify/remote_graph.py,sha256=Qzc-xOYY8upwjX92FozIyx040HRr2aeuJMUiUkK3tDk,4670
+indexify/http_client.py,sha256=rFYjv0rRMo4sXp-TjWoflQeJaxNZK-Di6Gk3G78xKQI,15838
+indexify/remote_graph.py,sha256=1JogFFcWCKc_Dq4Rh1s6IwAnxB-qrFnKuDQIbpJso1Y,4745
 indexify/remote_pipeline.py,sha256=oqx57rSPszNS3DToXO_nf-CKqkCZWptm1u_p3orV_gQ,790
 indexify/settings.py,sha256=Ny59mzYI4gbXoK8hjx66a_men6ndbd1J1zCTcKOoyzg,50
-indexify-0.2.28.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.2.28.dist-info/METADATA,sha256=4L3k1UBoIMJUzmSlnPfM4yqiQbGhRG15J95HNkg-r4s,6202
-indexify-0.2.28.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-indexify-0.2.28.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
-indexify-0.2.28.dist-info/RECORD,,
+indexify-0.2.29.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+indexify-0.2.29.dist-info/METADATA,sha256=3lC5Cqd9Uym3Bk0AvQVTT8JTlHsodWXyktGjEAU9Y7s,6202
+indexify-0.2.29.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+indexify-0.2.29.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
+indexify-0.2.29.dist-info/RECORD,,