@ainetwork/adk-provider-memory-mongodb 0.1.1

package/dist/chunk-2SB2M62A.js

@@ -0,0 +1,51 @@
+// models/chats.model.ts
+import mongoose, { Schema } from "mongoose";
+var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
+  ChatRole2["USER"] = "USER";
+  ChatRole2["SYSTEM"] = "SYSTEM";
+  ChatRole2["MODEL"] = "MODEL";
+  return ChatRole2;
+})(ChatRole || {});
+var ChatContentObjectSchema = new Schema(
+  {
+    type: { type: String, required: true },
+    parts: { type: [Schema.Types.Mixed], required: true }
+  },
+  { _id: false }
+);
+var 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
+  }
+);
+var ChatModel = mongoose.model("Chat", ChatObjectSchema);
+
+export {
+  ChatRole,
+  ChatModel
+};
+//# sourceMappingURL=chunk-2SB2M62A.js.map

package/dist/chunk-2SB2M62A.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../models/chats.model.ts"],"sourcesContent":["import mongoose, { type Document, Schema } from \"mongoose\";\n\n// ChatRole enum\nexport enum ChatRole {\n\tUSER = \"USER\",\n\tSYSTEM = \"SYSTEM\",\n\tMODEL = \"MODEL\",\n}\n\n// ChatContentObject schema\nconst ChatContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// ChatObject schema - 개별 문서로 저장\nconst ChatObjectSchema = new Schema(\n\t{\n\t\tsessionId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ChatRole),\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: ChatContentObjectSchema,\n\t\t\trequired: true,\n\t\t},\n\t\ttimestamp: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tmetadata: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t\tdefault: {},\n\t\t},\n\t},\n\t{\n\t\ttimestamps: true,\n\t},\n);\n\n// Chat Document interface\nexport interface ChatDocument extends Document {\n\tsessionId: string;\n\trole: ChatRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n\tcreatedAt: Date;\n\tupdatedAt: Date;\n}\n\n// Export the model\nexport const ChatModel = mongoose.model<ChatDocument>(\"Chat\", ChatObjectSchema);\n"],"mappings":";AAAA,OAAO,YAA2B,cAAc;AAGzC,IAAK,WAAL,kBAAKA,cAAL;AACN,EAAAA,UAAA,UAAO;AACP,EAAAA,UAAA,YAAS;AACT,EAAAA,UAAA,WAAQ;AAHG,SAAAA;AAAA,GAAA;AAOZ,IAAM,0BAA0B,IAAI;AAAA,EACnC;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,OAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGA,IAAM,mBAAmB,IAAI;AAAA,EAC5B;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,QAAQ;AAAA,MAC5B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,OAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAiBO,IAAM,YAAY,SAAS,MAAoB,QAAQ,gBAAgB;","names":["ChatRole"]}

package/dist/index.cjs

