@arvo-tools/agentic 0.3.0 → 0.4.1

package/dist/Agent/agentTool.d.ts

@@ -0,0 +1,4 @@
+import type z from 'zod';
+import type { AgentInternalTool } from './types.js';
+export declare const createAgentTool: <TInputSchema extends z.ZodTypeAny, TOutputSchema extends z.ZodTypeAny>(param: AgentInternalTool<TInputSchema, TOutputSchema>) => AgentInternalTool<TInputSchema, TOutputSchema>;
+//# sourceMappingURL=agentTool.d.ts.map

package/dist/Agent/agentTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agentTool.d.ts","sourceRoot":"","sources":["../../src/Agent/agentTool.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AAEzB,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAEpD,eAAO,MAAM,eAAe,GAC1B,YAAY,SAAS,CAAC,CAAC,UAAU,EACjC,aAAa,SAAS,CAAC,CAAC,UAAU,EAElC,OAAO,iBAAiB,CAAC,YAAY,EAAE,aAAa,CAAC,KAiD/C,iBAAiB,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC"}

package/dist/Agent/agentTool.js

@@ -0,0 +1,111 @@
+"use strict";
+var __assign = (this && this.__assign) || function () {
+    __assign = Object.assign || function(t) {
+        for (var s, i = 1, n = arguments.length; i < n; i++) {
+            s = arguments[i];
+            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
+                t[p] = s[p];
+        }
+        return t;
+    };
+    return __assign.apply(this, arguments);
+};
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createAgentTool = void 0;
+var openinference_semantic_conventions_1 = require("@arizeai/openinference-semantic-conventions");
+var api_1 = require("@opentelemetry/api");
+var arvo_core_1 = require("arvo-core");
+var createAgentTool = function (param) {
+    return (__assign(__assign({}, param), { fn: function (input, config) { return __awaiter(void 0, void 0, void 0, function () {
+            var _a;
+            return __generator(this, function (_b) {
+                switch (_b.label) {
+                    case 0: return [4 /*yield*/, arvo_core_1.ArvoOpenTelemetry.getInstance().startActiveSpan({
+                            name: "AgentTool<".concat(param.name, ">.execute"),
+                            disableSpanManagement: true,
+                            context: {
+                                inheritFrom: 'TRACE_HEADERS',
+                                traceHeaders: config.otelInfo.headers,
+                            },
+                            spanOptions: {
+                                attributes: (_a = {},
+                                    _a[openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND] = openinference_semantic_conventions_1.OpenInferenceSpanKind.TOOL,
+                                    _a),
+                            },
+                            fn: function (span) { return __awaiter(void 0, void 0, void 0, function () {
+                                var inputValidation, result, err_1;
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0:
+                                            _a.trys.push([0, 2, 3, 4]);
+                                            span.setAttribute(openinference_semantic_conventions_1.SemanticConventions.TOOL_CALL_FUNCTION_NAME, param.name);
+                                            span.setAttribute(openinference_semantic_conventions_1.SemanticConventions.TOOL_CALL_FUNCTION_ARGUMENTS_JSON, JSON.stringify(input));
+                                            span.setAttribute(openinference_semantic_conventions_1.SemanticConventions.INPUT_VALUE, JSON.stringify(input));
+                                            inputValidation = param.input.safeParse(input);
+                                            if (inputValidation.error)
+                                                throw new Error("Invalid tool input data. Please send the correct data and its structure as per the input schema. ".concat(inputValidation.error.toString()));
+                                            return [4 /*yield*/, param.fn(inputValidation.data, {
+                                                    otelInfo: {
+                                                        span: span,
+                                                        headers: (0, arvo_core_1.getOtelHeaderFromSpan)(span),
+                                                    },
+                                                })];
+                                        case 1:
+                                            result = _a.sent();
+                                            span.setAttribute(openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE, JSON.stringify(result));
+                                            return [2 /*return*/, result];
+                                        case 2:
+                                            err_1 = _a.sent();
+                                            span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: err_1 === null || err_1 === void 0 ? void 0 : err_1.message });
+                                            throw err_1;
+                                        case 3:
+                                            span.end();
+                                            return [7 /*endfinally*/];
+                                        case 4: return [2 /*return*/];
+                                    }
+                                });
+                            }); },
+                        })];
+                    case 1: return [2 /*return*/, _b.sent()];
+                }
+            });
+        }); } }));
+};
+exports.createAgentTool = createAgentTool;
+//# sourceMappingURL=agentTool.js.map

package/dist/Agent/agentTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agentTool.js","sourceRoot":"","sources":["../../src/Agent/agentTool.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,kGAGqD;AACrD,0CAAoD;AACpD,uCAAqE;AAK9D,IAAM,eAAe,GAAG,UAI7B,KAAqD;IAErD,OAAA,uBACK,KAAK,KACR,EAAE,EAAE,UAAO,KAA4B,EAAE,MAAkC;;;;4BACzE,qBAAM,6BAAiB,CAAC,WAAW,EAAE,CAAC,eAAe,CAAC;4BACpD,IAAI,EAAE,oBAAa,KAAK,CAAC,IAAI,cAAW;4BACxC,qBAAqB,EAAE,IAAI;4BAC3B,OAAO,EAAE;gCACP,WAAW,EAAE,eAAe;gCAC5B,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO;6BACtC;4BACD,WAAW,EAAE;gCACX,UAAU;oCACR,GAAC,wDAAgC,CAAC,uBAAuB,IAAG,0DAAqB,CAAC,IAAI;uCACvF;6BACF;4BACD,EAAE,EAAE,UAAO,IAAI;;;;;;4CAEX,IAAI,CAAC,YAAY,CAAC,wDAAgC,CAAC,uBAAuB,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;4CACxF,IAAI,CAAC,YAAY,CACf,wDAAgC,CAAC,iCAAiC,EAClE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CACtB,CAAC;4CACF,IAAI,CAAC,YAAY,CAAC,wDAAgC,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;4CACjF,eAAe,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;4CACrD,IAAI,eAAe,CAAC,KAAK;gDACvB,MAAM,IAAI,KAAK,CACb,2GAAoG,eAAe,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAE,CACvI,CAAC;4CACW,qBAAM,KAAK,CAAC,EAAE,CAAC,eAAe,CAAC,IAAI,EAAE;oDAClD,QAAQ,EAAE;wDACR,IAAI,MAAA;wDACJ,OAAO,EAAE,IAAA,iCAAqB,EAAC,IAAI,CAAC;qDACrC;iDACF,CAAC,EAAA;;4CALI,MAAM,GAAG,SAKb;4CACF,IAAI,CAAC,YAAY,CACf,wDAAgC,CAAC,YAAY,EAC7C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CACvB,CAAC;4CACF,sBAAO,MAAM,EAAC;;;4CAEd,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,KAAK,EAAE,OAAO,EAAG,KAAa,aAAb,KAAG,uBAAH,KAAG,CAAY,OAAO,EAAE,CAAC,CAAC;4CACjF,MAAM,KAAG,CAAC;;4CAEV,IAAI,CAAC,GAAG,EAAE,CAAC;;;;;iCAEd;yBACF,CAAC,EAAA;4BA3CF,sBAAA,SA2CE,EAAA;;;aAAA,IAC8C;AA/CpD,CA+CoD,CAAC;AArD1C,QAAA,eAAe,mBAqD2B"}

package/dist/Integrations/MCPClient.d.ts

