neoctl 0.1.13 → 0.1.14

package/dist/session/simple-session-runtime.d.ts

@@ -0,0 +1,74 @@
+import type { QueryEngineOptions } from "../core/query-engine.js";
+import { QueryEngine } from "../core/query-engine.js";
+import type { AgentEvent } from "../types/events.js";
+import type { Message, MessageBlock } from "../types/messages.js";
+import type { DisplayAgentEvent, DisplayImageAttachment, DisplayMessage, DisplayMessageOptions } from "../ui/display-message.js";
+import type { SessionStoreSnapshot, SessionSummary } from "./session-store.js";
+export type SimpleSessionRuntimeEventListener = (event: AgentEvent, context: SimpleSessionRuntimeEventContext) => void | Promise<void>;
+export type SimpleSessionRuntimeDisplayEventListener = (event: DisplayAgentEvent, context: SimpleSessionRuntimeEventContext) => void | Promise<void>;
+export interface SimpleSessionRuntimeEventContext {
+    sessionId: string;
+}
+export interface SimpleSessionRuntimeOptions extends Omit<QueryEngineOptions, "session"> {
+    /**
+     * Base directory for all sessions managed by this runtime.
+     * For simple multi-user isolation, create one runtime per user or pass a per-user rootDir.
+     */
+    sessionRootDir?: string;
+    session?: Omit<NonNullable<QueryEngineOptions["session"]>, "sessionId" | "rootDir" | "resume">;
+}
+export interface SimpleSessionRuntimeGetOptions {
+    /** Resume transcript when the engine is first created. Defaults to true. */
+    resume?: boolean;
+}
+export interface SimpleSessionRuntimeNewSessionOptions {
+    /** Optional concrete session id. If omitted, QueryEngine creates one. */
+    sessionId?: string;
+}
+export interface SimpleSessionRuntimeSendOptions {
+    abortSignal?: AbortSignal;
+    blocks?: MessageBlock[];
+    displayText?: string;
+    /**
+     * Behavior when another send is already running for the same session.
+     * Defaults to "reject".
+     */
+    busyBehavior?: "reject" | "interrupt";
+}
+export interface SimpleSessionRuntimeSnapshot {
+    activeSessions: string[];
+    busySessions: string[];
+}
+/**
+ * Lightweight session runtime for simple web/Vue integrations.
+ *
+ * It keeps exactly one active QueryEngine per session id, rejects concurrent writes
+ * to the same session by default, and can broadcast sendUserText events to listeners.
+ * Existing QueryEngine and SessionStore behavior is not changed.
+ */
+export declare class SimpleSessionRuntime {
+    private readonly options;
+    private readonly engines;
+    private readonly busy;
+    private readonly controllers;
+    private readonly listeners;
+    private readonly displayListeners;
+    constructor(options: SimpleSessionRuntimeOptions);
+    getOrCreateEngine(sessionId: string, options?: SimpleSessionRuntimeGetOptions): Promise<QueryEngine>;
+    newSession(options?: SimpleSessionRuntimeNewSessionOptions): Promise<SessionStoreSnapshot>;
+    resumeSession(sessionId: string): Promise<SessionStoreSnapshot>;
+    listSessions(limit?: number): Promise<SessionSummary[]>;
+    deleteSession(sessionId: string): Promise<boolean>;
+    getMessages(sessionId: string): Promise<Message[]>;
+    getDisplayMessages(sessionId: string, options?: DisplayMessageOptions): Promise<DisplayMessage[]>;
+    getDisplayImages(sessionId: string, options?: DisplayMessageOptions): Promise<DisplayImageAttachment[]>;
+    sendUserText(sessionId: string, text: string, options?: SimpleSessionRuntimeSendOptions): AsyncGenerator<AgentEvent>;
+    abort(sessionId: string): boolean;
+    isBusy(sessionId: string): boolean;
+    onEvent(listener: SimpleSessionRuntimeEventListener): () => void;
+    onDisplayEvent(listener: SimpleSessionRuntimeDisplayEventListener, options?: DisplayMessageOptions): () => void;
+    release(sessionId: string): void;
+    snapshot(): SimpleSessionRuntimeSnapshot;
+    private createEngine;
+    private emit;
+}

package/dist/session/simple-session-runtime.js