@@ -0,0 +1,167 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// index.ts
+var index_exports = {};
+__export(index_exports, {
+  MongoDBMemory: () => MongoDBMemory
+});
+module.exports = __toCommonJS(index_exports);
+var import_modules = require("@ainetwork/adk/modules");
+var import_logger = require("@ainetwork/adk/utils/logger");
+var import_mongoose2 = __toESM(require("mongoose"), 1);
+
+// models/chats.model.ts
+var import_mongoose = __toESM(require("mongoose"), 1);
+var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
+  ChatRole2["USER"] = "USER";
+  ChatRole2["SYSTEM"] = "SYSTEM";
+  ChatRole2["MODEL"] = "MODEL";
+  return ChatRole2;
+})(ChatRole || {});
+var ChatContentObjectSchema = new import_mongoose.Schema(
+  {
+    type: { type: String, required: true },
+    parts: { type: [import_mongoose.Schema.Types.Mixed], required: true }
+  },
+  { _id: false }
+);
+var ChatObjectSchema = new import_mongoose.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: import_mongoose.Schema.Types.Mixed,
+      default: {}
+    }
+  },
+  {
+    timestamps: true
+  }
+);
+var ChatModel = import_mongoose.default.model("Chat", ChatObjectSchema);
+
+// index.ts
+var MongoDBMemory = class extends import_modules.BaseMemory {
+  isConnected = false;
+  constructor(uri) {
+    super();
+    this.connect(uri);
+  }
+  async connect(uri) {
+    if (this.isConnected) {
+      return;
+    }
+    try {
+      await import_mongoose2.default.connect(uri);
+      this.isConnected = true;
+      import_logger.loggers.agent.info("MongoDB connected successfully");
+    } catch (error) {
+      import_logger.loggers.agent.error("Failed to connect to MongoDB:", error);
+      throw error;
+    }
+  }
+  async disconnect() {
+    if (!this.isConnected) {
+      return;
+    }
+    try {
+      await import_mongoose2.default.disconnect();
+      this.isConnected = false;
+      import_logger.loggers.agent.info("MongoDB disconnected successfully");
+    } catch (error) {
+      import_logger.loggers.agent.error("Failed to disconnect from MongoDB:", error);
+      throw error;
+    }
+  }
+  async getSessionHistory(sessionId) {
+    const chats = await ChatModel.find({ sessionId }).sort({
+      timestamp: 1
+    });
+    import_logger.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) {
+    import_logger.loggers.agent.info(`Updating session history for session ${sessionId}`);
+    import_logger.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) {
+    const chat = {
+      role: "USER" /* USER */,
+      content: {
+        type: "text",
+        parts: [query]
+      },
+      timestamp: Date.now(),
+      metadata: {
+        intent,
+        query
+      }
+    };
+    await this.updateSessionHistory(sessionId, chat);
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  MongoDBMemory
+});
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../index.ts","../models/chats.model.ts"],"sourcesContent":["import { BaseMemory } from \"@ainetwork/adk/modules\";\nimport type { ChatObject, SessionObject } from \"@ainetwork/adk/types/memory\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\nimport mongoose from \"mongoose\";\nimport {\n\ttype ChatDocument,\n\tChatModel,\n\tChatRole,\n} from \"./models/chats.model\";\n\nexport class MongoDBMemory extends BaseMemory {\n\tprivate isConnected = false;\n\n\tconstructor(uri: string) {\n\t\tsuper();\n\t\tthis.connect(uri);\n\t}\n\n\tpublic async connect(uri: string): Promise<void> {\n\t\tif (this.isConnected) {\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tawait mongoose.connect(uri);\n\t\t\tthis.isConnected = true;\n\t\t\tloggers.agent.info(\"MongoDB connected successfully\");\n\t\t} catch (error) {\n\t\t\tloggers.agent.error(\"Failed to connect to MongoDB:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tpublic async disconnect(): Promise<void> {\n\t\tif (!this.isConnected) {\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tawait mongoose.disconnect();\n\t\t\tthis.isConnected = false;\n\t\t\tloggers.agent.info(\"MongoDB disconnected successfully\");\n\t\t} catch (error) {\n\t\t\tloggers.agent.error(\"Failed to disconnect from MongoDB:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tpublic async getSessionHistory(sessionId: string): Promise<SessionObject> {\n\t\tconst chats = await ChatModel.find({ sessionId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n\n\t\tloggers.agent.info(`Found ${chats.length} chats for session ${sessionId}`);\n\n\t\tconst sessionObject: SessionObject = { chats: {} };\n\t\tchats.forEach((chat: ChatDocument) => {\n\t\t\tconst chatId = chat._id?.toString() || chat.id;\n\t\t\tsessionObject.chats[chatId] = {\n\t\t\t\trole: chat.role as ChatRole,\n\t\t\t\tcontent: chat.content,\n\t\t\t\ttimestamp: chat.timestamp,\n\t\t\t\tmetadata: chat.metadata,\n\t\t\t};\n\t\t});\n\n\t\treturn sessionObject;\n\t}\n\n\tpublic async updateSessionHistory(\n\t\tsessionId: string,\n\t\tchat: ChatObject,\n\t): Promise<void> {\n\t\tloggers.agent.info(`Updating session history for session ${sessionId}`);\n\t\tloggers.agent.info(`Chat: ${JSON.stringify(chat)}`);\n\n\t\tawait ChatModel.create({\n\t\t\tsessionId,\n\t\t\trole: chat.role,\n\t\t\tcontent: chat.content,\n\t\t\ttimestamp: chat.timestamp,\n\t\t\tmetadata: chat.metadata,\n\t\t});\n\t}\n\n\tpublic async storeQueryAndIntent(\n\t\tquery: string,\n\t\tintent: string,\n\t\tsessionId: string,\n\t): Promise<void> {\n\t\t// Intent 정보를 metadata에 저장\n\t\tconst chat: ChatObject = {\n\t\t\trole: ChatRole.USER,\n\t\t\tcontent: {\n\t\t\t\ttype: \"text\",\n\t\t\t\tparts: [query],\n\t\t\t},\n\t\t\ttimestamp: Date.now(),\n\t\t\tmetadata: {\n\t\t\t\tintent,\n\t\t\t\tquery,\n\t\t\t},\n\t\t};\n\n\t\tawait this.updateSessionHistory(sessionId, chat);\n\t}\n}\n","import mongoose, { type Document, Schema } from \"mongoose\";\n\n// ChatRole enum\nexport enum ChatRole {\n\tUSER = \"USER\",\n\tSYSTEM = \"SYSTEM\",\n\tMODEL = \"MODEL\",\n}\n\n// ChatContentObject schema\nconst ChatContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// ChatObject schema - 개별 문서로 저장\nconst ChatObjectSchema = new Schema(\n\t{\n\t\tsessionId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ChatRole),\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: ChatContentObjectSchema,\n\t\t\trequired: true,\n\t\t},\n\t\ttimestamp: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tmetadata: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t\tdefault: {},\n\t\t},\n\t},\n\t{\n\t\ttimestamps: true,\n\t},\n);\n\n// Chat Document interface\nexport interface ChatDocument extends Document {\n\tsessionId: string;\n\trole: ChatRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n\tcreatedAt: Date;\n\tupdatedAt: Date;\n}\n\n// Export the model\nexport const ChatModel = mongoose.model<ChatDocument>(\"Chat\", ChatObjectSchema);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,qBAA2B;AAE3B,oBAAwB;AACxB,IAAAA,mBAAqB;;;ACHrB,sBAAgD;AAGzC,IAAK,WAAL,kBAAKC,cAAL;AACN,EAAAA,UAAA,UAAO;AACP,EAAAA,UAAA,YAAS;AACT,EAAAA,UAAA,WAAQ;AAHG,SAAAA;AAAA,GAAA;AAOZ,IAAM,0BAA0B,IAAI;AAAA,EACnC;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,uBAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGA,IAAM,mBAAmB,IAAI;AAAA,EAC5B;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,QAAQ;AAAA,MAC5B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,uBAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAiBO,IAAM,YAAY,gBAAAC,QAAS,MAAoB,QAAQ,gBAAgB;;;ADtDvE,IAAM,gBAAN,cAA4B,0BAAW;AAAA,EACrC,cAAc;AAAA,EAEtB,YAAY,KAAa;AACxB,UAAM;AACN,SAAK,QAAQ,GAAG;AAAA,EACjB;AAAA,EAEA,MAAa,QAAQ,KAA4B;AAChD,QAAI,KAAK,aAAa;AACrB;AAAA,IACD;AAEA,QAAI;AACH,YAAM,iBAAAC,QAAS,QAAQ,GAAG;AAC1B,WAAK,cAAc;AACnB,4BAAQ,MAAM,KAAK,gCAAgC;AAAA,IACpD,SAAS,OAAO;AACf,4BAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACP;AAAA,EACD;AAAA,EAEA,MAAa,aAA4B;AACxC,QAAI,CAAC,KAAK,aAAa;AACtB;AAAA,IACD;AAEA,QAAI;AACH,YAAM,iBAAAA,QAAS,WAAW;AAC1B,WAAK,cAAc;AACnB,4BAAQ,MAAM,KAAK,mCAAmC;AAAA,IACvD,SAAS,OAAO;AACf,4BAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACP;AAAA,EACD;AAAA,EAEA,MAAa,kBAAkB,WAA2C;AACzE,UAAM,QAAQ,MAAM,UAAU,KAAK,EAAE,UAAU,CAAC,EAAE,KAAK;AAAA,MACtD,WAAW;AAAA,IACZ,CAAC;AAED,0BAAQ,MAAM,KAAK,SAAS,MAAM,MAAM,sBAAsB,SAAS,EAAE;AAEzE,UAAM,gBAA+B,EAAE,OAAO,CAAC,EAAE;AACjD,UAAM,QAAQ,CAAC,SAAuB;AACrC,YAAM,SAAS,KAAK,KAAK,SAAS,KAAK,KAAK;AAC5C,oBAAc,MAAM,MAAM,IAAI;AAAA,QAC7B,MAAM,KAAK;AAAA,QACX,SAAS,KAAK;AAAA,QACd,WAAW,KAAK;AAAA,QAChB,UAAU,KAAK;AAAA,MAChB;AAAA,IACD,CAAC;AAED,WAAO;AAAA,EACR;AAAA,EAEA,MAAa,qBACZ,WACA,MACgB;AAChB,0BAAQ,MAAM,KAAK,wCAAwC,SAAS,EAAE;AACtE,0BAAQ,MAAM,KAAK,SAAS,KAAK,UAAU,IAAI,CAAC,EAAE;AAElD,UAAM,UAAU,OAAO;AAAA,MACtB;AAAA,MACA,MAAM,KAAK;AAAA,MACX,SAAS,KAAK;AAAA,MACd,WAAW,KAAK;AAAA,MAChB,UAAU,KAAK;AAAA,IAChB,CAAC;AAAA,EACF;AAAA,EAEA,MAAa,oBACZ,OACA,QACA,WACgB;AAEhB,UAAM,OAAmB;AAAA,MACxB;AAAA,MACA,SAAS;AAAA,QACR,MAAM;AAAA,QACN,OAAO,CAAC,KAAK;AAAA,MACd;AAAA,MACA,WAAW,KAAK,IAAI;AAAA,MACpB,UAAU;AAAA,QACT;AAAA,QACA;AAAA,MACD;AAAA,IACD;AAEA,UAAM,KAAK,qBAAqB,WAAW,IAAI;AAAA,EAChD;AACD;","names":["import_mongoose","ChatRole","mongoose","mongoose"]}