@@ -0,0 +1,80 @@
+import type { IMCPClient } from '../interfaces.mcp.js';
+import type { OtelInfoType } from '../types.js';
+/**
+ * Model Context Protocol (MCP) client implementation for connecting to and interacting with MCP servers.
+ *
+ * This client provides a robust interface for establishing connections to MCP servers,
+ * discovering available tools, invoking those tools with arguments, and managing the
+ * connection lifecycle. It supports both Server-Sent Events (SSE) and streamable HTTP
+ * transport protocols, automatically selecting the appropriate transport based on the URL.
+ */
+export declare class MCPClient implements IMCPClient {
+    private client;
+    private isConnected;
+    private availableTools;
+    private readonly url;
+    private readonly requestInit;
+    constructor(param: {
+        url: string;
+        requestInit?: RequestInit;
+    } | (() => {
+        url: string;
+        requestInit?: RequestInit;
+    }));
+    /**
+     * Establishes a connection to the MCP server and discovers available tools.
+     *
+     * This method performs the following operations:
+     * 1. Selects the appropriate transport (SSE or HTTP streaming) based on URL
+     * 2. Creates and configures the MCP client with capabilities
+     * 3. Establishes the connection to the server
+     * 4. Retrieves and caches the list of available tools
+     * 5. Logs all operations to the provided OpenTelemetry span
+     *
+     * @returns Promise that resolves when connection is established and tools are discovered
+     * @throws {Error} Throws an error if connection fails, with details logged to the span
+     */
+    connect(config: {
+        otelInfo: OtelInfoType;
+    }): Promise<void>;
+    /**
+     * Retrieves the list of available tool definitions from the connected MCP server.
+     *
+     * Transforms the cached MCP tools into the AgentToolDefinition format required
+     * by the agent system. Tools listed in restrictedTools will have requires_approval set to true.
+     * This method returns an empty array if not connected.
+     */
+    getTools(config: {
+        otelInfo: OtelInfoType;
+    }): Promise<{
+        name: string;
+        description: string;
+        inputSchema: Record<string, any>;
+    }[]>;
+    /**
+     * Invokes a specific tool on the MCP server with the provided arguments.
+     *
+     * This method sends a tool invocation request to the connected MCP server and
+     * returns the result. All operations are logged to the OpenTelemetry span for
+     * observability. If an error occurs during invocation, it returns an error message
+     * rather than throwing, allowing the agent to handle tool failures gracefully.
+     */
+    invokeTool(param: {
+        name: string;
+        arguments?: Record<string, unknown> | null;
+    }, config: {
+        otelInfo: OtelInfoType;
+    }): Promise<string>;
+    /**
+     * Gracefully disconnects from the MCP server and cleans up resources.
+     *
+     * This method safely closes the connection to the MCP server if one exists,
+     * resets the connection state, and clears cached data. It's safe to call
+     * multiple times - subsequent calls will be no-ops if already disconnected.
+     */
+    disconnect(config: {
+        otelInfo: OtelInfoType;
+    }): Promise<void>;
+    getToolPriority(): Promise<{}>;
+}
+//# sourceMappingURL=MCPClient.d.ts.map

package/dist/Integrations/MCPClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MCPClient.d.ts","sourceRoot":"","sources":["../../src/Integrations/MCPClient.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAEhD;;;;;;;GAOG;AACH,qBAAa,SAAU,YAAW,UAAU;IAC1C,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,WAAW,CAAU;IAC7B,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAe;IACnC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAoB;gBAG9C,KAAK,EACD;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,WAAW,CAAA;KAAE,GAC1C,CAAC,MAAM;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,WAAW,CAAA;KAAE,CAAC;IASxD;;;;;;;;;;;;OAYG;IACG,OAAO,CAAC,MAAM,EAAE;QAAE,QAAQ,EAAE,YAAY,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA4ChE;;;;;;OAMG;IACG,QAAQ,CAAC,MAAM,EAAE;QAAE,QAAQ,EAAE,YAAY,CAAA;KAAE,GAAG,OAAO,CACzD;QACE,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QAEpB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAClC,EAAE,CACJ;IAsCD;;;;;;;OAOG;IACG,UAAU,CACd,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAA;KAAE,EACnE,MAAM,EAAE;QAAE,QAAQ,EAAE,YAAY,CAAA;KAAE,GACjC,OAAO,CAAC,MAAM,CAAC;IA6DlB;;;;;;OAMG;IACG,UAAU,CAAC,MAAM,EAAE;QAAE,QAAQ,EAAE,YAAY,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB7D,eAAe;CAGtB"}

package/dist/Integrations/MCPClient.js