@@ -0,0 +1,171 @@
+import { QueryEngine } from "../core/query-engine.js";
+import { extractMessageImages, toDisplayAgentEvent, toDisplayMessages } from "../ui/display-message.js";
+/**
+ * Lightweight session runtime for simple web/Vue integrations.
+ *
+ * It keeps exactly one active QueryEngine per session id, rejects concurrent writes
+ * to the same session by default, and can broadcast sendUserText events to listeners.
+ * Existing QueryEngine and SessionStore behavior is not changed.
+ */
+export class SimpleSessionRuntime {
+    options;
+    engines = new Map();
+    busy = new Set();
+    controllers = new Map();
+    listeners = new Set();
+    displayListeners = new Set();
+    constructor(options) {
+        this.options = options;
+    }
+    async getOrCreateEngine(sessionId, options = {}) {
+        const normalizedSessionId = normalizeSessionId(sessionId);
+        const existing = this.engines.get(normalizedSessionId);
+        if (existing) {
+            await existing.initialized;
+            return existing.engine;
+        }
+        const engine = this.createEngine(normalizedSessionId, options.resume ?? true);
+        const initialized = engine.initialize();
+        const managed = { engine, initialized };
+        this.engines.set(normalizedSessionId, managed);
+        try {
+            await initialized;
+        }
+        catch (error) {
+            this.engines.delete(normalizedSessionId);
+            throw error;
+        }
+        return engine;
+    }
+    async newSession(options = {}) {
+        const engine = this.createEngine(options.sessionId, false);
+        const snapshot = await engine.newSession();
+        this.engines.set(snapshot.sessionId, { engine, initialized: Promise.resolve() });
+        return snapshot;
+    }
+    async resumeSession(sessionId) {
+        const engine = await this.getOrCreateEngine(sessionId, { resume: true });
+        const snapshot = engine.snapshot().session;
+        if (!snapshot)
+            throw new Error("session transcripts are disabled");
+        return snapshot;
+    }
+    async listSessions(limit = 20) {
+        const engine = this.createEngine(undefined, false);
+        return engine.listSessions(limit);
+    }
+    async deleteSession(sessionId) {
+        const normalizedSessionId = normalizeSessionId(sessionId);
+        if (this.busy.has(normalizedSessionId))
+            throw new Error(`session is busy: ${normalizedSessionId}`);
+        this.abort(normalizedSessionId);
+        this.engines.delete(normalizedSessionId);
+        const engine = this.createEngine(undefined, false);
+        return engine.deleteSession(normalizedSessionId);
+    }
+    async getMessages(sessionId) {
+        const engine = await this.getOrCreateEngine(sessionId, { resume: true });
+        return engine.getHistoryMessages();
+    }
+    async getDisplayMessages(sessionId, options = {}) {
+        return toDisplayMessages(await this.getMessages(sessionId), options);
+    }
+    async getDisplayImages(sessionId, options = { imageMode: "data-url" }) {
+        return extractMessageImages(await this.getMessages(sessionId), options);
+    }
+    async *sendUserText(sessionId, text, options = {}) {
+        const normalizedSessionId = normalizeSessionId(sessionId);
+        if (this.busy.has(normalizedSessionId)) {
+            if (options.busyBehavior === "interrupt") {
+                this.abort(normalizedSessionId);
+            }
+            else {
+                throw new Error(`session is busy: ${normalizedSessionId}`);
+            }
+        }
+        const engine = await this.getOrCreateEngine(normalizedSessionId, { resume: true });
+        const controller = new AbortController();
+        const externalAbort = options.abortSignal;
+        const abortFromExternal = () => controller.abort(externalAbort?.reason);
+        if (externalAbort?.aborted)
+            abortFromExternal();
+        else
+            externalAbort?.addEventListener("abort", abortFromExternal, { once: true });
+        this.busy.add(normalizedSessionId);
+        this.controllers.set(normalizedSessionId, controller);
+        try {
+            for await (const event of engine.sendUserText(text, {
+                abortSignal: controller.signal,
+                blocks: options.blocks,
+                displayText: options.displayText,
+            })) {
+                await this.emit(event, { sessionId: normalizedSessionId });
+                yield event;
+            }
+        }
+        finally {
+            externalAbort?.removeEventListener("abort", abortFromExternal);
+            if (this.controllers.get(normalizedSessionId) === controller)
+                this.controllers.delete(normalizedSessionId);
+            this.busy.delete(normalizedSessionId);
+        }
+    }
+    abort(sessionId) {
+        const controller = this.controllers.get(normalizeSessionId(sessionId));
+        if (!controller)
+            return false;
+        controller.abort();
+        return true;
+    }
+    isBusy(sessionId) {
+        return this.busy.has(normalizeSessionId(sessionId));
+    }
+    onEvent(listener) {
+        this.listeners.add(listener);
+        return () => this.listeners.delete(listener);
+    }
+    onDisplayEvent(listener, options = { imageMode: "data-url" }) {
+        const entry = { listener, options };
+        this.displayListeners.add(entry);
+        return () => this.displayListeners.delete(entry);
+    }
+    release(sessionId) {
+        const normalizedSessionId = normalizeSessionId(sessionId);
+        this.abort(normalizedSessionId);
+        this.engines.delete(normalizedSessionId);
+        this.busy.delete(normalizedSessionId);
+    }
+    snapshot() {
+        return {
+            activeSessions: [...this.engines.keys()],
+            busySessions: [...this.busy],
+        };
+    }
+    createEngine(sessionId, resume) {
+        return new QueryEngine({
+            ...this.options,
+            session: {
+                ...this.options.session,
+                enabled: this.options.session?.enabled ?? true,
+                rootDir: this.options.sessionRootDir,
+                sessionId,
+                resume,
+            },
+        });
+    }
+    async emit(event, context) {
+        for (const listener of this.listeners) {
+            await listener(event, context);
+        }
+        for (const { listener, options } of this.displayListeners) {
+            await listener(toDisplayAgentEvent(event, options), context);
+        }
+    }
+}
+function normalizeSessionId(sessionId) {
+    const normalized = sessionId.trim();
+    if (!normalized)
+        throw new Error("sessionId is required");
+    return normalized;
+}
+//# sourceMappingURL=simple-session-runtime.js.map

