# atriamore

**Repository Path**: opencode-plugin/atriamore

## Basic Information

- **Project Name**: atriamore
- **Description**: atriamore-支持多key轮询
- **Primary Language**: TypeScript
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2026-09-19
- **Last Updated**: 2026-10-07

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Atria More OpenCode Provider Plugin

**提供商 ID**: `atriamore` | **显示名称**: Atria More | **Base URL**: `https://api.atria-asi.ai/v1`

OpenCode 插件，集成 Atria API（api.atria-asi.ai）网关，支持 API Key 多密钥轮询以缓解速率限制。

## 功能

- **多 Key 轮询** — 输入多个 API Key（逗号分隔），自动循环切换，减缓 `429` 速率限制
- **错误容错** — 单个 Key 连续错误 3 次自动跳过，全部不可用时重置计数
- **模型支持** — 内置 `Atria-Dawn-Preview`（744B MoE，GLM-5.2 基座，256K 上下文，纯文本）
- **思考内容分离** — 声明 `interleaved` 后 OpenCode 渲染独立思考区；插件注入 `thinking` 开关兜底
- **模型别名映射** — 上游模型 id 大小写敏感（`Atria-Dawn-Preview`），OpenCode 上送全小写别名，插件改写后请求
- **短暂错误重试** — 上游 5xx 与偶发 `422 upstream_error` 统一转为 429，交由 OpenCode 重试
- **未授权兜底** — 未配置 Key 时直接返回 401 并提示授权，不触发无意义重试
- **首 chunk 超时保护** — SSE 流 15 秒无首块时返回 429，交由 OpenCode 重试（防上游挂起）
- **调试日志** — `~/.atriamore/debug.log`，由环境变量控制开关
- **凭证存储** — API Key 持久化到 `~/.atriamore/auth.json`

## 安装

### 前提

- Node.js 22+
- OpenCode

### 步骤

```bash
# 1. 进入项目
cd atriamore

# 2. 安装依赖
npm install

# 3. 构建
npm run build
```

## 配置

### 1. 配置 opencode.json

```json
// opencode.json
{
  "plugins": [
    "/path/to/atriamore"
  ]
}
```

### 2. 添加凭据

在 OpenCode 中执行 `/connect` 命令，选择 **Atria More**：

```
┌ Atria-Dawn-Preview 模型，基座 glm-5.2，注册送1亿token
│
│ 创建 API Key：https://api.atria-asi.ai/console/keys
│
│ 查看官方文档：https://api.atria-asi.ai/docs#streaming
│
│ 输入 API Keys（多key用逗号分隔）
│ ────────────────────────────────
│ atr-key1,atr-key2,atr-key3
└ enter
```

> 多个 Key 用逗号分隔，插件会自动轮询使用。未授权时模型请求会返回
> `未授权：请通过 /connect 命令点击 Atria More 进行授权`。

### 3. 选择模型

执行 `/models` 命令，选择可用模型：

| Model ID | 上游模型 | 描述 | 上下文 | 最大输出 | 输入 | 价格（输入/输出，$/1M） |
|----------|----------|------|--------|----------|------|--------------------------|
| `atria-dawn-preview` | `Atria-Dawn-Preview` | Atria Dawn Preview（744B MoE，GLM-5.2 基座） | 256K | 64K | text | 0 / 0 |

> **模型别名映射**：OpenCode 上送的是表中 `Model ID` 列（全小写别名），插件通过它匹配
> `constants.ts` 中的模型定义，再用「上游模型」列的真实 id 请求上游。
> 上游模型 id **大小写敏感** —— 传小写 id 会返回
> `400 A supported model is required.`，因此改写是必需的。

## 思考内容处理

上游默认原生返回分离的 `reasoning_content`（非流式与流式均实测确认），
OpenCode 通过 `interleaved: { field: 'reasoning_content' }` 渲染为独立思考区。

插件额外在请求体注入 `thinking: { type: "enabled" }` 作为开关兜底
（防止上游偶发把推理混入 `content`）。`reasoning_effort` 由 OpenCode 按用户选择的
档位自动传入（SDK 在 `transformRequestBody` 之前写入），插件仅透传、不写死。

## 多 Key 轮询机制

```
请求 → chat.headers → KeyManager.getNextKey()
                        ↓
                 ┌──────────────┐
                 │  key1  ← 首次 │
                 │  key2  ← 下次 │
                 │  key3  ← 再下次 │
                 └──────────────┘
                        ↓
                 写入 Authorization → 发起 API 请求
```

- 循环轮询：`key1 → key2 → key3 → key1 → ...`
- 错误追踪：每个 Key 连续失败 3 次后跳过
- 自动恢复：所有 Key 都被跳过时重置计数
- 成功自愈：请求成功后清零该 Key 的错误计数

## 错误处理