@@ -0,0 +1,292 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __values = (this && this.__values) || function(o) {
+    var s = typeof Symbol === "function" && Symbol.iterator, m = s && o[s], i = 0;
+    if (m) return m.call(o);
+    if (o && typeof o.length === "number") return {
+        next: function () {
+            if (o && i >= o.length) o = void 0;
+            return { value: o && o[i++], done: !o };
+        }
+    };
+    throw new TypeError(s ? "Object is not iterable." : "Symbol.iterator is not defined.");
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MCPClient = void 0;
+var index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
+var sse_js_1 = require("@modelcontextprotocol/sdk/client/sse.js");
+var streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
+var api_1 = require("@opentelemetry/api");
+var arvo_core_1 = require("arvo-core");
+/**
+ * Model Context Protocol (MCP) client implementation for connecting to and interacting with MCP servers.
+ *
+ * This client provides a robust interface for establishing connections to MCP servers,
+ * discovering available tools, invoking those tools with arguments, and managing the
+ * connection lifecycle. It supports both Server-Sent Events (SSE) and streamable HTTP
+ * transport protocols, automatically selecting the appropriate transport based on the URL.
+ */
+var MCPClient = /** @class */ (function () {
+    function MCPClient(param) {
+        this.client = null;
+        this.isConnected = false;
+        this.availableTools = [];
+        this.url = function () { return (typeof param === 'function' ? param() : param).url; };
+        this.requestInit = function () { var _a; return (_a = (typeof param === 'function' ? param() : param).requestInit) !== null && _a !== void 0 ? _a : {}; };
+    }
+    /**
+     * Establishes a connection to the MCP server and discovers available tools.
+     *
+     * This method performs the following operations:
+     * 1. Selects the appropriate transport (SSE or HTTP streaming) based on URL
+     * 2. Creates and configures the MCP client with capabilities
+     * 3. Establishes the connection to the server
+     * 4. Retrieves and caches the list of available tools
+     * 5. Logs all operations to the provided OpenTelemetry span
+     *
+     * @returns Promise that resolves when connection is established and tools are discovered
+     * @throws {Error} Throws an error if connection fails, with details logged to the span
+     */
+    MCPClient.prototype.connect = function (config) {
+        return __awaiter(this, void 0, void 0, function () {
+            var url, requestInit, transport, tools, error_1;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        _a.trys.push([0, 3, , 4]);
+                        url = this.url();
+                        requestInit = this.requestInit();
+                        transport = url.includes('/mcp')
+                            ? new streamableHttp_js_1.StreamableHTTPClientTransport(new URL(url), { requestInit: requestInit })
+                            : new sse_js_1.SSEClientTransport(new URL(url), { requestInit: requestInit });
+                        this.client = new index_js_1.Client({
+                            name: 'arvo-tools-agentic-mcp-client',
+                            version: '1.0.0',
+                        });
+                        return [4 /*yield*/, this.client.connect(transport)];
+                    case 1:
+                        _a.sent();
+                        (0, arvo_core_1.logToSpan)({
+                            level: 'INFO',
+                            message: "Connected to MCP Server@".concat(url),
+                        }, config.otelInfo.span);
+                        return [4 /*yield*/, this.client.listTools()];
+                    case 2:
+                        tools = _a.sent();
+                        this.availableTools = tools.tools;
+                        this.isConnected = true;
+                        (0, arvo_core_1.logToSpan)({
+                            level: 'INFO',
+                            message: 'Available MCP tools:',
+                            tools: JSON.stringify(this.availableTools.map(function (t) { return ({ name: t.name, description: t.description }); })),
+                        }, config.otelInfo.span);
+                        return [3 /*break*/, 4];
+                    case 3:
+                        error_1 = _a.sent();
+                        (0, arvo_core_1.exceptionToSpan)(error_1, config.otelInfo.span);
+                        this.isConnected = false;
+                        throw new Error("Unable to connect to the MCP Server@".concat(this.url()));
+                    case 4: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * Retrieves the list of available tool definitions from the connected MCP server.
+     *
+     * Transforms the cached MCP tools into the AgentToolDefinition format required
+     * by the agent system. Tools listed in restrictedTools will have requires_approval set to true.
+     * This method returns an empty array if not connected.
+     */
+    MCPClient.prototype.getTools = function (config) {
+        return __awaiter(this, void 0, void 0, function () {
+            var toolDef, _a, _b, item;
+            var e_1, _c;
+            var _d;
+            return __generator(this, function (_e) {
+                toolDef = [];
+                if (!this.isConnected) {
+                    (0, arvo_core_1.logToSpan)({
+                        level: 'WARNING',
+                        message: "Cannot get tools - not connected to MCP Server@".concat(this.url()),
+                    }, config.otelInfo.span);
+                    return [2 /*return*/, toolDef];
+                }
+                try {
+                    for (_a = __values(this.availableTools), _b = _a.next(); !_b.done; _b = _a.next()) {
+                        item = _b.value;
+                        toolDef.push({
+                            name: item.name,
+                            description: (_d = item.description) !== null && _d !== void 0 ? _d : '',
+                            inputSchema: item.inputSchema,
+                        });
+                    }
+                }
+                catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                finally {
+                    try {
+                        if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
+                    }
+                    finally { if (e_1) throw e_1.error; }
+                }
+                (0, arvo_core_1.logToSpan)({
+                    level: 'INFO',
+                    message: "Retrieved ".concat(toolDef.length, " tool definitions from MCP Server@").concat(this.url()),
+                }, config.otelInfo.span);
+                return [2 /*return*/, toolDef];
+            });
+        });
+    };
+    /**
+     * Invokes a specific tool on the MCP server with the provided arguments.
+     *
+     * This method sends a tool invocation request to the connected MCP server and
+     * returns the result. All operations are logged to the OpenTelemetry span for
+     * observability. If an error occurs during invocation, it returns an error message
+     * rather than throwing, allowing the agent to handle tool failures gracefully.
+     */
+    MCPClient.prototype.invokeTool = function (param, config) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, arvo_core_1.ArvoOpenTelemetry.getInstance().startActiveSpan({
+                            name: "MCP.invoke<".concat(param.name, ">"),
+                            disableSpanManagement: true,
+                            context: {
+                                inheritFrom: 'TRACE_HEADERS',
+                                traceHeaders: config.otelInfo.headers,
+                            },
+                            fn: function (span) { return __awaiter(_this, void 0, void 0, function () {
+                                var result, error_2, err;
+                                var _a, _b;
+                                return __generator(this, function (_c) {
+                                    switch (_c.label) {
+                                        case 0:
+                                            _c.trys.push([0, 2, 3, 4]);
+                                            span.setAttribute(arvo_core_1.OpenInference.ATTR_SPAN_KIND, arvo_core_1.OpenInferenceSpanKind.TOOL);
+                                            span.setStatus({
+                                                code: api_1.SpanStatusCode.OK,
+                                            });
+                                            (0, arvo_core_1.logToSpan)({
+                                                level: 'INFO',
+                                                message: "Invoking tool<".concat(param.name, "> with arguments on MCP Server@").concat(this.url()),
+                                                param: JSON.stringify(param),
+                                            }, span);
+                                            if (!this.isConnected || !this.client) {
+                                                throw new Error("MCP Server@".concat(this.url(), " not connected"));
+                                            }
+                                            return [4 /*yield*/, this.client.callTool({
+                                                    name: param.name,
+                                                    arguments: (_a = param.arguments) !== null && _a !== void 0 ? _a : undefined,
+                                                })];
+                                        case 1:
+                                            result = _c.sent();
+                                            (0, arvo_core_1.logToSpan)({
+                                                level: 'INFO',
+                                                message: "Successfully invoked tool<".concat(param.name, "> on MCP Server@").concat(this.url()),
+                                            }, span);
+                                            return [2 /*return*/, JSON.stringify(result)];
+                                        case 2:
+                                            error_2 = _c.sent();
+                                            err = new Error("Error occurred while invoking MCP tool <".concat(param.name, "@").concat(this.url(), "> -> ").concat((_b = error_2 === null || error_2 === void 0 ? void 0 : error_2.message) !== null && _b !== void 0 ? _b : 'Something went wrong'));
+                                            (0, arvo_core_1.exceptionToSpan)(err, span);
+                                            span.setStatus({
+                                                code: api_1.SpanStatusCode.ERROR,
+                                                message: err.message,
+                                            });
+                                            return [2 /*return*/, err.message];
+                                        case 3:
+                                            span.end();
+                                            return [7 /*endfinally*/];
+                                        case 4: return [2 /*return*/];
+                                    }
+                                });
+                            }); },
+                        })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        });
+    };
+    /**
+     * Gracefully disconnects from the MCP server and cleans up resources.
+     *
+     * This method safely closes the connection to the MCP server if one exists,
+     * resets the connection state, and clears cached data. It's safe to call
+     * multiple times - subsequent calls will be no-ops if already disconnected.
+     */
+    MCPClient.prototype.disconnect = function (config) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        if (!(this.client && this.isConnected)) return [3 /*break*/, 2];
+                        return [4 /*yield*/, this.client.close()];
+                    case 1:
+                        _a.sent();
+                        this.isConnected = false;
+                        this.availableTools = [];
+                        this.client = null;
+                        (0, arvo_core_1.logToSpan)({
+                            level: 'INFO',
+                            message: "Disconnected from MCP Server@".concat(this.url()),
+                        }, config.otelInfo.span);
+                        return [3 /*break*/, 3];
+                    case 2:
+                        (0, arvo_core_1.logToSpan)({
+                            level: 'INFO',
+                            message: "MCP Server@".concat(this.url(), " already disconnected"),
+                        }, config.otelInfo.span);
+                        _a.label = 3;
+                    case 3: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    MCPClient.prototype.getToolPriority = function () {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, {}];
+            });
+        });
+    };
+    return MCPClient;
+}());
+exports.MCPClient = MCPClient;
+//# sourceMappingURL=MCPClient.js.map

