@forestadmin/ai-proxy 1.0.1 → 1.2.0

package/dist/index.js

@@ -18,13 +18,13 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.validMcpConfigurationOrThrow = validMcpConfigurationOrThrow;
-const mcp_config_checker_1 = __importDefault(require("./mcp-config-checker"));
+const mcp_config_checker_1 = __importDefault(require("./types/mcp-config-checker"));
 __exportStar(require("./provider-dispatcher"), exports);
 __exportStar(require("./remote-tools"), exports);
 __exportStar(require("./router"), exports);
 __exportStar(require("./mcp-client"), exports);
-__exportStar(require("./errors"), exports);
+__exportStar(require("./types/errors"), exports);
 function validMcpConfigurationOrThrow(mcpConfig) {
     return mcp_config_checker_1.default.check(mcpConfig);
 }
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvaW5kZXgudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQVdBLG9FQUVDO0FBWEQsOEVBQW9EO0FBRXBELHdEQUFzQztBQUN0QyxpREFBK0I7QUFDL0IsMkNBQXlCO0FBQ3pCLCtDQUE2QjtBQUU3QiwyQ0FBeUI7QUFFekIsU0FBZ0IsNEJBQTRCLENBQUMsU0FBMkI7SUFDdEUsT0FBTyw0QkFBZ0IsQ0FBQyxLQUFLLENBQUMsU0FBUyxDQUFDLENBQUM7QUFDM0MsQ0FBQyJ9
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvaW5kZXgudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQVdBLG9FQUVDO0FBWEQsb0ZBQTBEO0FBRTFELHdEQUFzQztBQUN0QyxpREFBK0I7QUFDL0IsMkNBQXlCO0FBQ3pCLCtDQUE2QjtBQUU3QixpREFBK0I7QUFFL0IsU0FBZ0IsNEJBQTRCLENBQUMsU0FBMkI7SUFDdEUsT0FBTyw0QkFBZ0IsQ0FBQyxLQUFLLENBQUMsU0FBUyxDQUFDLENBQUM7QUFDM0MsQ0FBQyJ9

package/dist/mcp-client.d.ts

@@ -1,15 +1,15 @@
 import type { Logger } from '@forestadmin/datasource-toolkit';
 import { MultiServerMCPClient } from '@langchain/mcp-adapters';
-import RemoteTool from './remote-tool';
+import McpServerRemoteTool from './types/mcp-server-remote-tool';
 export type McpConfiguration = {
     configs: MultiServerMCPClient['config']['mcpServers'];
 } & Omit<MultiServerMCPClient['config'], 'mcpServers'>;
 export default class McpClient {
     private readonly mcpClients;
     private readonly logger?;
-    readonly tools: RemoteTool[];
+    readonly tools: McpServerRemoteTool[];
     constructor(config: McpConfiguration, logger?: Logger);
-    loadTools(): Promise<RemoteTool[]>;
+    loadTools(): Promise<McpServerRemoteTool[]>;
     testConnections(): Promise<true>;
     closeConnections(): Promise<void>;
 }

package/dist/mcp-client.js

