@ainetwork/adk 0.3.1 → 0.3.3

package/dist/esm/chunk-YDMTU6UH.js

@@ -1,142 +0,0 @@
-import {
-  MCPConnector
-} from "./chunk-IKYOHZYW.js";
-import {
-  loggers
-} from "./chunk-N7VYM2R5.js";
-
-// src/modules/mcp/mcp.module.ts
-import { Client as MCPClient } from "@modelcontextprotocol/sdk/client/index.js";
-import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
-import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
-import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
-import { WebSocketClientTransport } from "@modelcontextprotocol/sdk/client/websocket.js";
-var MCPModule = class {
-  mcpConnectors = /* @__PURE__ */ new Map();
-  addMCPConnector(configs) {
-    for (const [name, config] of Object.entries(configs)) {
-      const conn = new MCPConnector(name, config);
-      this.mcpConnectors.set(name, conn);
-    }
-  }
-  getOrCreateClient(connector) {
-    connector.client ??= new MCPClient({
-      name: connector.name,
-      version: "1.0.0"
-    });
-    return connector.client;
-  }
-  async connectToServers() {
-    for (const [name, conn] of this.mcpConnectors.entries()) {
-      try {
-        const mcpClient = this.getOrCreateClient(conn);
-        const config = conn.config;
-        switch (config.type) {
-          case "stdio": {
-            const transport = new StdioClientTransport(config.params);
-            await mcpClient.connect(transport);
-            break;
-          }
-          case "websocket": {
-            const transport = new WebSocketClientTransport(config.url);
-            await mcpClient.connect(transport);
-            break;
-          }
-          case "sse": {
-            const transport = new SSEClientTransport(
-              config.url,
-              config.options
-            );
-            await mcpClient.connect(transport);
-            break;
-          }
-          case "streamableHttp": {
-            const transport = new StreamableHTTPClientTransport(
-              config.url,
-              config.options
-            );
-            await mcpClient.connect(transport);
-            break;
-          }
-          default:
-            loggers.mcp.error("Unsupported MCP config type");
-            break;
-        }
-        const toolList = await mcpClient.listTools();
-        conn.tools = toolList.tools.map((tool) => {
-          return {
-            toolName: `${name}-${tool.name}`,
-            // to avoid tool name duplication
-            connectorName: name,
-            protocol: "MCP" /* MCP */,
-            description: tool.description,
-            inputSchema: tool.inputSchema
-          };
-        });
-        loggers.mcp.info("Connected to MCP server with tools:", {
-          tools: conn.tools.map((tool) => tool.toolName)
-        });
-      } catch (error) {
-        loggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });
-      }
-    }
-  }
-  /**
-   * Returns all available tools from connected MCP servers.
-   *
-   * @returns Array of MCPTool instances representing available tools
-   */
-  getTools() {
-    const allTools = [];
-    for (const conn of this.mcpConnectors.values()) {
-      allTools.push(...conn.tools);
-    }
-    return allTools;
-  }
-  /**
-   * Executes a tool on its corresponding MCP server.
-   *
-   * @param tool - The MCPTool instance to execute
-   * @param _args - Arguments to pass to the tool
-   * @returns Promise resolving to the tool's execution result
-   * @throws Error if the MCP server for the tool is not found
-   */
-  async useTool(tool, _args) {
-    const { connectorName, toolName } = tool;
-    const client = this.mcpConnectors.get(connectorName)?.client;
-    try {
-      if (!client) {
-        throw new Error(`Invalid MCP Tool ${toolName}`);
-      }
-      const mcpToolName = toolName.slice(connectorName.length + 1);
-      const result = await client.callTool({
-        name: mcpToolName,
-        arguments: _args
-      });
-      const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
-` + JSON.stringify(result.content, null, 2);
-      return toolResult;
-    } catch (error) {
-      loggers.mcp.error("Failed to call tool", { error });
-      const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]
-${typeof error === "string" ? error : JSON.stringify(error, null, 2)}`;
-      return toolResult;
-    }
-  }
-  /**
-   * Closes all MCP client connections.
-   *
-   * Should be called when shutting down the application to ensure
-   * all MCP connections are properly closed.
-   */
-  async cleanup() {
-    for (const conn of this.mcpConnectors.values()) {
-      await conn.client?.close();
-    }
-  }
-};
-
-export {
-  MCPModule
-};
-//# sourceMappingURL=chunk-YDMTU6UH.js.map