package/dist/Integrations/MCPClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MCPClient.js","sourceRoot":"","sources":["../../src/Integrations/MCPClient.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,sEAAmE;AACnE,kEAA6E;AAC7E,wFAAmG;AAEnG,0CAAoD;AACpD,uCAMmB;AAInB;;;;;;;GAOG;AACH;IAOE,mBACE,KAEsD;QAEtD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,GAAG,GAAG,cAAM,OAAA,CAAC,OAAO,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAnD,CAAmD,CAAC;QACrE,IAAI,CAAC,WAAW,GAAG,sBAAM,OAAA,MAAA,CAAC,OAAO,KAAK,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,mCAAI,EAAE,CAAA,EAAA,CAAC;IAC7F,CAAC;IAED;;;;;;;;;;;;OAYG;IACG,2BAAO,GAAb,UAAc,MAAkC;;;;;;;wBAEtC,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBACjB,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;wBACjC,SAAS,GAAG,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;4BACpC,CAAC,CAAC,IAAI,iDAA6B,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,aAAA,EAAE,CAAC;4BAClE,CAAC,CAAC,IAAI,2BAAkB,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,aAAA,EAAE,CAAC,CAAC;wBAE1D,IAAI,CAAC,MAAM,GAAG,IAAI,iBAAM,CAAC;4BACvB,IAAI,EAAE,+BAA+B;4BACrC,OAAO,EAAE,OAAO;yBACjB,CAAC,CAAC;wBAEH,qBAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAA;;wBAApC,SAAoC,CAAC;wBAErC,IAAA,qBAAS,EACP;4BACE,KAAK,EAAE,MAAM;4BACb,OAAO,EAAE,kCAA2B,GAAG,CAAE;yBAC1C,EACD,MAAM,CAAC,QAAQ,CAAC,IAAI,CACrB,CAAC;wBAE6B,qBAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,EAAA;;wBAAtD,KAAK,GAAoB,SAA6B;wBAC5D,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,KAAK,CAAC;wBAClC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;wBAExB,IAAA,qBAAS,EACP;4BACE,KAAK,EAAE,MAAM;4BACb,OAAO,EAAE,sBAAsB;4BAC/B,KAAK,EAAE,IAAI,CAAC,SAAS,CACnB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,UAAC,CAAO,IAAK,OAAA,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,EAA9C,CAA8C,CAAC,CACrF;yBACF,EACD,MAAM,CAAC,QAAQ,CAAC,IAAI,CACrB,CAAC;;;;wBAEF,IAAA,2BAAe,EAAC,OAAc,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;wBACtD,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;wBACzB,MAAM,IAAI,KAAK,CAAC,8CAAuC,IAAI,CAAC,GAAG,EAAE,CAAE,CAAC,CAAC;;;;;KAExE;IAED;;;;;;OAMG;IACG,4BAAQ,GAAd,UAAe,MAAkC;;;;;;gBAQzC,OAAO,GAKP,EAAE,CAAC;gBAET,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;oBACtB,IAAA,qBAAS,EACP;wBACE,KAAK,EAAE,SAAS;wBAChB,OAAO,EAAE,yDAAkD,IAAI,CAAC,GAAG,EAAE,CAAE;qBACxE,EACD,MAAM,CAAC,QAAQ,CAAC,IAAI,CACrB,CAAC;oBACF,sBAAO,OAAO,EAAC;gBACjB,CAAC;;oBAED,KAAmB,KAAA,SAAA,IAAI,CAAC,cAAc,CAAA,4CAAE,CAAC;wBAA9B,IAAI;wBACb,OAAO,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,WAAW,EAAE,MAAA,IAAI,CAAC,WAAW,mCAAI,EAAE;4BACnC,WAAW,EAAE,IAAI,CAAC,WAAW;yBAC9B,CAAC,CAAC;oBACL,CAAC;;;;;;;;;gBAED,IAAA,qBAAS,EACP;oBACE,KAAK,EAAE,MAAM;oBACb,OAAO,EAAE,oBAAa,OAAO,CAAC,MAAM,+CAAqC,IAAI,CAAC,GAAG,EAAE,CAAE;iBACtF,EACD,MAAM,CAAC,QAAQ,CAAC,IAAI,CACrB,CAAC;gBAEF,sBAAO,OAAO,EAAC;;;KAChB;IAED;;;;;;;OAOG;IACG,8BAAU,GAAhB,UACE,KAAmE,EACnE,MAAkC;;;;;4BAE3B,qBAAM,6BAAiB,CAAC,WAAW,EAAE,CAAC,eAAe,CAAC;4BAC3D,IAAI,EAAE,qBAAc,KAAK,CAAC,IAAI,MAAG;4BACjC,qBAAqB,EAAE,IAAI;4BAC3B,OAAO,EAAE;gCACP,WAAW,EAAE,eAAe;gCAC5B,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO;6BACtC;4BACD,EAAE,EAAE,UAAO,IAAI;;;;;;;4CAEX,IAAI,CAAC,YAAY,CAAC,yBAAa,CAAC,cAAc,EAAE,iCAAqB,CAAC,IAAI,CAAC,CAAC;4CAC5E,IAAI,CAAC,SAAS,CAAC;gDACb,IAAI,EAAE,oBAAc,CAAC,EAAE;6CACxB,CAAC,CAAC;4CAEH,IAAA,qBAAS,EACP;gDACE,KAAK,EAAE,MAAM;gDACb,OAAO,EAAE,wBAAiB,KAAK,CAAC,IAAI,4CAAkC,IAAI,CAAC,GAAG,EAAE,CAAE;gDAClF,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;6CAC7B,EACD,IAAI,CACL,CAAC;4CAEF,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gDACtC,MAAM,IAAI,KAAK,CAAC,qBAAc,IAAI,CAAC,GAAG,EAAE,mBAAgB,CAAC,CAAC;4CAC5D,CAAC;4CAEc,qBAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC;oDACxC,IAAI,EAAE,KAAK,CAAC,IAAI;oDAChB,SAAS,EAAE,MAAA,KAAK,CAAC,SAAS,mCAAI,SAAS;iDACxC,CAAC,EAAA;;4CAHI,MAAM,GAAG,SAGb;4CAEF,IAAA,qBAAS,EACP;gDACE,KAAK,EAAE,MAAM;gDACb,OAAO,EAAE,oCAA6B,KAAK,CAAC,IAAI,6BAAmB,IAAI,CAAC,GAAG,EAAE,CAAE;6CAChF,EACD,IAAI,CACL,CAAC;4CAEF,sBAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAC;;;4CAExB,GAAG,GAAG,IAAI,KAAK,CACnB,kDAA2C,KAAK,CAAC,IAAI,cAAI,IAAI,CAAC,GAAG,EAAE,kBAAQ,MAAC,OAAe,aAAf,OAAK,uBAAL,OAAK,CAAY,OAAO,mCAAI,sBAAsB,CAAE,CACjI,CAAC;4CAEF,IAAA,2BAAe,EAAC,GAAG,EAAE,IAAI,CAAC,CAAC;4CAC3B,IAAI,CAAC,SAAS,CAAC;gDACb,IAAI,EAAE,oBAAc,CAAC,KAAK;gDAC1B,OAAO,EAAE,GAAG,CAAC,OAAO;6CACrB,CAAC,CAAC;4CAEH,sBAAO,GAAG,CAAC,OAAO,EAAC;;4CAEnB,IAAI,CAAC,GAAG,EAAE,CAAC;;;;;iCAEd;yBACF,CAAC,EAAA;4BAzDF,sBAAO,SAyDL,EAAC;;;;KACJ;IAED;;;;;;OAMG;IACG,8BAAU,GAAhB,UAAiB,MAAkC;;;;;6BAC7C,CAAA,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAA,EAA/B,wBAA+B;wBACjC,qBAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAA;;wBAAzB,SAAyB,CAAC;wBAC1B,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;wBACzB,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;wBACzB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;wBAEnB,IAAA,qBAAS,EACP;4BACE,KAAK,EAAE,MAAM;4BACb,OAAO,EAAE,uCAAgC,IAAI,CAAC,GAAG,EAAE,CAAE;yBACtD,EACD,MAAM,CAAC,QAAQ,CAAC,IAAI,CACrB,CAAC;;;wBAEF,IAAA,qBAAS,EACP;4BACE,KAAK,EAAE,MAAM;4BACb,OAAO,EAAE,qBAAc,IAAI,CAAC,GAAG,EAAE,0BAAuB;yBACzD,EACD,MAAM,CAAC,QAAQ,CAAC,IAAI,CACrB,CAAC;;;;;;KAEL;IACK,mCAAe,GAArB;;;gBACE,sBAAO,EAAE,EAAC;;;KACX;IACH,gBAAC;AAAD,CAAC,AA1OD,IA0OC;AA1OY,8BAAS"}

package/dist/Integrations/openai.d.ts

@@ -0,0 +1,10 @@
+import type OpenAI from 'openai';
+import type { ChatModel } from 'openai/resources/shared.mjs';
+import type { AgentLLMIntegration } from '../Agent/types';
+export declare const openaiLLMIntegration: (client: OpenAI, config?: {
+    model: ChatModel;
+    temperature?: number;
+    maxTokens?: number;
+    executionunits?: (prompt: number, completion: number) => number;
+}) => AgentLLMIntegration;
+//# sourceMappingURL=openai.d.ts.map

package/dist/Integrations/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/Integrations/openai.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AAEjC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAG7D,OAAO,KAAK,EACV,mBAAmB,EAMpB,MAAM,gBAAgB,CAAC;AA+FxB,eAAO,MAAM,oBAAoB,GAE7B,QAAQ,MAAM,EACd,SAAS;IACP,KAAK,EAAE,SAAS,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,KAAK,MAAM,CAAC;CACjE,KACA,mBA8KC,CAAC"}

package/dist/Integrations/openai.js

