corvo-worker 0.2.0__tar.gz

corvo_worker-0.2.0/PKG-INFO

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: corvo-worker
+Version: 0.2.0
+Summary: Python worker runtime for Corvo
+License-Expression: MIT
+Requires-Python: >=3.9
+Description-Content-Type: text/plain
+Requires-Dist: corvo-client>=0.1.0
+Requires-Dist: grpcio>=1.62.0
+Requires-Dist: protobuf>=4.25.0
+Provides-Extra: dev
+Requires-Dist: grpcio-tools>=1.62.0; extra == "dev"

corvo_worker-0.2.0/corvo_worker/__init__.py

@@ -0,0 +1,4 @@
+from .worker import CorvoWorker, WorkerConfig, JobContext
+from corvo_client import PayloadTooLargeError
+
+__all__ = ["CorvoWorker", "WorkerConfig", "JobContext", "PayloadTooLargeError"]

corvo_worker-0.2.0/corvo_worker/gen/corvo/v1/worker_pb2.py

@@ -0,0 +1,93 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: corvo/v1/worker.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'corvo/v1/worker.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x63orvo/v1/worker.proto\x12\x08\x63orvo.v1\x1a\x1fgoogle/protobuf/timestamp.proto\"[\n\x0e\x45nqueueRequest\x12\r\n\x05queue\x18\x01 \x01(\t\x12\x14\n\x0cpayload_json\x18\x02 \x01(\t\x12$\n\x05\x61gent\x18\x03 \x01(\x0b\x32\x15.corvo.v1.AgentConfig\"J\n\x0f\x45nqueueResponse\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x0e\n\x06status\x18\x02 \x01(\t\x12\x17\n\x0funique_existing\x18\x03 \x01(\x08\"[\n\x0c\x46\x65tchRequest\x12\x0e\n\x06queues\x18\x01 \x03(\t\x12\x11\n\tworker_id\x18\x02 \x01(\t\x12\x10\n\x08hostname\x18\x03 \x01(\t\x12\x16\n\x0elease_duration\x18\x04 \x01(\x05\"\xe2\x01\n\rFetchResponse\x12\r\n\x05\x66ound\x18\x01 \x01(\x08\x12\x0e\n\x06job_id\x18\x02 \x01(\t\x12\r\n\x05queue\x18\x03 \x01(\t\x12\x14\n\x0cpayload_json\x18\x04 \x01(\t\x12\x0f\n\x07\x61ttempt\x18\x05 \x01(\x05\x12\x13\n\x0bmax_retries\x18\x06 \x01(\x05\x12\x16\n\x0elease_duration\x18\x07 \x01(\x05\x12\x17\n\x0f\x63heckpoint_json\x18\x08 \x01(\t\x12\x11\n\ttags_json\x18\t \x01(\t\x12#\n\x05\x61gent\x18\n \x01(\x0b\x32\x14.corvo.v1.AgentState\"o\n\x11\x46\x65tchBatchRequest\x12\x0e\n\x06queues\x18\x01 \x03(\t\x12\x11\n\tworker_id\x18\x02 \x01(\t\x12\x10\n\x08hostname\x18\x03 \x01(\t\x12\x16\n\x0elease_duration\x18\x04 \x01(\x05\x12\r\n\x05\x63ount\x18\x05 \x01(\x05\"\xd3\x01\n\rFetchBatchJob\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\r\n\x05queue\x18\x02 \x01(\t\x12\x14\n\x0cpayload_json\x18\x03 \x01(\t\x12\x0f\n\x07\x61ttempt\x18\x04 \x01(\x05\x12\x13\n\x0bmax_retries\x18\x05 \x01(\x05\x12\x16\n\x0elease_duration\x18\x06 \x01(\x05\x12\x17\n\x0f\x63heckpoint_json\x18\x07 \x01(\t\x12\x11\n\ttags_json\x18\x08 \x01(\t\x12#\n\x05\x61gent\x18\t \x01(\x0b\x32\x14.corvo.v1.AgentState\";\n\x12\x46\x65tchBatchResponse\x12%\n\x04jobs\x18\x01 \x03(\x0b\x32\x17.corvo.v1.FetchBatchJob\"\xa7\x01\n\x0bUsageReport\x12\x14\n\x0cinput_tokens\x18\x01 \x01(\x03\x12\x15\n\routput_tokens\x18\x02 \x01(\x03\x12\x1d\n\x15\x63\x61\x63he_creation_tokens\x18\x03 \x01(\x03\x12\x19\n\x11\x63\x61\x63he_read_tokens\x18\x04 \x01(\x03\x12\r\n\x05model\x18\x05 \x01(\t\x12\x10\n\x08provider\x18\x06 \x01(\t\x12\x10\n\x08\x63ost_usd\x18\x07 \x01(\x01\"\xaf\x01\n\nAckRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x13\n\x0bresult_json\x18\x02 \x01(\t\x12$\n\x05usage\x18\x03 \x01(\x0b\x32\x15.corvo.v1.UsageReport\x12\x17\n\x0f\x63heckpoint_json\x18\x04 \x01(\t\x12\x14\n\x0c\x61gent_status\x18\x05 \x01(\t\x12\x13\n\x0bhold_reason\x18\x06 \x01(\t\x12\x12\n\ntrace_json\x18\x07 \x01(\t\"\r\n\x0b\x41\x63kResponse\"Y\n\x0c\x41\x63kBatchItem\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x13\n\x0bresult_json\x18\x02 \x01(\t\x12$\n\x05usage\x18\x03 \x01(\x0b\x32\x15.corvo.v1.UsageReport\"8\n\x0f\x41\x63kBatchRequest\x12%\n\x05items\x18\x01 \x03(\x0b\x32\x16.corvo.v1.AckBatchItem\"!\n\x10\x41\x63kBatchResponse\x12\r\n\x05\x61\x63ked\x18\x01 \x01(\x05\"\xe6\x01\n\x16LifecycleStreamRequest\x12\x12\n\nrequest_id\x18\x01 \x01(\x04\x12\x0e\n\x06queues\x18\x02 \x03(\t\x12\x11\n\tworker_id\x18\x03 \x01(\t\x12\x10\n\x08hostname\x18\x04 \x01(\t\x12\x16\n\x0elease_duration\x18\x05 \x01(\x05\x12\x13\n\x0b\x66\x65tch_count\x18\x06 \x01(\x05\x12$\n\x04\x61\x63ks\x18\x07 \x03(\x0b\x32\x16.corvo.v1.AckBatchItem\x12\x30\n\x08\x65nqueues\x18\x08 \x03(\x0b\x32\x1e.corvo.v1.LifecycleEnqueueItem\"a\n\x14LifecycleEnqueueItem\x12\r\n\x05queue\x18\x01 \x01(\t\x12\x14\n\x0cpayload_json\x18\x02 \x01(\t\x12$\n\x05\x61gent\x18\x03 \x01(\x0b\x32\x15.corvo.v1.AgentConfig\"V\n\x0b\x41gentConfig\x12\x16\n\x0emax_iterations\x18\x01 \x01(\x05\x12\x14\n\x0cmax_cost_usd\x18\x02 \x01(\x01\x12\x19\n\x11iteration_timeout\x18\x03 \x01(\t\"\x80\x01\n\nAgentState\x12\x16\n\x0emax_iterations\x18\x01 \x01(\x05\x12\x14\n\x0cmax_cost_usd\x18\x02 \x01(\x01\x12\x19\n\x11iteration_timeout\x18\x03 \x01(\t\x12\x11\n\titeration\x18\x04 \x01(\x05\x12\x16\n\x0etotal_cost_usd\x18\x05 \x01(\x01\"\xa1\x01\n\x17LifecycleStreamResponse\x12\x12\n\nrequest_id\x18\x01 \x01(\x04\x12%\n\x04jobs\x18\x02 \x03(\x0b\x32\x17.corvo.v1.FetchBatchJob\x12\r\n\x05\x61\x63ked\x18\x03 \x01(\x05\x12\r\n\x05\x65rror\x18\x04 \x01(\t\x12\x18\n\x10\x65nqueued_job_ids\x18\x05 \x03(\t\x12\x13\n\x0bleader_addr\x18\x06 \x01(\t\"?\n\x0b\x46\x61ilRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\r\n\x05\x65rror\x18\x02 \x01(\t\x12\x11\n\tbacktrace\x18\x03 \x01(\t\"o\n\x0c\x46\x61ilResponse\x12\x0e\n\x06status\x18\x01 \x01(\t\x12\x33\n\x0fnext_attempt_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x1a\n\x12\x61ttempts_remaining\x18\x03 \x01(\x05\"\x80\x01\n\x12HeartbeatJobUpdate\x12\x15\n\rprogress_json\x18\x01 \x01(\t\x12\x17\n\x0f\x63heckpoint_json\x18\x02 \x01(\t\x12$\n\x05usage\x18\x03 \x01(\x0b\x32\x15.corvo.v1.UsageReport\x12\x14\n\x0cstream_delta\x18\x04 \x01(\t\"\x91\x01\n\x10HeartbeatRequest\x12\x32\n\x04jobs\x18\x01 \x03(\x0b\x32$.corvo.v1.HeartbeatRequest.JobsEntry\x1aI\n\tJobsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b\x32\x1c.corvo.v1.HeartbeatJobUpdate:\x02\x38\x01\"&\n\x14HeartbeatJobResponse\x12\x0e\n\x06status\x18\x01 \x01(\t\"\x95\x01\n\x11HeartbeatResponse\x12\x33\n\x04jobs\x18\x01 \x03(\x0b\x32%.corvo.v1.HeartbeatResponse.JobsEntry\x1aK\n\tJobsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12-\n\x05value\x18\x02 \x01(\x0b\x32\x1e.corvo.v1.HeartbeatJobResponse:\x02\x38\x01\x32\xb2\x04\n\rWorkerService\x12@\n\x07\x45nqueue\x12\x18.corvo.v1.EnqueueRequest\x1a\x19.corvo.v1.EnqueueResponse\"\x00\x12:\n\x05\x46\x65tch\x12\x16.corvo.v1.FetchRequest\x1a\x17.corvo.v1.FetchResponse\"\x00\x12I\n\nFetchBatch\x12\x1b.corvo.v1.FetchBatchRequest\x1a\x1c.corvo.v1.FetchBatchResponse\"\x00\x12\x34\n\x03\x41\x63k\x12\x14.corvo.v1.AckRequest\x1a\x15.corvo.v1.AckResponse\"\x00\x12\x43\n\x08\x41\x63kBatch\x12\x19.corvo.v1.AckBatchRequest\x1a\x1a.corvo.v1.AckBatchResponse\"\x00\x12\\\n\x0fStreamLifecycle\x12 .corvo.v1.LifecycleStreamRequest\x1a!.corvo.v1.LifecycleStreamResponse\"\x00(\x01\x30\x01\x12\x37\n\x04\x46\x61il\x12\x15.corvo.v1.FailRequest\x1a\x16.corvo.v1.FailResponse\"\x00\x12\x46\n\tHeartbeat\x12\x1a.corvo.v1.HeartbeatRequest\x1a\x1b.corvo.v1.HeartbeatResponse\"\x00\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvo.v1.worker_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_HEARTBEATREQUEST_JOBSENTRY']._loaded_options = None
+  _globals['_HEARTBEATREQUEST_JOBSENTRY']._serialized_options = b'8\001'
+  _globals['_HEARTBEATRESPONSE_JOBSENTRY']._loaded_options = None
+  _globals['_HEARTBEATRESPONSE_JOBSENTRY']._serialized_options = b'8\001'
+  _globals['_ENQUEUEREQUEST']._serialized_start=68
+  _globals['_ENQUEUEREQUEST']._serialized_end=159
+  _globals['_ENQUEUERESPONSE']._serialized_start=161
+  _globals['_ENQUEUERESPONSE']._serialized_end=235
+  _globals['_FETCHREQUEST']._serialized_start=237
+  _globals['_FETCHREQUEST']._serialized_end=328
+  _globals['_FETCHRESPONSE']._serialized_start=331
+  _globals['_FETCHRESPONSE']._serialized_end=557
+  _globals['_FETCHBATCHREQUEST']._serialized_start=559
+  _globals['_FETCHBATCHREQUEST']._serialized_end=670
+  _globals['_FETCHBATCHJOB']._serialized_start=673
+  _globals['_FETCHBATCHJOB']._serialized_end=884
+  _globals['_FETCHBATCHRESPONSE']._serialized_start=886
+  _globals['_FETCHBATCHRESPONSE']._serialized_end=945
+  _globals['_USAGEREPORT']._serialized_start=948
+  _globals['_USAGEREPORT']._serialized_end=1115
+  _globals['_ACKREQUEST']._serialized_start=1118
+  _globals['_ACKREQUEST']._serialized_end=1293
+  _globals['_ACKRESPONSE']._serialized_start=1295
+  _globals['_ACKRESPONSE']._serialized_end=1308
+  _globals['_ACKBATCHITEM']._serialized_start=1310
+  _globals['_ACKBATCHITEM']._serialized_end=1399
+  _globals['_ACKBATCHREQUEST']._serialized_start=1401
+  _globals['_ACKBATCHREQUEST']._serialized_end=1457
+  _globals['_ACKBATCHRESPONSE']._serialized_start=1459
+  _globals['_ACKBATCHRESPONSE']._serialized_end=1492
+  _globals['_LIFECYCLESTREAMREQUEST']._serialized_start=1495
+  _globals['_LIFECYCLESTREAMREQUEST']._serialized_end=1725
+  _globals['_LIFECYCLEENQUEUEITEM']._serialized_start=1727
+  _globals['_LIFECYCLEENQUEUEITEM']._serialized_end=1824
+  _globals['_AGENTCONFIG']._serialized_start=1826
+  _globals['_AGENTCONFIG']._serialized_end=1912
+  _globals['_AGENTSTATE']._serialized_start=1915
+  _globals['_AGENTSTATE']._serialized_end=2043
+  _globals['_LIFECYCLESTREAMRESPONSE']._serialized_start=2046
+  _globals['_LIFECYCLESTREAMRESPONSE']._serialized_end=2207
+  _globals['_FAILREQUEST']._serialized_start=2209
+  _globals['_FAILREQUEST']._serialized_end=2272
+  _globals['_FAILRESPONSE']._serialized_start=2274
+  _globals['_FAILRESPONSE']._serialized_end=2385
+  _globals['_HEARTBEATJOBUPDATE']._serialized_start=2388
+  _globals['_HEARTBEATJOBUPDATE']._serialized_end=2516
+  _globals['_HEARTBEATREQUEST']._serialized_start=2519
+  _globals['_HEARTBEATREQUEST']._serialized_end=2664
+  _globals['_HEARTBEATREQUEST_JOBSENTRY']._serialized_start=2591
+  _globals['_HEARTBEATREQUEST_JOBSENTRY']._serialized_end=2664
+  _globals['_HEARTBEATJOBRESPONSE']._serialized_start=2666
+  _globals['_HEARTBEATJOBRESPONSE']._serialized_end=2704
+  _globals['_HEARTBEATRESPONSE']._serialized_start=2707
+  _globals['_HEARTBEATRESPONSE']._serialized_end=2856
+  _globals['_HEARTBEATRESPONSE_JOBSENTRY']._serialized_start=2781
+  _globals['_HEARTBEATRESPONSE_JOBSENTRY']._serialized_end=2856
+  _globals['_WORKERSERVICE']._serialized_start=2859
+  _globals['_WORKERSERVICE']._serialized_end=3421
+# @@protoc_insertion_point(module_scope)

