65 lines
2.5 KiB
TypeScript
65 lines
2.5 KiB
TypeScript
import { getModel } from "@earendil-works/pi-ai";
|
|
import { describe, expect, it } from "vitest";
|
|
import { NodeExecutionEnv } from "../../src/harness/execution-env.js";
|
|
import { createAgentHarness, createSession } from "../../src/harness/factory.js";
|
|
import { InMemorySessionStorage } from "../../src/harness/session/storage/memory.js";
|
|
|
|
describe("harness factories", () => {
|
|
it("creates sessions from storage", async () => {
|
|
const storage = new InMemorySessionStorage({
|
|
metadata: { id: "session-1", createdAt: "2026-01-01T00:00:00.000Z" },
|
|
});
|
|
const session = createSession(storage);
|
|
expect(session.getStorage()).toBe(storage);
|
|
expect(await session.getMetadata()).toEqual({ id: "session-1", createdAt: "2026-01-01T00:00:00.000Z" });
|
|
});
|
|
|
|
it("creates agent harnesses", () => {
|
|
const session = createSession(new InMemorySessionStorage());
|
|
const env = new NodeExecutionEnv({ cwd: process.cwd() });
|
|
const initialModel = getModel("anthropic", "claude-sonnet-4-5");
|
|
const harness = createAgentHarness({
|
|
env,
|
|
session,
|
|
model: initialModel,
|
|
systemPrompt: "You are helpful.",
|
|
steeringMode: "all",
|
|
followUpMode: "all",
|
|
});
|
|
expect(harness.env).toBe(env);
|
|
expect(harness.agent.state.model).toBe(initialModel);
|
|
expect(harness.steeringMode).toBe("all");
|
|
expect(harness.followUpMode).toBe("all");
|
|
harness.steeringMode = "one-at-a-time";
|
|
harness.followUpMode = "one-at-a-time";
|
|
expect(harness.agent.steeringMode).toBe("one-at-a-time");
|
|
expect(harness.agent.followUpMode).toBe("one-at-a-time");
|
|
});
|
|
|
|
it("updates and reads concrete resources", async () => {
|
|
const session = createSession(new InMemorySessionStorage());
|
|
const env = new NodeExecutionEnv({ cwd: process.cwd() });
|
|
const model = getModel("anthropic", "claude-sonnet-4-5");
|
|
const harness = createAgentHarness({ env, session, model });
|
|
const skill = {
|
|
name: "inspect",
|
|
description: "Inspect things",
|
|
content: "Use inspection tools.",
|
|
filePath: "/skills/inspect/SKILL.md",
|
|
};
|
|
const resources = { skills: [skill], promptTemplates: [{ name: "review", content: "Review $1" }] };
|
|
const updates: unknown[] = [];
|
|
harness.subscribe((event) => {
|
|
if (event.type === "resources_update") updates.push(event.resources);
|
|
});
|
|
|
|
await harness.setResources(resources);
|
|
const resolved = harness.getResources();
|
|
|
|
expect(updates).toEqual([resources]);
|
|
expect(resolved).toEqual(resources);
|
|
expect(resolved.skills).not.toBe(resources.skills);
|
|
expect(resolved.promptTemplates).not.toBe(resources.promptTemplates);
|
|
});
|
|
});
|