@@ -0,0 +1,353 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __values = (this && this.__values) || function(o) {
+    var s = typeof Symbol === "function" && Symbol.iterator, m = s && o[s], i = 0;
+    if (m) return m.call(o);
+    if (o && typeof o.length === "number") return {
+        next: function () {
+            if (o && i >= o.length) o = void 0;
+            return { value: o && o[i++], done: !o };
+        }
+    };
+    throw new TypeError(s ? "Object is not iterable." : "Symbol.iterator is not defined.");
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.openaiLLMIntegration = void 0;
+var openinference_semantic_conventions_1 = require("@arizeai/openinference-semantic-conventions");
+var api_1 = require("@opentelemetry/api");
+var arvo_core_1 = require("arvo-core");
+var zod_to_json_schema_1 = require("zod-to-json-schema");
+var utils_1 = require("../Agent/utils");
+var formatMessagesForOpenAI = function (messages, systemPrompt) {
+    var e_1, _a, e_2, _b;
+    var _c;
+    var formattedMessages = [];
+    if (systemPrompt) {
+        formattedMessages.push({
+            role: 'system',
+            content: systemPrompt,
+        });
+    }
+    var toolResponseMap = {};
+    try {
+        for (var messages_1 = __values(messages), messages_1_1 = messages_1.next(); !messages_1_1.done; messages_1_1 = messages_1.next()) {
+            var message = messages_1_1.value;
+            if (message.role === 'user' && message.content.type === 'tool_result') {
+                toolResponseMap[message.content.toolUseId] = message.content;
+            }
+        }
+    }
+    catch (e_1_1) { e_1 = { error: e_1_1 }; }
+    finally {
+        try {
+            if (messages_1_1 && !messages_1_1.done && (_a = messages_1.return)) _a.call(messages_1);
+        }
+        finally { if (e_1) throw e_1.error; }
+    }
+    try {
+        for (var messages_2 = __values(messages), messages_2_1 = messages_2.next(); !messages_2_1.done; messages_2_1 = messages_2.next()) {
+            var message = messages_2_1.value;
+            if (message.role === 'user') {
+                if (message.content.type === 'text') {
+                    formattedMessages.push({
+                        role: 'user',
+                        content: message.content.content,
+                    });
+                }
+                else if (message.content.type === 'media' && message.content.contentType.type === 'image') {
+                    formattedMessages.push({
+                        role: 'user',
+                        content: [
+                            {
+                                type: 'image_url',
+                                image_url: {
+                                    url: message.content.content,
+                                },
+                            },
+                        ],
+                    });
+                }
+                else if (message.content.type === 'media' && message.content.contentType.type === 'file') {
+                    formattedMessages.push({
+                        role: 'user',
+                        content: [
+                            {
+                                type: 'file',
+                                file: {
+                                    filename: message.content.contentType.filename,
+                                    file_data: message.content.content,
+                                },
+                            },
+                        ],
+                    });
+                }
+            }
+            else if (message.role === 'assistant') {
+                if (message.content.type === 'text') {
+                    formattedMessages.push({
+                        role: 'assistant',
+                        content: message.content.content,
+                    });
+                }
+                else if (message.content.type === 'tool_use') {
+                    formattedMessages.push({
+                        role: 'assistant',
+                        tool_calls: [
+                            {
+                                type: 'function',
+                                id: message.content.toolUseId,
+                                function: {
+                                    name: message.content.name,
+                                    arguments: JSON.stringify(message.content.input),
+                                },
+                            },
+                        ],
+                    });
+                    var toolResult = toolResponseMap[message.content.toolUseId];
+                    formattedMessages.push({
+                        role: 'tool',
+                        tool_call_id: message.content.toolUseId,
+                        content: (_c = toolResult === null || toolResult === void 0 ? void 0 : toolResult.content) !== null && _c !== void 0 ? _c : JSON.stringify({ error: 'No tool response' }),
+                    });
+                }
+            }
+        }
+    }
+    catch (e_2_1) { e_2 = { error: e_2_1 }; }
+    finally {
+        try {
+            if (messages_2_1 && !messages_2_1.done && (_b = messages_2.return)) _b.call(messages_2);
+        }
+        finally { if (e_2) throw e_2.error; }
+    }
+    return formattedMessages;
+};
+var openaiLLMIntegration = function (client, config) {
+    return function (_a, _b) { return __awaiter(void 0, [_a, _b], void 0, function (_c, _d) {
+        var _e;
+        var _messages = _c.messages, _system = _c.system, tools = _c.tools, outputFormat = _c.outputFormat, lifecycle = _c.lifecycle, toolInteractions = _c.toolInteractions;
+        var otelInfo = _d.otelInfo;
+        return __generator(this, function (_f) {
+            switch (_f.label) {
+                case 0: return [4 /*yield*/, arvo_core_1.ArvoOpenTelemetry.getInstance().startActiveSpan({
+                        name: "LLM.invoke<".concat(lifecycle === 'init' ? 'init' : lifecycle === 'tool_result' ? 'resume' : 'output_validation_feedback', ">"),
+                        disableSpanManagement: true,
+                        context: {
+                            inheritFrom: 'TRACE_HEADERS',
+                            traceHeaders: otelInfo.headers,
+                        },
+                        spanOptions: {
+                            attributes: (_e = {},
+                                _e[openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND] = openinference_semantic_conventions_1.OpenInferenceSpanKind.LLM,
+                                _e),
+                        },
+                        fn: function (span) { return __awaiter(void 0, void 0, void 0, function () {
+                            var llmModel, llmInvocationParams, messages, system, limitMessage, toolDef, tools_1, tools_1_1, tool, formattedMessages, responseFormat, completion, choice, llmUsage, executionUnits, toolRequests, _a, _b, toolCall, content, e_3;
+                            var e_4, _c, e_5, _d;
+                            var _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
+                            return __generator(this, function (_s) {
+                                switch (_s.label) {
+                                    case 0:
+                                        llmModel = (_e = config === null || config === void 0 ? void 0 : config.model) !== null && _e !== void 0 ? _e : 'gpt-4o';
+                                        llmInvocationParams = {
+                                            temperature: (_f = config === null || config === void 0 ? void 0 : config.temperature) !== null && _f !== void 0 ? _f : 0,
+                                            maxTokens: (_g = config === null || config === void 0 ? void 0 : config.maxTokens) !== null && _g !== void 0 ? _g : 4096,
+                                        };
+                                        messages = _messages.map(function (item) {
+                                            if (lifecycle === 'init')
+                                                return item;
+                                            if (item.content.type === 'media') {
+                                                return {
+                                                    role: item.role,
+                                                    content: {
+                                                        type: 'text',
+                                                        content: "Media file (type: ".concat(item.content.contentType.type, "@").concat(item.content.contentType.format, ") already parsed and looked at. No need for you to look at it again"),
+                                                    },
+                                                };
+                                            }
+                                            return item;
+                                        });
+                                        system = _system;
+                                        if (toolInteractions.exhausted) {
+                                            limitMessage = (0, arvo_core_1.cleanString)("\n            **CRITICAL WARNING: You have reached your tool interaction limit!**\n            You must answer the original question using all the data available to you. \n            You have run out of tool call budget. No more tool calls are allowed any more.\n            If you cannot answer the query well. Then mention what you have done briefly, what\n            can you answer based on the collected data, what data is missing and why you cannot \n            answer any further.  \n          ");
+                                            messages.push({
+                                                role: 'user',
+                                                content: {
+                                                    type: 'text',
+                                                    content: limitMessage,
+                                                },
+                                            });
+                                            system = "".concat(system, "\n\n").concat(limitMessage);
+                                        }
+                                        (0, utils_1.setOpenInferenceInputAttr)({
+                                            llm: {
+                                                provider: 'openai',
+                                                system: 'openai',
+                                                model: llmModel,
+                                                invocationParam: llmInvocationParams,
+                                            },
+                                            messages: messages,
+                                            system: system,
+                                            tools: tools,
+                                        }, span);
+                                        _s.label = 1;
+                                    case 1:
+                                        _s.trys.push([1, 3, 4, 5]);
+                                        toolDef = [];
+                                        try {
+                                            for (tools_1 = __values(tools), tools_1_1 = tools_1.next(); !tools_1_1.done; tools_1_1 = tools_1.next()) {
+                                                tool = tools_1_1.value;
+                                                toolDef.push({
+                                                    type: 'function',
+                                                    function: {
+                                                        name: tool.name,
+                                                        description: tool.description,
+                                                        parameters: tool.inputSchema,
+                                                    },
+                                                });
+                                            }
+                                        }
+                                        catch (e_4_1) { e_4 = { error: e_4_1 }; }
+                                        finally {
+                                            try {
+                                                if (tools_1_1 && !tools_1_1.done && (_c = tools_1.return)) _c.call(tools_1);
+                                            }
+                                            finally { if (e_4) throw e_4.error; }
+                                        }
+                                        formattedMessages = formatMessagesForOpenAI(messages, system);
+                                        responseFormat = outputFormat.type === 'json'
+                                            ? {
+                                                type: 'json_schema',
+                                                json_schema: {
+                                                    name: 'response_schema',
+                                                    description: 'The required response schema',
+                                                    // biome-ignore lint/suspicious/noExplicitAny: Make the typescript compiler ignore. Otherwise, it emits error "Type instantiation is excessively deep and possibly infinite. ts(2589)"
+                                                    schema: (0, zod_to_json_schema_1.zodToJsonSchema)(outputFormat.format),
+                                                },
+                                            }
+                                            : undefined;
+                                        return [4 /*yield*/, client.chat.completions.create({
+                                                model: llmModel,
+                                                max_tokens: llmInvocationParams.maxTokens,
+                                                temperature: llmInvocationParams.temperature,
+                                                tools: toolDef.length ? toolDef : undefined,
+                                                messages: formattedMessages,
+                                                response_format: responseFormat,
+                                            })];
+                                    case 2:
+                                        completion = _s.sent();
+                                        choice = completion.choices[0];
+                                        llmUsage = {
+                                            tokens: {
+                                                prompt: (_j = (_h = completion.usage) === null || _h === void 0 ? void 0 : _h.prompt_tokens) !== null && _j !== void 0 ? _j : 0,
+                                                completion: (_l = (_k = completion.usage) === null || _k === void 0 ? void 0 : _k.completion_tokens) !== null && _l !== void 0 ? _l : 0,
+                                            },
+                                        };
+                                        executionUnits = (_o = (_m = config === null || config === void 0 ? void 0 : config.executionunits) === null || _m === void 0 ? void 0 : _m.call(config, llmUsage.tokens.prompt, llmUsage.tokens.completion)) !== null && _o !== void 0 ? _o : llmUsage.tokens.prompt + llmUsage.tokens.completion;
+                                        (0, utils_1.setOpenInferenceUsageOutputAttr)(llmUsage, span);
+                                        if ((_p = choice === null || choice === void 0 ? void 0 : choice.message) === null || _p === void 0 ? void 0 : _p.tool_calls) {
+                                            toolRequests = [];
+                                            try {
+                                                for (_a = __values(choice.message
+                                                    .tool_calls), _b = _a.next(); !_b.done; _b = _a.next()) {
+                                                    toolCall = _b.value;
+                                                    try {
+                                                        toolRequests.push({
+                                                            toolUseId: toolCall.id,
+                                                            name: toolCall.function.name,
+                                                            input: JSON.parse(toolCall.function.arguments),
+                                                        });
+                                                    }
+                                                    catch (_t) {
+                                                        // Skip malformed tool calls
+                                                    }
+                                                }
+                                            }
+                                            catch (e_5_1) { e_5 = { error: e_5_1 }; }
+                                            finally {
+                                                try {
+                                                    if (_b && !_b.done && (_d = _a.return)) _d.call(_a);
+                                                }
+                                                finally { if (e_5) throw e_5.error; }
+                                            }
+                                            if (toolRequests.length) {
+                                                (0, utils_1.setOpenInferenceToolCallOutputAttr)({ toolCalls: toolRequests }, span);
+                                                return [2 /*return*/, {
+                                                        type: 'tool_call',
+                                                        toolRequests: toolRequests,
+                                                        usage: llmUsage,
+                                                        executionUnits: executionUnits,
+                                                    }];
+                                            }
+                                        }
+                                        content = (_r = (_q = choice === null || choice === void 0 ? void 0 : choice.message) === null || _q === void 0 ? void 0 : _q.content) !== null && _r !== void 0 ? _r : '';
+                                        (0, utils_1.setOpenInferenceResponseOutputAttr)({ response: content }, span);
+                                        if (outputFormat.type === 'json') {
+                                            return [2 /*return*/, {
+                                                    type: 'json',
+                                                    content: content || '{}',
+                                                    parsedContent: (0, utils_1.tryParseJson)(content || '{}'),
+                                                    usage: llmUsage,
+                                                    executionUnits: executionUnits,
+                                                }];
+                                        }
+                                        return [2 /*return*/, {
+                                                type: 'text',
+                                                content: content,
+                                                usage: llmUsage,
+                                                executionUnits: executionUnits,
+                                            }];
+                                    case 3:
+                                        e_3 = _s.sent();
+                                        span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: e_3 === null || e_3 === void 0 ? void 0 : e_3.message });
+                                        throw e_3;
+                                    case 4:
+                                        span.end();
+                                        return [7 /*endfinally*/];
+                                    case 5: return [2 /*return*/];
+                                }
+                            });
+                        }); },
+                    })];
+                case 1: return [2 /*return*/, _f.sent()];
+            }
+        });
+    }); };
+};
+exports.openaiLLMIntegration = openaiLLMIntegration;
+//# sourceMappingURL=openai.js.map