package/dist/session/simple-session-runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"simple-session-runtime.js","sourceRoot":"","sources":["../../src/session/simple-session-runtime.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAItD,OAAO,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAkDxG;;;;;;GAMG;AACH,MAAM,OAAO,oBAAoB;IAOF;IANZ,OAAO,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC3C,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IACzB,WAAW,GAAG,IAAI,GAAG,EAA2B,CAAC;IACjD,SAAS,GAAG,IAAI,GAAG,EAAqC,CAAC;IACzD,gBAAgB,GAAG,IAAI,GAAG,EAA0F,CAAC;IAEtI,YAA6B,OAAoC;QAApC,YAAO,GAAP,OAAO,CAA6B;IAAG,CAAC;IAErE,KAAK,CAAC,iBAAiB,CAAC,SAAiB,EAAE,UAA0C,EAAE;QACrF,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACvD,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,CAAC,WAAW,CAAC;YAC3B,OAAO,QAAQ,CAAC,MAAM,CAAC;QACzB,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,mBAAmB,EAAE,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;QAC9E,MAAM,WAAW,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QACxC,MAAM,OAAO,GAAkB,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;QACvD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC;YACH,MAAM,WAAW,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;YACzC,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,UAAiD,EAAE;QAClE,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC3D,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;QAC3C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACjF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACzE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC;QAC3C,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACnE,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAK,GAAG,EAAE;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACnD,OAAO,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,mBAAmB,EAAE,CAAC,CAAC;QACnG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAChC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACnD,OAAO,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,SAAiB;QACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACzE,OAAO,MAAM,CAAC,kBAAkB,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,SAAiB,EAAE,UAAiC,EAAE;QAC7E,OAAO,iBAAiB,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,SAAiB,EAAE,UAAiC,EAAE,SAAS,EAAE,UAAU,EAAE;QAClG,OAAO,oBAAoB,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,CAAC,YAAY,CAAC,SAAiB,EAAE,IAAY,EAAE,UAA2C,EAAE;QAChG,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC,EAAE,CAAC;YACvC,IAAI,OAAO,CAAC,YAAY,KAAK,WAAW,EAAE,CAAC;gBACzC,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,KAAK,CAAC,oBAAoB,mBAAmB,EAAE,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,mBAAmB,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACnF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,CAAC;QAC1C,MAAM,iBAAiB,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QACxE,IAAI,aAAa,EAAE,OAAO;YAAE,iBAAiB,EAAE,CAAC;;YAC3C,aAAa,EAAE,gBAAgB,CAAC,OAAO,EAAE,iBAAiB,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAEjF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACnC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,mBAAmB,EAAE,UAAU,CAAC,CAAC;QACtD,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE;gBAClD,WAAW,EAAE,UAAU,CAAC,MAAM;gBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,WAAW,EAAE,OAAO,CAAC,WAAW;aACjC,CAAC,EAAE,CAAC;gBACH,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,mBAAmB,EAAE,CAAC,CAAC;gBAC3D,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,aAAa,EAAE,mBAAmB,CAAC,OAAO,EAAE,iBAAiB,CAAC,CAAC;YAC/D,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,mBAAmB,CAAC,KAAK,UAAU;gBAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;YAC3G,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,SAAiB;QACrB,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC;QACvE,IAAI,CAAC,UAAU;YAAE,OAAO,KAAK,CAAC;QAC9B,UAAU,CAAC,KAAK,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,SAAiB;QACtB,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,OAAO,CAAC,QAA2C;QACjD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7B,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,cAAc,CAAC,QAAkD,EAAE,UAAiC,EAAE,SAAS,EAAE,UAAU,EAAE;QAC3H,MAAM,KAAK,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACjC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACnD,CAAC;IAED,OAAO,CAAC,SAAiB;QACvB,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;QAChC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACzC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;IACxC,CAAC;IAED,QAAQ;QACN,OAAO;YACL,cAAc,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACxC,YAAY,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC;SAC7B,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,SAA6B,EAAE,MAAe;QACjE,OAAO,IAAI,WAAW,CAAC;YACrB,GAAG,IAAI,CAAC,OAAO;YACf,OAAO,EAAE;gBACP,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO;gBACvB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,IAAI,IAAI;gBAC9C,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;gBACpC,SAAS;gBACT,MAAM;aACP;SACF,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,IAAI,CAAC,KAAiB,EAAE,OAAyC;QAC7E,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,MAAM,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACjC,CAAC;QACD,KAAK,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1D,MAAM,QAAQ,CAAC,mBAAmB,CAAC,KAAK,EAAE,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;CACF;AAED,SAAS,kBAAkB,CAAC,SAAiB;IAC3C,MAAM,UAAU,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC;IACpC,IAAI,CAAC,UAAU;QAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC1D,OAAO,UAAU,CAAC;AACpB,CAAC"}

package/dist/tools/builtins/image-generation-tool.d.ts

