@tschmz/imp 0.1.0

package/dist/runtime/create-oauth-api-key-resolver.js

@@ -0,0 +1,66 @@
+import { mkdir, readFile, writeFile } from "node:fs/promises";
+import { dirname } from "node:path";
+import { getOAuthApiKey, getOAuthProvider } from "@mariozechner/pi-ai/oauth";
+export function createOAuthApiKeyResolver(authFilePath, logger, dependencies = {}) {
+    const readTextFile = dependencies.readTextFile ?? defaultReadTextFile;
+    const writeTextFile = dependencies.writeTextFile ?? defaultWriteTextFile;
+    const resolveOAuthApiKey = dependencies.getOAuthApiKeyFn ?? getOAuthApiKey;
+    return async (provider) => {
+        if (!authFilePath || !getOAuthProvider(provider)) {
+            return undefined;
+        }
+        let auth;
+        try {
+            auth = await loadAuthFile(authFilePath, readTextFile);
+        }
+        catch (error) {
+            await logger?.error("failed to load oauth credentials", undefined, error);
+            return undefined;
+        }
+        try {
+            const result = await resolveOAuthApiKey(provider, auth);
+            if (!result) {
+                return undefined;
+            }
+            auth[provider] = {
+                type: "oauth",
+                ...result.newCredentials,
+            };
+            await writeTextFile(authFilePath, `${JSON.stringify(auth, null, 2)}\n`);
+            return result.apiKey;
+        }
+        catch (error) {
+            await logger?.error("failed to refresh oauth credentials", undefined, error);
+            return undefined;
+        }
+    };
+}
+async function loadAuthFile(path, readTextFile) {
+    try {
+        const raw = await readTextFile(path);
+        const parsed = JSON.parse(raw);
+        return isRecord(parsed)
+            ? parsed
+            : {};
+    }
+    catch (error) {
+        if (isMissingFileError(error)) {
+            return {};
+        }
+        throw error;
+    }
+}
+async function defaultReadTextFile(path) {
+    return readFile(path, "utf8");
+}
+async function defaultWriteTextFile(path, content) {
+    await mkdir(dirname(path), { recursive: true });
+    await writeFile(path, content, "utf8");
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null;
+}
+function isMissingFileError(error) {
+    return error instanceof Error && "code" in error && error.code === "ENOENT";
+}
+//# sourceMappingURL=create-oauth-api-key-resolver.js.map

package/dist/runtime/create-oauth-api-key-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create-oauth-api-key-resolver.js","sourceRoot":"","sources":["../../src/runtime/create-oauth-api-key-resolver.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAc7E,MAAM,UAAU,yBAAyB,CACvC,YAAgC,EAChC,MAAe,EACf,eAAgD,EAAE;IAElD,MAAM,YAAY,GAAG,YAAY,CAAC,YAAY,IAAI,mBAAmB,CAAC;IACtE,MAAM,aAAa,GAAG,YAAY,CAAC,aAAa,IAAI,oBAAoB,CAAC;IACzE,MAAM,kBAAkB,GAAG,YAAY,CAAC,gBAAgB,IAAI,cAAc,CAAC;IAE3E,OAAO,KAAK,EAAE,QAAQ,EAAE,EAAE;QACxB,IAAI,CAAC,YAAY,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,IAAmE,CAAC;QACxE,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,YAAY,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,MAAM,EAAE,KAAK,CAAC,kCAAkC,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YAC1E,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YACxD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,GAAG;gBACf,IAAI,EAAE,OAAO;gBACb,GAAG,MAAM,CAAC,cAAc;aACzB,CAAC;YACF,MAAM,aAAa,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;YACxE,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,MAAM,EAAE,KAAK,CAAC,qCAAqC,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YAC7E,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,IAAY,EACZ,YAA+C;IAE/C,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAY,CAAC;QAC1C,OAAO,QAAQ,CAAC,MAAM,CAAC;YACrB,CAAC,CAAE,MAAwE;YAC3E,CAAC,CAAC,EAAE,CAAC;IACT,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,IAAY;IAC7C,OAAO,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AAChC,CAAC;AAED,KAAK,UAAU,oBAAoB,CAAC,IAAY,EAAE,OAAe;IAC/D,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAChD,MAAM,SAAS,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAc;IACxC,OAAO,KAAK,YAAY,KAAK,IAAI,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC;AAC9E,CAAC"}

package/dist/runtime/create-pi-agent-engine.d.ts