package/dist/Integrations/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/Integrations/openai.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,kGAGqD;AACrD,0CAAoD;AACpD,uCAA2D;AAK3D,yDAAqD;AASrD,wCAMwB;AAExB,IAAM,uBAAuB,GAAG,UAC9B,QAA8C,EAC9C,YAA2B;;;IAE3B,IAAM,iBAAiB,GAAiC,EAAE,CAAC;IAE3D,IAAI,YAAY,EAAE,CAAC;QACjB,iBAAiB,CAAC,IAAI,CAAC;YACrB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,YAAY;SACtB,CAAC,CAAC;IACL,CAAC;IAED,IAAM,eAAe,GAA2C,EAAE,CAAC;;QACnE,KAAsB,IAAA,aAAA,SAAA,QAAQ,CAAA,kCAAA,wDAAE,CAAC;YAA5B,IAAM,OAAO,qBAAA;YAChB,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;gBACtE,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;YAC/D,CAAC;QACH,CAAC;;;;;;;;;;QAED,KAAsB,IAAA,aAAA,SAAA,QAAQ,CAAA,kCAAA,wDAAE,CAAC;YAA5B,IAAM,OAAO,qBAAA;YAChB,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACpC,iBAAiB,CAAC,IAAI,CAAC;wBACrB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO;qBACjC,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBAC5F,iBAAiB,CAAC,IAAI,CAAC;wBACrB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,WAAW;gCACjB,SAAS,EAAE;oCACT,GAAG,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO;iCAC7B;6BACF;yBACF;qBACF,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC3F,iBAAiB,CAAC,IAAI,CAAC;wBACrB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE;oCACJ,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ;oCAC9C,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO;iCACnC;6BACF;yBACF;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACxC,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACpC,iBAAiB,CAAC,IAAI,CAAC;wBACrB,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO;qBACjC,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,OAAO,CAAC,OAAO,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC/C,iBAAiB,CAAC,IAAI,CAAC;wBACrB,IAAI,EAAE,WAAW;wBACjB,UAAU,EAAE;4BACV;gCACE,IAAI,EAAE,UAAU;gCAChB,EAAE,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS;gCAC7B,QAAQ,EAAE;oCACR,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI;oCAC1B,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC;iCACjD;6BACF;yBACF;qBACF,CAAC,CAAC;oBACH,IAAM,UAAU,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;oBAC9D,iBAAiB,CAAC,IAAI,CAAC;wBACrB,IAAI,EAAE,MAAM;wBACZ,YAAY,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS;wBACvC,OAAO,EAAE,MAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,OAAO,mCAAI,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC;qBAC9E,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;;;;;;;;;IAED,OAAO,iBAAiB,CAAC;AAC3B,CAAC,CAAC;AAEK,IAAM,oBAAoB,GAC/B,UACE,MAAc,EACd,MAKC;IAEH,OAAA,yEACE,EAA0F,EAC1F,EAAY;;YADA,SAAS,cAAA,EAAU,OAAO,YAAA,EAAE,KAAK,WAAA,EAAE,YAAY,kBAAA,EAAE,SAAS,eAAA,EAAE,gBAAgB,sBAAA;YACtF,QAAQ,cAAA;;;wBAEV,qBAAM,6BAAiB,CAAC,WAAW,EAAE,CAAC,eAAe,CAAC;wBACpD,IAAI,EAAE,qBAAc,SAAS,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,KAAK,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,4BAA4B,MAAG;wBAC5H,qBAAqB,EAAE,IAAI;wBAC3B,OAAO,EAAE;4BACP,WAAW,EAAE,eAAe;4BAC5B,YAAY,EAAE,QAAQ,CAAC,OAAO;yBAC/B;wBACD,WAAW,EAAE;4BACX,UAAU;gCACR,GAAC,wDAAgC,CAAC,uBAAuB,IAAG,0DAAqB,CAAC,GAAG;mCACtF;yBACF;wBACD,EAAE,EAAE,UAAO,IAAI;;;;;;;wCACP,QAAQ,GAAc,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,KAAK,mCAAI,QAAQ,CAAC;wCAChD,mBAAmB,GAAG;4CAC1B,WAAW,EAAE,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,WAAW,mCAAI,CAAC;4CACrC,SAAS,EAAE,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,mCAAI,IAAI;yCACrC,CAAC;wCAEI,QAAQ,GAAmB,SAAS,CAAC,GAAG,CAAC,UAAC,IAAI;4CAClD,IAAI,SAAS,KAAK,MAAM;gDAAE,OAAO,IAAI,CAAC;4CACtC,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gDAClC,OAAO;oDACL,IAAI,EAAE,IAAI,CAAC,IAAI;oDACf,OAAO,EAAE;wDACP,IAAI,EAAE,MAAM;wDACZ,OAAO,EAAE,4BAAqB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,cAAI,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,MAAM,wEAAqE;qDACpK;iDACF,CAAC;4CACJ,CAAC;4CACD,OAAO,IAAI,CAAC;wCACd,CAAC,CAAC,CAAC;wCACC,MAAM,GAAG,OAAO,CAAC;wCAErB,IAAI,gBAAgB,CAAC,SAAS,EAAE,CAAC;4CACzB,YAAY,GAAG,IAAA,uBAAW,EAAC,yfAOhC,CAAC,CAAC;4CACH,QAAQ,CAAC,IAAI,CAAC;gDACZ,IAAI,EAAE,MAAM;gDACZ,OAAO,EAAE;oDACP,IAAI,EAAE,MAAM;oDACZ,OAAO,EAAE,YAAY;iDACtB;6CACF,CAAC,CAAC;4CACH,MAAM,GAAG,UAAG,MAAM,iBAAO,YAAY,CAAE,CAAC;wCAC1C,CAAC;wCAED,IAAA,iCAAyB,EACvB;4CACE,GAAG,EAAE;gDACH,QAAQ,EAAE,QAAQ;gDAClB,MAAM,EAAE,QAAQ;gDAChB,KAAK,EAAE,QAAQ;gDACf,eAAe,EAAE,mBAAmB;6CACrC;4CACD,QAAQ,UAAA;4CACR,MAAM,QAAA;4CACN,KAAK,OAAA;yCACN,EACD,IAAI,CACL,CAAC;;;;wCAGM,OAAO,GAAyB,EAAE,CAAC;;4CACzC,KAAmB,UAAA,SAAA,KAAK,CAAA,2EAAE,CAAC;gDAAhB,IAAI;gDACb,OAAO,CAAC,IAAI,CAAC;oDACX,IAAI,EAAE,UAAU;oDAChB,QAAQ,EAAE;wDACR,IAAI,EAAE,IAAI,CAAC,IAAI;wDACf,WAAW,EAAE,IAAI,CAAC,WAAW;wDAC7B,UAAU,EAAE,IAAI,CAAC,WAAW;qDAC7B;iDACF,CAAC,CAAC;4CACL,CAAC;;;;;;;;;wCAEK,iBAAiB,GAAG,uBAAuB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;wCAE9D,cAAc,GAClB,YAAY,CAAC,IAAI,KAAK,MAAM;4CAC1B,CAAC,CAAC;gDACE,IAAI,EAAE,aAAsB;gDAC5B,WAAW,EAAE;oDACX,IAAI,EAAE,iBAAiB;oDACvB,WAAW,EAAE,8BAA8B;oDAC3C,sLAAsL;oDACtL,MAAM,EAAE,IAAA,oCAAe,EAAC,YAAY,CAAC,MAAa,CAAC;iDACpD;6CACF;4CACH,CAAC,CAAC,SAAS,CAAC;wCAEG,qBAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gDACtD,KAAK,EAAE,QAAQ;gDACf,UAAU,EAAE,mBAAmB,CAAC,SAAS;gDACzC,WAAW,EAAE,mBAAmB,CAAC,WAAW;gDAC5C,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;gDAC3C,QAAQ,EAAE,iBAAiB;gDAC3B,eAAe,EAAE,cAAc;6CAChC,CAAC,EAAA;;wCAPI,UAAU,GAAG,SAOjB;wCAEI,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;wCAC/B,QAAQ,GAAoD;4CAChE,MAAM,EAAE;gDACN,MAAM,EAAE,MAAA,MAAA,UAAU,CAAC,KAAK,0CAAE,aAAa,mCAAI,CAAC;gDAC5C,UAAU,EAAE,MAAA,MAAA,UAAU,CAAC,KAAK,0CAAE,iBAAiB,mCAAI,CAAC;6CACrD;yCACF,CAAC;wCACI,cAAc,GAClB,MAAA,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,cAAc,uDAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,mCAC5E,QAAQ,CAAC,MAAM,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC;wCAEtD,IAAA,uCAA+B,EAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;wCAEhD,IAAI,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,0CAAE,UAAU,EAAE,CAAC;4CAC1B,YAAY,GAAyC,EAAE,CAAC;;gDAC9D,KAAuB,KAAA,SAAA,MAAM,CAAC,OAAO;qDAClC,UAAqD,CAAA,4CAAE,CAAC;oDADhD,QAAQ;oDAEjB,IAAI,CAAC;wDACH,YAAY,CAAC,IAAI,CAAC;4DAChB,SAAS,EAAE,QAAQ,CAAC,EAAE;4DACtB,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;4DAC5B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAA4B;yDAC1E,CAAC,CAAC;oDACL,CAAC;oDAAC,WAAM,CAAC;wDACP,4BAA4B;oDAC9B,CAAC;gDACH,CAAC;;;;;;;;;4CAED,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;gDACxB,IAAA,0CAAkC,EAAC,EAAE,SAAS,EAAE,YAAY,EAAE,EAAE,IAAI,CAAC,CAAC;gDACtE,sBAAO;wDACL,IAAI,EAAE,WAAW;wDACjB,YAAY,cAAA;wDACZ,KAAK,EAAE,QAAQ;wDACf,cAAc,gBAAA;qDACf,EAAC;4CACJ,CAAC;wCACH,CAAC;wCAEK,OAAO,GAAG,MAAA,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,0CAAE,OAAO,mCAAI,EAAE,CAAC;wCAC/C,IAAA,0CAAkC,EAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;wCAChE,IAAI,YAAY,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4CACjC,sBAAO;oDACL,IAAI,EAAE,MAAM;oDACZ,OAAO,EAAE,OAAO,IAAI,IAAI;oDACxB,aAAa,EAAE,IAAA,oBAAY,EAAC,OAAO,IAAI,IAAI,CAAC;oDAC5C,KAAK,EAAE,QAAQ;oDACf,cAAc,gBAAA;iDACf,EAAC;wCACJ,CAAC;wCAED,sBAAO;gDACL,IAAI,EAAE,MAAM;gDACZ,OAAO,SAAA;gDACP,KAAK,EAAE,QAAQ;gDACf,cAAc,gBAAA;6CACf,EAAC;;;wCAEF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,KAAK,EAAE,OAAO,EAAG,GAAW,aAAX,GAAC,uBAAD,GAAC,CAAY,OAAO,EAAE,CAAC,CAAC;wCAC/E,MAAM,GAAC,CAAC;;wCAER,IAAI,CAAC,GAAG,EAAE,CAAC;;;;;6BAEd;qBACF,CAAC,EAAA;wBAzKF,sBAAA,SAyKE,EAAA;;;SAAA;AA7KJ,CA6KI,CAAC;AAvLM,QAAA,oBAAoB,wBAuL1B"}