@@ -0,0 +1,51 @@
+import type { Tool } from "../tool.js";
+export type ImageGenerationSize = "auto" | "1024x1024" | "1536x1024" | "1024x1536";
+export type ImageGenerationQuality = "auto" | "low" | "medium" | "high";
+export type ImageGenerationFormat = "png" | "jpeg" | "webp";
+export type OpenAIImageGenerationResponseFormat = ImageGenerationFormat | "jpg";
+export type ImageGenerationBackground = "auto" | "transparent" | "opaque";
+export type ImageGenerationModeration = "auto" | "low";
+export interface ImageGenerationToolInput {
+    prompt: string;
+    model?: string;
+    size?: ImageGenerationSize;
+    quality?: ImageGenerationQuality;
+    outputFormat?: ImageGenerationFormat;
+    background?: ImageGenerationBackground;
+    moderation?: ImageGenerationModeration;
+    n?: number;
+}
+export interface ImageGenerationResult {
+    index: number;
+    mimeType: string;
+    base64: string;
+    dataUrl: string;
+    revisedPrompt?: string;
+}
+export interface ImageGenerationToolOutput {
+    provider: "openai";
+    model: string;
+    prompt: string;
+    size: ImageGenerationSize;
+    quality: ImageGenerationQuality;
+    outputFormat: ImageGenerationFormat;
+    background: ImageGenerationBackground;
+    returnedImages: number;
+    images: ImageGenerationResult[];
+    raw?: unknown;
+}
+export interface CreateOpenAIImageGenerationToolOptions {
+    apiKey?: string;
+    baseUrl?: string;
+    model?: string;
+    timeoutMs?: number;
+}
+/**
+ * OpenAI-only image generation tool backed by the Images API.
+ *
+ * The tool is intentionally exposed only when MODEL_PROVIDER=openai. Other providers
+ * should not receive a drawable tool definition; the default system prompt tells the
+ * model that image generation is unavailable in that configuration.
+ */
+export declare function createOpenAIImageGenerationTool(options?: CreateOpenAIImageGenerationToolOptions): Tool<ImageGenerationToolInput>;
+export declare const openAIImageGenerationTool: Tool<ImageGenerationToolInput>;

package/dist/tools/builtins/image-generation-tool.js