package/dist/index.d.cts

@@ -0,0 +1,14 @@
+import { BaseMemory } from '@ainetwork/adk/modules';
+import { SessionObject, ChatObject } from '@ainetwork/adk/types/memory';
+
+declare class MongoDBMemory extends BaseMemory {
+    private isConnected;
+    constructor(uri: string);
+    connect(uri: string): Promise<void>;
+    disconnect(): Promise<void>;
+    getSessionHistory(sessionId: string): Promise<SessionObject>;
+    updateSessionHistory(sessionId: string, chat: ChatObject): Promise<void>;
+    storeQueryAndIntent(query: string, intent: string, sessionId: string): Promise<void>;
+}
+
+export { MongoDBMemory };

package/dist/index.d.ts

@@ -0,0 +1,14 @@
+import { BaseMemory } from '@ainetwork/adk/modules';
+import { SessionObject, ChatObject } from '@ainetwork/adk/types/memory';
+
+declare class MongoDBMemory extends BaseMemory {
+    private isConnected;
+    constructor(uri: string);
+    connect(uri: string): Promise<void>;
+    disconnect(): Promise<void>;
+    getSessionHistory(sessionId: string): Promise<SessionObject>;
+    updateSessionHistory(sessionId: string, chat: ChatObject): Promise<void>;
+    storeQueryAndIntent(query: string, intent: string, sessionId: string): Promise<void>;
+}
+
+export { MongoDBMemory };

