indexify 0.4.22__py3-none-any.whl → 0.4.24__py3-none-any.whl

indexify/executor/function_executor_controller/task_output.py

@@ -1,19 +1,17 @@
 from typing import Any, Dict, List, Optional
 
 from tensorlake.function_executor.proto.function_executor_pb2 import (
-    SerializedObject,
+    BLOB,
+    SerializedObjectInsideBLOB,
 )
 
 from indexify.proto.executor_api_pb2 import (
-    DataPayload,
     FunctionExecutorTerminationReason,
     TaskAllocation,
     TaskFailureReason,
     TaskOutcomeCode,
 )
 
-from .function_executor_startup_output import FunctionExecutorStartupOutput
-
 
 class TaskMetrics:
     """Metrics for a task."""
@@ -30,33 +28,27 @@ class TaskOutput:
         self,
         allocation: TaskAllocation,
         outcome_code: TaskOutcomeCode,
-        # Optional[TaskFailureReason] is not supported in python 3.9
-        failure_reason: TaskFailureReason = None,
-        invocation_error_output: Optional[SerializedObject] = None,
-        function_outputs: List[SerializedObject] = [],
+        failure_reason: Optional[TaskFailureReason] = None,
+        function_outputs: List[SerializedObjectInsideBLOB] = [],
+        uploaded_function_outputs_blob: Optional[BLOB] = None,
+        invocation_error_output: Optional[SerializedObjectInsideBLOB] = None,
+        uploaded_invocation_error_blob: Optional[BLOB] = None,
         next_functions: List[str] = [],
-        stdout: Optional[str] = None,
-        stderr: Optional[str] = None,
         metrics: Optional[TaskMetrics] = None,
         execution_start_time: Optional[float] = None,
         execution_end_time: Optional[float] = None,
     ):
-        self.task = allocation.task
         self.allocation = allocation
-        self.function_outputs = function_outputs
-        self.next_functions = next_functions
-        self.stdout = stdout
-        self.stderr = stderr
         self.outcome_code = outcome_code
         self.failure_reason = failure_reason
+        self.function_outputs = function_outputs
+        self.uploaded_function_outputs_blob = uploaded_function_outputs_blob
         self.invocation_error_output = invocation_error_output
+        self.uploaded_invocation_error_blob = uploaded_invocation_error_blob
+        self.next_functions = next_functions
         self.metrics = metrics
         self.execution_start_time = execution_start_time
         self.execution_end_time = execution_end_time
-        self.uploaded_data_payloads: List[DataPayload] = []
-        self.uploaded_stdout: Optional[DataPayload] = None
-        self.uploaded_stderr: Optional[DataPayload] = None
-        self.uploaded_invocation_error_output: Optional[DataPayload] = None
 
     @classmethod
     def internal_error(
@@ -66,12 +58,10 @@ class TaskOutput:
         execution_end_time: Optional[float],
     ) -> "TaskOutput":
         """Creates a TaskOutput for an internal error."""
-        # We are not sharing internal error messages with the customer.
         return TaskOutput(
             allocation=allocation,
             outcome_code=TaskOutcomeCode.TASK_OUTCOME_CODE_FAILURE,
             failure_reason=TaskFailureReason.TASK_FAILURE_REASON_INTERNAL_ERROR,
-            stderr="Platform failed to execute the function.",
             execution_start_time=execution_start_time,
             execution_end_time=execution_end_time,
         )
@@ -80,17 +70,14 @@ class TaskOutput:
     def function_timeout(
         cls,
         allocation: TaskAllocation,
-        timeout_sec: float,
         execution_start_time: Optional[float],
         execution_end_time: Optional[float],
     ) -> "TaskOutput":
-        """Creates a TaskOutput for an function timeout error."""
-        # Task stdout, stderr is not available.
+        """Creates a TaskOutput for a function timeout error."""
         return TaskOutput(
             allocation=allocation,
             outcome_code=TaskOutcomeCode.TASK_OUTCOME_CODE_FAILURE,
             failure_reason=TaskFailureReason.TASK_FAILURE_REASON_FUNCTION_TIMEOUT,
-            stderr=f"Function exceeded its configured timeout of {timeout_sec:.3f} sec.",
             execution_start_time=execution_start_time,
             execution_end_time=execution_end_time,
         )
@@ -102,11 +89,13 @@ class TaskOutput:
         execution_start_time: Optional[float],
         execution_end_time: Optional[float],
     ) -> "TaskOutput":
