@compyle/unagent 1.0.0

package/dist/messages/message-utils.js

@@ -0,0 +1,343 @@
+import { MessageRole, } from "./message-types.js";
+export function getMessageRole(message) {
+    return message.type;
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null;
+}
+function isNonEmptyString(value) {
+    return typeof value === "string" && value.length > 0;
+}
+function getToolUseIdValue(record) {
+    return record.tool_use_id ?? record.toolUseId ?? record.id;
+}
+function getToolResultContentValue(record) {
+    return record.content ?? record.result;
+}
+function isToolResultLike(record) {
+    return Boolean(getToolUseIdValue(record)) && Boolean(getToolResultContentValue(record));
+}
+function getToolResultError(record) {
+    return Boolean(record.isError ?? record.is_error ?? record.isErr);
+}
+function stringifyValue(value) {
+    if (value === undefined || value === null) {
+        return "";
+    }
+    if (typeof value === "string") {
+        return value;
+    }
+    if (Array.isArray(value)) {
+        const parts = [];
+        for (const item of value) {
+            if (typeof item === "string") {
+                parts.push(item);
+                continue;
+            }
+            if (!item || typeof item !== "object") {
+                parts.push(String(item));
+                continue;
+            }
+            const record = item;
+            if (typeof record.text === "string") {
+                parts.push(record.text);
+                continue;
+            }
+            if (typeof record.content === "string") {
+                parts.push(record.content);
+                continue;
+            }
+            try {
+                parts.push(JSON.stringify(record));
+            }
+            catch {
+                parts.push(String(record));
+            }
+        }
+        if (parts.length > 0) {
+            return parts.join("\n");
+        }
+    }
+    if (typeof value === "object") {
+        const record = value;
+        if (typeof record.text === "string") {
+            return record.text;
+        }
+        if (typeof record.content === "string") {
+            return record.content;
+        }
+    }
+    try {
+        return JSON.stringify(value);
+    }
+    catch {
+        return String(value);
+    }
+}
+function getContentBlocks(message) {
+    if (Array.isArray(message.content)) {
+        return message.content;
+    }
+    if (message.content === undefined || message.content === null) {
+        return [];
+    }
+    return [message.content];
+}
+export function validateMessage(message) {
+    const errors = [];
+    const role = getMessageRole(message);
+    if (!role || !Object.values(MessageRole).includes(role)) {
+        errors.push(`Invalid message role: ${role}`);
+    }
+    const contentBlocks = getContentBlocks(message);
+    if (contentBlocks.length === 0) {
+        errors.push("Message content must be non-empty");
+    }
+    if (!Array.isArray(message.content)) {
+        const singleBlock = message.content;
+        const singleType = singleBlock?.type;
+        const isToolResultBlock = role === MessageRole.USER && singleType === "tool_result";
+        if (!isToolResultBlock) {
+            errors.push("Message content must be an array for USER/ASSISTANT messages");
+        }
+    }
+    for (let i = 0; i < contentBlocks.length; i++) {
+        const block = contentBlocks[i];
+        if (!block.type) {
+            errors.push(`Content block at index ${i} is missing type`);
+        }
+    }
+    if (role === MessageRole.USER) {
+        const blocks = getContentBlocks(message);
+        for (const block of blocks) {
+            const blockType = block.type;
+            if (blockType !== "text" && blockType !== "image" && blockType !== "tool_result") {
+                errors.push(`USER message cannot contain ${blockType} content block`);
+            }
+        }
+    }
+    else if (role === MessageRole.ASSISTANT) {
+        const blocks = getContentBlocks(message);
+        for (const block of blocks) {
+            const blockType = block.type;
+            const isToolUse = blockType === "tool_use" || blockType === "tool_call";
+            if (blockType !== "thinking" && blockType !== "text" && !isToolUse) {
+                errors.push(`ASSISTANT message cannot contain ${blockType} content block`);
+            }
+        }
+    }
+    return {
+        valid: errors.length === 0,
+        errors: errors.length > 0 ? errors : undefined,
+    };
+}
+export function createUserMessage(content) {
+    let contentBlocks;
+    if (typeof content === "string") {
+        contentBlocks = [{ type: "text", content }];
+    }
+    else {
+        contentBlocks = [{ type: "text", content: content.text }];
+        if (content.images && content.images.length > 0) {
+            for (const imageUrl of content.images) {
+                contentBlocks.push({ type: "image", url: imageUrl });
+            }
+        }
+    }
+    return {
+        type: MessageRole.USER,
+        content: contentBlocks,
+    };
+}
+export function createAssistantMessage(params) {
+    const contentBlocks = [];
+    if (params.thinking) {
+        contentBlocks.push({
+            type: "thinking",
+            content: params.thinking.content,
+            signature: params.thinking.signature,
+        });
+    }
+    if (params.text) {
+        contentBlocks.push({
+            type: "text",
+            content: params.text,
+        });
+    }
+    if (params.toolCalls && params.toolCalls.length > 0) {
+        contentBlocks.push({
+            type: "tool_use",
+            content: params.toolCalls,
+        });
+    }
+    return {
+        type: MessageRole.ASSISTANT,
+        content: contentBlocks,
+        ...(params.model && { model: params.model }),
+    };
+}
+export function createToolCallResultMessage(results) {
+    const isArrayInput = Array.isArray(results);
+    const resultList = isArrayInput ? results : [results];
+    const contentBlocks = resultList.map((result) => {
+        const isError = result.isError ?? result.is_error ?? false;
+        const toolUseId = result.tool_use_id ?? result.toolUseId ?? result.id ?? "";
+        const rawContent = result.content ?? result.result ?? "";
+        let content = "";
+        if (typeof rawContent !== "string") {
+            content = stringifyValue(rawContent);
+        }
+        else if (!rawContent && result.result !== undefined && result.result !== null) {
+            content = stringifyValue(result.result);
+        }
+        else {
+            content = rawContent;
+        }
+        const block = {
+            type: "tool_result",
+            tool_use_id: toolUseId,
+            content,
+        };
+        if (isError) {
+            block.is_error = true;
+        }
+        return block;
+    });
+    return {
+        type: MessageRole.USER,
+        content: isArrayInput ? contentBlocks : contentBlocks[0],
+    };
+}
+export function extractTextFromMessage(message) {
+    const contentBlocks = getContentBlocks(message);
+    const textBlocks = contentBlocks.filter((block) => block.type === "text");
+    return textBlocks.map((block) => block.content).join("\n");
+}
+export function extractToolCalls(message) {
+    const contentBlocks = getContentBlocks(message);
+    const toolCallBlocks = contentBlocks.filter((block) => block.type === "tool_use" || block.type === "tool_call");
+    return toolCallBlocks.flatMap((block) => block.content);
+}
+export function extractThinkingFromMessage(message) {
+    const contentBlocks = getContentBlocks(message);
+    return contentBlocks.filter((block) => block.type === "thinking");
+}
+export function extractImagesFromMessage(message) {
+    if (getMessageRole(message) !== MessageRole.USER)
+        return [];
+    const contentBlocks = getContentBlocks(message);
+    const imageBlocks = contentBlocks.filter((block) => block.type === "image");
+    return imageBlocks.map((block) => block.url);
+}
+export function extractToolResultsFromMessage(message) {
+    if (getMessageRole(message) !== MessageRole.USER)
+        return [];
+    const contentBlocks = getContentBlocks(message);
+    const toolResultBlocks = [];
+    for (const block of contentBlocks) {
+        if (block.type === "tool_result") {
+            toolResultBlocks.push({
+                type: "tool_result",
+                tool_use_id: block.tool_use_id,
+                content: block.content,
+                is_error: block.isError ?? block.is_error ?? false,
+            });
+            continue;
+        }
+        if (!isRecord(block)) {
+            continue;
+        }
+        if (!isToolResultLike(block)) {
+            continue;
+        }
+        const toolUseIdValue = getToolUseIdValue(block);
+        const toolUseId = isNonEmptyString(toolUseIdValue) ? toolUseIdValue : "";
+        const content = stringifyValue(getToolResultContentValue(block));
+        toolResultBlocks.push({
+            type: "tool_result",
+            tool_use_id: toolUseId,
+            content,
+            is_error: getToolResultError(block),
+        });
+    }
+    return toolResultBlocks;
+}
+export function normalizeMessage(message) {
+    const role = getMessageRole(message);
+    const contentBlocks = getContentBlocks(message);
+    if (!role) {
+        return {
+            ...message,
+            content: contentBlocks,
+        };
+    }
+    return {
+        ...message,
+        type: message.type ?? role,
+        content: contentBlocks,
+    };
+}
+export function messageToEvent(message, clientType, sessionId) {
+    const normalized = normalizeMessage(message);
+    const baseEvent = {
+        timestamp: Date.now(),
+        clientType,
+        sessionId,
+    };
+    const messageEvent = {
+        type: "item.end",
+        ...baseEvent,
+        message: normalized,
+    };
+    return messageEvent;
+}
+export function eventToMessage(event) {
+    if (event.type === "item.end") {
+        return normalizeMessage(event.message);
+    }
+    return null;
+}
+export function messageHasErrors(message) {
+    const results = extractToolResultsFromMessage(message);
+    return results.some((result) => result.isError ?? result.is_error);
+}
+export function getToolCallIds(message) {
+    const toolCalls = extractToolCalls(message);
+    return toolCalls.map((call) => call.id);
+}
+export function formatMessage(message) {
+    const parts = [];
+    const role = getMessageRole(message) ?? "UNKNOWN";
+    parts.push(`[${role}]`);
+    for (const block of getContentBlocks(message)) {
+        if (block.type === "tool_result") {
+            const result = block.content;
+            const isError = block.isError ?? block.is_error;
+            parts.push(`ToolResult: ${isError ? "ERROR" : "OK"} - "${result.substring(0, 50)}${result.length > 50 ? "..." : ""}"`);
+            continue;
+        }
+        if (isRecord(block) && isToolResultLike(block)) {
+            const result = stringifyValue(getToolResultContentValue(block));
+            const isError = getToolResultError(block);
+            parts.push(`ToolResult: ${isError ? "ERROR" : "OK"} - "${result.substring(0, 50)}${result.length > 50 ? "..." : ""}"`);
+            continue;
+        }
+        switch (block.type) {
+            case "text":
+                parts.push(`Text: "${block.content.substring(0, 50)}${block.content.length > 50 ? "..." : ""}"`);
+                break;
+            case "image":
+                parts.push(`Image: ${block.url.substring(0, 50)}...`);
+                break;
+            case "thinking":
+                parts.push(`Thinking: "${block.content.substring(0, 50)}${block.content.length > 50 ? "..." : ""}"`);
+                break;
+            case "tool_use":
+            case "tool_call":
+                parts.push(`ToolCalls: ${block.content.length} calls`);
+                break;
+        }
+    }
+    return parts.join(" | ");
+}
+//# sourceMappingURL=message-utils.js.map