package/dist/index.js

@@ -0,0 +1,88 @@
+import {
+  ChatModel
+} from "./chunk-2SB2M62A.js";
+
+// index.ts
+import { BaseMemory } from "@ainetwork/adk/modules";
+import { loggers } from "@ainetwork/adk/utils/logger";
+import mongoose from "mongoose";
+var MongoDBMemory = class extends BaseMemory {
+  isConnected = false;
+  constructor(uri) {
+    super();
+    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) {
+    const chat = {
+      role: "USER" /* USER */,
+      content: {
+        type: "text",
+        parts: [query]
+      },
+      timestamp: Date.now(),
+      metadata: {
+        intent,
+        query
+      }
+    };
+    await this.updateSessionHistory(sessionId, chat);
+  }
+};
+export {
+  MongoDBMemory
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../index.ts"],"sourcesContent":["import { BaseMemory } from \"@ainetwork/adk/modules\";\nimport type { ChatObject, SessionObject } from \"@ainetwork/adk/types/memory\";\nimport { loggers } from \"@ainetwork/adk/utils/logger\";\nimport mongoose from \"mongoose\";\nimport {\n\ttype ChatDocument,\n\tChatModel,\n\tChatRole,\n} from \"./models/chats.model\";\n\nexport class MongoDBMemory extends BaseMemory {\n\tprivate isConnected = false;\n\n\tconstructor(uri: string) {\n\t\tsuper();\n\t\tthis.connect(uri);\n\t}\n\n\tpublic async connect(uri: string): Promise<void> {\n\t\tif (this.isConnected) {\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tawait mongoose.connect(uri);\n\t\t\tthis.isConnected = true;\n\t\t\tloggers.agent.info(\"MongoDB connected successfully\");\n\t\t} catch (error) {\n\t\t\tloggers.agent.error(\"Failed to connect to MongoDB:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tpublic async disconnect(): Promise<void> {\n\t\tif (!this.isConnected) {\n\t\t\treturn;\n\t\t}\n\n\t\ttry {\n\t\t\tawait mongoose.disconnect();\n\t\t\tthis.isConnected = false;\n\t\t\tloggers.agent.info(\"MongoDB disconnected successfully\");\n\t\t} catch (error) {\n\t\t\tloggers.agent.error(\"Failed to disconnect from MongoDB:\", error);\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tpublic async getSessionHistory(sessionId: string): Promise<SessionObject> {\n\t\tconst chats = await ChatModel.find({ sessionId }).sort({\n\t\t\ttimestamp: 1,\n\t\t});\n\n\t\tloggers.agent.info(`Found ${chats.length} chats for session ${sessionId}`);\n\n\t\tconst sessionObject: SessionObject = { chats: {} };\n\t\tchats.forEach((chat: ChatDocument) => {\n\t\t\tconst chatId = chat._id?.toString() || chat.id;\n\t\t\tsessionObject.chats[chatId] = {\n\t\t\t\trole: chat.role as ChatRole,\n\t\t\t\tcontent: chat.content,\n\t\t\t\ttimestamp: chat.timestamp,\n\t\t\t\tmetadata: chat.metadata,\n\t\t\t};\n\t\t});\n\n\t\treturn sessionObject;\n\t}\n\n\tpublic async updateSessionHistory(\n\t\tsessionId: string,\n\t\tchat: ChatObject,\n\t): Promise<void> {\n\t\tloggers.agent.info(`Updating session history for session ${sessionId}`);\n\t\tloggers.agent.info(`Chat: ${JSON.stringify(chat)}`);\n\n\t\tawait ChatModel.create({\n\t\t\tsessionId,\n\t\t\trole: chat.role,\n\t\t\tcontent: chat.content,\n\t\t\ttimestamp: chat.timestamp,\n\t\t\tmetadata: chat.metadata,\n\t\t});\n\t}\n\n\tpublic async storeQueryAndIntent(\n\t\tquery: string,\n\t\tintent: string,\n\t\tsessionId: string,\n\t): Promise<void> {\n\t\t// Intent 정보를 metadata에 저장\n\t\tconst chat: ChatObject = {\n\t\t\trole: ChatRole.USER,\n\t\t\tcontent: {\n\t\t\t\ttype: \"text\",\n\t\t\t\tparts: [query],\n\t\t\t},\n\t\t\ttimestamp: Date.now(),\n\t\t\tmetadata: {\n\t\t\t\tintent,\n\t\t\t\tquery,\n\t\t\t},\n\t\t};\n\n\t\tawait this.updateSessionHistory(sessionId, chat);\n\t}\n}\n"],"mappings":";;;;;AAAA,SAAS,kBAAkB;AAE3B,SAAS,eAAe;AACxB,OAAO,cAAc;AAOd,IAAM,gBAAN,cAA4B,WAAW;AAAA,EACrC,cAAc;AAAA,EAEtB,YAAY,KAAa;AACxB,UAAM;AACN,SAAK,QAAQ,GAAG;AAAA,EACjB;AAAA,EAEA,MAAa,QAAQ,KAA4B;AAChD,QAAI,KAAK,aAAa;AACrB;AAAA,IACD;AAEA,QAAI;AACH,YAAM,SAAS,QAAQ,GAAG;AAC1B,WAAK,cAAc;AACnB,cAAQ,MAAM,KAAK,gCAAgC;AAAA,IACpD,SAAS,OAAO;AACf,cAAQ,MAAM,MAAM,iCAAiC,KAAK;AAC1D,YAAM;AAAA,IACP;AAAA,EACD;AAAA,EAEA,MAAa,aAA4B;AACxC,QAAI,CAAC,KAAK,aAAa;AACtB;AAAA,IACD;AAEA,QAAI;AACH,YAAM,SAAS,WAAW;AAC1B,WAAK,cAAc;AACnB,cAAQ,MAAM,KAAK,mCAAmC;AAAA,IACvD,SAAS,OAAO;AACf,cAAQ,MAAM,MAAM,sCAAsC,KAAK;AAC/D,YAAM;AAAA,IACP;AAAA,EACD;AAAA,EAEA,MAAa,kBAAkB,WAA2C;AACzE,UAAM,QAAQ,MAAM,UAAU,KAAK,EAAE,UAAU,CAAC,EAAE,KAAK;AAAA,MACtD,WAAW;AAAA,IACZ,CAAC;AAED,YAAQ,MAAM,KAAK,SAAS,MAAM,MAAM,sBAAsB,SAAS,EAAE;AAEzE,UAAM,gBAA+B,EAAE,OAAO,CAAC,EAAE;AACjD,UAAM,QAAQ,CAAC,SAAuB;AACrC,YAAM,SAAS,KAAK,KAAK,SAAS,KAAK,KAAK;AAC5C,oBAAc,MAAM,MAAM,IAAI;AAAA,QAC7B,MAAM,KAAK;AAAA,QACX,SAAS,KAAK;AAAA,QACd,WAAW,KAAK;AAAA,QAChB,UAAU,KAAK;AAAA,MAChB;AAAA,IACD,CAAC;AAED,WAAO;AAAA,EACR;AAAA,EAEA,MAAa,qBACZ,WACA,MACgB;AAChB,YAAQ,MAAM,KAAK,wCAAwC,SAAS,EAAE;AACtE,YAAQ,MAAM,KAAK,SAAS,KAAK,UAAU,IAAI,CAAC,EAAE;AAElD,UAAM,UAAU,OAAO;AAAA,MACtB;AAAA,MACA,MAAM,KAAK;AAAA,MACX,SAAS,KAAK;AAAA,MACd,WAAW,KAAK;AAAA,MAChB,UAAU,KAAK;AAAA,IAChB,CAAC;AAAA,EACF;AAAA,EAEA,MAAa,oBACZ,OACA,QACA,WACgB;AAEhB,UAAM,OAAmB;AAAA,MACxB;AAAA,MACA,SAAS;AAAA,QACR,MAAM;AAAA,QACN,OAAO,CAAC,KAAK;AAAA,MACd;AAAA,MACA,WAAW,KAAK,IAAI;AAAA,MACpB,UAAU;AAAA,QACT;AAAA,QACA;AAAA,MACD;AAAA,IACD;AAEA,UAAM,KAAK,qBAAqB,WAAW,IAAI;AAAA,EAChD;AACD;","names":[]}

