@multiplayer-app/ai-agent-mongo 0.1.0-beta.1 → 0.1.0-beta.3

package/dist/cjs/mongo/index.js

@@ -0,0 +1,112 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var _a;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.connected = exports.Config = exports.Model = exports.Schema = exports.ObjectId = exports.mongoose = exports.encryption = void 0;
+const mongoose_1 = __importDefault(require("mongoose"));
+const logger_js_1 = require("./logger.js");
+const _Config = __importStar(require("./config.js"));
+const _encryption = __importStar(require("./encryption.js"));
+const logger = (0, logger_js_1.getLogger)();
+exports.encryption = _encryption;
+const MONGODB_URI_MASKED = (_a = _Config.MONGODB_URI) === null || _a === void 0 ? void 0 : _a.replace(/:\/\/(.*):(.*)@/, '://***:***@');
+exports.mongoose = mongoose_1.default;
+exports.ObjectId = mongoose_1.default.Types.ObjectId;
+exports.Schema = mongoose_1.default.Schema;
+exports.Model = mongoose_1.default.Model;
+exports.Config = _Config;
+mongoose_1.default.set('debug', _Config.MONGO_DEBUG);
+mongoose_1.default.connection.on('connected', () => {
+    logger.info(`[MONGO] Connected to ${MONGODB_URI_MASKED}`);
+});
+mongoose_1.default.connection.on('error', error => {
+    logger.error(error, `[MONGO] Connection to MongoDB failed: ${error.message}`);
+});
+mongoose_1.default.connection.on('reconnected', () => {
+    logger.info(`[MONGO] Reconnected to ${MONGODB_URI_MASKED}`);
+});
+mongoose_1.default.connection.on('disconnected', () => {
+    logger.info(`[MONGO] Disconnected from ${MONGODB_URI_MASKED}`);
+});
+mongoose_1.default.connection.on('close', () => {
+    logger.info(`[MONGO] Connection closed ${MONGODB_URI_MASKED}`);
+});
+mongoose_1.default.connection.on('reconnectFailed', () => {
+    logger.error(`[MONGO] Reconnect Failed ${MONGODB_URI_MASKED}`);
+});
+const connect = async () => {
+    await exports.encryption.connect();
+    const encryptionKeyId = await exports.encryption.getKeyId();
+    const options = {
+        autoIndex: true,
+        connectTimeoutMS: 10000,
+        socketTimeoutMS: 25000,
+        minPoolSize: 3,
+        autoEncryption: {
+            bypassAutoEncryption: true,
+            keyVaultNamespace: exports.encryption.keyVaultNamespace,
+            kmsProviders: exports.encryption.kmsProviders,
+            schemaMap: exports.encryption.getEncryptionSchemaMap(encryptionKeyId),
+            extraOptions: {
+                mongocryptdURI: exports.Config.MONGODB_URI,
+            },
+        },
+    };
+    await mongoose_1.default.connect(_Config.MONGODB_URI, options);
+};
+const connected = () => mongoose_1.default.connection.readyState === 1;
+exports.connected = connected;
+const disconnect = async () => {
+    if (!(0, exports.connected)())
+        return;
+    await Promise.allSettled([
+        mongoose_1.default.connection.close(),
+        exports.encryption.disconnect(),
+    ]);
+};
+exports.default = {
+    encryption: exports.encryption,
+    connect,
+    disconnect,
+    connected: exports.connected,
+    mongoose: exports.mongoose,
+    ObjectId: mongoose_1.default.Types.ObjectId,
+    Schema: mongoose_1.default.Schema,
+    Model: mongoose_1.default.Model,
+};
+//# sourceMappingURL=index.js.map