package/dist/esm/chunk-YDMTU6UH.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../../src/modules/mcp/mcp.module.ts"],"sourcesContent":["import { Client as MCPClient } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { WebSocketClientTransport } from \"@modelcontextprotocol/sdk/client/websocket.js\";\nimport {\n\tCONNECTOR_PROTOCOL_TYPE,\n\ttype ConnectorTool,\n} from \"@/types/connector.js\";\nimport type { MCPConfig } from \"@/types/mcp.js\";\nimport { loggers } from \"@/utils/logger.js\";\nimport { MCPConnector } from \"./mcp.connector.js\";\n\n/**\n * Module for managing Model Context Protocol (MCP) server connections.\n *\n * This module handles the lifecycle of MCP client connections, discovers\n * available tools from connected servers, and provides an interface for\n * executing those tools. Multiple MCP servers can be connected simultaneously.\n */\nexport class MCPModule {\n\tprivate mcpConnectors: Map<string, MCPConnector> = new Map();\n\n\taddMCPConnector(configs: { [name: string]: MCPConfig }): void {\n\t\tfor (const [name, config] of Object.entries(configs)) {\n\t\t\tconst conn = new MCPConnector(name, config);\n\t\t\tthis.mcpConnectors.set(name, conn);\n\t\t}\n\t}\n\n\tprivate getOrCreateClient(connector: MCPConnector): MCPClient {\n\t\tconnector.client ??= new MCPClient({\n\t\t\tname: connector.name,\n\t\t\tversion: \"1.0.0\",\n\t\t});\n\t\treturn connector.client;\n\t}\n\n\tasync connectToServers(): Promise<void> {\n\t\tfor (const [name, conn] of this.mcpConnectors.entries()) {\n\t\t\ttry {\n\t\t\t\tconst mcpClient = this.getOrCreateClient(conn);\n\t\t\t\tconst config = conn.config;\n\t\t\t\tswitch (config.type) {\n\t\t\t\t\tcase \"stdio\": {\n\t\t\t\t\t\tconst transport = new StdioClientTransport(config.params);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"websocket\": {\n\t\t\t\t\t\tconst transport = new WebSocketClientTransport(config.url);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"sse\": {\n\t\t\t\t\t\tconst transport = new SSEClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tcase \"streamableHttp\": {\n\t\t\t\t\t\tconst transport = new StreamableHTTPClientTransport(\n\t\t\t\t\t\t\tconfig.url,\n\t\t\t\t\t\t\tconfig.options,\n\t\t\t\t\t\t);\n\t\t\t\t\t\tawait mcpClient.connect(transport);\n\t\t\t\t\t\tbreak;\n\t\t\t\t\t}\n\t\t\t\t\tdefault:\n\t\t\t\t\t\t// This cannot happen.\n\t\t\t\t\t\tloggers.mcp.error(\"Unsupported MCP config type\");\n\t\t\t\t\t\tbreak;\n\t\t\t\t}\n\n\t\t\t\tconst toolList = await mcpClient.listTools();\n\t\t\t\tconn.tools = toolList.tools.map((tool) => {\n\t\t\t\t\treturn {\n\t\t\t\t\t\ttoolName: `${name}-${tool.name}`, // to avoid tool name duplication\n\t\t\t\t\t\tconnectorName: name,\n\t\t\t\t\t\tprotocol: CONNECTOR_PROTOCOL_TYPE.MCP,\n\t\t\t\t\t\tdescription: tool.description,\n\t\t\t\t\t\tinputSchema: tool.inputSchema,\n\t\t\t\t\t};\n\t\t\t\t});\n\t\t\t\tloggers.mcp.info(\"Connected to MCP server with tools:\", {\n\t\t\t\t\ttools: conn.tools.map((tool) => tool.toolName),\n\t\t\t\t});\n\t\t\t} catch (error) {\n\t\t\t\tloggers.mcp.error(`Failed to connect to MCP server ${name}`, { error });\n\t\t\t}\n\t\t}\n\t}\n\n\t/**\n\t * Returns all available tools from connected MCP servers.\n\t *\n\t * @returns Array of MCPTool instances representing available tools\n\t */\n\tgetTools(): Array<ConnectorTool> {\n\t\tconst allTools: Array<ConnectorTool> = [];\n\t\tfor (const conn of this.mcpConnectors.values()) {\n\t\t\tallTools.push(...conn.tools);\n\t\t}\n\t\treturn allTools;\n\t}\n\n\t/**\n\t * Executes a tool on its corresponding MCP server.\n\t *\n\t * @param tool - The MCPTool instance to execute\n\t * @param _args - Arguments to pass to the tool\n\t * @returns Promise resolving to the tool's execution result\n\t * @throws Error if the MCP server for the tool is not found\n\t */\n\tasync useTool(tool: ConnectorTool, _args?: any): Promise<string> {\n\t\tconst { connectorName, toolName } = tool;\n\t\tconst client = this.mcpConnectors.get(connectorName)?.client;\n\n\t\ttry {\n\t\t\tif (!client) {\n\t\t\t\tthrow new Error(`Invalid MCP Tool ${toolName}`);\n\t\t\t}\n\n\t\t\t// `${name}-${tool.name}` => tool.name\n\t\t\tconst mcpToolName = toolName.slice(connectorName.length + 1);\n\t\t\tconst result = await client.callTool({\n\t\t\t\tname: mcpToolName,\n\t\t\t\targuments: _args,\n\t\t\t});\n\t\t\tconst toolResult =\n\t\t\t\t`[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n` +\n\t\t\t\tJSON.stringify(result.content, null, 2);\n\t\t\treturn toolResult;\n\t\t} catch (error) {\n\t\t\tloggers.mcp.error(\"Failed to call tool\", { error });\n\t\t\tconst toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\\n${typeof error === \"string\" ? error : JSON.stringify(error, null, 2)}`;\n\t\t\treturn toolResult;\n\t\t}\n\t}\n\n\t/**\n\t * Closes all MCP client connections.\n\t *\n\t * Should be called when shutting down the application to ensure\n\t * all MCP connections are properly closed.\n\t */\n\tasync cleanup() {\n\t\tfor (const conn of this.mcpConnectors.values()) {\n\t\t\tawait conn.client?.close();\n\t\t}\n\t}\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,UAAU,iBAAiB;AACpC,SAAS,0BAA0B;AACnC,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;AAC9C,SAAS,gCAAgC;AAgBlC,IAAM,YAAN,MAAgB;AAAA,EACd,gBAA2C,oBAAI,IAAI;AAAA,EAE3D,gBAAgB,SAA8C;AAC7D,eAAW,CAAC,MAAM,MAAM,KAAK,OAAO,QAAQ,OAAO,GAAG;AACrD,YAAM,OAAO,IAAI,aAAa,MAAM,MAAM;AAC1C,WAAK,cAAc,IAAI,MAAM,IAAI;AAAA,IAClC;AAAA,EACD;AAAA,EAEQ,kBAAkB,WAAoC;AAC7D,cAAU,WAAW,IAAI,UAAU;AAAA,MAClC,MAAM,UAAU;AAAA,MAChB,SAAS;AAAA,IACV,CAAC;AACD,WAAO,UAAU;AAAA,EAClB;AAAA,EAEA,MAAM,mBAAkC;AACvC,eAAW,CAAC,MAAM,IAAI,KAAK,KAAK,cAAc,QAAQ,GAAG;AACxD,UAAI;AACH,cAAM,YAAY,KAAK,kBAAkB,IAAI;AAC7C,cAAM,SAAS,KAAK;AACpB,gBAAQ,OAAO,MAAM;AAAA,UACpB,KAAK,SAAS;AACb,kBAAM,YAAY,IAAI,qBAAqB,OAAO,MAAM;AACxD,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,aAAa;AACjB,kBAAM,YAAY,IAAI,yBAAyB,OAAO,GAAG;AACzD,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,OAAO;AACX,kBAAM,YAAY,IAAI;AAAA,cACrB,OAAO;AAAA,cACP,OAAO;AAAA,YACR;AACA,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA,KAAK,kBAAkB;AACtB,kBAAM,YAAY,IAAI;AAAA,cACrB,OAAO;AAAA,cACP,OAAO;AAAA,YACR;AACA,kBAAM,UAAU,QAAQ,SAAS;AACjC;AAAA,UACD;AAAA,UACA;AAEC,oBAAQ,IAAI,MAAM,6BAA6B;AAC/C;AAAA,QACF;AAEA,cAAM,WAAW,MAAM,UAAU,UAAU;AAC3C,aAAK,QAAQ,SAAS,MAAM,IAAI,CAAC,SAAS;AACzC,iBAAO;AAAA,YACN,UAAU,GAAG,IAAI,IAAI,KAAK,IAAI;AAAA;AAAA,YAC9B,eAAe;AAAA,YACf;AAAA,YACA,aAAa,KAAK;AAAA,YAClB,aAAa,KAAK;AAAA,UACnB;AAAA,QACD,CAAC;AACD,gBAAQ,IAAI,KAAK,uCAAuC;AAAA,UACvD,OAAO,KAAK,MAAM,IAAI,CAAC,SAAS,KAAK,QAAQ;AAAA,QAC9C,CAAC;AAAA,MACF,SAAS,OAAO;AACf,gBAAQ,IAAI,MAAM,mCAAmC,IAAI,IAAI,EAAE,MAAM,CAAC;AAAA,MACvE;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,WAAiC;AAChC,UAAM,WAAiC,CAAC;AACxC,eAAW,QAAQ,KAAK,cAAc,OAAO,GAAG;AAC/C,eAAS,KAAK,GAAG,KAAK,KAAK;AAAA,IAC5B;AACA,WAAO;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,QAAQ,MAAqB,OAA8B;AAChE,UAAM,EAAE,eAAe,SAAS,IAAI;AACpC,UAAM,SAAS,KAAK,cAAc,IAAI,aAAa,GAAG;AAEtD,QAAI;AACH,UAAI,CAAC,QAAQ;AACZ,cAAM,IAAI,MAAM,oBAAoB,QAAQ,EAAE;AAAA,MAC/C;AAGA,YAAM,cAAc,SAAS,MAAM,cAAc,SAAS,CAAC;AAC3D,YAAM,SAAS,MAAM,OAAO,SAAS;AAAA,QACpC,MAAM;AAAA,QACN,WAAW;AAAA,MACZ,CAAC;AACD,YAAM,aACL,oBAAoB,QAAQ,cAAc,KAAK,UAAU,KAAK,CAAC;AAAA,IAC/D,KAAK,UAAU,OAAO,SAAS,MAAM,CAAC;AACvC,aAAO;AAAA,IACR,SAAS,OAAO;AACf,cAAQ,IAAI,MAAM,uBAAuB,EAAE,MAAM,CAAC;AAClD,YAAM,aAAa,oBAAoB,QAAQ,cAAc,KAAK,UAAU,KAAK,CAAC;AAAA,EAAM,OAAO,UAAU,WAAW,QAAQ,KAAK,UAAU,OAAO,MAAM,CAAC,CAAC;AAC1J,aAAO;AAAA,IACR;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,UAAU;AACf,eAAW,QAAQ,KAAK,cAAc,OAAO,GAAG;AAC/C,YAAM,KAAK,QAAQ,MAAM;AAAA,IAC1B;AAAA,EACD;AACD;","names":[]}

