@pasko70/pibo 1.1.0 → 1.2.0

package/dist/api/simple-agent-api.js

@@ -0,0 +1,136 @@
+import { randomUUID, timingSafeEqual } from "node:crypto";
+import { PiboWebHttpError, readJsonBody, responseJson } from "../web/http.js";
+const DEFAULT_SEND_MESSAGE_TIMEOUT_MS = 10 * 60 * 1000;
+function configuredApiKey(options) {
+    return options.apiKey ?? process.env.PIBO_SIMPLE_API_KEY;
+}
+function safeEqual(left, right) {
+    const leftBuffer = Buffer.from(left);
+    const rightBuffer = Buffer.from(right);
+    if (leftBuffer.length !== rightBuffer.length)
+        return false;
+    return timingSafeEqual(leftBuffer, rightBuffer);
+}
+function decodeBasicAuthCredential(value) {
+    const match = value.match(/^Basic\s+(.+)$/i);
+    if (!match)
+        return [];
+    try {
+        const decoded = Buffer.from(match[1] ?? "", "base64").toString("utf8");
+        return decoded.split(":").map((part) => part.trim()).filter(Boolean);
+    }
+    catch {
+        return [];
+    }
+}
+function requestApiKeyCandidates(request) {
+    const candidates = [];
+    const add = (value) => {
+        let trimmed = value?.trim();
+        if (!trimmed)
+            return;
+        candidates.push(trimmed);
+        while (/^(Bearer|Token)\s+/i.test(trimmed)) {
+            trimmed = trimmed.replace(/^(Bearer|Token)\s+/i, "").trim();
+            if (trimmed)
+                candidates.push(trimmed);
+        }
+    };
+    add(request.headers.get("x-api-key"));
+    add(request.headers.get("api-key"));
+    const authorization = request.headers.get("authorization")?.trim();
+    add(authorization);
+    for (const credential of decodeBasicAuthCredential(authorization ?? ""))
+        add(credential);
+    return candidates;
+}
+function requireApiKey(request, options) {
+    const expected = configuredApiKey(options);
+    if (!expected)
+        throw new PiboWebHttpError("Simple agent API key is not configured", 503);
+    const candidates = requestApiKeyCandidates(request);
+    if (!candidates.some((candidate) => safeEqual(candidate, expected))) {
+        throw new PiboWebHttpError("Invalid API key", 401);
+    }
+}
+function requireString(value, name) {
+    if (typeof value !== "string" || value.trim().length === 0) {
+        throw new PiboWebHttpError(`${name} is required`, 400);
+    }
+    return value;
+}
+function parseSendMessageBody(body) {
+    return {
+        sessionId: requireString(body.sessionId, "sessionId"),
+        message: requireString(body.message, "message"),
+    };
+}
+function awaitAgentMessage(context, input) {
+    return new Promise((resolve, reject) => {
+        let lastAssistantMessage = "";
+        let settled = false;
+        let unsubscribe;
+        const finish = (callback) => {
+            if (settled)
+                return;
+            settled = true;
+            clearTimeout(timeout);
+            unsubscribe?.();
+            callback();
+        };
+        const timeout = setTimeout(() => {
+            finish(() => reject(new PiboWebHttpError("Timed out waiting for agent response", 504)));
+        }, input.timeoutMs);
+        unsubscribe = context.subscribe((event) => {
+            if (event.piboSessionId !== input.sessionId || !("eventId" in event) || event.eventId !== input.eventId)
+                return;
+            if (event.type === "assistant_message") {
+                lastAssistantMessage = event.text;
+                return;
+            }
+            if (event.type === "session_error") {
+                finish(() => reject(new PiboWebHttpError(event.error, 500)));
+                return;
+            }
+            if (event.type === "message_finished") {
+                finish(() => resolve({ eventId: input.eventId, sessionId: input.sessionId, message: lastAssistantMessage }));
+            }
+        });
+    });
+}
+async function handleSendMessage(request, context, options) {
+    requireApiKey(request, options);
+    const input = parseSendMessageBody(await readJsonBody(request));
+    if (!context.getSession(input.sessionId)) {
+        throw new PiboWebHttpError("Unknown sessionId", 404);
+    }
+    const eventId = randomUUID();
+    const responsePromise = awaitAgentMessage(context, {
+        sessionId: input.sessionId,
+        eventId,
+        timeoutMs: options.timeoutMs ?? DEFAULT_SEND_MESSAGE_TIMEOUT_MS,
+    });
+    await context.emit({
+        type: "message",
+        piboSessionId: input.sessionId,
+        id: eventId,
+        text: input.message,
+        source: "service",
+    });
+    return responseJson(await responsePromise);
+}
+export async function handleSimpleAgentApiRequest(request, context, options = {}) {
+    const url = new URL(request.url);
+    if (url.pathname === "/api/health") {
+        if (request.method !== "GET" && request.method !== "HEAD") {
+            return responseJson({ error: "Method not allowed" }, { status: 405, headers: { allow: "GET, HEAD" } });
+        }
+        return responseJson({ status: "ok" });
+    }
+    if (url.pathname !== "/api/send-message")
+        return undefined;
+    if (request.method !== "POST") {
+        return responseJson({ error: "Method not allowed" }, { status: 405, headers: { allow: "POST" } });
+    }
+    return handleSendMessage(request, context, options);
+}

