@ainetwork/adk 0.0.1 → 0.0.2

package/dist/esm/modules/memory/providers/mongodb/index.js

@@ -0,0 +1,84 @@
+import mongoose from "mongoose";
+import { loggers } from "../../../../utils/logger.js";
+import { BaseMemory } from "../../base/index.js";
+import { ChatModel, ChatRole, } from "./models/chats.model.js";
+export class MongoDBMemory extends BaseMemory {
+    constructor(uri) {
+        super();
+        this.isConnected = false;
+        this.connect(uri);
+    }
+    async connect(uri) {
+        if (this.isConnected) {
+            return;
+        }
+        try {
+            await mongoose.connect(uri);
+            this.isConnected = true;
+            loggers.agent.info("MongoDB connected successfully");
+        }
+        catch (error) {
+            loggers.agent.error("Failed to connect to MongoDB:", error);
+            throw error;
+        }
+    }
+    async disconnect() {
+        if (!this.isConnected) {
+            return;
+        }
+        try {
+            await mongoose.disconnect();
+            this.isConnected = false;
+            loggers.agent.info("MongoDB disconnected successfully");
+        }
+        catch (error) {
+            loggers.agent.error("Failed to disconnect from MongoDB:", error);
+            throw error;
+        }
+    }
+    async getSessionHistory(sessionId) {
+        const chats = await ChatModel.find({ sessionId }).sort({
+            timestamp: 1,
+        });
+        loggers.agent.info(`Found ${chats.length} chats for session ${sessionId}`);
+        const sessionObject = { chats: {} };
+        chats.forEach((chat) => {
+            const chatId = chat._id?.toString() || chat.id;
+            sessionObject.chats[chatId] = {
+                role: chat.role,
+                content: chat.content,
+                timestamp: chat.timestamp,
+                metadata: chat.metadata,
+            };
+        });
+        return sessionObject;
+    }
+    async updateSessionHistory(sessionId, chat) {
+        loggers.agent.info(`Updating session history for session ${sessionId}`);
+        loggers.agent.info(`Chat: ${JSON.stringify(chat)}`);
+        await ChatModel.create({
+            sessionId,
+            role: chat.role,
+            content: chat.content,
+            timestamp: chat.timestamp,
+            metadata: chat.metadata,
+        });
+    }
+    async storeQueryAndIntent(query, intent, sessionId) {
+        // Intent 정보를 metadata에 저장
+        const chat = {
+            role: ChatRole.USER,
+            content: {
+                type: "text",
+                parts: [query],
+            },
+            timestamp: Date.now(),
+            metadata: {
+                intent,
+                query,
+            },
+        };
+        await this.updateSessionHistory(sessionId, chat);
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/esm/modules/memory/providers/mongodb/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../src/modules/memory/providers/mongodb/index.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,MAAM,UAAU,CAAC;AAEhC,OAAO,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAEN,SAAS,EACT,QAAQ,GACR,MAAM,yBAAyB,CAAC;AAEjC,MAAM,OAAO,aAAc,SAAQ,UAAU;IAG5C,YAAY,GAAW;QACtB,KAAK,EAAE,CAAC;QAHD,gBAAW,GAAG,KAAK,CAAC;QAI3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,OAAO,CAAC,GAAW;QAC/B,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,OAAO;QACR,CAAC;QAED,IAAI,CAAC;YACJ,MAAM,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAC5B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YACxB,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YAC5D,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAEM,KAAK,CAAC,UAAU;QACtB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACvB,OAAO;QACR,CAAC;QAED,IAAI,CAAC;YACJ,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;YACzB,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;QACzD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YACjE,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAEM,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QAC/C,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC;YACtD,SAAS,EAAE,CAAC;SACZ,CAAC,CAAC;QAEH,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,MAAM,sBAAsB,SAAS,EAAE,CAAC,CAAC;QAE3E,MAAM,aAAa,GAAkB,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QACnD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAkB,EAAE,EAAE;YACpC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,QAAQ,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC;YAC/C,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG;gBAC7B,IAAI,EAAE,IAAI,CAAC,IAAgB;gBAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACvB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,aAAa,CAAC;IACtB,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAChC,SAAiB,EACjB,IAAgB;QAEhB,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,wCAAwC,SAAS,EAAE,CAAC,CAAC;QACxE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEpD,MAAM,SAAS,CAAC,MAAM,CAAC;YACtB,SAAS;YACT,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACvB,CAAC,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAC/B,KAAa,EACb,MAAc,EACd,SAAiB;QAEjB,0BAA0B;QAC1B,MAAM,IAAI,GAAe;YACxB,IAAI,EAAE,QAAQ,CAAC,IAAI;YACnB,OAAO,EAAE;gBACR,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,KAAK,CAAC;aACd;YACD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE;gBACT,MAAM;gBACN,KAAK;aACL;SACD,CAAC;QAEF,MAAM,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;CACD"}

package/dist/esm/modules/memory/providers/mongodb/models/chats.model.d.ts

@@ -0,0 +1,26 @@
+import mongoose, { type Document } from "mongoose";
+export declare enum ChatRole {
+    USER = "USER",
+    SYSTEM = "SYSTEM",
+    MODEL = "MODEL"
+}
+export interface ChatDocument extends Document {
+    sessionId: string;
+    role: ChatRole;
+    content: {
+        type: string;
+        parts: any[];
+    };
+    timestamp: number;
+    metadata?: {
+        [key: string]: unknown;
+    };
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare const ChatModel: mongoose.Model<ChatDocument, {}, {}, {}, mongoose.Document<unknown, {}, ChatDocument, {}> & ChatDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+//# sourceMappingURL=chats.model.d.ts.map

package/dist/esm/modules/memory/providers/mongodb/models/chats.model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chats.model.d.ts","sourceRoot":"","sources":["../../../../../../../src/modules/memory/providers/mongodb/models/chats.model.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,EAAE,EAAE,KAAK,QAAQ,EAAU,MAAM,UAAU,CAAC;AAG3D,oBAAY,QAAQ;IACnB,IAAI,SAAS;IACb,MAAM,WAAW;IACjB,KAAK,UAAU;CACf;AA2CD,MAAM,WAAW,YAAa,SAAQ,QAAQ;IAC7C,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,QAAQ,CAAC;IACf,OAAO,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,GAAG,EAAE,CAAC;KACb,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE;QAAE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC;IACtC,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CAChB;AAGD,eAAO,MAAM,SAAS;;;;OAAyD,CAAC"}

package/dist/esm/modules/memory/providers/mongodb/models/chats.model.js

@@ -0,0 +1,43 @@
+import mongoose, { Schema } from "mongoose";
+// ChatRole enum
+export var ChatRole;
+(function (ChatRole) {
+    ChatRole["USER"] = "USER";
+    ChatRole["SYSTEM"] = "SYSTEM";
+    ChatRole["MODEL"] = "MODEL";
+})(ChatRole || (ChatRole = {}));
+// ChatContentObject schema
+const ChatContentObjectSchema = new Schema({
+    type: { type: String, required: true },
+    parts: { type: [Schema.Types.Mixed], required: true },
+}, { _id: false });
+// ChatObject schema - 개별 문서로 저장
+const ChatObjectSchema = new Schema({
+    sessionId: {
+        type: String,
+        required: true,
+        index: true,
+    },
+    role: {
+        type: String,
+        enum: Object.values(ChatRole),
+        required: true,
+    },
+    content: {
+        type: ChatContentObjectSchema,
+        required: true,
+    },
+    timestamp: {
+        type: Number,
+        required: true,
+    },
+    metadata: {
+        type: Schema.Types.Mixed,
+        default: {},
+    },
+}, {
+    timestamps: true,
+});
+// Export the model
+export const ChatModel = mongoose.model("Chat", ChatObjectSchema);
+//# sourceMappingURL=chats.model.js.map

package/dist/esm/modules/memory/providers/mongodb/models/chats.model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chats.model.js","sourceRoot":"","sources":["../../../../../../../src/modules/memory/providers/mongodb/models/chats.model.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,EAAE,EAAiB,MAAM,EAAE,MAAM,UAAU,CAAC;AAE3D,gBAAgB;AAChB,MAAM,CAAN,IAAY,QAIX;AAJD,WAAY,QAAQ;IACnB,yBAAa,CAAA;IACb,6BAAiB,CAAA;IACjB,2BAAe,CAAA;AAChB,CAAC,EAJW,QAAQ,KAAR,QAAQ,QAInB;AAED,2BAA2B;AAC3B,MAAM,uBAAuB,GAAG,IAAI,MAAM,CACzC;IACC,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE;IACtC,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE;CACrD,EACD,EAAE,GAAG,EAAE,KAAK,EAAE,CACd,CAAC;AAEF,gCAAgC;AAChC,MAAM,gBAAgB,GAAG,IAAI,MAAM,CAClC;IACC,SAAS,EAAE;QACV,IAAI,EAAE,MAAM;QACZ,QAAQ,EAAE,IAAI;QACd,KAAK,EAAE,IAAI;KACX;IACD,IAAI,EAAE;QACL,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC;QAC7B,QAAQ,EAAE,IAAI;KACd;IACD,OAAO,EAAE;QACR,IAAI,EAAE,uBAAuB;QAC7B,QAAQ,EAAE,IAAI;KACd;IACD,SAAS,EAAE;QACV,IAAI,EAAE,MAAM;QACZ,QAAQ,EAAE,IAAI;KACd;IACD,QAAQ,EAAE;QACT,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK;QACxB,OAAO,EAAE,EAAE;KACX;CACD,EACD;IACC,UAAU,EAAE,IAAI;CAChB,CACD,CAAC;AAgBF,mBAAmB;AACnB,MAAM,CAAC,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAe,MAAM,EAAE,gBAAgB,CAAC,CAAC"}

package/dist/esm/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.d.ts

@@ -0,0 +1,27 @@
+import mongoose, { type Document } from "mongoose";
+export interface IntentTriggeringInfoDocument extends Document {
+    context: {
+        messages: Array<{
+            role: string;
+            content: string | Array<{
+                type: "text";
+                text: string;
+            } | {
+                type: "image_url";
+                image_url: {
+                    url: string;
+                };
+            }>;
+        }>;
+    };
+    intent: {
+        name: string;
+        description: string;
+    };
+}
+export declare const IntentTriggeringInfoModel: mongoose.Model<IntentTriggeringInfoDocument, {}, {}, {}, mongoose.Document<unknown, {}, IntentTriggeringInfoDocument, {}> & IntentTriggeringInfoDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+//# sourceMappingURL=intentTriggeringInfos.model.d.ts.map

package/dist/esm/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"intentTriggeringInfos.model.d.ts","sourceRoot":"","sources":["../../../../../../../src/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,EAAE,EAAE,KAAK,QAAQ,EAAU,MAAM,UAAU,CAAC;AAiB3D,MAAM,WAAW,4BAA6B,SAAQ,QAAQ;IAC7D,OAAO,EAAE;QACR,QAAQ,EAAE,KAAK,CAAC;YACf,IAAI,EAAE,MAAM,CAAC;YACb,OAAO,EACJ,MAAM,GACN,KAAK,CACH;gBAAE,IAAI,EAAE,MAAM,CAAC;gBAAC,IAAI,EAAE,MAAM,CAAA;aAAE,GAC9B;gBAAE,IAAI,EAAE,WAAW,CAAC;gBAAC,SAAS,EAAE;oBAAE,GAAG,EAAE,MAAM,CAAA;iBAAE,CAAA;aAAE,CAClD,CAAC;SACL,CAAC,CAAC;KACH,CAAC;IACF,MAAM,EAAE;QACP,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;KACpB,CAAC;CACF;AAYD,eAAO,MAAM,yBAAyB;;;;OAIpC,CAAC"}

package/dist/esm/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.js

@@ -0,0 +1,23 @@
+import mongoose, { Schema } from "mongoose";
+const MessageSchema = new Schema({
+    role: {
+        type: String,
+        enum: ["system", "user", "assistant", "tool", "function"],
+        required: true,
+    },
+    content: {
+        type: Schema.Types.Mixed, // string 또는 object array
+        required: true,
+    },
+}, { _id: false });
+const IntentTriggeringInfoSchema = new Schema({
+    context: {
+        messages: [MessageSchema],
+    },
+    intent: {
+        name: { type: String, required: true },
+        description: { type: String, required: true },
+    },
+});
+export const IntentTriggeringInfoModel = mongoose.model("IntentTriggeringInfo", IntentTriggeringInfoSchema);
+//# sourceMappingURL=intentTriggeringInfos.model.js.map

package/dist/esm/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"intentTriggeringInfos.model.js","sourceRoot":"","sources":["../../../../../../../src/modules/memory/providers/mongodb/models/intentTriggeringInfos.model.ts"],"names":[],"mappings":"AAAA,OAAO,QAAQ,EAAE,EAAiB,MAAM,EAAE,MAAM,UAAU,CAAC;AAE3D,MAAM,aAAa,GAAG,IAAI,MAAM,CAC/B;IACC,IAAI,EAAE;QACL,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,CAAC;QACzD,QAAQ,EAAE,IAAI;KACd;IACD,OAAO,EAAE;QACR,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,yBAAyB;QACnD,QAAQ,EAAE,IAAI;KACd;CACD,EACD,EAAE,GAAG,EAAE,KAAK,EAAE,CACd,CAAC;AAoBF,MAAM,0BAA0B,GAAG,IAAI,MAAM,CAA+B;IAC3E,OAAO,EAAE;QACR,QAAQ,EAAE,CAAC,aAAa,CAAC;KACzB;IACD,MAAM,EAAE;QACP,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE;QACtC,WAAW,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE;KAC7C;CACD,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,yBAAyB,GACrC,QAAQ,CAAC,KAAK,CACb,sBAAsB,EACtB,0BAA0B,CAC1B,CAAC"}

package/dist/esm/modules/models/base/index.d.ts

@@ -0,0 +1,57 @@
+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(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        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>;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/esm/modules/models/base/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/modules/models/base/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEjE;;;;;;;;;GASG;AACH,8BAAsB,SAAS,CAAC,WAAW,EAAE,YAAY;IACxD;;;;;;;OAOG;IACH,QAAQ,CAAC,gBAAgB,CAAC,MAAM,EAAE;QACjC,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,CAAC,EAAE,aAAa,CAAC;QAC/B,YAAY,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,WAAW,EAAE;IAEjB;;;;;OAKG;IACH,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAEvE;;;;;OAKG;IACH,QAAQ,CAAC,uBAAuB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,YAAY,EAAE;IAErE;;;;;OAKG;IACH,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAE/D;;;;;;OAMG;IACH,QAAQ,CAAC,uBAAuB,CAC/B,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,YAAY,EAAE,GACvB,OAAO,CAAC,aAAa,CAAC;CACzB"}

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

@@ -0,0 +1,13 @@
+/**
+ * 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 {
+}
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/modules/models/base/index.ts"],"names":[],"mappings":"AAGA;;;;;;;;;GASG;AACH,MAAM,OAAgB,SAAS;CAkD9B"}

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

@@ -0,0 +1,57 @@
+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(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        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>;
+}
+//# sourceMappingURL=base.model.d.ts.map

package/dist/esm/modules/models/base.model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.model.d.ts","sourceRoot":"","sources":["../../../../src/modules/models/base.model.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEjE;;;;;;;;;GASG;AACH,8BAAsB,SAAS,CAAC,WAAW,EAAE,YAAY;IACxD;;;;;;;OAOG;IACH,QAAQ,CAAC,gBAAgB,CAAC,MAAM,EAAE;QACjC,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,CAAC,EAAE,aAAa,CAAC;QAC/B,YAAY,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,WAAW,EAAE;IAEjB;;;;;OAKG;IACH,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAEvE;;;;;OAKG;IACH,QAAQ,CAAC,uBAAuB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,YAAY,EAAE;IAErE;;;;;OAKG;IACH,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAE/D;;;;;;OAMG;IACH,QAAQ,CAAC,uBAAuB,CAC/B,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,YAAY,EAAE,GACvB,OAAO,CAAC,aAAa,CAAC;CACzB"}

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

@@ -0,0 +1,13 @@
+/**
+ * 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 {
+}
+//# sourceMappingURL=base.model.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"base.model.js","sourceRoot":"","sources":["../../../../src/modules/models/base.model.ts"],"names":[],"mappings":"AAGA;;;;;;;;;GASG;AACH,MAAM,OAAgB,SAAS;CAkD9B"}

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

@@ -1,71 +1,4 @@
-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");
- * ```
- */
+import type { BaseModel } from "./base.model.js";
 export declare class ModelModule {
     /** Registry of available models indexed by name */
     private models;
@@ -93,9 +26,7 @@ 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;
     };
 }

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

@@ -1 +1 @@
-{"version":3,"file":"model.module.d.ts","sourceRoot":"","sources":["../../../../src/modules/models/model.module.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEjE;;;;;;;;;GASG;AACH,8BAAsB,SAAS,CAAC,WAAW,EAAE,YAAY;IACxD;;;;;;;OAOG;IACH,QAAQ,CAAC,gBAAgB,CACxB,cAAc,EAAE,aAAa,EAC7B,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,GACnB,WAAW,EAAE;IAEhB;;;;;OAKG;IACH,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAEvE;;;;;OAKG;IACH,QAAQ,CAAC,uBAAuB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,YAAY,EAAE;IAErE;;;;;OAKG;IACH,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAE/D;;;;;;OAMG;IACH,QAAQ,CAAC,uBAAuB,CAC/B,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,YAAY,EAAE,GACvB,OAAO,CAAC,aAAa,CAAC;CACzB;AAED;;;;;;;;;;;;;;;GAeG;AACH,qBAAa,WAAW;IACvB,mDAAmD;IACnD,OAAO,CAAC,MAAM,CAAuD;IACrE,2DAA2D;IAC3D,OAAO,CAAC,gBAAgB,CAAC,CAAS;IAElC;;;;;;OAMG;IACI,QAAQ,CACd,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,EAClC,SAAS,CAAC,EAAE,OAAO;IAQpB;;;;;;OAMG;IACI,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC;IAW3D;;;;OAIG;IACI,YAAY;;;;;;CAMnB"}
+{"version":3,"file":"model.module.d.ts","sourceRoot":"","sources":["../../../../src/modules/models/model.module.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,qBAAa,WAAW;IACvB,mDAAmD;IACnD,OAAO,CAAC,MAAM,CAAuD;IACrE,2DAA2D;IAC3D,OAAO,CAAC,gBAAgB,CAAC,CAAS;IAElC;;;;;;OAMG;IACI,QAAQ,CACd,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,EAClC,SAAS,CAAC,EAAE,OAAO;IAQpB;;;;;;OAMG;IACI,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC;IAW3D;;;;OAIG;IACI,YAAY;;;;CAMnB"}

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

@@ -1,16 +1,4 @@
-/**
- * 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
@@ -67,7 +55,7 @@ export class ModelModule {
      */
     getModelList() {
         return {
-            models: this.models,
+            models: Object.keys(this.models),
             defaultModelName: this.defaultModelName,
         };
     }

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,"file":"model.module.js","sourceRoot":"","sources":["../../../../src/modules/models/model.module.ts"],"names":[],"mappings":"AAEA;;;;;;;;;;;;;;;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,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;YAChC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACvC,CAAC;IACH,CAAC;CACD"}

package/dist/esm/modules/models/{gemini.d.ts → providers/gemini.d.ts}

@@ -1,13 +1,17 @@
 import { type Content, type FunctionDeclaration } from "@google/genai";
-import { type SessionObject } from "../../types/memory.js";
-import type { FetchResponse, IAgentTool } from "../../types/tool.js";
-import { BaseModel } from "./model.module.js";
+import { type SessionObject } from "../../../types/memory.js";
+import type { FetchResponse, IAgentTool } from "../../../types/tool.js";
+import { BaseModel } from "../base/index.js";
 export declare class GeminiModel extends BaseModel<Content, FunctionDeclaration> {
     private client;
     private modelName;
     constructor(apiKey: string, modelName: string);
     private getMessageRole;
-    generateMessages(sessionHistory: SessionObject, query: string, systemPrompt?: string): Content[];
+    generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): Content[];
     appendMessages(messages: Content[], message: string): void;
     fetch(messages: Content[]): Promise<FetchResponse>;
     fetchWithContextMessage(messages: Content[], functions: FunctionDeclaration[]): Promise<FetchResponse>;

package/dist/esm/modules/models/providers/gemini.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../../../src/modules/models/providers/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,OAAO,EAEZ,KAAK,mBAAmB,EAExB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAY,KAAK,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,KAAK,EACX,aAAa,EAEb,UAAU,EAGV,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAE7C,qBAAa,WAAY,SAAQ,SAAS,CAAC,OAAO,EAAE,mBAAmB,CAAC;IACvE,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;IAM7C,OAAO,CAAC,cAAc;IAYtB,gBAAgB,CAAC,MAAM,EAAE;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,CAAC,EAAE,aAAa,CAAC;QAC/B,YAAY,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,EAAE;IAmBb,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAOpD,KAAK,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IASlD,uBAAuB,CAC5B,QAAQ,EAAE,OAAO,EAAE,EACnB,SAAS,EAAE,mBAAmB,EAAE,GAC9B,OAAO,CAAC,aAAa,CAAC;IAiCzB,uBAAuB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,mBAAmB,EAAE;CAwBnE"}

package/dist/esm/modules/models/{gemini.js → providers/gemini.js}

@@ -1,7 +1,7 @@
 import { GoogleGenAI, } from "@google/genai";
-import { ChatRole } from "../../types/memory.js";
-import { TOOL_PROTOCOL_TYPE } from "../../types/tool.js";
-import { BaseModel } from "./model.module.js";
+import { ChatRole } from "../../../types/memory.js";
+import { TOOL_PROTOCOL_TYPE } from "../../../types/tool.js";
+import { BaseModel } from "../base/index.js";
 export class GeminiModel extends BaseModel {
     constructor(apiKey, modelName) {
         super();
@@ -19,18 +19,21 @@ export class GeminiModel extends BaseModel {
                 return "model"; /*FIXME*/
         }
     }
-    generateMessages(sessionHistory, query, systemPrompt) {
+    generateMessages(params) {
+        const { query, sessionHistory, systemPrompt } = params;
         const messages = !systemPrompt
             ? []
             : [{ role: "model", parts: [{ text: systemPrompt.trim() }] }];
-        const sessionContent = Object.keys(sessionHistory.chats).map((chatId) => {
-            const chat = sessionHistory.chats[chatId];
-            // TODO: check message.content.type
-            return {
-                role: this.getMessageRole(chat.role),
-                parts: [{ text: chat.content.parts[0] }],
-            };
-        });
+        const sessionContent = !sessionHistory
+            ? []
+            : Object.keys(sessionHistory.chats).map((chatId) => {
+                const chat = sessionHistory.chats[chatId];
+                // TODO: check message.content.type
+                return {
+                    role: this.getMessageRole(chat.role),
+                    parts: [{ text: chat.content.parts[0] }],
+                };
+            });
         const userContent = { role: "user", parts: [{ text: query }] };
         return messages.concat(sessionContent).concat(userContent);
     }

package/dist/esm/modules/models/providers/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../../../src/modules/models/providers/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAIN,WAAW,GACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,QAAQ,EAAsB,MAAM,mBAAmB,CAAC;AAQjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAE7C,MAAM,OAAO,WAAY,SAAQ,SAAuC;IAIvE,YAAY,MAAc,EAAE,SAAiB;QAC5C,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC5B,CAAC;IAEO,cAAc,CAAC,IAAc;QACpC,QAAQ,IAAI,EAAE,CAAC;YACd,KAAK,QAAQ,CAAC,IAAI;gBACjB,OAAO,MAAM,CAAC;YACf,KAAK,QAAQ,CAAC,KAAK,CAAC;YACpB,KAAK,QAAQ,CAAC,MAAM;gBACnB,OAAO,OAAO,CAAC;YAChB;gBACC,OAAO,OAAO,CAAC,CAAC,SAAS;QAC3B,CAAC;IACF,CAAC;IAED,gBAAgB,CAAC,MAIhB;QACA,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,MAAM,CAAC;QACvD,MAAM,QAAQ,GAAc,CAAC,YAAY;YACxC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QAC/D,MAAM,cAAc,GAAc,CAAC,cAAc;YAChD,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,MAAc,EAAE,EAAE;gBACzD,MAAM,IAAI,GAAG,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC1C,mCAAmC;gBACnC,OAAO;oBACN,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;oBACpC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;iBACxC,CAAC;YACH,CAAC,CAAC,CAAC;QACL,MAAM,WAAW,GAAY,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QACxE,OAAO,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC5D,CAAC;IAED,cAAc,CAAC,QAAmB,EAAE,OAAe;QAClD,QAAQ,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;SAC1B,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,QAAmB;QAC9B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC;YACzD,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,QAAQ,EAAE,QAAQ;SAClB,CAAC,CAAC;QAEH,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,IAAI,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,uBAAuB,CAC5B,QAAmB,EACnB,SAAgC;QAEhC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC;gBACzD,KAAK,EAAE,IAAI,CAAC,SAAS;gBACrB,QAAQ,EAAE,QAAQ;gBAClB,MAAM,EAAE;oBACP,KAAK,EAAE,CAAC,EAAE,oBAAoB,EAAE,SAAS,EAAE,CAAC;iBAC5C;aACD,CAAC,CAAC;YAEH,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,QAAQ,CAAC;YACzC,MAAM,OAAO,GAAG,CACf,KAAmB,EACwB,EAAE;gBAC7C,OAAO,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC;YACjC,CAAC,CAAC;YACF,MAAM,SAAS,GAA2B,aAAa;gBACtD,EAAE,MAAM,CAAC,OAAO,CAAC;iBAChB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACd,OAAO;oBACN,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,SAAS,EAAE,KAAK,CAAC,IAAI;iBACrB,CAAC;YACH,CAAC,CAAC,CAAC;YAEJ,OAAO;gBACN,OAAO,EAAE,IAAI;gBACb,SAAS;aACT,CAAC;QACH,CAAC;QACD,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,uBAAuB,CAAC,KAAmB;QAC1C,MAAM,SAAS,GAA0B,EAAE,CAAC;QAC5C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACnB,SAAS;YACV,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,KAAK,kBAAkB,CAAC,GAAG,EAAE,CAAC;gBAC9C,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,GAAG,IAAgB,CAAC;gBACzC,SAAS,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,EAAE;oBACR,WAAW,EAAE,OAAO,CAAC,WAAW;oBAChC,oBAAoB,EAAE,OAAO,CAAC,WAAW;iBACzC,CAAC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACP,oBAAoB;gBACpB,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,IAAgB,CAAC;gBACtC,SAAS,CAAC,IAAI,CAAC;oBACd,IAAI,EAAE,EAAE;oBACR,WAAW,EAAE,IAAI,CAAC,WAAW;iBAC7B,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,OAAO,SAAS,CAAC;IAClB,CAAC;CACD"}

package/dist/{cjs/modules/models → esm/modules/models/providers}/openai.d.ts

@@ -1,13 +1,17 @@
 import type { ChatCompletionMessageParam as CCMessageParam, ChatCompletionTool } from "openai/resources";
-import { type SessionObject } from "../../types/memory.js";
-import type { FetchResponse, IAgentTool } from "../../types/tool.js";
-import { BaseModel } from "./model.module.js";
+import { type SessionObject } from "../../../types/memory.js";
+import type { FetchResponse, IAgentTool } from "../../../types/tool.js";
+import { BaseModel } from "../base/index.js";
 export declare class AzureOpenAI extends BaseModel<CCMessageParam, ChatCompletionTool> {
     private client;
     private modelName;
     constructor(baseUrl: string, apiKey: string, apiVersion: string, modelName: string);
     private getMessageRole;
-    generateMessages(sessionHistory: SessionObject, query: string, systemPrompt?: string): CCMessageParam[];
+    generateMessages(params: {
+        query: string;
+        sessionHistory?: SessionObject;
+        systemPrompt?: string;
+    }): CCMessageParam[];
     appendMessages(messages: CCMessageParam[], message: string): void;
     fetch(messages: CCMessageParam[]): Promise<FetchResponse>;
     fetchWithContextMessage(messages: CCMessageParam[], functions: ChatCompletionTool[]): Promise<FetchResponse>;

package/dist/esm/modules/models/providers/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../../../src/modules/models/providers/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACX,0BAA0B,IAAI,cAAc,EAE5C,kBAAkB,EAClB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAY,KAAK,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,KAAK,EACX,aAAa,EAEb,UAAU,EAGV,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAE7C,qBAAa,WAAY,SAAQ,SAAS,CAAC,cAAc,EAAE,kBAAkB,CAAC;IAC7E,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,SAAS,CAAS;gBAGzB,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM,EACd,UAAU,EAAE,MAAM,EAClB,SAAS,EAAE,MAAM;IAWlB,OAAO,CAAC,cAAc;IAYtB,gBAAgB,CAAC,MAAM,EAAE;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,cAAc,CAAC,EAAE,aAAa,CAAC;QAC/B,YAAY,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,cAAc,EAAE;IAkBpB,cAAc,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAO3D,KAAK,CAAC,QAAQ,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,aAAa,CAAC;IAWzD,uBAAuB,CAC5B,QAAQ,EAAE,cAAc,EAAE,EAC1B,SAAS,EAAE,kBAAkB,EAAE,GAC7B,OAAO,CAAC,aAAa,CAAC;IA6BzB,uBAAuB,CAAC,KAAK,EAAE,UAAU,EAAE,GAAG,kBAAkB,EAAE;CA8BlE"}