-        """Creates a TaskOutput for an unresponsive FE."""
-        # Task stdout, stderr is not available.
+        """Creates a TaskOutput for an unresponsive FE aka grey failure."""
+        # When FE is unresponsive we don't know exact cause of the failure.
         return TaskOutput(
             allocation=allocation,
             outcome_code=TaskOutcomeCode.TASK_OUTCOME_CODE_FAILURE,
+            # Treat the grey failure as a function error and thus charge the customer.
+            # This is to prevent service abuse by intentionally misbehaving functions.
             failure_reason=TaskFailureReason.TASK_FAILURE_REASON_FUNCTION_ERROR,
             execution_start_time=execution_start_time,
             execution_end_time=execution_end_time,
@@ -144,21 +133,17 @@ class TaskOutput:
     def function_executor_startup_failed(
         cls,
         allocation: TaskAllocation,
-        fe_startup_output: FunctionExecutorStartupOutput,
+        fe_termination_reason: FunctionExecutorTerminationReason,
         logger: Any,
     ) -> "TaskOutput":
         """Creates a TaskOutput for the case when we fail a task that didn't run because its FE startup failed."""
-        output = TaskOutput(
+        return TaskOutput(
             allocation=allocation,
             outcome_code=TaskOutcomeCode.TASK_OUTCOME_CODE_FAILURE,
             failure_reason=_fe_startup_failure_reason_to_task_failure_reason(
-                fe_startup_output.termination_reason, logger
+                fe_termination_reason, logger
             ),
         )
-        # Use FE startup stdout, stderr for allocations that we failed because FE startup failed.
-        output.uploaded_stdout = fe_startup_output.stdout
-        output.uploaded_stderr = fe_startup_output.stderr
-        return output
 
 
 def _fe_startup_failure_reason_to_task_failure_reason(
@@ -180,6 +165,12 @@ def _fe_startup_failure_reason_to_task_failure_reason(
         == FunctionExecutorTerminationReason.FUNCTION_EXECUTOR_TERMINATION_REASON_STARTUP_FAILED_INTERNAL_ERROR
     ):
         return TaskFailureReason.TASK_FAILURE_REASON_INTERNAL_ERROR
+    elif (
+        fe_termination_reason
+        == FunctionExecutorTerminationReason.FUNCTION_EXECUTOR_TERMINATION_REASON_FUNCTION_CANCELLED
+    ):
+        # This fe termination reason is used when FE gets deleted by Server from desired state while it's starting up.
+        return TaskFailureReason.TASK_FAILURE_REASON_FUNCTION_EXECUTOR_TERMINATED
     else:
         logger.error(
             "unexpected function executor startup failure reason",
@@ -187,4 +178,4 @@ def _fe_startup_failure_reason_to_task_failure_reason(
                 fe_termination_reason
             ),
         )
-        return TaskFailureReason.TASK_FAILURE_REASON_UNKNOWN
+        return TaskFailureReason.TASK_FAILURE_REASON_INTERNAL_ERROR

indexify/executor/function_executor_controller/terminate_function_executor.py

@@ -29,7 +29,12 @@ async def terminate_function_executor(
             logger.info(
                 "destroying function executor",
             )
-            await function_executor.destroy()
+            try:
+                # This await is a cancellation point, need to shield to ensure we destroyed the FE.
+                await asyncio.shield(function_executor.destroy())
+            except asyncio.CancelledError:
+                # We actually destroyed the FE so we can return without error.
+                pass
 
     return FunctionExecutorTerminated(
         is_success=True,

indexify/executor/logging.py

@@ -0,0 +1,69 @@
+# Executor logging configuration.
+import logging
+import sys
+import traceback
+from typing import TextIO
+
+import structlog
+
+
+def configure_logging_early():
+    """Configures standard Python logging module.
+    By default 3rd party modules that are using standard Python logging module
+    (logging.getLogger()) have their log lines dropped unless the module gets configured.
+    Not dropping log lines from 3rd party modules is useful for debugging. E.g. this helps
+    debugging errors in grpc servers if exceptions happen inside grpc module code.
+    """
+    logging.basicConfig(
+        level=logging.WARNING,
+        # This log message format is a bit similar to the default structlog format.
+        format="%(asctime)s [%(levelname)s] %(message)s logger=%(name)s",
+        datefmt="%Y-%m-%d %H:%M:%S",
+        handlers=[logging.StreamHandler(sys.stdout)],
+    )
+
+
+def configure_development_mode_logging(compact_tracebacks=True):
+    processors = [
+        structlog.contextvars.merge_contextvars,
+        structlog.processors.add_log_level,
+        structlog.dev.set_exc_info,
+        structlog.processors.TimeStamper(fmt="%Y-%m-%d %H:%M:%S", utc=False),
+        structlog.dev.ConsoleRenderer(
+            exception_formatter=(
+                _compact_traceback_formatter
+                if compact_tracebacks
+                else structlog.dev.rich_traceback
+            ),
+        ),
+    ]
+    structlog.configure(
+        processors=processors,
+    )
+
+
+def configure_production_mode_logging():
+    processors = [
+        structlog.contextvars.merge_contextvars,
+        structlog.processors.add_log_level,
+        structlog.dev.set_exc_info,
+        structlog.processors.format_exc_info,
+        structlog.processors.TimeStamper(fmt="%Y-%m-%d %H:%M:%S", utc=False),
+        structlog.processors.JSONRenderer(),
+    ]
+    structlog.configure(processors=processors)
+
+
+def _compact_traceback_formatter(
+    sio: TextIO, exc_info: structlog.typing.ExcInfo
+) -> None:
+    print(
+        "\n (only the frame where the exception was raised is printed by default)",
+        file=sio,
+    )
+    traceback.print_exception(
+        *exc_info,
+        file=sio,
+        limit=-1,  # Print only 1 frame where the exception was raised
+        chain=False,
+    )

indexify/executor/monitoring/metrics.py

@@ -243,3 +243,25 @@ def _hours_to_sec(hours: int) -> float:
 
 def _days_to_sec(days: int) -> float:
     return _hours_to_sec(days * 24)
+
+
+class IdempotentCounterChanger:
+    """A wrapper that ensures that inc/dec operations on a counter are done only once.
+
+    This is useful for tracking the number of in-progress operations or objects that exist.
+    """
+
+    def __init__(self, counter: prometheus_client.Counter) -> None:
+        self._counter: prometheus_client.Counter = counter
+        self.is_incremented: bool = False
+        self.is_decremented: bool = False
+
+    def inc(self) -> None:
+        if not self.is_incremented:
+            self._counter.inc()
+            self.is_incremented = True
+
+    def dec(self) -> None:
+        if self.is_incremented and not self.is_decremented:
+            self._counter.dec()
+            self.is_decremented = True

indexify/proto/executor_api.proto

@@ -16,12 +16,15 @@ enum DataPayloadEncoding {
 message DataPayload {
     optional uint64 size = 2;
     optional string sha256_hash = 3;
-    // URI of the data.
+    // URI of the BLOB where the data is stored.
     // S3 URI if the data is stored in S3.
     // Starts with "file://"" prefix if the data is stored on a local file system.
     optional string uri = 4;
     optional DataPayloadEncoding encoding = 5;
+    // Not set and ignored by Server right now.
     optional uint64 encoding_version = 6;
+    // Offset inside the BLOB.
+    optional uint64 offset = 7;
 }
 
 // ===== report_executor_state RPC =====
@@ -118,6 +121,7 @@ message FunctionExecutorDescription {
     // URI prefix for the startup output payloads.
     // S3 URI if the data is stored in S3.
     // Starts with "file://"" prefix followed by an absolute directory path if the data is stored on a local file system.
+    // Deprecated: most probably going to be removed once external FE logs ingestion pipeline gets implemented.
     optional string output_payload_uri_prefix = 12;
 }
 
@@ -128,6 +132,7 @@ message FunctionExecutorState {
     repeated string allocation_ids_caused_termination = 4;
 }
 
+// Deprecated: most probably going to be removed once external FE logs ingestion pipeline gets implemented.
 message FunctionExecutorUpdate {
     optional FunctionExecutorDescription description = 1;
     optional DataPayload startup_stdout = 2;
@@ -207,6 +212,8 @@ message Task {
     // Starts with "file://"" prefix followed by an absolute directory path if the data is stored on a local file system.
     optional string output_payload_uri_prefix = 13;
     optional TaskRetryPolicy retry_policy = 14;
+    // BLOB URI prefix for the invocation error payloads.
+    optional string invocation_error_payload_uri_prefix = 15;
 }
 
 message TaskAllocation {
@@ -240,9 +247,10 @@ enum TaskOutcomeCode {
 enum TaskFailureReason {
     TASK_FAILURE_REASON_UNKNOWN = 0;
     // Internal error on Executor aka platform error.
-    // Includes grey failures when we can't determine the exact cause.
     TASK_FAILURE_REASON_INTERNAL_ERROR = 1;
     // Clear function code failure typically by raising an exception from the function code.
+    // Also a grey failure where we can't determine the exact cause. We attribute these to
+    // functions to prevent service abuse but not billed intenionally failing functions.
     TASK_FAILURE_REASON_FUNCTION_ERROR = 2;
     // Function code run time exceeded its configured timeout.
     TASK_FAILURE_REASON_FUNCTION_TIMEOUT = 3;
@@ -268,11 +276,11 @@ message TaskResult {
     repeated string next_functions = 11;
     repeated DataPayload function_outputs = 12;
     // Standard output and error streams of the function.
+    // Deprecated: most probably going to be removed once external FE logs ingestion pipeline gets implemented.
     optional DataPayload stdout = 13;
     optional DataPayload stderr = 14;
     // User payload for invocation error if task failed with invocation error.
     optional DataPayload invocation_error_output = 15;
-
     optional uint64 execution_duration_ms = 16;
 }
 

indexify/proto/executor_api_pb2.py

@@ -19,7 +19,7 @@ _sym_db = _symbol_database.Default()
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n!indexify/proto/executor_api.proto\x12\x0f\x65xecutor_api_pb"\xeb\x01\n\x0b\x44\x61taPayload\x12\x11\n\x04size\x18\x02 \x01(\x04H\x00\x88\x01\x01\x12\x18\n\x0bsha256_hash\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x10\n\x03uri\x18\x04 \x01(\tH\x02\x88\x01\x01\x12;\n\x08\x65ncoding\x18\x05 \x01(\x0e\x32$.executor_api_pb.DataPayloadEncodingH\x03\x88\x01\x01\x12\x1d\n\x10\x65ncoding_version\x18\x06 \x01(\x04H\x04\x88\x01\x01\x42\x07\n\x05_sizeB\x0e\n\x0c_sha256_hashB\x06\n\x04_uriB\x0b\n\t_encodingB\x13\n\x11_encoding_version"e\n\x0cGPUResources\x12\x12\n\x05\x63ount\x18\x01 \x01(\rH\x00\x88\x01\x01\x12-\n\x05model\x18\x02 \x01(\x0e\x32\x19.executor_api_pb.GPUModelH\x01\x88\x01\x01\x42\x08\n\x06_countB\x08\n\x06_model"\xc2\x01\n\rHostResources\x12\x16\n\tcpu_count\x18\x01 \x01(\rH\x00\x88\x01\x01\x12\x19\n\x0cmemory_bytes\x18\x02 \x01(\x04H\x01\x88\x01\x01\x12\x17\n\ndisk_bytes\x18\x03 \x01(\x04H\x02\x88\x01\x01\x12/\n\x03gpu\x18\x04 \x01(\x0b\x32\x1d.executor_api_pb.GPUResourcesH\x03\x88\x01\x01\x42\x0c\n\n_cpu_countB\x0f\n\r_memory_bytesB\r\n\x0b_disk_bytesB\x06\n\x04_gpu"\xbb\x01\n\x0f\x41llowedFunction\x12\x16\n\tnamespace\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x17\n\ngraph_name\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x1a\n\rfunction_name\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x1a\n\rgraph_version\x18\x04 \x01(\tH\x03\x88\x01\x01\x42\x0c\n\n_namespaceB\r\n\x0b_graph_nameB\x10\n\x0e_function_nameB\x10\n\x0e_graph_version"\xd8\x01\n\x19\x46unctionExecutorResources\x12\x1b\n\x0e\x63pu_ms_per_sec\x18\x01 \x01(\rH\x00\x88\x01\x01\x12\x19\n\x0cmemory_bytes\x18\x02 \x01(\x04H\x01\x88\x01\x01\x12\x17\n\ndisk_bytes\x18\x03 \x01(\x04H\x02\x88\x01\x01\x12/\n\x03gpu\x18\x04 \x01(\x0b\x32\x1d.executor_api_pb.GPUResourcesH\x03\x88\x01\x01\x42\x11\n\x0f_cpu_ms_per_secB\x0f\n\r_memory_bytesB\r\n\x0b_disk_bytesB\x06\n\x04_gpu"\xb3\x04\n\x1b\x46unctionExecutorDescription\x12\x0f\n\x02id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x16\n\tnamespace\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x17\n\ngraph_name\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x1a\n\rgraph_version\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x1a\n\rfunction_name\x18\x05 \x01(\tH\x04\x88\x01\x01\x12\x16\n\timage_uri\x18\x06 \x01(\tH\x05\x88\x01\x01\x12\x14\n\x0csecret_names\x18\x07 \x03(\t\x12%\n\x18\x63ustomer_code_timeout_ms\x18\t \x01(\rH\x06\x88\x01\x01\x12\x30\n\x05graph\x18\n \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x07\x88\x01\x01\x12\x42\n\tresources\x18\x0b \x01(\x0b\x32*.executor_api_pb.FunctionExecutorResourcesH\x08\x88\x01\x01\x12&\n\x19output_payload_uri_prefix\x18\x0c \x01(\tH\t\x88\x01\x01\x42\x05\n\x03_idB\x0c\n\n_namespaceB\r\n\x0b_graph_nameB\x10\n\x0e_graph_versionB\x10\n\x0e_function_nameB\x0c\n\n_image_uriB\x1b\n\x19_customer_code_timeout_msB\x08\n\x06_graphB\x0c\n\n_resourcesB\x1c\n\x1a_output_payload_uri_prefix"\xcf\x02\n\x15\x46unctionExecutorState\x12\x46\n\x0b\x64\x65scription\x18\x01 \x01(\x0b\x32,.executor_api_pb.FunctionExecutorDescriptionH\x00\x88\x01\x01\x12<\n\x06status\x18\x02 \x01(\x0e\x32\'.executor_api_pb.FunctionExecutorStatusH\x01\x88\x01\x01\x12S\n\x12termination_reason\x18\x03 \x01(\x0e\x32\x32.executor_api_pb.FunctionExecutorTerminationReasonH\x02\x88\x01\x01\x12)\n!allocation_ids_caused_termination\x18\x04 \x03(\tB\x0e\n\x0c_descriptionB\t\n\x07_statusB\x15\n\x13_termination_reason"\x8c\x02\n\x16\x46unctionExecutorUpdate\x12\x46\n\x0b\x64\x65scription\x18\x01 \x01(\x0b\x32,.executor_api_pb.FunctionExecutorDescriptionH\x00\x88\x01\x01\x12\x39\n\x0estartup_stdout\x18\x02 \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x01\x88\x01\x01\x12\x39\n\x0estartup_stderr\x18\x03 \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x02\x88\x01\x01\x42\x0e\n\x0c_descriptionB\x11\n\x0f_startup_stdoutB\x11\n\x0f_startup_stderr"\xce\x05\n\rExecutorState\x12\x18\n\x0b\x65xecutor_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08hostname\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x14\n\x07version\x18\x05 \x01(\tH\x02\x88\x01\x01\x12\x34\n\x06status\x18\x06 \x01(\x0e\x32\x1f.executor_api_pb.ExecutorStatusH\x03\x88\x01\x01\x12<\n\x0ftotal_resources\x18\r \x01(\x0b\x32\x1e.executor_api_pb.HostResourcesH\x04\x88\x01\x01\x12N\n!total_function_executor_resources\x18\x07 \x01(\x0b\x32\x1e.executor_api_pb.HostResourcesH\x05\x88\x01\x01\x12;\n\x11\x61llowed_functions\x18\x08 \x03(\x0b\x32 .executor_api_pb.AllowedFunction\x12H\n\x18\x66unction_executor_states\x18\t \x03(\x0b\x32&.executor_api_pb.FunctionExecutorState\x12:\n\x06labels\x18\n \x03(\x0b\x32*.executor_api_pb.ExecutorState.LabelsEntry\x12\x17\n\nstate_hash\x18\x0b \x01(\tH\x06\x88\x01\x01\x12\x19\n\x0cserver_clock\x18\x0c \x01(\x04H\x07\x88\x01\x01\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x0e\n\x0c_executor_idB\x0b\n\t_hostnameB\n\n\x08_versionB\t\n\x07_statusB\x12\n\x10_total_resourcesB$\n"_total_function_executor_resourcesB\r\n\x0b_state_hashB\x0f\n\r_server_clock"\xb9\x01\n\x0e\x45xecutorUpdate\x12\x18\n\x0b\x65xecutor_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x31\n\x0ctask_results\x18\x02 \x03(\x0b\x32\x1b.executor_api_pb.TaskResult\x12J\n\x19\x66unction_executor_updates\x18\x03 \x03(\x0b\x32\'.executor_api_pb.FunctionExecutorUpdateB\x0e\n\x0c_executor_id"\xbf\x01\n\x1aReportExecutorStateRequest\x12;\n\x0e\x65xecutor_state\x18\x01 \x01(\x0b\x32\x1e.executor_api_pb.ExecutorStateH\x00\x88\x01\x01\x12=\n\x0f\x65xecutor_update\x18\x02 \x01(\x0b\x32\x1f.executor_api_pb.ExecutorUpdateH\x01\x88\x01\x01\x42\x11\n\x0f_executor_stateB\x12\n\x10_executor_update"\x1d\n\x1bReportExecutorStateResponse"\xcf\x01\n\x0fTaskRetryPolicy\x12\x18\n\x0bmax_retries\x18\x01 \x01(\rH\x00\x88\x01\x01\x12\x1d\n\x10initial_delay_ms\x18\x02 \x01(\rH\x01\x88\x01\x01\x12\x19\n\x0cmax_delay_ms\x18\x03 \x01(\rH\x02\x88\x01\x01\x12\x1d\n\x10\x64\x65lay_multiplier\x18\x04 \x01(\rH\x03\x88\x01\x01\x42\x0e\n\x0c_max_retriesB\x13\n\x11_initial_delay_msB\x0f\n\r_max_delay_msB\x13\n\x11_delay_multiplier"\xc6\x04\n\x04Task\x12\x0f\n\x02id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x16\n\tnamespace\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x17\n\ngraph_name\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x1a\n\rgraph_version\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x1a\n\rfunction_name\x18\x05 \x01(\tH\x04\x88\x01\x01\x12 \n\x13graph_invocation_id\x18\x06 \x01(\tH\x05\x88\x01\x01\x12\x17\n\ntimeout_ms\x18\n \x01(\rH\x06\x88\x01\x01\x12\x30\n\x05input\x18\x0b \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x07\x88\x01\x01\x12\x38\n\rreducer_input\x18\x0c \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x08\x88\x01\x01\x12&\n\x19output_payload_uri_prefix\x18\r \x01(\tH\t\x88\x01\x01\x12;\n\x0cretry_policy\x18\x0e \x01(\x0b\x32 .executor_api_pb.TaskRetryPolicyH\n\x88\x01\x01\x42\x05\n\x03_idB\x0c\n\n_namespaceB\r\n\x0b_graph_nameB\x10\n\x0e_graph_versionB\x10\n\x0e_function_nameB\x16\n\x14_graph_invocation_idB\r\n\x0b_timeout_msB\x08\n\x06_inputB\x10\n\x0e_reducer_inputB\x1c\n\x1a_output_payload_uri_prefixB\x0f\n\r_retry_policy"\xad\x01\n\x0eTaskAllocation\x12!\n\x14\x66unction_executor_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12(\n\x04task\x18\x02 \x01(\x0b\x32\x15.executor_api_pb.TaskH\x01\x88\x01\x01\x12\x1a\n\rallocation_id\x18\x03 \x01(\tH\x02\x88\x01\x01\x42\x17\n\x15_function_executor_idB\x07\n\x05_taskB\x10\n\x0e_allocation_id"K\n\x1fGetDesiredExecutorStatesRequest\x12\x18\n\x0b\x65xecutor_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\x0e\n\x0c_executor_id"\xb9\x01\n\x14\x44\x65siredExecutorState\x12H\n\x12\x66unction_executors\x18\x01 \x03(\x0b\x32,.executor_api_pb.FunctionExecutorDescription\x12\x39\n\x10task_allocations\x18\x02 \x03(\x0b\x32\x1f.executor_api_pb.TaskAllocation\x12\x12\n\x05\x63lock\x18\x03 \x01(\x04H\x00\x88\x01\x01\x42\x08\n\x06_clock"\xcc\x06\n\nTaskResult\x12\x14\n\x07task_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1a\n\rallocation_id\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x16\n\tnamespace\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x17\n\ngraph_name\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x1a\n\rgraph_version\x18\x05 \x01(\tH\x04\x88\x01\x01\x12\x1a\n\rfunction_name\x18\x06 \x01(\tH\x05\x88\x01\x01\x12 \n\x13graph_invocation_id\x18\x07 \x01(\tH\x06\x88\x01\x01\x12;\n\x0coutcome_code\x18\t \x01(\x0e\x32 .executor_api_pb.TaskOutcomeCodeH\x07\x88\x01\x01\x12?\n\x0e\x66\x61ilure_reason\x18\n \x01(\x0e\x32".executor_api_pb.TaskFailureReasonH\x08\x88\x01\x01\x12\x16\n\x0enext_functions\x18\x0b \x03(\t\x12\x36\n\x10\x66unction_outputs\x18\x0c \x03(\x0b\x32\x1c.executor_api_pb.DataPayload\x12\x31\n\x06stdout\x18\r \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\t\x88\x01\x01\x12\x31\n\x06stderr\x18\x0e \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\n\x88\x01\x01\x12\x42\n\x17invocation_error_output\x18\x0f \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x0b\x88\x01\x01\x12"\n\x15\x65xecution_duration_ms\x18\x10 \x01(\x04H\x0c\x88\x01\x01\x42\n\n\x08_task_idB\x10\n\x0e_allocation_idB\x0c\n\n_namespaceB\r\n\x0b_graph_nameB\x10\n\x0e_graph_versionB\x10\n\x0e_function_nameB\x16\n\x14_graph_invocation_idB\x0f\n\r_outcome_codeB\x11\n\x0f_failure_reasonB\t\n\x07_stdoutB\t\n\x07_stderrB\x1a\n\x18_invocation_error_outputB\x18\n\x16_execution_duration_ms*\xd1\x01\n\x13\x44\x61taPayloadEncoding\x12!\n\x1d\x44\x41TA_PAYLOAD_ENCODING_UNKNOWN\x10\x00\x12#\n\x1f\x44\x41TA_PAYLOAD_ENCODING_UTF8_JSON\x10\x01\x12#\n\x1f\x44\x41TA_PAYLOAD_ENCODING_UTF8_TEXT\x10\x02\x12\'\n#DATA_PAYLOAD_ENCODING_BINARY_PICKLE\x10\x03\x12$\n DATA_PAYLOAD_ENCODING_BINARY_ZIP\x10\x04*\xd6\x01\n\x08GPUModel\x12\x15\n\x11GPU_MODEL_UNKNOWN\x10\x00\x12\x1e\n\x1aGPU_MODEL_NVIDIA_A100_40GB\x10\x01\x12\x1e\n\x1aGPU_MODEL_NVIDIA_A100_80GB\x10\x02\x12\x1e\n\x1aGPU_MODEL_NVIDIA_H100_80GB\x10\x03\x12\x1d\n\x19GPU_MODEL_NVIDIA_TESLA_T4\x10\x04\x12\x1a\n\x16GPU_MODEL_NVIDIA_A6000\x10\x05\x12\x18\n\x14GPU_MODEL_NVIDIA_A10\x10\x06*\xb3\x01\n\x16\x46unctionExecutorStatus\x12$\n FUNCTION_EXECUTOR_STATUS_UNKNOWN\x10\x00\x12$\n FUNCTION_EXECUTOR_STATUS_PENDING\x10\x01\x12$\n FUNCTION_EXECUTOR_STATUS_RUNNING\x10\x02\x12\'\n#FUNCTION_EXECUTOR_STATUS_TERMINATED\x10\x03*\x94\x04\n!FunctionExecutorTerminationReason\x12\x30\n,FUNCTION_EXECUTOR_TERMINATION_REASON_UNKNOWN\x10\x00\x12\x46\nBFUNCTION_EXECUTOR_TERMINATION_REASON_STARTUP_FAILED_INTERNAL_ERROR\x10\x01\x12\x46\nBFUNCTION_EXECUTOR_TERMINATION_REASON_STARTUP_FAILED_FUNCTION_ERROR\x10\x02\x12H\nDFUNCTION_EXECUTOR_TERMINATION_REASON_STARTUP_FAILED_FUNCTION_TIMEOUT\x10\x03\x12\x32\n.FUNCTION_EXECUTOR_TERMINATION_REASON_UNHEALTHY\x10\x0c\x12\x37\n3FUNCTION_EXECUTOR_TERMINATION_REASON_INTERNAL_ERROR\x10\r\x12\x39\n5FUNCTION_EXECUTOR_TERMINATION_REASON_FUNCTION_TIMEOUT\x10\x0e\x12;\n7FUNCTION_EXECUTOR_TERMINATION_REASON_FUNCTION_CANCELLED\x10\x0f*\xa5\x01\n\x0e\x45xecutorStatus\x12\x1b\n\x17\x45XECUTOR_STATUS_UNKNOWN\x10\x00\x12\x1f\n\x1b\x45XECUTOR_STATUS_STARTING_UP\x10\x01\x12\x1b\n\x17\x45XECUTOR_STATUS_RUNNING\x10\x02\x12\x1b\n\x17\x45XECUTOR_STATUS_DRAINED\x10\x03\x12\x1b\n\x17\x45XECUTOR_STATUS_STOPPED\x10\x04*n\n\x0fTaskOutcomeCode\x12\x1d\n\x19TASK_OUTCOME_CODE_UNKNOWN\x10\x00\x12\x1d\n\x19TASK_OUTCOME_CODE_SUCCESS\x10\x01\x12\x1d\n\x19TASK_OUTCOME_CODE_FAILURE\x10\x02*\xb6\x02\n\x11TaskFailureReason\x12\x1f\n\x1bTASK_FAILURE_REASON_UNKNOWN\x10\x00\x12&\n"TASK_FAILURE_REASON_INTERNAL_ERROR\x10\x01\x12&\n"TASK_FAILURE_REASON_FUNCTION_ERROR\x10\x02\x12(\n$TASK_FAILURE_REASON_FUNCTION_TIMEOUT\x10\x03\x12(\n$TASK_FAILURE_REASON_INVOCATION_ERROR\x10\x04\x12&\n"TASK_FAILURE_REASON_TASK_CANCELLED\x10\x05\x12\x34\n0TASK_FAILURE_REASON_FUNCTION_EXECUTOR_TERMINATED\x10\x06\x32\xff\x01\n\x0b\x45xecutorAPI\x12t\n\x15report_executor_state\x12+.executor_api_pb.ReportExecutorStateRequest\x1a,.executor_api_pb.ReportExecutorStateResponse"\x00\x12z\n\x1bget_desired_executor_states\x12\x30.executor_api_pb.GetDesiredExecutorStatesRequest\x1a%.executor_api_pb.DesiredExecutorState"\x00\x30\x01\x62\x06proto3'
+    b'\n!indexify/proto/executor_api.proto\x12\x0f\x65xecutor_api_pb"\x8b\x02\n\x0b\x44\x61taPayload\x12\x11\n\x04size\x18\x02 \x01(\x04H\x00\x88\x01\x01\x12\x18\n\x0bsha256_hash\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x10\n\x03uri\x18\x04 \x01(\tH\x02\x88\x01\x01\x12;\n\x08\x65ncoding\x18\x05 \x01(\x0e\x32$.executor_api_pb.DataPayloadEncodingH\x03\x88\x01\x01\x12\x1d\n\x10\x65ncoding_version\x18\x06 \x01(\x04H\x04\x88\x01\x01\x12\x13\n\x06offset\x18\x07 \x01(\x04H\x05\x88\x01\x01\x42\x07\n\x05_sizeB\x0e\n\x0c_sha256_hashB\x06\n\x04_uriB\x0b\n\t_encodingB\x13\n\x11_encoding_versionB\t\n\x07_offset"e\n\x0cGPUResources\x12\x12\n\x05\x63ount\x18\x01 \x01(\rH\x00\x88\x01\x01\x12-\n\x05model\x18\x02 \x01(\x0e\x32\x19.executor_api_pb.GPUModelH\x01\x88\x01\x01\x42\x08\n\x06_countB\x08\n\x06_model"\xc2\x01\n\rHostResources\x12\x16\n\tcpu_count\x18\x01 \x01(\rH\x00\x88\x01\x01\x12\x19\n\x0cmemory_bytes\x18\x02 \x01(\x04H\x01\x88\x01\x01\x12\x17\n\ndisk_bytes\x18\x03 \x01(\x04H\x02\x88\x01\x01\x12/\n\x03gpu\x18\x04 \x01(\x0b\x32\x1d.executor_api_pb.GPUResourcesH\x03\x88\x01\x01\x42\x0c\n\n_cpu_countB\x0f\n\r_memory_bytesB\r\n\x0b_disk_bytesB\x06\n\x04_gpu"\xbb\x01\n\x0f\x41llowedFunction\x12\x16\n\tnamespace\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x17\n\ngraph_name\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x1a\n\rfunction_name\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x1a\n\rgraph_version\x18\x04 \x01(\tH\x03\x88\x01\x01\x42\x0c\n\n_namespaceB\r\n\x0b_graph_nameB\x10\n\x0e_function_nameB\x10\n\x0e_graph_version"\xd8\x01\n\x19\x46unctionExecutorResources\x12\x1b\n\x0e\x63pu_ms_per_sec\x18\x01 \x01(\rH\x00\x88\x01\x01\x12\x19\n\x0cmemory_bytes\x18\x02 \x01(\x04H\x01\x88\x01\x01\x12\x17\n\ndisk_bytes\x18\x03 \x01(\x04H\x02\x88\x01\x01\x12/\n\x03gpu\x18\x04 \x01(\x0b\x32\x1d.executor_api_pb.GPUResourcesH\x03\x88\x01\x01\x42\x11\n\x0f_cpu_ms_per_secB\x0f\n\r_memory_bytesB\r\n\x0b_disk_bytesB\x06\n\x04_gpu"\xb3\x04\n\x1b\x46unctionExecutorDescription\x12\x0f\n\x02id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x16\n\tnamespace\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x17\n\ngraph_name\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x1a\n\rgraph_version\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x1a\n\rfunction_name\x18\x05 \x01(\tH\x04\x88\x01\x01\x12\x16\n\timage_uri\x18\x06 \x01(\tH\x05\x88\x01\x01\x12\x14\n\x0csecret_names\x18\x07 \x03(\t\x12%\n\x18\x63ustomer_code_timeout_ms\x18\t \x01(\rH\x06\x88\x01\x01\x12\x30\n\x05graph\x18\n \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x07\x88\x01\x01\x12\x42\n\tresources\x18\x0b \x01(\x0b\x32*.executor_api_pb.FunctionExecutorResourcesH\x08\x88\x01\x01\x12&\n\x19output_payload_uri_prefix\x18\x0c \x01(\tH\t\x88\x01\x01\x42\x05\n\x03_idB\x0c\n\n_namespaceB\r\n\x0b_graph_nameB\x10\n\x0e_graph_versionB\x10\n\x0e_function_nameB\x0c\n\n_image_uriB\x1b\n\x19_customer_code_timeout_msB\x08\n\x06_graphB\x0c\n\n_resourcesB\x1c\n\x1a_output_payload_uri_prefix"\xcf\x02\n\x15\x46unctionExecutorState\x12\x46\n\x0b\x64\x65scription\x18\x01 \x01(\x0b\x32,.executor_api_pb.FunctionExecutorDescriptionH\x00\x88\x01\x01\x12<\n\x06status\x18\x02 \x01(\x0e\x32\'.executor_api_pb.FunctionExecutorStatusH\x01\x88\x01\x01\x12S\n\x12termination_reason\x18\x03 \x01(\x0e\x32\x32.executor_api_pb.FunctionExecutorTerminationReasonH\x02\x88\x01\x01\x12)\n!allocation_ids_caused_termination\x18\x04 \x03(\tB\x0e\n\x0c_descriptionB\t\n\x07_statusB\x15\n\x13_termination_reason"\x8c\x02\n\x16\x46unctionExecutorUpdate\x12\x46\n\x0b\x64\x65scription\x18\x01 \x01(\x0b\x32,.executor_api_pb.FunctionExecutorDescriptionH\x00\x88\x01\x01\x12\x39\n\x0estartup_stdout\x18\x02 \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x01\x88\x01\x01\x12\x39\n\x0estartup_stderr\x18\x03 \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x02\x88\x01\x01\x42\x0e\n\x0c_descriptionB\x11\n\x0f_startup_stdoutB\x11\n\x0f_startup_stderr"\xce\x05\n\rExecutorState\x12\x18\n\x0b\x65xecutor_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08hostname\x18\x03 \x01(\tH\x01\x88\x01\x01\x12\x14\n\x07version\x18\x05 \x01(\tH\x02\x88\x01\x01\x12\x34\n\x06status\x18\x06 \x01(\x0e\x32\x1f.executor_api_pb.ExecutorStatusH\x03\x88\x01\x01\x12<\n\x0ftotal_resources\x18\r \x01(\x0b\x32\x1e.executor_api_pb.HostResourcesH\x04\x88\x01\x01\x12N\n!total_function_executor_resources\x18\x07 \x01(\x0b\x32\x1e.executor_api_pb.HostResourcesH\x05\x88\x01\x01\x12;\n\x11\x61llowed_functions\x18\x08 \x03(\x0b\x32 .executor_api_pb.AllowedFunction\x12H\n\x18\x66unction_executor_states\x18\t \x03(\x0b\x32&.executor_api_pb.FunctionExecutorState\x12:\n\x06labels\x18\n \x03(\x0b\x32*.executor_api_pb.ExecutorState.LabelsEntry\x12\x17\n\nstate_hash\x18\x0b \x01(\tH\x06\x88\x01\x01\x12\x19\n\x0cserver_clock\x18\x0c \x01(\x04H\x07\x88\x01\x01\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x0e\n\x0c_executor_idB\x0b\n\t_hostnameB\n\n\x08_versionB\t\n\x07_statusB\x12\n\x10_total_resourcesB$\n"_total_function_executor_resourcesB\r\n\x0b_state_hashB\x0f\n\r_server_clock"\xb9\x01\n\x0e\x45xecutorUpdate\x12\x18\n\x0b\x65xecutor_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x31\n\x0ctask_results\x18\x02 \x03(\x0b\x32\x1b.executor_api_pb.TaskResult\x12J\n\x19\x66unction_executor_updates\x18\x03 \x03(\x0b\x32\'.executor_api_pb.FunctionExecutorUpdateB\x0e\n\x0c_executor_id"\xbf\x01\n\x1aReportExecutorStateRequest\x12;\n\x0e\x65xecutor_state\x18\x01 \x01(\x0b\x32\x1e.executor_api_pb.ExecutorStateH\x00\x88\x01\x01\x12=\n\x0f\x65xecutor_update\x18\x02 \x01(\x0b\x32\x1f.executor_api_pb.ExecutorUpdateH\x01\x88\x01\x01\x42\x11\n\x0f_executor_stateB\x12\n\x10_executor_update"\x1d\n\x1bReportExecutorStateResponse"\xcf\x01\n\x0fTaskRetryPolicy\x12\x18\n\x0bmax_retries\x18\x01 \x01(\rH\x00\x88\x01\x01\x12\x1d\n\x10initial_delay_ms\x18\x02 \x01(\rH\x01\x88\x01\x01\x12\x19\n\x0cmax_delay_ms\x18\x03 \x01(\rH\x02\x88\x01\x01\x12\x1d\n\x10\x64\x65lay_multiplier\x18\x04 \x01(\rH\x03\x88\x01\x01\x42\x0e\n\x0c_max_retriesB\x13\n\x11_initial_delay_msB\x0f\n\r_max_delay_msB\x13\n\x11_delay_multiplier"\xa0\x05\n\x04Task\x12\x0f\n\x02id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x16\n\tnamespace\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x17\n\ngraph_name\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x1a\n\rgraph_version\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x1a\n\rfunction_name\x18\x05 \x01(\tH\x04\x88\x01\x01\x12 \n\x13graph_invocation_id\x18\x06 \x01(\tH\x05\x88\x01\x01\x12\x17\n\ntimeout_ms\x18\n \x01(\rH\x06\x88\x01\x01\x12\x30\n\x05input\x18\x0b \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x07\x88\x01\x01\x12\x38\n\rreducer_input\x18\x0c \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x08\x88\x01\x01\x12&\n\x19output_payload_uri_prefix\x18\r \x01(\tH\t\x88\x01\x01\x12;\n\x0cretry_policy\x18\x0e \x01(\x0b\x32 .executor_api_pb.TaskRetryPolicyH\n\x88\x01\x01\x12\x30\n#invocation_error_payload_uri_prefix\x18\x0f \x01(\tH\x0b\x88\x01\x01\x42\x05\n\x03_idB\x0c\n\n_namespaceB\r\n\x0b_graph_nameB\x10\n\x0e_graph_versionB\x10\n\x0e_function_nameB\x16\n\x14_graph_invocation_idB\r\n\x0b_timeout_msB\x08\n\x06_inputB\x10\n\x0e_reducer_inputB\x1c\n\x1a_output_payload_uri_prefixB\x0f\n\r_retry_policyB&\n$_invocation_error_payload_uri_prefix"\xad\x01\n\x0eTaskAllocation\x12!\n\x14\x66unction_executor_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12(\n\x04task\x18\x02 \x01(\x0b\x32\x15.executor_api_pb.TaskH\x01\x88\x01\x01\x12\x1a\n\rallocation_id\x18\x03 \x01(\tH\x02\x88\x01\x01\x42\x17\n\x15_function_executor_idB\x07\n\x05_taskB\x10\n\x0e_allocation_id"K\n\x1fGetDesiredExecutorStatesRequest\x12\x18\n\x0b\x65xecutor_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\x0e\n\x0c_executor_id"\xb9\x01\n\x14\x44\x65siredExecutorState\x12H\n\x12\x66unction_executors\x18\x01 \x03(\x0b\x32,.executor_api_pb.FunctionExecutorDescription\x12\x39\n\x10task_allocations\x18\x02 \x03(\x0b\x32\x1f.executor_api_pb.TaskAllocation\x12\x12\n\x05\x63lock\x18\x03 \x01(\x04H\x00\x88\x01\x01\x42\x08\n\x06_clock"\xcc\x06\n\nTaskResult\x12\x14\n\x07task_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1a\n\rallocation_id\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x16\n\tnamespace\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x17\n\ngraph_name\x18\x04 \x01(\tH\x03\x88\x01\x01\x12\x1a\n\rgraph_version\x18\x05 \x01(\tH\x04\x88\x01\x01\x12\x1a\n\rfunction_name\x18\x06 \x01(\tH\x05\x88\x01\x01\x12 \n\x13graph_invocation_id\x18\x07 \x01(\tH\x06\x88\x01\x01\x12;\n\x0coutcome_code\x18\t \x01(\x0e\x32 .executor_api_pb.TaskOutcomeCodeH\x07\x88\x01\x01\x12?\n\x0e\x66\x61ilure_reason\x18\n \x01(\x0e\x32".executor_api_pb.TaskFailureReasonH\x08\x88\x01\x01\x12\x16\n\x0enext_functions\x18\x0b \x03(\t\x12\x36\n\x10\x66unction_outputs\x18\x0c \x03(\x0b\x32\x1c.executor_api_pb.DataPayload\x12\x31\n\x06stdout\x18\r \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\t\x88\x01\x01\x12\x31\n\x06stderr\x18\x0e \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\n\x88\x01\x01\x12\x42\n\x17invocation_error_output\x18\x0f \x01(\x0b\x32\x1c.executor_api_pb.DataPayloadH\x0b\x88\x01\x01\x12"\n\x15\x65xecution_duration_ms\x18\x10 \x01(\x04H\x0c\x88\x01\x01\x42\n\n\x08_task_idB\x10\n\x0e_allocation_idB\x0c\n\n_namespaceB\r\n\x0b_graph_nameB\x10\n\x0e_graph_versionB\x10\n\x0e_function_nameB\x16\n\x14_graph_invocation_idB\x0f\n\r_outcome_codeB\x11\n\x0f_failure_reasonB\t\n\x07_stdoutB\t\n\x07_stderrB\x1a\n\x18_invocation_error_outputB\x18\n\x16_execution_duration_ms*\xd1\x01\n\x13\x44\x61taPayloadEncoding\x12!\n\x1d\x44\x41TA_PAYLOAD_ENCODING_UNKNOWN\x10\x00\x12#\n\x1f\x44\x41TA_PAYLOAD_ENCODING_UTF8_JSON\x10\x01\x12#\n\x1f\x44\x41TA_PAYLOAD_ENCODING_UTF8_TEXT\x10\x02\x12\'\n#DATA_PAYLOAD_ENCODING_BINARY_PICKLE\x10\x03\x12$\n DATA_PAYLOAD_ENCODING_BINARY_ZIP\x10\x04*\xd6\x01\n\x08GPUModel\x12\x15\n\x11GPU_MODEL_UNKNOWN\x10\x00\x12\x1e\n\x1aGPU_MODEL_NVIDIA_A100_40GB\x10\x01\x12\x1e\n\x1aGPU_MODEL_NVIDIA_A100_80GB\x10\x02\x12\x1e\n\x1aGPU_MODEL_NVIDIA_H100_80GB\x10\x03\x12\x1d\n\x19GPU_MODEL_NVIDIA_TESLA_T4\x10\x04\x12\x1a\n\x16GPU_MODEL_NVIDIA_A6000\x10\x05\x12\x18\n\x14GPU_MODEL_NVIDIA_A10\x10\x06*\xb3\x01\n\x16\x46unctionExecutorStatus\x12$\n FUNCTION_EXECUTOR_STATUS_UNKNOWN\x10\x00\x12$\n FUNCTION_EXECUTOR_STATUS_PENDING\x10\x01\x12$\n FUNCTION_EXECUTOR_STATUS_RUNNING\x10\x02\x12\'\n#FUNCTION_EXECUTOR_STATUS_TERMINATED\x10\x03*\x94\x04\n!FunctionExecutorTerminationReason\x12\x30\n,FUNCTION_EXECUTOR_TERMINATION_REASON_UNKNOWN\x10\x00\x12\x46\nBFUNCTION_EXECUTOR_TERMINATION_REASON_STARTUP_FAILED_INTERNAL_ERROR\x10\x01\x12\x46\nBFUNCTION_EXECUTOR_TERMINATION_REASON_STARTUP_FAILED_FUNCTION_ERROR\x10\x02\x12H\nDFUNCTION_EXECUTOR_TERMINATION_REASON_STARTUP_FAILED_FUNCTION_TIMEOUT\x10\x03\x12\x32\n.FUNCTION_EXECUTOR_TERMINATION_REASON_UNHEALTHY\x10\x0c\x12\x37\n3FUNCTION_EXECUTOR_TERMINATION_REASON_INTERNAL_ERROR\x10\r\x12\x39\n5FUNCTION_EXECUTOR_TERMINATION_REASON_FUNCTION_TIMEOUT\x10\x0e\x12;\n7FUNCTION_EXECUTOR_TERMINATION_REASON_FUNCTION_CANCELLED\x10\x0f*\xa5\x01\n\x0e\x45xecutorStatus\x12\x1b\n\x17\x45XECUTOR_STATUS_UNKNOWN\x10\x00\x12\x1f\n\x1b\x45XECUTOR_STATUS_STARTING_UP\x10\x01\x12\x1b\n\x17\x45XECUTOR_STATUS_RUNNING\x10\x02\x12\x1b\n\x17\x45XECUTOR_STATUS_DRAINED\x10\x03\x12\x1b\n\x17\x45XECUTOR_STATUS_STOPPED\x10\x04*n\n\x0fTaskOutcomeCode\x12\x1d\n\x19TASK_OUTCOME_CODE_UNKNOWN\x10\x00\x12\x1d\n\x19TASK_OUTCOME_CODE_SUCCESS\x10\x01\x12\x1d\n\x19TASK_OUTCOME_CODE_FAILURE\x10\x02*\xb6\x02\n\x11TaskFailureReason\x12\x1f\n\x1bTASK_FAILURE_REASON_UNKNOWN\x10\x00\x12&\n"TASK_FAILURE_REASON_INTERNAL_ERROR\x10\x01\x12&\n"TASK_FAILURE_REASON_FUNCTION_ERROR\x10\x02\x12(\n$TASK_FAILURE_REASON_FUNCTION_TIMEOUT\x10\x03\x12(\n$TASK_FAILURE_REASON_INVOCATION_ERROR\x10\x04\x12&\n"TASK_FAILURE_REASON_TASK_CANCELLED\x10\x05\x12\x34\n0TASK_FAILURE_REASON_FUNCTION_EXECUTOR_TERMINATED\x10\x06\x32\xff\x01\n\x0b\x45xecutorAPI\x12t\n\x15report_executor_state\x12+.executor_api_pb.ReportExecutorStateRequest\x1a,.executor_api_pb.ReportExecutorStateResponse"\x00\x12z\n\x1bget_desired_executor_states\x12\x30.executor_api_pb.GetDesiredExecutorStatesRequest\x1a%.executor_api_pb.DesiredExecutorState"\x00\x30\x01\x62\x06proto3'
 )
 
 _globals = globals()
@@ -31,58 +31,58 @@ if not _descriptor._USE_C_DESCRIPTORS:
     DESCRIPTOR._loaded_options = None
     _globals["_EXECUTORSTATE_LABELSENTRY"]._loaded_options = None
     _globals["_EXECUTORSTATE_LABELSENTRY"]._serialized_options = b"8\001"
-    _globals["_DATAPAYLOADENCODING"]._serialized_start = 5394
-    _globals["_DATAPAYLOADENCODING"]._serialized_end = 5603
-    _globals["_GPUMODEL"]._serialized_start = 5606
-    _globals["_GPUMODEL"]._serialized_end = 5820
-    _globals["_FUNCTIONEXECUTORSTATUS"]._serialized_start = 5823
-    _globals["_FUNCTIONEXECUTORSTATUS"]._serialized_end = 6002
-    _globals["_FUNCTIONEXECUTORTERMINATIONREASON"]._serialized_start = 6005
-    _globals["_FUNCTIONEXECUTORTERMINATIONREASON"]._serialized_end = 6537
-    _globals["_EXECUTORSTATUS"]._serialized_start = 6540
-    _globals["_EXECUTORSTATUS"]._serialized_end = 6705
-    _globals["_TASKOUTCOMECODE"]._serialized_start = 6707
-    _globals["_TASKOUTCOMECODE"]._serialized_end = 6817
-    _globals["_TASKFAILUREREASON"]._serialized_start = 6820
-    _globals["_TASKFAILUREREASON"]._serialized_end = 7130
+    _globals["_DATAPAYLOADENCODING"]._serialized_start = 5516
+    _globals["_DATAPAYLOADENCODING"]._serialized_end = 5725
+    _globals["_GPUMODEL"]._serialized_start = 5728
+    _globals["_GPUMODEL"]._serialized_end = 5942
+    _globals["_FUNCTIONEXECUTORSTATUS"]._serialized_start = 5945
+    _globals["_FUNCTIONEXECUTORSTATUS"]._serialized_end = 6124
+    _globals["_FUNCTIONEXECUTORTERMINATIONREASON"]._serialized_start = 6127
+    _globals["_FUNCTIONEXECUTORTERMINATIONREASON"]._serialized_end = 6659
+    _globals["_EXECUTORSTATUS"]._serialized_start = 6662
+    _globals["_EXECUTORSTATUS"]._serialized_end = 6827
+    _globals["_TASKOUTCOMECODE"]._serialized_start = 6829
+    _globals["_TASKOUTCOMECODE"]._serialized_end = 6939
+    _globals["_TASKFAILUREREASON"]._serialized_start = 6942
+    _globals["_TASKFAILUREREASON"]._serialized_end = 7252
     _globals["_DATAPAYLOAD"]._serialized_start = 55
-    _globals["_DATAPAYLOAD"]._serialized_end = 290
-    _globals["_GPURESOURCES"]._serialized_start = 292
-    _globals["_GPURESOURCES"]._serialized_end = 393
-    _globals["_HOSTRESOURCES"]._serialized_start = 396
-    _globals["_HOSTRESOURCES"]._serialized_end = 590
-    _globals["_ALLOWEDFUNCTION"]._serialized_start = 593
-    _globals["_ALLOWEDFUNCTION"]._serialized_end = 780
-    _globals["_FUNCTIONEXECUTORRESOURCES"]._serialized_start = 783
-    _globals["_FUNCTIONEXECUTORRESOURCES"]._serialized_end = 999
-    _globals["_FUNCTIONEXECUTORDESCRIPTION"]._serialized_start = 1002
-    _globals["_FUNCTIONEXECUTORDESCRIPTION"]._serialized_end = 1565
-    _globals["_FUNCTIONEXECUTORSTATE"]._serialized_start = 1568
-    _globals["_FUNCTIONEXECUTORSTATE"]._serialized_end = 1903
-    _globals["_FUNCTIONEXECUTORUPDATE"]._serialized_start = 1906
-    _globals["_FUNCTIONEXECUTORUPDATE"]._serialized_end = 2174
-    _globals["_EXECUTORSTATE"]._serialized_start = 2177
-    _globals["_EXECUTORSTATE"]._serialized_end = 2895
-    _globals["_EXECUTORSTATE_LABELSENTRY"]._serialized_start = 2708
-    _globals["_EXECUTORSTATE_LABELSENTRY"]._serialized_end = 2753
-    _globals["_EXECUTORUPDATE"]._serialized_start = 2898
-    _globals["_EXECUTORUPDATE"]._serialized_end = 3083
-    _globals["_REPORTEXECUTORSTATEREQUEST"]._serialized_start = 3086
-    _globals["_REPORTEXECUTORSTATEREQUEST"]._serialized_end = 3277
-    _globals["_REPORTEXECUTORSTATERESPONSE"]._serialized_start = 3279
-    _globals["_REPORTEXECUTORSTATERESPONSE"]._serialized_end = 3308
-    _globals["_TASKRETRYPOLICY"]._serialized_start = 3311
-    _globals["_TASKRETRYPOLICY"]._serialized_end = 3518
-    _globals["_TASK"]._serialized_start = 3521
-    _globals["_TASK"]._serialized_end = 4103
-    _globals["_TASKALLOCATION"]._serialized_start = 4106
-    _globals["_TASKALLOCATION"]._serialized_end = 4279
-    _globals["_GETDESIREDEXECUTORSTATESREQUEST"]._serialized_start = 4281
-    _globals["_GETDESIREDEXECUTORSTATESREQUEST"]._serialized_end = 4356
-    _globals["_DESIREDEXECUTORSTATE"]._serialized_start = 4359
-    _globals["_DESIREDEXECUTORSTATE"]._serialized_end = 4544
-    _globals["_TASKRESULT"]._serialized_start = 4547
-    _globals["_TASKRESULT"]._serialized_end = 5391
-    _globals["_EXECUTORAPI"]._serialized_start = 7133
-    _globals["_EXECUTORAPI"]._serialized_end = 7388
+    _globals["_DATAPAYLOAD"]._serialized_end = 322
+    _globals["_GPURESOURCES"]._serialized_start = 324
+    _globals["_GPURESOURCES"]._serialized_end = 425
+    _globals["_HOSTRESOURCES"]._serialized_start = 428
+    _globals["_HOSTRESOURCES"]._serialized_end = 622
+    _globals["_ALLOWEDFUNCTION"]._serialized_start = 625
+    _globals["_ALLOWEDFUNCTION"]._serialized_end = 812
+    _globals["_FUNCTIONEXECUTORRESOURCES"]._serialized_start = 815
+    _globals["_FUNCTIONEXECUTORRESOURCES"]._serialized_end = 1031
+    _globals["_FUNCTIONEXECUTORDESCRIPTION"]._serialized_start = 1034
+    _globals["_FUNCTIONEXECUTORDESCRIPTION"]._serialized_end = 1597
+    _globals["_FUNCTIONEXECUTORSTATE"]._serialized_start = 1600
+    _globals["_FUNCTIONEXECUTORSTATE"]._serialized_end = 1935
+    _globals["_FUNCTIONEXECUTORUPDATE"]._serialized_start = 1938
+    _globals["_FUNCTIONEXECUTORUPDATE"]._serialized_end = 2206
+    _globals["_EXECUTORSTATE"]._serialized_start = 2209
+    _globals["_EXECUTORSTATE"]._serialized_end = 2927
+    _globals["_EXECUTORSTATE_LABELSENTRY"]._serialized_start = 2740
+    _globals["_EXECUTORSTATE_LABELSENTRY"]._serialized_end = 2785
+    _globals["_EXECUTORUPDATE"]._serialized_start = 2930
+    _globals["_EXECUTORUPDATE"]._serialized_end = 3115
+    _globals["_REPORTEXECUTORSTATEREQUEST"]._serialized_start = 3118
+    _globals["_REPORTEXECUTORSTATEREQUEST"]._serialized_end = 3309
+    _globals["_REPORTEXECUTORSTATERESPONSE"]._serialized_start = 3311
+    _globals["_REPORTEXECUTORSTATERESPONSE"]._serialized_end = 3340
+    _globals["_TASKRETRYPOLICY"]._serialized_start = 3343
+    _globals["_TASKRETRYPOLICY"]._serialized_end = 3550
+    _globals["_TASK"]._serialized_start = 3553
+    _globals["_TASK"]._serialized_end = 4225
+    _globals["_TASKALLOCATION"]._serialized_start = 4228
+    _globals["_TASKALLOCATION"]._serialized_end = 4401
+    _globals["_GETDESIREDEXECUTORSTATESREQUEST"]._serialized_start = 4403
+    _globals["_GETDESIREDEXECUTORSTATESREQUEST"]._serialized_end = 4478
+    _globals["_DESIREDEXECUTORSTATE"]._serialized_start = 4481
+    _globals["_DESIREDEXECUTORSTATE"]._serialized_end = 4666
+    _globals["_TASKRESULT"]._serialized_start = 4669
+    _globals["_TASKRESULT"]._serialized_end = 5513
+    _globals["_EXECUTORAPI"]._serialized_start = 7255
+    _globals["_EXECUTORAPI"]._serialized_end = 7510
 # @@protoc_insertion_point(module_scope)

indexify/proto/executor_api_pb2.pyi

@@ -138,17 +138,19 @@ TASK_FAILURE_REASON_TASK_CANCELLED: TaskFailureReason
 TASK_FAILURE_REASON_FUNCTION_EXECUTOR_TERMINATED: TaskFailureReason
 
 class DataPayload(_message.Message):
-    __slots__ = ("size", "sha256_hash", "uri", "encoding", "encoding_version")
+    __slots__ = ("size", "sha256_hash", "uri", "encoding", "encoding_version", "offset")
     SIZE_FIELD_NUMBER: _ClassVar[int]
     SHA256_HASH_FIELD_NUMBER: _ClassVar[int]
     URI_FIELD_NUMBER: _ClassVar[int]
     ENCODING_FIELD_NUMBER: _ClassVar[int]
     ENCODING_VERSION_FIELD_NUMBER: _ClassVar[int]
+    OFFSET_FIELD_NUMBER: _ClassVar[int]
     size: int
     sha256_hash: str
     uri: str
     encoding: DataPayloadEncoding
     encoding_version: int
+    offset: int
     def __init__(
         self,
         size: _Optional[int] = ...,
@@ -156,6 +158,7 @@ class DataPayload(_message.Message):
         uri: _Optional[str] = ...,
         encoding: _Optional[_Union[DataPayloadEncoding, str]] = ...,
         encoding_version: _Optional[int] = ...,
+        offset: _Optional[int] = ...,
     ) -> None: ...
 
 class GPUResources(_message.Message):
@@ -449,6 +452,7 @@ class Task(_message.Message):
         "reducer_input",
         "output_payload_uri_prefix",
         "retry_policy",
+        "invocation_error_payload_uri_prefix",
     )
     ID_FIELD_NUMBER: _ClassVar[int]
     NAMESPACE_FIELD_NUMBER: _ClassVar[int]
@@ -461,6 +465,7 @@ class Task(_message.Message):
     REDUCER_INPUT_FIELD_NUMBER: _ClassVar[int]
     OUTPUT_PAYLOAD_URI_PREFIX_FIELD_NUMBER: _ClassVar[int]
     RETRY_POLICY_FIELD_NUMBER: _ClassVar[int]
+    INVOCATION_ERROR_PAYLOAD_URI_PREFIX_FIELD_NUMBER: _ClassVar[int]
     id: str
     namespace: str
     graph_name: str
@@ -472,6 +477,7 @@ class Task(_message.Message):
     reducer_input: DataPayload
     output_payload_uri_prefix: str
     retry_policy: TaskRetryPolicy
+    invocation_error_payload_uri_prefix: str
     def __init__(
         self,
         id: _Optional[str] = ...,
@@ -485,6 +491,7 @@ class Task(_message.Message):
         reducer_input: _Optional[_Union[DataPayload, _Mapping]] = ...,
         output_payload_uri_prefix: _Optional[str] = ...,
         retry_policy: _Optional[_Union[TaskRetryPolicy, _Mapping]] = ...,
+        invocation_error_payload_uri_prefix: _Optional[str] = ...,
     ) -> None: ...
 
 class TaskAllocation(_message.Message):

{indexify-0.4.22.dist-info → indexify-0.4.24.dist-info}/METADATA

@@ -1,20 +1,19 @@
 Metadata-Version: 2.3
 Name: indexify
-Version: 0.4.22
+Version: 0.4.24
 Summary: Open Source Indexify components and helper tools
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0
 Author: Tensorlake Inc.
 Author-email: support@tensorlake.ai
-Requires-Python: >=3.10,<4.0
+Requires-Python: >=3.11,<4.0
 Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: aiohttp (>=3.12.14,<4.0.0)
-Requires-Dist: boto3 (>=1.39.15,<2.0.0)
+Requires-Dist: aiohttp (>=3.12.15,<4.0.0)
+Requires-Dist: boto3 (>=1.40.6,<2.0.0)
 Requires-Dist: docker (>=7.1.0,<8.0.0)
 Requires-Dist: httpx[http2] (==0.27.2)
 Requires-Dist: nanoid (>=2.0.0,<3.0.0)
@@ -22,7 +21,8 @@ Requires-Dist: prometheus-client (>=0.22.1,<0.23.0)
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
 Requires-Dist: pydantic (>=2.11,<3.0)
 Requires-Dist: requests (>=2.32.4,<3.0.0)
-Requires-Dist: tensorlake (==0.2.37)
+Requires-Dist: structlog (==25.4.0)
+Requires-Dist: tensorlake (==0.2.39)
 Requires-Dist: urllib3 (>=2.5.0,<3.0.0)
 Project-URL: Repository, https://github.com/tensorlakeai/indexify
 Description-Content-Type: text/markdown