indexify 0.2.36__py3-none-any.whl → 0.2.37__py3-none-any.whl

indexify/executor/agent.py

@@ -299,7 +299,7 @@ class ExtractorAgent:
                 image_version=image_version,
                 labels=runtime_probe.labels,
             ).model_dump()
-            logging.info("registering_executor", executor_id=self._executor_id)
+            logging.info("registering_executor", executor_id=self._executor_id, url=url)
             try:
                 async with get_httpx_client(self._config_path, True) as client:
                     async with aconnect_sse(

indexify/functions_sdk/graph.py

@@ -147,6 +147,11 @@ class Graph:
         from_node: Union[Type[IndexifyFunction], Type[IndexifyRouter]],
         to_node: List[Union[Type[IndexifyFunction], Type[IndexifyRouter]]],
     ) -> "Graph":
+        if issubclass(from_node, IndexifyRouter):
+            raise ValueError(
+                "Cannot add edges from a router node, use route method instead"
+            )
+
         self.add_node(from_node)
         from_node_name = from_node.name
         for node in to_node:

indexify/functions_sdk/indexify_functions.py

@@ -112,11 +112,32 @@ class IndexifyRouter:
         pass
 
 
-from inspect import signature
+from inspect import Parameter, signature
 
 from typing_extensions import get_type_hints
 
 