corvo_worker-0.2.0/corvo_worker/gen/corvo/v1/worker_pb2.pyi

@@ -0,0 +1,304 @@
+import datetime
+
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class EnqueueRequest(_message.Message):
+    __slots__ = ("queue", "payload_json", "agent")
+    QUEUE_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_JSON_FIELD_NUMBER: _ClassVar[int]
+    AGENT_FIELD_NUMBER: _ClassVar[int]
+    queue: str
+    payload_json: str
+    agent: AgentConfig
+    def __init__(self, queue: _Optional[str] = ..., payload_json: _Optional[str] = ..., agent: _Optional[_Union[AgentConfig, _Mapping]] = ...) -> None: ...
+
+class EnqueueResponse(_message.Message):
+    __slots__ = ("job_id", "status", "unique_existing")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    UNIQUE_EXISTING_FIELD_NUMBER: _ClassVar[int]
+    job_id: str
+    status: str
+    unique_existing: bool
+    def __init__(self, job_id: _Optional[str] = ..., status: _Optional[str] = ..., unique_existing: bool = ...) -> None: ...
+
+class FetchRequest(_message.Message):
+    __slots__ = ("queues", "worker_id", "hostname", "lease_duration")
+    QUEUES_FIELD_NUMBER: _ClassVar[int]
+    WORKER_ID_FIELD_NUMBER: _ClassVar[int]
+    HOSTNAME_FIELD_NUMBER: _ClassVar[int]
+    LEASE_DURATION_FIELD_NUMBER: _ClassVar[int]
+    queues: _containers.RepeatedScalarFieldContainer[str]
+    worker_id: str
+    hostname: str
+    lease_duration: int
+    def __init__(self, queues: _Optional[_Iterable[str]] = ..., worker_id: _Optional[str] = ..., hostname: _Optional[str] = ..., lease_duration: _Optional[int] = ...) -> None: ...
+
+class FetchResponse(_message.Message):
+    __slots__ = ("found", "job_id", "queue", "payload_json", "attempt", "max_retries", "lease_duration", "checkpoint_json", "tags_json", "agent")
+    FOUND_FIELD_NUMBER: _ClassVar[int]
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    QUEUE_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_JSON_FIELD_NUMBER: _ClassVar[int]
+    ATTEMPT_FIELD_NUMBER: _ClassVar[int]
+    MAX_RETRIES_FIELD_NUMBER: _ClassVar[int]
+    LEASE_DURATION_FIELD_NUMBER: _ClassVar[int]
+    CHECKPOINT_JSON_FIELD_NUMBER: _ClassVar[int]
+    TAGS_JSON_FIELD_NUMBER: _ClassVar[int]
+    AGENT_FIELD_NUMBER: _ClassVar[int]
+    found: bool
+    job_id: str
+    queue: str
+    payload_json: str
+    attempt: int
+    max_retries: int
+    lease_duration: int
+    checkpoint_json: str
+    tags_json: str
+    agent: AgentState
+    def __init__(self, found: bool = ..., job_id: _Optional[str] = ..., queue: _Optional[str] = ..., payload_json: _Optional[str] = ..., attempt: _Optional[int] = ..., max_retries: _Optional[int] = ..., lease_duration: _Optional[int] = ..., checkpoint_json: _Optional[str] = ..., tags_json: _Optional[str] = ..., agent: _Optional[_Union[AgentState, _Mapping]] = ...) -> None: ...
+
+class FetchBatchRequest(_message.Message):
+    __slots__ = ("queues", "worker_id", "hostname", "lease_duration", "count")
+    QUEUES_FIELD_NUMBER: _ClassVar[int]
+    WORKER_ID_FIELD_NUMBER: _ClassVar[int]
+    HOSTNAME_FIELD_NUMBER: _ClassVar[int]
+    LEASE_DURATION_FIELD_NUMBER: _ClassVar[int]
+    COUNT_FIELD_NUMBER: _ClassVar[int]
+    queues: _containers.RepeatedScalarFieldContainer[str]
+    worker_id: str
+    hostname: str
+    lease_duration: int
+    count: int
+    def __init__(self, queues: _Optional[_Iterable[str]] = ..., worker_id: _Optional[str] = ..., hostname: _Optional[str] = ..., lease_duration: _Optional[int] = ..., count: _Optional[int] = ...) -> None: ...
+
+class FetchBatchJob(_message.Message):
+    __slots__ = ("job_id", "queue", "payload_json", "attempt", "max_retries", "lease_duration", "checkpoint_json", "tags_json", "agent")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    QUEUE_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_JSON_FIELD_NUMBER: _ClassVar[int]
+    ATTEMPT_FIELD_NUMBER: _ClassVar[int]
+    MAX_RETRIES_FIELD_NUMBER: _ClassVar[int]
+    LEASE_DURATION_FIELD_NUMBER: _ClassVar[int]
+    CHECKPOINT_JSON_FIELD_NUMBER: _ClassVar[int]
+    TAGS_JSON_FIELD_NUMBER: _ClassVar[int]
+    AGENT_FIELD_NUMBER: _ClassVar[int]
+    job_id: str
+    queue: str
+    payload_json: str
+    attempt: int
+    max_retries: int
+    lease_duration: int
+    checkpoint_json: str
+    tags_json: str
+    agent: AgentState
+    def __init__(self, job_id: _Optional[str] = ..., queue: _Optional[str] = ..., payload_json: _Optional[str] = ..., attempt: _Optional[int] = ..., max_retries: _Optional[int] = ..., lease_duration: _Optional[int] = ..., checkpoint_json: _Optional[str] = ..., tags_json: _Optional[str] = ..., agent: _Optional[_Union[AgentState, _Mapping]] = ...) -> None: ...
+
+class FetchBatchResponse(_message.Message):
+    __slots__ = ("jobs",)
+    JOBS_FIELD_NUMBER: _ClassVar[int]
+    jobs: _containers.RepeatedCompositeFieldContainer[FetchBatchJob]
+    def __init__(self, jobs: _Optional[_Iterable[_Union[FetchBatchJob, _Mapping]]] = ...) -> None: ...
+
+class UsageReport(_message.Message):
+    __slots__ = ("input_tokens", "output_tokens", "cache_creation_tokens", "cache_read_tokens", "model", "provider", "cost_usd")
+    INPUT_TOKENS_FIELD_NUMBER: _ClassVar[int]
+    OUTPUT_TOKENS_FIELD_NUMBER: _ClassVar[int]
+    CACHE_CREATION_TOKENS_FIELD_NUMBER: _ClassVar[int]
+    CACHE_READ_TOKENS_FIELD_NUMBER: _ClassVar[int]
+    MODEL_FIELD_NUMBER: _ClassVar[int]
+    PROVIDER_FIELD_NUMBER: _ClassVar[int]
+    COST_USD_FIELD_NUMBER: _ClassVar[int]
+    input_tokens: int
+    output_tokens: int
+    cache_creation_tokens: int
+    cache_read_tokens: int
+    model: str
+    provider: str
+    cost_usd: float
+    def __init__(self, input_tokens: _Optional[int] = ..., output_tokens: _Optional[int] = ..., cache_creation_tokens: _Optional[int] = ..., cache_read_tokens: _Optional[int] = ..., model: _Optional[str] = ..., provider: _Optional[str] = ..., cost_usd: _Optional[float] = ...) -> None: ...
+
+class AckRequest(_message.Message):
+    __slots__ = ("job_id", "result_json", "usage", "checkpoint_json", "agent_status", "hold_reason", "trace_json")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    RESULT_JSON_FIELD_NUMBER: _ClassVar[int]
+    USAGE_FIELD_NUMBER: _ClassVar[int]
+    CHECKPOINT_JSON_FIELD_NUMBER: _ClassVar[int]
+    AGENT_STATUS_FIELD_NUMBER: _ClassVar[int]
+    HOLD_REASON_FIELD_NUMBER: _ClassVar[int]
+    TRACE_JSON_FIELD_NUMBER: _ClassVar[int]
+    job_id: str
+    result_json: str
+    usage: UsageReport
+    checkpoint_json: str
+    agent_status: str
+    hold_reason: str
+    trace_json: str
+    def __init__(self, job_id: _Optional[str] = ..., result_json: _Optional[str] = ..., usage: _Optional[_Union[UsageReport, _Mapping]] = ..., checkpoint_json: _Optional[str] = ..., agent_status: _Optional[str] = ..., hold_reason: _Optional[str] = ..., trace_json: _Optional[str] = ...) -> None: ...
+
+class AckResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class AckBatchItem(_message.Message):
+    __slots__ = ("job_id", "result_json", "usage")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    RESULT_JSON_FIELD_NUMBER: _ClassVar[int]
+    USAGE_FIELD_NUMBER: _ClassVar[int]
+    job_id: str
+    result_json: str
+    usage: UsageReport
+    def __init__(self, job_id: _Optional[str] = ..., result_json: _Optional[str] = ..., usage: _Optional[_Union[UsageReport, _Mapping]] = ...) -> None: ...
+
+class AckBatchRequest(_message.Message):
+    __slots__ = ("items",)
+    ITEMS_FIELD_NUMBER: _ClassVar[int]
+    items: _containers.RepeatedCompositeFieldContainer[AckBatchItem]
+    def __init__(self, items: _Optional[_Iterable[_Union[AckBatchItem, _Mapping]]] = ...) -> None: ...
+
+class AckBatchResponse(_message.Message):
+    __slots__ = ("acked",)
+    ACKED_FIELD_NUMBER: _ClassVar[int]
+    acked: int
+    def __init__(self, acked: _Optional[int] = ...) -> None: ...
+
+class LifecycleStreamRequest(_message.Message):
+    __slots__ = ("request_id", "queues", "worker_id", "hostname", "lease_duration", "fetch_count", "acks", "enqueues")
+    REQUEST_ID_FIELD_NUMBER: _ClassVar[int]
+    QUEUES_FIELD_NUMBER: _ClassVar[int]
+    WORKER_ID_FIELD_NUMBER: _ClassVar[int]
+    HOSTNAME_FIELD_NUMBER: _ClassVar[int]
+    LEASE_DURATION_FIELD_NUMBER: _ClassVar[int]
+    FETCH_COUNT_FIELD_NUMBER: _ClassVar[int]
+    ACKS_FIELD_NUMBER: _ClassVar[int]
+    ENQUEUES_FIELD_NUMBER: _ClassVar[int]
+    request_id: int
+    queues: _containers.RepeatedScalarFieldContainer[str]
+    worker_id: str
+    hostname: str
+    lease_duration: int
+    fetch_count: int
+    acks: _containers.RepeatedCompositeFieldContainer[AckBatchItem]
+    enqueues: _containers.RepeatedCompositeFieldContainer[LifecycleEnqueueItem]
+    def __init__(self, request_id: _Optional[int] = ..., queues: _Optional[_Iterable[str]] = ..., worker_id: _Optional[str] = ..., hostname: _Optional[str] = ..., lease_duration: _Optional[int] = ..., fetch_count: _Optional[int] = ..., acks: _Optional[_Iterable[_Union[AckBatchItem, _Mapping]]] = ..., enqueues: _Optional[_Iterable[_Union[LifecycleEnqueueItem, _Mapping]]] = ...) -> None: ...
+
+class LifecycleEnqueueItem(_message.Message):
+    __slots__ = ("queue", "payload_json", "agent")
+    QUEUE_FIELD_NUMBER: _ClassVar[int]
+    PAYLOAD_JSON_FIELD_NUMBER: _ClassVar[int]
+    AGENT_FIELD_NUMBER: _ClassVar[int]
+    queue: str
+    payload_json: str
+    agent: AgentConfig
+    def __init__(self, queue: _Optional[str] = ..., payload_json: _Optional[str] = ..., agent: _Optional[_Union[AgentConfig, _Mapping]] = ...) -> None: ...
+
+class AgentConfig(_message.Message):
+    __slots__ = ("max_iterations", "max_cost_usd", "iteration_timeout")
+    MAX_ITERATIONS_FIELD_NUMBER: _ClassVar[int]
+    MAX_COST_USD_FIELD_NUMBER: _ClassVar[int]
+    ITERATION_TIMEOUT_FIELD_NUMBER: _ClassVar[int]
+    max_iterations: int
+    max_cost_usd: float
+    iteration_timeout: str
+    def __init__(self, max_iterations: _Optional[int] = ..., max_cost_usd: _Optional[float] = ..., iteration_timeout: _Optional[str] = ...) -> None: ...
+
+class AgentState(_message.Message):
+    __slots__ = ("max_iterations", "max_cost_usd", "iteration_timeout", "iteration", "total_cost_usd")
+    MAX_ITERATIONS_FIELD_NUMBER: _ClassVar[int]
+    MAX_COST_USD_FIELD_NUMBER: _ClassVar[int]
+    ITERATION_TIMEOUT_FIELD_NUMBER: _ClassVar[int]
+    ITERATION_FIELD_NUMBER: _ClassVar[int]
+    TOTAL_COST_USD_FIELD_NUMBER: _ClassVar[int]
+    max_iterations: int
+    max_cost_usd: float
+    iteration_timeout: str
+    iteration: int
+    total_cost_usd: float
+    def __init__(self, max_iterations: _Optional[int] = ..., max_cost_usd: _Optional[float] = ..., iteration_timeout: _Optional[str] = ..., iteration: _Optional[int] = ..., total_cost_usd: _Optional[float] = ...) -> None: ...
+
+class LifecycleStreamResponse(_message.Message):
+    __slots__ = ("request_id", "jobs", "acked", "error", "enqueued_job_ids", "leader_addr")
+    REQUEST_ID_FIELD_NUMBER: _ClassVar[int]
+    JOBS_FIELD_NUMBER: _ClassVar[int]
+    ACKED_FIELD_NUMBER: _ClassVar[int]
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    ENQUEUED_JOB_IDS_FIELD_NUMBER: _ClassVar[int]
+    LEADER_ADDR_FIELD_NUMBER: _ClassVar[int]
+    request_id: int
+    jobs: _containers.RepeatedCompositeFieldContainer[FetchBatchJob]
+    acked: int
+    error: str
+    enqueued_job_ids: _containers.RepeatedScalarFieldContainer[str]
+    leader_addr: str
+    def __init__(self, request_id: _Optional[int] = ..., jobs: _Optional[_Iterable[_Union[FetchBatchJob, _Mapping]]] = ..., acked: _Optional[int] = ..., error: _Optional[str] = ..., enqueued_job_ids: _Optional[_Iterable[str]] = ..., leader_addr: _Optional[str] = ...) -> None: ...
+
+class FailRequest(_message.Message):
+    __slots__ = ("job_id", "error", "backtrace")
+    JOB_ID_FIELD_NUMBER: _ClassVar[int]
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    BACKTRACE_FIELD_NUMBER: _ClassVar[int]
+    job_id: str
+    error: str
+    backtrace: str
+    def __init__(self, job_id: _Optional[str] = ..., error: _Optional[str] = ..., backtrace: _Optional[str] = ...) -> None: ...
+
+class FailResponse(_message.Message):
+    __slots__ = ("status", "next_attempt_at", "attempts_remaining")
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    NEXT_ATTEMPT_AT_FIELD_NUMBER: _ClassVar[int]
+    ATTEMPTS_REMAINING_FIELD_NUMBER: _ClassVar[int]
+    status: str
+    next_attempt_at: _timestamp_pb2.Timestamp
+    attempts_remaining: int
+    def __init__(self, status: _Optional[str] = ..., next_attempt_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., attempts_remaining: _Optional[int] = ...) -> None: ...
+
+class HeartbeatJobUpdate(_message.Message):
+    __slots__ = ("progress_json", "checkpoint_json", "usage", "stream_delta")
+    PROGRESS_JSON_FIELD_NUMBER: _ClassVar[int]
+    CHECKPOINT_JSON_FIELD_NUMBER: _ClassVar[int]
+    USAGE_FIELD_NUMBER: _ClassVar[int]
+    STREAM_DELTA_FIELD_NUMBER: _ClassVar[int]
+    progress_json: str
+    checkpoint_json: str
+    usage: UsageReport
+    stream_delta: str
+    def __init__(self, progress_json: _Optional[str] = ..., checkpoint_json: _Optional[str] = ..., usage: _Optional[_Union[UsageReport, _Mapping]] = ..., stream_delta: _Optional[str] = ...) -> None: ...
+
+class HeartbeatRequest(_message.Message):
+    __slots__ = ("jobs",)
+    class JobsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: HeartbeatJobUpdate
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[HeartbeatJobUpdate, _Mapping]] = ...) -> None: ...
+    JOBS_FIELD_NUMBER: _ClassVar[int]
+    jobs: _containers.MessageMap[str, HeartbeatJobUpdate]
+    def __init__(self, jobs: _Optional[_Mapping[str, HeartbeatJobUpdate]] = ...) -> None: ...
+
+class HeartbeatJobResponse(_message.Message):
+    __slots__ = ("status",)
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    status: str
+    def __init__(self, status: _Optional[str] = ...) -> None: ...
+
+class HeartbeatResponse(_message.Message):
+    __slots__ = ("jobs",)
+    class JobsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: HeartbeatJobResponse
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[HeartbeatJobResponse, _Mapping]] = ...) -> None: ...
+    JOBS_FIELD_NUMBER: _ClassVar[int]
+    jobs: _containers.MessageMap[str, HeartbeatJobResponse]
+    def __init__(self, jobs: _Optional[_Mapping[str, HeartbeatJobResponse]] = ...) -> None: ...