arkruntime 0.1.0__py3-none-any.whl

arkruntime/types/session/update_session_request.py

@@ -0,0 +1,28 @@
+# 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 List, Optional
+
+from arkruntime._models import BaseModel
+
+from .tag import Tag
+
+
+class UpdateSessionRequest(BaseModel):
+    """
+    局部更新会话；所有字段均 optional，omit 表示保留。
+    """
+
+    tags: Optional[List[Tag]] = None
+    """
+    资源标签（整体替换）。
+    """
+    title: Optional[str] = None
+    """
+    会话人类可读标题。
+    """

arkruntime/types/session/url_document_source.py

@@ -0,0 +1,23 @@
+# 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 UrlDocumentSource(BaseModel):
+    """
+    通过 URL 引用的文档。
+    """
+
+    url: str
+    """
+    文档 URL。
+    """
+    type: Literal["url"]

arkruntime/types/session/url_image_source.py

@@ -0,0 +1,23 @@
+# 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 UrlImageSource(BaseModel):
+    """
+    通过 URL 引用的图片。
+    """
+
+    url: str
+    """
+    图片 URL。
+    """
+    type: Literal["url"]

arkruntime/types/skill/__init__.py

@@ -0,0 +1,18 @@
+# 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 .create_skill_request import CreateSkillRequest
+from .skill import Skill
+
+__all__ = [
+    "CreateSkillRequest",
+    "Skill",
+]
+
+# Hand-written extras (preserved across regen via Makefile rsync --exclude=*_shim.py).
+from ._init_extras_shim import *  # noqa: F401,F403

arkruntime/types/skill/_init_extras_shim.py

@@ -0,0 +1,12 @@
+# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+
+"""Hand-written extras shim for the skill API (empty stub).
+
+Preserved across regen via the Makefile rsync --exclude=*_shim.py rule.
+Add re-exports here as needed.
+"""
+
+from __future__ import annotations
+
+__all__: list[str] = []

arkruntime/types/skill/create_skill_request.py

@@ -0,0 +1,18 @@
+# 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 Optional
+
+from arkruntime._models import BaseModel
+
+
+class CreateSkillRequest(BaseModel):
+    display_title: Optional[str] = None
+    """
+    Skill 展示名。
+    """

arkruntime/types/skill/skill.py

@@ -0,0 +1,46 @@
+# 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
+
+
+class Skill(BaseModel):
+    """
+    用户上传到 Managed Agents 控制面的技能摘要。
+
+    与其他资源不同：Skill 用 `object` 而不是 `type` 作为对象类型判别字段，
+    且 `created_at` 是 Unix 秒（int64）而非 RFC 3339 字符串 —— 与
+    ark-managed-agents thrift 契约保持一致。
+    """
+
+    id: str
+    """
+    Skill ID，形如 `skill_01XJ5...`。
+    """
+    object: Literal["skill"]
+    """
+    固定 `"skill"`。
+    """
+    created_at: int
+    """
+    创建时间（Unix 秒）。
+    """
+    description: Optional[str] = None
+    """
+    描述信息。
+    """
+    latest_version: str
+    """
+    最新版本号。
+    """
+    name: str
+    """
+    人类可读名称。
+    """

arkruntime/types/tokenization/__init__.py

@@ -0,0 +1,24 @@
+# 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 .tokenization_input import TokenizationInput
+from .tokenization_item import TokenizationItem
+from .tokenization_request import TokenizationRequest
+from .tokenization_response import TokenizationResponse
+
+__all__ = [
+    "TokenizationInput",
+    "TokenizationItem",
+    "TokenizationRequest",
+    "TokenizationResponse",
+]
+from .tokenization_input_param import TokenizationInputParam
+from .tokenization_request_param import TokenizationRequestParam
+
+# Hand-written extras (preserved across regen via Makefile rsync --exclude=*_shim.py).
+from ._init_extras_shim import *  # noqa: F401,F403

