@ainetwork/adk 0.0.1 → 0.1.0

package/dist/esm/modules/mcp/mcp.module.js

@@ -1,114 +1,10 @@
-import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
-import { loggers } from "../../utils/logger.js";
-import { MCPTool } from "./mcp.tool.js";
-/**
- * Module for managing Model Context Protocol (MCP) server connections.
- *
- * This module handles the lifecycle of MCP client connections, discovers
- * available tools from connected servers, and provides an interface for
- * executing those tools. Multiple MCP servers can be connected simultaneously.
- *
- * @example
- * ```typescript
- * const mcpModule = new MCPModule();
- * await mcpModule.addMCPConfig({
- *   "filesystem": {
- *     command: "npx",
- *     args: ["@modelcontextprotocol/server-filesystem", "/path/to/files"]
- *   }
- * });
- *
- * const tools = mcpModule.getTools();
- * const result = await mcpModule.useTool(tools[0], { path: "/example.txt" });
- * ```
- */
-export class MCPModule {
-    constructor() {
-        /** Map of MCP server names to their client instances */
-        this.mcpMap = new Map();
-        /** Map of MCP server names to their transport instances */
-        this.transportMap = new Map();
-        /** Array of all discovered tools from connected MCP servers */
-        this.tools = [];
-    }
-    /**
-     * Connects to MCP servers based on the provided configuration.
-     *
-     * For each server in the config, establishes a connection, discovers
-     * available tools, and adds them to the module's tool collection.
-     * Skips servers that are already connected.
-     *
-     * @param mcpConfig - Configuration object mapping server names to connection details
-     * @throws Error if connection to any MCP server fails
-     */
-    async addMCPConfig(mcpConfig) {
-        try {
-            for (const [name, conf] of Object.entries(mcpConfig)) {
-                // FIXME(yoojin): Need strict duplication check.
-                if (this.mcpMap.get(name) && this.transportMap.get(name))
-                    continue; // Duplicated mcp: skip
-                const transport = new StdioClientTransport(conf);
-                this.transportMap.set(name, transport);
-                const mcp = new Client({ name: "mcp-client-cli", version: "1.0.0" });
-                await mcp.connect(transport);
-                this.mcpMap.set(name, mcp);
-                const toolsResult = await mcp.listTools();
-                this.tools.push(...toolsResult.tools.map((tool) => {
-                    return new MCPTool(name, tool);
-                }));
-            }
-            loggers.mcp.info("Connected to MCP server with tools:", {
-                tools: this.tools.map((tool) => tool.id),
-            });
-        }
-        catch (error) {
-            loggers.mcp.error("Failed to connect to MCP server:", { error });
-            throw error;
-        }
-    }
-    /**
-     * Returns all available tools from connected MCP servers.
-     *
-     * @returns Array of MCPTool instances representing available tools
-     */
-    getTools() {
-        return this.tools;
-    }
-    /**
-     * 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 { serverName, mcpTool } = tool;
-        const toolName = mcpTool.name;
-        const mcp = this.mcpMap.get(serverName);
-        if (!mcp) {
-            throw new Error(`Invalid MCP Tool ${serverName}-${mcpTool.name}`);
-        }
-        const result = await mcp.callTool({
-            name: toolName,
-            arguments: _args,
-        });
-        const toolResult = `[Bot Called Tool ${toolName} with args ${JSON.stringify(_args)}]\n` +
-            JSON.stringify(result.content, null, 2);
-        loggers.mcp.debug("MCP useTool result:", toolResult);
-        return result;
-    }
-    /**
-     * Closes all MCP client connections.
-     *
-     * Should be called when shutting down the application to ensure
-     * all MCP connections are properly closed.
-     */
-    async cleanup() {
-        this.mcpMap.forEach((mcp) => {
-            mcp.close();
-        });
-    }
-}
+import {
+  MCPModule
+} from "../../chunk-BTZVK3EA.js";
+import "../../chunk-3LLHEUY2.js";
+import "../../chunk-FI2UQLVC.js";
+import "../../chunk-64GEXWSE.js";
+export {
+  MCPModule
+};
 //# sourceMappingURL=mcp.module.js.map