package/dist/cjs/mongo/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/mongo/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,wDAAmD;AACnD,2CAAuC;AACvC,qDAAsC;AACtC,6DAA8C;AAE9C,MAAM,MAAM,GAAG,IAAA,qBAAS,GAAE,CAAA;AAEb,QAAA,UAAU,GAAG,WAAW,CAAA;AAErC,MAAM,kBAAkB,GAAG,MAAA,OAAO,CAAC,WAAW,0CAAE,OAAO,CAAC,iBAAiB,EAAE,aAAa,CAAC,CAAA;AAE5E,QAAA,QAAQ,GAAG,kBAAQ,CAAA;AAEnB,QAAA,QAAQ,GAAG,kBAAQ,CAAC,KAAK,CAAC,QAAQ,CAAA;AAQlC,QAAA,MAAM,GAAG,kBAAQ,CAAC,MAAM,CAAA;AAExB,QAAA,KAAK,GAAG,kBAAQ,CAAC,KAAK,CAAA;AAEtB,QAAA,MAAM,GAAG,OAAO,CAAA;AAE7B,kBAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,WAAW,CAAC,CAAA;AAE1C,kBAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;IACvC,MAAM,CAAC,IAAI,CAAC,wBAAwB,kBAAkB,EAAE,CAAC,CAAA;AAC3D,CAAC,CAAC,CAAA;AAEF,kBAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE;IACtC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,yCAAyC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAA;AAC/E,CAAC,CAAC,CAAA;AAEF,kBAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,aAAa,EAAE,GAAG,EAAE;IACzC,MAAM,CAAC,IAAI,CAAC,0BAA0B,kBAAkB,EAAE,CAAC,CAAA;AAC7D,CAAC,CAAC,CAAA;AAEF,kBAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE;IAC1C,MAAM,CAAC,IAAI,CAAC,6BAA6B,kBAAkB,EAAE,CAAC,CAAA;AAChE,CAAC,CAAC,CAAA;AAEF,kBAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;IACnC,MAAM,CAAC,IAAI,CAAC,6BAA6B,kBAAkB,EAAE,CAAC,CAAA;AAChE,CAAC,CAAC,CAAA;AAEF,kBAAQ,CAAC,UAAU,CAAC,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC7C,MAAM,CAAC,KAAK,CAAC,4BAA4B,kBAAkB,EAAE,CAAC,CAAA;AAChE,CAAC,CAAC,CAAA;AAEF,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;IACzB,MAAM,kBAAU,CAAC,OAAO,EAAE,CAAA;IAE1B,MAAM,eAAe,GAAG,MAAM,kBAAU,CAAC,QAAQ,EAAE,CAAA;IAEnD,MAAM,OAAO,GAAmB;QAC9B,SAAS,EAAE,IAAI;QACf,gBAAgB,EAAE,KAAK;QACvB,eAAe,EAAE,KAAK;QACtB,WAAW,EAAE,CAAC;QACd,cAAc,EAAE;YACd,oBAAoB,EAAE,IAAI;YAC1B,iBAAiB,EAAE,kBAAU,CAAC,iBAAiB;YAC/C,YAAY,EAAE,kBAAU,CAAC,YAAY;YACrC,SAAS,EAAE,kBAAU,CAAC,sBAAsB,CAAC,eAAe,CAAC;YAC7D,YAAY,EAAE;gBACZ,cAAc,EAAE,cAAM,CAAC,WAAW;aACnC;SACF;KACF,CAAA;IACD,MAAM,kBAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC,CAAA;AACtD,CAAC,CAAA;AAEM,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,kBAAQ,CAAC,UAAU,CAAC,UAAU,KAAK,CAAC,CAAA;AAAtD,QAAA,SAAS,aAA6C;AAEnE,MAAM,UAAU,GAAG,KAAK,IAAI,EAAE;IAC5B,IAAI,CAAC,IAAA,iBAAS,GAAE;QAAE,OAAM;IACxB,MAAM,OAAO,CAAC,UAAU,CAAC;QACvB,kBAAQ,CAAC,UAAU,CAAC,KAAK,EAAE;QAC3B,kBAAU,CAAC,UAAU,EAAE;KACxB,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,kBAAe;IACb,UAAU,EAAV,kBAAU;IACV,OAAO;IACP,UAAU;IACV,SAAS,EAAT,iBAAS;IACT,QAAQ,EAAR,gBAAQ;IACR,QAAQ,EAAE,kBAAQ,CAAC,KAAK,CAAC,QAAQ;IACjC,MAAM,EAAE,kBAAQ,CAAC,MAAM;IACvB,KAAK,EAAE,kBAAQ,CAAC,KAAK;CACtB,CAAA"}

package/dist/cjs/mongo/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../src/mongo/logger.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,WAAW,WAAW;IAC1B,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,KAAK,CAAC,cAAc,EAAE,KAAK,GAAG,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxE,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,KAAK,CAAC,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;CAC/B;AAeD;;;;;;;;;;;;;GAaG;AACH,wBAAgB,SAAS,CAAC,YAAY,EAAE,WAAW,GAAG,IAAI,CAEzD;AAED;;;GAGG;AACH,wBAAgB,SAAS,IAAI,WAAW,CAEvC"}

package/dist/cjs/mongo/logger.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setLogger = setLogger;
+exports.getLogger = getLogger;
+/**
+ * No-op logger that discards all log messages.
+ * Used as default when no logger is provided.
+ */
+const noOpLogger = {
+    info: () => { },
+    error: () => { },
+    warn: () => { },
+    debug: () => { },
+};
+let logger = noOpLogger;
+/**
+ * Sets a custom logger for MongoDB operations.
+ * If not called, all logging is silently discarded.
+ *
+ * @param customLogger - Logger instance implementing MongoLogger interface
+ *
+ * @example
+ * ```ts
+ * import { setLogger } from '@multiplayer-app/ai-agent-mongo';
+ * import pino from 'pino';
+ *
+ * setLogger(pino());
+ * ```
+ */
+function setLogger(customLogger) {
+    logger = customLogger;
+}
+/**
+ * Gets the current logger instance.
+ * @internal
+ */
+function getLogger() {
+    return logger;
+}
+//# sourceMappingURL=logger.js.map