arkruntime/types/tokenization/_init_extras_shim.py

@@ -0,0 +1,12 @@
+# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+
+"""Hand-written extras shim for the tokenization API (empty stub).
+
+Preserved across regen via the Makefile rsync --exclude=*_shim.py rule.
+Add re-exports here as needed.
+"""
+
+from __future__ import annotations
+
+__all__: list[str] = []

arkruntime/types/tokenization/tokenization_input.py

@@ -0,0 +1,17 @@
+# 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 List, Union
+
+from typing_extensions import TypeAliasType
+
+TokenizationInput = TypeAliasType("TokenizationInput", Union[str, List[str]])
+"""
+Input text(s) to tokenize. A single string tokenizes one input; an array
+of strings tokenizes each element independently.
+"""

arkruntime/types/tokenization/tokenization_input_param.py

@@ -0,0 +1,15 @@
+# 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 List, TypeAlias, Union
+
+TokenizationInputParam: TypeAlias = Union[str, List[str]]
+"""
+Input text(s) to tokenize. A single string tokenizes one input; an array
+of strings tokenizes each element independently.
+"""

arkruntime/types/tokenization/tokenization_item.py

@@ -0,0 +1,39 @@
+# 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 List, Literal
+
+from arkruntime._models import BaseModel
+
+
+class TokenizationItem(BaseModel):
+    """
+    A single tokenization result for one input string.
+    """
+
+    index: int
+    """
+    The index of this result in the request `text` array.
+    """
+    object: Literal["tokenization"]
+    """
+    The object type. Always `tokenization`.
+    """
+    total_tokens: int
+    """
+    The total number of tokens produced for this input.
+    """
+    token_ids: List[int]
+    """
+    The token IDs produced by the tokenizer, in order.
+    """
+    offset_mapping: List[List[int]]
+    """
+    Per-token character offsets into the input string, as
+    `[start, end]` pairs aligned with `token_ids`.
+    """

arkruntime/types/tokenization/tokenization_request.py

@@ -0,0 +1,30 @@
+# 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 Optional
+
+from arkruntime._models import BaseModel
+
+from .tokenization_input import TokenizationInput
+
+
+class TokenizationRequest(BaseModel):
+    model: str
+    """
+    Endpoint ID or model name to tokenize for.
+    """
+    text: TokenizationInput
+    """
+    Input text to tokenize, as a string or array of strings. To tokenize
+    multiple inputs in one request, pass an array.
+    """
+    user: Optional[str] = None
+    """
+    A unique identifier representing the end-user. Currently accepted but
+    not used by the server; reserved for parity with other endpoints.
+    """

arkruntime/types/tokenization/tokenization_request_param.py

@@ -0,0 +1,30 @@
+# 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 Optional
+
+from typing_extensions import Required, TypedDict
+
+from .tokenization_input_param import TokenizationInputParam
+
+
+class TokenizationRequestParam(TypedDict, total=False):
+    model: Required[str]
+    """Endpoint ID or model name to tokenize for."""
+
+    text: Required[TokenizationInputParam]
+    """
+    Input text to tokenize, as a string or array of strings. To tokenize
+    multiple inputs in one request, pass an array.
+    """
+
+    user: Optional[str]
+    """
+    A unique identifier representing the end-user. Currently accepted but
+    not used by the server; reserved for parity with other endpoints.
+    """

arkruntime/types/tokenization/tokenization_response.py

@@ -0,0 +1,36 @@
+# 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 List, Literal
+
+from arkruntime._models import BaseModel
+
+from .tokenization_item import TokenizationItem
+
+
+class TokenizationResponse(BaseModel):
+    id: str
+    """
+    A unique identifier for the tokenization response.
+    """
+    object: Literal["list"]
+    """
+    The object type. Always `list`.
+    """
+    created: int
+    """
+    Unix timestamp (in seconds) of when the response was created.
+    """
+    model: str
+    """
+    The model that produced the tokenization.
+    """
+    data: List[TokenizationItem]
+    """
+    The list of tokenization results, one per input.
+    """

