langgraph-api 0.5.4__py3-none-any.whl → 0.7.3__py3-none-any.whl

langgraph_api/{grpc_ops → grpc}/generated/core_api_pb2.pyi

@@ -4,6 +4,13 @@ from google.protobuf import empty_pb2 as _empty_pb2
 from google.protobuf import struct_pb2 as _struct_pb2
 from google.protobuf import timestamp_pb2 as _timestamp_pb2
 from . import engine_common_pb2 as _engine_common_pb2
+from . import enum_run_status_pb2 as _enum_run_status_pb2
+from . import enum_multitask_strategy_pb2 as _enum_multitask_strategy_pb2
+from . import enum_stream_mode_pb2 as _enum_stream_mode_pb2
+from . import enum_cancel_run_action_pb2 as _enum_cancel_run_action_pb2
+from . import enum_thread_status_pb2 as _enum_thread_status_pb2
+from . import enum_thread_stream_mode_pb2 as _enum_thread_stream_mode_pb2
+from . import enum_control_signal_pb2 as _enum_control_signal_pb2
 from google.protobuf.internal import containers as _containers
 from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
@@ -32,16 +39,10 @@ class AssistantsSortBy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     CREATED_AT: _ClassVar[AssistantsSortBy]
     UPDATED_AT: _ClassVar[AssistantsSortBy]
 
-class ThreadStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    THREAD_STATUS_IDLE: _ClassVar[ThreadStatus]
-    THREAD_STATUS_BUSY: _ClassVar[ThreadStatus]
-    THREAD_STATUS_INTERRUPTED: _ClassVar[ThreadStatus]
-    THREAD_STATUS_ERROR: _ClassVar[ThreadStatus]
-
 class ThreadTTLStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
     THREAD_TTL_STRATEGY_DELETE: _ClassVar[ThreadTTLStrategy]
