indexify 0.2.19__py3-none-any.whl → 0.2.20__py3-none-any.whl

indexify/executor/task_reporter.py

@@ -29,10 +29,10 @@ class TaskReporter:
     def report_task_outcome(self, completed_task: CompletedTask):
         fn_outputs = []
         for output in completed_task.outputs or []:
-            output_bytes = MsgPackSerializer.serialize(output)
             print(
                 f"[bold]task-reporter[/bold] uploading output of size: {len(output_bytes)} bytes"
             )
+            output_bytes = MsgPackSerializer.serialize(output) 
             fn_outputs.append(
                 ("node_outputs", (nanoid.generate(), io.BytesIO(output_bytes)))
             )

indexify/remote_graph.py

@@ -11,23 +11,41 @@ class RemoteGraph:
         self,
         name: str,
         server_url: Optional[str] = DEFAULT_SERVICE_URL,
+        client: Optional[IndexifyClient] = None,
     ):
+        """
+        Create a handle to call a RemoteGraph by name.
+
+        Note: Use the class methods RemoteGraph.deploy or RemoteGraph.by_name to create a RemoteGraph object.
+
+        :param name: The name of the graph.
+        :param server_url: The URL of the server where the graph will be registered.
+            Not used if client is provided.
+        :param client: The IndexifyClient used to communicate with the server.
+            Prefered over server_url.
+        """
         self._name = name
-        self._client = IndexifyClient(service_url=server_url)
+        if client:
+            self._client = client
+        else:
+            self._client = IndexifyClient(service_url=server_url)
 
     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.
+
         :param block_until_done: If True, the function will block until the graph execution is complete.
         :param kwargs: The input to the start function of the graph. Pass the input as keyword arguments.
         :return: The invocation ID of the graph execution.
 
         Example:
-        @indexify_function()
-        def foo(x: int) -> int:
-            return x + 1
-        remote_graph = RemoteGraph.by_name("test")
-        invocation_id = remote_graph.run(x=1)
+
+            @indexify_function()
+            def foo(x: int) -> int:
+                return x + 1
+
+            remote_graph = RemoteGraph.by_name("test")
+            invocation_id = remote_graph.run(x=1)
         """
         return self._client.invoke_graph_with_object(
             self._name, block_until_done, **kwargs
@@ -36,6 +54,7 @@ class RemoteGraph:
     def rerun(self):
         """
         Rerun the graph with the given invocation ID.
+
         :param invocation_id: The invocation ID of the graph execution.
         """
         self._client.rerun_graph(self._name)
@@ -45,26 +64,38 @@ class RemoteGraph:
         cls,
         g: Graph,
         additional_modules=[],
-        server_url: Optional[str] = "http://localhost:8900",
+        server_url: Optional[str] = DEFAULT_SERVICE_URL,
+        client: Optional[IndexifyClient] = None,
     ):
         """
         Create a new RemoteGraph from a local Graph object.
+
         :param g: The local Graph object.
+        :param additional_modules: List of additional modules to be registered with the graph.
+            Needed for modules that are imported outside of an indexify function.
         :param server_url: The URL of the server where the graph will be registered.
+            Not used if client is provided.
+        :param client: The IndexifyClient used to communicate with the server.
+            Prefered over server_url.
         """
-        client = IndexifyClient(service_url=server_url)
+        if not client:
+            client = IndexifyClient(service_url=server_url)
         client.register_compute_graph(g, additional_modules)
-        return cls(name=g.name, server_url=server_url)
+        return cls(name=g.name, server_url=server_url, client=client)
 
     @classmethod
-    def by_name(cls, name: str, server_url: Optional[str] = "http://localhost:8900"):
+    def by_name(cls, name: str, server_url: Optional[str] = DEFAULT_SERVICE_URL, client: Optional[IndexifyClient] = None):
         """
         Create a handle to call a RemoteGraph by name.
+
         :param name: The name of the graph.
-        :param server_url: The URL of the server where the graph is registered.
+        :param server_url: The URL of the server where the graph will be registered.
+            Not used if client is provided.
+        :param client: The IndexifyClient used to communicate with the server.
+            Prefered over server_url.
         :return: A RemoteGraph object.
         """
-        return cls(name=name, server_url=server_url)
+        return cls(name=name, server_url=server_url, client=client)
 
     def output(
         self,
@@ -72,11 +103,14 @@ class RemoteGraph:
         fn_name: str,
     ) -> List[Any]:
         """