@@ -0,0 +1,28 @@
+import type { AgentOptions } from "@mariozechner/pi-agent-core";
+import type { AgentDefinition } from "../domain/agent.js";
+import type { Logger } from "../logging/types.js";
+import type { AgentEngineLifecycleHooks, HookRegistration } from "../plugins/types.js";
+import type { ToolRegistry } from "../tools/registry.js";
+import type { AgentHandle } from "./agent-execution.js";
+import { type ResolvedMcpTools } from "./mcp-tool-runtime.js";
+import { type ModelResolver } from "./model-resolution.js";
+import { type PromptTemplateSystemContext } from "./prompt-template.js";
+import type { AgentEngine } from "./types.js";
+import { type WorkingDirectoryState } from "./tool-resolution.js";
+interface PiAgentEngineDependencies {
+    logger?: Logger;
+    resolveModel?: ModelResolver;
+    getApiKey?: (provider: string, agent: AgentDefinition) => Promise<string | undefined> | string | undefined;
+    createAgent?: (options: AgentOptions) => AgentHandle;
+    readTextFile?: (path: string) => Promise<string>;
+    getContextFileFingerprint?: (path: string) => Promise<string>;
+    toolRegistry?: ToolRegistry;
+    createBuiltInToolRegistry?: (workingDirectory: string | WorkingDirectoryState, agent?: AgentDefinition) => ToolRegistry;
+    resolveMcpTools?: (agent: AgentDefinition, options: {
+        logger?: Logger;
+    }) => Promise<ResolvedMcpTools>;
+    promptTemplateSystemContext?: PromptTemplateSystemContext;
+    agentEngineHooks?: ReadonlyArray<HookRegistration<AgentEngineLifecycleHooks>>;
+}
+export declare function createPiAgentEngine(dependencies?: PiAgentEngineDependencies): AgentEngine;
+export { createBuiltInToolRegistry, mergeShellPathEntries, resolveAgentTools, resolveWorkingDirectory, type WorkingDirectoryState, } from "./tool-resolution.js";

package/dist/runtime/create-pi-agent-engine.js