package/dist/messages/message-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-utils.js","sourceRoot":"","sources":["../../src/messages/message-utils.ts"],"names":[],"mappings":"AAKA,OAAO,EAGL,WAAW,GAUZ,MAAM,oBAAoB,CAAC;AAM5B,MAAM,UAAU,cAAc,CAAC,OAA+B;IAC5D,OAAO,OAAO,CAAC,IAAI,CAAC;AACtB,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,iBAAiB,CAAC,MAA+B;IACxD,OAAO,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,EAAE,CAAC;AAC7D,CAAC;AAED,SAAS,yBAAyB,CAAC,MAA+B;IAChE,OAAO,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,CAAC;AACzC,CAAC;AAED,SAAS,gBAAgB,CAAC,MAA+B;IACvD,OAAO,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,IAAI,OAAO,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC;AAC1F,CAAC;AAED,SAAS,kBAAkB,CAAC,MAA+B;IACzD,OAAO,OAAO,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC1C,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC7B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACjB,SAAS;YACX,CAAC;YACD,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;gBACzB,SAAS;YACX,CAAC;YACD,MAAM,MAAM,GAAG,IAA6C,CAAC;YAC7D,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACxB,SAAS;YACX,CAAC;YACD,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACvC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAC3B,SAAS;YACX,CAAC;YACD,IAAI,CAAC;gBACH,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;YACrC,CAAC;YAAC,MAAM,CAAC;gBACP,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,KAA8C,CAAC;QAC9D,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpC,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,CAAC;QACD,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACvC,OAAO,MAAM,CAAC,OAAO,CAAC;QACxB,CAAC;IACH,CAAC;IACD,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAgB;IACxC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACnC,OAAO,OAAO,CAAC,OAAO,CAAC;IACzB,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;QAC9D,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AAC3B,CAAC;AAKD,MAAM,UAAU,eAAe,CAAC,OAAgB;IAC9C,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IAGrC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAGhD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;IACnD,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACpC,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;QACpC,MAAM,UAAU,GAAG,WAAW,EAAE,IAAI,CAAC;QACrC,MAAM,iBAAiB,GAAG,IAAI,KAAK,WAAW,CAAC,IAAI,IAAI,UAAU,KAAK,aAAa,CAAC;QACpF,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,8DAA8D,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IAGD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,kBAAkB,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAGD,IAAI,IAAI,KAAK,WAAW,CAAC,IAAI,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACzC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;YAC7B,IAAI,SAAS,KAAK,MAAM,IAAI,SAAS,KAAK,OAAO,IAAI,SAAS,KAAK,aAAa,EAAE,CAAC;gBACjF,MAAM,CAAC,IAAI,CAAC,+BAA+B,SAAS,gBAAgB,CAAC,CAAC;YACxE,CAAC;QACH,CAAC;IACH,CAAC;SAAM,IAAI,IAAI,KAAK,WAAW,CAAC,SAAS,EAAE,CAAC;QAC1C,MAAM,MAAM,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACzC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;YAC7B,MAAM,SAAS,GAAG,SAAS,KAAK,UAAU,IAAI,SAAS,KAAK,WAAW,CAAC;YACxE,IAAI,SAAS,KAAK,UAAU,IAAI,SAAS,KAAK,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnE,MAAM,CAAC,IAAI,CAAC,oCAAoC,SAAS,gBAAgB,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,KAAK,EAAE,MAAM,CAAC,MAAM,KAAK,CAAC;QAC1B,MAAM,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;KAC/C,CAAC;AACJ,CAAC;AAKD,MAAM,UAAU,iBAAiB,CAC/B,OAAqD;IAErD,IAAI,aAA6C,CAAC;IAElD,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,aAAa,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;IAC9C,CAAC;SAAM,CAAC;QACN,aAAa,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACtC,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,IAAI,EAAE,WAAW,CAAC,IAAI;QACtB,OAAO,EAAE,aAAa;KACvB,CAAC;AACJ,CAAC;AAKD,MAAM,UAAU,sBAAsB,CAAC,MAKtC;IACC,MAAM,aAAa,GAAuD,EAAE,CAAC;IAG7E,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,aAAa,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO;YAChC,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;SACrC,CAAC,CAAC;IACL,CAAC;IAGD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QAChB,aAAa,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,MAAM,CAAC,IAAI;SACrB,CAAC,CAAC;IACL,CAAC;IAGD,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpD,aAAa,CAAC,IAAI,CAAC;YACjB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,MAAM,CAAC,SAAS;SAC1B,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,IAAI,EAAE,WAAW,CAAC,SAAS;QAC3B,OAAO,EAAE,aAAa;QACtB,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;KAC7C,CAAC;AACJ,CAAC;AAKD,MAAM,UAAU,2BAA2B,CACzC,OAkBK;IAEL,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5C,MAAM,UAAU,GAAG,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IACtD,MAAM,aAAa,GAAwB,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QACnE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC;QAC3D,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC;QAC5E,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;QACzD,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;YACnC,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC;QACvC,CAAC;aAAM,IAAI,CAAC,UAAU,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;YAChF,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,UAAU,CAAC;QACvB,CAAC;QACD,MAAM,KAAK,GAAsB;YAC/B,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,SAAS;YACtB,OAAO;SACR,CAAC;QACF,IAAI,OAAO,EAAE,CAAC;YACZ,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC;QACxB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,IAAI,EAAE,WAAW,CAAC,IAAI;QACtB,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;KACzD,CAAC;AACJ,CAAC;AAKD,MAAM,UAAU,sBAAsB,CAAC,OAAgB;IACrD,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CACrC,CAAC,KAAK,EAAwB,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CACvD,CAAC;IACF,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC7D,CAAC;AAKD,MAAM,UAAU,gBAAgB,CAAC,OAAgB;IAC/C,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,cAAc,GAAG,aAAa,CAAC,MAAM,CACzC,CAAC,KAAK,EAA2B,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,CAC5F,CAAC;IAEF,OAAO,cAAc,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AAC1D,CAAC;AAKD,MAAM,UAAU,0BAA0B,CAAC,OAAgB;IACzD,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAChD,OAAO,aAAa,CAAC,MAAM,CACzB,CAAC,KAAK,EAA4B,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAC/D,CAAC;AACJ,CAAC;AAKD,MAAM,UAAU,wBAAwB,CAAC,OAAgB;IACvD,IAAI,cAAc,CAAC,OAAO,CAAC,KAAK,WAAW,CAAC,IAAI;QAAE,OAAO,EAAE,CAAC;IAE5D,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,WAAW,GAAG,aAAa,CAAC,MAAM,CACtC,CAAC,KAAK,EAAyB,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,OAAO,CACzD,CAAC;IACF,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;AAC/C,CAAC;AAKD,MAAM,UAAU,6BAA6B,CAAC,OAAgB;IAC5D,IAAI,cAAc,CAAC,OAAO,CAAC,KAAK,WAAW,CAAC,IAAI;QAAE,OAAO,EAAE,CAAC;IAE5D,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAChD,MAAM,gBAAgB,GAAwB,EAAE,CAAC;IAEjD,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;QAClC,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACjC,gBAAgB,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ,EAAE,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,IAAI,KAAK;aACnD,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACrB,SAAS;QACX,CAAC;QACD,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7B,SAAS;QACX,CAAC;QACD,MAAM,cAAc,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,gBAAgB,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;QACzE,MAAM,OAAO,GAAG,cAAc,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC;QACjE,gBAAgB,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,SAAS;YACtB,OAAO;YACP,QAAQ,EAAE,kBAAkB,CAAC,KAAK,CAAC;SACpC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAKD,MAAM,UAAU,gBAAgB,CAAC,OAAgB;IAC/C,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;IACrC,MAAM,aAAa,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEhD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,GAAG,OAAO;YACV,OAAO,EAAE,aAAa;SACZ,CAAC;IACf,CAAC;IAED,OAAO;QACL,GAAG,OAAO;QACV,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI;QAC1B,OAAO,EAAE,aAAa;KACZ,CAAC;AACf,CAAC;AAMD,MAAM,UAAU,cAAc,CAC5B,OAAgB,EAChB,UAAsB,EACtB,SAAkB;IAElB,MAAM,UAAU,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC7C,MAAM,SAAS,GAAG;QAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,UAAU;QACV,SAAS;KACV,CAAC;IAEF,MAAM,YAAY,GAAiB;QACjC,IAAI,EAAE,UAAU;QAChB,GAAG,SAAS;QACZ,OAAO,EAAE,UAAU;KACpB,CAAC;IAEF,OAAO,YAAY,CAAC;AACtB,CAAC;AAOD,MAAM,UAAU,cAAc,CAAC,KAAkB;IAC/C,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAC9B,OAAO,gBAAgB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAKD,MAAM,UAAU,gBAAgB,CAAC,OAAgB;IAC/C,MAAM,OAAO,GAAG,6BAA6B,CAAC,OAAO,CAAC,CAAC;IACvD,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC;AACrE,CAAC;AAKD,MAAM,UAAU,cAAc,CAAC,OAAgB;IAC7C,MAAM,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC5C,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAC1C,CAAC;AAKD,MAAM,UAAU,aAAa,CAAC,OAAgB;IAC5C,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,MAAM,IAAI,GAAG,cAAc,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC;IAClD,KAAK,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;IAExB,KAAK,MAAM,KAAK,IAAI,gBAAgB,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9C,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC;YAC7B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,CAAC;YAChD,KAAK,CAAC,IAAI,CACR,eAAe,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GACnE,MAAM,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAC/B,GAAG,CACJ,CAAC;YACF,SAAS;QACX,CAAC;QAED,IAAI,QAAQ,CAAC,KAAK,CAAC,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/C,MAAM,MAAM,GAAG,cAAc,CAAC,yBAAyB,CAAC,KAAK,CAAC,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC1C,KAAK,CAAC,IAAI,CACR,eAAe,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,OAAO,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GACnE,MAAM,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAC/B,GAAG,CACJ,CAAC;YACF,SAAS;QACX,CAAC;QAED,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,MAAM;gBACT,KAAK,CAAC,IAAI,CACR,UAAU,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,CACrF,CAAC;gBACF,MAAM;YACR,KAAK,OAAO;gBACV,KAAK,CAAC,IAAI,CAAC,UAAU,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC;gBACtD,MAAM;YACR,KAAK,UAAU;gBACb,KAAK,CAAC,IAAI,CACR,cAAc,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,CACzF,CAAC;gBACF,MAAM;YACR,KAAK,UAAU,CAAC;YAChB,KAAK,WAAW;gBACd,KAAK,CAAC,IAAI,CAAC,cAAc,KAAK,CAAC,OAAO,CAAC,MAAM,QAAQ,CAAC,CAAC;gBACvD,MAAM;QACV,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC3B,CAAC"}

