mirror of
https://github.com/deepseek-ai/deepseek-harness
synced 2026-08-15 21:04:50 +00:00
# Conflicts: # docs/cordis-catalog/services.md # docs/event-producer-consumer.md # examples/acp-agent/tests/snapshots/advanced-toolchain/session.1.jsonl # examples/acp-agent/tests/snapshots/advanced-toolchain/session.2.jsonl # examples/acp-agent/tests/snapshots/advanced-toolchain/session.jsonl # examples/acp-agent/tests/snapshots/both-mode-turn/session.jsonl # examples/acp-agent/tests/snapshots/code-mode-turn/session.jsonl # examples/acp-agent/tests/snapshots/permission-switching/session.jsonl # examples/acp-agent/tests/snapshots/skill-load/session.jsonl # examples/acp-agent/tests/snapshots/text-turn/session.jsonl
4.5 KiB
4.5 KiB
实操手册:添加 LLM 适配器
English | 中文
如何接入一个新的模型提供方。参考实现:packages/llm/llm-deepseek(手写 HTTP/SSE)与 packages/llm/llm-pi-ai(封装 LLM 库)。请先阅读 packages/llm/llm/src/types.ts 中的 StreamChunk 文档——它记录了两个适配器都经过验证的协议约定。
基本形态
class MyAdapter extends LlmAdapter {
async * stream(options: GenerateOptions): AsyncIterable<StreamChunk> { … }
}
export const name = 'llm-myprovider'
export const inject = ['llm']
export const Config: z<Config> = z.object({ apiKey: z.string(), … })
export function apply(ctx: Context, config: Config) {
ctx.llm.registerAdapter(['my-provider'], new MyAdapter(…))
}
注册基于副作用(HMR 安全);每个提供方路由仅对应一个适配器,重复注册会抛出异常,多路由注册要么全部成功,要么全部失败。options.provider 用于选择适配器,options.model 是提供方模型 ID,因此动态模型目录适配器无需重新配置生命周期即可提供新模型。密钥采用 Cordis 原生方式管理:schemastery Config 带环境变量回退,通过 cordis.yml 的 !!js process.env.MY_KEY 注入。代码中禁止临时读取密钥文件。
协议义务(两个实现共同验证的契约)
- 在
finish之前发出usage;finish之后不再发出任何内容。稳健做法:缓冲 finish/usage 直到提供方的流结束标记,再统一 flush(可处理提供方在末尾发送仅含 usage 的分片的情况)。 - 工具调用的
arguments全程为原始 JSON 字符串;流式片段以argumentsDelta发送。如果你的提供方返回已解析的对象,请在block-end时重新 stringify。 - 按首次出现的流顺序分配块
index;同一个块的每次 delta 复用该 index。 - 错误有且仅有两条合法路径:从
stream()抛出(传输与协议故障——使用带稳定 code 的LlmError),或以finish {kind: 'error' | 'aborted'}结束流(提供方带内故障)。消费方两者都处理;按故障类别选择路径并加以文档化。 - 遵守
options.signal(将其传递给 fetch 或你的 SDK)。 - 如果
GenerateOptions中某个字段你的提供方无法支持(例如提供方不支持 stop sequences 时收到stop列表):抛出LlmError(..., 'UNSUPPORTED'),而非静默丢弃。 - 如果提供方在后续调用中需要响应 ID、签名或其他原生元数据,请将其最小无损 JSON 投影作为
finish.replayState发出。重建历史时验证该状态。只有历史提供方路由和目标提供方路由当前由完全相同的适配器实例拥有时,LlmService才会传递该状态;由适配器决定同模型、跨模型或跨提供方恢复是否合法。状态缺失时,切勿仅根据提供方/模型名称推断原生回放。
提供方特有的请求旋钮(thinking 模式、effort 级别)放在适配器的 Config 中,而非 GenerateOptions 中——核心词汇保持提供方无关。
经验证有效的结构
将适配器拆分为可测试的阶段(llm-deepseek 的布局):协议格式(wire format)类型(types.ts,豁免覆盖率)→ 请求序列化器 → SSE/传输解析器 → 分片转换状态机 → 一个将它们串联的薄适配器类。每个阶段配备独立的单元测试套件。
测试
- 单元测试:mock 提供方,而非 harness。 用脚本化的
node:http服务器模拟提供方的协议格式,覆盖正常路径、所有错误状态码、畸形载荷、连接提前关闭和中止——无需网络,且能满足 100% 逐文件覆盖率门禁。对基于 SDK 的适配器同样适用(将 SDK 的 baseURL 指向 mock 服务器)。 - 恶意分帧测试。 在任意字节位置(包括 UTF-8 字符中间)切割流载荷——真实网络环境正是如此。
- E2E:
tests/*.e2e.ts,通过pnpm run test:e2e运行,以describe.skipIf(!process.env.MY_KEY)守卫,确保无密钥的 CI 保持绿色。覆盖具有代表性的模型/提供方/API 系列以及你映射的每种提供方模式、一次包含后续轮次(历史中带工具结果)的工具调用往返,以及仅做宽松断言(子串/结构匹配、有界的 maxTokens——真实模型是非确定性的)。 - 在
knip.json中注册 e2e 文件模式(per-workspaceentry覆盖),否则 knip 会将其标记为未使用。