xiaoju
e5e6de2fad
- Replace bun:test with vitest across all packages - Replace bun build with esbuild - Replace bun:sqlite with better-sqlite3 - Fix OCAS Store API: store.put/get → store.cas.put/get - Fix vitest vi.mock hoisting (vi.hoisted) - Add pnpm-workspace.yaml and pnpm-lock.yaml - Update all package.json test/build scripts WIP: 8 failures remain in agent-hermes (bun engines check + sqlite migration) Refs #26
43 lines
1.3 KiB
TypeScript
43 lines
1.3 KiB
TypeScript
import { describe, expect, test } from 'vitest';
|
|
import type { WorkflowConfig } from "@united-workforce/protocol";
|
|
import { resolveExtractModelAlias } from "../src/extract.js";
|
|
|
|
function baseConfig(overrides: Partial<WorkflowConfig> = {}): WorkflowConfig {
|
|
return {
|
|
providers: {},
|
|
models: {
|
|
sonnet: { provider: "openrouter", name: "anthropic/claude-sonnet-4" },
|
|
"gpt4o-mini": { provider: "openai", name: "gpt-4o-mini" },
|
|
},
|
|
agents: {},
|
|
defaultAgent: "hermes",
|
|
agentOverrides: null,
|
|
defaultModel: "sonnet",
|
|
modelOverrides: null,
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
describe("resolveExtractModelAlias", () => {
|
|
test("uses modelOverrides.extract when set", () => {
|
|
const config = baseConfig({
|
|
modelOverrides: { extract: "gpt4o-mini" },
|
|
});
|
|
expect(resolveExtractModelAlias(config)).toBe("gpt4o-mini");
|
|
});
|
|
|
|
test("falls back to models.extract alias when present", () => {
|
|
const config = baseConfig({
|
|
models: {
|
|
extract: { provider: "openai", name: "gpt-4o-mini" },
|
|
sonnet: { provider: "openrouter", name: "anthropic/claude-sonnet-4" },
|
|
},
|
|
});
|
|
expect(resolveExtractModelAlias(config)).toBe("extract");
|
|
});
|
|
|
|
test("falls back to defaultModel", () => {
|
|
expect(resolveExtractModelAlias(baseConfig())).toBe("sonnet");
|
|
});
|
|
});
|