mirror of
https://github.com/badlogic/pi-mono.git
synced 2026-11-05 23:37:25 +00:00
837d8a16bd
Submissions with request-ID deduplication and handles, the pi.live run control and presentation document, the built-in pi.generation task (system prompt preparation, streamed durable partials, retries, deferred polling, abort), typed entry tokens, stream options and retry policy, TaskRuntime reads, Tx.settleSubmission, registry conversation setups, Storage.scanSubmissions, Harness.inspect, and byTaskId attribution. Waiting calls enable scheduling. The scratch tour is split into test/examples.
54 lines
2 KiB
TypeScript
54 lines
2 KiB
TypeScript
// Open a Harness with a registry.
|
|
// Run from packages/durable:
|
|
// node --conditions=source --experimental-strip-types test/examples/06-harness.ts
|
|
import { BACKGROUND_CONTEXT } from "@earendil-works/chord/context";
|
|
import { createModels, Type } from "@earendil-works/pi-ai";
|
|
import {
|
|
ConversationConfig,
|
|
createRegistry,
|
|
defineDoc,
|
|
Harness,
|
|
MemoryStorage,
|
|
type ToolRegistration,
|
|
} from "../../src/index.ts";
|
|
|
|
const context = BACKGROUND_CONTEXT;
|
|
|
|
// A Harness is a Session plus conversation handles and durable tasks. Extension
|
|
// code (tools, hooks, tasks, system prompt sections, conversation setups) lives
|
|
// in a registry the application owns. Nothing in the registry is saved; it is
|
|
// this process's code.
|
|
const registry = createRegistry();
|
|
const read: ToolRegistration = {
|
|
name: "read",
|
|
description: "Read a file",
|
|
parameters: Type.Object({ path: Type.String() }),
|
|
execute: async (args) => ({ content: [{ type: "text", text: `read ${JSON.stringify(args)}` }] }),
|
|
};
|
|
registry.tools.add(read);
|
|
|
|
// `models` is pi-ai's model access; generation calls models through it.
|
|
const harness = await Harness.open(new MemoryStorage(), { models: createModels(), registry }, context);
|
|
|
|
const Notes = defineDoc<{ text: string }>({
|
|
kind: "example.notes",
|
|
version: 1,
|
|
scope: "conversation",
|
|
history: "rewindable",
|
|
fork: "asOf",
|
|
initial: () => ({ text: "" }),
|
|
});
|
|
|
|
// The root conversation always has ID 1. The first root() call creates it,
|
|
// its built-in documents, and whatever `init` writes, all in one commit. Later
|
|
// calls, including after a restart, return it and ignore `init`.
|
|
const root = await harness.root(context, {
|
|
init: async (tx, rootId) => {
|
|
(await tx.doc(Notes, rootId)).text = "root notes";
|
|
(await tx.doc(ConversationConfig, rootId)).thinkingLevel = "low";
|
|
},
|
|
});
|
|
console.log("root:", root.id, await harness.snapshot(Notes, root.id, context));
|
|
console.log("root config:", await harness.snapshot(ConversationConfig, root.id, context));
|
|
|
|
await harness.close(context);
|