package/dist/index.d.ts

@@ -1,7 +1,11 @@
 export { createArvoAgent } from './Agent';
 export { AgentDefaults } from './Agent/AgentDefaults';
+export { createAgentTool } from './Agent/agentTool';
 export { AgentMediaContentSchema, AgentMessageContentSchema, AgentMessageSchema, AgentTextContentSchema, AgentToolCallContentSchema, AgentToolResultContentSchema, } from './Agent/schema';
 export type { AgentContextBuilder, AgentInternalTool, AgentLLMContext, AgentLLMIntegration, AgentLLMIntegrationOutput, AgentLLMIntegrationParam, AgentMediaContent, AgentMessage, AgentMessageContent, AgentOutputBuilder, AgentServiceContract, AgentTextContent, AgentToolCallContent, AgentToolDefinition, AgentToolResultContent, AnyArvoContract, AnyArvoOrchestratorContract, CreateArvoAgentParam, NonEmptyArray, PromiseLike, } from './Agent/types';
+export { setOpenInferenceInputAttr, setOpenInferenceResponseOutputAttr, setOpenInferenceToolCallOutputAttr, setOpenInferenceUsageOutputAttr, tryParseJson, } from './Agent/utils';
+export { MCPClient } from './Integrations/MCPClient';
+export { openaiLLMIntegration } from './Integrations/openai';
 export type { IMCPClient } from './interfaces.mcp';
 export type { OtelInfoType } from './types';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE1C,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAEtD,OAAO,EACL,uBAAuB,EACvB,yBAAyB,EACzB,kBAAkB,EAClB,sBAAsB,EACtB,0BAA0B,EAC1B,4BAA4B,GAC7B,MAAM,gBAAgB,CAAC;AAExB,YAAY,EACV,mBAAmB,EACnB,iBAAiB,EACjB,eAAe,EACf,mBAAmB,EACnB,yBAAyB,EACzB,wBAAwB,EACxB,iBAAiB,EACjB,YAAY,EACZ,mBAAmB,EACnB,kBAAkB,EAClB,oBAAoB,EACpB,gBAAgB,EAChB,oBAAoB,EACpB,mBAAmB,EACnB,sBAAsB,EACtB,eAAe,EACf,2BAA2B,EAC3B,oBAAoB,EACpB,aAAa,EACb,WAAW,GACZ,MAAM,eAAe,CAAC;AAEvB,YAAY,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,YAAY,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EACL,uBAAuB,EACvB,yBAAyB,EACzB,kBAAkB,EAClB,sBAAsB,EACtB,0BAA0B,EAC1B,4BAA4B,GAC7B,MAAM,gBAAgB,CAAC;AACxB,YAAY,EACV,mBAAmB,EACnB,iBAAiB,EACjB,eAAe,EACf,mBAAmB,EACnB,yBAAyB,EACzB,wBAAwB,EACxB,iBAAiB,EACjB,YAAY,EACZ,mBAAmB,EACnB,kBAAkB,EAClB,oBAAoB,EACpB,gBAAgB,EAChB,oBAAoB,EACpB,mBAAmB,EACnB,sBAAsB,EACtB,eAAe,EACf,2BAA2B,EAC3B,oBAAoB,EACpB,aAAa,EACb,WAAW,GACZ,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,yBAAyB,EACzB,kCAAkC,EAClC,kCAAkC,EAClC,+BAA+B,EAC/B,YAAY,GACb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AACrD,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAE7D,YAAY,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,YAAY,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC"}