package/dist/esm/modules/mcp/mcp.module.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp.module.js","sourceRoot":"","sources":["../../../../src/modules/mcp/mcp.module.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAEjF,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAExC;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,OAAO,SAAS;IAAtB;QACC,wDAAwD;QAChD,WAAM,GAAwB,IAAI,GAAG,EAAE,CAAC;QAChD,2DAA2D;QACnD,iBAAY,GAAsC,IAAI,GAAG,EAAE,CAAC;QACpE,+DAA+D;QACvD,UAAK,GAAc,EAAE,CAAC;IAyF/B,CAAC;IAvFA;;;;;;;;;OASG;IACH,KAAK,CAAC,YAAY,CAAC,SAAoB;QACtC,IAAI,CAAC;YACJ,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;gBACtD,gDAAgD;gBAChD,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;oBAAE,SAAS,CAAC,uBAAuB;gBAE3F,MAAM,SAAS,GAAG,IAAI,oBAAoB,CAAC,IAAI,CAAC,CAAC;gBACjD,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;gBACvC,MAAM,GAAG,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBACrE,MAAM,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC7B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;gBAE3B,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;gBAC1C,IAAI,CAAC,KAAK,CAAC,IAAI,CACd,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;oBACjC,OAAO,IAAI,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAChC,CAAC,CAAC,CACF,CAAC;YACH,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,qCAAqC,EAAE;gBACvD,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;aACxC,CAAC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACzB,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACjE,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAED;;;;OAIG;IACH,QAAQ;QACP,OAAO,IAAI,CAAC,KAAK,CAAC;IACnB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,OAAO,CAAC,IAAa,EAAE,KAAW;QACvC,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;QACrC,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAExC,IAAI,CAAC,GAAG,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,oBAAoB,UAAU,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC;YACjC,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,KAAK;SAChB,CAAC,CAAC;QACH,MAAM,UAAU,GACf,oBAAoB,QAAQ,cAAc,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK;YACpE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAEzC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,EAAE,UAAU,CAAC,CAAC;QACrD,OAAO,MAAM,CAAC;IACf,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,OAAO;QACZ,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAW,EAAE,EAAE;YACnC,GAAG,CAAC,KAAK,EAAE,CAAC;QACb,CAAC,CAAC,CAAC;IACJ,CAAC;CACD"}
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/esm/modules/mcp/mcp.tool.d.ts

