xiaoju
cfebd07124
Expose the store created during context build on AgentContext so agents reuse the same in-memory cache instead of opening a second store. Co-authored-by: Cursor <cursoragent@cursor.com>
29 lines
801 B
TypeScript
29 lines
801 B
TypeScript
#!/usr/bin/env bun
|
|
// Mock agent for smoke testing
|
|
import { bootstrap, type JSONSchema, putSchema } from "@uncaged/json-cas";
|
|
import { createAgent } from "../packages/uwf-agent-kit/src/index.js";
|
|
|
|
const MOCK_RAW_OUTPUT_SCHEMA: JSONSchema = {
|
|
title: "mock-raw-output",
|
|
type: "object",
|
|
required: ["text"],
|
|
properties: {
|
|
text: { type: "string" },
|
|
},
|
|
additionalProperties: false,
|
|
};
|
|
|
|
const agent = createAgent({
|
|
name: "mock",
|
|
run: async (ctx) => {
|
|
const output = `Mock output for role ${ctx.role}: task was "${ctx.prompt}"`;
|
|
const { store } = ctx;
|
|
await bootstrap(store);
|
|
const schemaHash = await putSchema(store, MOCK_RAW_OUTPUT_SCHEMA);
|
|
const detailHash = await store.put(schemaHash, { text: output });
|
|
return { output, detailHash };
|
|
},
|
|
});
|
|
|
|
await agent();
|