arkruntime/types/vault/__init__.py

@@ -0,0 +1,62 @@
+# 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 .create_credential_request import CreateCredentialRequest
+from .create_vault_request import CreateVaultRequest
+from .credential import Credential
+from .credential_auth import CredentialAuth
+from .credential_auth_type import CredentialAuthType
+from .credential_networking import CredentialNetworking
+from .credential_networking_type import CredentialNetworkingType
+from .credential_validation import CredentialValidation
+from .delete_credential_response import DeleteCredentialResponse
+from .delete_vault_response import DeleteVaultResponse
+from .limited_credential_networking import LimitedCredentialNetworking
+from .list_credentials_response import ListCredentialsResponse
+from .list_vaults_response import ListVaultsResponse
+from .mcp_probe import MCPProbe
+from .mcpo_auth_refresh import MCPOAuthRefresh
+from .refresh_http_response import RefreshHTTPResponse
+from .refresh_object import RefreshObject
+from .refresh_status import RefreshStatus
+from .token_endpoint_auth import TokenEndpointAuth
+from .token_endpoint_auth_type import TokenEndpointAuthType
+from .update_credential_request import UpdateCredentialRequest
+from .update_vault_request import UpdateVaultRequest
+from .validation_status import ValidationStatus
+from .vault import Vault
+
+__all__ = [
+    "CreateCredentialRequest",
+    "CreateVaultRequest",
+    "Credential",
+    "CredentialAuth",
+    "CredentialAuthType",
+    "CredentialNetworking",
+    "CredentialNetworkingType",
+    "CredentialValidation",
+    "DeleteCredentialResponse",
+    "DeleteVaultResponse",
+    "LimitedCredentialNetworking",
+    "ListCredentialsResponse",
+    "ListVaultsResponse",
+    "MCPOAuthRefresh",
+    "MCPProbe",
+    "RefreshHTTPResponse",
+    "RefreshObject",
+    "RefreshStatus",
+    "TokenEndpointAuth",
+    "TokenEndpointAuthType",
+    "UpdateCredentialRequest",
+    "UpdateVaultRequest",
+    "ValidationStatus",
+    "Vault",
+]
+
+# Hand-written extras (preserved across regen via Makefile rsync --exclude=*_shim.py).
+from ._init_extras_shim import *  # noqa: F401,F403

arkruntime/types/vault/_init_extras_shim.py

@@ -0,0 +1,12 @@
+# Copyright (c) 2026 ByteDance Ltd. and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+
+"""Hand-written extras shim for the vault API (empty stub).
+
+Preserved across regen via the Makefile rsync --exclude=*_shim.py rule.
+Add re-exports here as needed.
+"""
+
+from __future__ import annotations
+
+__all__: list[str] = []

arkruntime/types/vault/create_credential_request.py

@@ -0,0 +1,32 @@
+# 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 Dict, Optional
+
+from arkruntime._models import BaseModel
+
+from .credential_auth import CredentialAuth
+
+
+class CreateCredentialRequest(BaseModel):
+    """
+    创建 Credential 的请求体。
+    """
+
+    auth: CredentialAuth
+    """
+    鉴权细节。所有 secret 字段是 write-only。
+    """
+    display_name: Optional[str] = None
+    """
+    人类可读名称。
+    """
+    metadata: Optional[Dict[str, str]] = None
+    """
+    用户自定义键值对元数据。
+    """

arkruntime/types/vault/create_vault_request.py

@@ -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 Dict, Optional
+
+from arkruntime._models import BaseModel
+
+
+class CreateVaultRequest(BaseModel):
+    """
+    创建 Vault 的请求体。
+    """
+
+    display_name: str
+    """
+    人类可读名称，1–255 字符。
+    """
+    metadata: Optional[Dict[str, str]] = None
+    """
+    用户自定义键值对元数据。
+    """