package/dist/esm/modules/auth/base.auth.d.ts

@@ -1,8 +0,0 @@
-import { Request, Response } from 'express';
-import { AuthResponse } from '../../types/auth.js';
-
-declare abstract class BaseAuth {
-    abstract authenticate(req: Request, res: Response): Promise<AuthResponse>;
-}
-
-export { BaseAuth };

package/dist/esm/modules/auth/base.auth.js

@@ -1,7 +0,0 @@
-import {
-  BaseAuth
-} from "../../chunk-FF5RJ4L2.js";
-export {
-  BaseAuth
-};
-//# sourceMappingURL=base.auth.js.map

package/dist/esm/services/intents/fulfill-stream.service.d.ts

@@ -1,50 +0,0 @@
-import { A2AModule } from '../../modules/a2a/a2a.module.js';
-import { MCPModule } from '../../modules/mcp/mcp.module.js';
-import { TriggeredIntent, ThreadObject } from '../../types/memory.js';
-import { MemoryModule } from '../../modules/memory/memory.module.js';
-import { StreamEvent } from '../../types/stream.js';
-import { ModelModule } from '../../modules/models/model.module.js';
-import '@a2a-js/sdk';
-import '../../types/connector.js';
-import '../../types/mcp.js';
-import '@modelcontextprotocol/sdk/client/sse.js';
-import '@modelcontextprotocol/sdk/client/stdio.js';
-import '@modelcontextprotocol/sdk/client/streamableHttp.js';
-import '../../modules/memory/base.memory.js';
-import '../../modules/models/base.model.js';
-
-declare class IntentFulfillStreamService {
-    private modelModule;
-    private a2aModule?;
-    private mcpModule?;
-    private memoryModule?;
-    constructor(modelModule: ModelModule, a2aModule?: A2AModule, mcpModule?: MCPModule, memoryModule?: MemoryModule);
-    private addToThreadMessages;
-    /**
-     * Fulfills the detected intent by generating a streaming response.
-     *
-     * Manages the complete inference loop including:
-     * - Loading prompts and conversation history
-     * - Collecting available tools from modules
-     * - Executing model inference with tool support
-     * - Processing tool calls iteratively until completion
-     * - Streaming results as Server-Sent Events
-     *
-     * @param query - The user's input query
-     * @param threadId - Thread identifier for context
-     * @param thread - Previous conversation history
-     * @param intent - Optional detected intent with custom prompt
-     * @returns AsyncGenerator yielding StreamEvent objects
-     */
-    private intentFulfilling;
-    /**
-     * Detects the intent from context.
-     *
-     * @param intents - The user's input query
-     * @param thread - The thread history
-     * @returns The detected intent
-     */
-    intentFulfillStream(intents: Array<TriggeredIntent>, thread: ThreadObject): AsyncGenerator<StreamEvent>;
-}
-
-export { IntentFulfillStreamService };

