feat: integrate Pi backend and Pi Web

2026-10-03 02:59:35 +00:00 · 2026-07-30 19:37:53 +08:00
commit 7392ab9dd7
1390 changed files with 337197 additions and 0 deletions
@@ -0,0 +1,79 @@
+import { homedir } from "node:os";
+import { join } from "node:path";
+import { createModels } from "@earendil-works/pi-ai";
+import { cloudflareAIGatewayProvider } from "@earendil-works/pi-ai/providers/cloudflare-ai-gateway";
+import { openaiProvider } from "@earendil-works/pi-ai/providers/openai";
+import { NodeExecutionEnv } from "../../src/harness/env/nodejs.ts";
+import { InMemorySessionStorage } from "../../src/harness/session/memory-storage.ts";
+import {
+	AgentHarness,
+	createBashTool,
+	createEditTool,
+	createReadTool,
+	createWriteTool,
+	formatSkillsForSystemPrompt,
+	loadSourcedPromptTemplates,
+	loadSourcedSkills,
+	type PromptTemplate,
+	Session,
+	type Skill,
+} from "../../src/index.ts";
+
+type Source = { type: "project" | "user" | "path"; dir: string };
+type SourcedSkill = Skill & { source: Source };
+type SourcedPromptTemplate = PromptTemplate & { source: Source };
+
+const env = new NodeExecutionEnv({ cwd: process.cwd() });
+
+const source = (type: Source["type"], dir: string) => ({ path: dir, source: { type, dir } });
+const { skills: sourcedSkills } = await loadSourcedSkills<Source, SourcedSkill>(
+	env,
+	[
+		source("project", join(env.cwd, ".pi/skills")),
+		source("user", join(homedir(), ".pi/agent/skills")),
+		source("path", join(env.cwd, "../../../pi-skills")),
+	],
+	(skill, source) => ({ ...skill, source }),
+);
+const { promptTemplates: sourcedPromptTemplates } = await loadSourcedPromptTemplates<Source, SourcedPromptTemplate>(
+	env,
+	[source("project", join(env.cwd, ".pi/prompts")), source("user", join(homedir(), ".pi/agent/prompts"))],
+	(promptTemplate, source) => ({ ...promptTemplate, source }),
+);
+
+const models = createModels();
+models.setProvider(openaiProvider());
+models.setProvider(cloudflareAIGatewayProvider());
+const model = models.getModel("openai", "gpt-5.5");
+// const model = models.getModel("cloudflare-ai-gateway", "claude-haiku-4-5");
+if (!model) {
+	console.log("Model not found");
+	process.exit(-1);
+}
+
+const session = new Session(new InMemorySessionStorage());
+const agent = new AgentHarness({
+	session,
+	models,
+	model,
+	thinkingLevel: "low",
+	tools: [createReadTool(), createWriteTool(), createEditTool(), createBashTool()],
+	toolContext: { env },
+	systemPrompt: ({ resources }) =>
+		[
+			"You are a helpful assistant.",
+			formatSkillsForSystemPrompt(resources.skills ?? []),
+			`Current working directory: ${env.cwd}`,
+		]
+			.filter((part) => part.length > 0)
+			.join("\n\n"),
+	resources: {
+		promptTemplates: sourcedPromptTemplates.map(({ promptTemplate }) => promptTemplate),
+		skills: sourcedSkills.map(({ skill }) => skill),
+	},
+});
+
+const response = await agent.prompt(
+	"What skills do you have? Any duplicates? Also use bash to get the current date and time, then read README.md and tell me what this project is about.",
+);
+console.log(response);