arkruntime/types/vault/credential.py

@@ -0,0 +1,52 @@
+# 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 Dict, Literal, Optional
+
+from arkruntime._models import BaseModel
+
+from .credential_auth import CredentialAuth
+
+
+class Credential(BaseModel):
+    """
+    一条挂在 Vault 上的凭证。
+    """
+
+    id: str
+    """
+    Credential ID。
+    """
+    type: Literal["vault_credential"]
+    """
+    固定 `"vault_credential"`。
+    """
+    vault_id: str
+    """
+    归属的 Vault ID。
+    """
+    auth: CredentialAuth
+    """
+    鉴权细节。响应中所有 secret 字段被抹掉。
+    """
+    display_name: Optional[str] = None
+    """
+    人类可读名称。
+    """
+    metadata: Optional[Dict[str, str]] = None
+    """
+    用户自定义键值对元数据。
+    """
+    created_at: str
+    """
+    RFC 3339 时间。
+    """
+    updated_at: str
+    """
+    RFC 3339 时间。
+    """

arkruntime/types/vault/credential_auth.py

@@ -0,0 +1,64 @@
+# 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 Optional
+
+from arkruntime._models import BaseModel
+
+from .credential_auth_type import CredentialAuthType
+from .credential_networking import CredentialNetworking
+from .mcpo_auth_refresh import MCPOAuthRefresh
+
+
+class CredentialAuth(BaseModel):
+    """
+    凭证的鉴权细节。按 `type` 决定哪些字段非空：
+    - `mcp_oauth` → `mcp_server_url` + `access_token`(write-only) + 可选 `expires_at` / `refresh`
+    - `static_bearer` → `mcp_server_url` + `token`(write-only)
+    - `environment_variable` → `secret_name` + `secret_value`(write-only) + `networking`
+
+    所有 secret 字段（`access_token` / `token` / `secret_value` / `refresh.refresh_token` /
+    `token_endpoint_auth.client_secret`）都是 write-only：响应永不返回。
+    """
+
+    type: CredentialAuthType
+    """
+    凭证 auth 类型。
+    """
+    mcp_server_url: Optional[str] = None
+    """
+    MCP server URL；`mcp_oauth` / `static_bearer` 用。
+    """
+    access_token: Optional[str] = None
+    """
+    OAuth access token；`mcp_oauth` 专用（write-only）。
+    """
+    expires_at: Optional[str] = None
+    """
+    access token 过期时间（RFC 3339）；`mcp_oauth` 专用。
+    """
+    refresh: Optional[MCPOAuthRefresh] = None
+    """
+    OAuth refresh 配置；`mcp_oauth` 专用。
+    """
+    token: Optional[str] = None
+    """
+    Bearer token；`static_bearer` 专用（write-only）。
+    """
+    secret_name: Optional[str] = None
+    """
+    环境变量名；`environment_variable` 专用。
+    """
+    secret_value: Optional[str] = None
+    """
+    环境变量值；`environment_variable` 专用（write-only）。
+    """
+    networking: Optional[CredentialNetworking] = None
+    """
+    注入时的网络策略；`environment_variable` 专用。
+    """

arkruntime/types/vault/credential_auth_type.py

@@ -0,0 +1,22 @@
+# 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 enum import Enum
+
+
+class CredentialAuthType(str, Enum):
+    """
+    凭证 auth 类型。按值决定 `CredentialAuth` 里哪些字段被使用：
+    - `mcp_oauth`：`mcp_server_url` + `access_token`(w) + 可选 `expires_at` / `refresh`
+    - `static_bearer`：`mcp_server_url` + `token`(w)
+    - `environment_variable`：`secret_name` + `secret_value`(w) + `networking`
+    """
+
+    mcp_oauth = "mcp_oauth"
+    static_bearer = "static_bearer"
+    environment_variable = "environment_variable"