@@ -0,0 +1,150 @@
+import { readFile, stat } from "node:fs/promises";
+import { createHookRunner } from "../plugins/hook-runner.js";
+import { createMcpToolCache } from "./mcp-tool-cache.js";
+import { resolveMcpTools } from "./mcp-tool-runtime.js";
+import { defaultResolveModel } from "./model-resolution.js";
+import { createDefaultPromptTemplateSystemContext, } from "./prompt-template.js";
+import { executeAgentStage } from "./pipeline/execute-agent-stage.js";
+import { resolveModelStage } from "./pipeline/resolve-model-stage.js";
+import { resolvePromptStage } from "./pipeline/resolve-prompt-stage.js";
+import { resolveToolsStage } from "./pipeline/resolve-tools-stage.js";
+import { InMemoryCacheStrategy, SystemPromptCache } from "./system-prompt-cache.js";
+import { createBuiltInToolRegistry, } from "./tool-resolution.js";
+export function createPiAgentEngine(dependencies = {}) {
+    const resolveModel = dependencies.resolveModel ?? defaultResolveModel;
+    const readTextFile = dependencies.readTextFile ?? defaultReadTextFile;
+    const getContextFileFingerprint = dependencies.getContextFileFingerprint ?? defaultGetContextFileFingerprint;
+    const buildToolRegistry = dependencies.createBuiltInToolRegistry ?? createBuiltInToolRegistry;
+    const loadMcpTools = dependencies.resolveMcpTools ?? resolveMcpTools;
+    const promptTemplateSystemContext = dependencies.promptTemplateSystemContext ?? createDefaultPromptTemplateSystemContext();
+    const logger = dependencies.logger;
+    const systemPromptCache = new SystemPromptCache({
+        readTextFile,
+        getContextFileFingerprint,
+        strategy: new InMemoryCacheStrategy(),
+    });
+    const mcpToolCache = createMcpToolCache({
+        logger,
+        resolveMcpTools: loadMcpTools,
+    });
+    const hookRunner = createHookRunner(dependencies.agentEngineHooks, { logger });
+    let closed = false;
+    return {
+        async run(input) {
+            if (closed) {
+                throw new Error("Agent engine is closed.");
+            }
+            const context = {
+                endpointId: input.message.endpointId,
+                transport: input.message.conversation.transport,
+                conversationId: input.message.conversation.externalId,
+                messageId: input.message.messageId,
+                correlationId: input.message.correlationId,
+                agentId: input.agent.id,
+            };
+            const startedAt = Date.now();
+            try {
+                await hookRunner.run("onAgentEngineRunStart", (hooks) => hooks.onAgentEngineRunStart, { input });
+                const runContext = {
+                    input,
+                    agent: input.agent,
+                    conversation: input.conversation,
+                };
+                await logPipelineEvent(logger, context, { step: "model-resolution", status: "started" });
+                const modelContext = resolveModelStage(runContext, { resolveModel });
+                await logPipelineEvent(logger, context, { step: "model-resolution", status: "completed" });
+                await logPipelineEvent(logger, context, { step: "system-prompt-resolution", status: "started" });
+                const promptContext = await resolvePromptStage(modelContext, {
+                    readTextFile,
+                    systemPromptCache,
+                    promptTemplateSystemContext,
+                });
+                await logPipelineEvent(logger, context, {
+                    step: "system-prompt-resolution",
+                    status: "completed",
+                    cacheHit: promptContext.systemPromptResolution.cacheHit,
+                });
+                await logPipelineEvent(logger, context, { step: "tool-resolution", status: "started" });
+                const toolContext = await resolveToolsStage(promptContext, {
+                    toolRegistry: dependencies.toolRegistry,
+                    createBuiltInToolRegistry: buildToolRegistry,
+                    mcpToolCache,
+                });
+                await logPipelineEvent(logger, context, {
+                    step: "tool-resolution",
+                    status: "completed",
+                    initialWorkingDirectory: toolContext.initialWorkingDirectory,
+                    configuredBuiltInTools: toolContext.toolResolution.configuredBuiltInTools,
+                    resolvedBuiltInTools: toolContext.toolResolution.resolvedBuiltInTools,
+                    missingBuiltInTools: toolContext.toolResolution.missingBuiltInTools,
+                    configuredMcpServers: toolContext.toolResolution.configuredMcpServers,
+                    initializedMcpServers: toolContext.toolResolution.initializedMcpServers,
+                    failedMcpServers: toolContext.toolResolution.failedMcpServers,
+                    resolvedMcpTools: toolContext.toolResolution.resolvedMcpTools,
+                    resolvedTools: toolContext.toolResolution.resolvedTools,
+                });
+                await logPipelineEvent(logger, context, { step: "agent-execution", status: "started" });
+                const executionContext = await executeAgentStage(toolContext, {
+                    createAgent: dependencies.createAgent,
+                    getApiKey: dependencies.getApiKey,
+                });
+                await logPipelineEvent(logger, context, { step: "agent-execution", status: "completed" });
+                await logPipelineEvent(logger, context, {
+                    step: "pipeline",
+                    status: "completed",
+                    durationMs: Date.now() - startedAt,
+                });
+                await hookRunner.run("onAgentEngineRunSuccess", (hooks) => hooks.onAgentEngineRunSuccess, {
+                    input,
+                    result: executionContext.result,
+                    durationMs: Date.now() - startedAt,
+                });
+                return executionContext.result;
+            }
+            catch (error) {
+                await logPipelineEvent(logger, context, {
+                    step: "pipeline",
+                    status: "failed",
+                    durationMs: Date.now() - startedAt,
+                    errorType: error instanceof Error ? error.name : typeof error,
+                    error,
+                });
+                await hookRunner.runErrorHook("onAgentEngineRunError", (hooks) => hooks.onAgentEngineRunError, {
+                    input,
+                    error,
+                    durationMs: Date.now() - startedAt,
+                });
+                throw error;
+            }
+        },
+        async close() {
+            if (closed) {
+                return;
+            }
+            closed = true;
+            await mcpToolCache.close();
+        },
+    };
+}
+async function logPipelineEvent(logger, context, event) {
+    await logger?.debug("agent-engine.pipeline", {
+        ...context,
+        ...event,
+    });
+    if (event.status === "failed") {
+        await logger?.error("agent engine run failed", {
+            ...context,
+            durationMs: event.durationMs,
+            errorType: event.errorType,
+        }, event.error instanceof Error ? event.error : new Error(String(event.error ?? "Unknown pipeline error")));
+    }
+}
+async function defaultReadTextFile(path) {
+    return readFile(path, "utf8");
+}
+async function defaultGetContextFileFingerprint(path) {
+    const fileStats = await stat(path);
+    return `${fileStats.mtimeMs}:${fileStats.size}`;
+}
+export { createBuiltInToolRegistry, mergeShellPathEntries, resolveAgentTools, resolveWorkingDirectory, } from "./tool-resolution.js";
+//# sourceMappingURL=create-pi-agent-engine.js.map

