diff --git a/src/arkruntime/types/session/__init__.py b/src/arkruntime/types/session/__init__.py index 8857e13..d26a553 100644 --- a/src/arkruntime/types/session/__init__.py +++ b/src/arkruntime/types/session/__init__.py @@ -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 @@ -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, @@ -95,6 +103,11 @@ "ListSessionThreadsResponse", "ListSessionsOrder", "ListSessionsResponse", + "ManagedAgentsAgentMessagePreview", + "ManagedAgentsAgentThinkingPreview", + "ManagedAgentsDeltaContent", + "ManagedAgentsDeltaEvent", + "ManagedAgentsDeltaTextBlock", "ManagedAgentsDocumentBlock", "ManagedAgentsEventParams", "ManagedAgentsFileRubricParams", @@ -103,6 +116,9 @@ "ManagedAgentsRubric", "ManagedAgentsSearchResultBlock", "ManagedAgentsSessionEvent", + "ManagedAgentsStartEvent", + "ManagedAgentsStartEventPreview", + "ManagedAgentsStreamPreviewEvent", "ManagedAgentsSystemContentBlock", "ManagedAgentsSystemMessageEventParams", "ManagedAgentsTextBlock", diff --git a/src/arkruntime/types/session/managed_agents_agent_message_preview.py b/src/arkruntime/types/session/managed_agents_agent_message_preview.py new file mode 100644 index 0000000..88781ee --- /dev/null +++ b/src/arkruntime/types/session/managed_agents_agent_message_preview.py @@ -0,0 +1,27 @@ +# Generated by datamodel-code-generator from ark-apis typespec/openapi. +# DO NOT EDIT — regenerate with `make gen-py-` 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"`。 + """ diff --git a/src/arkruntime/types/session/managed_agents_agent_thinking_preview.py b/src/arkruntime/types/session/managed_agents_agent_thinking_preview.py new file mode 100644 index 0000000..df3d884 --- /dev/null +++ b/src/arkruntime/types/session/managed_agents_agent_thinking_preview.py @@ -0,0 +1,27 @@ +# Generated by datamodel-code-generator from ark-apis typespec/openapi. +# DO NOT EDIT — regenerate with `make gen-py-` 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"`。 + """ diff --git a/src/arkruntime/types/session/managed_agents_delta_content.py b/src/arkruntime/types/session/managed_agents_delta_content.py new file mode 100644 index 0000000..40ff52d --- /dev/null +++ b/src/arkruntime/types/session/managed_agents_delta_content.py @@ -0,0 +1,34 @@ +# Generated by datamodel-code-generator from ark-apis typespec/openapi. +# DO NOT EDIT — regenerate with `make gen-py-` 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 以兼容历史帧。 + """ diff --git a/src/arkruntime/types/session/managed_agents_delta_event.py b/src/arkruntime/types/session/managed_agents_delta_event.py new file mode 100644 index 0000000..42976ac --- /dev/null +++ b/src/arkruntime/types/session/managed_agents_delta_event.py @@ -0,0 +1,34 @@ +# Generated by datamodel-code-generator from ark-apis typespec/openapi. +# DO NOT EDIT — regenerate with `make gen-py-` 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 一致。 + """ diff --git a/src/arkruntime/types/session/managed_agents_delta_text_block.py b/src/arkruntime/types/session/managed_agents_delta_text_block.py new file mode 100644 index 0000000..ad0f823 --- /dev/null +++ b/src/arkruntime/types/session/managed_agents_delta_text_block.py @@ -0,0 +1,27 @@ +# Generated by datamodel-code-generator from ark-apis typespec/openapi. +# DO NOT EDIT — regenerate with `make gen-py-` 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 + """ + 本帧新增的文本。 + """ diff --git a/src/arkruntime/types/session/managed_agents_session_event.py b/src/arkruntime/types/session/managed_agents_session_event.py index 7529034..824cba5 100644 --- a/src/arkruntime/types/session/managed_agents_session_event.py +++ b/src/arkruntime/types/session/managed_agents_session_event.py @@ -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 """ diff --git a/src/arkruntime/types/session/managed_agents_start_event.py b/src/arkruntime/types/session/managed_agents_start_event.py new file mode 100644 index 0000000..1e69781 --- /dev/null +++ b/src/arkruntime/types/session/managed_agents_start_event.py @@ -0,0 +1,31 @@ +# Generated by datamodel-code-generator from ark-apis typespec/openapi. +# DO NOT EDIT — regenerate with `make gen-py-` 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`。 + """ diff --git a/src/arkruntime/types/session/managed_agents_start_event_preview.py b/src/arkruntime/types/session/managed_agents_start_event_preview.py new file mode 100644 index 0000000..6007df2 --- /dev/null +++ b/src/arkruntime/types/session/managed_agents_start_event_preview.py @@ -0,0 +1,27 @@ +# Generated by datamodel-code-generator from ark-apis typespec/openapi. +# DO NOT EDIT — regenerate with `make gen-py-` 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`。 +""" diff --git a/src/arkruntime/types/session/managed_agents_stream_preview_event.py b/src/arkruntime/types/session/managed_agents_stream_preview_event.py new file mode 100644 index 0000000..6f561aa --- /dev/null +++ b/src/arkruntime/types/session/managed_agents_stream_preview_event.py @@ -0,0 +1,26 @@ +# Generated by datamodel-code-generator from ark-apis typespec/openapi. +# DO NOT EDIT — regenerate with `make gen-py-` 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 帧。 +"""