-        Returns the extracted objects by a graph for an ingested object. If the extractor name is provided, only the objects extracted by that extractor are returned.
-        If the extractor name is not provided, all the extracted objects are returned for the input object.
-        invocation_id: str: The ID of the ingested object
-        fn_name: Optional[str]: The name of the function whose output is to be returned if provided
-        return: List[Any]: Output of the function.
+        Returns the extracted objects by a graph for an ingested object.
+ 
+        - If the extractor name is provided, only the objects extracted by that extractor are returned.
+        - If the extractor name is not provided, all the extracted objects are returned for the input object.
+
+        :param invocation_id (str): The ID of the ingested object
+        :param fn_name (Optional[str]): The name of the function whose output is to be returned if provided
+        :return (List[Any]): Output of the function.
         """
 
         return self._client.graph_outputs(

{indexify-0.2.19.dist-info → indexify-0.2.20.dist-info}/METADATA

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

{indexify-0.2.19.dist-info → indexify-0.2.20.dist-info}/RECORD

@@ -11,7 +11,7 @@ indexify/executor/executor_tasks.py,sha256=gAZ2pvza1YwGlaR1o_tJW4SXtdCgK7sLJgp4W
 indexify/executor/function_worker.py,sha256=pCGn13rg4dEykzmwYNyCTkewrpyQXQR1cH6n2Hx5Lfc,5813
 indexify/executor/indexify_executor.py,sha256=2Ut_VX-Su_lm4b4aEROyRJ3gXx-uFHA-V7EN0sWiARE,771
 indexify/executor/runtime_probes.py,sha256=JY0FoxtlQ9sgsE8gBKWM5h3R1TWkYENhNF0HR2KkV4Q,1704
-indexify/executor/task_reporter.py,sha256=kbWoqY5IbAUdCzXDkzfuXEUznQN968ABPCYt-zCGJY4,3718
+indexify/executor/task_reporter.py,sha256=OsF-D88BIY3wzhBjZeKl4O5-BH8Hl6C8s1R8i4MHi8I,3719
 indexify/executor/task_store.py,sha256=q8s2gImsFffWeXQR0mk1Xlo1Aj_2GfclNPjQ2EA_YBo,3984
 indexify/functions_sdk/data_objects.py,sha256=CQZMzYiV7l6dyzFkYquWQHqdte6JnC7XA3i2ZyvPvgQ,844
 indexify/functions_sdk/graph.py,sha256=L8jtuSGaUk05JgT_cQX-n_1H9t84lcPa1v4XXlpuJa0,11143
@@ -23,11 +23,11 @@ indexify/functions_sdk/local_cache.py,sha256=cNWF67zbhbTJe3g86hyLBy3Rqzs6dNvp2Sj
 indexify/functions_sdk/object_serializer.py,sha256=Zz4GobW3ZamBBtFDF76QxU3TP6oJNdWnhsfKd0OUFoc,1660
 indexify/functions_sdk/pipeline.py,sha256=7hDatRK-SCHYvttf2Vj5YFyiJEVU0OOXEZBOIQenSmk,847
 indexify/http_client.py,sha256=MflqHbkkzYWw64nnmZCeJE76Q9p-o5hFnlI3h7kbJFI,13729
-indexify/remote_graph.py,sha256=ILg6IY6EFgyvnfz1DSzicBZhqvPkg2-UUwgI6lxp6sA,3094
+indexify/remote_graph.py,sha256=NOotayhw_x4-mAw1c3ooNe7gFE_q3QJ9qaO0t0Sb8sE,4397
 indexify/remote_pipeline.py,sha256=FW7IAv3r24OOpiqlprw3kuFrpdkqi6Ic4_tT26FThjA,761
 indexify/settings.py,sha256=LSaWZ0ADIVmUv6o6dHWRC3-Ry5uLbCw2sBSg1e_U7UM,99
-indexify-0.2.19.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.2.19.dist-info/METADATA,sha256=TdRtdNYEXcffmOtFnSvN3FDH631dPxUd-ZLJrZWvNOo,6199
-indexify-0.2.19.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-indexify-0.2.19.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
-indexify-0.2.19.dist-info/RECORD,,
+indexify-0.2.20.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+indexify-0.2.20.dist-info/METADATA,sha256=T8Qbsc59Y_uLGCVlRgUw_N-OblvDCM0g7SscDDVZ6jg,6199
+indexify-0.2.20.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+indexify-0.2.20.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
+indexify-0.2.20.dist-info/RECORD,,