package/dist/tools/define-tool.d.ts

@@ -0,0 +1,6 @@
+import { z } from "zod";
+import type { ToolDefinition, RegisteredTool } from "./tool-types.js";
+export declare function defineTool<Schema extends z.ZodRawShape>(definition: ToolDefinition<Schema>): RegisteredTool;
+export declare function isRegisteredTool(value: unknown): value is RegisteredTool;
+export declare function extractCustomTools(tools: unknown): RegisteredTool[];
+//# sourceMappingURL=define-tool.d.ts.map

package/dist/tools/define-tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"define-tool.d.ts","sourceRoot":"","sources":["../../src/tools/define-tool.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAyBtE,wBAAgB,UAAU,CAAC,MAAM,SAAS,CAAC,CAAC,WAAW,EACrD,UAAU,EAAE,cAAc,CAAC,MAAM,CAAC,GACjC,cAAc,CA8EhB;AAKD,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,cAAc,CAWxE;AAKD,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,cAAc,EAAE,CAkBnE"}

package/dist/tools/define-tool.js

@@ -0,0 +1,83 @@
+import { z } from "zod";
+import { toolRegistry } from "./tool-registry.js";
+import { convertToCallToolResult } from "./tool-result-converter.js";
+export function defineTool(definition) {
+    const { name, description, handler } = definition;
+    const inputSchema = definition.inputSchema ?? definition.input_schema;
+    if (!name || typeof name !== 'string') {
+        throw new Error('Tool name must be a non-empty string');
+    }
+    if (!description || typeof description !== 'string') {
+        throw new Error('Tool description must be a non-empty string');
+    }
+    if (!inputSchema || typeof inputSchema !== "object") {
+        throw new Error("Tool inputSchema/input_schema must be a valid Zod schema shape");
+    }
+    if (typeof handler !== 'function') {
+        throw new Error('Tool handler must be a function');
+    }
+    const id = `${name}_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;
+    const schema = z.object(inputSchema);
+    const wrappedHandler = async (args) => {
+        try {
+            const validatedArgs = schema.parse(args);
+            const result = await handler(validatedArgs);
+            return convertToCallToolResult(result);
+        }
+        catch (error) {
+            if (error instanceof Error && error.name === 'ZodError') {
+                return {
+                    content: [{
+                            type: 'text',
+                            text: `Input validation error: ${error.message}`
+                        }],
+                    isError: true
+                };
+            }
+            return {
+                content: [{
+                        type: 'text',
+                        text: `Error: ${error instanceof Error ? error.message : String(error)}`
+                    }],
+                isError: true
+            };
+        }
+    };
+    const registeredTool = {
+        id,
+        name,
+        description,
+        inputSchema,
+        handler: wrappedHandler,
+        createdAt: new Date(),
+    };
+    toolRegistry.register(registeredTool);
+    return registeredTool;
+}
+export function isRegisteredTool(value) {
+    return (typeof value === 'object' &&
+        value !== null &&
+        'id' in value &&
+        'name' in value &&
+        'description' in value &&
+        'inputSchema' in value &&
+        'handler' in value &&
+        'createdAt' in value);
+}
+export function extractCustomTools(tools) {
+    if (tools === undefined || tools === null) {
+        return [];
+    }
+    if (Array.isArray(tools)) {
+        const invalidTools = tools.filter((tool) => !isRegisteredTool(tool));
+        if (invalidTools.length > 0) {
+            throw new Error("Tools must be RegisteredTool instances created by defineTool");
+        }
+        return tools;
+    }
+    if (isRegisteredTool(tools)) {
+        return [tools];
+    }
+    throw new Error("Tools must be RegisteredTool instances created by defineTool");
+}
+//# sourceMappingURL=define-tool.js.map

package/dist/tools/define-tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"define-tool.js","sourceRoot":"","sources":["../../src/tools/define-tool.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAuBrE,MAAM,UAAU,UAAU,CACxB,UAAkC;IAElC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC;IAClD,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,YAAY,CAAC;IAGtE,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAC1D,CAAC;IAGD,IAAI,CAAC,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;QACpD,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACjE,CAAC;IAGD,IAAI,CAAC,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;QACpD,MAAM,IAAI,KAAK,CAAC,gEAAgE,CAAC,CAAC;IACpF,CAAC;IAGD,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;IACrD,CAAC;IAGD,MAAM,EAAE,GAAG,GAAG,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAG7E,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAGrC,MAAM,cAAc,GAA8B,KAAK,EAAE,IAAI,EAAE,EAAE;QAC/D,IAAI,CAAC;YAEH,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAGzC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,CAAC;YAG5C,OAAO,uBAAuB,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAExD,OAAO;oBACL,OAAO,EAAE,CAAC;4BACR,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,2BAA2B,KAAK,CAAC,OAAO,EAAE;yBACjD,CAAC;oBACF,OAAO,EAAE,IAAI;iBACd,CAAC;YACJ,CAAC;YAGD,OAAO;gBACL,OAAO,EAAE,CAAC;wBACR,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,UAAU,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;qBACzE,CAAC;gBACF,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC,CAAC;IAGF,MAAM,cAAc,GAAmB;QACrC,EAAE;QACF,IAAI;QACJ,WAAW;QACX,WAAW;QACX,OAAO,EAAE,cAAc;QACvB,SAAS,EAAE,IAAI,IAAI,EAAE;KACtB,CAAC;IAGF,YAAY,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;IAEtC,OAAO,cAAc,CAAC;AACxB,CAAC;AAKD,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,IAAI,IAAI,KAAK;QACb,MAAM,IAAI,KAAK;QACf,aAAa,IAAI,KAAK;QACtB,aAAa,IAAI,KAAK;QACtB,SAAS,IAAI,KAAK;QAClB,WAAW,IAAI,KAAK,CACrB,CAAC;AACJ,CAAC;AAKD,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC1C,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC;QACrE,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;QAClF,CAAC;QACD,OAAO,KAAyB,CAAC;IACnC,CAAC;IAED,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,CAAC,KAAK,CAAC,CAAC;IACjB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;AAClF,CAAC"}

package/dist/tools/index.d.ts

@@ -0,0 +1,6 @@
+export * from './tool-types.js';
+export * from './define-tool.js';
+export { toolRegistry } from './tool-registry.js';
+export { convertToCallToolResult } from './tool-result-converter.js';
+export { ToolStdioMcpService } from "./tool-stdio-mcp-service.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/tools/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAKA,cAAc,iBAAiB,CAAC;AAGhC,cAAc,kBAAkB,CAAC;AAGjC,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGlD,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAGrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC"}

package/dist/tools/index.js

@@ -0,0 +1,6 @@
+export * from './tool-types.js';
+export * from './define-tool.js';
+export { toolRegistry } from './tool-registry.js';
+export { convertToCallToolResult } from './tool-result-converter.js';
+export { ToolStdioMcpService } from "./tool-stdio-mcp-service.js";
+//# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAKA,cAAc,iBAAiB,CAAC;AAGhC,cAAc,kBAAkB,CAAC;AAGjC,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGlD,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAGrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,6BAA6B,CAAC"}

package/dist/tools/mcp-stdio-proxy.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=mcp-stdio-proxy.d.ts.map

package/dist/tools/mcp-stdio-proxy.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-stdio-proxy.d.ts","sourceRoot":"","sources":["../../src/tools/mcp-stdio-proxy.ts"],"names":[],"mappings":""}

package/dist/tools/mcp-stdio-proxy.js

@@ -0,0 +1,31 @@
+import net from "node:net";
+const socketPath = process.env.UNAGENT_MCP_SOCKET;
+if (!socketPath) {
+    console.error("[mcp-stdio-proxy] Missing UNAGENT_MCP_SOCKET");
+    process.exit(1);
+}
+const socket = net.createConnection(socketPath);
+socket.on("connect", () => {
+    process.stdin.pipe(socket);
+    socket.pipe(process.stdout);
+    process.stdin.resume();
+});
+socket.on("close", () => {
+    process.exit(0);
+});
+socket.on("error", (error) => {
+    console.error("[mcp-stdio-proxy] Socket error:", error);
+    process.exit(1);
+});
+process.stdin.on("end", () => {
+    socket.end();
+});
+process.on("SIGINT", () => {
+    socket.end();
+    process.exit(0);
+});
+process.on("SIGTERM", () => {
+    socket.end();
+    process.exit(0);
+});
+//# sourceMappingURL=mcp-stdio-proxy.js.map

package/dist/tools/mcp-stdio-proxy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-stdio-proxy.js","sourceRoot":"","sources":["../../src/tools/mcp-stdio-proxy.ts"],"names":[],"mappings":"AAKA,OAAO,GAAG,MAAM,UAAU,CAAC;AAE3B,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;AAElD,IAAI,CAAC,UAAU,EAAE,CAAC;IAChB,OAAO,CAAC,KAAK,CAAC,8CAA8C,CAAC,CAAC;IAC9D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC;AAED,MAAM,MAAM,GAAG,GAAG,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;AAEhD,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;IACxB,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC3B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC5B,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;AACzB,CAAC,CAAC,CAAC;AAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;IACtB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;IAC3B,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;IACxD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;IAC3B,MAAM,CAAC,GAAG,EAAE,CAAC;AACf,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE;IACxB,MAAM,CAAC,GAAG,EAAE,CAAC;IACb,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC;AAEH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;IACzB,MAAM,CAAC,GAAG,EAAE,CAAC;IACb,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/tools/tool-call-context.d.ts

@@ -0,0 +1,8 @@
+type ToolCallContext = {
+    parentToolUseId?: string;
+};
+export declare function runWithToolCallContext<T>(context: ToolCallContext | undefined, fn: () => T): T;
+export declare function getParentToolUseId(): string | undefined;
+export declare function resolveParentToolUseId(extra: unknown): string | undefined;
+export {};
+//# sourceMappingURL=tool-call-context.d.ts.map

package/dist/tools/tool-call-context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-context.d.ts","sourceRoot":"","sources":["../../src/tools/tool-call-context.ts"],"names":[],"mappings":"AAOA,KAAK,eAAe,GAAG;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B,CAAC;AAIF,wBAAgB,sBAAsB,CAAC,CAAC,EACtC,OAAO,EAAE,eAAe,GAAG,SAAS,EACpC,EAAE,EAAE,MAAM,CAAC,GACV,CAAC,CAKH;AAED,wBAAgB,kBAAkB,IAAI,MAAM,GAAG,SAAS,CAEvD;AAoBD,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAiBzE"}

package/dist/tools/tool-call-context.js

@@ -0,0 +1,44 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+const toolCallContext = new AsyncLocalStorage();
+export function runWithToolCallContext(context, fn) {
+    if (!context || !context.parentToolUseId) {
+        return fn();
+    }
+    return toolCallContext.run(context, fn);
+}
+export function getParentToolUseId() {
+    return toolCallContext.getStore()?.parentToolUseId;
+}
+function normalizeParentToolUseId(value) {
+    if (typeof value === "string") {
+        const trimmed = value.trim();
+        return trimmed ? trimmed : undefined;
+    }
+    if (typeof value === "number" && Number.isFinite(value)) {
+        return String(value);
+    }
+    return undefined;
+}
+function getMetaValue(meta) {
+    if (!meta) {
+        return undefined;
+    }
+    return meta.parent_tool_use_id ?? meta.parentToolUseId ?? meta.tool_use_id ?? meta.toolUseId;
+}
+export function resolveParentToolUseId(extra) {
+    if (!extra || typeof extra !== "object") {
+        return undefined;
+    }
+    const record = extra;
+    const meta = record._meta;
+    const metaRecord = meta && typeof meta === "object" ? meta : undefined;
+    const candidate = getMetaValue(metaRecord)
+        ?? record.parent_tool_use_id
+        ?? record.parentToolUseId
+        ?? record.tool_use_id
+        ?? record.toolUseId
+        ?? record.requestId
+        ?? record.request_id;
+    return normalizeParentToolUseId(candidate);
+}
+//# sourceMappingURL=tool-call-context.js.map

package/dist/tools/tool-call-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-context.js","sourceRoot":"","sources":["../../src/tools/tool-call-context.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAMrD,MAAM,eAAe,GAAG,IAAI,iBAAiB,EAAmB,CAAC;AAEjE,MAAM,UAAU,sBAAsB,CACpC,OAAoC,EACpC,EAAW;IAEX,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC;QACzC,OAAO,EAAE,EAAE,CAAC;IACd,CAAC;IACD,OAAO,eAAe,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,OAAO,eAAe,CAAC,QAAQ,EAAE,EAAE,eAAe,CAAC;AACrD,CAAC;AAED,SAAS,wBAAwB,CAAC,KAAc;IAC9C,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAC7B,OAAO,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;IACvC,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACxD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,YAAY,CAAC,IAA8B;IAClD,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,IAAI,CAAC,kBAAkB,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,SAAS,CAAC;AAC/F,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,KAAc;IACnD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,MAAM,GAAG,KAAgC,CAAC;IAChD,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC;IAC1B,MAAM,UAAU,GAAG,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAE,IAAgC,CAAC,CAAC,CAAC,SAAS,CAAC;IACpG,MAAM,SAAS,GAAG,YAAY,CAAC,UAAU,CAAC;WACrC,MAAM,CAAC,kBAAkB;WACzB,MAAM,CAAC,eAAe;WACtB,MAAM,CAAC,WAAW;WAClB,MAAM,CAAC,SAAS;WAChB,MAAM,CAAC,SAAS;WAChB,MAAM,CAAC,UAAU,CAAC;IAEvB,OAAO,wBAAwB,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC"}

package/dist/tools/tool-registry.d.ts

@@ -0,0 +1,14 @@
+import type { RegisteredTool } from './tool-types.js';
+declare class ToolRegistry {
+    private toolsById;
+    private toolsByName;
+    register(tool: RegisteredTool): void;
+    registerMultiple(tools: RegisteredTool[]): void;
+    getAllTools(): RegisteredTool[];
+    getToolByName(name: string): RegisteredTool | undefined;
+    getToolById(id: string): RegisteredTool | undefined;
+    cleanup(): void;
+}
+export declare const toolRegistry: ToolRegistry;
+export {};
+//# sourceMappingURL=tool-registry.d.ts.map

package/dist/tools/tool-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-registry.d.ts","sourceRoot":"","sources":["../../src/tools/tool-registry.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAKtD,cAAM,YAAY;IAChB,OAAO,CAAC,SAAS,CAAqC;IACtD,OAAO,CAAC,WAAW,CAAqC;IAKxD,QAAQ,CAAC,IAAI,EAAE,cAAc,GAAG,IAAI;IAiBpC,gBAAgB,CAAC,KAAK,EAAE,cAAc,EAAE,GAAG,IAAI;IAS/C,WAAW,IAAI,cAAc,EAAE;IAO/B,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAOvD,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAOnD,OAAO,IAAI,IAAI;CAOhB;AAKD,eAAO,MAAM,YAAY,cAAqB,CAAC"}