lix/mnote
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: checkpoint pi lab rust integration work

Browse Source
This commit is contained in:
Agent Board
2026-07-11 20:39:18 +08:00
parent d47f6447fd
commit d16eccfe10
39 changed files with 4843 additions and 455 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/pi-mnote/extensions/pi-rust-official/subagent/agents.ts
+13 -1
View File
@@ -5,6 +5,7 @@
import * as fs from "node:fs";
import * as os from "node:os";
import * as path from "node:path";
import { fileURLToPath } from "node:url";
import { parseFrontmatter } from "@mariozechner/pi-coding-agent";
export type AgentScope = "user" | "project" | "both";
@@ -97,9 +98,20 @@ function findNearestProjectAgentsDir(cwd: string): string | null {
export function discoverAgents(cwd: string, scope: AgentScope): AgentDiscoveryResult {
const userDir = path.join(os.homedir(), ".pi", "agent", "agents");
const bundledDir = path.join(path.dirname(fileURLToPath(import.meta.url)), "agents");
const projectAgentsDir = findNearestProjectAgentsDir(cwd);
const bundledAgents = loadAgentsFromDir(bundledDir, "user").map((agent) => ({
...agent,
model: undefined,
}));
const userAgents = scope === "project" ? [] : loadAgentsFromDir(userDir, "user");
const userAgents =
scope === "project"
? []
: [
...bundledAgents,
...loadAgentsFromDir(userDir, "user"),
];
const projectAgents = scope === "user" || !projectAgentsDir ? [] : loadAgentsFromDir(projectAgentsDir, "project");
const agentMap = new Map<string, AgentConfig>();
packages/pi-mnote/extensions/pi-rust-official/subagent/index.ts
+45 -14
View File
@@ -16,6 +16,7 @@ import { spawn } from "node:child_process";
import * as fs from "node:fs";
import * as os from "node:os";
import * as path from "node:path";
import { fileURLToPath } from "node:url";
import type { AgentToolResult } from "@mariozechner/pi-agent-core";
import type { Message } from "@mariozechner/pi-ai";
import { StringEnum } from "@mariozechner/pi-ai";
@@ -27,6 +28,22 @@ import { type AgentConfig, type AgentScope, discoverAgents } from "./agents.js";
const MAX_PARALLEL_TASKS = 8;
const MAX_CONCURRENCY = 4;
const COLLAPSED_ITEM_COUNT = 10;
const EXTENSION_DIR = path.dirname(fileURLToPath(import.meta.url));
const MNOTE_CONTEXT_PATH = path.join(path.dirname(EXTENSION_DIR), "mnote-bridge", "mnote-context.json");
function readMnoteParentModel(): { provider?: string; id?: string } | undefined {
try {
const payload = JSON.parse(fs.readFileSync(MNOTE_CONTEXT_PATH, "utf-8")) as {
modelProvider?: unknown;
modelId?: unknown;
};
const provider = typeof payload.modelProvider === "string" ? payload.modelProvider.trim() : "";
const id = typeof payload.modelId === "string" ? payload.modelId.trim() : "";
return provider || id ? { provider: provider || undefined, id: id || undefined } : undefined;
} catch {
return undefined;
}
}
function formatTokens(count: number): string {
if (count < 1000) return count.toString();
@@ -227,6 +244,7 @@ async function runSingleAgent(
signal: AbortSignal | undefined,
onUpdate: OnUpdateCallback | undefined,
makeDetails: (results: SingleResult[]) => SubagentDetails,
parentModel: { provider?: string; id?: string } | undefined,
): Promise<SingleResult> {
const agent = agents.find((a) => a.name === agentName);
@@ -244,7 +262,12 @@ async function runSingleAgent(
}
const args: string[] = ["--mode", "json", "-p", "--no-session"];
if (agent.model) args.push("--model", agent.model);
if (agent.model) {
args.push("--model", agent.model);
} else {
if (parentModel?.provider) args.push("--provider", parentModel.provider);
if (parentModel?.id) args.push("--model", parentModel.id);
}
if (agent.tools && agent.tools.length > 0) args.push("--tools", agent.tools.join(","));
let tmpPromptDir: string | null = null;
@@ -258,7 +281,7 @@ async function runSingleAgent(
messages: [],
stderr: "",
usage: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0, contextTokens: 0, turns: 0 },
model: agent.model,
model: agent.model ?? (parentModel?.provider && parentModel.id ? `${parentModel.provider}/${parentModel.id}` : undefined),
step,
};
@@ -411,12 +434,17 @@ export default function (pi: ExtensionAPI) {
description: [
"Delegate tasks to specialized subagents with isolated context.",
"Modes: single (agent + task), parallel (tasks array), chain (sequential with {previous} placeholder).",
'Default agent scope is "user" (from ~/.pi/agent/agents).',
'Default agent scope is "user" (bundled agents plus ~/.pi/agent/agents overrides).',
'To enable project-local agents in .pi/agents, set agentScope: "both" (or "project").',
].join(" "),
parameters: SubagentParams,
async execute(_toolCallId, params, signal, onUpdate, ctx) {
const mnoteModel = readMnoteParentModel();
const parentModel = {
provider: mnoteModel?.provider ?? process.env.MNOTE_PI_MODEL_PROVIDER ?? ctx.model?.provider,
id: mnoteModel?.id ?? process.env.MNOTE_PI_MODEL_ID ?? ctx.model?.id,
};
const agentScope: AgentScope = params.agentScope ?? "user";
const discovery = discoverAgents(ctx.cwd, agentScope);
const agents = discovery.agents;
@@ -504,10 +532,11 @@ export default function (pi: ExtensionAPI) {
taskWithContext,
step.cwd,
i + 1,
signal,
chainUpdate,
makeDetails("chain"),
);
signal,
chainUpdate,
makeDetails("chain"),
parentModel,
);
results.push(result);
const isError =
@@ -585,9 +614,10 @@ export default function (pi: ExtensionAPI) {
allResults[index] = partial.details.results[0];
emitParallelUpdate();
}
},
makeDetails("parallel"),
);
},
makeDetails("parallel"),
parentModel,
);
allResults[index] = result;
emitParallelUpdate();
return result;
@@ -618,10 +648,11 @@ export default function (pi: ExtensionAPI) {
params.task,
params.cwd,
undefined,
signal,
onUpdate,
makeDetails("single"),
);
signal,
onUpdate,
makeDetails("single"),
parentModel,
);
const isError = result.exitCode !== 0 || result.stopReason === "error" || result.stopReason === "aborted";
if (isError) {
const errorMsg =