自定义 `fetch` 统一将上游错误转换为 OpenCode 可解析的标准 JSON 错误流：

| 场景 | 行为 |
|------|------|
| 401 / 403 | 原样返回上游 Response；`isRetryable=false` 立即报错（不重试），保留上游原始错误消息 |
| 429 | 原样返回上游 Response + `Retry-After`；由 OpenCode 判断重试 |
| 5xx | 返回固定 429 错误流，触发 OpenCode 重试 |
| 422 `upstream_error` | 返回固定 429 错误流，触发 OpenCode 重试（上游推理偶发失败） |
| 400 `invalid_request` | 保留原状态码立即报错（参数校验错误，重试无意义） |
| 400 `upstream_error` | 保留原状态码立即报错（请求体层面的确定性失败，重试无意义） |
| 408 / 409 | 返回原状态码，OpenCode 视为可重试 |
| 网络异常 | 返回 502 标准错误流，不向 UI 抛异常 |
| 未授权（无 Key） | 返回 401 并提示 `/connect` 授权，不发真实请求 |
| 首 chunk 15s 超时 | 返回 429 触发 OpenCode 重试（防上游挂起） |

> 判据来自实测：
> - genuine 参数校验错误 → `400` + `code: "invalid_request"`（如 `messages must be a non-empty array.`）
> - 请求体层面确定性失败 → `400` + `code: "upstream_error"`（如畸形 tool arguments，重放 6/6 稳定复现）
> - 偶发推理失败 → `422` + `code: "upstream_error"`（message `Inference request failed.`，
>   同一请求多数 200、偶发 422，重试可恢复）
>
> 因此仅 `422 upstream_error` 归类为短暂性错误转 429；`400` 一律原样立即报错。

## 调试日志

通过环境变量开启调试日志：

```bash
# 方式一：全局 DEBUG
DEBUG=1 opencode

# 方式二：仅 Atria More
OPENCODE_ATRIAMORE_DEBUG=1 opencode
```

日志位置：`%USERPROFILE%\.atriamore\debug.log`

> 环境变量在**插件加载时**读取，需在启动 OpenCode 前设置。

### 日志格式

可读 TEXT 单行格式（对齐 agnesmore/sensemore 约定），按 append 追加：

```
[2026-09-19T08:25:05.573Z] [INFO ] Auth loader initialized  {"keyCount":8,"keys":["1:atr_EXAM...a1b2","2:atr_EXAM...c3d4"]}
[2026-09-19T08:25:05.595Z] [INFO ] Request start  {"seq":1,"key":"atr_EXAM...a1b2","keyPos":"1/8","model":"Atria-Dawn-Preview","endpoint":"/v1/chat/completions"}
[2026-09-19T08:25:09.864Z] [INFO ] Request success  {"seq":1,"key":"atr_EXAM...a1b2","keyPos":"1/8","status":200,"stream":false,"ms":4046}
```

### 确认 Key 轮询情况

**每个请求（无论成功或失败）都会记录所用 key**，用于确认轮询是否生效：

| 字段 | 含义 |
|------|------|
| `seq` | 请求序号（插件生命周期内自增） |
| `key` | 该请求实际使用的 API Key（掩码，保留前 8 位 + 末 4 位） |
| `keyPos` | 该 key 在池中的位置，如 `3/8` |
| `model` | 请求的模型（已改写为上游真实 id） |
| `endpoint` | 请求路径，如 `/v1/chat/completions` |
| `status` / `stream` / `ms` | 响应状态码 / 是否流式 / 耗时毫秒 |

观察 `seq` 与 `keyPos` 的对应关系即可确认轮询：

```
seq:1 keyPos:1/8   seq:2 keyPos:2/8   seq:3 keyPos:3/8   ...
seq:9 keyPos:1/8   ← 8 把 key 用完一轮后回到第 1 把
```

相关事件：

| 事件 | 级别 | 说明 |
|------|------|------|
| `Provider config updated` | INFO | config 钩子注册 provider 与模型 |
| `Auth loader initialized` | INFO | 载入 key 池，列出全部掩码 key 及序号 |
| `Request start` | INFO | 请求发出前（含 key / keyPos / model） |
| `Request success` | INFO | 请求成功（含状态码 / 是否流式 / 耗时） |
| `Upstream error response` | WARN | 上游返回错误（含状态码 / 错误码 / 原始消息） |
| `Key rotated` | DEBUG | chat.headers 轮换到下一把 key |
| `Key error count increased` | WARN | key 连续失败计数（≥3 次将被跳过） |
| `Key recovered, error count reset` | INFO | 成功请求后清零该 key 错误计数 |
| `Unauthenticated request blocked` | WARN | 未授权兜底拦截 |
| `First chunk timeout` | WARN | SSE 首块 15s 超时，转 429 |
| `Fetch network error` | ERROR | 网络异常 |
| `Dispose hook called` | INFO | 插件卸载，全局状态清空 |