package/dist/cjs/mongo/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../../src/mongo/logger.ts"],"names":[],"mappings":";;AAsCA,8BAEC;AAMD,8BAEC;AArCD;;;GAGG;AACH,MAAM,UAAU,GAAgB;IAC9B,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC;IACd,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;IACf,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC;IACd,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;CAChB,CAAC;AAEF,IAAI,MAAM,GAAgB,UAAU,CAAC;AAErC;;;;;;;;;;;;;GAaG;AACH,SAAgB,SAAS,CAAC,YAAyB;IACjD,MAAM,GAAG,YAAY,CAAC;AACxB,CAAC;AAED;;;GAGG;AACH,SAAgB,SAAS;IACvB,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/cjs/repositories/MongoAgentChatRepository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentChatRepository.d.ts","sourceRoot":"","sources":["../../../src/repositories/MongoAgentChatRepository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAC;AACpF,OAAO,EAAY,SAAS,EAAE,MAAM,iCAAiC,CAAC;AACtE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AA0BxE;;GAEG;AACH,qBAAa,wBAAyB,YAAW,mBAAmB;IAC5D,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAK/C,IAAI,CACR,MAAM,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,EAC3B,OAAO,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,SAAS,CAAA;SAAE,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,OAAO,CAAC,SAAS,EAAE,CAAC;IA4BjB,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAa9D,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;IAgBrF,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAUpG,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAUpC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpC,KAAK,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAcnD,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAKlD,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAK1D,yBAAyB,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAKnF,0BAA0B,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAQpE,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAUjE,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAK/C,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAKvD,gBAAgB,CACpB,MAAM,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,EAC3B,OAAO,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,SAAS,CAAA;SAAE,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACvE,OAAO,CAAC,iBAAiB,EAAE,CAAC;CA2GhC"}

package/dist/cjs/repositories/MongoAgentChatRepository.js