package/dist/app-context.js

@@ -0,0 +1,4 @@
+export const PIBO_APP_CONTEXT = Object.freeze({
+    kind: "app-context",
+    id: "app",
+});

package/dist/apps/chat/agent-profiles.js

@@ -1,5 +1,6 @@
 import { InitialSessionContextBuilder } from "../../core/profiles.js";
-export function createCustomAgentProfileDefinition(agent) {
+export function createCustomAgentProfileDefinition(agent, options = {}) {
+    const shouldWarnMissingReferences = options.missingReferenceMode !== "silent";
     return {
         name: agent.profileName,
         aliases: [agent.id, `custom-agent:${agent.id}`],
@@ -16,6 +17,18 @@ export function createCustomAgentProfileDefinition(agent) {
                 builder.withMainModel(agent.mainModel);
             if (agent.subagentModel)
                 builder.withSubagentModel(agent.subagentModel);
+            if (agent.thinkingLevel)
+                builder.withThinkingLevel(agent.thinkingLevel);
+            if (agent.mainThinkingLevel)
+                builder.withMainThinkingLevel(agent.mainThinkingLevel);
+            if (agent.subagentThinkingLevel)
+                builder.withSubagentThinkingLevel(agent.subagentThinkingLevel);
+            if (agent.fast !== undefined)
+                builder.withFastMode(agent.fast);
+            if (agent.mainFast !== undefined)
+                builder.withMainFastMode(agent.mainFast);
+            if (agent.subagentFast !== undefined)
+                builder.withSubagentFastMode(agent.subagentFast);
             for (const skillName of agent.skills) {
                 try {
                     builder.addSkill(context.getSkill(skillName));
@@ -23,7 +36,8 @@ export function createCustomAgentProfileDefinition(agent) {
                 catch (error) {
                     if (!isUnknownSkillError(error, skillName))
                         throw error;
-                    console.warn(`Skipping unknown skill "${skillName}" for custom agent "${agent.profileName}"`);
+                    if (shouldWarnMissingReferences)
+                        console.warn(`Skipping unknown skill "${skillName}" for custom agent "${agent.profileName}"`);
                 }
             }
             for (const contextFileKey of agent.contextFiles) {
@@ -33,7 +47,8 @@ export function createCustomAgentProfileDefinition(agent) {
                 catch (error) {
                     if (!isUnknownContextFileError(error, contextFileKey))
                         throw error;
-                    console.warn(`Skipping unknown context file "${contextFileKey}" for custom agent "${agent.profileName}"`);
+                    if (shouldWarnMissingReferences)
+                        console.warn(`Skipping unknown context file "${contextFileKey}" for custom agent "${agent.profileName}"`);
                 }
             }
             for (const toolName of agent.nativeTools)

package/dist/apps/chat/agent-store.js

@@ -1,8 +1,10 @@
-import { randomUUID } from "node:crypto";
+import { createHash, randomUUID } from "node:crypto";
 import { mkdirSync } from "node:fs";
 import { dirname, resolve } from "node:path";
+import { piboHomePath } from "../../core/pibo-home.js";
 import { DatabaseSync } from "node:sqlite";
 import { DEFAULT_BUILTIN_TOOL_NAMES } from "../../core/profiles.js";
+import { isPiboThinkingLevel } from "../../core/thinking.js";
 import { findPiPackage } from "../../pi-packages/store.js";
 const CUSTOM_AGENT_NAME_PATTERN = /^[a-z][a-z0-9]*(?:-[a-z0-9]+)*$/;
 export class CustomAgentStore {
@@ -19,7 +21,6 @@ export class CustomAgentStore {
 			CREATE TABLE IF NOT EXISTS chat_agents (
 				id TEXT PRIMARY KEY,
 				profile_name TEXT NOT NULL UNIQUE,
-				owner_scope TEXT NOT NULL,
 				display_name TEXT NOT NULL,
 				description TEXT,
 				native_tools_json TEXT NOT NULL,
@@ -30,6 +31,12 @@ export class CustomAgentStore {
 				pi_packages_json TEXT NOT NULL DEFAULT '[]',
 				main_model_json TEXT,
 				subagent_model_json TEXT,
+				thinking_level TEXT,
+				main_thinking_level TEXT,
+				subagent_thinking_level TEXT,
+				fast INTEGER,
+				main_fast INTEGER,
+				subagent_fast INTEGER,
 				builtin_tools TEXT NOT NULL,
 				builtin_tool_names_json TEXT NOT NULL DEFAULT '["read","bash","edit","write"]',
 				auto_context_files INTEGER NOT NULL DEFAULT 1,
@@ -39,23 +46,23 @@ export class CustomAgentStore {
 				archived_at TEXT
 			);
 
-			CREATE INDEX IF NOT EXISTS idx_chat_agents_owner
-				ON chat_agents(owner_scope, updated_at);
 		`);
         this.migrateArchivedAtColumn();
         this.migrateAutoContextFilesColumn();
         this.migrateMcpServersColumn();
         this.migratePiPackagesColumn();
         this.migrateModelColumns();
+        this.migrateThinkingLevelColumn();
+        this.migrateThinkingOptionColumns();
         this.migrateBuiltinToolNamesColumn();
         this.migrateLegacyProfileNames();
+        this.migrateDuplicateProfileNames();
     }
-    list(ownerScope, options = {}) {
+    list(options = {}) {
         this.migrateLegacyProfileNames();
+        this.migrateDuplicateProfileNames();
         const archivedClause = options.includeArchived ? "" : " AND archived_at IS NULL";
-        const rows = ownerScope
-            ? this.db.prepare(`SELECT * FROM chat_agents WHERE owner_scope = ?${archivedClause} ORDER BY updated_at DESC`).all(ownerScope)
-            : this.db.prepare(`SELECT * FROM chat_agents WHERE 1 = 1${archivedClause} ORDER BY updated_at DESC`).all();
+        const rows = this.db.prepare(`SELECT * FROM chat_agents WHERE 1 = 1${archivedClause} ORDER BY updated_at DESC`).all();
         return rows.map(agentFromRow);
     }
     get(id) {
@@ -72,7 +79,6 @@ export class CustomAgentStore {
         const agent = {
             id,
             profileName,
-            ownerScope: input.ownerScope,
             displayName: input.displayName,
             description: input.description,
             nativeTools: [...(input.nativeTools ?? [])],
@@ -83,6 +89,12 @@ export class CustomAgentStore {
             piPackages: sanitizePiPackages(input.piPackages ?? []),
             mainModel: sanitizeModelProfile(input.mainModel),
             subagentModel: sanitizeModelProfile(input.subagentModel),
+            thinkingLevel: sanitizeThinkingLevel(input.thinkingLevel),
+            mainThinkingLevel: sanitizeThinkingLevel(input.mainThinkingLevel),
+            subagentThinkingLevel: sanitizeThinkingLevel(input.subagentThinkingLevel),
+            fast: sanitizeBoolean(input.fast),
+            mainFast: sanitizeBoolean(input.mainFast),
+            subagentFast: sanitizeBoolean(input.subagentFast),
             builtinTools: input.builtinTools ?? "default",
             builtinToolNames: sanitizeBuiltinToolNames(input.builtinToolNames),
             autoContextFiles: input.autoContextFiles ?? true,
@@ -116,6 +128,12 @@ export class CustomAgentStore {
             piPackages: input.piPackages ? sanitizePiPackages(input.piPackages) : existing.piPackages,
             mainModel: input.mainModel === undefined ? existing.mainModel : sanitizeModelProfile(input.mainModel),
             subagentModel: input.subagentModel === undefined ? existing.subagentModel : sanitizeModelProfile(input.subagentModel),
+            thinkingLevel: input.thinkingLevel === undefined ? existing.thinkingLevel : sanitizeThinkingLevel(input.thinkingLevel),
+            mainThinkingLevel: input.mainThinkingLevel === undefined ? existing.mainThinkingLevel : sanitizeThinkingLevel(input.mainThinkingLevel),
+            subagentThinkingLevel: input.subagentThinkingLevel === undefined ? existing.subagentThinkingLevel : sanitizeThinkingLevel(input.subagentThinkingLevel),
+            fast: input.fast === undefined ? existing.fast : sanitizeBoolean(input.fast),
+            mainFast: input.mainFast === undefined ? existing.mainFast : sanitizeBoolean(input.mainFast),
+            subagentFast: input.subagentFast === undefined ? existing.subagentFast : sanitizeBoolean(input.subagentFast),
             builtinTools: input.builtinTools ?? existing.builtinTools,
             builtinToolNames: input.builtinToolNames ? sanitizeBuiltinToolNames(input.builtinToolNames) : existing.builtinToolNames,
             autoContextFiles: input.autoContextFiles ?? existing.autoContextFiles,
@@ -136,6 +154,12 @@ export class CustomAgentStore {
 					pi_packages_json = ?,
 					main_model_json = ?,
 					subagent_model_json = ?,
+					thinking_level = ?,
+					main_thinking_level = ?,
+					subagent_thinking_level = ?,
+					fast = ?,
+					main_fast = ?,
+					subagent_fast = ?,
 					builtin_tools = ?,
 					builtin_tool_names_json = ?,
 					auto_context_files = ?,
@@ -143,7 +167,7 @@ export class CustomAgentStore {
 					updated_at = ?
 				WHERE id = ?
 			`)
-            .run(updated.profileName, updated.displayName, updated.description ?? null, JSON.stringify(updated.nativeTools), JSON.stringify(updated.skills), JSON.stringify(updated.contextFiles), JSON.stringify(sanitizeSubagents(updated.subagents)), JSON.stringify(updated.mcpServers), JSON.stringify(updated.piPackages), updated.mainModel ? JSON.stringify(updated.mainModel) : null, updated.subagentModel ? JSON.stringify(updated.subagentModel) : null, updated.builtinTools, JSON.stringify(updated.builtinToolNames), updated.autoContextFiles ? 1 : 0, updated.runControl ? 1 : 0, updated.updatedAt, id);
+            .run(updated.profileName, updated.displayName, updated.description ?? null, JSON.stringify(updated.nativeTools), JSON.stringify(updated.skills), JSON.stringify(updated.contextFiles), JSON.stringify(sanitizeSubagents(updated.subagents)), JSON.stringify(updated.mcpServers), JSON.stringify(updated.piPackages), updated.mainModel ? JSON.stringify(updated.mainModel) : null, updated.subagentModel ? JSON.stringify(updated.subagentModel) : null, updated.thinkingLevel ?? null, updated.mainThinkingLevel ?? null, updated.subagentThinkingLevel ?? null, serializeBoolean(updated.fast), serializeBoolean(updated.mainFast), serializeBoolean(updated.subagentFast), updated.builtinTools, JSON.stringify(updated.builtinToolNames), updated.autoContextFiles ? 1 : 0, updated.runControl ? 1 : 0, updated.updatedAt, id);
         return this.get(id);
     }
     setArchived(id, archived) {
@@ -170,7 +194,6 @@ export class CustomAgentStore {
 				INSERT INTO chat_agents (
 					id,
 					profile_name,
-					owner_scope,
 					display_name,
 					description,
 					native_tools_json,
@@ -181,6 +204,12 @@ export class CustomAgentStore {
 					pi_packages_json,
 					main_model_json,
 					subagent_model_json,
+					thinking_level,
+					main_thinking_level,
+					subagent_thinking_level,
+					fast,
+					main_fast,
+					subagent_fast,
 					builtin_tools,
 					builtin_tool_names_json,
 					auto_context_files,
@@ -188,9 +217,9 @@ export class CustomAgentStore {
 					created_at,
 					updated_at,
 					archived_at
-				) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+				) VALUES (${Array.from({ length: 25 }, () => "?").join(", ")})
 			`)
-            .run(agent.id, agent.profileName, agent.ownerScope, agent.displayName, agent.description ?? null, JSON.stringify(agent.nativeTools), JSON.stringify(agent.skills), JSON.stringify(agent.contextFiles), JSON.stringify(sanitizeSubagents(agent.subagents)), JSON.stringify(agent.mcpServers), JSON.stringify(agent.piPackages), agent.mainModel ? JSON.stringify(agent.mainModel) : null, agent.subagentModel ? JSON.stringify(agent.subagentModel) : null, agent.builtinTools, JSON.stringify(agent.builtinToolNames), agent.autoContextFiles ? 1 : 0, agent.runControl ? 1 : 0, agent.createdAt, agent.updatedAt, agent.archivedAt ?? null);
+            .run(agent.id, agent.profileName, agent.displayName, agent.description ?? null, JSON.stringify(agent.nativeTools), JSON.stringify(agent.skills), JSON.stringify(agent.contextFiles), JSON.stringify(sanitizeSubagents(agent.subagents)), JSON.stringify(agent.mcpServers), JSON.stringify(agent.piPackages), agent.mainModel ? JSON.stringify(agent.mainModel) : null, agent.subagentModel ? JSON.stringify(agent.subagentModel) : null, agent.thinkingLevel ?? null, agent.mainThinkingLevel ?? null, agent.subagentThinkingLevel ?? null, serializeBoolean(agent.fast), serializeBoolean(agent.mainFast), serializeBoolean(agent.subagentFast), agent.builtinTools, JSON.stringify(agent.builtinToolNames), agent.autoContextFiles ? 1 : 0, agent.runControl ? 1 : 0, agent.createdAt, agent.updatedAt, agent.archivedAt ?? null);
     }
     requireProfileNameAvailable(profileName, currentId) {
         const row = this.db.prepare("SELECT id FROM chat_agents WHERE profile_name = ?").get(profileName);
@@ -211,6 +240,24 @@ export class CustomAgentStore {
                 .run(nextName, nextName, row.id);
         }
     }
+    migrateDuplicateProfileNames() {
+        const rows = this.db.prepare("SELECT id, profile_name, display_name, created_at, updated_at FROM chat_agents ORDER BY profile_name ASC, updated_at DESC, created_at DESC, id DESC").all();
+        const used = new Set();
+        for (const row of rows) {
+            if (!used.has(row.profile_name)) {
+                used.add(row.profile_name);
+                continue;
+            }
+            const nextName = uniqueAgentName(legacyAgentNameCandidate(row.profile_name, row.id), used);
+            used.add(nextName);
+            this.db
+                .prepare("UPDATE chat_agents SET profile_name = ?, display_name = ? WHERE id = ?")
+                .run(nextName, nextName, row.id);
+        }
+    }
+    tableColumns() {
+        return new Set(this.db.prepare("PRAGMA table_info(chat_agents)").all().map((column) => column.name));
+    }
     migrateArchivedAtColumn() {
         const columns = new Set(this.db.prepare("PRAGMA table_info(chat_agents)").all().map((column) => column.name));
         if (!columns.has("archived_at")) {
@@ -244,6 +291,30 @@ export class CustomAgentStore {
             this.db.prepare("ALTER TABLE chat_agents ADD COLUMN subagent_model_json TEXT").run();
         }
     }
+    migrateThinkingLevelColumn() {
+        const columns = new Set(this.db.prepare("PRAGMA table_info(chat_agents)").all().map((column) => column.name));
+        if (!columns.has("thinking_level")) {
+            this.db.prepare("ALTER TABLE chat_agents ADD COLUMN thinking_level TEXT").run();
+        }
+    }
+    migrateThinkingOptionColumns() {
+        const columns = new Set(this.db.prepare("PRAGMA table_info(chat_agents)").all().map((column) => column.name));
+        if (!columns.has("main_thinking_level")) {
+            this.db.prepare("ALTER TABLE chat_agents ADD COLUMN main_thinking_level TEXT").run();
+        }
+        if (!columns.has("subagent_thinking_level")) {
+            this.db.prepare("ALTER TABLE chat_agents ADD COLUMN subagent_thinking_level TEXT").run();
+        }
+        if (!columns.has("fast")) {
+            this.db.prepare("ALTER TABLE chat_agents ADD COLUMN fast INTEGER").run();
+        }
+        if (!columns.has("main_fast")) {
+            this.db.prepare("ALTER TABLE chat_agents ADD COLUMN main_fast INTEGER").run();
+        }
+        if (!columns.has("subagent_fast")) {
+            this.db.prepare("ALTER TABLE chat_agents ADD COLUMN subagent_fast INTEGER").run();
+        }
+    }
     migrateBuiltinToolNamesColumn() {
         const columns = new Set(this.db.prepare("PRAGMA table_info(chat_agents)").all().map((column) => column.name));
         if (!columns.has("builtin_tool_names_json")) {
@@ -251,14 +322,13 @@ export class CustomAgentStore {
         }
     }
 }
-export function createDefaultCustomAgentStore(cwd = process.cwd()) {
-    return new CustomAgentStore(resolve(cwd, ".pibo/chat-agents.sqlite"));
+export function createDefaultCustomAgentStore(_cwd) {
+    return new CustomAgentStore(piboHomePath("chat-agents.sqlite"));
 }
 function agentFromRow(row) {
     return {
         id: row.id,
         profileName: row.profile_name,
-        ownerScope: row.owner_scope,
         displayName: row.display_name,
         description: row.description ?? undefined,
         nativeTools: parseStringArray(row.native_tools_json),
@@ -269,6 +339,12 @@ function agentFromRow(row) {
         piPackages: parseStringArray(row.pi_packages_json),
         mainModel: parseModelProfile(row.main_model_json),
         subagentModel: parseModelProfile(row.subagent_model_json),
+        thinkingLevel: sanitizeThinkingLevel(row.thinking_level),
+        mainThinkingLevel: sanitizeThinkingLevel(row.main_thinking_level),
+        subagentThinkingLevel: sanitizeThinkingLevel(row.subagent_thinking_level),
+        fast: parseBoolean(row.fast),
+        mainFast: parseBoolean(row.main_fast),
+        subagentFast: parseBoolean(row.subagent_fast),
         builtinTools: row.builtin_tools,
         builtinToolNames: sanitizeBuiltinToolNames(parseStringArray(row.builtin_tool_names_json)),
         autoContextFiles: row.auto_context_files !== 0,
@@ -298,6 +374,22 @@ function sanitizePiPackages(value) {
     }
     return packages;
 }
+function sanitizeThinkingLevel(value) {
+    return typeof value === "string" && isPiboThinkingLevel(value) ? value : undefined;
+}
+function sanitizeBoolean(value) {
+    return typeof value === "boolean" ? value : undefined;
+}
+function serializeBoolean(value) {
+    if (value === undefined)
+        return null;
+    return value ? 1 : 0;
+}
+function parseBoolean(value) {
+    if (value === null)
+        return undefined;
+    return value === 1;
+}
 function sanitizeBuiltinToolNames(value) {
     const selected = new Set(uniqueStrings(value ?? DEFAULT_BUILTIN_TOOL_NAMES));
     return DEFAULT_BUILTIN_TOOL_NAMES.filter((name) => selected.has(name));
@@ -368,6 +460,11 @@ function agentNameCandidate(displayName, id) {
         return candidate;
     return `agent-${id.replace(/^agent_/, "").replace(/[^a-z0-9]+/g, "-").replace(/^-+|-+$/g, "")}`;
 }
+function legacyAgentNameCandidate(profileName, id) {
+    const baseName = agentNameCandidate(profileName, id);
+    const hash = createHash("sha256").update(id).digest("hex").slice(0, 8);
+    return `${baseName}-legacy-${hash}`;
+}
 function uniqueAgentName(baseName, used) {
     let name = baseName;
     let suffix = 2;