@femtomc/mu-agent 26.2.33

package/dist/pi_sdk_backend.js

@@ -0,0 +1,152 @@
+import { existsSync } from "node:fs";
+import { mkdir, open } from "node:fs/promises";
+import { basename, dirname, join } from "node:path";
+import { getModels, getProviders } from "@mariozechner/pi-ai";
+import { AuthStorage, createAgentSession, createBashTool, createEditTool, createReadTool, createWriteTool, DefaultResourceLoader, SessionManager, SettingsManager, } from "@mariozechner/pi-coding-agent";
+import { piStreamHasError } from "./pi_backend.js";
+/**
+ * Resolve a bare model ID (e.g. "gpt-5.3-codex") to a pi-ai Model object.
+ *
+ * When multiple providers offer the same model ID, prefer providers that
+ * have auth configured (env var, OAuth, or stored API key).
+ */
+function resolveModel(modelId, authStorage, providerConstraint) {
+    if (providerConstraint) {
+        const providers = getProviders();
+        if (!providers.includes(providerConstraint)) {
+            throw new Error(`Unknown provider "${providerConstraint}". Available: ${providers.join(", ")}`);
+        }
+        const models = getModels(providerConstraint);
+        return models.find((m) => m.id === modelId);
+    }
+    let fallback;
+    for (const provider of getProviders()) {
+        const models = getModels(provider);
+        const match = models.find((m) => m.id === modelId);
+        if (!match)
+            continue;
+        // Prefer providers that have auth configured.
+        if (authStorage.hasAuth(provider)) {
+            return match;
+        }
+        // Keep first match as fallback.
+        if (!fallback) {
+            fallback = match;
+        }
+    }
+    return fallback;
+}
+/**
+ * In-process backend using the pi SDK.
+ *
+ * Replaces subprocess spawning of the `pi` CLI with direct use of
+ * `createAgentSession` from `@mariozechner/pi-coding-agent`.
+ */
+export class PiSdkBackend {
+    async run(opts) {
+        const authStorage = new AuthStorage();
+        const model = resolveModel(opts.model, authStorage, opts.provider);
+        if (!model) {
+            const scope = opts.provider ? ` in provider "${opts.provider}"` : "";
+            throw new Error(`Model "${opts.model}" not found${scope} in pi-ai registry.`);
+        }
+        const settingsManager = SettingsManager.inMemory();
+        const resourceLoader = createMuResourceLoader({
+            cwd: opts.cwd,
+            systemPrompt: opts.systemPrompt,
+            settingsManager,
+        });
+        await resourceLoader.reload();
+        const tools = [
+            createBashTool(opts.cwd),
+            createReadTool(opts.cwd),
+            createWriteTool(opts.cwd),
+            createEditTool(opts.cwd),
+        ];
+        const sessionOpts = {
+            cwd: opts.cwd,
+            model,
+            thinkingLevel: opts.thinking,
+            tools,
+            sessionManager: SessionManager.inMemory(opts.cwd),
+            settingsManager,
+            resourceLoader,
+            authStorage,
+        };
+        const { session } = await createAgentSession(sessionOpts);
+        let teeFh = null;
+        try {
+            if (opts.teePath) {
+                await mkdir(dirname(opts.teePath), { recursive: true });
+                teeFh = await open(opts.teePath, "w");
+            }
+            // Bind extensions (required for tools to work in print mode).
+            await session.bindExtensions({
+                commandContextActions: {
+                    waitForIdle: () => session.agent.waitForIdle(),
+                    newSession: async () => ({ cancelled: true }),
+                    fork: async () => ({ cancelled: true }),
+                    navigateTree: async () => ({ cancelled: true }),
+                    switchSession: async () => ({ cancelled: true }),
+                    reload: async () => { },
+                },
+                onError: () => { },
+            });
+            let sawError = false;
+            const DELTA_TYPES = new Set(["thinking_delta", "toolcall_delta", "text_delta"]);
+            // Subscribe to events — serialize to JSONL for tee and error detection.
+            const unsub = session.subscribe((event) => {
+                const line = JSON.stringify(event);
+                if (piStreamHasError(line)) {
+                    sawError = true;
+                }
+                // onLine gets everything (CLI needs deltas for live rendering).
+                opts.onLine?.(line);
+                // Tee file: skip streaming deltas (they carry the full accumulated
+                // message state on every token, causing quadratic log growth).
+                // Structural events (message_start/end, turn_start/end, tool_execution_*,
+                // thinking_start/end, toolcall_start/end) are kept.
+                if (teeFh) {
+                    const aType = event?.assistantMessageEvent?.type;
+                    if (!DELTA_TYPES.has(aType)) {
+                        teeFh.write(`${line}\n`).catch(() => { });
+                    }
+                }
+            });
+            try {
+                await session.prompt(opts.prompt, { expandPromptTemplates: false });
+            }
+            finally {
+                unsub();
+            }
+            return sawError ? 1 : 0;
+        }
+        finally {
+            session.dispose();
+            if (teeFh) {
+                await teeFh.close();
+            }
+        }
+    }
+}
+export function createMuResourceLoader(opts) {
+    const skillPaths = new Set();
+    for (const p of opts.additionalSkillPaths ?? []) {
+        skillPaths.add(p);
+    }
+    // If a repo has a top-level `skills/` dir (like workshop/), load it.
+    const repoSkills = join(opts.cwd, "skills");
+    if (existsSync(repoSkills)) {
+        skillPaths.add(repoSkills);
+    }
+    return new DefaultResourceLoader({
+        cwd: opts.cwd,
+        agentDir: opts.agentDir,
+        settingsManager: opts.settingsManager ?? SettingsManager.inMemory(),
+        additionalSkillPaths: [...skillPaths],
+        systemPromptOverride: (_base) => opts.systemPrompt,
+        agentsFilesOverride: (base) => ({
+            agentsFiles: base.agentsFiles.filter((f) => basename(f.path) === "AGENTS.md"),
+        }),
+    });
+}