+def _process_dict_arg(dict_arg: dict, sig: inspect.Signature) -> Tuple[list, dict]:
+    new_args = []
+    new_kwargs = {}
+    remaining_kwargs = dict_arg.copy()
+
+    # Match dictionary keys to function parameters
+    for param_name, param in sig.parameters.items():
+        if param_name in dict_arg:
+            new_args.append(dict_arg[param_name])
+            remaining_kwargs.pop(param_name, None)
+
+    if any(v.kind == Parameter.VAR_KEYWORD for v in sig.parameters.values()):
+        # Combine remaining dict items with additional kwargs
+        new_kwargs.update(remaining_kwargs)
+    elif len(remaining_kwargs) > 0:
+        # If there are remaining kwargs, add them as a single dict argument
+        new_args.append(remaining_kwargs)
+
+    return new_args, new_kwargs
+
+
 def indexify_router(
     name: Optional[str] = None,
     description: Optional[str] = "",
@@ -132,6 +153,13 @@ def indexify_router(
 
         # Create run method that preserves signature
         def run(self, *args, **kwargs):
+            # Process dictionary argument mapping it to args or to kwargs.
+            if len(args) == 1 and isinstance(args[0], dict):
+                sig = inspect.signature(fn)
+                dict_arg = args[0]
+                new_args, new_kwargs = _process_dict_arg(dict_arg, sig)
+                return fn(*new_args, **new_kwargs)
+
             return fn(*args, **kwargs)
 
         # Apply original signature and annotations to run method
@@ -173,6 +201,20 @@ def indexify_function(
 
         # Create run method that preserves signature
         def run(self, *args, **kwargs):
+            # Process dictionary argument mapping it to args or to kwargs.
+            if self.accumulate and len(args) == 2 and isinstance(args[1], dict):
+                sig = inspect.signature(fn)
+                new_args = [args[0]]  # Keep the accumulate argument
+                dict_arg = args[1]
+                new_args_from_dict, new_kwargs = _process_dict_arg(dict_arg, sig)
+                new_args.extend(new_args_from_dict)
+                return fn(*new_args, **new_kwargs)
+            elif len(args) == 1 and isinstance(args[0], dict):
+                sig = inspect.signature(fn)
+                dict_arg = args[0]
+                new_args, new_kwargs = _process_dict_arg(dict_arg, sig)
+                return fn(*new_args, **new_kwargs)
+
             return fn(*args, **kwargs)
 
         # Apply original signature and annotations to run method
@@ -252,14 +294,15 @@ class IndexifyFunctionWrapper:
     def run_router(
         self, input: Union[Dict, Type[BaseModel]]
     ) -> Tuple[List[str], Optional[str]]:
-        kwargs = input if isinstance(input, dict) else {"input": input}
         args = []
         kwargs = {}
-        if isinstance(input, dict):
-            kwargs = input
-        else:
-            args.append(input)
         try:
+            # tuple and list are considered positional arguments, list is used for compatibility
+            # with json encoding which won't deserialize in tuple.
+            if isinstance(input, tuple) or isinstance(input, list):
+                args += input
+            else:
+                args.append(input)
             extracted_data = self.indexify_function.run(*args, **kwargs)
         except Exception as e:
             return [], traceback.format_exc()
@@ -271,14 +314,18 @@ class IndexifyFunctionWrapper:
         return edges, None
 
     def run_fn(
-        self, input: Union[Dict, Type[BaseModel]], acc: Type[Any] = None
+        self, input: Union[Dict, Type[BaseModel], List, Tuple], acc: Type[Any] = None
     ) -> Tuple[List[Any], Optional[str]]:
         args = []
         kwargs = {}
+
         if acc is not None:
             args.append(acc)
-        if isinstance(input, dict):
-            kwargs = input
+
+        # tuple and list are considered positional arguments, list is used for compatibility
+        # with json encoding which won't deserialize in tuple.
+        if isinstance(input, tuple) or isinstance(input, list):
+            args += input
         else:
             args.append(input)
 

indexify/remote_graph.py

@@ -23,7 +23,7 @@ class RemoteGraph:
         :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.
+            Preferred over server_url.
         """
         self._name = name
         if client:
@@ -88,7 +88,7 @@ class RemoteGraph:
         :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.
+            Preferred over server_url.
         """
         g.validate_graph()
         if not client:
@@ -110,7 +110,7 @@ class RemoteGraph:
         :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.
+            Preferred over server_url.
         :return: A RemoteGraph object.
         """
         return cls(name=name, server_url=server_url, client=client)

{indexify-0.2.36.dist-info → indexify-0.2.37.dist-info}/METADATA

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

{indexify-0.2.36.dist-info → indexify-0.2.37.dist-info}/RECORD

@@ -5,7 +5,7 @@ indexify/data_loaders/__init__.py,sha256=Y5NEuseTcYAICRiweYw5wBQ2m2YplbsY21I7df-
 indexify/data_loaders/local_directory_loader.py,sha256=fCrgj5drnW71ZUdDDvcB1-VJjIs1w6Q8sEW0HSGSAiA,1247
 indexify/data_loaders/url_loader.py,sha256=32SERljcq1Xsi4RdLz2dgyk2TER5pQPTtXl3gUzwHbY,1533
 indexify/error.py,sha256=qAWr8R6AxPkjsxHSzXTc8zqYnNO_AjOqqYEPsQvF1Zs,238
-indexify/executor/agent.py,sha256=7fGywWXziuJkwNvqgyjMX3eKCv4XrpzDLkn0yoCALp8,14044
+indexify/executor/agent.py,sha256=zWdcroELAA-C8LaChR518ey5UKaUhfTHtkyawpdOkIE,14053
 indexify/executor/api_objects.py,sha256=mvmwGbK4paJNQGFvbtNHMPpiH_LpVhrlRnCcrqS6HOQ,859
 indexify/executor/downloader.py,sha256=dHLxoBnX8-Bh4yZtFDYptZNF6rlVtmTk_70JK8Ect5w,4184
 indexify/executor/executor_tasks.py,sha256=A0UIEZ5VaB6zSkFQG81UmTW0E57MTYhGlaXuAbRV8lQ,1884
@@ -16,20 +16,20 @@ indexify/executor/runtime_probes.py,sha256=mjw2_mGQ622wRT_39WPGGgPEZQTgtrf3-ICcU
 indexify/executor/task_reporter.py,sha256=XlEhNf_ScNnzG67zbtVwL7_9Bo8MvPZiHLI5UHymUnM,5305
 indexify/executor/task_store.py,sha256=JlRlWwAm4YjFRkTNRx-6GsUcmOzcyvzb5Csa5XDpRTI,3982
 indexify/functions_sdk/data_objects.py,sha256=wXbUa9hjU6rsXmmk19vQ5Kixf3FsI59VBWPNmHasAX0,854
-indexify/functions_sdk/graph.py,sha256=pca6LKbPAfYT-BMGTVL5sbPuuOhvvr_Yve6u3H0NWVk,13126
+indexify/functions_sdk/graph.py,sha256=K7ZF8fPhISVWoNzUTkDF8dMfpF-NKw88FSe9dZHISME,13301
 indexify/functions_sdk/graph_definition.py,sha256=rJmGcy9u5A_Sme6Ol33NsCnSKQVjyUfeN9LnH3bU88Y,1732
 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=J-etsuC_IGNCsUfbsJMrolyZVHZsurIMC5IaMQGiZnM,11187
+indexify/functions_sdk/indexify_functions.py,sha256=kjGmkRYnA_KO5ypkyaIhqrz0-3UJOFTPGldgOrHSRHY,13401
 indexify/functions_sdk/local_cache.py,sha256=cNWF67zbhbTJe3g86hyLBy3Rqzs6dNvp2SjLazGZWvw,1348
 indexify/functions_sdk/object_serializer.py,sha256=R58ALsl2Lb87ii6km4D6hBBsqRs_CHNISxhUICE2d9o,1931
 indexify/functions_sdk/pipeline.py,sha256=KmxZE8eBFAQ4bbEcYURXXR26HSyoAT3O6iu9H38-OXE,974
 indexify/http_client.py,sha256=iLafZagCFnlTS6uHfOjInogjg0uXW_zXEspIN7ttB5I,15903
-indexify/remote_graph.py,sha256=aox9NibZIU8YDiP92syerFSjiZc2jVPkF6g2kfqOvCA,5003
+indexify/remote_graph.py,sha256=OzBucU4buR5UdTViNJvh1RfnOTmYA34Uel5qXnRsQsA,5006
 indexify/remote_pipeline.py,sha256=oqx57rSPszNS3DToXO_nf-CKqkCZWptm1u_p3orV_gQ,790
 indexify/settings.py,sha256=Ny59mzYI4gbXoK8hjx66a_men6ndbd1J1zCTcKOoyzg,50
-indexify-0.2.36.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-indexify-0.2.36.dist-info/METADATA,sha256=cz7LrjMiw617WSpDsEw2wecgXW36fXItd8sxEi0rzic,6197
-indexify-0.2.36.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-indexify-0.2.36.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
-indexify-0.2.36.dist-info/RECORD,,
+indexify-0.2.37.dist-info/LICENSE.txt,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+indexify-0.2.37.dist-info/METADATA,sha256=UXkVvR1kLUGn34f_mxmQtlEFB4XpB078_Ob3Nbu5eGs,6197
+indexify-0.2.37.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+indexify-0.2.37.dist-info/entry_points.txt,sha256=Pih7WV-XMpAzI5dEvROcpLr-ybVhd9Y-AtuzBKUdcDs,49
+indexify-0.2.37.dist-info/RECORD,,