+    THREAD_TTL_STRATEGY_KEEP_LATEST: _ClassVar[ThreadTTLStrategy]
 
 class CheckpointSource(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
@@ -59,45 +60,11 @@ class ThreadsSortBy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     THREADS_SORT_BY_UPDATED_AT: _ClassVar[ThreadsSortBy]
     THREADS_SORT_BY_STATUS: _ClassVar[ThreadsSortBy]
 
-class RunStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    RUN_STATUS_PENDING: _ClassVar[RunStatus]
-    RUN_STATUS_RUNNING: _ClassVar[RunStatus]
-    RUN_STATUS_ERROR: _ClassVar[RunStatus]
-    RUN_STATUS_SUCCESS: _ClassVar[RunStatus]
-    RUN_STATUS_TIMEOUT: _ClassVar[RunStatus]
-    RUN_STATUS_INTERRUPTED: _ClassVar[RunStatus]
-
-class MultitaskStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    MULTITASK_STRATEGY_UNSPECIFIED: _ClassVar[MultitaskStrategy]
-    MULTITASK_STRATEGY_REJECT: _ClassVar[MultitaskStrategy]
-    MULTITASK_STRATEGY_ROLLBACK: _ClassVar[MultitaskStrategy]
-    MULTITASK_STRATEGY_INTERRUPT: _ClassVar[MultitaskStrategy]
-    MULTITASK_STRATEGY_ENQUEUE: _ClassVar[MultitaskStrategy]
-
-class StreamMode(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    STREAM_MODE_UNSPECIFIED: _ClassVar[StreamMode]
-    STREAM_MODE_VALUES: _ClassVar[StreamMode]
-    STREAM_MODE_MESSAGES: _ClassVar[StreamMode]
-    STREAM_MODE_UPDATES: _ClassVar[StreamMode]
-    STREAM_MODE_EVENTS: _ClassVar[StreamMode]
-    STREAM_MODE_DEBUG: _ClassVar[StreamMode]
-    STREAM_MODE_TASKS: _ClassVar[StreamMode]
-    STREAM_MODE_CHECKPOINTS: _ClassVar[StreamMode]
-    STREAM_MODE_CUSTOM: _ClassVar[StreamMode]
-
 class CreateRunBehavior(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
     REJECT_RUN_IF_THREAD_NOT_EXISTS: _ClassVar[CreateRunBehavior]
     CREATE_THREAD_IF_THREAD_NOT_EXISTS: _ClassVar[CreateRunBehavior]
 
-class CancelRunAction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    CANCEL_RUN_ACTION_INTERRUPT: _ClassVar[CancelRunAction]
-    CANCEL_RUN_ACTION_ROLLBACK: _ClassVar[CancelRunAction]
-
 class CancelRunStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
     CANCEL_RUN_STATUS_PENDING: _ClassVar[CancelRunStatus]
@@ -113,11 +80,8 @@ GRAPH_ID: AssistantsSortBy
 NAME: AssistantsSortBy
 CREATED_AT: AssistantsSortBy
 UPDATED_AT: AssistantsSortBy
-THREAD_STATUS_IDLE: ThreadStatus
-THREAD_STATUS_BUSY: ThreadStatus
-THREAD_STATUS_INTERRUPTED: ThreadStatus
-THREAD_STATUS_ERROR: ThreadStatus
 THREAD_TTL_STRATEGY_DELETE: ThreadTTLStrategy
+THREAD_TTL_STRATEGY_KEEP_LATEST: ThreadTTLStrategy
 CHECKPOINT_SOURCE_UNSPECIFIED: CheckpointSource
 CHECKPOINT_SOURCE_INPUT: CheckpointSource
 CHECKPOINT_SOURCE_LOOP: CheckpointSource
@@ -128,30 +92,8 @@ THREADS_SORT_BY_THREAD_ID: ThreadsSortBy
 THREADS_SORT_BY_CREATED_AT: ThreadsSortBy
 THREADS_SORT_BY_UPDATED_AT: ThreadsSortBy
 THREADS_SORT_BY_STATUS: ThreadsSortBy
-RUN_STATUS_PENDING: RunStatus
-RUN_STATUS_RUNNING: RunStatus
-RUN_STATUS_ERROR: RunStatus
-RUN_STATUS_SUCCESS: RunStatus
-RUN_STATUS_TIMEOUT: RunStatus
-RUN_STATUS_INTERRUPTED: RunStatus
-MULTITASK_STRATEGY_UNSPECIFIED: MultitaskStrategy
-MULTITASK_STRATEGY_REJECT: MultitaskStrategy
-MULTITASK_STRATEGY_ROLLBACK: MultitaskStrategy
-MULTITASK_STRATEGY_INTERRUPT: MultitaskStrategy
-MULTITASK_STRATEGY_ENQUEUE: MultitaskStrategy
-STREAM_MODE_UNSPECIFIED: StreamMode
-STREAM_MODE_VALUES: StreamMode
-STREAM_MODE_MESSAGES: StreamMode
-STREAM_MODE_UPDATES: StreamMode
-STREAM_MODE_EVENTS: StreamMode
-STREAM_MODE_DEBUG: StreamMode
-STREAM_MODE_TASKS: StreamMode
-STREAM_MODE_CHECKPOINTS: StreamMode
-STREAM_MODE_CUSTOM: StreamMode
 REJECT_RUN_IF_THREAD_NOT_EXISTS: CreateRunBehavior
 CREATE_THREAD_IF_THREAD_NOT_EXISTS: CreateRunBehavior
-CANCEL_RUN_ACTION_INTERRUPT: CancelRunAction
-CANCEL_RUN_ACTION_ROLLBACK: CancelRunAction
 CANCEL_RUN_STATUS_PENDING: CancelRunStatus
 CANCEL_RUN_STATUS_RUNNING: CancelRunStatus
 CANCEL_RUN_STATUS_ALL: CancelRunStatus
@@ -163,24 +105,44 @@ class Tags(_message.Message):
     def __init__(self, values: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class EqAuthFilter(_message.Message):
-    __slots__ = ("match",)
+    __slots__ = ("key", "match")
+    KEY_FIELD_NUMBER: _ClassVar[int]
     MATCH_FIELD_NUMBER: _ClassVar[int]
+    key: str
     match: str
-    def __init__(self, match: _Optional[str] = ...) -> None: ...
+    def __init__(self, key: _Optional[str] = ..., match: _Optional[str] = ...) -> None: ...
 
 class ContainsAuthFilter(_message.Message):
-    __slots__ = ("matches",)
+    __slots__ = ("key", "matches")
+    KEY_FIELD_NUMBER: _ClassVar[int]
     MATCHES_FIELD_NUMBER: _ClassVar[int]
-    matches: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Value]
-    def __init__(self, matches: _Optional[_Iterable[_Union[_struct_pb2.Value, _Mapping]]] = ...) -> None: ...
+    key: str
+    matches: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, key: _Optional[str] = ..., matches: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class OrAuthFilter(_message.Message):
+    __slots__ = ("filters",)
+    FILTERS_FIELD_NUMBER: _ClassVar[int]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
+
+class AndAuthFilter(_message.Message):
+    __slots__ = ("filters",)
+    FILTERS_FIELD_NUMBER: _ClassVar[int]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
 
 class AuthFilter(_message.Message):
-    __slots__ = ("eq", "contains")
+    __slots__ = ("eq", "contains", "or_filter", "and_filter")
     EQ_FIELD_NUMBER: _ClassVar[int]
     CONTAINS_FIELD_NUMBER: _ClassVar[int]
+    OR_FILTER_FIELD_NUMBER: _ClassVar[int]
+    AND_FILTER_FIELD_NUMBER: _ClassVar[int]
     eq: EqAuthFilter
     contains: ContainsAuthFilter
-    def __init__(self, eq: _Optional[_Union[EqAuthFilter, _Mapping]] = ..., contains: _Optional[_Union[ContainsAuthFilter, _Mapping]] = ...) -> None: ...
+    or_filter: OrAuthFilter
+    and_filter: AndAuthFilter
+    def __init__(self, eq: _Optional[_Union[EqAuthFilter, _Mapping]] = ..., contains: _Optional[_Union[ContainsAuthFilter, _Mapping]] = ..., or_filter: _Optional[_Union[OrAuthFilter, _Mapping]] = ..., and_filter: _Optional[_Union[AndAuthFilter, _Mapping]] = ...) -> None: ...
 
 class UUID(_message.Message):
     __slots__ = ("value",)
@@ -194,16 +156,26 @@ class CountResponse(_message.Message):
     count: int
     def __init__(self, count: _Optional[int] = ...) -> None: ...
 
+class StreamEvent(_message.Message):
+    __slots__ = ("event_type", "message", "stream_id")
+    EVENT_TYPE_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    STREAM_ID_FIELD_NUMBER: _ClassVar[int]
+    event_type: str
+    message: bytes
+    stream_id: str
+    def __init__(self, event_type: _Optional[str] = ..., message: _Optional[bytes] = ..., stream_id: _Optional[str] = ...) -> None: ...
+
 class Assistant(_message.Message):
-    __slots__ = ("assistant_id", "graph_id", "version", "created_at", "updated_at", "config", "context", "metadata", "name", "description")
+    __slots__ = ("assistant_id", "graph_id", "version", "created_at", "updated_at", "config", "context_json", "metadata_json", "name", "description")
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
     GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
     CREATED_AT_FIELD_NUMBER: _ClassVar[int]
     UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
     CONFIG_FIELD_NUMBER: _ClassVar[int]
-    CONTEXT_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    CONTEXT_JSON_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
     assistant_id: str
@@ -212,21 +184,21 @@ class Assistant(_message.Message):
     created_at: _timestamp_pb2.Timestamp
     updated_at: _timestamp_pb2.Timestamp
     config: _engine_common_pb2.EngineRunnableConfig
-    context: _struct_pb2.Struct
-    metadata: _struct_pb2.Struct
+    context_json: bytes
+    metadata_json: bytes
     name: str
     description: str
-    def __init__(self, assistant_id: _Optional[str] = ..., graph_id: _Optional[str] = ..., version: _Optional[int] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., name: _Optional[str] = ..., description: _Optional[str] = ...) -> None: ...
+    def __init__(self, assistant_id: _Optional[str] = ..., graph_id: _Optional[str] = ..., version: _Optional[int] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context_json: _Optional[bytes] = ..., metadata_json: _Optional[bytes] = ..., name: _Optional[str] = ..., description: _Optional[str] = ...) -> None: ...
 
 class AssistantVersion(_message.Message):
-    __slots__ = ("assistant_id", "graph_id", "version", "created_at", "config", "context", "metadata", "name", "description")
+    __slots__ = ("assistant_id", "graph_id", "version", "created_at", "config", "context_json", "metadata_json", "name", "description")
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
     GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
     CREATED_AT_FIELD_NUMBER: _ClassVar[int]
     CONFIG_FIELD_NUMBER: _ClassVar[int]
-    CONTEXT_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    CONTEXT_JSON_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
     assistant_id: str
@@ -234,97 +206,71 @@ class AssistantVersion(_message.Message):
     version: int
     created_at: _timestamp_pb2.Timestamp
     config: _engine_common_pb2.EngineRunnableConfig
-    context: _struct_pb2.Struct
-    metadata: _struct_pb2.Struct
+    context_json: bytes
+    metadata_json: bytes
     name: str
     description: str
-    def __init__(self, assistant_id: _Optional[str] = ..., graph_id: _Optional[str] = ..., version: _Optional[int] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., name: _Optional[str] = ..., description: _Optional[str] = ...) -> None: ...
+    def __init__(self, assistant_id: _Optional[str] = ..., graph_id: _Optional[str] = ..., version: _Optional[int] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context_json: _Optional[bytes] = ..., metadata_json: _Optional[bytes] = ..., name: _Optional[str] = ..., description: _Optional[str] = ...) -> None: ...
 
 class CreateAssistantRequest(_message.Message):
-    __slots__ = ("assistant_id", "graph_id", "filters", "if_exists", "config", "context", "name", "description", "metadata")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("assistant_id", "graph_id", "filters", "if_exists", "config", "context_json", "name", "description", "metadata_json")
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
     GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     IF_EXISTS_FIELD_NUMBER: _ClassVar[int]
     CONFIG_FIELD_NUMBER: _ClassVar[int]
-    CONTEXT_FIELD_NUMBER: _ClassVar[int]
+    CONTEXT_JSON_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     assistant_id: str
     graph_id: str
-    filters: _containers.MessageMap[str, AuthFilter]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
     if_exists: OnConflictBehavior
     config: _engine_common_pb2.EngineRunnableConfig
-    context: _struct_pb2.Struct
+    context_json: bytes
     name: str
     description: str
-    metadata: _struct_pb2.Struct
-    def __init__(self, assistant_id: _Optional[str] = ..., graph_id: _Optional[str] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ..., if_exists: _Optional[_Union[OnConflictBehavior, str]] = ..., config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    metadata_json: bytes
+    def __init__(self, assistant_id: _Optional[str] = ..., graph_id: _Optional[str] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., if_exists: _Optional[_Union[OnConflictBehavior, str]] = ..., config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context_json: _Optional[bytes] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., metadata_json: _Optional[bytes] = ...) -> None: ...
 
 class GetAssistantRequest(_message.Message):
     __slots__ = ("assistant_id", "filters")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     assistant_id: str
-    filters: _containers.MessageMap[str, AuthFilter]
-    def __init__(self, assistant_id: _Optional[str] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, assistant_id: _Optional[str] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
 
 class PatchAssistantRequest(_message.Message):
-    __slots__ = ("assistant_id", "filters", "graph_id", "config", "context", "name", "description", "metadata")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("assistant_id", "filters", "graph_id", "config", "context_json", "name", "description", "metadata_json")
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
     CONFIG_FIELD_NUMBER: _ClassVar[int]
-    CONTEXT_FIELD_NUMBER: _ClassVar[int]
+    CONTEXT_JSON_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     assistant_id: str
-    filters: _containers.MessageMap[str, AuthFilter]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
     graph_id: str
     config: _engine_common_pb2.EngineRunnableConfig
-    context: _struct_pb2.Struct
+    context_json: bytes
     name: str
     description: str
-    metadata: _struct_pb2.Struct
-    def __init__(self, assistant_id: _Optional[str] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ..., graph_id: _Optional[str] = ..., config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    metadata_json: bytes
+    def __init__(self, assistant_id: _Optional[str] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., graph_id: _Optional[str] = ..., config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context_json: _Optional[bytes] = ..., name: _Optional[str] = ..., description: _Optional[str] = ..., metadata_json: _Optional[bytes] = ...) -> None: ...
 
 class DeleteAssistantRequest(_message.Message):
-    __slots__ = ("assistant_id", "filters")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("assistant_id", "filters", "delete_threads")
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
+    DELETE_THREADS_FIELD_NUMBER: _ClassVar[int]
     assistant_id: str
-    filters: _containers.MessageMap[str, AuthFilter]
-    def __init__(self, assistant_id: _Optional[str] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    delete_threads: bool
+    def __init__(self, assistant_id: _Optional[str] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., delete_threads: bool = ...) -> None: ...
 
 class DeleteAssistantsResponse(_message.Message):
     __slots__ = ("assistant_ids",)
@@ -334,47 +280,35 @@ class DeleteAssistantsResponse(_message.Message):
 
 class SetLatestAssistantRequest(_message.Message):
     __slots__ = ("assistant_id", "version", "filters")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
     VERSION_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     assistant_id: str
     version: int
-    filters: _containers.MessageMap[str, AuthFilter]
-    def __init__(self, assistant_id: _Optional[str] = ..., version: _Optional[int] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, assistant_id: _Optional[str] = ..., version: _Optional[int] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
 
 class SearchAssistantsRequest(_message.Message):
-    __slots__ = ("filters", "graph_id", "metadata", "limit", "offset", "sort_by", "sort_order", "select")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("filters", "graph_id", "metadata_json", "limit", "offset", "sort_by", "sort_order", "select", "name")
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     LIMIT_FIELD_NUMBER: _ClassVar[int]
     OFFSET_FIELD_NUMBER: _ClassVar[int]
     SORT_BY_FIELD_NUMBER: _ClassVar[int]
     SORT_ORDER_FIELD_NUMBER: _ClassVar[int]
     SELECT_FIELD_NUMBER: _ClassVar[int]
-    filters: _containers.MessageMap[str, AuthFilter]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
     graph_id: str
-    metadata: _struct_pb2.Struct
+    metadata_json: bytes
     limit: int
     offset: int
     sort_by: AssistantsSortBy
     sort_order: SortOrder
     select: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(self, filters: _Optional[_Mapping[str, AuthFilter]] = ..., graph_id: _Optional[str] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., sort_by: _Optional[_Union[AssistantsSortBy, str]] = ..., sort_order: _Optional[_Union[SortOrder, str]] = ..., select: _Optional[_Iterable[str]] = ...) -> None: ...
+    name: str
+    def __init__(self, filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., graph_id: _Optional[str] = ..., metadata_json: _Optional[bytes] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., sort_by: _Optional[_Union[AssistantsSortBy, str]] = ..., sort_order: _Optional[_Union[SortOrder, str]] = ..., select: _Optional[_Iterable[str]] = ..., name: _Optional[str] = ...) -> None: ...
 
 class SearchAssistantsResponse(_message.Message):
     __slots__ = ("assistants",)
@@ -383,25 +317,18 @@ class SearchAssistantsResponse(_message.Message):
     def __init__(self, assistants: _Optional[_Iterable[_Union[Assistant, _Mapping]]] = ...) -> None: ...
 
 class GetAssistantVersionsRequest(_message.Message):
-    __slots__ = ("assistant_id", "filters", "metadata", "limit", "offset")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("assistant_id", "filters", "metadata_json", "limit", "offset")
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     LIMIT_FIELD_NUMBER: _ClassVar[int]
     OFFSET_FIELD_NUMBER: _ClassVar[int]
     assistant_id: str
-    filters: _containers.MessageMap[str, AuthFilter]
-    metadata: _struct_pb2.Struct
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    metadata_json: bytes
     limit: int
     offset: int
-    def __init__(self, assistant_id: _Optional[str] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ...) -> None: ...
+    def __init__(self, assistant_id: _Optional[str] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., metadata_json: _Optional[bytes] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ...) -> None: ...
 
 class GetAssistantVersionsResponse(_message.Message):
     __slots__ = ("versions",)
@@ -410,21 +337,16 @@ class GetAssistantVersionsResponse(_message.Message):
     def __init__(self, versions: _Optional[_Iterable[_Union[AssistantVersion, _Mapping]]] = ...) -> None: ...
 
 class CountAssistantsRequest(_message.Message):
-    __slots__ = ("filters", "graph_id", "metadata")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("filters", "graph_id", "metadata_json", "name")
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
-    filters: _containers.MessageMap[str, AuthFilter]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
     graph_id: str
-    metadata: _struct_pb2.Struct
-    def __init__(self, filters: _Optional[_Mapping[str, AuthFilter]] = ..., graph_id: _Optional[str] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    metadata_json: bytes
+    name: str
+    def __init__(self, filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., graph_id: _Optional[str] = ..., metadata_json: _Optional[bytes] = ..., name: _Optional[str] = ...) -> None: ...
 
 class TruncateRequest(_message.Message):
     __slots__ = ("runs", "threads", "assistants", "checkpointer", "store")
@@ -457,18 +379,18 @@ class Fragment(_message.Message):
     def __init__(self, value: _Optional[bytes] = ...) -> None: ...
 
 class CheckpointTask(_message.Message):
-    __slots__ = ("id", "name", "error", "interrupts", "state")
+    __slots__ = ("id", "name", "error", "interrupts_json", "state_json")
     ID_FIELD_NUMBER: _ClassVar[int]
     NAME_FIELD_NUMBER: _ClassVar[int]
     ERROR_FIELD_NUMBER: _ClassVar[int]
-    INTERRUPTS_FIELD_NUMBER: _ClassVar[int]
-    STATE_FIELD_NUMBER: _ClassVar[int]
+    INTERRUPTS_JSON_FIELD_NUMBER: _ClassVar[int]
+    STATE_JSON_FIELD_NUMBER: _ClassVar[int]
     id: str
     name: str
     error: str
-    interrupts: _containers.RepeatedCompositeFieldContainer[_struct_pb2.Struct]
-    state: _struct_pb2.Struct
-    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., error: _Optional[str] = ..., interrupts: _Optional[_Iterable[_Union[_struct_pb2.Struct, _Mapping]]] = ..., state: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    interrupts_json: _containers.RepeatedScalarFieldContainer[bytes]
+    state_json: bytes
+    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., error: _Optional[str] = ..., interrupts_json: _Optional[_Iterable[bytes]] = ..., state_json: _Optional[bytes] = ...) -> None: ...
 
 class CheckpointMetadata(_message.Message):
     __slots__ = ("source", "step", "parents")
@@ -488,20 +410,30 @@ class CheckpointMetadata(_message.Message):
     def __init__(self, source: _Optional[_Union[CheckpointSource, str]] = ..., step: _Optional[int] = ..., parents: _Optional[_Mapping[str, str]] = ...) -> None: ...
 
 class CheckpointPayload(_message.Message):
-    __slots__ = ("config", "metadata", "values", "next", "parent_config", "tasks")
+    __slots__ = ("config", "metadata", "values_json", "next", "parent_config", "tasks")
     CONFIG_FIELD_NUMBER: _ClassVar[int]
     METADATA_FIELD_NUMBER: _ClassVar[int]
-    VALUES_FIELD_NUMBER: _ClassVar[int]
+    VALUES_JSON_FIELD_NUMBER: _ClassVar[int]
     NEXT_FIELD_NUMBER: _ClassVar[int]
     PARENT_CONFIG_FIELD_NUMBER: _ClassVar[int]
     TASKS_FIELD_NUMBER: _ClassVar[int]
     config: _engine_common_pb2.EngineRunnableConfig
     metadata: CheckpointMetadata
-    values: _struct_pb2.Struct
+    values_json: bytes
     next: _containers.RepeatedScalarFieldContainer[str]
     parent_config: _engine_common_pb2.EngineRunnableConfig
     tasks: _containers.RepeatedCompositeFieldContainer[CheckpointTask]
-    def __init__(self, config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., metadata: _Optional[_Union[CheckpointMetadata, _Mapping]] = ..., values: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., next: _Optional[_Iterable[str]] = ..., parent_config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., tasks: _Optional[_Iterable[_Union[CheckpointTask, _Mapping]]] = ...) -> None: ...
+    def __init__(self, config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., metadata: _Optional[_Union[CheckpointMetadata, _Mapping]] = ..., values_json: _Optional[bytes] = ..., next: _Optional[_Iterable[str]] = ..., parent_config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., tasks: _Optional[_Iterable[_Union[CheckpointTask, _Mapping]]] = ...) -> None: ...
+
+class ThreadStatusCheckpoint(_message.Message):
+    __slots__ = ("values_json", "next", "interrupts_json")
+    VALUES_JSON_FIELD_NUMBER: _ClassVar[int]
+    NEXT_FIELD_NUMBER: _ClassVar[int]
+    INTERRUPTS_JSON_FIELD_NUMBER: _ClassVar[int]
+    values_json: bytes
+    next: _containers.RepeatedScalarFieldContainer[str]
+    interrupts_json: bytes
+    def __init__(self, values_json: _Optional[bytes] = ..., next: _Optional[_Iterable[str]] = ..., interrupts_json: _Optional[bytes] = ...) -> None: ...
 
 class Interrupt(_message.Message):
     __slots__ = ("id", "value", "when", "resumable", "ns")
@@ -546,125 +478,83 @@ class Thread(_message.Message):
     updated_at: _timestamp_pb2.Timestamp
     metadata: Fragment
     config: Fragment
-    status: ThreadStatus
+    status: _enum_thread_status_pb2.ThreadStatus
     values: Fragment
     interrupts: _containers.MessageMap[str, Interrupts]
     error: Fragment
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., metadata: _Optional[_Union[Fragment, _Mapping]] = ..., config: _Optional[_Union[Fragment, _Mapping]] = ..., status: _Optional[_Union[ThreadStatus, str]] = ..., values: _Optional[_Union[Fragment, _Mapping]] = ..., interrupts: _Optional[_Mapping[str, Interrupts]] = ..., error: _Optional[_Union[Fragment, _Mapping]] = ...) -> None: ...
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., metadata: _Optional[_Union[Fragment, _Mapping]] = ..., config: _Optional[_Union[Fragment, _Mapping]] = ..., status: _Optional[_Union[_enum_thread_status_pb2.ThreadStatus, str]] = ..., values: _Optional[_Union[Fragment, _Mapping]] = ..., interrupts: _Optional[_Mapping[str, Interrupts]] = ..., error: _Optional[_Union[Fragment, _Mapping]] = ...) -> None: ...
 
 class CreateThreadRequest(_message.Message):
-    __slots__ = ("thread_id", "filters", "if_exists", "metadata", "ttl")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("thread_id", "filters", "if_exists", "metadata_json", "ttl")
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     IF_EXISTS_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     TTL_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
-    filters: _containers.MessageMap[str, AuthFilter]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
     if_exists: OnConflictBehavior
-    metadata: _struct_pb2.Struct
+    metadata_json: bytes
     ttl: ThreadTTLConfig
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ..., if_exists: _Optional[_Union[OnConflictBehavior, str]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., ttl: _Optional[_Union[ThreadTTLConfig, _Mapping]] = ...) -> None: ...
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., if_exists: _Optional[_Union[OnConflictBehavior, str]] = ..., metadata_json: _Optional[bytes] = ..., ttl: _Optional[_Union[ThreadTTLConfig, _Mapping]] = ...) -> None: ...
 
 class GetThreadRequest(_message.Message):
     __slots__ = ("thread_id", "filters")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
-    filters: _containers.MessageMap[str, AuthFilter]
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
 
 class PatchThreadRequest(_message.Message):
-    __slots__ = ("thread_id", "filters", "metadata", "ttl")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("thread_id", "filters", "metadata_json", "ttl")
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     TTL_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
-    filters: _containers.MessageMap[str, AuthFilter]
-    metadata: _struct_pb2.Struct
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    metadata_json: bytes
     ttl: ThreadTTLConfig
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., ttl: _Optional[_Union[ThreadTTLConfig, _Mapping]] = ...) -> None: ...
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., metadata_json: _Optional[bytes] = ..., ttl: _Optional[_Union[ThreadTTLConfig, _Mapping]] = ...) -> None: ...
 
 class DeleteThreadRequest(_message.Message):
     __slots__ = ("thread_id", "filters")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
-    filters: _containers.MessageMap[str, AuthFilter]
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
 
 class CopyThreadRequest(_message.Message):
     __slots__ = ("thread_id", "filters")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
-    filters: _containers.MessageMap[str, AuthFilter]
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
 
 class SearchThreadsRequest(_message.Message):
-    __slots__ = ("filters", "metadata", "values", "status", "limit", "offset", "sort_by", "sort_order", "select")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("filters", "metadata_json", "values_json", "status", "limit", "offset", "sort_by", "sort_order", "select")
     FILTERS_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
-    VALUES_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
+    VALUES_JSON_FIELD_NUMBER: _ClassVar[int]
     STATUS_FIELD_NUMBER: _ClassVar[int]
     LIMIT_FIELD_NUMBER: _ClassVar[int]
     OFFSET_FIELD_NUMBER: _ClassVar[int]
     SORT_BY_FIELD_NUMBER: _ClassVar[int]
     SORT_ORDER_FIELD_NUMBER: _ClassVar[int]
     SELECT_FIELD_NUMBER: _ClassVar[int]
-    filters: _containers.MessageMap[str, AuthFilter]
-    metadata: _struct_pb2.Struct
-    values: _struct_pb2.Struct
-    status: ThreadStatus
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    metadata_json: bytes
+    values_json: bytes
+    status: _enum_thread_status_pb2.ThreadStatus
     limit: int
     offset: int
     sort_by: ThreadsSortBy
     sort_order: SortOrder
     select: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(self, filters: _Optional[_Mapping[str, AuthFilter]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., values: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., status: _Optional[_Union[ThreadStatus, str]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., sort_by: _Optional[_Union[ThreadsSortBy, str]] = ..., sort_order: _Optional[_Union[SortOrder, str]] = ..., select: _Optional[_Iterable[str]] = ...) -> None: ...
+    def __init__(self, filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., metadata_json: _Optional[bytes] = ..., values_json: _Optional[bytes] = ..., status: _Optional[_Union[_enum_thread_status_pb2.ThreadStatus, str]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., sort_by: _Optional[_Union[ThreadsSortBy, str]] = ..., sort_order: _Optional[_Union[SortOrder, str]] = ..., select: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class SearchThreadsResponse(_message.Message):
     __slots__ = ("threads",)
@@ -673,23 +563,16 @@ class SearchThreadsResponse(_message.Message):
     def __init__(self, threads: _Optional[_Iterable[_Union[Thread, _Mapping]]] = ...) -> None: ...
 
 class CountThreadsRequest(_message.Message):
-    __slots__ = ("filters", "metadata", "values", "status")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("filters", "metadata_json", "values_json", "status")
     FILTERS_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
-    VALUES_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
+    VALUES_JSON_FIELD_NUMBER: _ClassVar[int]
     STATUS_FIELD_NUMBER: _ClassVar[int]
-    filters: _containers.MessageMap[str, AuthFilter]
-    metadata: _struct_pb2.Struct
-    values: _struct_pb2.Struct
-    status: ThreadStatus
-    def __init__(self, filters: _Optional[_Mapping[str, AuthFilter]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., values: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., status: _Optional[_Union[ThreadStatus, str]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    metadata_json: bytes
+    values_json: bytes
+    status: _enum_thread_status_pb2.ThreadStatus
+    def __init__(self, filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., metadata_json: _Optional[bytes] = ..., values_json: _Optional[bytes] = ..., status: _Optional[_Union[_enum_thread_status_pb2.ThreadStatus, str]] = ...) -> None: ...
 
 class SweepThreadsTTLResponse(_message.Message):
     __slots__ = ("expired", "deleted")
@@ -708,55 +591,51 @@ class SweepThreadsTTLRequest(_message.Message):
     def __init__(self, batch_size: _Optional[int] = ..., limit: _Optional[int] = ...) -> None: ...
 
 class SetThreadStatusRequest(_message.Message):
-    __slots__ = ("thread_id", "status", "checkpoint", "exception", "expected_status")
+    __slots__ = ("thread_id", "checkpoint", "exception_json", "expected_status")
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
-    STATUS_FIELD_NUMBER: _ClassVar[int]
     CHECKPOINT_FIELD_NUMBER: _ClassVar[int]
-    EXCEPTION_FIELD_NUMBER: _ClassVar[int]
+    EXCEPTION_JSON_FIELD_NUMBER: _ClassVar[int]
     EXPECTED_STATUS_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
-    status: ThreadStatus
-    checkpoint: CheckpointPayload
-    exception: _struct_pb2.Struct
-    expected_status: _containers.RepeatedScalarFieldContainer[ThreadStatus]
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., status: _Optional[_Union[ThreadStatus, str]] = ..., checkpoint: _Optional[_Union[CheckpointPayload, _Mapping]] = ..., exception: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., expected_status: _Optional[_Iterable[_Union[ThreadStatus, str]]] = ...) -> None: ...
+    checkpoint: ThreadStatusCheckpoint
+    exception_json: bytes
+    expected_status: _containers.RepeatedScalarFieldContainer[_enum_thread_status_pb2.ThreadStatus]
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., checkpoint: _Optional[_Union[ThreadStatusCheckpoint, _Mapping]] = ..., exception_json: _Optional[bytes] = ..., expected_status: _Optional[_Iterable[_Union[_enum_thread_status_pb2.ThreadStatus, str]]] = ...) -> None: ...
 
 class SetThreadJointStatusRequest(_message.Message):
-    __slots__ = ("thread_id", "run_id", "run_status", "graph_id", "checkpoint", "exception")
+    __slots__ = ("thread_id", "run_id", "run_status", "graph_id", "checkpoint", "exception_json")
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     RUN_ID_FIELD_NUMBER: _ClassVar[int]
     RUN_STATUS_FIELD_NUMBER: _ClassVar[int]
     GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
     CHECKPOINT_FIELD_NUMBER: _ClassVar[int]
-    EXCEPTION_FIELD_NUMBER: _ClassVar[int]
+    EXCEPTION_JSON_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
     run_id: UUID
     run_status: str
     graph_id: str
     checkpoint: CheckpointPayload
-    exception: _struct_pb2.Struct
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., run_id: _Optional[_Union[UUID, _Mapping]] = ..., run_status: _Optional[str] = ..., graph_id: _Optional[str] = ..., checkpoint: _Optional[_Union[CheckpointPayload, _Mapping]] = ..., exception: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    exception_json: bytes
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., run_id: _Optional[_Union[UUID, _Mapping]] = ..., run_status: _Optional[str] = ..., graph_id: _Optional[str] = ..., checkpoint: _Optional[_Union[CheckpointPayload, _Mapping]] = ..., exception_json: _Optional[bytes] = ...) -> None: ...
 
-class JointRollbackRequest(_message.Message):
-    __slots__ = ("thread_id", "run_id", "graph_id", "checkpoint", "exception")
+class StreamThreadRequest(_message.Message):
+    __slots__ = ("thread_id", "filters", "last_event_id", "stream_modes")
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
-    RUN_ID_FIELD_NUMBER: _ClassVar[int]
-    GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
-    CHECKPOINT_FIELD_NUMBER: _ClassVar[int]
-    EXCEPTION_FIELD_NUMBER: _ClassVar[int]
+    FILTERS_FIELD_NUMBER: _ClassVar[int]
+    LAST_EVENT_ID_FIELD_NUMBER: _ClassVar[int]
+    STREAM_MODES_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
-    run_id: UUID
-    graph_id: str
-    checkpoint: CheckpointPayload
-    exception: _struct_pb2.Struct
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., run_id: _Optional[_Union[UUID, _Mapping]] = ..., graph_id: _Optional[str] = ..., checkpoint: _Optional[_Union[CheckpointPayload, _Mapping]] = ..., exception: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    last_event_id: str
+    stream_modes: _containers.RepeatedScalarFieldContainer[_enum_thread_stream_mode_pb2.ThreadStreamMode]
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., last_event_id: _Optional[str] = ..., stream_modes: _Optional[_Iterable[_Union[_enum_thread_stream_mode_pb2.ThreadStreamMode, str]]] = ...) -> None: ...
 
 class RunKwargs(_message.Message):
-    __slots__ = ("config", "context", "input", "command", "stream_mode", "interrupt_before", "interrupt_after", "webhook", "feedback_keys", "temporary", "subgraphs", "resumable", "checkpoint_during", "durability")
+    __slots__ = ("config", "context_json", "input_json", "command_json", "stream_mode", "interrupt_before", "interrupt_after", "webhook", "feedback_keys", "temporary", "subgraphs", "resumable", "checkpoint_during", "durability")
     CONFIG_FIELD_NUMBER: _ClassVar[int]
-    CONTEXT_FIELD_NUMBER: _ClassVar[int]
-    INPUT_FIELD_NUMBER: _ClassVar[int]
-    COMMAND_FIELD_NUMBER: _ClassVar[int]
+    CONTEXT_JSON_FIELD_NUMBER: _ClassVar[int]
+    INPUT_JSON_FIELD_NUMBER: _ClassVar[int]
+    COMMAND_JSON_FIELD_NUMBER: _ClassVar[int]
     STREAM_MODE_FIELD_NUMBER: _ClassVar[int]
     INTERRUPT_BEFORE_FIELD_NUMBER: _ClassVar[int]
     INTERRUPT_AFTER_FIELD_NUMBER: _ClassVar[int]
@@ -768,12 +647,12 @@ class RunKwargs(_message.Message):
     CHECKPOINT_DURING_FIELD_NUMBER: _ClassVar[int]
     DURABILITY_FIELD_NUMBER: _ClassVar[int]
     config: _engine_common_pb2.EngineRunnableConfig
-    context: _struct_pb2.Struct
-    input: _struct_pb2.Struct
-    command: _struct_pb2.Struct
-    stream_mode: StreamMode
-    interrupt_before: _containers.RepeatedScalarFieldContainer[str]
-    interrupt_after: _containers.RepeatedScalarFieldContainer[str]
+    context_json: bytes
+    input_json: bytes
+    command_json: bytes
+    stream_mode: _enum_stream_mode_pb2.StreamMode
+    interrupt_before: _engine_common_pb2.StaticInterruptConfig
+    interrupt_after: _engine_common_pb2.StaticInterruptConfig
     webhook: str
     feedback_keys: _containers.RepeatedScalarFieldContainer[str]
     temporary: bool
@@ -781,7 +660,7 @@ class RunKwargs(_message.Message):
     resumable: bool
     checkpoint_during: bool
     durability: str
-    def __init__(self, config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., input: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., command: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., stream_mode: _Optional[_Union[StreamMode, str]] = ..., interrupt_before: _Optional[_Iterable[str]] = ..., interrupt_after: _Optional[_Iterable[str]] = ..., webhook: _Optional[str] = ..., feedback_keys: _Optional[_Iterable[str]] = ..., temporary: bool = ..., subgraphs: bool = ..., resumable: bool = ..., checkpoint_during: bool = ..., durability: _Optional[str] = ...) -> None: ...
+    def __init__(self, config: _Optional[_Union[_engine_common_pb2.EngineRunnableConfig, _Mapping]] = ..., context_json: _Optional[bytes] = ..., input_json: _Optional[bytes] = ..., command_json: _Optional[bytes] = ..., stream_mode: _Optional[_Union[_enum_stream_mode_pb2.StreamMode, str]] = ..., interrupt_before: _Optional[_Union[_engine_common_pb2.StaticInterruptConfig, _Mapping]] = ..., interrupt_after: _Optional[_Union[_engine_common_pb2.StaticInterruptConfig, _Mapping]] = ..., webhook: _Optional[str] = ..., feedback_keys: _Optional[_Iterable[str]] = ..., temporary: bool = ..., subgraphs: bool = ..., resumable: bool = ..., checkpoint_during: bool = ..., durability: _Optional[str] = ...) -> None: ...
 
 class Run(_message.Message):
     __slots__ = ("run_id", "thread_id", "assistant_id", "created_at", "updated_at", "status", "metadata", "kwargs", "multitask_strategy")
@@ -799,23 +678,25 @@ class Run(_message.Message):
     assistant_id: UUID
     created_at: _timestamp_pb2.Timestamp
     updated_at: _timestamp_pb2.Timestamp
-    status: RunStatus
+    status: _enum_run_status_pb2.RunStatus
     metadata: Fragment
     kwargs: RunKwargs
-    multitask_strategy: MultitaskStrategy
-    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., assistant_id: _Optional[_Union[UUID, _Mapping]] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., status: _Optional[_Union[RunStatus, str]] = ..., metadata: _Optional[_Union[Fragment, _Mapping]] = ..., kwargs: _Optional[_Union[RunKwargs, _Mapping]] = ..., multitask_strategy: _Optional[_Union[MultitaskStrategy, str]] = ...) -> None: ...
+    multitask_strategy: _enum_multitask_strategy_pb2.MultitaskStrategy
+    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., assistant_id: _Optional[_Union[UUID, _Mapping]] = ..., created_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., updated_at: _Optional[_Union[datetime.datetime, _timestamp_pb2.Timestamp, _Mapping]] = ..., status: _Optional[_Union[_enum_run_status_pb2.RunStatus, str]] = ..., metadata: _Optional[_Union[Fragment, _Mapping]] = ..., kwargs: _Optional[_Union[RunKwargs, _Mapping]] = ..., multitask_strategy: _Optional[_Union[_enum_multitask_strategy_pb2.MultitaskStrategy, str]] = ...) -> None: ...
 
-class QueueStats(_message.Message):
-    __slots__ = ("n_pending", "n_running", "max_age_secs", "med_age_secs")
+class RunStats(_message.Message):
+    __slots__ = ("n_pending", "n_running", "pending_runs_wait_time_max_secs", "pending_runs_wait_time_med_secs", "pending_unblocked_runs_wait_time_max_secs")
     N_PENDING_FIELD_NUMBER: _ClassVar[int]
     N_RUNNING_FIELD_NUMBER: _ClassVar[int]
-    MAX_AGE_SECS_FIELD_NUMBER: _ClassVar[int]
-    MED_AGE_SECS_FIELD_NUMBER: _ClassVar[int]
+    PENDING_RUNS_WAIT_TIME_MAX_SECS_FIELD_NUMBER: _ClassVar[int]
+    PENDING_RUNS_WAIT_TIME_MED_SECS_FIELD_NUMBER: _ClassVar[int]
+    PENDING_UNBLOCKED_RUNS_WAIT_TIME_MAX_SECS_FIELD_NUMBER: _ClassVar[int]
     n_pending: int
     n_running: int
-    max_age_secs: float
-    med_age_secs: float
-    def __init__(self, n_pending: _Optional[int] = ..., n_running: _Optional[int] = ..., max_age_secs: _Optional[float] = ..., med_age_secs: _Optional[float] = ...) -> None: ...
+    pending_runs_wait_time_max_secs: float
+    pending_runs_wait_time_med_secs: float
+    pending_unblocked_runs_wait_time_max_secs: float
+    def __init__(self, n_pending: _Optional[int] = ..., n_running: _Optional[int] = ..., pending_runs_wait_time_max_secs: _Optional[float] = ..., pending_runs_wait_time_med_secs: _Optional[float] = ..., pending_unblocked_runs_wait_time_max_secs: _Optional[float] = ...) -> None: ...
 
 class NextRunRequest(_message.Message):
     __slots__ = ("wait", "limit")
@@ -840,73 +721,56 @@ class NextRunResponse(_message.Message):
     def __init__(self, runs: _Optional[_Iterable[_Union[RunWithAttempt, _Mapping]]] = ...) -> None: ...
 
 class CreateRunRequest(_message.Message):
-    __slots__ = ("assistant_id", "kwargs", "filters", "thread_id", "user_id", "run_id", "status", "metadata", "prevent_insert_if_inflight", "multitask_strategy", "if_not_exists", "after_seconds")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
+    __slots__ = ("assistant_id", "kwargs_json", "thread_filters", "assistant_filters", "thread_id", "user_id", "run_id", "status", "metadata_json", "prevent_insert_if_inflight", "multitask_strategy", "if_not_exists", "after_seconds", "thread_ttl")
     ASSISTANT_ID_FIELD_NUMBER: _ClassVar[int]
-    KWARGS_FIELD_NUMBER: _ClassVar[int]
-    FILTERS_FIELD_NUMBER: _ClassVar[int]
+    KWARGS_JSON_FIELD_NUMBER: _ClassVar[int]
+    THREAD_FILTERS_FIELD_NUMBER: _ClassVar[int]
+    ASSISTANT_FILTERS_FIELD_NUMBER: _ClassVar[int]
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     USER_ID_FIELD_NUMBER: _ClassVar[int]
     RUN_ID_FIELD_NUMBER: _ClassVar[int]
     STATUS_FIELD_NUMBER: _ClassVar[int]
-    METADATA_FIELD_NUMBER: _ClassVar[int]
+    METADATA_JSON_FIELD_NUMBER: _ClassVar[int]
     PREVENT_INSERT_IF_INFLIGHT_FIELD_NUMBER: _ClassVar[int]
     MULTITASK_STRATEGY_FIELD_NUMBER: _ClassVar[int]
     IF_NOT_EXISTS_FIELD_NUMBER: _ClassVar[int]
     AFTER_SECONDS_FIELD_NUMBER: _ClassVar[int]
+    THREAD_TTL_FIELD_NUMBER: _ClassVar[int]
     assistant_id: UUID
-    kwargs: _struct_pb2.Struct
-    filters: _containers.MessageMap[str, AuthFilter]
+    kwargs_json: bytes
+    thread_filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    assistant_filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
     thread_id: UUID
     user_id: str
     run_id: UUID
-    status: RunStatus
-    metadata: _struct_pb2.Struct
+    status: _enum_run_status_pb2.RunStatus
+    metadata_json: bytes
     prevent_insert_if_inflight: bool
-    multitask_strategy: MultitaskStrategy
+    multitask_strategy: _enum_multitask_strategy_pb2.MultitaskStrategy
     if_not_exists: CreateRunBehavior
     after_seconds: int
-    def __init__(self, assistant_id: _Optional[_Union[UUID, _Mapping]] = ..., kwargs: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., user_id: _Optional[str] = ..., run_id: _Optional[_Union[UUID, _Mapping]] = ..., status: _Optional[_Union[RunStatus, str]] = ..., metadata: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ..., prevent_insert_if_inflight: bool = ..., multitask_strategy: _Optional[_Union[MultitaskStrategy, str]] = ..., if_not_exists: _Optional[_Union[CreateRunBehavior, str]] = ..., after_seconds: _Optional[int] = ...) -> None: ...
+    thread_ttl: ThreadTTLConfig
+    def __init__(self, assistant_id: _Optional[_Union[UUID, _Mapping]] = ..., kwargs_json: _Optional[bytes] = ..., thread_filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., assistant_filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., user_id: _Optional[str] = ..., run_id: _Optional[_Union[UUID, _Mapping]] = ..., status: _Optional[_Union[_enum_run_status_pb2.RunStatus, str]] = ..., metadata_json: _Optional[bytes] = ..., prevent_insert_if_inflight: bool = ..., multitask_strategy: _Optional[_Union[_enum_multitask_strategy_pb2.MultitaskStrategy, str]] = ..., if_not_exists: _Optional[_Union[CreateRunBehavior, str]] = ..., after_seconds: _Optional[int] = ..., thread_ttl: _Optional[_Union[ThreadTTLConfig, _Mapping]] = ...) -> None: ...
 
 class GetRunRequest(_message.Message):
     __slots__ = ("run_id", "thread_id", "filters")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     RUN_ID_FIELD_NUMBER: _ClassVar[int]
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     run_id: UUID
     thread_id: UUID
-    filters: _containers.MessageMap[str, AuthFilter]
-    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
 
 class DeleteRunRequest(_message.Message):
     __slots__ = ("run_id", "thread_id", "filters")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     RUN_ID_FIELD_NUMBER: _ClassVar[int]
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     run_id: UUID
     thread_id: UUID
-    filters: _containers.MessageMap[str, AuthFilter]
-    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ...) -> None: ...
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ...) -> None: ...
 
 class CancelRunIdsTarget(_message.Message):
     __slots__ = ("thread_id", "run_ids")
@@ -924,32 +788,18 @@ class CancelStatusTarget(_message.Message):
 
 class CancelRunRequest(_message.Message):
     __slots__ = ("filters", "run_ids", "status", "action")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     RUN_IDS_FIELD_NUMBER: _ClassVar[int]
     STATUS_FIELD_NUMBER: _ClassVar[int]
     ACTION_FIELD_NUMBER: _ClassVar[int]
-    filters: _containers.MessageMap[str, AuthFilter]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
     run_ids: CancelRunIdsTarget
     status: CancelStatusTarget
-    action: CancelRunAction
-    def __init__(self, filters: _Optional[_Mapping[str, AuthFilter]] = ..., run_ids: _Optional[_Union[CancelRunIdsTarget, _Mapping]] = ..., status: _Optional[_Union[CancelStatusTarget, _Mapping]] = ..., action: _Optional[_Union[CancelRunAction, str]] = ...) -> None: ...
+    action: _enum_cancel_run_action_pb2.CancelRunAction
+    def __init__(self, filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., run_ids: _Optional[_Union[CancelRunIdsTarget, _Mapping]] = ..., status: _Optional[_Union[CancelStatusTarget, _Mapping]] = ..., action: _Optional[_Union[_enum_cancel_run_action_pb2.CancelRunAction, str]] = ...) -> None: ...
 
 class SearchRunsRequest(_message.Message):
     __slots__ = ("thread_id", "filters", "limit", "offset", "status", "select")
-    class FiltersEntry(_message.Message):
-        __slots__ = ("key", "value")
-        KEY_FIELD_NUMBER: _ClassVar[int]
-        VALUE_FIELD_NUMBER: _ClassVar[int]
-        key: str
-        value: AuthFilter
-        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[AuthFilter, _Mapping]] = ...) -> None: ...
     THREAD_ID_FIELD_NUMBER: _ClassVar[int]
     FILTERS_FIELD_NUMBER: _ClassVar[int]
     LIMIT_FIELD_NUMBER: _ClassVar[int]