package/dist/prompt.d.ts

@@ -0,0 +1,17 @@
+import type { Issue } from "@femtomc/mu-core";
+export type PromptMeta = Record<string, unknown>;
+export declare function splitFrontmatter(text: string): {
+    meta: PromptMeta;
+    body: string;
+};
+export declare function extractDescription(meta: PromptMeta, body: string): {
+    description: string;
+    source: string;
+};
+export declare function readPromptMeta(path: string): Promise<PromptMeta>;
+export declare function buildRoleCatalog(repoRoot: string): Promise<string>;
+export declare function renderPromptTemplate(path: string, issue: Pick<Issue, "id" | "title" | "body">, opts?: {
+    repoRoot?: string;
+}): Promise<string>;
+export declare function resolvePromptPath(repoRoot: string, promptPath: string): string;
+//# sourceMappingURL=prompt.d.ts.map

package/dist/prompt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt.d.ts","sourceRoot":"","sources":["../src/prompt.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AAE9C,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAyCjD,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,GAAG;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CA6BjF;AAYD,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,GAAG;IAAE,WAAW,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAW1G;AAED,wBAAsB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAItE;AAMD,wBAAsB,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAwCxE;AAED,wBAAsB,oBAAoB,CACzC,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,GAAG,OAAO,GAAG,MAAM,CAAC,EAC3C,IAAI,GAAE;IAAE,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAO,GAC9B,OAAO,CAAC,MAAM,CAAC,CAmBjB;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM,CAK9E"}

package/dist/prompt.js