@@ -0,0 +1,239 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MongoAgentChatRepository = void 0;
+const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
+const AgentChatSchema_js_1 = require("../schemas/AgentChatSchema.js");
+const AgentMessageSchema_js_1 = require("../schemas/AgentMessageSchema.js");
+const index_js_1 = require("../mongo/index.js");
+/**
+ * Helper function to convert Mongoose document to AgentChat type
+ */
+function toAgentChat(doc) {
+    if (!doc)
+        return null;
+    const chat = doc.toObject ? doc.toObject() : doc;
+    return {
+        id: doc._id.toString(),
+        title: chat.title,
+        type: chat.type || ai_agent_types_1.ChatType.Chat,
+        status: chat.status,
+        contextKey: chat.contextKey,
+        model: chat.model,
+        createdAt: chat.createdAt,
+        updatedAt: chat.updatedAt,
+        metadata: chat.metadata,
+        userId: chat.userId
+    };
+}
+/**
+ * MongoDB implementation of AgentChatRepository
+ */
+class MongoAgentChatRepository {
+    async findById(id) {
+        const doc = await AgentChatSchema_js_1.AgentChatModel.findById(id).exec();
+        return toAgentChat(doc);
+    }
+    async find(filter, options) {
+        const query = {};
+        if (filter) {
+            if (filter.id) {
+                query._id = filter.id;
+                delete filter.id;
+            }
+            Object.assign(query, filter);
+        }
+        let queryBuilder = AgentChatSchema_js_1.AgentChatModel.find(query);
+        // Apply sorting
+        if (options === null || options === void 0 ? void 0 : options.sort) {
+            const sortOrder = options.sort.order === ai_agent_types_1.SortOrder.Asc ? 1 : -1;
+            queryBuilder = queryBuilder.sort({ [options.sort.field]: sortOrder });
+        }
+        // Apply limit
+        if (options === null || options === void 0 ? void 0 : options.limit) {
+            queryBuilder = queryBuilder.limit(options.limit);
+        }
+        const docs = await queryBuilder.exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async findOne(filter) {
+        const query = {};
+        if (filter.id) {
+            query._id = filter.id;
+            delete filter.id;
+        }
+        Object.assign(query, filter);
+        const doc = await AgentChatSchema_js_1.AgentChatModel.findOne(query).exec();
+        return toAgentChat(doc);
+    }
+    async create(entity) {
+        const now = new Date().toISOString();
+        const doc = new AgentChatSchema_js_1.AgentChatModel({
+            ...entity,
+            createdAt: now,
+            updatedAt: now,
+        });
+        const saved = await doc.save();
+        const result = toAgentChat(saved);
+        if (!result) {
+            throw new Error('Failed to create chat');
+        }
+        return result;
+    }
+    async update(id, updates) {
+        const doc = await AgentChatSchema_js_1.AgentChatModel.findByIdAndUpdate(id, { $set: { ...updates, updatedAt: new Date().toISOString() } }, { new: true }).exec();
+        return toAgentChat(doc);
+    }
+    async delete(id) {
+        // Delete all related messages first
+        await AgentMessageSchema_js_1.AgentMessageModel.deleteMany({ chat: id }).exec();
+        // Then delete the chat
+        const result = await AgentChatSchema_js_1.AgentChatModel.findByIdAndDelete(id).exec();
+        return !!result;
+    }
+    async exists(id) {
+        const count = await AgentChatSchema_js_1.AgentChatModel.countDocuments({ _id: id }).exec();
+        return count > 0;
+    }
+    async count(filter) {
+        const query = {};
+        if (filter) {
+            if (filter.id) {
+                query._id = filter.id;
+                delete filter.id;
+            }
+            Object.assign(query, filter);
+        }
+        return AgentChatSchema_js_1.AgentChatModel.countDocuments(query).exec();
+    }
+    async findByUserId(userId) {
+        const docs = await AgentChatSchema_js_1.AgentChatModel.find({ userId }).exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async findByContextKey(contextKey) {
+        const docs = await AgentChatSchema_js_1.AgentChatModel.find({ contextKey }).exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async findByUserIdAndContextKey(userId, contextKey) {
+        const docs = await AgentChatSchema_js_1.AgentChatModel.find({ userId, contextKey }).exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async findGuestChatsByContextKey(contextKey) {
+        const docs = await AgentChatSchema_js_1.AgentChatModel.find({
+            contextKey,
+            userId: { $exists: false }
+        }).exec();
+        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
+    }
+    async updateTitle(id, title) {
+        const doc = await AgentChatSchema_js_1.AgentChatModel.findByIdAndUpdate(id, { $set: { title, updatedAt: new Date().toISOString() } }, { new: true }).exec();
+        return toAgentChat(doc);
+    }
+    async deleteByUserId(userId) {
+        const result = await AgentChatSchema_js_1.AgentChatModel.deleteMany({ userId }).exec();
+        return result.deletedCount || 0;
+    }
+    async deleteByContextKey(contextKey) {
+        const result = await AgentChatSchema_js_1.AgentChatModel.deleteMany({ contextKey }).exec();
+        return result.deletedCount || 0;
+    }
+    async findWithMessages(filter, options) {
+        // Build match stage for filtering
+        const matchStage = {};
+        if (filter) {
+            if (filter.id) {
+                matchStage._id = new index_js_1.ObjectId(filter.id);
+            }
+            if (filter.userId !== undefined) {
+                matchStage.userId = filter.userId;
+            }
+            if (filter.contextKey !== undefined) {
+                matchStage.contextKey = filter.contextKey;
+            }
+            if (filter.type !== undefined) {
+                matchStage.type = filter.type;
+            }
+            if (filter.status !== undefined) {
+                matchStage.status = filter.status;
+            }
+            if (filter.model !== undefined) {
+                matchStage.model = filter.model;
+            }
+        }
+        // Build aggregation pipeline
+        const pipeline = [
+            { $match: matchStage }
+        ];
+        // Add sort stage before lookup (for better performance)
+        if (options === null || options === void 0 ? void 0 : options.sort) {
+            const sortField = options.sort.field === 'id' ? '_id' : options.sort.field;
+            const sortOrder = options.sort.order === ai_agent_types_1.SortOrder.Asc ? 1 : -1;
+            pipeline.push({ $sort: { [sortField]: sortOrder } });
+        }
+        // Add limit before lookup if specified (for better performance)
+        if (options === null || options === void 0 ? void 0 : options.limit) {
+            pipeline.push({ $limit: options.limit });
+        }
+        // Lookup messages
+        pipeline.push({
+            $lookup: {
+                from: 'agentmessages',
+                localField: '_id',
+                foreignField: 'chat',
+                as: 'messages'
+            }
+        });
+        // Transform messages
+        pipeline.push({
+            $addFields: {
+                messages: {
+                    $map: {
+                        input: { $sortArray: { input: '$messages', sortBy: { createdAt: 1 } } },
+                        as: 'msg',
+                        in: {
+                            id: { $toString: '$$msg._id' },
+                            chat: { $toString: '$$msg.chat' },
+                            role: '$$msg.role',
+                            content: '$$msg.content',
+                            createdAt: '$$msg.createdAt',
+                            toolCalls: { $ifNull: ['$$msg.toolCalls', []] },
+                            attachments: { $ifNull: ['$$msg.attachments', []] },
+                            annotations: '$$msg.annotations'
+                        }
+                    }
+                }
+            }
+        });
+        // Project final shape
+        pipeline.push({
+            $project: {
+                id: { $toString: '$_id' },
+                title: 1,
+                type: { $ifNull: ['$type', ai_agent_types_1.ChatType.Chat] },
+                status: 1,
+                contextKey: 1,
+                model: 1,
+                createdAt: 1,
+                updatedAt: 1,
+                metadata: 1,
+                userId: 1,
+                messages: 1
+            }
+        });
+        const results = await AgentChatSchema_js_1.AgentChatModel.aggregate(pipeline).exec();
+        return results.map((result) => ({
+            id: result.id,
+            title: result.title,
+            type: result.type || ai_agent_types_1.ChatType.Chat,
+            status: result.status,
+            contextKey: result.contextKey,
+            model: result.model,
+            createdAt: result.createdAt,
+            updatedAt: result.updatedAt,
+            metadata: result.metadata,
+            userId: result.userId,
+            messages: result.messages || []
+        }));
+    }
+}
+exports.MongoAgentChatRepository = MongoAgentChatRepository;
+//# sourceMappingURL=MongoAgentChatRepository.js.map

package/dist/cjs/repositories/MongoAgentChatRepository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentChatRepository.js","sourceRoot":"","sources":["../../../src/repositories/MongoAgentChatRepository.ts"],"names":[],"mappings":";;;AACA,oEAAsE;AAEtE,sEAAuF;AACvF,4EAAqE;AACrE,gDAA6C;AAE7C;;GAEG;AACH,SAAS,WAAW,CAAC,GAA6B;IAChD,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IAEtB,MAAM,IAAI,GAAI,GAAW,CAAC,QAAQ,CAAC,CAAC,CAAE,GAAW,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;IACnE,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE;QACtB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,yBAAQ,CAAC,IAAI;QAChC,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAa,wBAAwB;IACnC,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,GAAG,GAAG,MAAM,mCAAc,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,IAAI,CACR,MAA2B,EAC3B,OAAwE;QAExE,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC,EAAE,CAAC;YACnB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,YAAY,GAAG,mCAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE9C,gBAAgB;QAChB,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,0BAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChE,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,cAAc;QACd,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,KAAK,EAAE,CAAC;YACnB,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,CAAC;QACvC,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA0B;QACtC,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;YACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;YACtB,OAAO,MAAM,CAAC,EAAE,CAAC;QACnB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAE7B,MAAM,GAAG,GAAG,MAAM,mCAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACvD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAyD;QACpE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,GAAG,GAAG,IAAI,mCAAc,CAAC;YAC7B,GAAG,MAAM;YACT,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,OAAqD;QAC5E,MAAM,GAAG,GAAG,MAAM,mCAAc,CAAC,iBAAiB,CAChD,EAAE,EACF,EAAE,IAAI,EAAE,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,EAAE,EAC7D,EAAE,GAAG,EAAE,IAAI,EAAE,CACd,CAAC,IAAI,EAAE,CAAC;QAET,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,oCAAoC;QACpC,MAAM,yCAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAExD,uBAAuB;QACvB,MAAM,MAAM,GAAG,MAAM,mCAAc,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAEjE,OAAO,CAAC,CAAC,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,KAAK,GAAG,MAAM,mCAAc,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,KAAK,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAA2B;QACrC,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC,EAAE,CAAC;YACnB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,mCAAc,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,IAAI,GAAG,MAAM,mCAAc,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1D,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,UAAkB;QACvC,MAAM,IAAI,GAAG,MAAM,mCAAc,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC9D,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,MAAc,EAAE,UAAkB;QAChE,MAAM,IAAI,GAAG,MAAM,mCAAc,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,0BAA0B,CAAC,UAAkB;QACjD,MAAM,IAAI,GAAG,MAAM,mCAAc,CAAC,IAAI,CAAC;YACrC,UAAU;YACV,MAAM,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE;SAC3B,CAAC,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU,EAAE,KAAa;QACzC,MAAM,GAAG,GAAG,MAAM,mCAAc,CAAC,iBAAiB,CAChD,EAAE,EACF,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,EAAE,EACxD,EAAE,GAAG,EAAE,IAAI,EAAE,CACd,CAAC,IAAI,EAAE,CAAC;QAET,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAAc;QACjC,MAAM,MAAM,GAAG,MAAM,mCAAc,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAClE,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,UAAkB;QACzC,MAAM,MAAM,GAAG,MAAM,mCAAc,CAAC,UAAU,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,MAA2B,EAC3B,OAAwE;QAExE,kCAAkC;QAClC,MAAM,UAAU,GAAQ,EAAE,CAAC;QAE3B,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,UAAU,CAAC,GAAG,GAAG,IAAI,mBAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC3C,CAAC;YACD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAChC,UAAU,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YACpC,CAAC;YACD,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;gBACpC,UAAU,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;YAC5C,CAAC;YACD,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC9B,UAAU,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YAChC,CAAC;YACD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAChC,UAAU,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YACpC,CAAC;YACD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;gBAC/B,UAAU,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;YAClC,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,MAAM,QAAQ,GAAU;YACtB,EAAE,MAAM,EAAE,UAAU,EAAE;SACvB,CAAC;QAEF,wDAAwD;QACxD,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;YAC3E,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,0BAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChE,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,gEAAgE;QAChE,IAAI,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,KAAK,EAAE,CAAC;YACnB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,kBAAkB;QAClB,QAAQ,CAAC,IAAI,CAAC;YACZ,OAAO,EAAE;gBACP,IAAI,EAAE,eAAe;gBACrB,UAAU,EAAE,KAAK;gBACjB,YAAY,EAAE,MAAM;gBACpB,EAAE,EAAE,UAAU;aACf;SACF,CAAC,CAAC;QAEH,qBAAqB;QACrB,QAAQ,CAAC,IAAI,CAAC;YACZ,UAAU,EAAE;gBACV,QAAQ,EAAE;oBACR,IAAI,EAAE;wBACJ,KAAK,EAAE,EAAE,UAAU,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE;wBACvE,EAAE,EAAE,KAAK;wBACT,EAAE,EAAE;4BACF,EAAE,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE;4BAC9B,IAAI,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE;4BACjC,IAAI,EAAE,YAAY;4BAClB,OAAO,EAAE,eAAe;4BACxB,SAAS,EAAE,iBAAiB;4BAC5B,SAAS,EAAE,EAAE,OAAO,EAAE,CAAC,iBAAiB,EAAE,EAAE,CAAC,EAAE;4BAC/C,WAAW,EAAE,EAAE,OAAO,EAAE,CAAC,mBAAmB,EAAE,EAAE,CAAC,EAAE;4BACnD,WAAW,EAAE,mBAAmB;yBACjC;qBACF;iBACF;aACF;SACF,CAAC,CAAC;QAEH,sBAAsB;QACtB,QAAQ,CAAC,IAAI,CAAC;YACZ,QAAQ,EAAE;gBACR,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;gBACzB,KAAK,EAAE,CAAC;gBACR,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,yBAAQ,CAAC,IAAI,CAAC,EAAE;gBAC3C,MAAM,EAAE,CAAC;gBACT,UAAU,EAAE,CAAC;gBACb,KAAK,EAAE,CAAC;gBACR,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,CAAC;gBACZ,QAAQ,EAAE,CAAC;gBACX,MAAM,EAAE,CAAC;gBACT,QAAQ,EAAE,CAAC;aACZ;SACF,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,mCAAc,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QAEhE,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC9B,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,yBAAQ,CAAC,IAAI;YAClC,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE;SAChC,CAAC,CAAC,CAAC;IACN,CAAC;CACF;AAlQD,4DAkQC"}

package/dist/cjs/repositories/MongoAgentMessageRepository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentMessageRepository.d.ts","sourceRoot":"","sources":["../../../src/repositories/MongoAgentMessageRepository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AACjF,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AA0B3E;;GAEG;AACH,qBAAa,2BAA4B,YAAW,sBAAsB;IAClE,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAKlD,IAAI,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAe7D,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAapE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,GAAG,WAAW,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAgB1F,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,YAAY,GAAG,IAAI,CAAC;IAU1G,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpC,KAAK,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IActD,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAKrD,UAAU,CAAC,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAKtD,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAShF,iBAAiB,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAa3D,mBAAmB,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAa7D,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAK/C,YAAY,CAAC,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;IAKhD,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;CAIhF"}

package/dist/cjs/repositories/MongoAgentMessageRepository.js

@@ -0,0 +1,144 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MongoAgentMessageRepository = void 0;
+const AgentMessageSchema_js_1 = require("../schemas/AgentMessageSchema.js");
+/**
+ * Helper function to convert Mongoose document to AgentMessage type
+ */
+function toAgentMessage(doc) {
+    if (!doc)
+        return null;
+    // Use toObject with proper options to ensure all fields are included
+    const message = doc.toObject ? doc.toObject({ getters: true, virtuals: false }) : doc;
+    return {
+        id: doc._id.toString(),
+        chat: doc.chat.toString(),
+        role: message.role,
+        content: message.content,
+        createdAt: message.createdAt,
+        reasoning: message.reasoning,
+        toolCalls: message.toolCalls || [],
+        attachments: message.attachments || [],
+        annotations: message.annotations,
+        tokens: message.tokens,
+        agentName: message.agentName,
+    };
+}
+/**
+ * MongoDB implementation of AgentMessageRepository
+ */
+class MongoAgentMessageRepository {
+    async findById(id) {
+        const doc = await AgentMessageSchema_js_1.AgentMessageModel.findById(id).exec();
+        return toAgentMessage(doc);
+    }
+    async find(filter) {
+        const query = {};
+        if (filter) {
+            if (filter.id) {
+                query._id = filter.id;
+                delete filter.id;
+            }
+            Object.assign(query, filter);
+        }
+        const docs = await AgentMessageSchema_js_1.AgentMessageModel.find(query).exec();
+        return docs.map(doc => toAgentMessage(doc)).filter(Boolean);
+    }
+    async findOne(filter) {
+        const query = {};
+        if (filter.id) {
+            query._id = filter.id;
+            delete filter.id;
+        }
+        Object.assign(query, filter);
+        const doc = await AgentMessageSchema_js_1.AgentMessageModel.findOne(query).exec();
+        return toAgentMessage(doc);
+    }
+    async create(entity, id) {
+        const now = new Date().toISOString();
+        const doc = new AgentMessageSchema_js_1.AgentMessageModel({
+            ...entity,
+            ...(id ? { _id: id } : {}),
+            createdAt: now
+        });
+        const saved = await doc.save();
+        const result = toAgentMessage(saved);
+        if (!result) {
+            throw new Error('Failed to create message');
+        }
+        return result;
+    }
+    async update(id, updates) {
+        const doc = await AgentMessageSchema_js_1.AgentMessageModel.findByIdAndUpdate(id, { $set: updates }, { new: true }).exec();
+        return toAgentMessage(doc);
+    }
+    async delete(id) {
+        const result = await AgentMessageSchema_js_1.AgentMessageModel.findByIdAndDelete(id).exec();
+        return !!result;
+    }
+    async exists(id) {
+        const count = await AgentMessageSchema_js_1.AgentMessageModel.countDocuments({ _id: id }).exec();
+        return count > 0;
+    }
+    async count(filter) {
+        const query = {};
+        if (filter) {
+            if (filter.id) {
+                query._id = filter.id;
+                delete filter.id;
+            }
+            Object.assign(query, filter);
+        }
+        return AgentMessageSchema_js_1.AgentMessageModel.countDocuments(query).exec();
+    }
+    async findByChatId(chatId) {
+        const docs = await AgentMessageSchema_js_1.AgentMessageModel.find({ chat: chatId }).exec();
+        return docs.map(doc => toAgentMessage(doc)).filter(Boolean);
+    }
+    async findByRole(role) {
+        const docs = await AgentMessageSchema_js_1.AgentMessageModel.find({ role }).exec();
+        return docs.map(doc => toAgentMessage(doc)).filter(Boolean);
+    }
+    async findByChatIdAndRole(chatId, role) {
+        const query = { chat: chatId };
+        if (role) {
+            query.role = role;
+        }
+        const docs = await AgentMessageSchema_js_1.AgentMessageModel.find(query).exec();
+        return docs.map(doc => toAgentMessage(doc)).filter(Boolean);
+    }
+    async findWithToolCalls(chatId) {
+        const query = {
+            toolCalls: { $exists: true, $ne: [] }
+        };
+        if (chatId) {
+            query.chat = chatId;
+        }
+        const docs = await AgentMessageSchema_js_1.AgentMessageModel.find(query).exec();
+        return docs.map(doc => toAgentMessage(doc)).filter(Boolean);
+    }
+    async findWithAttachments(chatId) {
+        const query = {
+            attachments: { $exists: true, $ne: [] }
+        };
+        if (chatId) {
+            query.chat = chatId;
+        }
+        const docs = await AgentMessageSchema_js_1.AgentMessageModel.find(query).exec();
+        return docs.map(doc => toAgentMessage(doc)).filter(Boolean);
+    }
+    async deleteByChatId(chatId) {
+        const result = await AgentMessageSchema_js_1.AgentMessageModel.deleteMany({ chat: chatId }).exec();
+        return result.deletedCount || 0;
+    }
+    async deleteByRole(role) {
+        const result = await AgentMessageSchema_js_1.AgentMessageModel.deleteMany({ role }).exec();
+        return result.deletedCount || 0;
+    }
+    async deleteByChatIdAndRole(chatId, role) {
+        const result = await AgentMessageSchema_js_1.AgentMessageModel.deleteMany({ chat: chatId, role }).exec();
+        return result.deletedCount || 0;
+    }
+}
+exports.MongoAgentMessageRepository = MongoAgentMessageRepository;
+//# sourceMappingURL=MongoAgentMessageRepository.js.map

package/dist/cjs/repositories/MongoAgentMessageRepository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentMessageRepository.js","sourceRoot":"","sources":["../../../src/repositories/MongoAgentMessageRepository.ts"],"names":[],"mappings":";;;AAEA,4EAAgG;AAEhG;;GAEG;AACH,SAAS,cAAc,CAAC,GAAgC;IACtD,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IAEtB,qEAAqE;IACrE,MAAM,OAAO,GAAI,GAAW,CAAC,QAAQ,CAAC,CAAC,CAAE,GAAW,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IACxG,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE;QACtB,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE;QACzB,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;QAC5B,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,EAAE;QAClC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,EAAE;QACtC,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,SAAS,EAAE,OAAO,CAAC,SAAS;KAC7B,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAa,2BAA2B;IACtC,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,GAAG,GAAG,MAAM,yCAAiB,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACxD,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAA8B;QACvC,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC,EAAE,CAAC;YACnB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,yCAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACxD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,cAAc,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA6B;QACzC,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;YACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;YACtB,OAAO,MAAM,CAAC,EAAE,CAAC;QACnB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAE7B,MAAM,GAAG,GAAG,MAAM,yCAAiB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QAC1D,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAA8C,EAAE,EAAW;QACtE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,GAAG,GAAG,IAAI,yCAAiB,CAAC;YAChC,GAAG,MAAM;YACT,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1B,SAAS,EAAE,GAAG;SACf,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,OAAwD;QAC/E,MAAM,GAAG,GAAG,MAAM,yCAAiB,CAAC,iBAAiB,CACnD,EAAE,EACF,EAAE,IAAI,EAAE,OAAO,EAAE,EACjB,EAAE,GAAG,EAAE,IAAI,EAAE,CACd,CAAC,IAAI,EAAE,CAAC;QAET,OAAO,cAAc,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,MAAM,GAAG,MAAM,yCAAiB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACpE,OAAO,CAAC,CAAC,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,KAAK,GAAG,MAAM,yCAAiB,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACzE,OAAO,KAAK,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAA8B;QACxC,MAAM,KAAK,GAAQ,EAAE,CAAC;QAEtB,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;gBACtB,OAAO,MAAM,CAAC,EAAE,CAAC;YACnB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,yCAAiB,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACxD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,IAAI,GAAG,MAAM,yCAAiB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,cAAc,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,IAAiB;QAChC,MAAM,IAAI,GAAG,MAAM,yCAAiB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC3D,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,cAAc,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,MAAc,EAAE,IAAkB;QAC1D,MAAM,KAAK,GAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACpC,IAAI,IAAI,EAAE,CAAC;YACT,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;QACpB,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,yCAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACxD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,cAAc,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,MAAe;QACrC,MAAM,KAAK,GAAQ;YACjB,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,EAAE;SACtC,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,IAAI,GAAG,MAAM,CAAC;QACtB,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,yCAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACxD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,cAAc,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,MAAe;QACvC,MAAM,KAAK,GAAQ;YACjB,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,EAAE;SACxC,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,IAAI,GAAG,MAAM,CAAC;QACtB,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,yCAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACxD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,cAAc,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAAc;QACjC,MAAM,MAAM,GAAG,MAAM,yCAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC3E,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,IAAiB;QAClC,MAAM,MAAM,GAAG,MAAM,yCAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACnE,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAAc,EAAE,IAAiB;QAC3D,MAAM,MAAM,GAAG,MAAM,yCAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACjF,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;CACF;AA/ID,kEA+IC"}

package/dist/cjs/repositories/index.d.ts

@@ -0,0 +1,3 @@
+export * from './MongoAgentChatRepository.js';
+export * from './MongoAgentMessageRepository.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/cjs/repositories/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/repositories/index.ts"],"names":[],"mappings":"AAAA,cAAc,+BAA+B,CAAC;AAC9C,cAAc,kCAAkC,CAAC"}

package/dist/cjs/repositories/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./MongoAgentChatRepository.js"), exports);
+__exportStar(require("./MongoAgentMessageRepository.js"), exports);
+//# sourceMappingURL=index.js.map

package/dist/cjs/repositories/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/repositories/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,gEAA8C;AAC9C,mEAAiD"}

package/dist/{schemas → cjs/schemas}/AgentChatSchema.d.ts

@@ -1,4 +1,4 @@
-import { type ObjectId, type Document } from '../mongo';
+import { type ObjectId, type Document } from '../mongo/index.js';
 import type { AgentChat } from '@multiplayer-app/ai-agent-types';
 /**
  * Mongoose document interface for AgentChat

package/dist/cjs/schemas/AgentChatSchema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"AgentChatSchema.d.ts","sourceRoot":"","sources":["../../../src/schemas/AgentChatSchema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,KAAK,QAAQ,EAAE,KAAK,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AACnF,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAGjE;;GAEG;AACH,MAAM,WAAW,iBAAkB,SAAQ,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC;IACvF,GAAG,EAAE,QAAQ,CAAC;CACf;AAED;;GAEG;AACH,eAAO,MAAM,eAAe;;;;;;;;EAa1B,CAAC;AAKH;;GAEG;AACH,eAAO,MAAM,cAAc;;;;OAAkE,CAAC"}