Files
UHD_Media_Tool/packages/contracts/test/pi.test.ts
T
rootandClaude Opus 5.5 3e37ac8ae7 refactor(agent): runAgent uses Agent upstreams only; deprecate legacy settings.pi fields
pi.provider / baseUrl / model / thinkingLevel / apiKeySecretRef are now
DEPRECATED_SETTINGS_FIELDS: kept in the schema for stored JSON, dropped
from PATCH /settings and from the settings sections, never read by the
server. runAgent fails with a clear message when no upstream is set.
The agent demo gate checks the default upstream provider; e2e seeds a
fake upstream as default. OpenAPI summaries say Agent upstream.

Co-Authored-By: Claude Opus 5.5 <[email protected]>
2026-09-28 13:45:40 +08:00

78 lines
3.5 KiB
TypeScript

import { describe, expect, it } from "vitest";
import * as C from "../src/index.ts";
// W7-A Pi 上游合同:跨字段约束的正反例、设置兼容。
describe("PiUpstreamCreate", () => {
const ok = (v: unknown) => C.PiUpstreamCreate.safeParse(v).success;
it("正例:api_key / oauth / none / 自定义端点", () => {
expect(ok({ name: "OpenAI 主号", provider: "openai", authKind: "api_key", apiKey: "sk-1" })).toBe(true);
expect(ok({ name: "Codex", provider: "openai-codex", authKind: "oauth", thinkingLevel: "max" })).toBe(true);
expect(
ok({
name: "Bedrock",
provider: "amazon-bedrock",
authKind: "none",
extra: { env: { AWS_REGION: "us-east-1" } },
}),
).toBe(true);
expect(
ok({
name: "本地 vLLM",
provider: "openai-compatible",
authKind: "none",
baseUrl: "http://llm.local/v1",
defaultModel: "qwen3",
extra: { api: "openai-completions", compat: { supportsDeveloperRole: false } },
}),
).toBe(true);
expect(ok({ name: "fake", provider: "fake", authKind: "none" })).toBe(true);
});
it("反例:未知 provider、oauth 带 apiKey / baseUrl、自定义端点缺 baseUrl 或模型、extra 多余字段、名字为空", () => {
expect(ok({ name: "x", provider: "google-gemini-cli", authKind: "oauth" })).toBe(false);
expect(ok({ name: "x", provider: "anthropic", authKind: "oauth", apiKey: "sk" })).toBe(false);
expect(ok({ name: "x", provider: "anthropic", authKind: "oauth", baseUrl: "https://p.local" })).toBe(false);
expect(ok({ name: "x", provider: "openai-compatible", authKind: "api_key", defaultModel: "m" })).toBe(false);
expect(ok({ name: "x", provider: "openai-compatible", authKind: "api_key", baseUrl: "http://a/v1" })).toBe(false);
expect(ok({ name: "x", provider: "openai", authKind: "api_key", extra: { secret: "sk" } })).toBe(false);
expect(ok({ name: " ", provider: "openai", authKind: "api_key" })).toBe(false);
expect(ok({ name: "x", provider: "openai", authKind: "token" })).toBe(false);
});
it("provider 枚举 = 41 个内置 + openai-compatible + fake", () => {
expect(C.PI_BUILTIN_PROVIDER_IDS).toHaveLength(41);
expect(C.PiProviderId.options).toHaveLength(43);
});
});
describe("OAuth 请求", () => {
it("start/complete 的边界", () => {
expect(C.PiOAuthStartRequest.safeParse({}).success).toBe(true);
expect(C.PiOAuthStartRequest.safeParse({ mode: "popup" }).success).toBe(false);
expect(C.PiOAuthCompleteRequest.safeParse({ sessionId: "s" }).success).toBe(true);
expect(C.PiOAuthCompleteRequest.safeParse({ sessionId: "s", waitMs: 120_000 }).success).toBe(false);
expect(C.PiOAuthCompleteRequest.safeParse({ pastedUrlOrCode: "x" }).success).toBe(false);
});
});
describe("settings.pi 兼容", () => {
it("缺省 defaultUpstreamId = null;PATCH 不能直接改 defaultUpstreamId,旧字段被丢弃", () => {
expect(C.DEFAULT_SETTINGS.pi.defaultUpstreamId).toBeNull();
expect(C.GlobalSettings.safeParse(C.DEFAULT_SETTINGS).success).toBe(true);
const patch = C.GlobalSettingsUpdate.parse({
pi: {
defaultUpstreamId: "7d3f7f5e-8a8c-4b8e-9f0a-1b2c3d4e5f60",
provider: "fake",
apiKey: "sk-x",
model: "m",
contextBudget: 64_000,
},
});
expect(patch.pi).toEqual({ contextBudget: 64_000 });
const current = { ...C.DEFAULT_SETTINGS, pi: { ...C.DEFAULT_SETTINGS.pi, defaultUpstreamId: "keep-me" } };
expect(C.mergeSettings(current as never, patch).pi.defaultUpstreamId).toBe("keep-me");
});
});