@@ -4,8 +4,8 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const mcp_adapters_1 = require("@langchain/mcp-adapters");
-const errors_1 = require("./errors");
-const remote_tool_1 = __importDefault(require("./remote-tool"));
+const errors_1 = require("./types/errors");
+const mcp_server_remote_tool_1 = __importDefault(require("./types/mcp-server-remote-tool"));
 class McpClient {
     constructor(config, logger) {
         this.mcpClients = {};
@@ -25,11 +25,7 @@ class McpClient {
         await Promise.all(Object.entries(this.mcpClients).map(async ([name, client]) => {
             try {
                 const tools = (await client.getTools()) ?? [];
-                const extendedTools = tools.map(tool => new remote_tool_1.default({
-                    tool,
-                    sourceId: name,
-                    sourceType: 'mcp-server',
-                }));
+                const extendedTools = tools.map(tool => new mcp_server_remote_tool_1.default({ tool, sourceId: name }));
                 this.tools.push(...extendedTools);
             }
             catch (error) {
@@ -80,4 +76,4 @@ class McpClient {
     }
 }
 exports.default = McpClient;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibWNwLWNsaWVudC5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9tY3AtY2xpZW50LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7O0FBRUEsMERBQStEO0FBRS9ELHFDQUE4QztBQUM5QyxnRUFBdUM7QUFNdkMsTUFBcUIsU0FBUztJQU01QixZQUFZLE1BQXdCLEVBQUUsTUFBZTtRQUxwQyxlQUFVLEdBQXlDLEVBQUUsQ0FBQztRQUc5RCxVQUFLLEdBQWlCLEVBQUUsQ0FBQztRQUdoQyxJQUFJLENBQUMsTUFBTSxHQUFHLE1BQU0sQ0FBQztRQUNyQiw2REFBNkQ7UUFDN0Qsc0RBQXNEO1FBQ3RELE1BQU0sQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDLE9BQU8sQ0FBQyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsSUFBSSxFQUFFLFlBQVksQ0FBQyxFQUFFLEVBQUU7WUFDOUQsSUFBSSxDQUFDLFVBQVUsQ0FBQyxJQUFJLENBQUMsR0FBRyxJQUFJLG1DQUFvQixDQUFDO2dCQUMvQyxVQUFVLEVBQUUsRUFBRSxDQUFDLElBQUksQ0FBQyxFQUFFLFlBQVksRUFBRTtnQkFDcEMsR0FBRyxNQUFNO2FBQ1YsQ0FBQyxDQUFDO1FBQ0wsQ0FBQyxDQUFDLENBQUM7SUFDTCxDQUFDO0lBRUQsS0FBSyxDQUFDLFNBQVM7UUFDYixNQUFNLE1BQU0sR0FBNEMsRUFBRSxDQUFDO1FBRTNELE1BQU0sT0FBTyxDQUFDLEdBQUcsQ0FDZixNQUFNLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsQ0FBQyxHQUFHLENBQUMsS0FBSyxFQUFFLENBQUMsSUFBSSxFQUFFLE1BQU0sQ0FBQyxFQUFFLEVBQUU7WUFDM0QsSUFBSSxDQUFDO2dCQUNILE1BQU0sS0FBSyxHQUFHLENBQUMsTUFBTSxNQUFNLENBQUMsUUFBUSxFQUFFLENBQUMsSUFBSSxFQUFFLENBQUM7Z0JBQzlDLE1BQU0sYUFBYSxHQUFHLEtBQUssQ0FBQyxHQUFHLENBQzdCLElBQUksQ0FBQyxFQUFFLENBQ0wsSUFBSSxxQkFBVSxDQUFDO29CQUNiLElBQUk7b0JBQ0osUUFBUSxFQUFFLElBQUk7b0JBQ2QsVUFBVSxFQUFFLFlBQVk7aUJBQ3pCLENBQUMsQ0FDTCxDQUFDO2dCQUNGLElBQUksQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLEdBQUcsYUFBYSxDQUFDLENBQUM7WUFDcEMsQ0FBQztZQUFDLE9BQU8sS0FBSyxFQUFFLENBQUM7Z0JBQ2YsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDLE9BQU8sRUFBRSwyQkFBMkIsSUFBSSxFQUFFLEVBQUUsS0FBYyxDQUFDLENBQUM7Z0JBQzFFLE1BQU0sQ0FBQyxJQUFJLENBQUMsRUFBRSxNQUFNLEVBQUUsSUFBSSxFQUFFLEtBQUssRUFBRSxLQUFjLEVBQUUsQ0FBQyxDQUFDO1lBQ3ZELENBQUM7UUFDSCxDQUFDLENBQUMsQ0FDSCxDQUFDO1FBRUYsc0RBQXNEO1FBQ3RELElBQUksTUFBTSxDQUFDLE1BQU0sR0FBRyxDQUFDLEVBQUUsQ0FBQztZQUN0QixNQUFNLFlBQVksR0FBRyxNQUFNLENBQUMsR0FBRyxDQUFDLENBQUMsQ0FBQyxFQUFFLENBQUMsR0FBRyxDQUFDLENBQUMsTUFBTSxLQUFLLENBQUMsQ0FBQyxLQUFLLENBQUMsT0FBTyxFQUFFLENBQUMsQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLENBQUM7WUFDbkYsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUNYLE9BQU8sRUFDUCw2QkFBNkIsTUFBTSxDQUFDLE1BQU0sSUFBSSxNQUFNLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsQ0FBQyxNQUFNLEdBQUc7Z0JBQ2xGLGtCQUFrQixZQUFZLEVBQUUsQ0FDbkMsQ0FBQztRQUNKLENBQUM7UUFFRCxPQUFPLElBQUksQ0FBQyxLQUFLLENBQUM7SUFDcEIsQ0FBQztJQUVELEtBQUssQ0FBQyxlQUFlO1FBQ25CLElBQUksQ0FBQztZQUNILE1BQU0sT0FBTyxDQUFDLEdBQUcsQ0FDZixNQUFNLENBQUMsTUFBTSxDQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsQ0FBQyxHQUFHLENBQUMsTUFBTSxDQUFDLEVBQUUsQ0FBQyxNQUFNLENBQUMscUJBQXFCLEVBQUUsQ0FBQyxDQUM3RSxDQUFDO1lBRUYsT0FBTyxJQUFJLENBQUM7UUFDZCxDQUFDO1FBQUMsT0FBTyxLQUFLLEVBQUUsQ0FBQztZQUNmLE1BQU0sSUFBSSwyQkFBa0IsQ0FBRSxLQUFlLENBQUMsT0FBTyxDQUFDLENBQUM7UUFDekQsQ0FBQztnQkFBUyxDQUFDO1lBQ1QsSUFBSSxDQUFDO2dCQUNILE1BQU0sSUFBSSxDQUFDLGdCQUFnQixFQUFFLENBQUM7WUFDaEMsQ0FBQztZQUFDLE9BQU8sWUFBWSxFQUFFLENBQUM7Z0JBQ3RCLDRFQUE0RTtnQkFDNUUsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDLE9BQU8sRUFBRSxzQ0FBc0MsRUFBRSxZQUFxQixDQUFDLENBQUM7WUFDeEYsQ0FBQztRQUNILENBQUM7SUFDSCxDQUFDO0lBRUQsS0FBSyxDQUFDLGdCQUFnQjtRQUNwQixNQUFNLE9BQU8sR0FBRyxNQUFNLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsQ0FBQztRQUNoRCxNQUFNLE9BQU8sR0FBRyxNQUFNLE9BQU8sQ0FBQyxVQUFVLENBQUMsT0FBTyxDQUFDLEdBQUcsQ0FBQyxDQUFDLENBQUMsRUFBRSxNQUFNLENBQUMsRUFBRSxFQUFFLENBQUMsTUFBTSxDQUFDLEtBQUssRUFBRSxDQUFDLENBQUMsQ0FBQztRQUV0RixxRUFBcUU7UUFDckUsTUFBTSxRQUFRLEdBQUcsT0FBTzthQUNyQixHQUFHLENBQUMsQ0FBQyxNQUFNLEVBQUUsS0FBSyxFQUFFLEVBQUUsQ0FBQyxDQUFDLEVBQUUsTUFBTSxFQUFFLElBQUksRUFBRSxPQUFPLENBQUMsS0FBSyxDQUFDLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQyxDQUFDO2FBQzdELE1BQU0sQ0FBQyxDQUFDLEVBQUUsTUFBTSxFQUFFLEVBQUUsRUFBRSxDQUFDLE1BQU0sQ0FBQyxNQUFNLEtBQUssVUFBVSxDQUFDLENBQUM7UUFFeEQsSUFBSSxRQUFRLENBQUMsTUFBTSxHQUFHLENBQUMsRUFBRSxDQUFDO1lBQ3hCLFFBQVEsQ0FBQyxPQUFPLENBQUMsQ0FBQyxFQUFFLElBQUksRUFBRSxNQUFNLEVBQUUsRUFBRSxFQUFFO2dCQUNwQyxJQUFJLENBQUMsTUFBTSxFQUFFLENBQ1gsT0FBTyxFQUNQLHNDQUFzQyxJQUFJLEVBQUUsRUFDM0MsTUFBZ0MsQ0FBQyxNQUFNLENBQ3pDLENBQUM7WUFDSixDQUFDLENBQUMsQ0FBQztZQUNILElBQUksQ0FBQyxNQUFNLEVBQUUsQ0FDWCxPQUFPLEVBQ1AsbUJBQW1CLFFBQVEsQ0FBQyxNQUFNLElBQUksT0FBTyxDQUFDLE1BQU0sb0JBQW9CO2dCQUN0RSxvQ0FBb0MsQ0FDdkMsQ0FBQztRQUNKLENBQUM7SUFDSCxDQUFDO0NBQ0Y7QUFqR0QsNEJBaUdDIn0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibWNwLWNsaWVudC5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9tY3AtY2xpZW50LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7O0FBRUEsMERBQStEO0FBRS9ELDJDQUFvRDtBQUNwRCw0RkFBaUU7QUFNakUsTUFBcUIsU0FBUztJQU01QixZQUFZLE1BQXdCLEVBQUUsTUFBZTtRQUxwQyxlQUFVLEdBQXlDLEVBQUUsQ0FBQztRQUc5RCxVQUFLLEdBQTBCLEVBQUUsQ0FBQztRQUd6QyxJQUFJLENBQUMsTUFBTSxHQUFHLE1BQU0sQ0FBQztRQUNyQiw2REFBNkQ7UUFDN0Qsc0RBQXNEO1FBQ3RELE1BQU0sQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDLE9BQU8sQ0FBQyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsSUFBSSxFQUFFLFlBQVksQ0FBQyxFQUFFLEVBQUU7WUFDOUQsSUFBSSxDQUFDLFVBQVUsQ0FBQyxJQUFJLENBQUMsR0FBRyxJQUFJLG1DQUFvQixDQUFDO2dCQUMvQyxVQUFVLEVBQUUsRUFBRSxDQUFDLElBQUksQ0FBQyxFQUFFLFlBQVksRUFBRTtnQkFDcEMsR0FBRyxNQUFNO2FBQ1YsQ0FBQyxDQUFDO1FBQ0wsQ0FBQyxDQUFDLENBQUM7SUFDTCxDQUFDO0lBRUQsS0FBSyxDQUFDLFNBQVM7UUFDYixNQUFNLE1BQU0sR0FBNEMsRUFBRSxDQUFDO1FBRTNELE1BQU0sT0FBTyxDQUFDLEdBQUcsQ0FDZixNQUFNLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxVQUFVLENBQUMsQ0FBQyxHQUFHLENBQUMsS0FBSyxFQUFFLENBQUMsSUFBSSxFQUFFLE1BQU0sQ0FBQyxFQUFFLEVBQUU7WUFDM0QsSUFBSSxDQUFDO2dCQUNILE1BQU0sS0FBSyxHQUFHLENBQUMsTUFBTSxNQUFNLENBQUMsUUFBUSxFQUFFLENBQUMsSUFBSSxFQUFFLENBQUM7Z0JBQzlDLE1BQU0sYUFBYSxHQUFHLEtBQUssQ0FBQyxHQUFHLENBQzdCLElBQUksQ0FBQyxFQUFFLENBQUMsSUFBSSxnQ0FBbUIsQ0FBQyxFQUFFLElBQUksRUFBRSxRQUFRLEVBQUUsSUFBSSxFQUFFLENBQUMsQ0FDMUQsQ0FBQztnQkFDRixJQUFJLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxHQUFHLGFBQWEsQ0FBQyxDQUFDO1lBQ3BDLENBQUM7WUFBQyxPQUFPLEtBQUssRUFBRSxDQUFDO2dCQUNmLElBQUksQ0FBQyxNQUFNLEVBQUUsQ0FBQyxPQUFPLEVBQUUsMkJBQTJCLElBQUksRUFBRSxFQUFFLEtBQWMsQ0FBQyxDQUFDO2dCQUMxRSxNQUFNLENBQUMsSUFBSSxDQUFDLEVBQUUsTUFBTSxFQUFFLElBQUksRUFBRSxLQUFLLEVBQUUsS0FBYyxFQUFFLENBQUMsQ0FBQztZQUN2RCxDQUFDO1FBQ0gsQ0FBQyxDQUFDLENBQ0gsQ0FBQztRQUVGLHNEQUFzRDtRQUN0RCxJQUFJLE1BQU0sQ0FBQyxNQUFNLEdBQUcsQ0FBQyxFQUFFLENBQUM7WUFDdEIsTUFBTSxZQUFZLEdBQUcsTUFBTSxDQUFDLEdBQUcsQ0FBQyxDQUFDLENBQUMsRUFBRSxDQUFDLEdBQUcsQ0FBQyxDQUFDLE1BQU0sS0FBSyxDQUFDLENBQUMsS0FBSyxDQUFDLE9BQU8sRUFBRSxDQUFDLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxDQUFDO1lBQ25GLElBQUksQ0FBQyxNQUFNLEVBQUUsQ0FDWCxPQUFPLEVBQ1AsNkJBQTZCLE1BQU0sQ0FBQyxNQUFNLElBQUksTUFBTSxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsVUFBVSxDQUFDLENBQUMsTUFBTSxHQUFHO2dCQUNsRixrQkFBa0IsWUFBWSxFQUFFLENBQ25DLENBQUM7UUFDSixDQUFDO1FBRUQsT0FBTyxJQUFJLENBQUMsS0FBSyxDQUFDO0lBQ3BCLENBQUM7SUFFRCxLQUFLLENBQUMsZUFBZTtRQUNuQixJQUFJLENBQUM7WUFDSCxNQUFNLE9BQU8sQ0FBQyxHQUFHLENBQ2YsTUFBTSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsVUFBVSxDQUFDLENBQUMsR0FBRyxDQUFDLE1BQU0sQ0FBQyxFQUFFLENBQUMsTUFBTSxDQUFDLHFCQUFxQixFQUFFLENBQUMsQ0FDN0UsQ0FBQztZQUVGLE9BQU8sSUFBSSxDQUFDO1FBQ2QsQ0FBQztRQUFDLE9BQU8sS0FBSyxFQUFFLENBQUM7WUFDZixNQUFNLElBQUksMkJBQWtCLENBQUUsS0FBZSxDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQ3pELENBQUM7Z0JBQVMsQ0FBQztZQUNULElBQUksQ0FBQztnQkFDSCxNQUFNLElBQUksQ0FBQyxnQkFBZ0IsRUFBRSxDQUFDO1lBQ2hDLENBQUM7WUFBQyxPQUFPLFlBQVksRUFBRSxDQUFDO2dCQUN0Qiw0RUFBNEU7Z0JBQzVFLElBQUksQ0FBQyxNQUFNLEVBQUUsQ0FBQyxPQUFPLEVBQUUsc0NBQXNDLEVBQUUsWUFBcUIsQ0FBQyxDQUFDO1lBQ3hGLENBQUM7UUFDSCxDQUFDO0lBQ0gsQ0FBQztJQUVELEtBQUssQ0FBQyxnQkFBZ0I7UUFDcEIsTUFBTSxPQUFPLEdBQUcsTUFBTSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsVUFBVSxDQUFDLENBQUM7UUFDaEQsTUFBTSxPQUFPLEdBQUcsTUFBTSxPQUFPLENBQUMsVUFBVSxDQUFDLE9BQU8sQ0FBQyxHQUFHLENBQUMsQ0FBQyxDQUFDLEVBQUUsTUFBTSxDQUFDLEVBQUUsRUFBRSxDQUFDLE1BQU0sQ0FBQyxLQUFLLEVBQUUsQ0FBQyxDQUFDLENBQUM7UUFFdEYscUVBQXFFO1FBQ3JFLE1BQU0sUUFBUSxHQUFHLE9BQU87YUFDckIsR0FBRyxDQUFDLENBQUMsTUFBTSxFQUFFLEtBQUssRUFBRSxFQUFFLENBQUMsQ0FBQyxFQUFFLE1BQU0sRUFBRSxJQUFJLEVBQUUsT0FBTyxDQUFDLEtBQUssQ0FBQyxDQUFDLENBQUMsQ0FBQyxFQUFFLENBQUMsQ0FBQzthQUM3RCxNQUFNLENBQUMsQ0FBQyxFQUFFLE1BQU0sRUFBRSxFQUFFLEVBQUUsQ0FBQyxNQUFNLENBQUMsTUFBTSxLQUFLLFVBQVUsQ0FBQyxDQUFDO1FBRXhELElBQUksUUFBUSxDQUFDLE1BQU0sR0FBRyxDQUFDLEVBQUUsQ0FBQztZQUN4QixRQUFRLENBQUMsT0FBTyxDQUFDLENBQUMsRUFBRSxJQUFJLEVBQUUsTUFBTSxFQUFFLEVBQUUsRUFBRTtnQkFDcEMsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUNYLE9BQU8sRUFDUCxzQ0FBc0MsSUFBSSxFQUFFLEVBQzNDLE1BQWdDLENBQUMsTUFBTSxDQUN6QyxDQUFDO1lBQ0osQ0FBQyxDQUFDLENBQUM7WUFDSCxJQUFJLENBQUMsTUFBTSxFQUFFLENBQ1gsT0FBTyxFQUNQLG1CQUFtQixRQUFRLENBQUMsTUFBTSxJQUFJLE9BQU8sQ0FBQyxNQUFNLG9CQUFvQjtnQkFDdEUsb0NBQW9DLENBQ3ZDLENBQUM7UUFDSixDQUFDO0lBQ0gsQ0FBQztDQUNGO0FBNUZELDRCQTRGQyJ9

package/dist/provider-dispatcher.d.ts

@@ -1,19 +1,41 @@
 import type { RemoteTools } from './remote-tools';
-import type { ClientOptions } from 'openai';
-import type { ChatCompletionCreateParamsNonStreaming } from 'openai/resources/chat/completions';
-export type OpenAiConfiguration = ClientOptions & {
+import type { ChatOpenAIFields, OpenAIChatModelId } from '@langchain/openai';
+import type OpenAI from 'openai';
+export declare function isModelSupportingTools(model: string): boolean;
+/**
+ * Base configuration common to all AI providers.
+ */
+export type BaseAiConfiguration = {
+    name: string;
+    provider: AiProvider;
+    model: string;
+    apiKey?: string;
+};
+/**
+ * OpenAI-specific configuration.
+ * Extends base with all ChatOpenAI options (temperature, maxTokens, configuration, etc.)
+ */
+export type OpenAiConfiguration = BaseAiConfiguration & Omit<ChatOpenAIFields, 'model' | 'apiKey'> & {
     provider: 'openai';
-    model: ChatCompletionCreateParamsNonStreaming['model'];
+    model: OpenAIChatModelId | (string & NonNullable<unknown>);
 };
+export type AiProvider = 'openai';
 export type AiConfiguration = OpenAiConfiguration;
-export type AiProvider = AiConfiguration['provider'];
-export type OpenAIBody = Pick<ChatCompletionCreateParamsNonStreaming, 'tools' | 'messages' | 'tool_choice'>;
-export type DispatchBody = OpenAIBody;
+export type ChatCompletionResponse = OpenAI.Chat.Completions.ChatCompletion;
+export type ChatCompletionMessage = OpenAI.Chat.Completions.ChatCompletionMessageParam;
+export type ChatCompletionTool = OpenAI.Chat.Completions.ChatCompletionTool;
+export type ChatCompletionToolChoice = OpenAI.Chat.Completions.ChatCompletionToolChoiceOption;
+export type DispatchBody = {
+    messages: ChatCompletionMessage[];
+    tools?: ChatCompletionTool[];
+    tool_choice?: ChatCompletionToolChoice;
+    parallel_tool_calls?: boolean;
+};
 export declare class ProviderDispatcher {
-    private readonly aiClient;
+    private readonly chatModel;
     private readonly remoteTools;
     constructor(configuration: AiConfiguration | null, remoteTools: RemoteTools);
-    dispatch(body: DispatchBody): Promise<unknown>;
-    private enhanceOpenAIRemoteTools;
+    dispatch(body: DispatchBody): Promise<ChatCompletionResponse>;
+    private enrichToolDefinitions;
 }
 //# sourceMappingURL=provider-dispatcher.d.ts.map

package/dist/provider-dispatcher.js

@@ -1,56 +1,96 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ProviderDispatcher = void 0;
+exports.isModelSupportingTools = isModelSupportingTools;
 const function_calling_1 = require("@langchain/core/utils/function_calling");
-const openai_1 = __importDefault(require("openai"));
-const errors_1 = require("./errors");
+const openai_1 = require("@langchain/openai");
+const errors_1 = require("./types/errors");
+/**
+ * OpenAI model prefixes that do NOT support function calling (tools).
+ * Unknown models are allowed.
+ * @see https://platform.openai.com/docs/guides/function-calling
+ */
+const OPENAI_MODELS_WITHOUT_TOOLS_SUPPORT = [
+    'gpt-4',
+    'gpt-3.5-turbo',
+    'gpt-3.5',
+    'text-davinci',
+    'davinci',
+    'curie',
+    'babbage',
+    'ada',
+];
+/**
+ * Exceptions to the unsupported list - these models DO support tools
+ * even though they start with an unsupported prefix.
+ */
+const OPENAI_MODELS_EXCEPTIONS = ['gpt-4-turbo', 'gpt-4o', 'gpt-4.1'];
+function isModelSupportingTools(model) {
+    const isException = OPENAI_MODELS_EXCEPTIONS.some(exception => model === exception || model.startsWith(`${exception}-`));
+    if (isException)
+        return true;
+    const isKnownUnsupported = OPENAI_MODELS_WITHOUT_TOOLS_SUPPORT.some(unsupported => model === unsupported || model.startsWith(`${unsupported}-`));
+    return !isKnownUnsupported;
+}
 class ProviderDispatcher {
     constructor(configuration, remoteTools) {
-        this.aiClient = null;
+        this.chatModel = null;
         this.remoteTools = remoteTools;
-        if (configuration?.provider === 'openai' && configuration.apiKey) {
-            const { provider, model, ...clientOptions } = configuration;
-            this.aiClient = {
-                client: new openai_1.default(clientOptions),
-                model,
-            };
+        if (configuration?.provider === 'openai') {
+            const { provider, name, ...chatOpenAIOptions } = configuration;
+            this.chatModel = new openai_1.ChatOpenAI({
+                ...chatOpenAIOptions,
+                __includeRawResponse: true,
+            });
         }
     }
     async dispatch(body) {
-        if (!this.aiClient) {
+        if (!this.chatModel) {
             throw new errors_1.AINotConfiguredError();
         }
-        // tools, messages and tool_choice must be extracted from the body and passed as options
-        // because we don't want to let users to pass any other option
-        const { tools, messages, tool_choice: toolChoice } = body;
-        const options = {
-            model: this.aiClient.model,
-            // Add the remote tools to the tools to be used by the AI
-            tools: this.enhanceOpenAIRemoteTools(tools),
-            messages,
-            tool_choice: toolChoice,
-        };
+        const { tools, messages, tool_choice: toolChoice, parallel_tool_calls: parallelToolCalls, } = body;
+        const enrichedTools = this.enrichToolDefinitions(tools);
+        const model = enrichedTools?.length
+            ? this.chatModel.bindTools(enrichedTools, {
+                tool_choice: toolChoice,
+                parallel_tool_calls: parallelToolCalls,
+            })
+            : this.chatModel;
         try {
-            return await this.aiClient.client.chat.completions.create(options);
+            const response = await model.invoke(messages);
+            // eslint-disable-next-line no-underscore-dangle
+            const rawResponse = response.additional_kwargs.__raw_response;
+            if (!rawResponse) {
+                throw new errors_1.OpenAIUnprocessableError('OpenAI response missing raw response data. This may indicate an API change.');
+            }
+            return rawResponse;
         }
         catch (error) {
-            throw new errors_1.OpenAIUnprocessableError(`Error while calling OpenAI: ${error.message}`);
+            if (error instanceof errors_1.OpenAIUnprocessableError)
+                throw error;
+            const err = error;
+            if (err.status === 429) {
+                throw new errors_1.OpenAIUnprocessableError(`Rate limit exceeded: ${err.message}`);
+            }
+            if (err.status === 401) {
+                throw new errors_1.OpenAIUnprocessableError(`Authentication failed: ${err.message}`);
+            }
+            throw new errors_1.OpenAIUnprocessableError(`Error while calling OpenAI: ${err.message}`);
         }
     }
-    enhanceOpenAIRemoteTools(tools) {
+    enrichToolDefinitions(tools) {
         if (!tools || !Array.isArray(tools))
             return tools;
-        const remoteToolFunctions = this.remoteTools.tools.map(extendedTools => (0, function_calling_1.convertToOpenAIFunction)(extendedTools.base));
+        const remoteToolSchemas = this.remoteTools.tools.map(remoteTool => (0, function_calling_1.convertToOpenAIFunction)(remoteTool.base));
         return tools.map(tool => {
-            const remoteFunction = remoteToolFunctions.find(functionDefinition => functionDefinition.name === tool.function.name);
-            if (remoteFunction)
-                return { ...tool, function: remoteFunction };
+            if (tool.type !== 'function')
+                return tool;
+            const remoteSchema = remoteToolSchemas.find(schema => schema.name === tool.function.name);
+            if (remoteSchema)
+                return { ...tool, function: remoteSchema };
             return tool;
         });
     }
 }
 exports.ProviderDispatcher = ProviderDispatcher;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicHJvdmlkZXItZGlzcGF0Y2hlci5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9wcm92aWRlci1kaXNwYXRjaGVyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7OztBQUlBLDZFQUFpRjtBQUNqRixvREFBNEI7QUFFNUIscUNBQTBFO0FBb0IxRSxNQUFhLGtCQUFrQjtJQUs3QixZQUFZLGFBQXFDLEVBQUUsV0FBd0I7UUFKMUQsYUFBUSxHQUF3QixJQUFJLENBQUM7UUFLcEQsSUFBSSxDQUFDLFdBQVcsR0FBRyxXQUFXLENBQUM7UUFFL0IsSUFBSSxhQUFhLEVBQUUsUUFBUSxLQUFLLFFBQVEsSUFBSSxhQUFhLENBQUMsTUFBTSxFQUFFLENBQUM7WUFDakUsTUFBTSxFQUFFLFFBQVEsRUFBRSxLQUFLLEVBQUUsR0FBRyxhQUFhLEVBQUUsR0FBRyxhQUFhLENBQUM7WUFDNUQsSUFBSSxDQUFDLFFBQVEsR0FBRztnQkFDZCxNQUFNLEVBQUUsSUFBSSxnQkFBTSxDQUFDLGFBQWEsQ0FBQztnQkFDakMsS0FBSzthQUNOLENBQUM7UUFDSixDQUFDO0lBQ0gsQ0FBQztJQUVELEtBQUssQ0FBQyxRQUFRLENBQUMsSUFBa0I7UUFDL0IsSUFBSSxDQUFDLElBQUksQ0FBQyxRQUFRLEVBQUUsQ0FBQztZQUNuQixNQUFNLElBQUksNkJBQW9CLEVBQUUsQ0FBQztRQUNuQyxDQUFDO1FBRUQsd0ZBQXdGO1FBQ3hGLDhEQUE4RDtRQUM5RCxNQUFNLEVBQUUsS0FBSyxFQUFFLFFBQVEsRUFBRSxXQUFXLEVBQUUsVUFBVSxFQUFFLEdBQUcsSUFBSSxDQUFDO1FBRTFELE1BQU0sT0FBTyxHQUFHO1lBQ2QsS0FBSyxFQUFFLElBQUksQ0FBQyxRQUFRLENBQUMsS0FBSztZQUMxQix5REFBeUQ7WUFDekQsS0FBSyxFQUFFLElBQUksQ0FBQyx3QkFBd0IsQ0FBQyxLQUFLLENBQUM7WUFDM0MsUUFBUTtZQUNSLFdBQVcsRUFBRSxVQUFVO1NBQ2tCLENBQUM7UUFFNUMsSUFBSSxDQUFDO1lBQ0gsT0FBTyxNQUFNLElBQUksQ0FBQyxRQUFRLENBQUMsTUFBTSxDQUFDLElBQUksQ0FBQyxXQUFXLENBQUMsTUFBTSxDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQ3JFLENBQUM7UUFBQyxPQUFPLEtBQUssRUFBRSxDQUFDO1lBQ2YsTUFBTSxJQUFJLGlDQUF3QixDQUFDLCtCQUFnQyxLQUFlLENBQUMsT0FBTyxFQUFFLENBQUMsQ0FBQztRQUNoRyxDQUFDO0lBQ0gsQ0FBQztJQUVPLHdCQUF3QixDQUFDLEtBQXVEO1FBQ3RGLElBQUksQ0FBQyxLQUFLLElBQUksQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLEtBQUssQ0FBQztZQUFFLE9BQU8sS0FBSyxDQUFDO1FBRWxELE1BQU0sbUJBQW1CLEdBQUcsSUFBSSxDQUFDLFdBQVcsQ0FBQyxLQUFLLENBQUMsR0FBRyxDQUFDLGFBQWEsQ0FBQyxFQUFFLENBQ3JFLElBQUEsMENBQXVCLEVBQUMsYUFBYSxDQUFDLElBQUksQ0FBQyxDQUM1QyxDQUFDO1FBRUYsT0FBTyxLQUFLLENBQUMsR0FBRyxDQUFDLElBQUksQ0FBQyxFQUFFO1lBQ3RCLE1BQU0sY0FBYyxHQUFHLG1CQUFtQixDQUFDLElBQUksQ0FDN0Msa0JBQWtCLENBQUMsRUFBRSxDQUFDLGtCQUFrQixDQUFDLElBQUksS0FBSyxJQUFJLENBQUMsUUFBUSxDQUFDLElBQUksQ0FDckUsQ0FBQztZQUNGLElBQUksY0FBYztnQkFBRSxPQUFPLEVBQUUsR0FBRyxJQUFJLEVBQUUsUUFBUSxFQUFFLGNBQWMsRUFBRSxDQUFDO1lBRWpFLE9BQU8sSUFBSSxDQUFDO1FBQ2QsQ0FBQyxDQUFDLENBQUM7SUFDTCxDQUFDO0NBQ0Y7QUF6REQsZ0RBeURDIn0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicHJvdmlkZXItZGlzcGF0Y2hlci5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9wcm92aWRlci1kaXNwYXRjaGVyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7OztBQWdDQSx3REFXQztBQXRDRCw2RUFBaUY7QUFDakYsOENBQStDO0FBRS9DLDJDQUFnRjtBQUVoRjs7OztHQUlHO0FBQ0gsTUFBTSxtQ0FBbUMsR0FBRztJQUMxQyxPQUFPO0lBQ1AsZUFBZTtJQUNmLFNBQVM7SUFDVCxjQUFjO0lBQ2QsU0FBUztJQUNULE9BQU87SUFDUCxTQUFTO0lBQ1QsS0FBSztDQUNOLENBQUM7QUFFRjs7O0dBR0c7QUFDSCxNQUFNLHdCQUF3QixHQUFHLENBQUMsYUFBYSxFQUFFLFFBQVEsRUFBRSxTQUFTLENBQUMsQ0FBQztBQUV0RSxTQUFnQixzQkFBc0IsQ0FBQyxLQUFhO0lBQ2xELE1BQU0sV0FBVyxHQUFHLHdCQUF3QixDQUFDLElBQUksQ0FDL0MsU0FBUyxDQUFDLEVBQUUsQ0FBQyxLQUFLLEtBQUssU0FBUyxJQUFJLEtBQUssQ0FBQyxVQUFVLENBQUMsR0FBRyxTQUFTLEdBQUcsQ0FBQyxDQUN0RSxDQUFDO0lBQ0YsSUFBSSxXQUFXO1FBQUUsT0FBTyxJQUFJLENBQUM7SUFFN0IsTUFBTSxrQkFBa0IsR0FBRyxtQ0FBbUMsQ0FBQyxJQUFJLENBQ2pFLFdBQVcsQ0FBQyxFQUFFLENBQUMsS0FBSyxLQUFLLFdBQVcsSUFBSSxLQUFLLENBQUMsVUFBVSxDQUFDLEdBQUcsV0FBVyxHQUFHLENBQUMsQ0FDNUUsQ0FBQztJQUVGLE9BQU8sQ0FBQyxrQkFBa0IsQ0FBQztBQUM3QixDQUFDO0FBdUNELE1BQWEsa0JBQWtCO0lBSzdCLFlBQVksYUFBcUMsRUFBRSxXQUF3QjtRQUoxRCxjQUFTLEdBQXNCLElBQUksQ0FBQztRQUtuRCxJQUFJLENBQUMsV0FBVyxHQUFHLFdBQVcsQ0FBQztRQUUvQixJQUFJLGFBQWEsRUFBRSxRQUFRLEtBQUssUUFBUSxFQUFFLENBQUM7WUFDekMsTUFBTSxFQUFFLFFBQVEsRUFBRSxJQUFJLEVBQUUsR0FBRyxpQkFBaUIsRUFBRSxHQUFHLGFBQWEsQ0FBQztZQUMvRCxJQUFJLENBQUMsU0FBUyxHQUFHLElBQUksbUJBQVUsQ0FBQztnQkFDOUIsR0FBRyxpQkFBaUI7Z0JBQ3BCLG9CQUFvQixFQUFFLElBQUk7YUFDM0IsQ0FBQyxDQUFDO1FBQ0wsQ0FBQztJQUNILENBQUM7SUFFRCxLQUFLLENBQUMsUUFBUSxDQUFDLElBQWtCO1FBQy9CLElBQUksQ0FBQyxJQUFJLENBQUMsU0FBUyxFQUFFLENBQUM7WUFDcEIsTUFBTSxJQUFJLDZCQUFvQixFQUFFLENBQUM7UUFDbkMsQ0FBQztRQUVELE1BQU0sRUFDSixLQUFLLEVBQ0wsUUFBUSxFQUNSLFdBQVcsRUFBRSxVQUFVLEVBQ3ZCLG1CQUFtQixFQUFFLGlCQUFpQixHQUN2QyxHQUFHLElBQUksQ0FBQztRQUVULE1BQU0sYUFBYSxHQUFHLElBQUksQ0FBQyxxQkFBcUIsQ0FBQyxLQUFLLENBQUMsQ0FBQztRQUN4RCxNQUFNLEtBQUssR0FBRyxhQUFhLEVBQUUsTUFBTTtZQUNqQyxDQUFDLENBQUMsSUFBSSxDQUFDLFNBQVMsQ0FBQyxTQUFTLENBQUMsYUFBYSxFQUFFO2dCQUN0QyxXQUFXLEVBQUUsVUFBVTtnQkFDdkIsbUJBQW1CLEVBQUUsaUJBQWlCO2FBQ3ZDLENBQUM7WUFDSixDQUFDLENBQUMsSUFBSSxDQUFDLFNBQVMsQ0FBQztRQUVuQixJQUFJLENBQUM7WUFDSCxNQUFNLFFBQVEsR0FBRyxNQUFNLEtBQUssQ0FBQyxNQUFNLENBQUMsUUFBNkIsQ0FBQyxDQUFDO1lBRW5FLGdEQUFnRDtZQUNoRCxNQUFNLFdBQVcsR0FBRyxRQUFRLENBQUMsaUJBQWlCLENBQUMsY0FBd0MsQ0FBQztZQUV4RixJQUFJLENBQUMsV0FBVyxFQUFFLENBQUM7Z0JBQ2pCLE1BQU0sSUFBSSxpQ0FBd0IsQ0FDaEMsNkVBQTZFLENBQzlFLENBQUM7WUFDSixDQUFDO1lBRUQsT0FBTyxXQUFXLENBQUM7UUFDckIsQ0FBQztRQUFDLE9BQU8sS0FBSyxFQUFFLENBQUM7WUFDZixJQUFJLEtBQUssWUFBWSxpQ0FBd0I7Z0JBQUUsTUFBTSxLQUFLLENBQUM7WUFFM0QsTUFBTSxHQUFHLEdBQUcsS0FBb0MsQ0FBQztZQUVqRCxJQUFJLEdBQUcsQ0FBQyxNQUFNLEtBQUssR0FBRyxFQUFFLENBQUM7Z0JBQ3ZCLE1BQU0sSUFBSSxpQ0FBd0IsQ0FBQyx3QkFBd0IsR0FBRyxDQUFDLE9BQU8sRUFBRSxDQUFDLENBQUM7WUFDNUUsQ0FBQztZQUVELElBQUksR0FBRyxDQUFDLE1BQU0sS0FBSyxHQUFHLEVBQUUsQ0FBQztnQkFDdkIsTUFBTSxJQUFJLGlDQUF3QixDQUFDLDBCQUEwQixHQUFHLENBQUMsT0FBTyxFQUFFLENBQUMsQ0FBQztZQUM5RSxDQUFDO1lBRUQsTUFBTSxJQUFJLGlDQUF3QixDQUFDLCtCQUErQixHQUFHLENBQUMsT0FBTyxFQUFFLENBQUMsQ0FBQztRQUNuRixDQUFDO0lBQ0gsQ0FBQztJQUVPLHFCQUFxQixDQUFDLEtBQTRCO1FBQ3hELElBQUksQ0FBQyxLQUFLLElBQUksQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLEtBQUssQ0FBQztZQUFFLE9BQU8sS0FBSyxDQUFDO1FBRWxELE1BQU0saUJBQWlCLEdBQUcsSUFBSSxDQUFDLFdBQVcsQ0FBQyxLQUFLLENBQUMsR0FBRyxDQUFDLFVBQVUsQ0FBQyxFQUFFLENBQ2hFLElBQUEsMENBQXVCLEVBQUMsVUFBVSxDQUFDLElBQUksQ0FBQyxDQUN6QyxDQUFDO1FBRUYsT0FBTyxLQUFLLENBQUMsR0FBRyxDQUFDLElBQUksQ0FBQyxFQUFFO1lBQ3RCLElBQUksSUFBSSxDQUFDLElBQUksS0FBSyxVQUFVO2dCQUFFLE9BQU8sSUFBSSxDQUFDO1lBRTFDLE1BQU0sWUFBWSxHQUFHLGlCQUFpQixDQUFDLElBQUksQ0FBQyxNQUFNLENBQUMsRUFBRSxDQUFDLE1BQU0sQ0FBQyxJQUFJLEtBQUssSUFBSSxDQUFDLFFBQVEsQ0FBQyxJQUFJLENBQUMsQ0FBQztZQUMxRixJQUFJLFlBQVk7Z0JBQUUsT0FBTyxFQUFFLEdBQUcsSUFBSSxFQUFFLFFBQVEsRUFBRSxZQUFZLEVBQUUsQ0FBQztZQUU3RCxPQUFPLElBQUksQ0FBQztRQUNkLENBQUMsQ0FBQyxDQUFDO0lBQ0wsQ0FBQztDQUNGO0FBbkZELGdEQW1GQyJ9

package/dist/remote-tools.d.ts

@@ -1,6 +1,6 @@
+import type RemoteTool from './types/remote-tool';
 import type { ResponseFormat } from '@langchain/core/tools';
 import type { ChatCompletionCreateParamsNonStreaming } from 'openai/resources/chat/completions';
-import RemoteTool from './remote-tool';
 export type Messages = ChatCompletionCreateParamsNonStreaming['messages'];
 export type RemoteToolsApiKeys = {
     ['AI_REMOTE_TOOL_BRAVE_SEARCH_API_KEY']: string;
@@ -15,7 +15,7 @@ export declare class RemoteTools {
         responseFormat: ResponseFormat;
         schema: import("@langchain/core/utils/json_schema").JSONSchema;
         sourceId: string;
-        sourceType: import("./remote-tool").SourceType;
+        sourceType: string;
     }[];
     invokeTool(toolName: string, messages: ChatCompletionCreateParamsNonStreaming['messages']): Promise<unknown>;
 }

package/dist/remote-tools.js

@@ -6,17 +6,16 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.RemoteTools = void 0;
 const brave_search_1 = require("@langchain/community/tools/brave_search");
 const json_schema_1 = require("@langchain/core/utils/json_schema");
-const errors_1 = require("./errors");
-const remote_tool_1 = __importDefault(require("./remote-tool"));
+const errors_1 = require("./types/errors");
+const server_remote_tool_1 = __importDefault(require("./types/server-remote-tool"));
 class RemoteTools {
     constructor(apiKeys, tools) {
         this.tools = [];
         this.apiKeys = apiKeys;
         this.tools.push(...(tools ?? []));
         if (this.apiKeys?.AI_REMOTE_TOOL_BRAVE_SEARCH_API_KEY) {
-            this.tools.push(new remote_tool_1.default({
+            this.tools.push(new server_remote_tool_1.default({
                 sourceId: 'brave_search',
-                sourceType: 'server',
                 tool: new brave_search_1.BraveSearch({ apiKey: this.apiKeys.AI_REMOTE_TOOL_BRAVE_SEARCH_API_KEY }),
             }));
         }
@@ -46,4 +45,4 @@ class RemoteTools {
     }
 }
 exports.RemoteTools = RemoteTools;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicmVtb3RlLXRvb2xzLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vc3JjL3JlbW90ZS10b29scy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7QUFHQSwwRUFBc0U7QUFDdEUsbUVBQWlFO0FBRWpFLHFDQUF5RTtBQUN6RSxnRUFBdUM7QUFRdkMsTUFBYSxXQUFXO0lBSXRCLFlBQVksT0FBMkIsRUFBRSxLQUFvQjtRQUZwRCxVQUFLLEdBQWlCLEVBQUUsQ0FBQztRQUdoQyxJQUFJLENBQUMsT0FBTyxHQUFHLE9BQU8sQ0FBQztRQUN2QixJQUFJLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxHQUFHLENBQUMsS0FBSyxJQUFJLEVBQUUsQ0FBQyxDQUFDLENBQUM7UUFFbEMsSUFBSSxJQUFJLENBQUMsT0FBTyxFQUFFLG1DQUFtQyxFQUFFLENBQUM7WUFDdEQsSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQ2IsSUFBSSxxQkFBVSxDQUFDO2dCQUNiLFFBQVEsRUFBRSxjQUFjO2dCQUN4QixVQUFVLEVBQUUsUUFBUTtnQkFDcEIsSUFBSSxFQUFFLElBQUksMEJBQVcsQ0FBQyxFQUFFLE1BQU0sRUFBRSxJQUFJLENBQUMsT0FBTyxDQUFDLG1DQUFtQyxFQUFFLENBQUM7YUFDcEYsQ0FBQyxDQUNILENBQUM7UUFDSixDQUFDO0lBQ0gsQ0FBQztJQUVELElBQUksMEJBQTBCO1FBQzVCLE9BQU8sSUFBSSxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsWUFBWSxDQUFDLEVBQUU7WUFDbkMsT0FBTztnQkFDTCxJQUFJLEVBQUUsWUFBWSxDQUFDLGFBQWE7Z0JBQ2hDLFdBQVcsRUFBRSxZQUFZLENBQUMsSUFBSSxDQUFDLFdBQVc7Z0JBQzFDLGNBQWMsRUFBRSxTQUEyQjtnQkFDM0MsTUFBTSxFQUFFLElBQUEsMEJBQVksRUFBQyxZQUFZLENBQUMsSUFBSSxDQUFDLE1BQU0sQ0FBQztnQkFDOUMsUUFBUSxFQUFFLFlBQVksQ0FBQyxRQUFRO2dCQUMvQixVQUFVLEVBQUUsWUFBWSxDQUFDLFVBQVU7YUFDcEMsQ0FBQztRQUNKLENBQUMsQ0FBQyxDQUFDO0lBQ0wsQ0FBQztJQUVELEtBQUssQ0FBQyxVQUFVLENBQUMsUUFBZ0IsRUFBRSxRQUE0RDtRQUM3RixNQUFNLFlBQVksR0FBRyxJQUFJLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxNQUFNLENBQUMsRUFBRSxDQUFDLE1BQU0sQ0FBQyxhQUFhLEtBQUssUUFBUSxDQUFDLENBQUM7UUFFbEYsSUFBSSxDQUFDLFlBQVk7WUFBRSxNQUFNLElBQUksNEJBQW1CLENBQUMsUUFBUSxRQUFRLFlBQVksQ0FBQyxDQUFDO1FBRS9FLElBQUksQ0FBQztZQUNILE9BQU8sQ0FBQyxNQUFNLFlBQVksQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLFFBQVEsQ0FBQyxDQUFZLENBQUM7UUFDL0QsQ0FBQztRQUFDLE9BQU8sS0FBSyxFQUFFLENBQUM7WUFDZixNQUFNLElBQUksaUNBQXdCLENBQ2hDLDRCQUE0QixRQUFRLEtBQU0sS0FBZSxDQUFDLE9BQU8sRUFBRSxDQUNwRSxDQUFDO1FBQ0osQ0FBQztJQUNILENBQUM7Q0FDRjtBQTdDRCxrQ0E2Q0MifQ==
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicmVtb3RlLXRvb2xzLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vc3JjL3JlbW90ZS10b29scy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7QUFJQSwwRUFBc0U7QUFDdEUsbUVBQWlFO0FBRWpFLDJDQUErRTtBQUMvRSxvRkFBMEQ7QUFRMUQsTUFBYSxXQUFXO0lBSXRCLFlBQVksT0FBMkIsRUFBRSxLQUFvQjtRQUZwRCxVQUFLLEdBQWlCLEVBQUUsQ0FBQztRQUdoQyxJQUFJLENBQUMsT0FBTyxHQUFHLE9BQU8sQ0FBQztRQUN2QixJQUFJLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxHQUFHLENBQUMsS0FBSyxJQUFJLEVBQUUsQ0FBQyxDQUFDLENBQUM7UUFFbEMsSUFBSSxJQUFJLENBQUMsT0FBTyxFQUFFLG1DQUFtQyxFQUFFLENBQUM7WUFDdEQsSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQ2IsSUFBSSw0QkFBZ0IsQ0FBQztnQkFDbkIsUUFBUSxFQUFFLGNBQWM7Z0JBQ3hCLElBQUksRUFBRSxJQUFJLDBCQUFXLENBQUMsRUFBRSxNQUFNLEVBQUUsSUFBSSxDQUFDLE9BQU8sQ0FBQyxtQ0FBbUMsRUFBRSxDQUFDO2FBQ3BGLENBQUMsQ0FDSCxDQUFDO1FBQ0osQ0FBQztJQUNILENBQUM7SUFFRCxJQUFJLDBCQUEwQjtRQUM1QixPQUFPLElBQUksQ0FBQyxLQUFLLENBQUMsR0FBRyxDQUFDLFlBQVksQ0FBQyxFQUFFO1lBQ25DLE9BQU87Z0JBQ0wsSUFBSSxFQUFFLFlBQVksQ0FBQyxhQUFhO2dCQUNoQyxXQUFXLEVBQUUsWUFBWSxDQUFDLElBQUksQ0FBQyxXQUFXO2dCQUMxQyxjQUFjLEVBQUUsU0FBMkI7Z0JBQzNDLE1BQU0sRUFBRSxJQUFBLDBCQUFZLEVBQUMsWUFBWSxDQUFDLElBQUksQ0FBQyxNQUFNLENBQUM7Z0JBQzlDLFFBQVEsRUFBRSxZQUFZLENBQUMsUUFBUTtnQkFDL0IsVUFBVSxFQUFFLFlBQVksQ0FBQyxVQUFVO2FBQ3BDLENBQUM7UUFDSixDQUFDLENBQUMsQ0FBQztJQUNMLENBQUM7SUFFRCxLQUFLLENBQUMsVUFBVSxDQUFDLFFBQWdCLEVBQUUsUUFBNEQ7UUFDN0YsTUFBTSxZQUFZLEdBQUcsSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLEVBQUUsQ0FBQyxNQUFNLENBQUMsYUFBYSxLQUFLLFFBQVEsQ0FBQyxDQUFDO1FBRWxGLElBQUksQ0FBQyxZQUFZO1lBQUUsTUFBTSxJQUFJLDRCQUFtQixDQUFDLFFBQVEsUUFBUSxZQUFZLENBQUMsQ0FBQztRQUUvRSxJQUFJLENBQUM7WUFDSCxPQUFPLENBQUMsTUFBTSxZQUFZLENBQUMsSUFBSSxDQUFDLE1BQU0sQ0FBQyxRQUFRLENBQUMsQ0FBWSxDQUFDO1FBQy9ELENBQUM7UUFBQyxPQUFPLEtBQUssRUFBRSxDQUFDO1lBQ2YsTUFBTSxJQUFJLGlDQUF3QixDQUNoQyw0QkFBNEIsUUFBUSxLQUFNLEtBQWUsQ0FBQyxPQUFPLEVBQUUsQ0FDcEUsQ0FBQztRQUNKLENBQUM7SUFDSCxDQUFDO0NBQ0Y7QUE1Q0Qsa0NBNENDIn0=

package/dist/router.d.ts

@@ -9,29 +9,26 @@ export type Body = DispatchBody | InvokeRemoteToolBody | undefined;
 export type Route = 'ai-query' | 'remote-tools' | 'invoke-remote-tool';
 export type Query = {
     'tool-name'?: string;
+    'ai-name'?: string;
 };
 export type ApiKeys = RemoteToolsApiKeys;
 export declare class Router {
     private readonly localToolsApiKeys?;
-    private readonly aiConfiguration;
+    private readonly aiConfigurations;
     private readonly logger?;
     constructor(params?: {
-        aiConfiguration?: AiConfiguration;
+        aiConfigurations?: AiConfiguration[];
         localToolsApiKeys?: ApiKeys;
         logger?: Logger;
     });
+    private getAiConfiguration;
     /**
      * Route the request to the appropriate handler
      *
-     * List of routes:
-     * // dispatch the query to the AI
-     * - /ai-query
-     *
-     * // invoke a remote tool by name
-     * - /invoke-remote-tool?tool-name=:toolName
-     *
-     * // get the list of available remote tools
-     * - /remote-tools
+     * Routes:
+     * - ai-query: Dispatch messages to the configured AI provider and return the response
+     * - invoke-remote-tool: Execute a remote tool by name with the provided inputs
+     * - remote-tools: Return the list of available remote tools definitions
      */
     route(args: {
         body?: Body;

package/dist/router.js

@@ -9,22 +9,31 @@ const mcp_client_1 = __importDefault(require("./mcp-client"));
 const remote_tools_1 = require("./remote-tools");
 class Router {
     constructor(params) {
-        this.aiConfiguration = params?.aiConfiguration ?? null;
+        this.aiConfigurations = params?.aiConfigurations ?? [];
         this.localToolsApiKeys = params?.localToolsApiKeys;
         this.logger = params?.logger;
     }
+    getAiConfiguration(aiName) {
+        if (this.aiConfigurations.length === 0)
+            return null;
+        if (aiName) {
+            const config = this.aiConfigurations.find(c => c.name === aiName);
+            if (!config) {
+                const fallback = this.aiConfigurations[0];
+                this.logger?.('Warn', `AI configuration '${aiName}' not found. Falling back to '${fallback.name}'.`);
+                return fallback;
+            }
+            return config;
+        }
+        return this.aiConfigurations[0];
+    }
     /**
      * Route the request to the appropriate handler
      *
-     * List of routes:
-     * // dispatch the query to the AI
-     * - /ai-query
-     *
-     * // invoke a remote tool by name
-     * - /invoke-remote-tool?tool-name=:toolName
-     *
-     * // get the list of available remote tools
-     * - /remote-tools
+     * Routes:
+     * - ai-query: Dispatch messages to the configured AI provider and return the response
+     * - invoke-remote-tool: Execute a remote tool by name with the provided inputs
+     * - remote-tools: Return the list of available remote tools definitions
      */
     async route(args) {
         let mcpClient;
@@ -32,12 +41,21 @@ class Router {
             if (args.mcpConfigs) {
                 mcpClient = new mcp_client_1.default(args.mcpConfigs, this.logger);
             }
-            const remoteTools = new remote_tools_1.RemoteTools(this.localToolsApiKeys, await mcpClient?.loadTools());
+            const remoteTools = new remote_tools_1.RemoteTools(this.localToolsApiKeys ?? {}, await mcpClient?.loadTools());
             if (args.route === 'ai-query') {
-                return await new index_1.ProviderDispatcher(this.aiConfiguration, remoteTools).dispatch(args.body);
+                const aiConfiguration = this.getAiConfiguration(args.query?.['ai-name']);
+                return await new index_1.ProviderDispatcher(aiConfiguration, remoteTools).dispatch(args.body);
             }
             if (args.route === 'invoke-remote-tool') {
-                return await remoteTools.invokeTool(args.query['tool-name'], args.body.inputs);
+                const toolName = args.query?.['tool-name'];
+                if (!toolName) {
+                    throw new index_1.AIBadRequestError('Missing required query parameter: tool-name');
+                }
+                const body = args.body;
+                if (!body?.inputs) {
+                    throw new index_1.AIBadRequestError('Missing required body parameter: inputs');
+                }
+                return await remoteTools.invokeTool(toolName, body.inputs);
             }
             if (args.route === 'remote-tools') {
                 return remoteTools.toolDefinitionsForFrontend;
@@ -55,11 +73,12 @@ class Router {
                     await mcpClient.closeConnections();
                 }
                 catch (cleanupError) {
-                    this.logger?.('Error', 'Error during MCP connection cleanup', cleanupError);
+                    const error = cleanupError instanceof Error ? cleanupError : new Error(String(cleanupError));
+                    this.logger?.('Error', 'Error during MCP connection cleanup', error);
                 }
             }
         }
     }
 }
 exports.Router = Router;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicm91dGVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vc3JjL3JvdXRlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7QUFLQSxtQ0FBbUU7QUFDbkUsOERBQXFDO0FBQ3JDLGlEQUE2QztBQVU3QyxNQUFhLE1BQU07SUFLakIsWUFBWSxNQUlYO1FBQ0MsSUFBSSxDQUFDLGVBQWUsR0FBRyxNQUFNLEVBQUUsZUFBZSxJQUFJLElBQUksQ0FBQztRQUN2RCxJQUFJLENBQUMsaUJBQWlCLEdBQUcsTUFBTSxFQUFFLGlCQUFpQixDQUFDO1FBQ25ELElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxFQUFFLE1BQU0sQ0FBQztJQUMvQixDQUFDO0lBRUQ7Ozs7Ozs7Ozs7OztPQVlHO0lBQ0gsS0FBSyxDQUFDLEtBQUssQ0FBQyxJQUFpRjtRQUMzRixJQUFJLFNBQWdDLENBQUM7UUFFckMsSUFBSSxDQUFDO1lBQ0gsSUFBSSxJQUFJLENBQUMsVUFBVSxFQUFFLENBQUM7Z0JBQ3BCLFNBQVMsR0FBRyxJQUFJLG9CQUFTLENBQUMsSUFBSSxDQUFDLFVBQVUsRUFBRSxJQUFJLENBQUMsTUFBTSxDQUFDLENBQUM7WUFDMUQsQ0FBQztZQUVELE1BQU0sV0FBVyxHQUFHLElBQUksMEJBQVcsQ0FBQyxJQUFJLENBQUMsaUJBQWlCLEVBQUUsTUFBTSxTQUFTLEVBQUUsU0FBUyxFQUFFLENBQUMsQ0FBQztZQUUxRixJQUFJLElBQUksQ0FBQyxLQUFLLEtBQUssVUFBVSxFQUFFLENBQUM7Z0JBQzlCLE9BQU8sTUFBTSxJQUFJLDBCQUFrQixDQUFDLElBQUksQ0FBQyxlQUFlLEVBQUUsV0FBVyxDQUFDLENBQUMsUUFBUSxDQUM3RSxJQUFJLENBQUMsSUFBb0IsQ0FDMUIsQ0FBQztZQUNKLENBQUM7WUFFRCxJQUFJLElBQUksQ0FBQyxLQUFLLEtBQUssb0JBQW9CLEVBQUUsQ0FBQztnQkFDeEMsT0FBTyxNQUFNLFdBQVcsQ0FBQyxVQUFVLENBQ2pDLElBQUksQ0FBQyxLQUFLLENBQUMsV0FBVyxDQUFDLEVBQ3RCLElBQUksQ0FBQyxJQUE2QixDQUFDLE1BQU0sQ0FDM0MsQ0FBQztZQUNKLENBQUM7WUFFRCxJQUFJLElBQUksQ0FBQyxLQUFLLEtBQUssY0FBYyxFQUFFLENBQUM7Z0JBQ2xDLE9BQU8sV0FBVyxDQUFDLDBCQUEwQixDQUFDO1lBQ2hELENBQUM7WUFFRCxxRkFBcUY7WUFDckYsTUFBTSxJQUFJLDRCQUFvQixDQUM1Qix5QkFBeUIsSUFBSSxDQUFDLFNBQVMsQ0FBQztnQkFDdEMsSUFBSSxFQUFFLElBQUksQ0FBQyxJQUFJO2dCQUNmLEtBQUssRUFBRSxJQUFJLENBQUMsS0FBSztnQkFDakIsS0FBSyxFQUFFLElBQUksQ0FBQyxLQUFLO2FBQ2xCLENBQUMsRUFBRSxDQUNMLENBQUM7UUFDSixDQUFDO2dCQUFTLENBQUM7WUFDVCxJQUFJLFNBQVMsRUFBRSxDQUFDO2dCQUNkLElBQUksQ0FBQztvQkFDSCxNQUFNLFNBQVMsQ0FBQyxnQkFBZ0IsRUFBRSxDQUFDO2dCQUNyQyxDQUFDO2dCQUFDLE9BQU8sWUFBWSxFQUFFLENBQUM7b0JBQ3RCLElBQUksQ0FBQyxNQUFNLEVBQUUsQ0FBQyxPQUFPLEVBQUUscUNBQXFDLEVBQUUsWUFBcUIsQ0FBQyxDQUFDO2dCQUN2RixDQUFDO1lBQ0gsQ0FBQztRQUNILENBQUM7SUFDSCxDQUFDO0NBQ0Y7QUF6RUQsd0JBeUVDIn0=
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicm91dGVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vc3JjL3JvdXRlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7QUFLQSxtQ0FBc0Y7QUFDdEYsOERBQXFDO0FBQ3JDLGlEQUE2QztBQVc3QyxNQUFhLE1BQU07SUFLakIsWUFBWSxNQUlYO1FBQ0MsSUFBSSxDQUFDLGdCQUFnQixHQUFHLE1BQU0sRUFBRSxnQkFBZ0IsSUFBSSxFQUFFLENBQUM7UUFDdkQsSUFBSSxDQUFDLGlCQUFpQixHQUFHLE1BQU0sRUFBRSxpQkFBaUIsQ0FBQztRQUNuRCxJQUFJLENBQUMsTUFBTSxHQUFHLE1BQU0sRUFBRSxNQUFNLENBQUM7SUFDL0IsQ0FBQztJQUVPLGtCQUFrQixDQUFDLE1BQWU7UUFDeEMsSUFBSSxJQUFJLENBQUMsZ0JBQWdCLENBQUMsTUFBTSxLQUFLLENBQUM7WUFBRSxPQUFPLElBQUksQ0FBQztRQUVwRCxJQUFJLE1BQU0sRUFBRSxDQUFDO1lBQ1gsTUFBTSxNQUFNLEdBQUcsSUFBSSxDQUFDLGdCQUFnQixDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsRUFBRSxDQUFDLENBQUMsQ0FBQyxJQUFJLEtBQUssTUFBTSxDQUFDLENBQUM7WUFFbEUsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDO2dCQUNaLE1BQU0sUUFBUSxHQUFHLElBQUksQ0FBQyxnQkFBZ0IsQ0FBQyxDQUFDLENBQUMsQ0FBQztnQkFDMUMsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUNYLE1BQU0sRUFDTixxQkFBcUIsTUFBTSxpQ0FBaUMsUUFBUSxDQUFDLElBQUksSUFBSSxDQUM5RSxDQUFDO2dCQUVGLE9BQU8sUUFBUSxDQUFDO1lBQ2xCLENBQUM7WUFFRCxPQUFPLE1BQU0sQ0FBQztRQUNoQixDQUFDO1FBRUQsT0FBTyxJQUFJLENBQUMsZ0JBQWdCLENBQUMsQ0FBQyxDQUFDLENBQUM7SUFDbEMsQ0FBQztJQUVEOzs7Ozs7O09BT0c7SUFDSCxLQUFLLENBQUMsS0FBSyxDQUFDLElBQWlGO1FBQzNGLElBQUksU0FBZ0MsQ0FBQztRQUVyQyxJQUFJLENBQUM7WUFDSCxJQUFJLElBQUksQ0FBQyxVQUFVLEVBQUUsQ0FBQztnQkFDcEIsU0FBUyxHQUFHLElBQUksb0JBQVMsQ0FBQyxJQUFJLENBQUMsVUFBVSxFQUFFLElBQUksQ0FBQyxNQUFNLENBQUMsQ0FBQztZQUMxRCxDQUFDO1lBRUQsTUFBTSxXQUFXLEdBQUcsSUFBSSwwQkFBVyxDQUNqQyxJQUFJLENBQUMsaUJBQWlCLElBQUksRUFBRSxFQUM1QixNQUFNLFNBQVMsRUFBRSxTQUFTLEVBQUUsQ0FDN0IsQ0FBQztZQUVGLElBQUksSUFBSSxDQUFDLEtBQUssS0FBSyxVQUFVLEVBQUUsQ0FBQztnQkFDOUIsTUFBTSxlQUFlLEdBQUcsSUFBSSxDQUFDLGtCQUFrQixDQUFDLElBQUksQ0FBQyxLQUFLLEVBQUUsQ0FBQyxTQUFTLENBQUMsQ0FBQyxDQUFDO2dCQUV6RSxPQUFPLE1BQU0sSUFBSSwwQkFBa0IsQ0FBQyxlQUFlLEVBQUUsV0FBVyxDQUFDLENBQUMsUUFBUSxDQUN4RSxJQUFJLENBQUMsSUFBb0IsQ0FDMUIsQ0FBQztZQUNKLENBQUM7WUFFRCxJQUFJLElBQUksQ0FBQyxLQUFLLEtBQUssb0JBQW9CLEVBQUUsQ0FBQztnQkFDeEMsTUFBTSxRQUFRLEdBQUcsSUFBSSxDQUFDLEtBQUssRUFBRSxDQUFDLFdBQVcsQ0FBQyxDQUFDO2dCQUUzQyxJQUFJLENBQUMsUUFBUSxFQUFFLENBQUM7b0JBQ2QsTUFBTSxJQUFJLHlCQUFpQixDQUFDLDZDQUE2QyxDQUFDLENBQUM7Z0JBQzdFLENBQUM7Z0JBRUQsTUFBTSxJQUFJLEdBQUcsSUFBSSxDQUFDLElBQXdDLENBQUM7Z0JBRTNELElBQUksQ0FBQyxJQUFJLEVBQUUsTUFBTSxFQUFFLENBQUM7b0JBQ2xCLE1BQU0sSUFBSSx5QkFBaUIsQ0FBQyx5Q0FBeUMsQ0FBQyxDQUFDO2dCQUN6RSxDQUFDO2dCQUVELE9BQU8sTUFBTSxXQUFXLENBQUMsVUFBVSxDQUFDLFFBQVEsRUFBRSxJQUFJLENBQUMsTUFBTSxDQUFDLENBQUM7WUFDN0QsQ0FBQztZQUVELElBQUksSUFBSSxDQUFDLEtBQUssS0FBSyxjQUFjLEVBQUUsQ0FBQztnQkFDbEMsT0FBTyxXQUFXLENBQUMsMEJBQTBCLENBQUM7WUFDaEQsQ0FBQztZQUVELHFGQUFxRjtZQUNyRixNQUFNLElBQUksNEJBQW9CLENBQzVCLHlCQUF5QixJQUFJLENBQUMsU0FBUyxDQUFDO2dCQUN0QyxJQUFJLEVBQUUsSUFBSSxDQUFDLElBQUk7Z0JBQ2YsS0FBSyxFQUFFLElBQUksQ0FBQyxLQUFLO2dCQUNqQixLQUFLLEVBQUUsSUFBSSxDQUFDLEtBQUs7YUFDbEIsQ0FBQyxFQUFFLENBQ0wsQ0FBQztRQUNKLENBQUM7Z0JBQVMsQ0FBQztZQUNULElBQUksU0FBUyxFQUFFLENBQUM7Z0JBQ2QsSUFBSSxDQUFDO29CQUNILE1BQU0sU0FBUyxDQUFDLGdCQUFnQixFQUFFLENBQUM7Z0JBQ3JDLENBQUM7Z0JBQUMsT0FBTyxZQUFZLEVBQUUsQ0FBQztvQkFDdEIsTUFBTSxLQUFLLEdBQ1QsWUFBWSxZQUFZLEtBQUssQ0FBQyxDQUFDLENBQUMsWUFBWSxDQUFDLENBQUMsQ0FBQyxJQUFJLEtBQUssQ0FBQyxNQUFNLENBQUMsWUFBWSxDQUFDLENBQUMsQ0FBQztvQkFDakYsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDLE9BQU8sRUFBRSxxQ0FBcUMsRUFBRSxLQUFLLENBQUMsQ0FBQztnQkFDdkUsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO0lBQ0gsQ0FBQztDQUNGO0FBMUdELHdCQTBHQyJ9

package/dist/{errors.d.ts → types/errors.d.ts}

@@ -10,6 +10,13 @@
  * -------------------------------------
  */
 export declare class AIError extends Error {
+    readonly status: number;
+    constructor(message: string, status?: number);
+}
+export declare class AIBadRequestError extends AIError {
+    constructor(message: string);
+}
+export declare class AINotFoundError extends AIError {
     constructor(message: string);
 }
 export declare class AIUnprocessableError extends AIError {
@@ -24,7 +31,7 @@ export declare class OpenAIUnprocessableError extends AIUnprocessableError {
 export declare class AIToolUnprocessableError extends AIUnprocessableError {
     constructor(message: string);
 }
-export declare class AIToolNotFoundError extends AIError {
+export declare class AIToolNotFoundError extends AINotFoundError {
     constructor(message: string);
 }
 export declare class McpError extends AIError {

package/dist/types/errors.js

@@ -0,0 +1,104 @@
+"use strict";
+/**
+ * -------------------------------------
+ * -------------------------------------
+ * -------------------------------------
+ * All custom errors must extend the AIError class.
+ * This inheritance is crucial for proper error translation
+ * and consistent handling throughout the system.
+ * -------------------------------------
+ * -------------------------------------
+ * -------------------------------------
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.McpConfigError = exports.McpConflictError = exports.McpConnectionError = exports.McpError = exports.AIToolNotFoundError = exports.AIToolUnprocessableError = exports.OpenAIUnprocessableError = exports.AINotConfiguredError = exports.AIUnprocessableError = exports.AINotFoundError = exports.AIBadRequestError = exports.AIError = void 0;
+// eslint-disable-next-line max-classes-per-file
+class AIError extends Error {
+    constructor(message, status = 422) {
+        if (status < 100 || status > 599) {
+            throw new RangeError(`Invalid HTTP status code: ${status}`);
+        }
+        super(message);
+        this.name = 'AIError';
+        this.status = status;
+    }
+}
+exports.AIError = AIError;
+class AIBadRequestError extends AIError {
+    constructor(message) {
+        super(message, 400);
+        this.name = 'AIBadRequestError';
+    }
+}
+exports.AIBadRequestError = AIBadRequestError;
+class AINotFoundError extends AIError {
+    constructor(message) {
+        super(message, 404);
+        this.name = 'AINotFoundError';
+    }
+}
+exports.AINotFoundError = AINotFoundError;
+class AIUnprocessableError extends AIError {
+    constructor(message) {
+        super(message, 422);
+        this.name = 'AIUnprocessableError';
+    }
+}
+exports.AIUnprocessableError = AIUnprocessableError;
+class AINotConfiguredError extends AIError {
+    constructor() {
+        super('AI is not configured. Please call addAI() on your agent.', 422);
+        this.name = 'AINotConfiguredError';
+    }
+}
+exports.AINotConfiguredError = AINotConfiguredError;
+class OpenAIUnprocessableError extends AIUnprocessableError {
+    constructor(message) {
+        super(message);
+        this.name = 'OpenAIError';
+    }
+}
+exports.OpenAIUnprocessableError = OpenAIUnprocessableError;
+class AIToolUnprocessableError extends AIUnprocessableError {
+    constructor(message) {
+        super(message);
+        this.name = 'AIToolError';
+    }
+}
+exports.AIToolUnprocessableError = AIToolUnprocessableError;
+class AIToolNotFoundError extends AINotFoundError {
+    constructor(message) {
+        super(message);
+        this.name = 'AIToolNotFoundError';
+    }
+}
+exports.AIToolNotFoundError = AIToolNotFoundError;
+class McpError extends AIError {
+    constructor(message) {
+        super(message);
+        this.name = 'McpError';
+    }
+}
+exports.McpError = McpError;
+class McpConnectionError extends McpError {
+    constructor(message) {
+        super(message);
+        this.name = 'McpConnectionError';
+    }
+}
+exports.McpConnectionError = McpConnectionError;
+class McpConflictError extends McpError {
+    constructor(entityName) {
+        super(`"${entityName}" already exists for your project`);
+        this.name = 'McpConflictError';
+    }
+}
+exports.McpConflictError = McpConflictError;
+class McpConfigError extends McpError {
+    constructor(message) {
+        super(message);
+        this.name = 'McpConfigError';
+    }
+}
+exports.McpConfigError = McpConfigError;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZXJyb3JzLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL3R5cGVzL2Vycm9ycy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiO0FBQUE7Ozs7Ozs7Ozs7R0FVRzs7O0FBRUgsZ0RBQWdEO0FBQ2hELE1BQWEsT0FBUSxTQUFRLEtBQUs7SUFHaEMsWUFBWSxPQUFlLEVBQUUsTUFBTSxHQUFHLEdBQUc7UUFDdkMsSUFBSSxNQUFNLEdBQUcsR0FBRyxJQUFJLE1BQU0sR0FBRyxHQUFHLEVBQUUsQ0FBQztZQUNqQyxNQUFNLElBQUksVUFBVSxDQUFDLDZCQUE2QixNQUFNLEVBQUUsQ0FBQyxDQUFDO1FBQzlELENBQUM7UUFFRCxLQUFLLENBQUMsT0FBTyxDQUFDLENBQUM7UUFDZixJQUFJLENBQUMsSUFBSSxHQUFHLFNBQVMsQ0FBQztRQUN0QixJQUFJLENBQUMsTUFBTSxHQUFHLE1BQU0sQ0FBQztJQUN2QixDQUFDO0NBQ0Y7QUFaRCwwQkFZQztBQUVELE1BQWEsaUJBQWtCLFNBQVEsT0FBTztJQUM1QyxZQUFZLE9BQWU7UUFDekIsS0FBSyxDQUFDLE9BQU8sRUFBRSxHQUFHLENBQUMsQ0FBQztRQUNwQixJQUFJLENBQUMsSUFBSSxHQUFHLG1CQUFtQixDQUFDO0lBQ2xDLENBQUM7Q0FDRjtBQUxELDhDQUtDO0FBRUQsTUFBYSxlQUFnQixTQUFRLE9BQU87SUFDMUMsWUFBWSxPQUFlO1FBQ3pCLEtBQUssQ0FBQyxPQUFPLEVBQUUsR0FBRyxDQUFDLENBQUM7UUFDcEIsSUFBSSxDQUFDLElBQUksR0FBRyxpQkFBaUIsQ0FBQztJQUNoQyxDQUFDO0NBQ0Y7QUFMRCwwQ0FLQztBQUVELE1BQWEsb0JBQXFCLFNBQVEsT0FBTztJQUMvQyxZQUFZLE9BQWU7UUFDekIsS0FBSyxDQUFDLE9BQU8sRUFBRSxHQUFHLENBQUMsQ0FBQztRQUNwQixJQUFJLENBQUMsSUFBSSxHQUFHLHNCQUFzQixDQUFDO0lBQ3JDLENBQUM7Q0FDRjtBQUxELG9EQUtDO0FBRUQsTUFBYSxvQkFBcUIsU0FBUSxPQUFPO0lBQy9DO1FBQ0UsS0FBSyxDQUFDLDBEQUEwRCxFQUFFLEdBQUcsQ0FBQyxDQUFDO1FBQ3ZFLElBQUksQ0FBQyxJQUFJLEdBQUcsc0JBQXNCLENBQUM7SUFDckMsQ0FBQztDQUNGO0FBTEQsb0RBS0M7QUFFRCxNQUFhLHdCQUF5QixTQUFRLG9CQUFvQjtJQUNoRSxZQUFZLE9BQWU7UUFDekIsS0FBSyxDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQ2YsSUFBSSxDQUFDLElBQUksR0FBRyxhQUFhLENBQUM7SUFDNUIsQ0FBQztDQUNGO0FBTEQsNERBS0M7QUFFRCxNQUFhLHdCQUF5QixTQUFRLG9CQUFvQjtJQUNoRSxZQUFZLE9BQWU7UUFDekIsS0FBSyxDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQ2YsSUFBSSxDQUFDLElBQUksR0FBRyxhQUFhLENBQUM7SUFDNUIsQ0FBQztDQUNGO0FBTEQsNERBS0M7QUFFRCxNQUFhLG1CQUFvQixTQUFRLGVBQWU7SUFDdEQsWUFBWSxPQUFlO1FBQ3pCLEtBQUssQ0FBQyxPQUFPLENBQUMsQ0FBQztRQUNmLElBQUksQ0FBQyxJQUFJLEdBQUcscUJBQXFCLENBQUM7SUFDcEMsQ0FBQztDQUNGO0FBTEQsa0RBS0M7QUFFRCxNQUFhLFFBQVMsU0FBUSxPQUFPO0lBQ25DLFlBQVksT0FBZTtRQUN6QixLQUFLLENBQUMsT0FBTyxDQUFDLENBQUM7UUFDZixJQUFJLENBQUMsSUFBSSxHQUFHLFVBQVUsQ0FBQztJQUN6QixDQUFDO0NBQ0Y7QUFMRCw0QkFLQztBQUVELE1BQWEsa0JBQW1CLFNBQVEsUUFBUTtJQUM5QyxZQUFZLE9BQWU7UUFDekIsS0FBSyxDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQ2YsSUFBSSxDQUFDLElBQUksR0FBRyxvQkFBb0IsQ0FBQztJQUNuQyxDQUFDO0NBQ0Y7QUFMRCxnREFLQztBQUVELE1BQWEsZ0JBQWlCLFNBQVEsUUFBUTtJQUM1QyxZQUFZLFVBQWtCO1FBQzVCLEtBQUssQ0FBQyxJQUFJLFVBQVUsbUNBQW1DLENBQUMsQ0FBQztRQUN6RCxJQUFJLENBQUMsSUFBSSxHQUFHLGtCQUFrQixDQUFDO0lBQ2pDLENBQUM7Q0FDRjtBQUxELDRDQUtDO0FBRUQsTUFBYSxjQUFlLFNBQVEsUUFBUTtJQUMxQyxZQUFZLE9BQWU7UUFDekIsS0FBSyxDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQ2YsSUFBSSxDQUFDLElBQUksR0FBRyxnQkFBZ0IsQ0FBQztJQUMvQixDQUFDO0NBQ0Y7QUFMRCx3Q0FLQyJ9

package/dist/{mcp-config-checker.d.ts → types/mcp-config-checker.d.ts}

@@ -1,4 +1,4 @@
-import type { McpConfiguration } from './mcp-client';
+import type { McpConfiguration } from '../mcp-client';
 export default class McpConfigChecker {
     static check(mcpConfig: McpConfiguration): Promise<true>;
 }

package/dist/{mcp-config-checker.js → types/mcp-config-checker.js}

@@ -3,11 +3,11 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-const mcp_client_1 = __importDefault(require("./mcp-client"));
+const mcp_client_1 = __importDefault(require("../mcp-client"));
 class McpConfigChecker {
     static check(mcpConfig) {
         return new mcp_client_1.default(mcpConfig).testConnections();
     }
 }
 exports.default = McpConfigChecker;
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibWNwLWNvbmZpZy1jaGVja2VyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vc3JjL21jcC1jb25maWctY2hlY2tlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7OztBQUVBLDhEQUFxQztBQUVyQyxNQUFxQixnQkFBZ0I7SUFDbkMsTUFBTSxDQUFDLEtBQUssQ0FBQyxTQUEyQjtRQUN0QyxPQUFPLElBQUksb0JBQVMsQ0FBQyxTQUFTLENBQUMsQ0FBQyxlQUFlLEVBQUUsQ0FBQztJQUNwRCxDQUFDO0NBQ0Y7QUFKRCxtQ0FJQyJ9
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibWNwLWNvbmZpZy1jaGVja2VyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL3R5cGVzL21jcC1jb25maWctY2hlY2tlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7OztBQUVBLCtEQUFzQztBQUV0QyxNQUFxQixnQkFBZ0I7SUFDbkMsTUFBTSxDQUFDLEtBQUssQ0FBQyxTQUEyQjtRQUN0QyxPQUFPLElBQUksb0JBQVMsQ0FBQyxTQUFTLENBQUMsQ0FBQyxlQUFlLEVBQUUsQ0FBQztJQUNwRCxDQUFDO0NBQ0Y7QUFKRCxtQ0FJQyJ9

package/dist/types/mcp-server-remote-tool.d.ts

@@ -0,0 +1,9 @@
+import type { StructuredToolInterface } from '@langchain/core/tools';
+import RemoteTool from './remote-tool';
+export default class McpServerRemoteTool<ToolType = unknown> extends RemoteTool {
+    constructor(options: {
+        tool: StructuredToolInterface<ToolType>;
+        sourceId?: string;
+    });
+}
+//# sourceMappingURL=mcp-server-remote-tool.d.ts.map