@@ -1,6 +1,9 @@
-import type { Tool } from "@modelcontextprotocol/sdk/types.js";
-import { type IMCPTool, TOOL_PROTOCOL_TYPE } from "../../types/tool.js";
-export declare class MCPTool implements IMCPTool {
+import { Tool } from '@modelcontextprotocol/sdk/types.js';
+import { IMCPTool, TOOL_PROTOCOL_TYPE } from '../../types/tool.js';
+import '@a2a-js/sdk';
+import '@a2a-js/sdk/client';
+
+declare class MCPTool implements IMCPTool {
     id: string;
     protocol: TOOL_PROTOCOL_TYPE;
     enabled: boolean;
@@ -10,4 +13,5 @@ export declare class MCPTool implements IMCPTool {
     enable(): void;
     disable(): void;
 }
-//# sourceMappingURL=mcp.tool.d.ts.map
+
+export { MCPTool };

package/dist/esm/modules/mcp/mcp.tool.js

@@ -1,18 +1,8 @@
-import { TOOL_PROTOCOL_TYPE } from "../../types/tool.js";
-export class MCPTool {
-    constructor(serverName, tool) {
-        this.id = `${serverName}_${tool.name}`;
-        this.protocol = TOOL_PROTOCOL_TYPE.MCP;
-        this.enabled = true;
-        // NOTE(yoojin): MCP Server name. ex) notionApi
-        this.serverName = serverName;
-        this.mcpTool = tool;
-    }
-    enable() {
-        this.enabled = true;
-    }
-    disable() {
-        this.enabled = false;
-    }
-}
+import {
+  MCPTool
+} from "../../chunk-3LLHEUY2.js";
+import "../../chunk-FI2UQLVC.js";
+export {
+  MCPTool
+};
 //# sourceMappingURL=mcp.tool.js.map

package/dist/esm/modules/mcp/mcp.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp.tool.js","sourceRoot":"","sources":["../../../../src/modules/mcp/mcp.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAiB,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAEpE,MAAM,OAAO,OAAO;IAOnB,YAAY,UAAkB,EAAE,IAAU;QACzC,IAAI,CAAC,EAAE,GAAG,GAAG,UAAU,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,QAAQ,GAAG,kBAAkB,CAAC,GAAG,CAAC;QACvC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,+CAA+C;QAC/C,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;IACrB,CAAC;IAEM,MAAM;QACZ,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;IACrB,CAAC;IAEM,OAAO;QACb,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;IACtB,CAAC;CACD"}
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/{cjs/modules/memory/memory.module.d.ts → esm/modules/memory/base.memory.d.ts}

@@ -1,4 +1,5 @@
-import type { ChatObject, SessionObject } from "../../types/memory.js";
+import { SessionObject, ChatObject } from '../../types/memory.js';
+
 /**
  * Abstract base class for memory storage implementations.
  *
@@ -6,7 +7,7 @@ import type { ChatObject, SessionObject } from "../../types/memory.js";
  * and query-intent pairs. Implementations can use various backends
  * such as in-memory storage, databases, or file systems.
  */
-export declare abstract class BaseMemory {
+declare abstract class BaseMemory {
     /**
      * Retrieves the conversation history for a specific session.
      *
@@ -30,36 +31,5 @@ export declare abstract class BaseMemory {
      */
     abstract storeQueryAndIntent(query: string, intent: string, sessionId: string): Promise<void>;
 }
-/**
- * Module wrapper for memory storage implementations.
- *
- * Provides a consistent interface for accessing memory storage
- * functionality throughout the application. Currently supports
- * a single memory implementation but designed for future extensibility.
- *
- * @example
- * ```typescript
- * const inMemoryStorage = new InMemoryStorage();
- * const memoryModule = new MemoryModule(inMemoryStorage);
- *
- * const memory = memoryModule.getMemory();
- * await memory.updateSessionHistory("session-123", chatMessage);
- * ```
- */
-export declare class MemoryModule {
-    /** The memory storage implementation */
-    private memory;
-    /**
-     * Creates a new MemoryModule with the specified storage implementation.
-     *
-     * @param memory - The memory storage implementation to use
-     */
-    constructor(memory: BaseMemory);
-    /**
-     * Returns the current memory storage implementation.
-     *
-     * @returns The active memory storage instance
-     */
-    getMemory(): BaseMemory;
-}
-//# sourceMappingURL=memory.module.d.ts.map
+
+export { BaseMemory };

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

@@ -0,0 +1,7 @@
+import {
+  BaseMemory
+} from "../../chunk-KUMCCCED.js";
+export {
+  BaseMemory
+};
+//# sourceMappingURL=base.memory.js.map

package/dist/esm/modules/memory/base.memory.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/esm/modules/memory/memory.module.d.ts

@@ -1,35 +1,6 @@
-import type { ChatObject, SessionObject } from "../../types/memory.js";
-/**
- * Abstract base class for memory storage implementations.
- *
- * Provides an interface for storing and retrieving conversation history
- * and query-intent pairs. Implementations can use various backends
- * such as in-memory storage, databases, or file systems.
- */
-export declare abstract class BaseMemory {
-    /**
-     * Retrieves the conversation history for a specific session.
-     *
-     * @param sessionId - Unique identifier for the session
-     * @returns Promise resolving to the session's conversation history
-     */
-    abstract getSessionHistory(sessionId: string): Promise<SessionObject>;
-    /**
-     * Adds a new chat message to the session history.
-     *
-     * @param sessionId - Unique identifier for the session
-     * @param chat - The chat object to add to the history
-     */
-    abstract updateSessionHistory(sessionId: string, chat: ChatObject): Promise<void>;
-    /**
-     * Stores a query-intent pair for analysis or training purposes.
-     *
-     * @param query - The user's input query
-     * @param intent - The detected intent for the query
-     * @param sessionId - Unique identifier for the session
-     */
-    abstract storeQueryAndIntent(query: string, intent: string, sessionId: string): Promise<void>;
-}
+import { MemoryType } from '../../types/memory.js';
+import { BaseMemory } from './base.memory.js';
+
 /**
  * Module wrapper for memory storage implementations.
  *
@@ -46,9 +17,9 @@ export declare abstract class BaseMemory {
  * await memory.updateSessionHistory("session-123", chatMessage);
  * ```
  */
-export declare class MemoryModule {
+declare class MemoryModule {
     /** The memory storage implementation */
-    private memory;
+    private memoryMap;
     /**
      * Creates a new MemoryModule with the specified storage implementation.
      *
@@ -60,6 +31,7 @@ export declare class MemoryModule {
      *
      * @returns The active memory storage instance
      */
-    getMemory(): BaseMemory;
+    getMemory(type?: MemoryType): BaseMemory | undefined;
 }
-//# sourceMappingURL=memory.module.d.ts.map
+
+export { MemoryModule };

package/dist/esm/modules/memory/memory.module.js

@@ -1,45 +1,8 @@
-/**
- * Abstract base class for memory storage implementations.
- *
- * Provides an interface for storing and retrieving conversation history
- * and query-intent pairs. Implementations can use various backends
- * such as in-memory storage, databases, or file systems.
- */
-export class BaseMemory {
-}
-/**
- * Module wrapper for memory storage implementations.
- *
- * Provides a consistent interface for accessing memory storage
- * functionality throughout the application. Currently supports
- * a single memory implementation but designed for future extensibility.
- *
- * @example
- * ```typescript
- * const inMemoryStorage = new InMemoryStorage();
- * const memoryModule = new MemoryModule(inMemoryStorage);
- *
- * const memory = memoryModule.getMemory();
- * await memory.updateSessionHistory("session-123", chatMessage);
- * ```
- */
-export class MemoryModule {
-    /**
-     * Creates a new MemoryModule with the specified storage implementation.
-     *
-     * @param memory - The memory storage implementation to use
-     */
-    constructor(memory) {
-        this.memory = memory;
-    }
-    /**
-     * Returns the current memory storage implementation.
-     *
-     * @returns The active memory storage instance
-     */
-    getMemory() {
-        // TODO: Support multi-memory for each type of memory?
-        return this.memory;
-    }
-}
+import {
+  MemoryModule
+} from "../../chunk-XPJK5O3I.js";
+import "../../chunk-TMHRCGTE.js";
+export {
+  MemoryModule
+};
 //# sourceMappingURL=memory.module.js.map

package/dist/esm/modules/memory/memory.module.js.map

@@ -1 +1 @@
-{"version":3,"file":"memory.module.js","sourceRoot":"","sources":["../../../../src/modules/memory/memory.module.ts"],"names":[],"mappings":"AAEA;;;;;;GAMG;AACH,MAAM,OAAgB,UAAU;CA8B/B;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,OAAO,YAAY;IAIxB;;;;OAIG;IACH,YAAY,MAAkB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACI,SAAS;QACf,sDAAsD;QACtD,OAAO,IAAI,CAAC,MAAM,CAAC;IACpB,CAAC;CACD"}
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/{cjs/modules/models/model.module.d.ts → esm/modules/models/base.model.d.ts}

@@ -1,5 +1,9 @@
-import type { SessionObject } from "../../types/memory.js";
-import type { FetchResponse, IAgentTool } from "../../types/tool.js";
+import { SessionObject } from '../../types/memory.js';
+import { IAgentTool, FetchResponse } from '../../types/tool.js';
+import '@a2a-js/sdk';
+import '@a2a-js/sdk/client';
+import '@modelcontextprotocol/sdk/types.js';
+
 /**
  * Abstract base class for AI model implementations.
  *
@@ -10,7 +14,7 @@ import type { FetchResponse, IAgentTool } from "../../types/tool.js";
  * @typeParam MessageType - The message format used by the specific model API
  * @typeParam FunctionType - The function/tool format used by the specific model API
  */
-export declare abstract class BaseModel<MessageType, FunctionType> {
+declare abstract class BaseModel<MessageType, FunctionType> {
     /**
      * Generates an array of messages from session history and current query.
      *
@@ -19,7 +23,11 @@ export declare abstract class BaseModel<MessageType, FunctionType> {
      * @param systemPrompt - Optional system prompt to set context
      * @returns Array of messages formatted for the specific model API
      */
-    abstract generateMessages(sessionHistory: SessionObject, query: string, systemPrompt?: string): MessageType[];
+    abstract generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): MessageType[];
     /**
      * Appends a new message to the existing message array.
      *
@@ -50,53 +58,5 @@ export declare abstract class BaseModel<MessageType, FunctionType> {
      */
     abstract fetchWithContextMessage(messages: MessageType[], functions: FunctionType[]): Promise<FetchResponse>;
 }
-/**
- * Module for managing multiple AI model implementations.
- *
- * Allows registration of multiple models and provides a unified interface
- * for accessing them. Supports setting a default model for convenience.
- *
- * @example
- * ```typescript
- * const modelModule = new ModelModule();
- * modelModule.addModel("gpt-4", new OpenAIModel("gpt-4"), true);
- * modelModule.addModel("gemini", new GeminiModel("gemini-pro"));
- *
- * const defaultModel = modelModule.getModel();
- * const specificModel = modelModule.getModel("gemini");
- * ```
- */
-export declare class ModelModule {
-    /** Registry of available models indexed by name */
-    private models;
-    /** Name of the default model to use when none specified */
-    private defaultModelName?;
-    /**
-     * Registers a new model with the module.
-     *
-     * @param name - Unique identifier for the model
-     * @param model - Model instance implementing BaseModel
-     * @param isDefault - Whether to set this as the default model
-     */
-    addModel(name: string, model: BaseModel<unknown, unknown>, isDefault?: boolean): void;
-    /**
-     * Retrieves a model by name or returns the default model.
-     *
-     * @param name - Optional model name to retrieve
-     * @returns The requested model or default model if name not provided
-     * @throws Error if no default model is set and name is not provided
-     */
-    getModel(name?: string): BaseModel<unknown, unknown>;
-    /**
-     * Returns information about all registered models.
-     *
-     * @returns Object containing all models and the default model name
-     */
-    getModelList(): {
-        models: {
-            [name: string]: BaseModel<unknown, unknown>;
-        };
-        defaultModelName: string | undefined;
-    };
-}
-//# sourceMappingURL=model.module.d.ts.map
+
+export { BaseModel };

package/dist/esm/modules/models/base.model.js

@@ -0,0 +1,7 @@
+import {
+  BaseModel
+} from "../../chunk-GZFRRFOE.js";
+export {
+  BaseModel
+};
+//# sourceMappingURL=base.model.js.map

package/dist/esm/modules/models/base.model.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/esm/modules/models/model.module.d.ts

@@ -1,72 +1,11 @@
-import type { SessionObject } from "../../types/memory.js";
-import type { FetchResponse, IAgentTool } from "../../types/tool.js";
-/**
- * Abstract base class for AI model implementations.
- *
- * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)
- * to integrate with the AIN-ADK framework. Each model implementation must handle
- * message formatting, tool conversion, and API communication.
- *
- * @typeParam MessageType - The message format used by the specific model API
- * @typeParam FunctionType - The function/tool format used by the specific model API
- */
-export declare abstract class BaseModel<MessageType, FunctionType> {
-    /**
-     * Generates an array of messages from session history and current query.
-     *
-     * @param sessionHistory - Previous conversation history
-     * @param query - Current user query
-     * @param systemPrompt - Optional system prompt to set context
-     * @returns Array of messages formatted for the specific model API
-     */
-    abstract generateMessages(sessionHistory: SessionObject, query: string, systemPrompt?: string): MessageType[];
-    /**
-     * Appends a new message to the existing message array.
-     *
-     * @param messages - Existing message array to expand
-     * @param message - New message content to append
-     */
-    abstract appendMessages(messages: MessageType[], message: string): void;
-    /**
-     * Converts protocol-agnostic tools to model-specific function format.
-     *
-     * @param tools - Array of agent tools from MCP or A2A sources
-     * @returns Array of functions in the format required by the model API
-     */
-    abstract convertToolsToFunctions(tools: IAgentTool[]): FunctionType[];
-    /**
-     * Fetches a response from the model API without tool support.
-     *
-     * @param messages - Array of messages to send to the model
-     * @returns Promise resolving to the model's response
-     */
-    abstract fetch(messages: MessageType[]): Promise<FetchResponse>;
-    /**
-     * Fetches a response from the model API with tool/function support.
-     *
-     * @param messages - Array of messages to send to the model
-     * @param functions - Array of available functions/tools the model can call
-     * @returns Promise resolving to the model's response, possibly including tool calls
-     */
-    abstract fetchWithContextMessage(messages: MessageType[], functions: FunctionType[]): Promise<FetchResponse>;
-}
-/**
- * Module for managing multiple AI model implementations.
- *
- * Allows registration of multiple models and provides a unified interface
- * for accessing them. Supports setting a default model for convenience.
- *
- * @example
- * ```typescript
- * const modelModule = new ModelModule();
- * modelModule.addModel("gpt-4", new OpenAIModel("gpt-4"), true);
- * modelModule.addModel("gemini", new GeminiModel("gemini-pro"));
- *
- * const defaultModel = modelModule.getModel();
- * const specificModel = modelModule.getModel("gemini");
- * ```
- */
-export declare class ModelModule {
+import { BaseModel } from './base.model.js';
+import '../../types/memory.js';
+import '../../types/tool.js';
+import '@a2a-js/sdk';
+import '@a2a-js/sdk/client';
+import '@modelcontextprotocol/sdk/types.js';
+
+declare class ModelModule {
     /** Registry of available models indexed by name */
     private models;
     /** Name of the default model to use when none specified */
@@ -93,10 +32,9 @@ export declare class ModelModule {
      * @returns Object containing all models and the default model name
      */
     getModelList(): {
-        models: {
-            [name: string]: BaseModel<unknown, unknown>;
-        };
+        models: string[];
         defaultModelName: string | undefined;
     };
 }
-//# sourceMappingURL=model.module.d.ts.map
+
+export { ModelModule };

package/dist/esm/modules/models/model.module.js

@@ -1,75 +1,7 @@
-/**
- * Abstract base class for AI model implementations.
- *
- * Provides a common interface for different AI model providers (OpenAI, Gemini, etc.)
- * to integrate with the AIN-ADK framework. Each model implementation must handle
- * message formatting, tool conversion, and API communication.
- *
- * @typeParam MessageType - The message format used by the specific model API
- * @typeParam FunctionType - The function/tool format used by the specific model API
- */
-export class BaseModel {
-}
-/**
- * Module for managing multiple AI model implementations.
- *
- * Allows registration of multiple models and provides a unified interface
- * for accessing them. Supports setting a default model for convenience.
- *
- * @example
- * ```typescript
- * const modelModule = new ModelModule();
- * modelModule.addModel("gpt-4", new OpenAIModel("gpt-4"), true);
- * modelModule.addModel("gemini", new GeminiModel("gemini-pro"));
- *
- * const defaultModel = modelModule.getModel();
- * const specificModel = modelModule.getModel("gemini");
- * ```
- */
-export class ModelModule {
-    constructor() {
-        /** Registry of available models indexed by name */
-        this.models = {};
-    }
-    /**
-     * Registers a new model with the module.
-     *
-     * @param name - Unique identifier for the model
-     * @param model - Model instance implementing BaseModel
-     * @param isDefault - Whether to set this as the default model
-     */
-    addModel(name, model, isDefault) {
-        this.models[name] = model;
-        if (isDefault || !this.defaultModelName) {
-            this.defaultModelName = name;
-        }
-    }
-    /**
-     * Retrieves a model by name or returns the default model.
-     *
-     * @param name - Optional model name to retrieve
-     * @returns The requested model or default model if name not provided
-     * @throws Error if no default model is set and name is not provided
-     */
-    getModel(name) {
-        if (!this.defaultModelName) {
-            throw Error("No default model");
-        }
-        if (!name || !this.models[name]) {
-            return this.models[this.defaultModelName];
-        }
-        return this.models[name];
-    }
-    /**
-     * Returns information about all registered models.
-     *
-     * @returns Object containing all models and the default model name
-     */
-    getModelList() {
-        return {
-            models: this.models,
-            defaultModelName: this.defaultModelName,
-        };
-    }
-}
+import {
+  ModelModule
+} from "../../chunk-PHKK5NHK.js";
+export {
+  ModelModule
+};
 //# sourceMappingURL=model.module.js.map

package/dist/esm/modules/models/model.module.js.map

@@ -1 +1 @@
-{"version":3,"file":"model.module.js","sourceRoot":"","sources":["../../../../src/modules/models/model.module.ts"],"names":[],"mappings":"AAGA;;;;;;;;;GASG;AACH,MAAM,OAAgB,SAAS;CAkD9B;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,OAAO,WAAW;IAAxB;QACC,mDAAmD;QAC3C,WAAM,GAAoD,EAAE,CAAC;IAmDtE,CAAC;IA/CA;;;;;;OAMG;IACI,QAAQ,CACd,IAAY,EACZ,KAAkC,EAClC,SAAmB;QAEnB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;QAC1B,IAAI,SAAS,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACzC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC9B,CAAC;IACF,CAAC;IAED;;;;;;OAMG;IACI,QAAQ,CAAC,IAAa;QAC5B,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC5B,MAAM,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACjC,CAAC;QAED,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACjC,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;;;OAIG;IACI,YAAY;QAClB,OAAO;YACN,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACvC,CAAC;IACH,CAAC;CACD"}
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}