package/dist/runtime/create-pi-agent-engine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create-pi-agent-engine.js","sourceRoot":"","sources":["../../src/runtime/create-pi-agent-engine.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAIlD,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAI7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAyB,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EAAE,mBAAmB,EAAsB,MAAM,uBAAuB,CAAC;AAChF,OAAO,EACL,wCAAwC,GAEzC,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AACtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AACxE,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AACtE,OAAO,EAAE,qBAAqB,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAEpF,OAAO,EACL,yBAAyB,GAE1B,MAAM,sBAAsB,CAAC;AAsD9B,MAAM,UAAU,mBAAmB,CACjC,eAA0C,EAAE;IAE5C,MAAM,YAAY,GAAG,YAAY,CAAC,YAAY,IAAI,mBAAmB,CAAC;IACtE,MAAM,YAAY,GAAG,YAAY,CAAC,YAAY,IAAI,mBAAmB,CAAC;IACtE,MAAM,yBAAyB,GAC7B,YAAY,CAAC,yBAAyB,IAAI,gCAAgC,CAAC;IAC7E,MAAM,iBAAiB,GACrB,YAAY,CAAC,yBAAyB,IAAI,yBAAyB,CAAC;IACtE,MAAM,YAAY,GAAG,YAAY,CAAC,eAAe,IAAI,eAAe,CAAC;IACrE,MAAM,2BAA2B,GAC/B,YAAY,CAAC,2BAA2B,IAAI,wCAAwC,EAAE,CAAC;IACzF,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;IACnC,MAAM,iBAAiB,GAAG,IAAI,iBAAiB,CAAC;QAC9C,YAAY;QACZ,yBAAyB;QACzB,QAAQ,EAAE,IAAI,qBAAqB,EAAU;KAC9C,CAAC,CAAC;IACH,MAAM,YAAY,GAAG,kBAAkB,CAAC;QACtC,MAAM;QACN,eAAe,EAAE,YAAY;KAC9B,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,gBAAgB,CAAC,YAAY,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IAC/E,IAAI,MAAM,GAAG,KAAK,CAAC;IAEnB,OAAO;QACL,KAAK,CAAC,GAAG,CAAC,KAAK;YACb,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAC7C,CAAC;YAED,MAAM,OAAO,GAAqB;gBAChC,UAAU,EAAE,KAAK,CAAC,OAAO,CAAC,UAAU;gBACpC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS;gBAC/C,cAAc,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU;gBACrD,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,SAAS;gBAClC,aAAa,EAAE,KAAK,CAAC,OAAO,CAAC,aAAa;gBAC1C,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE;aACxB,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,IAAI,CAAC;gBACH,MAAM,UAAU,CAAC,GAAG,CAClB,uBAAuB,EACvB,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,qBAAqB,EACtC,EAAE,KAAK,EAAE,CACV,CAAC;gBAEF,MAAM,UAAU,GAAoB;oBAClC,KAAK;oBACL,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,YAAY,EAAE,KAAK,CAAC,YAAY;iBACjC,CAAC;gBAEF,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;gBACzF,MAAM,YAAY,GAAG,iBAAiB,CAAC,UAAU,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;gBACrE,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;gBAE3F,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,0BAA0B,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;gBACjG,MAAM,aAAa,GAAG,MAAM,kBAAkB,CAAC,YAAY,EAAE;oBAC3D,YAAY;oBACZ,iBAAiB;oBACjB,2BAA2B;iBAC5B,CAAC,CAAC;gBACH,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE;oBACtC,IAAI,EAAE,0BAA0B;oBAChC,MAAM,EAAE,WAAW;oBACnB,QAAQ,EAAE,aAAa,CAAC,sBAAsB,CAAC,QAAQ;iBACxD,CAAC,CAAC;gBAEH,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;gBACxF,MAAM,WAAW,GAAG,MAAM,iBAAiB,CAAC,aAAa,EAAE;oBACzD,YAAY,EAAE,YAAY,CAAC,YAAY;oBACvC,yBAAyB,EAAE,iBAAiB;oBAC5C,YAAY;iBACb,CAAC,CAAC;gBACH,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE;oBACtC,IAAI,EAAE,iBAAiB;oBACvB,MAAM,EAAE,WAAW;oBACnB,uBAAuB,EAAE,WAAW,CAAC,uBAAuB;oBAC5D,sBAAsB,EAAE,WAAW,CAAC,cAAc,CAAC,sBAAsB;oBACzE,oBAAoB,EAAE,WAAW,CAAC,cAAc,CAAC,oBAAoB;oBACrE,mBAAmB,EAAE,WAAW,CAAC,cAAc,CAAC,mBAAmB;oBACnE,oBAAoB,EAAE,WAAW,CAAC,cAAc,CAAC,oBAAoB;oBACrE,qBAAqB,EAAE,WAAW,CAAC,cAAc,CAAC,qBAAqB;oBACvE,gBAAgB,EAAE,WAAW,CAAC,cAAc,CAAC,gBAAgB;oBAC7D,gBAAgB,EAAE,WAAW,CAAC,cAAc,CAAC,gBAAgB;oBAC7D,aAAa,EAAE,WAAW,CAAC,cAAc,CAAC,aAAa;iBACxD,CAAC,CAAC;gBAEH,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;gBACxF,MAAM,gBAAgB,GAAG,MAAM,iBAAiB,CAAC,WAAW,EAAE;oBAC5D,WAAW,EAAE,YAAY,CAAC,WAAW;oBACrC,SAAS,EAAE,YAAY,CAAC,SAAS;iBAClC,CAAC,CAAC;gBACH,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;gBAE1F,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE;oBACtC,IAAI,EAAE,UAAU;oBAChB,MAAM,EAAE,WAAW;oBACnB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBACnC,CAAC,CAAC;gBAEH,MAAM,UAAU,CAAC,GAAG,CAClB,yBAAyB,EACzB,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,uBAAuB,EACxC;oBACE,KAAK;oBACL,MAAM,EAAE,gBAAgB,CAAC,MAAM;oBAC/B,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBACnC,CACF,CAAC;gBAEF,OAAO,gBAAgB,CAAC,MAAM,CAAC;YACjC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE;oBACtC,IAAI,EAAE,UAAU;oBAChB,MAAM,EAAE,QAAQ;oBAChB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;oBAClC,SAAS,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,KAAK;oBAC7D,KAAK;iBACN,CAAC,CAAC;gBACH,MAAM,UAAU,CAAC,YAAY,CAC3B,uBAAuB,EACvB,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,qBAAqB,EACtC;oBACE,KAAK;oBACL,KAAK;oBACL,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBACnC,CACF,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,KAAK,CAAC,KAAK;YACT,IAAI,MAAM,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YAED,MAAM,GAAG,IAAI,CAAC;YACd,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;QAC7B,CAAC;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,gBAAgB,CAC7B,MAA0B,EAC1B,OAAyB,EACzB,KAAoB;IAEpB,MAAM,MAAM,EAAE,KAAK,CAAC,uBAAuB,EAAE;QAC3C,GAAG,OAAO;QACV,GAAG,KAAK;KACT,CAAC,CAAC;IAEH,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,MAAM,EAAE,KAAK,CACjB,yBAAyB,EACzB;YACE,GAAG,OAAO;YACV,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,SAAS,EAAE,KAAK,CAAC,SAAS;SAC3B,EACD,KAAK,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,IAAI,wBAAwB,CAAC,CAAC,CACxG,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,IAAY;IAC7C,OAAO,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AAChC,CAAC;AAED,KAAK,UAAU,gCAAgC,CAAC,IAAY;IAC1D,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC;IACnC,OAAO,GAAG,SAAS,CAAC,OAAO,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC;AAClD,CAAC;AAED,OAAO,EACL,yBAAyB,EACzB,qBAAqB,EACrB,iBAAiB,EACjB,uBAAuB,GAExB,MAAM,sBAAsB,CAAC"}