@@ -957,12 +807,12 @@ class SearchRunsRequest(_message.Message):
     STATUS_FIELD_NUMBER: _ClassVar[int]
     SELECT_FIELD_NUMBER: _ClassVar[int]
     thread_id: UUID
-    filters: _containers.MessageMap[str, AuthFilter]
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
     limit: int
     offset: int
-    status: RunStatus
+    status: _enum_run_status_pb2.RunStatus
     select: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Mapping[str, AuthFilter]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., status: _Optional[_Union[RunStatus, str]] = ..., select: _Optional[_Iterable[str]] = ...) -> None: ...
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., limit: _Optional[int] = ..., offset: _Optional[int] = ..., status: _Optional[_Union[_enum_run_status_pb2.RunStatus, str]] = ..., select: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class SearchRunsResponse(_message.Message):
     __slots__ = ("runs",)
@@ -975,11 +825,81 @@ class SetRunStatusRequest(_message.Message):
     RUN_ID_FIELD_NUMBER: _ClassVar[int]
     STATUS_FIELD_NUMBER: _ClassVar[int]
     run_id: UUID
-    status: RunStatus
-    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., status: _Optional[_Union[RunStatus, str]] = ...) -> None: ...
+    status: _enum_run_status_pb2.RunStatus
+    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., status: _Optional[_Union[_enum_run_status_pb2.RunStatus, str]] = ...) -> None: ...
 
 class SweepRunsResponse(_message.Message):
     __slots__ = ("run_ids",)
     RUN_IDS_FIELD_NUMBER: _ClassVar[int]
     run_ids: _containers.RepeatedCompositeFieldContainer[UUID]
     def __init__(self, run_ids: _Optional[_Iterable[_Union[UUID, _Mapping]]] = ...) -> None: ...
