xiaoju
26cf51366f
- Add buildThreadContext(headHash, cas) to workflow-runtime
- Expand extract phase to return { meta, contentPayload, refs[] }
- Add parseCasThreadNode() to workflow-cas for node type parsing
- Update createWorkflow to write ContentMerkleNode with artifact refs
- Tests: 4 pass (build-context + extract-refs)
- Biome format pass on all files
Refs #155, closes #158
小橘 <xiaoju@shazhou.work>
73 lines
2.3 KiB
TypeScript
73 lines
2.3 KiB
TypeScript
import { afterEach, describe, expect, test } from "bun:test";
|
|
import { mkdtemp, rm } from "node:fs/promises";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { createCasStore } from "@uncaged/workflow-cas";
|
|
import { type ExtractContext, START } from "@uncaged/workflow-runtime";
|
|
import * as z from "zod/v4";
|
|
|
|
import { createExtract } from "../src/extract/extract-fn.js";
|
|
|
|
function installPlainJsonExtractMock(meta: Record<string, unknown>): () => void {
|
|
const origFetch = globalThis.fetch;
|
|
const mockFetch = async (): Promise<Response> =>
|
|
new Response(
|
|
JSON.stringify({
|
|
choices: [{ message: { content: JSON.stringify(meta) } }],
|
|
}),
|
|
{ status: 200, headers: { "Content-Type": "application/json" } },
|
|
);
|
|
globalThis.fetch = Object.assign(mockFetch, {
|
|
preconnect: origFetch.preconnect.bind(origFetch),
|
|
}) as typeof fetch;
|
|
return () => {
|
|
globalThis.fetch = origFetch;
|
|
};
|
|
}
|
|
|
|
describe("createExtract — ExtractResult shape", () => {
|
|
let restoreFetch: (() => void) | null = null;
|
|
|
|
afterEach(() => {
|
|
restoreFetch?.();
|
|
restoreFetch = null;
|
|
});
|
|
|
|
test("returns meta, contentPayload, and refs[]", async () => {
|
|
restoreFetch = installPlainJsonExtractMock({ confidence: 0.9 });
|
|
|
|
const dir = await mkdtemp(join(tmpdir(), "wf-extract-refs-"));
|
|
try {
|
|
const cas = createCasStore(join(dir, "cas"));
|
|
const extract = createExtract(
|
|
{ baseUrl: "http://127.0.0.1:9", apiKey: "key", model: "m" },
|
|
{ cas },
|
|
);
|
|
|
|
const schema = z.object({ confidence: z.number() });
|
|
const ctx: ExtractContext = {
|
|
threadId: "01THREADTESTAAAAAAAAAAAAAA",
|
|
depth: 0,
|
|
start: {
|
|
role: START,
|
|
content: "task text",
|
|
meta: { maxRounds: 10 },
|
|
timestamp: 100,
|
|
},
|
|
steps: [],
|
|
currentRole: { name: "analyst", systemPrompt: "be precise" },
|
|
agentContent: "model says hello",
|
|
};
|
|
|
|
const out = await extract(schema, "extract fields", ctx);
|
|
|
|
expect(out.meta).toEqual({ confidence: 0.9 });
|
|
expect(out.contentPayload).toBe("model says hello");
|
|
expect(Array.isArray(out.refs)).toBe(true);
|
|
expect(out.refs).toEqual([]);
|
|
} finally {
|
|
await rm(dir, { recursive: true, force: true });
|
|
}
|
|
});
|
|
});
|