package/dist/runtime/mcp-tool-cache.d.ts

@@ -0,0 +1,13 @@
+import type { AgentDefinition } from "../domain/agent.js";
+import type { Logger } from "../logging/types.js";
+import type { ResolvedMcpTools } from "./mcp-tool-runtime.js";
+export interface McpToolCache {
+    resolve(agent: AgentDefinition): Promise<ResolvedMcpTools>;
+    close(): Promise<void>;
+}
+export declare function createMcpToolCache(options: {
+    logger?: Logger;
+    resolveMcpTools: (agent: AgentDefinition, options: {
+        logger?: Logger;
+    }) => Promise<ResolvedMcpTools>;
+}): McpToolCache;

package/dist/runtime/mcp-tool-cache.js

@@ -0,0 +1,43 @@
+export function createMcpToolCache(options) {
+    const cache = new Map();
+    return {
+        async resolve(agent) {
+            if (!agent.mcp || agent.mcp.servers.length === 0) {
+                return {
+                    tools: [],
+                    initializedServerIds: [],
+                    failedServerIds: [],
+                    async close() { },
+                };
+            }
+            const cached = cache.get(agent.id);
+            if (cached) {
+                await options.logger?.debug(`reusing cached MCP runtime for agent "${agent.id}"`);
+                return cached.promise;
+            }
+            await options.logger?.debug(`initializing cached MCP runtime for agent "${agent.id}"`);
+            const promise = options.resolveMcpTools(agent, { logger: options.logger }).catch((error) => {
+                cache.delete(agent.id);
+                throw error;
+            });
+            cache.set(agent.id, { promise });
+            const resolution = await promise;
+            await options.logger?.debug(`cached MCP runtime ready for agent "${agent.id}"`);
+            return resolution;
+        },
+        async close() {
+            const resolutions = [...cache.values()];
+            cache.clear();
+            if (resolutions.length === 0) {
+                return;
+            }
+            await options.logger?.debug("closing cached MCP runtimes");
+            await Promise.all(resolutions.map(async ({ promise }) => {
+                const resolution = await promise;
+                await resolution.close();
+            }));
+            await options.logger?.debug("closed cached MCP runtimes");
+        },
+    };
+}
+//# sourceMappingURL=mcp-tool-cache.js.map

package/dist/runtime/mcp-tool-cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-tool-cache.js","sourceRoot":"","sources":["../../src/runtime/mcp-tool-cache.ts"],"names":[],"mappings":"AAaA,MAAM,UAAU,kBAAkB,CAAC,OAQlC;IACC,MAAM,KAAK,GAAG,IAAI,GAAG,EAAmC,CAAC;IAEzD,OAAO;QACL,KAAK,CAAC,OAAO,CAAC,KAAK;YACjB,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjD,OAAO;oBACL,KAAK,EAAE,EAAE;oBACT,oBAAoB,EAAE,EAAE;oBACxB,eAAe,EAAE,EAAE;oBACnB,KAAK,CAAC,KAAK,KAAI,CAAC;iBACjB,CAAC;YACJ,CAAC;YAED,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACnC,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,yCAAyC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;gBAClF,OAAO,MAAM,CAAC,OAAO,CAAC;YACxB,CAAC;YAED,MAAM,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,8CAA8C,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;YACvF,MAAM,OAAO,GAAG,OAAO,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACzF,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACvB,MAAM,KAAK,CAAC;YACd,CAAC,CAAC,CAAC;YACH,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAEjC,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC;YACjC,MAAM,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,uCAAuC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;YAChF,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,KAAK,CAAC,KAAK;YACT,MAAM,WAAW,GAAG,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;YACxC,KAAK,CAAC,KAAK,EAAE,CAAC;YAEd,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,OAAO;YACT,CAAC;YAED,MAAM,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,6BAA6B,CAAC,CAAC;YAE3D,MAAM,OAAO,CAAC,GAAG,CACf,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;gBACpC,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC;gBACjC,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;YAC3B,CAAC,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAC5D,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/runtime/mcp-tool-runtime.d.ts