@@ -0,0 +1,153 @@
+import { readdir, readFile } from "node:fs/promises";
+import { isAbsolute, join, relative } from "node:path";
+function stripQuotes(s) {
+    const trimmed = s.trim();
+    if (trimmed.length >= 2) {
+        const first = trimmed[0];
+        const last = trimmed[trimmed.length - 1];
+        if ((first === `"` && last === `"`) || (first === `'` && last === `'`)) {
+            return trimmed.slice(1, -1);
+        }
+    }
+    return trimmed;
+}
+function parseSimpleYamlFrontmatter(text) {
+    // We only need a small subset: flat `key: value` mappings.
+    // If parsing fails, we return {} (mirrors Python behavior).
+    const out = {};
+    const lines = text.split(/\r?\n/);
+    for (const rawLine of lines) {
+        const line = rawLine.trim();
+        if (line.length === 0) {
+            continue;
+        }
+        if (line.startsWith("#")) {
+            continue;
+        }
+        const idx = line.indexOf(":");
+        if (idx <= 0) {
+            continue;
+        }
+        const key = line.slice(0, idx).trim();
+        if (!key) {
+            continue;
+        }
+        const value = stripQuotes(line.slice(idx + 1));
+        out[key] = value;
+    }
+    return out;
+}
+export function splitFrontmatter(text) {
+    const lines = text.split(/\r?\n/);
+    if (lines.length === 0 || lines[0]?.trim() !== "---") {
+        return { meta: {}, body: text };
+    }
+    // Find the terminating `---` line.
+    let endIdx = -1;
+    for (let i = 1; i < lines.length; i++) {
+        if (lines[i]?.trim() === "---") {
+            endIdx = i;
+            break;
+        }
+    }
+    if (endIdx < 0) {
+        return { meta: {}, body: text };
+    }
+    try {
+        const metaText = lines.slice(1, endIdx).join("\n");
+        const body = lines
+            .slice(endIdx + 1)
+            .join("\n")
+            .replace(/^\n+/, "");
+        const meta = parseSimpleYamlFrontmatter(metaText);
+        return { meta, body };
+    }
+    catch {
+        return { meta: {}, body: text };
+    }
+}
+function firstNonEmptyLine(text) {
+    for (const line of text.split(/\r?\n/)) {
+        const stripped = line.trim();
+        if (stripped) {
+            return stripped;
+        }
+    }
+    return "";
+}
+export function extractDescription(meta, body) {
+    const raw = meta.description;
+    const desc = typeof raw === "string" ? raw.trim() : "";
+    if (desc) {
+        return { description: desc, source: "frontmatter" };
+    }
+    const bodyDesc = firstNonEmptyLine(body);
+    if (bodyDesc) {
+        return { description: bodyDesc, source: "body" };
+    }
+    return { description: "", source: "none" };
+}
+export async function readPromptMeta(path) {
+    const text = await readFile(path, "utf8");
+    const { meta } = splitFrontmatter(text);
+    return meta;
+}
+function toPosixPath(path) {
+    return path.replaceAll("\\", "/");
+}
+export async function buildRoleCatalog(repoRoot) {
+    const rolesDir = join(repoRoot, ".mu", "roles");
+    let entries;
+    try {
+        entries = await readdir(rolesDir);
+    }
+    catch {
+        return "";
+    }
+    const roleFiles = entries.filter((e) => e.endsWith(".md")).sort();
+    const sections = [];
+    for (const file of roleFiles) {
+        const abs = join(rolesDir, file);
+        const text = await readFile(abs, "utf8");
+        const { meta, body } = splitFrontmatter(text);
+        const name = file.replace(/\.md$/, "");
+        const promptPath = toPosixPath(relative(repoRoot, abs));
+        const { description, source } = extractDescription(meta, body);
+        const parts = [];
+        for (const key of ["cli", "model", "reasoning"]) {
+            if (key in meta) {
+                parts.push(`${key}: ${String(meta[key])}`);
+            }
+        }
+        const configLine = parts.length > 0 ? parts.join(" | ") : "default config";
+        const catalogDesc = description || "No description provided.";
+        sections.push(`### ${name}\n` +
+            `description: ${catalogDesc}\n` +
+            `description_source: ${source}\n` +
+            `prompt: ${promptPath}\n` +
+            `config: ${configLine}`);
+    }
+    return sections.join("\n\n");
+}
+export async function renderPromptTemplate(path, issue, opts = {}) {
+    const text = await readFile(path, "utf8");
+    const { body } = splitFrontmatter(text);
+    let promptText = issue.title ?? "";
+    if (issue.body) {
+        promptText += `\n\n${issue.body}`;
+    }
+    let rendered = body;
+    rendered = rendered.replaceAll("{{PROMPT}}", promptText);
+    rendered = rendered.replaceAll("{{ISSUE_ID}}", issue.id ?? "");
+    if (rendered.includes("{{ROLES}}")) {
+        const catalog = opts.repoRoot ? await buildRoleCatalog(opts.repoRoot) : "";
+        rendered = rendered.replaceAll("{{ROLES}}", catalog);
+    }
+    return rendered;
+}
+export function resolvePromptPath(repoRoot, promptPath) {
+    if (isAbsolute(promptPath)) {
+        return promptPath;
+    }
+    return join(repoRoot, promptPath);
+}

package/package.json

@@ -0,0 +1,23 @@
+{
+  "name": "@femtomc/mu-agent",
+  "version": "26.2.33",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist/**"
+  ],
+  "dependencies": {
+    "@femtomc/mu-core": "26.2.33",
+    "@mariozechner/pi-agent-core": "^0.52.12",
+    "@mariozechner/pi-ai": "^0.52.12",
+    "@mariozechner/pi-coding-agent": "^0.52.12",
+    "zod": "^4.1.9"
+  }
+}