Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 16 additions & 0 deletions src/arkruntime/types/session/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,11 @@
from .list_session_threads_response import ListSessionThreadsResponse
from .list_sessions_order import ListSessionsOrder
from .list_sessions_response import ListSessionsResponse
from .managed_agents_agent_message_preview import ManagedAgentsAgentMessagePreview
from .managed_agents_agent_thinking_preview import ManagedAgentsAgentThinkingPreview
from .managed_agents_delta_content import ManagedAgentsDeltaContent
from .managed_agents_delta_event import ManagedAgentsDeltaEvent
from .managed_agents_delta_text_block import ManagedAgentsDeltaTextBlock
from .managed_agents_document_block import ManagedAgentsDocumentBlock
from .managed_agents_event_params import ManagedAgentsEventParams
from .managed_agents_file_rubric_params import ManagedAgentsFileRubricParams
Expand All @@ -31,6 +36,9 @@
from .managed_agents_rubric import ManagedAgentsRubric
from .managed_agents_search_result_block import ManagedAgentsSearchResultBlock
from .managed_agents_session_event import ManagedAgentsSessionEvent
from .managed_agents_start_event import ManagedAgentsStartEvent
from .managed_agents_start_event_preview import ManagedAgentsStartEventPreview
from .managed_agents_stream_preview_event import ManagedAgentsStreamPreviewEvent
from .managed_agents_system_content_block import ManagedAgentsSystemContentBlock
from .managed_agents_system_message_event_params import (
ManagedAgentsSystemMessageEventParams,
Expand Down Expand Up @@ -95,6 +103,11 @@
"ListSessionThreadsResponse",
"ListSessionsOrder",
"ListSessionsResponse",
"ManagedAgentsAgentMessagePreview",
"ManagedAgentsAgentThinkingPreview",
"ManagedAgentsDeltaContent",
"ManagedAgentsDeltaEvent",
"ManagedAgentsDeltaTextBlock",
"ManagedAgentsDocumentBlock",
"ManagedAgentsEventParams",
"ManagedAgentsFileRubricParams",
Expand All @@ -103,6 +116,9 @@
"ManagedAgentsRubric",
"ManagedAgentsSearchResultBlock",
"ManagedAgentsSessionEvent",
"ManagedAgentsStartEvent",
"ManagedAgentsStartEventPreview",
"ManagedAgentsStreamPreviewEvent",
"ManagedAgentsSystemContentBlock",
"ManagedAgentsSystemMessageEventParams",
"ManagedAgentsTextBlock",
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# Generated by datamodel-code-generator from ark-apis typespec/openapi.
# DO NOT EDIT — regenerate with `make gen-py-<api>` in ark-apis.
#
# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import Literal

from arkruntime._models import BaseModel


class ManagedAgentsAgentMessagePreview(BaseModel):
"""
`agent.message` 的流式预览标识。`id` 同时关联后续 `event_delta.event_id`
与最终完整 `agent.message.id`。
"""

id: str
"""
被预览事件的 id。
"""
type: Literal["agent.message"]
"""
固定 `"agent.message"`。
"""
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# Generated by datamodel-code-generator from ark-apis typespec/openapi.
# DO NOT EDIT — regenerate with `make gen-py-<api>` in ark-apis.
#
# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import Literal

from arkruntime._models import BaseModel


class ManagedAgentsAgentThinkingPreview(BaseModel):
"""
`agent.thinking` 的流式预览标识。`id` 同时关联后续 `event_delta.event_id`
与最终完整 `agent.thinking.id`。
"""

id: str
"""
被预览事件的 id。
"""
type: Literal["agent.thinking"]
"""
固定 `"agent.thinking"`。
"""
34 changes: 34 additions & 0 deletions src/arkruntime/types/session/managed_agents_delta_content.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
# Generated by datamodel-code-generator from ark-apis typespec/openapi.
# DO NOT EDIT — regenerate with `make gen-py-<api>` in ark-apis.
#
# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import Literal, Optional

from arkruntime._models import BaseModel

from .managed_agents_delta_text_block import ManagedAgentsDeltaTextBlock


class ManagedAgentsDeltaContent(BaseModel):
"""
`event_delta.delta` 的 `content_delta` 载荷。`content` 是一个文本片段,
`index` 指定它落到最终事件 `content[]` 的位置。
"""

type: Literal["content_delta"]
"""
固定 `"content_delta"`,区别于 Messages API 的 `content_block_delta`。
"""
content: ManagedAgentsDeltaTextBlock
"""
与最终 `agent.message.content[]` / `agent.thinking.content[]` 元素同形的文本片段。
"""
index: Optional[int] = None
"""
最终 `content[]` 的下标。新 index 插入条目,已见过的 index 追加文本。
当前 agent-loop 总会写出该字段;公开层保留 optional 以兼容历史帧。
"""
34 changes: 34 additions & 0 deletions src/arkruntime/types/session/managed_agents_delta_event.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
# Generated by datamodel-code-generator from ark-apis typespec/openapi.
# DO NOT EDIT — regenerate with `make gen-py-<api>` in ark-apis.
#
# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import Literal

from arkruntime._models import BaseModel

from .managed_agents_delta_content import ManagedAgentsDeltaContent


class ManagedAgentsDeltaEvent(BaseModel):
"""
`event_delta`:仍在流式生成的事件增量。客户端用 `event_id` 反查前面的
`event_start.event.type`,并按 `(event_id, index)` 累积文本。该帧是
best-effort 预览,不进入事件历史,也不携带顶层 `id` / `processed_at`。
"""

type: Literal["event_delta"]
"""
固定 `"event_delta"`。
"""
delta: ManagedAgentsDeltaContent
"""
被预览事件的一个内容增量。
"""
event_id: str
"""
对应 `event_start.event.id`,也与最终完整 buffered 事件的 id 一致。
"""
27 changes: 27 additions & 0 deletions src/arkruntime/types/session/managed_agents_delta_text_block.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# Generated by datamodel-code-generator from ark-apis typespec/openapi.
# DO NOT EDIT — regenerate with `make gen-py-<api>` in ark-apis.
#
# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import Literal

from arkruntime._models import BaseModel


class ManagedAgentsDeltaTextBlock(BaseModel):
"""
`content_delta.content` 的文本片段。wire 形态与 buffered 事件中的 text block
相同;单独建模可在直接引用时保留必需的 `type: "text"` discriminator。
"""

type: Literal["text"]
"""
固定 `"text"`。
"""
text: str
"""
本帧新增的文本。
"""
6 changes: 4 additions & 2 deletions src/arkruntime/types/session/managed_agents_session_event.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,12 +15,14 @@ class ManagedAgentsSessionEvent(BaseModel):
"""
会话流上的一个事件。`type` 是判别式,其余字段随 type 变化。SDK 层建议
把这个模型作为出流事件的公共 envelope;具体 variant 的 payload 可以在
消费侧按 `type` 再解构。
消费侧按 `type` 再解构。`event_start` / `event_delta` 的强类型结构见
`ManagedAgentsStreamPreviewEvent`;这两种瞬时预览帧不携带 `id` /
`processed_at`。
"""

type: str
"""
事件类型判别式,例如 `agent.message` / `span.outcome_evaluation_end` / `session.status_idle`。
事件类型判别式,例如 `agent.message` / `event_start` / `event_delta` / `session.status_idle`。
"""
id: Optional[str] = None
"""
Expand Down
31 changes: 31 additions & 0 deletions src/arkruntime/types/session/managed_agents_start_event.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
# Generated by datamodel-code-generator from ark-apis typespec/openapi.
# DO NOT EDIT — regenerate with `make gen-py-<api>` in ark-apis.
#
# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import Literal

from arkruntime._models import BaseModel

from .managed_agents_start_event_preview import ManagedAgentsStartEventPreview


class ManagedAgentsStartEvent(BaseModel):
"""
`event_start`:打开一次 buffered 事件的流式预览。该帧仅在流式接口通过
`event_deltas` opt-in 后出现,不进入事件历史,也不携带顶层 `id` /
`processed_at`。通常由相同 id 的完整 buffered 事件收尾;若模型请求中途
结束,则由对应的 `span.model_request_end` 关闭预览。
"""

type: Literal["event_start"]
"""
固定 `"event_start"`。
"""
event: ManagedAgentsStartEventPreview
"""
被预览的 `agent.message` 或 `agent.thinking`。
"""
27 changes: 27 additions & 0 deletions src/arkruntime/types/session/managed_agents_start_event_preview.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# Generated by datamodel-code-generator from ark-apis typespec/openapi.
# DO NOT EDIT — regenerate with `make gen-py-<api>` in ark-apis.
#
# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import Annotated, Union

from pydantic import Field
from typing_extensions import TypeAliasType

from .managed_agents_agent_message_preview import ManagedAgentsAgentMessagePreview
from .managed_agents_agent_thinking_preview import ManagedAgentsAgentThinkingPreview

ManagedAgentsStartEventPreview = TypeAliasType(
"ManagedAgentsStartEventPreview",
Annotated[
Union[ManagedAgentsAgentMessagePreview, ManagedAgentsAgentThinkingPreview],
Field(discriminator="type"),
],
)
"""
`event_start.event` 的 tagged union。目标事件的 `type` 决定后续
`event_delta` 应累积到 `agent.message` 还是 `agent.thinking`。
"""
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
# Generated by datamodel-code-generator from ark-apis typespec/openapi.
# DO NOT EDIT — regenerate with `make gen-py-<api>` in ark-apis.
#
# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import Annotated, Union

from pydantic import Field
from typing_extensions import TypeAliasType

from .managed_agents_delta_event import ManagedAgentsDeltaEvent
from .managed_agents_start_event import ManagedAgentsStartEvent

ManagedAgentsStreamPreviewEvent = TypeAliasType(
"ManagedAgentsStreamPreviewEvent",
Annotated[
Union[ManagedAgentsStartEvent, ManagedAgentsDeltaEvent],
Field(discriminator="type"),
],
)
"""
流式预览新增的两种 SSE 帧。
"""
Loading