@@ -0,0 +1,76 @@
+import { type StdioServerParameters } from "@modelcontextprotocol/sdk/client/stdio.js";
+import type { ToolDefinition } from "../tools/types.js";
+import type { AgentDefinition } from "../domain/agent.js";
+import type { Logger } from "../logging/types.js";
+interface McpListedTool {
+    name: string;
+    title?: string;
+    description?: string;
+    inputSchema?: Record<string, unknown>;
+}
+type McpContentBlock = {
+    type: "text";
+    text: string;
+} | {
+    type: "image";
+    data: string;
+    mimeType: string;
+} | {
+    type: "audio";
+    data: string;
+    mimeType: string;
+} | {
+    type: "resource";
+    resource: {
+        uri: string;
+        text: string;
+        mimeType?: string;
+    } | {
+        uri: string;
+        blob: string;
+        mimeType?: string;
+    };
+} | {
+    type: "resource_link";
+    uri: string;
+    name: string;
+    mimeType?: string;
+    description?: string;
+    title?: string;
+};
+interface McpCallToolSuccess {
+    content: McpContentBlock[];
+    structuredContent?: Record<string, unknown>;
+    isError?: boolean;
+}
+interface McpCallToolCompatibilityResult {
+    toolResult: unknown;
+}
+interface McpClientLike {
+    connect(transport: unknown): Promise<void>;
+    listTools(params?: {
+        cursor?: string;
+    }): Promise<{
+        tools: McpListedTool[];
+        nextCursor?: string;
+    }>;
+    callTool(params: {
+        name: string;
+        arguments?: Record<string, unknown>;
+    }): Promise<McpCallToolSuccess | McpCallToolCompatibilityResult>;
+    close(): Promise<void>;
+}
+interface McpToolRuntimeDependencies {
+    createClient?: () => McpClientLike;
+    createTransport?: (server: StdioServerParameters) => unknown;
+}
+export interface ResolvedMcpTools {
+    tools: ToolDefinition[];
+    initializedServerIds: string[];
+    failedServerIds: string[];
+    close(): Promise<void>;
+}
+export declare function resolveMcpTools(agent: AgentDefinition, options?: {
+    logger?: Logger;
+} & McpToolRuntimeDependencies): Promise<ResolvedMcpTools>;
+export {};

package/dist/runtime/mcp-tool-runtime.js