package/dist/esm/services/intents/fulfill-stream.service.js

@@ -1,11 +0,0 @@
-import {
-  IntentFulfillStreamService
-} from "../../chunk-V24ATQLM.js";
-import "../../chunk-7BH6AYP7.js";
-import "../../chunk-7CULQDJF.js";
-import "../../chunk-GRN6CHNO.js";
-import "../../chunk-N7VYM2R5.js";
-export {
-  IntentFulfillStreamService
-};
-//# sourceMappingURL=fulfill-stream.service.js.map

package/dist/esm/services/query-stream.service.d.ts

@@ -1,53 +0,0 @@
-import { A2AModule } from '../modules/a2a/a2a.module.js';
-import { MCPModule } from '../modules/mcp/mcp.module.js';
-import { ThreadType } from '../types/memory.js';
-import { MemoryModule } from '../modules/memory/memory.module.js';
-import { ModelFetchOptions } from '../modules/models/base.model.js';
-import { ModelModule } from '../modules/models/model.module.js';
-import { StreamEvent } from '../types/stream.js';
-import '@a2a-js/sdk';
-import '../types/connector.js';
-import '../types/mcp.js';
-import '@modelcontextprotocol/sdk/client/sse.js';
-import '@modelcontextprotocol/sdk/client/stdio.js';
-import '@modelcontextprotocol/sdk/client/streamableHttp.js';
-import '../modules/memory/base.memory.js';
-
-/**
- * Service for processing user queries through the agent's AI pipeline.
- *
- * Orchestrates the query processing workflow including intent detection,
- * model inference, tool execution, and response generation. Manages
- * conversation context and coordinates between different modules.
- */
-declare class QueryStreamService {
-    private modelModule;
-    private memoryModule?;
-    private intentTriggerService;
-    private intentFulfillStreamService;
-    constructor(modelModule: ModelModule, a2aModule?: A2AModule, mcpModule?: MCPModule, memoryModule?: MemoryModule);
-    /**
-     * Main entry point for processing streaming user queries.
-     *
-     * Handles the complete query lifecycle:
-     * 1. Loads or creates thread from memory
-     * 2. Detects intent from the query
-     * 3. Fulfills the intent with streaming AI response
-     * 4. Updates conversation history in real-time
-     *
-     * @param threadMetadata - Metadata containing type, userId, and optional threadId
-     * @param threadMetadata.type - The type of thread (e.g., chat, workflow)
-     * @param threadMetadata.userId - The user's unique identifier
-     * @param threadMetadata.threadId - Optional thread identifier
-     * @param query - The user's input query
-     * @returns AsyncGenerator yielding StreamEvent objects for SSE
-     */
-    handleQueryStream(threadMetadata: {
-        type: ThreadType;
-        userId: string;
-        threadId?: string;
-        options?: ModelFetchOptions;
-    }, query: string, isA2A?: boolean): AsyncGenerator<StreamEvent>;
-}
-
-export { QueryStreamService };

package/dist/esm/services/query-stream.service.js

@@ -1,15 +0,0 @@
-import {
-  QueryStreamService
-} from "../chunk-RJZ4QSDX.js";
-import "../chunk-V24ATQLM.js";
-import "../chunk-DIW5VDAQ.js";
-import "../chunk-7BH6AYP7.js";
-import "../chunk-J2UFYVKK.js";
-import "../chunk-WUNZG3WP.js";
-import "../chunk-7CULQDJF.js";
-import "../chunk-GRN6CHNO.js";
-import "../chunk-N7VYM2R5.js";
-export {
-  QueryStreamService
-};
-//# sourceMappingURL=query-stream.service.js.map