indexify 0.3.12__py3-none-any.whl → 0.3.14__py3-none-any.whl

indexify/task_scheduler/proto/task_scheduler_pb2.pyi

@@ -0,0 +1,286 @@
+from typing import ClassVar as _ClassVar
+from typing import Iterable as _Iterable
+from typing import Mapping as _Mapping
+from typing import Optional as _Optional
+from typing import Union as _Union
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class GPUModel(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    GPU_MODEL_UNKNOWN: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_TESLA_T4_16GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_TESLA_V100_16GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_A10_24GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_A6000_48GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_A100_SXM4_40GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_A100_SXM4_80GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_A100_PCI_40GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_H100_SXM5_80GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_H100_PCI_80GB: _ClassVar[GPUModel]
+    GPU_MODEL_NVIDIA_RTX_6000_24GB: _ClassVar[GPUModel]
+
+class FunctionExecutorStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    FUNCTION_EXECUTOR_STATUS_UNKNOWN: _ClassVar[FunctionExecutorStatus]
+    FUNCTION_EXECUTOR_STATUS_STOPPED: _ClassVar[FunctionExecutorStatus]
+    FUNCTION_EXECUTOR_STATUS_STARTING_UP: _ClassVar[FunctionExecutorStatus]
+    FUNCTION_EXECUTOR_STATUS_STARTUP_FAILED_CUSTOMER_ERROR: _ClassVar[
+        FunctionExecutorStatus
+    ]
+    FUNCTION_EXECUTOR_STATUS_STARTUP_FAILED_PLATFORM_ERROR: _ClassVar[
+        FunctionExecutorStatus
+    ]
+    FUNCTION_EXECUTOR_STATUS_IDLE: _ClassVar[FunctionExecutorStatus]
+    FUNCTION_EXECUTOR_STATUS_RUNNING_TASK: _ClassVar[FunctionExecutorStatus]
+    FUNCTION_EXECUTOR_STATUS_UNHEALTHY: _ClassVar[FunctionExecutorStatus]
+    FUNCTION_EXECUTOR_STATUS_STOPPING: _ClassVar[FunctionExecutorStatus]
+
+class ExecutorStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    EXECUTOR_STATUS_UNKNOWN: _ClassVar[ExecutorStatus]
+    EXECUTOR_STATUS_STARTING: _ClassVar[ExecutorStatus]
+    EXECUTOR_STATUS_RUNNING: _ClassVar[ExecutorStatus]
+    EXECUTOR_STATUS_DRAINED: _ClassVar[ExecutorStatus]
+    EXECUTOR_STATUS_SHUTTING_DOWN: _ClassVar[ExecutorStatus]
+
+GPU_MODEL_UNKNOWN: GPUModel
+GPU_MODEL_NVIDIA_TESLA_T4_16GB: GPUModel
+GPU_MODEL_NVIDIA_TESLA_V100_16GB: GPUModel
+GPU_MODEL_NVIDIA_A10_24GB: GPUModel
+GPU_MODEL_NVIDIA_A6000_48GB: GPUModel
+GPU_MODEL_NVIDIA_A100_SXM4_40GB: GPUModel
+GPU_MODEL_NVIDIA_A100_SXM4_80GB: GPUModel
+GPU_MODEL_NVIDIA_A100_PCI_40GB: GPUModel
+GPU_MODEL_NVIDIA_H100_SXM5_80GB: GPUModel
+GPU_MODEL_NVIDIA_H100_PCI_80GB: GPUModel
+GPU_MODEL_NVIDIA_RTX_6000_24GB: GPUModel
+FUNCTION_EXECUTOR_STATUS_UNKNOWN: FunctionExecutorStatus
+FUNCTION_EXECUTOR_STATUS_STOPPED: FunctionExecutorStatus
+FUNCTION_EXECUTOR_STATUS_STARTING_UP: FunctionExecutorStatus
+FUNCTION_EXECUTOR_STATUS_STARTUP_FAILED_CUSTOMER_ERROR: FunctionExecutorStatus
+FUNCTION_EXECUTOR_STATUS_STARTUP_FAILED_PLATFORM_ERROR: FunctionExecutorStatus
+FUNCTION_EXECUTOR_STATUS_IDLE: FunctionExecutorStatus
+FUNCTION_EXECUTOR_STATUS_RUNNING_TASK: FunctionExecutorStatus
+FUNCTION_EXECUTOR_STATUS_UNHEALTHY: FunctionExecutorStatus
+FUNCTION_EXECUTOR_STATUS_STOPPING: FunctionExecutorStatus
+EXECUTOR_STATUS_UNKNOWN: ExecutorStatus
+EXECUTOR_STATUS_STARTING: ExecutorStatus
+EXECUTOR_STATUS_RUNNING: ExecutorStatus
+EXECUTOR_STATUS_DRAINED: ExecutorStatus
+EXECUTOR_STATUS_SHUTTING_DOWN: ExecutorStatus
+
+class GPUResources(_message.Message):
+    __slots__ = ("count", "model")
+    COUNT_FIELD_NUMBER: _ClassVar[int]
+    MODEL_FIELD_NUMBER: _ClassVar[int]
+    count: int
+    model: GPUModel
+    def __init__(
+        self, count: _Optional[int] = ..., model: _Optional[_Union[GPUModel, str]] = ...
+    ) -> None: ...
+
+class HostResources(_message.Message):
+    __slots__ = ("cpu_count", "memory_bytes", "disk_bytes", "gpu")
+    CPU_COUNT_FIELD_NUMBER: _ClassVar[int]
+    MEMORY_BYTES_FIELD_NUMBER: _ClassVar[int]
+    DISK_BYTES_FIELD_NUMBER: _ClassVar[int]
+    GPU_FIELD_NUMBER: _ClassVar[int]
+    cpu_count: int
+    memory_bytes: int
+    disk_bytes: int
+    gpu: GPUResources
+    def __init__(
+        self,
+        cpu_count: _Optional[int] = ...,
+        memory_bytes: _Optional[int] = ...,
+        disk_bytes: _Optional[int] = ...,
+        gpu: _Optional[_Union[GPUResources, _Mapping]] = ...,
+    ) -> None: ...
+
+class AllowedFunction(_message.Message):
+    __slots__ = ("namespace", "graph_name", "function_name", "graph_version")
+    NAMESPACE_FIELD_NUMBER: _ClassVar[int]
+    GRAPH_NAME_FIELD_NUMBER: _ClassVar[int]
+    FUNCTION_NAME_FIELD_NUMBER: _ClassVar[int]
+    GRAPH_VERSION_FIELD_NUMBER: _ClassVar[int]
+    namespace: str
+    graph_name: str
+    function_name: str
+    graph_version: str
+    def __init__(
+        self,
+        namespace: _Optional[str] = ...,
+        graph_name: _Optional[str] = ...,
+        function_name: _Optional[str] = ...,
+        graph_version: _Optional[str] = ...,
+    ) -> None: ...
+
+class FunctionExecutorDescription(_message.Message):
+    __slots__ = (
+        "id",
+        "namespace",
+        "graph_name",
+        "graph_version",
+        "function_name",
+        "image_uri",
+    )
+    ID_FIELD_NUMBER: _ClassVar[int]
+    NAMESPACE_FIELD_NUMBER: _ClassVar[int]
+    GRAPH_NAME_FIELD_NUMBER: _ClassVar[int]
+    GRAPH_VERSION_FIELD_NUMBER: _ClassVar[int]
+    FUNCTION_NAME_FIELD_NUMBER: _ClassVar[int]
+    IMAGE_URI_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    namespace: str
+    graph_name: str
+    graph_version: str
+    function_name: str
+    image_uri: str
+    def __init__(
+        self,
+        id: _Optional[str] = ...,
+        namespace: _Optional[str] = ...,
+        graph_name: _Optional[str] = ...,
+        graph_version: _Optional[str] = ...,
+        function_name: _Optional[str] = ...,
+        image_uri: _Optional[str] = ...,
+    ) -> None: ...
+
+class FunctionExecutorState(_message.Message):
+    __slots__ = ("description", "status")
+    DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    description: FunctionExecutorDescription
+    status: FunctionExecutorStatus
+    def __init__(
+        self,
+        description: _Optional[_Union[FunctionExecutorDescription, _Mapping]] = ...,
+        status: _Optional[_Union[FunctionExecutorStatus, str]] = ...,
+    ) -> None: ...
+
+class ExecutorState(_message.Message):
+    __slots__ = (
+        "executor_id",
+        "executor_status",
+        "host_resources",
+        "allowed_functions",
+        "function_executor_states",
+    )
+    EXECUTOR_ID_FIELD_NUMBER: _ClassVar[int]
+    EXECUTOR_STATUS_FIELD_NUMBER: _ClassVar[int]
+    HOST_RESOURCES_FIELD_NUMBER: _ClassVar[int]
+    ALLOWED_FUNCTIONS_FIELD_NUMBER: _ClassVar[int]
+    FUNCTION_EXECUTOR_STATES_FIELD_NUMBER: _ClassVar[int]
+    executor_id: str
+    executor_status: ExecutorStatus
+    host_resources: HostResources
+    allowed_functions: _containers.RepeatedCompositeFieldContainer[AllowedFunction]
+    function_executor_states: _containers.RepeatedCompositeFieldContainer[
+        FunctionExecutorState
+    ]
+    def __init__(
+        self,
+        executor_id: _Optional[str] = ...,
+        executor_status: _Optional[_Union[ExecutorStatus, str]] = ...,
+        host_resources: _Optional[_Union[HostResources, _Mapping]] = ...,
+        allowed_functions: _Optional[
+            _Iterable[_Union[AllowedFunction, _Mapping]]
+        ] = ...,
+        function_executor_states: _Optional[
+            _Iterable[_Union[FunctionExecutorState, _Mapping]]
+        ] = ...,
+    ) -> None: ...
+
+class ReportExecutorStateRequest(_message.Message):
+    __slots__ = ("executor_state",)
+    EXECUTOR_STATE_FIELD_NUMBER: _ClassVar[int]
+    executor_state: ExecutorState
+    def __init__(
+        self, executor_state: _Optional[_Union[ExecutorState, _Mapping]] = ...
+    ) -> None: ...
+
+class ReportExecutorStateResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class Task(_message.Message):
+    __slots__ = (
+        "id",
+        "namespace",
+        "graph_name",
+        "graph_version",
+        "function_name",
+        "graph_invocation_id",
+        "input_key",
+        "reducer_output_key",
+    )
+    ID_FIELD_NUMBER: _ClassVar[int]
+    NAMESPACE_FIELD_NUMBER: _ClassVar[int]
+    GRAPH_NAME_FIELD_NUMBER: _ClassVar[int]
+    GRAPH_VERSION_FIELD_NUMBER: _ClassVar[int]
+    FUNCTION_NAME_FIELD_NUMBER: _ClassVar[int]
+    GRAPH_INVOCATION_ID_FIELD_NUMBER: _ClassVar[int]
+    INPUT_KEY_FIELD_NUMBER: _ClassVar[int]
+    REDUCER_OUTPUT_KEY_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    namespace: str
+    graph_name: str
+    graph_version: str
+    function_name: str
+    graph_invocation_id: str
+    input_key: str
+    reducer_output_key: str
+    def __init__(
+        self,
+        id: _Optional[str] = ...,
+        namespace: _Optional[str] = ...,
+        graph_name: _Optional[str] = ...,
+        graph_version: _Optional[str] = ...,
+        function_name: _Optional[str] = ...,
+        graph_invocation_id: _Optional[str] = ...,
+        input_key: _Optional[str] = ...,
+        reducer_output_key: _Optional[str] = ...,
+    ) -> None: ...
+
+class TaskAllocation(_message.Message):
+    __slots__ = ("function_executor_id", "task")
+    FUNCTION_EXECUTOR_ID_FIELD_NUMBER: _ClassVar[int]
+    TASK_FIELD_NUMBER: _ClassVar[int]
+    function_executor_id: str
+    task: Task
+    def __init__(
+        self,
+        function_executor_id: _Optional[str] = ...,
+        task: _Optional[_Union[Task, _Mapping]] = ...,
+    ) -> None: ...
+
+class GetDesiredExecutorStatesRequest(_message.Message):
+    __slots__ = ("executor_id",)
+    EXECUTOR_ID_FIELD_NUMBER: _ClassVar[int]
+    executor_id: str
+    def __init__(self, executor_id: _Optional[str] = ...) -> None: ...
+
+class DesiredExecutorState(_message.Message):
+    __slots__ = ("function_executors", "task_allocations", "clock")
+    FUNCTION_EXECUTORS_FIELD_NUMBER: _ClassVar[int]
+    TASK_ALLOCATIONS_FIELD_NUMBER: _ClassVar[int]
+    CLOCK_FIELD_NUMBER: _ClassVar[int]
+    function_executors: _containers.RepeatedCompositeFieldContainer[
+        FunctionExecutorDescription
+    ]
+    task_allocations: _containers.RepeatedCompositeFieldContainer[TaskAllocation]
+    clock: int
+    def __init__(
+        self,
+        function_executors: _Optional[
+            _Iterable[_Union[FunctionExecutorDescription, _Mapping]]
+        ] = ...,
+        task_allocations: _Optional[_Iterable[_Union[TaskAllocation, _Mapping]]] = ...,
+        clock: _Optional[int] = ...,
+    ) -> None: ...

indexify/task_scheduler/proto/task_scheduler_pb2_grpc.py

@@ -0,0 +1,170 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import warnings
+
+import grpc
+
+from indexify.task_scheduler.proto import (
+    task_scheduler_pb2 as indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2,
+)
+
+GRPC_GENERATED_VERSION = "1.70.0"
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+
+    _version_not_supported = first_version_is_lower(
+        GRPC_VERSION, GRPC_GENERATED_VERSION
+    )
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f"The grpc package installed is at version {GRPC_VERSION},"
+        + f" but the generated code in indexify/task_scheduler/proto/task_scheduler_pb2_grpc.py depends on"
+        + f" grpcio>={GRPC_GENERATED_VERSION}."
+        + f" Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}"
+        + f" or downgrade your generated code using grpcio-tools<={GRPC_VERSION}."
+    )
+
+
+class TaskSchedulerServiceStub(object):
+    """Internal API for scheduling and running tasks on Executors. Executors are acting as clients of this API.
+    Server is responsible for scheduling tasks on Executors and Executors are responsible for running the tasks.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.report_executor_state = channel.unary_unary(
+            "/task_scheduler_service.TaskSchedulerService/report_executor_state",
+            request_serializer=indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.ReportExecutorStateRequest.SerializeToString,
+            response_deserializer=indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.ReportExecutorStateResponse.FromString,
+            _registered_method=True,
+        )
+        self.get_desired_executor_states = channel.unary_stream(
+            "/task_scheduler_service.TaskSchedulerService/get_desired_executor_states",
+            request_serializer=indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.GetDesiredExecutorStatesRequest.SerializeToString,
+            response_deserializer=indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.DesiredExecutorState.FromString,
+            _registered_method=True,
+        )
+
+
+class TaskSchedulerServiceServicer(object):
+    """Internal API for scheduling and running tasks on Executors. Executors are acting as clients of this API.
+    Server is responsible for scheduling tasks on Executors and Executors are responsible for running the tasks.
+    """
+
+    def report_executor_state(self, request, context):
+        """Called by Executor every 5 seconds to report that it's still alive and provide its current state.
+
+        Missing 3 reports will result in the Executor being deregistered by Server.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def get_desired_executor_states(self, request, context):
+        """Called by Executor to open a stream of its desired states. When Server wants Executor to change something
+        it puts a message on the stream with the new desired state of the Executor.
+
+        Depricated HTTP API is used to download the serialized graph and task inputs.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+
+def add_TaskSchedulerServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "report_executor_state": grpc.unary_unary_rpc_method_handler(
+            servicer.report_executor_state,
+            request_deserializer=indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.ReportExecutorStateRequest.FromString,
+            response_serializer=indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.ReportExecutorStateResponse.SerializeToString,
+        ),
+        "get_desired_executor_states": grpc.unary_stream_rpc_method_handler(
+            servicer.get_desired_executor_states,
+            request_deserializer=indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.GetDesiredExecutorStatesRequest.FromString,
+            response_serializer=indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.DesiredExecutorState.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+        "task_scheduler_service.TaskSchedulerService", rpc_method_handlers
+    )
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers(
+        "task_scheduler_service.TaskSchedulerService", rpc_method_handlers
+    )
+
+
+# This class is part of an EXPERIMENTAL API.
+class TaskSchedulerService(object):
+    """Internal API for scheduling and running tasks on Executors. Executors are acting as clients of this API.
+    Server is responsible for scheduling tasks on Executors and Executors are responsible for running the tasks.
+    """
+
+    @staticmethod
+    def report_executor_state(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/task_scheduler_service.TaskSchedulerService/report_executor_state",
+            indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.ReportExecutorStateRequest.SerializeToString,
+            indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.ReportExecutorStateResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True,
+        )
+
+    @staticmethod
+    def get_desired_executor_states(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_stream(
+            request,
+            target,
+            "/task_scheduler_service.TaskSchedulerService/get_desired_executor_states",
+            indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.GetDesiredExecutorStatesRequest.SerializeToString,
+            indexify_dot_task__scheduler_dot_proto_dot_task__scheduler__pb2.DesiredExecutorState.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True,
+        )

{indexify-0.3.12.dist-info → indexify-0.3.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: indexify
-Version: 0.3.12
+Version: 0.3.14
 Summary: Open Source Indexify components and helper tools
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0

{indexify-0.3.12.dist-info → indexify-0.3.14.dist-info}/RECORD

@@ -1,14 +1,14 @@
-indexify/cli/cli.py,sha256=SQOVWIr-ZNPilFAVQw1qRC42V8T0VEbVt_E1eClsnpM,10946
+indexify/cli/cli.py,sha256=Y0PwJKhc3MWOV4Sw9c01WjS0M43QSMcOP6IVMKUnjME,11027
 indexify/executor/README.md,sha256=ozC6_hMkhQQNVCMEpBxwiUALz6lwErPQxNxQfQDqnG4,2029
 indexify/executor/api_objects.py,sha256=TaYwDoo7EjuLBusxH512-KdvAJRtBwbEP2IObWraabU,1100
-indexify/executor/downloader.py,sha256=XjaGCzsGM3ex2HxbKGkIsB50OhlXgdnywWGCjUBdW2k,8127
-indexify/executor/executor.py,sha256=r5do6k7Izp9SIzebgOn2psYjzKZmNqHx3iuTW3vWrhg,10035
-indexify/executor/function_executor/function_executor.py,sha256=BxNhsW0uXxKjdKT5ixMAyIo5F7otdIT2c0FMQ09Lsrs,10234
-indexify/executor/function_executor/function_executor_state.py,sha256=IWPLWa7LaN0Eq8PDu-0kFzkuKJB0COShu7wCO1oyiNA,3141
-indexify/executor/function_executor/function_executor_states_container.py,sha256=x4hlF7ZBWswk9dkA06Rvgeqar6H9TWjZ7Etyy2CzBDE,2682
-indexify/executor/function_executor/health_checker.py,sha256=YT24ajPLdYOpdt2UNyJGviGbivCDXIJpeQOxQofcl50,3258
+indexify/executor/downloader.py,sha256=LkvAXfKxddnDzgfmwHcpDB_n795-eVKzn-hLjq4nUEM,9412
+indexify/executor/executor.py,sha256=stF2gTi0b2jUf5jtogh4rFz3jC1-jMOyhk9eaUno4dg,15200
+indexify/executor/function_executor/function_executor.py,sha256=sQ5FOdrjybDDsjagghlfjV06IXTpWWDBSTHqQXI-w9M,11245
+indexify/executor/function_executor/function_executor_state.py,sha256=_mqWPHI442ak2hwo21BbDclXIUhjg_M7tfg7lyOoxg8,3666
+indexify/executor/function_executor/function_executor_states_container.py,sha256=Z4tMQfOoGqaWotQXTIBvsujXkXAAw2UGsIxUnlUm8Wo,3067
+indexify/executor/function_executor/health_checker.py,sha256=qUUpG4oeVsPLibiCspAiRm-2Ldg46ulnnpj9EBXr1NQ,3916
 indexify/executor/function_executor/invocation_state_client.py,sha256=p-xgM4__cHR1ApvMV9hShrGWee_Je0VDhICZUGjpQY4,9644
-indexify/executor/function_executor/metrics/function_executor.py,sha256=y36cgZEp949HPCBVFF6uFQIxk2PvnSI74QSbxcAalI4,5659
+indexify/executor/function_executor/metrics/function_executor.py,sha256=KHzf4cMh_HqnOAE7TZ6_oIqUH3nExOkt9LSekAEN69w,6304
 indexify/executor/function_executor/metrics/function_executor_state.py,sha256=M7cMA7JY8_8FW9xjuSqtp6o2xxUgB31LJowo7kzcexg,352
 indexify/executor/function_executor/metrics/function_executor_state_container.py,sha256=6rrAfml-TivjkHatCM4BLY7jmVs523Wzb6QIysncc-0,302
 indexify/executor/function_executor/metrics/health_checker.py,sha256=EaeIYJPrQ-qqNMGZVGkvjPoeQSCl4FzPKXEv3Cly1NE,456
@@ -16,12 +16,12 @@ indexify/executor/function_executor/metrics/invocation_state_client.py,sha256=6F
 indexify/executor/function_executor/metrics/single_task_runner.py,sha256=7BJlGkdPGKeufMs3zWNO_1GRVzjINRY5rW3Mp4oWWec,805
 indexify/executor/function_executor/server/client_configuration.py,sha256=gOywMus0cotlX6NKIadEJwvOmBE-LbGE_wvoMi5-HzY,994
 indexify/executor/function_executor/server/function_executor_server.py,sha256=_DLivLDikupZusRk8gVWDk7fWPT9XjZ4un1yWSlOObs,883
-indexify/executor/function_executor/server/function_executor_server_factory.py,sha256=oBEuOwuGsZrvOZsv9v8M4JEtfproirE-cH28XCr0HoY,1739
+indexify/executor/function_executor/server/function_executor_server_factory.py,sha256=Xw-t6ZK-nAQUaRHf-peOnHP_CvLVH22oBYecEZ34GRw,1842
 indexify/executor/function_executor/server/subprocess_function_executor_server.py,sha256=JekDOqF7oFD4J6zcN3xB0Dxd1cgpEXMOsb_rKZOeBlI,668
 indexify/executor/function_executor/server/subprocess_function_executor_server_factory.py,sha256=xbH0-73tBM-i0dh1x_k0fDrNAi8WgDKjqiQ9xveU4Zg,4214
-indexify/executor/function_executor/single_task_runner.py,sha256=Td28vF6pNemKZEs4mNWeLKzy22XRmmMVgp2ipYO_OXU,9262
+indexify/executor/function_executor/single_task_runner.py,sha256=l02KDO4s7KMZxm2AazKC9SWqKBdy_mmrXTfZEyhNXAU,10205
 indexify/executor/function_executor/task_input.py,sha256=wSrHR4m0juiGClQyeVdhRC37QzDt6Rrjq-ZXJkfBi9k,584
-indexify/executor/function_executor/task_output.py,sha256=Qg7vojYi0WmeHRf2qlC-5h46jcwis13EgiN63OmWpcg,1229
+indexify/executor/function_executor/task_output.py,sha256=SQJSlrknB7Ylf5IOeINfBEgiplS5hAPJh1hYulhyvfU,1962
 indexify/executor/metrics/downloader.py,sha256=lctPh8xjkXeLEFJnl1hNrD1yEhLhIl5sggsR4Yoe_Zc,2746
 indexify/executor/metrics/executor.py,sha256=ua-Vv_k1CB4juJdF7tEBQbBMksqWAA3iXKKMKXZUCLk,2369
 indexify/executor/metrics/task_fetcher.py,sha256=iJEwCLzYr2cuz7hRvNiqaa2nvQP4OrA0hm0iJY0YKG0,736
@@ -37,10 +37,16 @@ indexify/executor/monitoring/prometheus_metrics_handler.py,sha256=KiGqSf7rkXTfbD
 indexify/executor/monitoring/server.py,sha256=yzdYhcxnmY6uTQUMt3vatF5jilN52ZtfFseOmHyQpTo,1254
 indexify/executor/monitoring/startup_probe_handler.py,sha256=zXXsBU15SMlBx1bSFpxWDfed1VHtKKnwvLQ8-frpG98,425
 indexify/executor/runtime_probes.py,sha256=bo6Dq6AGZpJH099j0DHtVSDEH80tv3j9MXf3VXSx_p8,2182
+indexify/executor/state_reconciler.py,sha256=hIO3BDjXJKicvS7Sgdd5eht9nTsZtleZi7beBPI8d8E,11764
+indexify/executor/state_reporter.py,sha256=83ImGX4b78bEDBJzTDsiEhSd8780a0uob5v6RgA2kzA,4514
 indexify/executor/task_fetcher.py,sha256=NpFfHgaY99bSL-K2D5kcDAMNUG2FArq0-qF_mgF-LBQ,3375
-indexify/executor/task_reporter.py,sha256=t7FzSKV7fG1fEtxyTcFh-sTWZ8WXApbG-qXaAMq_gRQ,7363
-indexify/executor/task_runner.py,sha256=tcS7hHdSVQexC3XEIsDTwX51bJK4iSQqC8CbWeNoud0,5590
-indexify-0.3.12.dist-info/METADATA,sha256=OkLYjGB3yomMTO21dsw1YR5q3swzqRsaWvsOzrPQTKg,1158
-indexify-0.3.12.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-indexify-0.3.12.dist-info/entry_points.txt,sha256=GU9wmsgvN7nQw3N2X0PMYn1RSvF6CrhH9RuC2D8d3Gk,53
-indexify-0.3.12.dist-info/RECORD,,
+indexify/executor/task_reporter.py,sha256=mYgwozUO95PEwYMmeeIS0-HfMrO4z3Nhy6IduMsMahM,7367
+indexify/executor/task_runner.py,sha256=ggV1IcOI2x4IWzzsulmClJP4mcgAA4z64W0Beo2GM2U,5938
+indexify/task_scheduler/proto/task_scheduler.proto,sha256=6zpGLqhITlbXtwYQwqZms7Ok5FYhjF6tWA1h-QbktSQ,5410
+indexify/task_scheduler/proto/task_scheduler_pb2.py,sha256=NeDSktCe7cs6E-_W_cL25qbIgp9uBJccv0bgts2Oufo,8767
+indexify/task_scheduler/proto/task_scheduler_pb2.pyi,sha256=MX4oyoqNQfhikOBSE7bnMLvFbvmdUqL3aBlEsC5Uo24,10620
+indexify/task_scheduler/proto/task_scheduler_pb2_grpc.py,sha256=a-KpbGWSGZIZ9F35OgwEjp25C4jySf5PW8BOf8UFR2A,7144
+indexify-0.3.14.dist-info/METADATA,sha256=M1Q7n7lwImSZ9c5J0SAMp5Agu79Ys9r0Nt9gQRFTdCs,1158
+indexify-0.3.14.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+indexify-0.3.14.dist-info/entry_points.txt,sha256=GU9wmsgvN7nQw3N2X0PMYn1RSvF6CrhH9RuC2D8d3Gk,53
+indexify-0.3.14.dist-info/RECORD,,