@@ -0,0 +1,298 @@
+const DEFAULT_OPENAI_BASE_URL = "https://api.openai.com";
+const DEFAULT_IMAGE_MODEL = "gpt-image-1";
+const DEFAULT_TIMEOUT_MS = 120_000;
+const MAX_IMAGES = 4;
+/**
+ * OpenAI-only image generation tool backed by the Images API.
+ *
+ * The tool is intentionally exposed only when MODEL_PROVIDER=openai. Other providers
+ * should not receive a drawable tool definition; the default system prompt tells the
+ * model that image generation is unavailable in that configuration.
+ */
+export function createOpenAIImageGenerationTool(options = {}) {
+    return {
+        name: "image2",
+        aliases: ["draw_image", "generate_image"],
+        description: "Generate images with OpenAI's Images API. Use this when the user asks you to draw, create, generate, or render an image. Return the generated image data URLs in the tool result for the UI to display. This tool is available only when MODEL_PROVIDER=openai; with other providers, state that this model does not have a drawing tool.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                prompt: { type: "string", description: "Detailed image prompt. Include subject, style, composition, colors, text to render, aspect ratio intent, and any safety-preserving constraints." },
+                model: { type: "string", description: `Optional OpenAI image model. Defaults to OPENAI_IMAGE_MODEL or ${DEFAULT_IMAGE_MODEL}.` },
+                size: { type: "string", enum: ["auto", "1024x1024", "1536x1024", "1024x1536"], description: "Output image size. Use 1024x1024 for square, 1536x1024 for landscape, 1024x1536 for portrait. Defaults to auto." },
+                quality: { type: "string", enum: ["auto", "low", "medium", "high"], description: "Rendering quality. Defaults to auto." },
+                outputFormat: { type: "string", enum: ["png", "jpeg", "webp"], description: "Returned image format. Defaults to png." },
+                background: { type: "string", enum: ["auto", "transparent", "opaque"], description: "Background handling for supported models. Defaults to auto." },
+                moderation: { type: "string", enum: ["auto", "low"], description: "OpenAI image moderation setting for supported models. Defaults to auto." },
+                n: { type: "integer", description: `Number of images to generate, 1-${MAX_IMAGES}. Defaults to 1.` },
+            },
+            required: ["prompt"],
+            additionalProperties: false,
+        },
+        metadata: {
+            readOnly: false,
+            concurrent: true,
+            visible: true,
+            maxResultSizeChars: 24000,
+        },
+        validate(input) {
+            const record = input;
+            return {
+                prompt: record.prompt ?? "",
+                model: record.model,
+                size: record.size ?? "auto",
+                quality: record.quality ?? "auto",
+                outputFormat: record.outputFormat ?? "png",
+                background: record.background ?? "auto",
+                moderation: record.moderation ?? "auto",
+                n: record.n ?? 1,
+            };
+        },
+        validateInput(input) {
+            if (!input.prompt.trim())
+                return { ok: false, message: "image2.prompt cannot be empty" };
+            if (input.model !== undefined && !input.model.trim())
+                return { ok: false, message: "image2.model cannot be empty" };
+            if (!isImageSize(input.size))
+                return { ok: false, message: "image2.size must be auto, 1024x1024, 1536x1024, or 1024x1536" };
+            if (!isImageQuality(input.quality))
+                return { ok: false, message: "image2.quality must be auto, low, medium, or high" };
+            if (!isImageFormat(input.outputFormat))
+                return { ok: false, message: "image2.outputFormat must be png, jpeg, or webp" };
+            if (!isImageBackground(input.background))
+                return { ok: false, message: "image2.background must be auto, transparent, or opaque" };
+            if (!isImageModeration(input.moderation))
+                return { ok: false, message: "image2.moderation must be auto or low" };
+            const count = input.n ?? 1;
+            if (!Number.isInteger(count) || count < 1 || count > MAX_IMAGES)
+                return { ok: false, message: `image2.n must be between 1 and ${MAX_IMAGES}` };
+            return { ok: true, value: { ...input, n: count } };
+        },
+        isConcurrencySafe() {
+            return true;
+        },
+        async call(input, context, callOptions) {
+            const apiKey = resolveApiKey(options.apiKey);
+            if (!apiKey) {
+                return {
+                    ok: false,
+                    output: {
+                        provider: "openai",
+                        error: "OpenAI image generation requires OPENAI_IMAGE_API_KEY or OPENAI_API_KEY.",
+                    },
+                };
+            }
+            const baseUrl = stripTrailingSlash(options.baseUrl?.trim() || process.env.OPENAI_IMAGE_BASE_URL?.trim() || process.env.OPENAI_BASE_URL?.trim() || DEFAULT_OPENAI_BASE_URL);
+            const model = input.model?.trim() || options.model?.trim() || process.env.OPENAI_IMAGE_MODEL?.trim() || DEFAULT_IMAGE_MODEL;
+            const timeoutMs = options.timeoutMs ?? parsePositiveNumber(process.env.OPENAI_IMAGE_TIMEOUT_MS) ?? parsePositiveNumber(process.env.MODEL_TIMEOUT_MS) ?? DEFAULT_TIMEOUT_MS;
+            callOptions.onProgress?.({ toolName: "image2", message: `Generating image with OpenAI ${model}` });
+            try {
+                const response = await callOpenAIImageGeneration({
+                    apiKey,
+                    baseUrl,
+                    timeoutMs,
+                    signal: context.abortSignal,
+                    input: { ...input, model },
+                });
+                const images = extractGeneratedImages(response, input.outputFormat ?? "png");
+                const output = {
+                    provider: "openai",
+                    model,
+                    prompt: input.prompt,
+                    size: input.size ?? "auto",
+                    quality: input.quality ?? "auto",
+                    outputFormat: input.outputFormat ?? "png",
+                    background: input.background ?? "auto",
+                    returnedImages: images.length,
+                    images,
+                    raw: compactRawResponse(response),
+                };
+                return {
+                    ok: images.length > 0,
+                    output,
+                    summary: images.length ? `${images.length} image(s) generated` : "OpenAI returned no image data",
+                };
+            }
+            catch (error) {
+                return {
+                    ok: false,
+                    output: {
+                        provider: "openai",
+                        model,
+                        prompt: input.prompt,
+                        error: error instanceof Error ? error.message : String(error),
+                    },
+                };
+            }
+        },
+        mapResult(result) {
+            return compactImageGenerationOutput(result.output);
+        },
+        renderToolResultMessage(result, request) {
+            return createImageGenerationToolResultMessage(result, request?.id ?? "");
+        },
+    };
+}
+export const openAIImageGenerationTool = createOpenAIImageGenerationTool();
+async function callOpenAIImageGeneration(options) {
+    const controller = new AbortController();
+    const abort = () => controller.abort(options.signal?.reason);
+    const timeout = setTimeout(() => controller.abort(new Error(`Image generation request timed out after ${options.timeoutMs}ms`)), options.timeoutMs);
+    options.signal?.addEventListener("abort", abort, { once: true });
+    try {
+        const response = await fetch(`${options.baseUrl}/v1/images/generations`, {
+            method: "POST",
+            headers: {
+                Accept: "application/json",
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${options.apiKey}`,
+            },
+            body: JSON.stringify(buildOpenAIImageRequestBody(options.input)),
+            signal: controller.signal,
+        });
+        const text = await response.text();
+        const body = text ? parseJsonObject(text) : {};
+        if (!response.ok) {
+            throw new Error(`OpenAI image generation HTTP ${response.status}: ${openAIErrorMessage(body) ?? text.slice(0, 1000)}`);
+        }
+        return body;
+    }
+    finally {
+        clearTimeout(timeout);
+        options.signal?.removeEventListener("abort", abort);
+    }
+}
+function buildOpenAIImageRequestBody(input) {
+    return dropUndefined({
+        model: input.model,
+        prompt: input.prompt,
+        n: input.n ?? 1,
+        size: input.size ?? "auto",
+        quality: input.quality ?? "auto",
+        output_format: input.outputFormat ?? "png",
+        background: input.background === "auto" ? undefined : input.background,
+        moderation: input.moderation === "auto" ? undefined : input.moderation,
+    });
+}
+function createImageGenerationToolResultMessage(result, toolUseId) {
+    const output = result.output;
+    const blocks = [{
+            type: "tool_result",
+            toolUseId,
+            name: "image2",
+            ok: result.ok,
+            output: compactImageGenerationOutput(output),
+        }];
+    if (result.ok && isImageGenerationToolOutput(output)) {
+        for (const image of output.images) {
+            blocks.push({ type: "image", mimeType: image.mimeType, data: image.base64, label: `Generated image ${image.index + 1}` });
+        }
+    }
+    return {
+        id: `${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 10)}`,
+        role: "tool_result",
+        createdAt: new Date().toISOString(),
+        blocks,
+        metadata: result.ok ? { generatedImages: true, tool: "image2" } : undefined,
+    };
+}
+function compactImageGenerationOutput(output) {
+    if (!isImageGenerationToolOutput(output))
+        return output;
+    return {
+        ...output,
+        images: output.images.map((image) => ({
+            ...image,
+            base64: `[base64 image omitted from model context; image ${image.index}, ${image.mimeType}]`,
+            dataUrl: `[data URL omitted from model context; image ${image.index}, ${image.mimeType}]`,
+        })),
+    };
+}
+function isImageGenerationToolOutput(value) {
+    return isRecord(value) && value.provider === "openai" && Array.isArray(value.images);
+}
+function extractGeneratedImages(response, format) {
+    const data = Array.isArray(response.data) ? response.data : [];
+    return data.flatMap((item, index) => {
+        if (!isRecord(item))
+            return [];
+        const base64 = stringFrom(item.b64_json ?? item.image_base64 ?? item.base64_json ?? item.base64);
+        if (!base64)
+            return [];
+        const mimeType = `image/${format === "jpg" ? "jpeg" : format}`;
+        return [{
+                index,
+                mimeType,
+                base64,
+                dataUrl: `data:${mimeType};base64,${base64}`,
+                revisedPrompt: stringFrom(item.revised_prompt),
+            }];
+    });
+}
+function resolveApiKey(configured) {
+    return configured?.trim() || process.env.OPENAI_IMAGE_API_KEY?.trim() || process.env.OPENAI_API_KEY?.trim() || undefined;
+}
+function stripTrailingSlash(value) {
+    return value.replace(/\/+$/, "");
+}
+function parsePositiveNumber(value) {
+    if (!value)
+        return undefined;
+    const parsed = Number(value);
+    return Number.isFinite(parsed) && parsed > 0 ? parsed : undefined;
+}
+function parseJsonObject(text) {
+    try {
+        const value = JSON.parse(text);
+        return isRecord(value) ? value : { value };
+    }
+    catch {
+        return { text };
+    }
+}
+function openAIErrorMessage(body) {
+    const error = isRecord(body.error) ? body.error : undefined;
+    return stringFrom(error?.message);
+}
+function compactRawResponse(response) {
+    const data = Array.isArray(response.data) ? response.data : undefined;
+    return {
+        ...response,
+        data: data?.map((item) => {
+            if (!isRecord(item))
+                return item;
+            return {
+                ...item,
+                b64_json: item.b64_json ? "[base64 image omitted]" : undefined,
+                image_base64: item.image_base64 ? "[base64 image omitted]" : undefined,
+                base64_json: item.base64_json ? "[base64 image omitted]" : undefined,
+                base64: item.base64 ? "[base64 image omitted]" : undefined,
+            };
+        }),
+    };
+}
+function dropUndefined(value) {
+    return Object.fromEntries(Object.entries(value).filter(([, entry]) => entry !== undefined));
+}
+function stringFrom(value) {
+    return typeof value === "string" ? value : undefined;
+}
+function isRecord(value) {
+    return !!value && typeof value === "object" && !Array.isArray(value);
+}
+function isImageSize(value) {
+    return value === "auto" || value === "1024x1024" || value === "1536x1024" || value === "1024x1536";
+}
+function isImageQuality(value) {
+    return value === "auto" || value === "low" || value === "medium" || value === "high";
+}
+function isImageFormat(value) {
+    return value === "png" || value === "jpeg" || value === "webp";
+}
+function isImageBackground(value) {
+    return value === "auto" || value === "transparent" || value === "opaque";
+}
+function isImageModeration(value) {
+    return value === "auto" || value === "low";
+}
+//# sourceMappingURL=image-generation-tool.js.map

package/dist/tools/builtins/image-generation-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-generation-tool.js","sourceRoot":"","sources":["../../../src/tools/builtins/image-generation-tool.ts"],"names":[],"mappings":"AAiDA,MAAM,uBAAuB,GAAG,wBAAwB,CAAC;AACzD,MAAM,mBAAmB,GAAG,aAAa,CAAC;AAC1C,MAAM,kBAAkB,GAAG,OAAO,CAAC;AACnC,MAAM,UAAU,GAAG,CAAC,CAAC;AAErB;;;;;;GAMG;AACH,MAAM,UAAU,+BAA+B,CAAC,UAAkD,EAAE;IAClG,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,OAAO,EAAE,CAAC,YAAY,EAAE,gBAAgB,CAAC;QACzC,WAAW,EAAE,2UAA2U;QACxV,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,iJAAiJ,EAAE;gBAC1L,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,kEAAkE,mBAAmB,GAAG,EAAE;gBAChI,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,WAAW,EAAE,iHAAiH,EAAE;gBAC/M,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,WAAW,EAAE,sCAAsC,EAAE;gBACzH,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,EAAE,yCAAyC,EAAE;gBACvH,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,aAAa,EAAE,QAAQ,CAAC,EAAE,WAAW,EAAE,6DAA6D,EAAE;gBACnJ,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,WAAW,EAAE,yEAAyE,EAAE;gBAC7I,CAAC,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,mCAAmC,UAAU,kBAAkB,EAAE;aACrG;YACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;YACpB,oBAAoB,EAAE,KAAK;SAC5B;QACD,QAAQ,EAAE;YACR,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,IAAI;YACb,kBAAkB,EAAE,KAAK;SAC1B;QACD,QAAQ,CAAC,KAAK;YACZ,MAAM,MAAM,GAAG,KAA0C,CAAC;YAC1D,OAAO;gBACL,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,EAAE;gBAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM;gBAC3B,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,MAAM;gBACjC,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,KAAK;gBAC1C,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM;gBACvC,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM;gBACvC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC;aACjB,CAAC;QACJ,CAAC;QACD,aAAa,CAAC,KAAK;YACjB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,+BAA+B,EAAE,CAAC;YACzF,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,8BAA8B,EAAE,CAAC;YACpH,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,8DAA8D,EAAE,CAAC;YAC5H,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,OAAO,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,mDAAmD,EAAE,CAAC;YACvH,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,YAAY,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,gDAAgD,EAAE,CAAC;YACxH,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,UAAU,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,wDAAwD,EAAE,CAAC;YAClI,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,UAAU,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,uCAAuC,EAAE,CAAC;YACjH,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,GAAG,UAAU;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,kCAAkC,UAAU,EAAE,EAAE,CAAC;YAC/I,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QACrD,CAAC;QACD,iBAAiB;YACf,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,WAAW;YACpC,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO;oBACL,EAAE,EAAE,KAAK;oBACT,MAAM,EAAE;wBACN,QAAQ,EAAE,QAAQ;wBAClB,KAAK,EAAE,0EAA0E;qBAClF;iBACF,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAAG,kBAAkB,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,EAAE,IAAI,uBAAuB,CAAC,CAAC;YAC3K,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,IAAI,EAAE,IAAI,mBAAmB,CAAC;YAC5H,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,IAAI,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,kBAAkB,CAAC;YAE3K,WAAW,CAAC,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,gCAAgC,KAAK,EAAE,EAAE,CAAC,CAAC;YAEnG,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,yBAAyB,CAAC;oBAC/C,MAAM;oBACN,OAAO;oBACP,SAAS;oBACT,MAAM,EAAE,OAAO,CAAC,WAAW;oBAC3B,KAAK,EAAE,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE;iBAC3B,CAAC,CAAC;gBACH,MAAM,MAAM,GAAG,sBAAsB,CAAC,QAAQ,EAAE,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,CAAC;gBAC7E,MAAM,MAAM,GAA8B;oBACxC,QAAQ,EAAE,QAAQ;oBAClB,KAAK;oBACL,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,MAAM;oBAC1B,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,MAAM;oBAChC,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,KAAK;oBACzC,UAAU,EAAE,KAAK,CAAC,UAAU,IAAI,MAAM;oBACtC,cAAc,EAAE,MAAM,CAAC,MAAM;oBAC7B,MAAM;oBACN,GAAG,EAAE,kBAAkB,CAAC,QAAQ,CAAC;iBAClC,CAAC;gBACF,OAAO;oBACL,EAAE,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC;oBACrB,MAAM;oBACN,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,qBAAqB,CAAC,CAAC,CAAC,+BAA+B;iBACjG,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO;oBACL,EAAE,EAAE,KAAK;oBACT,MAAM,EAAE;wBACN,QAAQ,EAAE,QAAQ;wBAClB,KAAK;wBACL,MAAM,EAAE,KAAK,CAAC,MAAM;wBACpB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QACD,SAAS,CAAC,MAAM;YACd,OAAO,4BAA4B,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACrD,CAAC;QACD,uBAAuB,CAAC,MAAM,EAAE,OAAO;YACrC,OAAO,sCAAsC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3E,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,yBAAyB,GAAG,+BAA+B,EAAE,CAAC;AAU3E,KAAK,UAAU,yBAAyB,CAAC,OAA4C;IACnF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7D,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,4CAA4C,OAAO,CAAC,SAAS,IAAI,CAAC,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;IACpJ,OAAO,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAEjE,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,CAAC,OAAO,wBAAwB,EAAE;YACvE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,MAAM,EAAE,kBAAkB;gBAC1B,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,OAAO,CAAC,MAAM,EAAE;aAC1C;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,2BAA2B,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAChE,MAAM,EAAE,UAAU,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/C,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,gCAAgC,QAAQ,CAAC,MAAM,KAAK,kBAAkB,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QACzH,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,OAAO,CAAC,CAAC;QACtB,OAAO,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IACtD,CAAC;AACH,CAAC;AAED,SAAS,2BAA2B,CAAC,KAAmD;IACtF,OAAO,aAAa,CAAC;QACnB,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC;QACf,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,MAAM;QAC1B,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,MAAM;QAChC,aAAa,EAAE,KAAK,CAAC,YAAY,IAAI,KAAK;QAC1C,UAAU,EAAE,KAAK,CAAC,UAAU,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU;QACtE,UAAU,EAAE,KAAK,CAAC,UAAU,KAAK,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU;KACvE,CAAC,CAAC;AACL,CAAC;AAED,SAAS,sCAAsC,CAAC,MAAkB,EAAE,SAAiB;IACnF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC7B,MAAM,MAAM,GAAsB,CAAC;YACjC,IAAI,EAAE,aAAa;YACnB,SAAS;YACT,IAAI,EAAE,QAAQ;YACd,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,MAAM,EAAE,4BAA4B,CAAC,MAAM,CAAC;SAC7C,CAAC,CAAC;IAEH,IAAI,MAAM,CAAC,EAAE,IAAI,2BAA2B,CAAC,MAAM,CAAC,EAAE,CAAC;QACrD,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,mBAAmB,KAAK,CAAC,KAAK,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5H,CAAC;IACH,CAAC;IAED,OAAO;QACL,EAAE,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE;QAC3E,IAAI,EAAE,aAAa;QACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,MAAM;QACN,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS;KAC5E,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAe;IACnD,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IACxD,OAAO;QACL,GAAG,MAAM;QACT,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACpC,GAAG,KAAK;YACR,MAAM,EAAE,mDAAmD,KAAK,CAAC,KAAK,KAAK,KAAK,CAAC,QAAQ,GAAG;YAC5F,OAAO,EAAE,+CAA+C,KAAK,CAAC,KAAK,KAAK,KAAK,CAAC,QAAQ,GAAG;SAC1F,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC;AAED,SAAS,2BAA2B,CAAC,KAAc;IACjD,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AACvF,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAiC,EAAE,MAA2C;IAC5G,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/D,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAA2B,EAAE;QAC3D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,OAAO,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC;QACjG,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,SAAS,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAC/D,OAAO,CAAC;gBACN,KAAK;gBACL,QAAQ;gBACR,MAAM;gBACN,OAAO,EAAE,QAAQ,QAAQ,WAAW,MAAM,EAAE;gBAC5C,aAAa,EAAE,UAAU,CAAC,IAAI,CAAC,cAAc,CAAC;aAC/C,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,UAAmB;IACxC,OAAO,UAAU,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,IAAI,EAAE,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;AAC3H,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAa;IACvC,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAyB;IACpD,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACpE,CAAC;AAED,SAAS,eAAe,CAAC,IAAY;IACnC,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/B,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;IAC7C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,IAAI,EAAE,CAAC;IAClB,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,IAA6B;IACvD,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5D,OAAO,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,kBAAkB,CAAC,QAAiC;IAC3D,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACtE,OAAO;QACL,GAAG,QAAQ;QACX,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAAE,OAAO,IAAI,CAAC;YACjC,OAAO;gBACL,GAAG,IAAI;gBACP,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,SAAS;gBAC9D,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,SAAS;gBACtE,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,SAAS;gBACpE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,wBAAwB,CAAC,CAAC,CAAC,SAAS;aAC3D,CAAC;QACJ,CAAC,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAA8B;IACnD,OAAO,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC;AAC9F,CAAC;AAED,SAAS,UAAU,CAAC,KAAc;IAChC,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACvD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,CAAC,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AACvE,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,WAAW,IAAI,KAAK,KAAK,WAAW,IAAI,KAAK,KAAK,WAAW,CAAC;AACrG,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,KAAK,IAAI,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,MAAM,CAAC;AACvF,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,KAAK,KAAK,KAAK,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,MAAM,CAAC;AACjE,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,OAAO,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,aAAa,IAAI,KAAK,KAAK,QAAQ,CAAC;AAC3E,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,OAAO,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,KAAK,CAAC;AAC7C,CAAC"}

package/dist/tools/registry.d.ts

@@ -10,6 +10,7 @@ export declare class ToolRegistry {
     register(tool: Tool<any>): void;
     get(name: string): Tool<any> | undefined;
     getByAlias(name: string): Tool<any> | undefined;
+    unregister(name: string): boolean;
     definitions(context?: ToolUseContext, options?: ToolPoolOptions): ToolDefinition[];
     names(): string[];
     list(context?: ToolUseContext, options?: ToolPoolOptions): Tool<any>[];

package/dist/tools/registry.js

@@ -19,6 +19,17 @@ export class ToolRegistry {
         const canonical = this.aliases.get(name);
         return canonical ? this.tools.get(canonical) : undefined;
     }
+    unregister(name) {
+        const tool = this.tools.get(name);
+        if (!tool)
+            return false;
+        this.tools.delete(name);
+        for (const alias of tool.aliases ?? []) {
+            if (this.aliases.get(alias) === name)
+                this.aliases.delete(alias);
+        }
+        return true;
+    }
     definitions(context, options = {}) {
         return this.visibleTools(context, options).map((tool) => ({
             name: tool.name,