package/dist/models/chats.model.cjs

@@ -0,0 +1,86 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// models/chats.model.ts
+var chats_model_exports = {};
+__export(chats_model_exports, {
+  ChatModel: () => ChatModel,
+  ChatRole: () => ChatRole
+});
+module.exports = __toCommonJS(chats_model_exports);
+var import_mongoose = __toESM(require("mongoose"), 1);
+var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
+  ChatRole2["USER"] = "USER";
+  ChatRole2["SYSTEM"] = "SYSTEM";
+  ChatRole2["MODEL"] = "MODEL";
+  return ChatRole2;
+})(ChatRole || {});
+var ChatContentObjectSchema = new import_mongoose.Schema(
+  {
+    type: { type: String, required: true },
+    parts: { type: [import_mongoose.Schema.Types.Mixed], required: true }
+  },
+  { _id: false }
+);
+var ChatObjectSchema = new import_mongoose.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: import_mongoose.Schema.Types.Mixed,
+      default: {}
+    }
+  },
+  {
+    timestamps: true
+  }
+);
+var ChatModel = import_mongoose.default.model("Chat", ChatObjectSchema);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ChatModel,
+  ChatRole
+});
+//# sourceMappingURL=chats.model.cjs.map

package/dist/models/chats.model.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../models/chats.model.ts"],"sourcesContent":["import mongoose, { type Document, Schema } from \"mongoose\";\n\n// ChatRole enum\nexport enum ChatRole {\n\tUSER = \"USER\",\n\tSYSTEM = \"SYSTEM\",\n\tMODEL = \"MODEL\",\n}\n\n// ChatContentObject schema\nconst ChatContentObjectSchema = new Schema(\n\t{\n\t\ttype: { type: String, required: true },\n\t\tparts: { type: [Schema.Types.Mixed], required: true },\n\t},\n\t{ _id: false },\n);\n\n// ChatObject schema - 개별 문서로 저장\nconst ChatObjectSchema = new Schema(\n\t{\n\t\tsessionId: {\n\t\t\ttype: String,\n\t\t\trequired: true,\n\t\t\tindex: true,\n\t\t},\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: Object.values(ChatRole),\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: ChatContentObjectSchema,\n\t\t\trequired: true,\n\t\t},\n\t\ttimestamp: {\n\t\t\ttype: Number,\n\t\t\trequired: true,\n\t\t},\n\t\tmetadata: {\n\t\t\ttype: Schema.Types.Mixed,\n\t\t\tdefault: {},\n\t\t},\n\t},\n\t{\n\t\ttimestamps: true,\n\t},\n);\n\n// Chat Document interface\nexport interface ChatDocument extends Document {\n\tsessionId: string;\n\trole: ChatRole;\n\tcontent: {\n\t\ttype: string;\n\t\tparts: any[];\n\t};\n\ttimestamp: number;\n\tmetadata?: { [key: string]: unknown };\n\tcreatedAt: Date;\n\tupdatedAt: Date;\n}\n\n// Export the model\nexport const ChatModel = mongoose.model<ChatDocument>(\"Chat\", ChatObjectSchema);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,sBAAgD;AAGzC,IAAK,WAAL,kBAAKA,cAAL;AACN,EAAAA,UAAA,UAAO;AACP,EAAAA,UAAA,YAAS;AACT,EAAAA,UAAA,WAAQ;AAHG,SAAAA;AAAA,GAAA;AAOZ,IAAM,0BAA0B,IAAI;AAAA,EACnC;AAAA,IACC,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,OAAO,EAAE,MAAM,CAAC,uBAAO,MAAM,KAAK,GAAG,UAAU,KAAK;AAAA,EACrD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAGA,IAAM,mBAAmB,IAAI;AAAA,EAC5B;AAAA,IACC,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,MACV,OAAO;AAAA,IACR;AAAA,IACA,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,OAAO,OAAO,QAAQ;AAAA,MAC5B,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,WAAW;AAAA,MACV,MAAM;AAAA,MACN,UAAU;AAAA,IACX;AAAA,IACA,UAAU;AAAA,MACT,MAAM,uBAAO,MAAM;AAAA,MACnB,SAAS,CAAC;AAAA,IACX;AAAA,EACD;AAAA,EACA;AAAA,IACC,YAAY;AAAA,EACb;AACD;AAiBO,IAAM,YAAY,gBAAAC,QAAS,MAAoB,QAAQ,gBAAgB;","names":["ChatRole","mongoose"]}