package/dist/index.js

@@ -1,10 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AgentToolResultContentSchema = exports.AgentToolCallContentSchema = exports.AgentTextContentSchema = exports.AgentMessageSchema = exports.AgentMessageContentSchema = exports.AgentMediaContentSchema = exports.AgentDefaults = exports.createArvoAgent = void 0;
+exports.openaiLLMIntegration = exports.MCPClient = exports.tryParseJson = exports.setOpenInferenceUsageOutputAttr = exports.setOpenInferenceToolCallOutputAttr = exports.setOpenInferenceResponseOutputAttr = exports.setOpenInferenceInputAttr = exports.AgentToolResultContentSchema = exports.AgentToolCallContentSchema = exports.AgentTextContentSchema = exports.AgentMessageSchema = exports.AgentMessageContentSchema = exports.AgentMediaContentSchema = exports.createAgentTool = exports.AgentDefaults = exports.createArvoAgent = void 0;
 var Agent_1 = require("./Agent");
 Object.defineProperty(exports, "createArvoAgent", { enumerable: true, get: function () { return Agent_1.createArvoAgent; } });
 var AgentDefaults_1 = require("./Agent/AgentDefaults");
 Object.defineProperty(exports, "AgentDefaults", { enumerable: true, get: function () { return AgentDefaults_1.AgentDefaults; } });
+var agentTool_1 = require("./Agent/agentTool");
+Object.defineProperty(exports, "createAgentTool", { enumerable: true, get: function () { return agentTool_1.createAgentTool; } });
 var schema_1 = require("./Agent/schema");
 Object.defineProperty(exports, "AgentMediaContentSchema", { enumerable: true, get: function () { return schema_1.AgentMediaContentSchema; } });
 Object.defineProperty(exports, "AgentMessageContentSchema", { enumerable: true, get: function () { return schema_1.AgentMessageContentSchema; } });
@@ -12,4 +14,14 @@ Object.defineProperty(exports, "AgentMessageSchema", { enumerable: true, get: fu
 Object.defineProperty(exports, "AgentTextContentSchema", { enumerable: true, get: function () { return schema_1.AgentTextContentSchema; } });
 Object.defineProperty(exports, "AgentToolCallContentSchema", { enumerable: true, get: function () { return schema_1.AgentToolCallContentSchema; } });
 Object.defineProperty(exports, "AgentToolResultContentSchema", { enumerable: true, get: function () { return schema_1.AgentToolResultContentSchema; } });
+var utils_1 = require("./Agent/utils");
+Object.defineProperty(exports, "setOpenInferenceInputAttr", { enumerable: true, get: function () { return utils_1.setOpenInferenceInputAttr; } });
+Object.defineProperty(exports, "setOpenInferenceResponseOutputAttr", { enumerable: true, get: function () { return utils_1.setOpenInferenceResponseOutputAttr; } });
+Object.defineProperty(exports, "setOpenInferenceToolCallOutputAttr", { enumerable: true, get: function () { return utils_1.setOpenInferenceToolCallOutputAttr; } });
+Object.defineProperty(exports, "setOpenInferenceUsageOutputAttr", { enumerable: true, get: function () { return utils_1.setOpenInferenceUsageOutputAttr; } });
+Object.defineProperty(exports, "tryParseJson", { enumerable: true, get: function () { return utils_1.tryParseJson; } });
+var MCPClient_1 = require("./Integrations/MCPClient");
+Object.defineProperty(exports, "MCPClient", { enumerable: true, get: function () { return MCPClient_1.MCPClient; } });
+var openai_1 = require("./Integrations/openai");
+Object.defineProperty(exports, "openaiLLMIntegration", { enumerable: true, get: function () { return openai_1.openaiLLMIntegration; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,iCAA0C;AAAjC,wGAAA,eAAe,OAAA;AAExB,uDAAsD;AAA7C,8GAAA,aAAa,OAAA;AAEtB,yCAOwB;AANtB,iHAAA,uBAAuB,OAAA;AACvB,mHAAA,yBAAyB,OAAA;AACzB,4GAAA,kBAAkB,OAAA;AAClB,gHAAA,sBAAsB,OAAA;AACtB,oHAAA,0BAA0B,OAAA;AAC1B,sHAAA,4BAA4B,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,iCAA0C;AAAjC,wGAAA,eAAe,OAAA;AACxB,uDAAsD;AAA7C,8GAAA,aAAa,OAAA;AACtB,+CAAoD;AAA3C,4GAAA,eAAe,OAAA;AACxB,yCAOwB;AANtB,iHAAA,uBAAuB,OAAA;AACvB,mHAAA,yBAAyB,OAAA;AACzB,4GAAA,kBAAkB,OAAA;AAClB,gHAAA,sBAAsB,OAAA;AACtB,oHAAA,0BAA0B,OAAA;AAC1B,sHAAA,4BAA4B,OAAA;AAwB9B,uCAMuB;AALrB,kHAAA,yBAAyB,OAAA;AACzB,2HAAA,kCAAkC,OAAA;AAClC,2HAAA,kCAAkC,OAAA;AAClC,wHAAA,+BAA+B,OAAA;AAC/B,qGAAA,YAAY,OAAA;AAEd,sDAAqD;AAA5C,sGAAA,SAAS,OAAA;AAClB,gDAA6D;AAApD,8GAAA,oBAAoB,OAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arvo-tools/agentic",
-  "version": "0.3.0",
+  "version": "0.4.1",
   "description": "Agentic toolset for building applications using arvo-core and arvo-event-handler",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -26,11 +26,12 @@
   },
   "dependencies": {
     "@arizeai/openinference-semantic-conventions": "^2.1.2",
+    "@modelcontextprotocol/sdk": "^1.22.0",
     "@opentelemetry/api": "1.9.0",
     "arvo-core": "^3.0.19",
     "arvo-event-handler": "^3.0.19",
     "openai": "^6.9.1",
-    "uuid": "^13.0.0",
+    "uuid": "^11.1.0",
     "zod": "^3.25.74",
     "zod-to-json-schema": "^3.24.6"
   },