+
+class CountRunsRequest(_message.Message):
+    __slots__ = ("thread_id", "statuses")
+    THREAD_ID_FIELD_NUMBER: _ClassVar[int]
+    STATUSES_FIELD_NUMBER: _ClassVar[int]
+    thread_id: UUID
+    statuses: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., statuses: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class StreamRunRequest(_message.Message):
+    __slots__ = ("thread_id", "run_id", "filters", "last_event_id", "stream_modes", "ignore_run_not_found", "cancel_on_disconnect")
+    THREAD_ID_FIELD_NUMBER: _ClassVar[int]
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    FILTERS_FIELD_NUMBER: _ClassVar[int]
+    LAST_EVENT_ID_FIELD_NUMBER: _ClassVar[int]
+    STREAM_MODES_FIELD_NUMBER: _ClassVar[int]
+    IGNORE_RUN_NOT_FOUND_FIELD_NUMBER: _ClassVar[int]
+    CANCEL_ON_DISCONNECT_FIELD_NUMBER: _ClassVar[int]
+    thread_id: UUID
+    run_id: UUID
+    filters: _containers.RepeatedCompositeFieldContainer[AuthFilter]
+    last_event_id: str
+    stream_modes: _containers.RepeatedScalarFieldContainer[_enum_stream_mode_pb2.StreamMode]
+    ignore_run_not_found: bool
+    cancel_on_disconnect: bool
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ..., run_id: _Optional[_Union[UUID, _Mapping]] = ..., filters: _Optional[_Iterable[_Union[AuthFilter, _Mapping]]] = ..., last_event_id: _Optional[str] = ..., stream_modes: _Optional[_Iterable[_Union[_enum_stream_mode_pb2.StreamMode, str]]] = ..., ignore_run_not_found: bool = ..., cancel_on_disconnect: bool = ...) -> None: ...
+
+class EnterRunRequest(_message.Message):
+    __slots__ = ("run_id", "thread_id", "resumable")
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    THREAD_ID_FIELD_NUMBER: _ClassVar[int]
+    RESUMABLE_FIELD_NUMBER: _ClassVar[int]
+    run_id: UUID
+    thread_id: UUID
+    resumable: bool
+    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., resumable: bool = ...) -> None: ...
+
+class MarkRunDoneRequest(_message.Message):
+    __slots__ = ("run_id", "thread_id", "resumable")
+    RUN_ID_FIELD_NUMBER: _ClassVar[int]
+    THREAD_ID_FIELD_NUMBER: _ClassVar[int]
+    RESUMABLE_FIELD_NUMBER: _ClassVar[int]
+    run_id: UUID
+    thread_id: UUID
+    resumable: bool
+    def __init__(self, run_id: _Optional[_Union[UUID, _Mapping]] = ..., thread_id: _Optional[_Union[UUID, _Mapping]] = ..., resumable: bool = ...) -> None: ...
+
+class GetGraphIDRequest(_message.Message):
+    __slots__ = ("thread_id",)
+    THREAD_ID_FIELD_NUMBER: _ClassVar[int]
+    thread_id: UUID
+    def __init__(self, thread_id: _Optional[_Union[UUID, _Mapping]] = ...) -> None: ...
+
+class GetGraphIDResponse(_message.Message):
+    __slots__ = ("graph_id",)
+    GRAPH_ID_FIELD_NUMBER: _ClassVar[int]
+    graph_id: str
+    def __init__(self, graph_id: _Optional[str] = ...) -> None: ...
+
+class CreateRunResponse(_message.Message):
+    __slots__ = ("runs",)
+    RUNS_FIELD_NUMBER: _ClassVar[int]
+    runs: _containers.RepeatedCompositeFieldContainer[Run]
+    def __init__(self, runs: _Optional[_Iterable[_Union[Run, _Mapping]]] = ...) -> None: ...
+
+class ControlEvent(_message.Message):
+    __slots__ = ("action",)
+    ACTION_FIELD_NUMBER: _ClassVar[int]
+    action: _enum_control_signal_pb2.ControlSignal
+    def __init__(self, action: _Optional[_Union[_enum_control_signal_pb2.ControlSignal, str]] = ...) -> None: ...