@@ -0,0 +1,180 @@
+import { Client } from "@modelcontextprotocol/sdk/client";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+const MCP_CLIENT_INFO = {
+    name: "imp",
+    version: "0.1.0",
+};
+export async function resolveMcpTools(agent, options = {}) {
+    const servers = agent.mcp?.servers ?? [];
+    if (servers.length === 0) {
+        return createEmptyResolution();
+    }
+    const createClient = options.createClient ?? createDefaultClient;
+    const createTransport = options.createTransport ?? createDefaultTransport;
+    const clients = [];
+    const tools = [];
+    const initializedServerIds = [];
+    const failedServerIds = [];
+    for (const server of servers) {
+        const client = createClient();
+        const transport = createTransport({
+            command: server.command,
+            ...(server.args ? { args: server.args } : {}),
+            ...(server.env ? { env: server.env } : {}),
+            ...(server.cwd ? { cwd: server.cwd } : {}),
+            stderr: "pipe",
+        });
+        try {
+            await client.connect(transport);
+            const listedTools = await listAllTools(client);
+            for (const tool of listedTools) {
+                tools.push(createMcpToolDefinition(server.id, tool, client));
+            }
+            clients.push(client);
+            initializedServerIds.push(server.id);
+            await options.logger?.debug(`initialized MCP server "${server.id}" for agent "${agent.id}"`);
+        }
+        catch (error) {
+            failedServerIds.push(server.id);
+            await safeCloseClient(client);
+            await options.logger?.error(`failed to initialize MCP server "${server.id}" for agent "${agent.id}"`, undefined, error);
+        }
+    }
+    return {
+        tools,
+        initializedServerIds,
+        failedServerIds,
+        async close() {
+            await Promise.all(clients.map(async (client) => safeCloseClient(client)));
+        },
+    };
+}
+function createDefaultClient() {
+    return new Client(MCP_CLIENT_INFO);
+}
+function createDefaultTransport(server) {
+    return new StdioClientTransport(server);
+}
+function createEmptyResolution() {
+    return {
+        tools: [],
+        initializedServerIds: [],
+        failedServerIds: [],
+        async close() { },
+    };
+}
+async function listAllTools(client) {
+    const tools = [];
+    let cursor;
+    do {
+        const response = await client.listTools(cursor ? { cursor } : undefined);
+        tools.push(...response.tools);
+        cursor = response.nextCursor;
+    } while (cursor);
+    return tools;
+}
+function createMcpToolDefinition(serverId, tool, client) {
+    const prefixedName = `${serverId}__${tool.name}`;
+    return {
+        name: prefixedName,
+        label: tool.title ?? prefixedName,
+        description: tool.description ?? `Tool imported from MCP server "${serverId}".`,
+        parameters: normalizeToolSchema(tool.inputSchema),
+        async execute(_toolCallId, params) {
+            const result = await client.callTool({
+                name: tool.name,
+                ...(isRecord(params) ? { arguments: params } : {}),
+            });
+            if ("toolResult" in result) {
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: JSON.stringify(result.toolResult, null, 2),
+                        },
+                    ],
+                    details: {
+                        serverId,
+                        toolName: tool.name,
+                        toolResult: result.toolResult,
+                    },
+                };
+            }
+            if (result.isError) {
+                throw new Error(getErrorMessage(result.content, prefixedName));
+            }
+            return {
+                content: mapContentBlocks(result.content),
+                details: {
+                    serverId,
+                    toolName: tool.name,
+                    ...(result.structuredContent ? { structuredContent: result.structuredContent } : {}),
+                },
+            };
+        },
+    };
+}
+function normalizeToolSchema(inputSchema) {
+    return (inputSchema && typeof inputSchema === "object"
+        ? inputSchema
+        : {
+            type: "object",
+            properties: {},
+            additionalProperties: true,
+        });
+}
+function mapContentBlocks(content) {
+    const mapped = [];
+    for (const block of content) {
+        switch (block.type) {
+            case "text":
+                mapped.push({ type: "text", text: block.text });
+                break;
+            case "image":
+                mapped.push({ type: "image", data: block.data, mimeType: block.mimeType });
+                break;
+            case "audio":
+                mapped.push({
+                    type: "text",
+                    text: `[audio output omitted: ${block.mimeType}]`,
+                });
+                break;
+            case "resource":
+                if ("text" in block.resource) {
+                    mapped.push({ type: "text", text: block.resource.text });
+                    break;
+                }
+                mapped.push({
+                    type: "text",
+                    text: `[resource blob omitted: ${block.resource.uri}]`,
+                });
+                break;
+            case "resource_link":
+                mapped.push({
+                    type: "text",
+                    text: block.uri,
+                });
+                break;
+        }
+    }
+    return mapped;
+}
+function getErrorMessage(content, prefixedToolName) {
+    const text = content
+        .flatMap((block) => (block.type === "text" ? [block.text] : []))
+        .join("\n")
+        .trim();
+    return text.length > 0 ? text : `MCP tool "${prefixedToolName}" failed.`;
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+async function safeCloseClient(client) {
+    try {
+        await client.close();
+    }
+    catch {
+        // Ignore shutdown errors so runtime cleanup does not mask prior failures.
+    }
+}
+//# sourceMappingURL=mcp-tool-runtime.js.map

package/dist/runtime/mcp-tool-runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-tool-runtime.js","sourceRoot":"","sources":["../../src/runtime/mcp-tool-runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,kCAAkC,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAA8B,MAAM,2CAA2C,CAAC;AAoE7G,MAAM,eAAe,GAAG;IACtB,IAAI,EAAE,KAAK;IACX,OAAO,EAAE,OAAO;CACR,CAAC;AAEX,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,KAAsB,EACtB,UAEiC,EAAE;IAEnC,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,EAAE,OAAO,IAAI,EAAE,CAAC;IACzC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,qBAAqB,EAAE,CAAC;IACjC,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,mBAAmB,CAAC;IACjE,MAAM,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,sBAAsB,CAAC;IAC1E,MAAM,OAAO,GAAoB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAqB,EAAE,CAAC;IACnC,MAAM,oBAAoB,GAAa,EAAE,CAAC;IAC1C,MAAM,eAAe,GAAa,EAAE,CAAC;IAErC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;QAC9B,MAAM,SAAS,GAAG,eAAe,CAAC;YAChC,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1C,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1C,MAAM,EAAE,MAAM;SACf,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAChC,MAAM,WAAW,GAAG,MAAM,YAAY,CAAC,MAAM,CAAC,CAAC;YAE/C,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;gBAC/B,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;YAC/D,CAAC;YAED,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACrB,oBAAoB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACrC,MAAM,OAAO,CAAC,MAAM,EAAE,KAAK,CACzB,2BAA2B,MAAM,CAAC,EAAE,gBAAgB,KAAK,CAAC,EAAE,GAAG,CAChE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAChC,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;YAC9B,MAAM,OAAO,CAAC,MAAM,EAAE,KAAK,CACzB,oCAAoC,MAAM,CAAC,EAAE,gBAAgB,KAAK,CAAC,EAAE,GAAG,EACxE,SAAS,EACT,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO;QACL,KAAK;QACL,oBAAoB;QACpB,eAAe;QACf,KAAK,CAAC,KAAK;YACT,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAC5E,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB;IAC1B,OAAO,IAAI,MAAM,CAAC,eAAe,CAAC,CAAC;AACrC,CAAC;AAED,SAAS,sBAAsB,CAAC,MAA6B;IAC3D,OAAO,IAAI,oBAAoB,CAAC,MAAM,CAAC,CAAC;AAC1C,CAAC;AAED,SAAS,qBAAqB;IAC5B,OAAO;QACL,KAAK,EAAE,EAAE;QACT,oBAAoB,EAAE,EAAE;QACxB,eAAe,EAAE,EAAE;QACnB,KAAK,CAAC,KAAK,KAAI,CAAC;KACjB,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,MAAqB;IAC/C,MAAM,KAAK,GAAoB,EAAE,CAAC;IAClC,IAAI,MAA0B,CAAC;IAE/B,GAAG,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACzE,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC9B,MAAM,GAAG,QAAQ,CAAC,UAAU,CAAC;IAC/B,CAAC,QAAQ,MAAM,EAAE;IAEjB,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,uBAAuB,CAC9B,QAAgB,EAChB,IAAmB,EACnB,MAAqB;IAErB,MAAM,YAAY,GAAG,GAAG,QAAQ,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;IAEjD,OAAO;QACL,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,YAAY;QACjC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,kCAAkC,QAAQ,IAAI;QAC/E,UAAU,EAAE,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC;QACjD,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM;YAC/B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;gBACnC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACnD,CAAC,CAAC;YAEH,IAAI,YAAY,IAAI,MAAM,EAAE,CAAC;gBAC3B,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC;yBACjD;qBACF;oBACD,OAAO,EAAE;wBACP,QAAQ;wBACR,QAAQ,EAAE,IAAI,CAAC,IAAI;wBACnB,UAAU,EAAE,MAAM,CAAC,UAAU;qBAC9B;iBACF,CAAC;YACJ,CAAC;YAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC;YACjE,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC;gBACzC,OAAO,EAAE;oBACP,QAAQ;oBACR,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,GAAG,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACrF;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAAC,WAAgD;IAC3E,OAAQ,CAAC,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ;QACrD,CAAC,CAAC,WAAW;QACb,CAAC,CAAC;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;YACd,oBAAoB,EAAE,IAAI;SAC3B,CAAkC,CAAC;AAC1C,CAAC;AAED,SAAS,gBAAgB,CAAC,OAA0B;IAIlD,MAAM,MAAM,GAGR,EAAE,CAAC;IAEP,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,MAAM;gBACT,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBAChD,MAAM;YACR,KAAK,OAAO;gBACV,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC3E,MAAM;YACR,KAAK,OAAO;gBACV,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,0BAA0B,KAAK,CAAC,QAAQ,GAAG;iBAClD,CAAC,CAAC;gBACH,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAC7B,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;oBACzD,MAAM;gBACR,CAAC;gBAED,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,2BAA2B,KAAK,CAAC,QAAQ,CAAC,GAAG,GAAG;iBACvD,CAAC,CAAC;gBACH,MAAM;YACR,KAAK,eAAe;gBAClB,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,KAAK,CAAC,GAAG;iBAChB,CAAC,CAAC;gBACH,MAAM;QACV,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,eAAe,CAAC,OAA0B,EAAE,gBAAwB;IAC3E,MAAM,IAAI,GAAG,OAAO;SACjB,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;SAC/D,IAAI,CAAC,IAAI,CAAC;SACV,IAAI,EAAE,CAAC;IAEV,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,gBAAgB,WAAW,CAAC;AAC3E,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,MAAqB;IAClD,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAAC,MAAM,CAAC;QACP,0EAA0E;IAC5E,CAAC;AACH,CAAC"}

package/dist/runtime/message-mapping.d.ts

@@ -0,0 +1,11 @@
+import type { AgentMessage } from "@mariozechner/pi-agent-core";
+import type { Api as AiApi, AssistantMessage, Model, ToolResultMessage, Usage, UserMessage } from "@mariozechner/pi-ai";
+import type { ConversationEvent } from "../domain/conversation.js";
+declare const EMPTY_USAGE: Usage;
+export declare function toAgentMessages(messages: ConversationEvent[], model: Model<AiApi>): Array<UserMessage | AssistantMessage | ToolResultMessage>;
+export declare function toConversationEvents(messages: AgentMessage[], options: {
+    parentMessageId: string;
+    correlationId: string;
+}): ConversationEvent[];
+export declare function getAssistantText(message: AssistantMessage): string;
+export { EMPTY_USAGE };