indexify 0.2.4__tar.gz → 0.2.6__tar.gz

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

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

{indexify-0.2.4 → indexify-0.2.6}/indexify/cli.py

@@ -122,6 +122,11 @@ def server_dev_mode():
 def build_image(workflow_file_path: str, func_names: List[str]):
     globals_dict = {}
 
+    # Add the folder in the workflow file path to the current Python path
+    folder_path = os.path.dirname(workflow_file_path)
+    if folder_path not in sys.path:
+        sys.path.append(folder_path)
+
     try:
         exec(open(workflow_file_path).read(), globals_dict)
     except FileNotFoundError as e:
@@ -240,8 +245,10 @@ WORKDIR /app
     console.print(f"{docker_file}", style="magenta")
 
     client = docker.from_env()
+    image_name =f"{image._image_name}:{image._tag}"
     client.images.build(
         fileobj=io.BytesIO(docker_file.encode()),
-        tag=f"{image._image_name}:{image._tag}",
+        tag=image_name,
         rm=True,
     )
+    print(f"built image: {image_name}")

{indexify-0.2.4 → indexify-0.2.6}/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.6}/indexify/functions_sdk/graph.py

@@ -132,8 +132,23 @@ 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]
+            print(f"registering module {module_name} with cloudpickle")
+            cloudpickle.register_pickle_by_value(module)
+
 
     def serialize(self):
+        self._register_cloudpickle()
         return cloudpickle.dumps(self)
 
     @staticmethod

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

@@ -4,9 +4,9 @@ class Image:
 
         self._tag = "latest"
 
-        self._base_image = None
+        self._base_image = "python:3.10.15-slim-bookworm"
 
-        self._run_strs = []
+        self._run_strs = ["pip install indexify"]
 
     def name(self, image_name):
         self._image_name = image_name

{indexify-0.2.4 → indexify-0.2.6}/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.6}/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.6}/pyproject.toml

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