package/dist/models/chats.model.d.cts

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

package/dist/models/chats.model.d.ts

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

package/dist/models/chats.model.js

@@ -0,0 +1,9 @@
+import {
+  ChatModel,
+  ChatRole
+} from "../chunk-2SB2M62A.js";
+export {
+  ChatModel,
+  ChatRole
+};
+//# sourceMappingURL=chats.model.js.map

package/dist/models/chats.model.js.map

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

package/dist/models/intentTriggeringInfos.model.cjs

@@ -0,0 +1,69 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// models/intentTriggeringInfos.model.ts
+var intentTriggeringInfos_model_exports = {};
+__export(intentTriggeringInfos_model_exports, {
+  IntentTriggeringInfoModel: () => IntentTriggeringInfoModel
+});
+module.exports = __toCommonJS(intentTriggeringInfos_model_exports);
+var import_mongoose = __toESM(require("mongoose"), 1);
+var MessageSchema = new import_mongoose.Schema(
+  {
+    role: {
+      type: String,
+      enum: ["system", "user", "assistant", "tool", "function"],
+      required: true
+    },
+    content: {
+      type: import_mongoose.Schema.Types.Mixed,
+      // string 또는 object array
+      required: true
+    }
+  },
+  { _id: false }
+);
+var IntentTriggeringInfoSchema = new import_mongoose.Schema({
+  context: {
+    messages: [MessageSchema]
+  },
+  intent: {
+    name: { type: String, required: true },
+    description: { type: String, required: true }
+  }
+});
+var IntentTriggeringInfoModel = import_mongoose.default.model(
+  "IntentTriggeringInfo",
+  IntentTriggeringInfoSchema
+);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  IntentTriggeringInfoModel
+});
+//# sourceMappingURL=intentTriggeringInfos.model.cjs.map

package/dist/models/intentTriggeringInfos.model.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../models/intentTriggeringInfos.model.ts"],"sourcesContent":["import mongoose, { type Document, Schema } from \"mongoose\";\n\nconst MessageSchema = new Schema(\n\t{\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: [\"system\", \"user\", \"assistant\", \"tool\", \"function\"],\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: Schema.Types.Mixed, // string 또는 object array\n\t\t\trequired: true,\n\t\t},\n\t},\n\t{ _id: false },\n);\n\nexport interface IntentTriggeringInfoDocument extends Document {\n\tcontext: {\n\t\tmessages: Array<{\n\t\t\trole: string;\n\t\t\tcontent:\n\t\t\t\t| string\n\t\t\t\t| Array<\n\t\t\t\t\t\t| { type: \"text\"; text: string }\n\t\t\t\t\t\t| { type: \"image_url\"; image_url: { url: string } }\n\t\t\t\t  >;\n\t\t}>;\n\t};\n\tintent: {\n\t\tname: string;\n\t\tdescription: string;\n\t};\n}\n\nconst IntentTriggeringInfoSchema = new Schema<IntentTriggeringInfoDocument>({\n\tcontext: {\n\t\tmessages: [MessageSchema],\n\t},\n\tintent: {\n\t\tname: { type: String, required: true },\n\t\tdescription: { type: String, required: true },\n\t},\n});\n\nexport const IntentTriggeringInfoModel =\n\tmongoose.model<IntentTriggeringInfoDocument>(\n\t\t\"IntentTriggeringInfo\",\n\t\tIntentTriggeringInfoSchema,\n\t);\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,sBAAgD;AAEhD,IAAM,gBAAgB,IAAI;AAAA,EACzB;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,CAAC,UAAU,QAAQ,aAAa,QAAQ,UAAU;AAAA,MACxD,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM,uBAAO,MAAM;AAAA;AAAA,MACnB,UAAU;AAAA,IACX;AAAA,EACD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAoBA,IAAM,6BAA6B,IAAI,uBAAqC;AAAA,EAC3E,SAAS;AAAA,IACR,UAAU,CAAC,aAAa;AAAA,EACzB;AAAA,EACA,QAAQ;AAAA,IACP,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,aAAa,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,EAC7C;AACD,CAAC;AAEM,IAAM,4BACZ,gBAAAA,QAAS;AAAA,EACR;AAAA,EACA;AACD;","names":["mongoose"]}