> 注：`Request start` / `Request success` 为成功路径的主要可见记录。
> 若日志中只有 `Dispose` 一行，说明当次会话未产生任何请求或错误。

## 上游接口（实测记录）

网关同时提供三种 API，插件使用 **Chat Completions**（OpenAI 兼容）：

| API | 路径 | 输入字段 | 认证头 |
|-----|------|----------|--------|
| Chat Completions | `/v1/chat/completions` | `messages` | `Authorization: Bearer` |
| Messages | `/v1/messages` | `messages` | `x-api-key` + `anthropic-version` |
| Responses | `/v1/responses` | `input` | `Authorization: Bearer` |

实测要点：

- **模型 id 大小写敏感**：`Atria-Dawn-Preview`；小写 → `400 invalid_request`
- **流式**：标准 SSE，`reasoning_content` 与 `content` 分离，**无空 `tool_calls` 占位**
- **工具调用**：支持；历史 assistant 缺失/空 `reasoning_content` 时上游仍返回 200（宽容）
- **图片输入**：纯文本模型，图片会被忽略（不报错）
- **限流响应头**：`x-rpm-limit` / `x-rpm-remaining`（成功响应亦携带）
- **短暂性错误**：偶发 `422 upstream_error`，以及偶发请求挂起（45s 无响应）

## 项目结构

```
atriamore/
├── package.json
├── tsconfig.json
├── src/
│   ├── index.ts              # 主入口：auth hook + config hook + chat.headers + 共享 fetch
│   ├── constants.ts          # 提供商配置、模型定义
│   ├── plugin/
│   │   ├── auth.ts           # 凭证读写 ~/.atriamore/auth.json
│   │   └── key-manager.ts    # 多 Key 轮询算法
│   └── utils/
│       └── debug-logger.ts   # 文件日志（无 console.error）
├── scripts/                  # 验证脚本
└── dist/                     # 构建产物
```

## 验证脚本

| 脚本 | 用途 |
|------|------|
| `scripts/check-e2e.mjs` | 端到端（需真实 Key）：别名改写 / 推理分离 / 流式 / 工具 / 多轮兜底 / 未授权 / 错误映射 |
| `scripts/check-key-rotation.mjs` | **多 key 轮询 + 日志归因验证**（可用假 key，不发真实请求；不写凭证文件） |
| `scripts/check-error-mapping.mjs` | 错误映射离线验证（mock Response，无需 Key）：401/403/429/5xx/422/400/408/409/网络异常 |
| `scripts/probe-s5-body.mjs` | 捕获多轮 + tools 的实际请求体并原样重放，判定失败归因 |

```bash
# 错误映射（离线，无需 Key）
node scripts/check-error-mapping.mjs

# 多 key 轮询（用假 key 即可，验证轮询顺序与日志归因）
node scripts/check-key-rotation.mjs keyA,keyB,keyC

# 端到端（需真实 Key）
node scripts/check-e2e.mjs atr_xxxxxxxx

# 未授权兜底（独立进程，验证 401 拦截）
node scripts/check-e2e.mjs dummy --no-auth
```

## 开发

```bash
npm run typecheck   # 类型检查
npm run build       # 构建到 dist/
```

## Key 黑名单（blacklist.json）

KeyManager 会为每个 Key 维护错误计数，持久性错误（`401/403/400` 等）触发 `reportError`（计数 +1），
当 `errorCount >= 3` 时该 Key 被临时拉黑，轮询时自动跳过；请求成功一次即触发 `reportSuccess` 复位计数。
黑名单状态通过快照文件跨重启保留，重启后自动恢复各 Key 的错误计数。

快照文件格式（与插件写盘格式一致）：

```json
{
  "updatedAt": 1759800000000,
  "keys": [
    { "key": "sk-xxx1", "errorCount": 3, "blocked": true },
    { "key": "sk-xxx2", "errorCount": 0, "blocked": false }
  ]
}
```

目标路径：`%USERPROFILE%\.atriamore\blacklist.json`

复制方法（可选）：将项目 `jsonc/blacklist.jsonc` 复制到 `%USERPROFILE%\.atriamore\` 并重命名为
`blacklist.json`。不复制也不影响运行——插件仅在内存中计数，不跨重启保留；快照读写失败仅记录
debug 警告，绝不中断主流程。

## 配置文件

插件运行所需的配置文件位于本仓库 ./jsonc/ 目录下（provider.jsonc 与 models.jsonc）。使用前将这两个文件复制到用户目录 %USERPROFILE%\.atriamore/ 下（缺失时插件启动会 fail fast 报错）。复制后可按需编辑模型列表与提供商信息（支持注释），修改后重启插件生效。