indexify 0.2.4__tar.gz → 0.2.5__tar.gz

{indexify-0.2.4 → indexify-0.2.5}/PKG-INFO

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

{indexify-0.2.4 → indexify-0.2.5}/indexify/executor/function_worker.py

@@ -135,6 +135,7 @@ def _run_function(
     fn_output = None
     has_failed = False
     exception_msg = None
+    print(f"[bold] function_worker: [/bold] invoking function {fn_name} in graph {graph_name}")
     with redirect_stdout(stdout_capture), redirect_stderr(stderr_capture):
         try:
             key = f"{namespace}/{graph_name}/{version}/{fn_name}"

{indexify-0.2.4 → indexify-0.2.5}/indexify/functions_sdk/graph.py

@@ -132,8 +132,22 @@ class Graph:
             self.add_node(node)
             self.routers[from_node.name].append(node.name)
         return self
+    
+    
+    def _register_cloudpickle(self):
+        # Get all unique modules from nodes and edges
+        modules = set()
+        for node in self.nodes.values():
+            modules.add(node.__module__)
+
+        # Register each module with cloudpickle
+        for module_name in modules:
+            module = sys.modules[module_name]
+            cloudpickle.register_pickle_by_value(module)
+
 
     def serialize(self):
+        self._register_cloudpickle()
         return cloudpickle.dumps(self)
 
     @staticmethod

{indexify-0.2.4 → indexify-0.2.5}/indexify/functions_sdk/image.py

@@ -4,7 +4,7 @@ class Image:
 
         self._tag = "latest"
 
-        self._base_image = None
+        self._base_image = "python:3.10.15-slim-bookworm"
 
         self._run_strs = []
 

{indexify-0.2.4 → indexify-0.2.5}/indexify/http_client.py

@@ -85,7 +85,7 @@ class IndexifyClient:
             message = (
                 f"Make sure the server is running and accesible at {self._service_url}"
             )
-            ex = ApiException(status="ConnectionError", message=message)
+            ex = ApiException(message=message)
             print(ex)
             raise ex
         return response

{indexify-0.2.4 → indexify-0.2.5}/indexify/remote_graph.py

@@ -34,18 +34,21 @@ class RemoteGraph:
         )
 
     @classmethod
-    def deploy(cls, g: Graph, server_url: Optional[str] = "http://localhost:8090"):
+    def deploy(cls, g: Graph, additional_modules=[], server_url: Optional[str] = "http://localhost:8900"):
         """
         Create a new RemoteGraph from a local Graph object.
         :param g: The local Graph object.
         :param server_url: The URL of the server where the graph will be registered.
         """
+        import cloudpickle
+        for module in additional_modules:
+            cloudpickle.register_pickle_by_value(module)
         client = IndexifyClient(service_url=server_url)
         client.register_compute_graph(g)
         return cls(name=g.name, server_url=server_url)
 
     @classmethod
-    def by_name(cls, name: str, server_url: Optional[str] = "http://localhost:8090"):
+    def by_name(cls, name: str, server_url: Optional[str] = "http://localhost:8900"):
         """
         Create a handle to call a RemoteGraph by name.
         :param name: The name of the graph.

{indexify-0.2.4 → indexify-0.2.5}/pyproject.toml

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