package/dist/models/intentTriggeringInfos.model.d.cts

@@ -0,0 +1,29 @@
+import mongoose, { Document } from 'mongoose';
+
+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;
+    };
+}
+declare const IntentTriggeringInfoModel: mongoose.Model<IntentTriggeringInfoDocument, {}, {}, {}, mongoose.Document<unknown, {}, IntentTriggeringInfoDocument, {}> & IntentTriggeringInfoDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+
+export { type IntentTriggeringInfoDocument, IntentTriggeringInfoModel };

package/dist/models/intentTriggeringInfos.model.d.ts

@@ -0,0 +1,29 @@
+import mongoose, { Document } from 'mongoose';
+
+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;
+    };
+}
+declare const IntentTriggeringInfoModel: mongoose.Model<IntentTriggeringInfoDocument, {}, {}, {}, mongoose.Document<unknown, {}, IntentTriggeringInfoDocument, {}> & IntentTriggeringInfoDocument & Required<{
+    _id: unknown;
+}> & {
+    __v: number;
+}, any>;
+
+export { type IntentTriggeringInfoDocument, IntentTriggeringInfoModel };

package/dist/models/intentTriggeringInfos.model.js

@@ -0,0 +1,34 @@
+// models/intentTriggeringInfos.model.ts
+import mongoose, { Schema } from "mongoose";
+var 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 }
+);
+var IntentTriggeringInfoSchema = new Schema({
+  context: {
+    messages: [MessageSchema]
+  },
+  intent: {
+    name: { type: String, required: true },
+    description: { type: String, required: true }
+  }
+});
+var IntentTriggeringInfoModel = mongoose.model(
+  "IntentTriggeringInfo",
+  IntentTriggeringInfoSchema
+);
+export {
+  IntentTriggeringInfoModel
+};
+//# sourceMappingURL=intentTriggeringInfos.model.js.map

package/dist/models/intentTriggeringInfos.model.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../models/intentTriggeringInfos.model.ts"],"sourcesContent":["import mongoose, { type Document, Schema } from \"mongoose\";\n\nconst MessageSchema = new Schema(\n\t{\n\t\trole: {\n\t\t\ttype: String,\n\t\t\tenum: [\"system\", \"user\", \"assistant\", \"tool\", \"function\"],\n\t\t\trequired: true,\n\t\t},\n\t\tcontent: {\n\t\t\ttype: Schema.Types.Mixed, // string 또는 object array\n\t\t\trequired: true,\n\t\t},\n\t},\n\t{ _id: false },\n);\n\nexport interface IntentTriggeringInfoDocument extends Document {\n\tcontext: {\n\t\tmessages: Array<{\n\t\t\trole: string;\n\t\t\tcontent:\n\t\t\t\t| string\n\t\t\t\t| Array<\n\t\t\t\t\t\t| { type: \"text\"; text: string }\n\t\t\t\t\t\t| { type: \"image_url\"; image_url: { url: string } }\n\t\t\t\t  >;\n\t\t}>;\n\t};\n\tintent: {\n\t\tname: string;\n\t\tdescription: string;\n\t};\n}\n\nconst IntentTriggeringInfoSchema = new Schema<IntentTriggeringInfoDocument>({\n\tcontext: {\n\t\tmessages: [MessageSchema],\n\t},\n\tintent: {\n\t\tname: { type: String, required: true },\n\t\tdescription: { type: String, required: true },\n\t},\n});\n\nexport const IntentTriggeringInfoModel =\n\tmongoose.model<IntentTriggeringInfoDocument>(\n\t\t\"IntentTriggeringInfo\",\n\t\tIntentTriggeringInfoSchema,\n\t);\n"],"mappings":";AAAA,OAAO,YAA2B,cAAc;AAEhD,IAAM,gBAAgB,IAAI;AAAA,EACzB;AAAA,IACC,MAAM;AAAA,MACL,MAAM;AAAA,MACN,MAAM,CAAC,UAAU,QAAQ,aAAa,QAAQ,UAAU;AAAA,MACxD,UAAU;AAAA,IACX;AAAA,IACA,SAAS;AAAA,MACR,MAAM,OAAO,MAAM;AAAA;AAAA,MACnB,UAAU;AAAA,IACX;AAAA,EACD;AAAA,EACA,EAAE,KAAK,MAAM;AACd;AAoBA,IAAM,6BAA6B,IAAI,OAAqC;AAAA,EAC3E,SAAS;AAAA,IACR,UAAU,CAAC,aAAa;AAAA,EACzB;AAAA,EACA,QAAQ;AAAA,IACP,MAAM,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,IACrC,aAAa,EAAE,MAAM,QAAQ,UAAU,KAAK;AAAA,EAC7C;AACD,CAAC;AAEM,IAAM,4BACZ,SAAS;AAAA,EACR;AAAA,EACA;AACD;","names":[]}

