Kris
2ab65f153f
新增 QQ Bot 渠道扩展，支持在 Yuxi 平台中集成 QQ 机器人渠道。 包含以下功能模块： - api_client: QQ API 客户端封装 - api_routes: API 路由管理 - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - websocket: WebSocket 实时连接 - credentials: 凭证管理 - token: Token 管理 - outbound: 外发消息管理 - outbound_media: 媒体外发 - streaming: 流式消息处理 - streaming_media: 媒体流处理 - pairing: 用户配对与绑定 - security: 安全校验 - dedupe: 消息去重 - monitor: 渠道状态监控 - status: 会话状态管理 - session: 会话管理 - pipeline: 消息管道 - pipeline_stages: 管道阶段 - commands: 指令处理 - commands_builtin: 内置指令 - interaction: 交互处理 - approval: 审批流程 - ark: ARK 消息 - audio: 音频处理 - media: 媒体资源 - media_chunked: 分块媒体 - media_tags: 媒体标签 - message_queue: 消息队列 - delivery: 消息送达确认 - reconnect: 重连机制 - typing_keepalive: 输入状态保活 - group_activation: 群激活 - group_gating: 群门控 - group_history: 群历史 - known_users: 已知用户 - ref_index: 引用索引 - tools: Agent 工具集成 - types: 类型定义
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from enum import StrEnum
|
|
|
|
|
|
class QQBotErrorCode(StrEnum):
|
|
AUTH_FAILED = "auth_failed"
|
|
INVALID_SESSION = "invalid_session"
|
|
RATE_LIMITED = "rate_limited"
|
|
SESSION_TIMEOUT = "session_timeout"
|
|
INSUFFICIENT_INTENTS = "insufficient_intents"
|
|
DISALLOWED_INTENTS = "disallowed_intents"
|
|
NETWORK_ERROR = "network_error"
|
|
CONNECTION_CLOSED = "connection_closed"
|
|
CONFIG_ERROR = "config_error"
|
|
TOKEN_EXPIRED = "token_expired"
|
|
MEDIA_UPLOAD_FAILED = "media_upload_failed"
|
|
STREAMING_FAILED = "streaming_failed"
|
|
SEND_FAILED = "send_failed"
|
|
UNKNOWN = "unknown"
|
|
|
|
|
|
class QQBotError(Exception):
|
|
def __init__(self, code: QQBotErrorCode, message: str, retryable: bool = False):
|
|
self.code = code
|
|
self.retryable = retryable
|
|
super().__init__(f"[{code.value}] {message}")
|
|
|
|
|
|
CLOSE_CODE_RETRY_MAP: dict[int, tuple[bool, float | None]] = {
|
|
1000: (False, None),
|
|
4004: (True, 0),
|
|
4006: (True, 0),
|
|
4007: (True, 0),
|
|
4008: (True, 60_000),
|
|
4009: (True, 0),
|
|
4914: (False, None),
|
|
4915: (False, None),
|
|
}
|
|
|
|
|
|
SERVER_ERROR_RANGE = range(4900, 4914)
|
|
|
|
|
|
def classify_close_code(code: int) -> tuple[bool, float | None]:
|
|
if code in CLOSE_CODE_RETRY_MAP:
|
|
return CLOSE_CODE_RETRY_MAP[code]
|
|
if code in SERVER_ERROR_RANGE:
|
|
return (True, 0)
|
|
return (True, 0) |