package/index.ts

@@ -0,0 +1,107 @@
+import { BaseMemory } from "@ainetwork/adk/modules";
+import type { ChatObject, SessionObject } from "@ainetwork/adk/types/memory";
+import { loggers } from "@ainetwork/adk/utils/logger";
+import mongoose from "mongoose";
+import {
+	type ChatDocument,
+	ChatModel,
+	ChatRole,
+} from "./models/chats.model";
+
+export class MongoDBMemory extends BaseMemory {
+	private isConnected = false;
+
+	constructor(uri: string) {
+		super();
+		this.connect(uri);
+	}
+
+	public async connect(uri: string): Promise<void> {
+		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;
+		}
+	}
+
+	public async disconnect(): Promise<void> {
+		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;
+		}
+	}
+
+	public async getSessionHistory(sessionId: string): Promise<SessionObject> {
+		const chats = await ChatModel.find({ sessionId }).sort({
+			timestamp: 1,
+		});
+
+		loggers.agent.info(`Found ${chats.length} chats for session ${sessionId}`);
+
+		const sessionObject: SessionObject = { chats: {} };
+		chats.forEach((chat: ChatDocument) => {
+			const chatId = chat._id?.toString() || chat.id;
+			sessionObject.chats[chatId] = {
+				role: chat.role as ChatRole,
+				content: chat.content,
+				timestamp: chat.timestamp,
+				metadata: chat.metadata,
+			};
+		});
+
+		return sessionObject;
+	}
+
+	public async updateSessionHistory(
+		sessionId: string,
+		chat: ChatObject,
+	): Promise<void> {
+		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,
+		});
+	}
+
+	public async storeQueryAndIntent(
+		query: string,
+		intent: string,
+		sessionId: string,
+	): Promise<void> {
+		// Intent 정보를 metadata에 저장
+		const chat: ChatObject = {
+			role: ChatRole.USER,
+			content: {
+				type: "text",
+				parts: [query],
+			},
+			timestamp: Date.now(),
+			metadata: {
+				intent,
+				query,
+			},
+		};
+
+		await this.updateSessionHistory(sessionId, chat);
+	}
+}

package/models/chats.model.ts

@@ -0,0 +1,65 @@
+import mongoose, { type Document, Schema } from "mongoose";
+
+// ChatRole enum
+export enum ChatRole {
+	USER = "USER",
+	SYSTEM = "SYSTEM",
+	MODEL = "MODEL",
+}
+
+// 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,
+	},
+);
+
+// Chat Document interface
+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 the model
+export const ChatModel = mongoose.model<ChatDocument>("Chat", ChatObjectSchema);

package/models/intentTriggeringInfos.model.ts

@@ -0,0 +1,50 @@
+import mongoose, { type Document, 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 },
+);
+
+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;
+	};
+}
+
+const IntentTriggeringInfoSchema = new Schema<IntentTriggeringInfoDocument>({
+	context: {
+		messages: [MessageSchema],
+	},
+	intent: {
+		name: { type: String, required: true },
+		description: { type: String, required: true },
+	},
+});
+
+export const IntentTriggeringInfoModel =
+	mongoose.model<IntentTriggeringInfoDocument>(
+		"IntentTriggeringInfo",
+		IntentTriggeringInfoSchema,
+	);

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@ainetwork/adk-provider-memory-mongodb",
+  "version": "0.1.1",
+  "author": "AI Network (https://ainetwork.ai)",
+  "type": "module",
+  "engines": {
+    "node": ">=20"
+  },
+  "main": "./dist/index.cjs",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    },
+    "./models/*": {
+      "types": "./dist/models/*.d.ts",
+      "import": "./dist/models/*.js",
+      "require": "./dist/models/*.cjs"
+    }
+  },
+  "scripts": {
+    "build": "tsup",
+    "clean": "rm -rf dist"
+  },
+  "dependencies": {
+    "@ainetwork/adk": "^0.1.2",
+    "mongoose": "^8.16.5"
+  },
+  "devDependencies": {
+    "typescript": "^5.0.0"
+  },
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "gitHead": "cb6205738fd84533ce56bbeeab00ea9dfe2aaf3a"
+}

package/tsconfig.json

@@ -0,0 +1,9 @@
+{
+	"extends": "../../../tsconfig.json",
+	"compilerOptions": {
+		"outDir": "./dist",
+		"rootDir": ".",
+	},
+	"include": ["index.ts", "models/**/*.ts"],
+	"exclude": ["node_modules", "dist"]
+}

package/tsup.config.ts

@@ -0,0 +1,9 @@
+import { defineConfig } from 'tsup'
+
+export default defineConfig({
+  entry: ['index.ts', 'models/**/*.ts'],
+  format: ['cjs', 'esm'],
+  dts: true,
+  sourcemap: true,
+  clean: true,
+})