@multiplayer-app/ai-agent-mongo 0.1.0-beta.7 → 0.1.0-beta.71

package/dist/cjs/repositories/MongoAgentChatRepository.cjs

@@ -4,7 +4,6 @@ exports.MongoAgentChatRepository = void 0;
 const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
 const AgentChatSchema_1 = require("../schemas/AgentChatSchema.cjs");
 const AgentMessageSchema_1 = require("../schemas/AgentMessageSchema.cjs");
-const mongo_1 = require("../mongo/index.cjs");
 /**
  * Helper function to convert Mongoose document to AgentChat type
  */
@@ -12,6 +11,7 @@ function toAgentChat(doc) {
     if (!doc)
         return null;
     const chat = doc.toObject ? doc.toObject() : doc;
+    const tenants = chat.tenants instanceof Map ? Object.fromEntries(chat.tenants.entries()) : (chat.tenants ?? {});
     return {
         id: doc._id.toString(),
         title: chat.title,
@@ -22,7 +22,7 @@ function toAgentChat(doc) {
         createdAt: chat.createdAt,
         updatedAt: chat.updatedAt,
         metadata: chat.metadata,
-        userId: chat.userId
+        tenants
     };
 }
 /**
@@ -33,35 +33,13 @@ class MongoAgentChatRepository {
         const doc = await AgentChatSchema_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_1.AgentChatModel.find(query);
-        // Apply sorting
-        if (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?.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;
+        const { id, ...rest } = filter;
+        if (id) {
+            query._id = id;
         }
-        Object.assign(query, filter);
+        Object.assign(query, rest);
         const doc = await AgentChatSchema_1.AgentChatModel.findOne(query).exec();
         return toAgentChat(doc);
     }
@@ -97,142 +75,59 @@ class MongoAgentChatRepository {
     async count(filter) {
         const query = {};
         if (filter) {
-            if (filter.id) {
-                query._id = filter.id;
-                delete filter.id;
+            const { id, ...rest } = filter;
+            if (id) {
+                query._id = id;
             }
-            Object.assign(query, filter);
+            Object.assign(query, rest);
         }
         return AgentChatSchema_1.AgentChatModel.countDocuments(query).exec();
     }
-    async findByUserId(userId) {
-        const docs = await AgentChatSchema_1.AgentChatModel.find({ userId }).exec();
-        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
-    }
-    async findByContextKey(contextKey) {
-        const docs = await AgentChatSchema_1.AgentChatModel.find({ contextKey }).exec();
-        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
-    }
-    async findByUserIdAndContextKey(userId, contextKey) {
-        const docs = await AgentChatSchema_1.AgentChatModel.find({ userId, contextKey }).exec();
-        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
-    }
-    async findGuestChatsByContextKey(contextKey) {
-        const docs = await AgentChatSchema_1.AgentChatModel.find({
-            contextKey,
-            userId: { $exists: false }
-        }).exec();
-        return docs.map(doc => toAgentChat(doc)).filter(Boolean);
-    }
     async updateTitle(id, title) {
         const doc = await AgentChatSchema_1.AgentChatModel.findByIdAndUpdate(id, { $set: { title, updatedAt: new Date().toISOString() } }, { new: true }).exec();
         return toAgentChat(doc);
     }
     async deleteByUserId(userId) {
-        const result = await AgentChatSchema_1.AgentChatModel.deleteMany({ userId }).exec();
+        const result = await AgentChatSchema_1.AgentChatModel.deleteMany({ 'tenants.userId': userId }).exec();
         return result.deletedCount || 0;
     }
     async deleteByContextKey(contextKey) {
         const result = await AgentChatSchema_1.AgentChatModel.deleteMany({ contextKey }).exec();
         return result.deletedCount || 0;
     }
-    async findWithMessages(filter, options) {
-        // Build match stage for filtering
-        const matchStage = {};
+    async find(filter, options) {
+        const query = {};
         if (filter) {
-            if (filter.id) {
-                matchStage._id = new mongo_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;
-            }
+            if (filter.id)
+                query._id = filter.id;
+            if (filter.tenants !== undefined)
+                query.tenants = filter.tenants;
+            const legacyUserId = filter.userId;
+            if (legacyUserId !== undefined)
+                query['tenants.userId'] = legacyUserId;
+            if (filter.contextKey !== undefined)
+                query.contextKey = filter.contextKey;
+            if (filter.type !== undefined)
+                query.type = filter.type;
+            if (filter.status !== undefined)
+                query.status = filter.status;
+            if (filter.model !== undefined)
+                query.model = filter.model;
         }
-        // Build aggregation pipeline
-        const pipeline = [
-            { $match: matchStage }
-        ];
-        // Add sort stage before lookup (for better performance)
+        let qb = AgentChatSchema_1.AgentChatModel.find(query);
         if (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 } });
+            qb = qb.sort({ [sortField]: sortOrder });
+        }
+        if (options?.skip) {
+            qb = qb.skip(options.skip);
         }
-        // Add limit before lookup if specified (for better performance)
         if (options?.limit) {
-            pipeline.push({ $limit: options.limit });
+            qb = qb.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_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 || []
-        }));
+        const docs = await qb.exec();
+        return docs.map((doc) => toAgentChat(doc)).filter(Boolean);
     }
 }
 exports.MongoAgentChatRepository = MongoAgentChatRepository;

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

@@ -1 +1 @@
-{"version":3,"file":"MongoAgentChatRepository.js","sourceRoot":"","sources":["../../../src/repositories/MongoAgentChatRepository.ts"],"names":[],"mappings":";;;AACA,oEAAsE;AAEtE,gEAAoF;AACpF,sEAAkE;AAClE,oCAAoC;AAEpC;;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,gCAAc,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,gCAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE9C,gBAAgB;QAChB,IAAI,OAAO,EAAE,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,EAAE,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,gCAAc,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,gCAAc,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,gCAAc,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,sCAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAExD,uBAAuB;QACvB,MAAM,MAAM,GAAG,MAAM,gCAAc,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,gCAAc,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,gCAAc,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,IAAI,GAAG,MAAM,gCAAc,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,gCAAc,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,gCAAc,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,gCAAc,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,gCAAc,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,gCAAc,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,gCAAc,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,gBAAQ,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,EAAE,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,EAAE,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,gCAAc,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"}
+{"version":3,"file":"MongoAgentChatRepository.js","sourceRoot":"","sources":["../../../src/repositories/MongoAgentChatRepository.ts"],"names":[],"mappings":";;;AACA,oEAAsE;AAEtE,gEAAoF;AACpF,sEAAkE;AAElE;;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,MAAM,OAAO,GACX,IAAI,CAAC,OAAO,YAAY,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;IAClG,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,OAAO;KACR,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAa,wBAAwB;IACnC,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,GAAG,GAAG,MAAM,gCAAc,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAA0B;QACtC,MAAM,KAAK,GAAQ,EAAE,CAAC;QACtB,MAAM,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;QAE/B,IAAI,EAAE,EAAE,CAAC;YACP,KAAK,CAAC,GAAG,GAAG,EAAE,CAAC;QACjB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAE3B,MAAM,GAAG,GAAG,MAAM,gCAAc,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,gCAAc,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,gCAAc,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,sCAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAExD,uBAAuB;QACvB,MAAM,MAAM,GAAG,MAAM,gCAAc,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,gCAAc,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,MAAM,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,GAAG,MAAM,CAAC;YAC/B,IAAI,EAAE,EAAE,CAAC;gBACP,KAAK,CAAC,GAAG,GAAG,EAAE,CAAC;YACjB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,gCAAc,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU,EAAE,KAAa;QACzC,MAAM,GAAG,GAAG,MAAM,gCAAc,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,gCAAc,CAAC,UAAU,CAAC,EAAE,gBAAgB,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACpF,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,UAAkB;QACzC,MAAM,MAAM,GAAG,MAAM,gCAAc,CAAC,UAAU,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,IAAI,CACR,MAA2B,EAC3B,OAAuF;QAEvF,MAAM,KAAK,GAA4B,EAAE,CAAC;QAE1C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE;gBAAE,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;YACrC,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS;gBAAE,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;YACjE,MAAM,YAAY,GAAI,MAAc,CAAC,MAAM,CAAC;YAC5C,IAAI,YAAY,KAAK,SAAS;gBAAE,KAAK,CAAC,gBAAgB,CAAC,GAAG,YAAY,CAAC;YACvE,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS;gBAAE,KAAK,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;YAC1E,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS;gBAAE,KAAK,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;YACxD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;gBAAE,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YAC9D,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS;gBAAE,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC7D,CAAC;QAED,IAAI,EAAE,GAAG,gCAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEpC,IAAI,OAAO,EAAE,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,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;YAClB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;YACnB,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC9D,CAAC;CACF;AAlID,4DAkIC"}

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

@@ -1,4 +1,4 @@
-import type { AgentChat, AgentChatResponse } from '@multiplayer-app/ai-agent-types';
+import type { AgentChat } from '@multiplayer-app/ai-agent-types';
 import { SortOrder } from '@multiplayer-app/ai-agent-types';
 import type { AgentChatRepository } from '@multiplayer-app/ai-agent-db';
 /**
@@ -6,32 +6,22 @@ import type { AgentChatRepository } from '@multiplayer-app/ai-agent-db';
  */
 export declare class MongoAgentChatRepository implements AgentChatRepository {
     findById(id: string): Promise<AgentChat | null>;
-    find(filter?: Partial<AgentChat>, options?: {
-        sort?: {
-            field: string;
-            order: SortOrder;
-        };
-        limit?: number;
-    }): Promise<AgentChat[]>;
     findOne(filter: Partial<AgentChat>): Promise<AgentChat | null>;
     create(entity: Omit<AgentChat, 'id' | 'createdAt' | 'updatedAt'>): Promise<AgentChat>;
     update(id: string, updates: Partial<Omit<AgentChat, 'id' | 'createdAt'>>): Promise<AgentChat | null>;
     delete(id: string): Promise<boolean>;
     exists(id: string): Promise<boolean>;
     count(filter?: Partial<AgentChat>): Promise<number>;
-    findByUserId(userId: string): Promise<AgentChat[]>;
-    findByContextKey(contextKey: string): Promise<AgentChat[]>;
-    findByUserIdAndContextKey(userId: string, contextKey: string): Promise<AgentChat[]>;
-    findGuestChatsByContextKey(contextKey: string): Promise<AgentChat[]>;
     updateTitle(id: string, title: string): Promise<AgentChat | null>;
     deleteByUserId(userId: string): Promise<number>;
     deleteByContextKey(contextKey: string): Promise<number>;
-    findWithMessages(filter?: Partial<AgentChat>, options?: {
+    find(filter?: Partial<AgentChat>, options?: {
         sort?: {
             field: string;
             order: SortOrder;
         };
+        skip?: number;
         limit?: number;
-    }): Promise<AgentChatResponse[]>;
+    }): Promise<AgentChat[]>;
 }
 //# sourceMappingURL=MongoAgentChatRepository.d.ts.map

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

@@ -1 +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"}
+{"version":3,"file":"MongoAgentChatRepository.d.ts","sourceRoot":"","sources":["../../../src/repositories/MongoAgentChatRepository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AACjE,OAAO,EAAY,SAAS,EAAE,MAAM,iCAAiC,CAAC;AACtE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AA2BxE;;GAEG;AACH,qBAAa,wBAAyB,YAAW,mBAAmB;IAC5D,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAK/C,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,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,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,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACtF,OAAO,CAAC,SAAS,EAAE,CAAC;CAiCxB"}

package/dist/cjs/repositories/MongoAgentConfigRepository.cjs

@@ -0,0 +1,126 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MongoAgentConfigRepository = void 0;
+const ai_agent_types_1 = require("@multiplayer-app/ai-agent-types");
+const AgentConfigPreferenceSchema_1 = require("../schemas/AgentConfigPreferenceSchema.cjs");
+function toAgentConfigPreference(doc) {
+    if (!doc)
+        return null;
+    const value = doc.toObject ? doc.toObject() : doc;
+    const toolsValue = value.tools;
+    const tools = toolsValue instanceof Map
+        ? Object.fromEntries(toolsValue.entries())
+        : (toolsValue ?? {});
+    return {
+        id: doc._id.toString(),
+        userId: value.userId,
+        agentName: value.agentName,
+        tools,
+        createdAt: value.createdAt?.toISOString?.() ?? value.createdAt,
+        updatedAt: value.updatedAt?.toISOString?.() ?? value.updatedAt,
+    };
+}
+class MongoAgentConfigRepository {
+    async findById(id) {
+        const doc = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.findById(id).exec();
+        return toAgentConfigPreference(doc);
+    }
+    async find(filter, options) {
+        const query = {};
+        if (filter) {
+            if (filter.id)
+                query._id = filter.id;
+            if (filter.userId !== undefined)
+                query.userId = filter.userId;
+            if (filter.agentName !== undefined)
+                query.agentName = filter.agentName;
+        }
+        let qb = AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.find(query);
+        if (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;
+            qb = qb.sort({ [sortField]: sortOrder });
+        }
+        if (options?.skip)
+            qb = qb.skip(options.skip);
+        if (options?.limit)
+            qb = qb.limit(options.limit);
+        const docs = await qb.exec();
+        return docs.map((doc) => toAgentConfigPreference(doc)).filter(Boolean);
+    }
+    async findOne(filter) {
+        const query = {};
+        if (filter.id)
+            query._id = filter.id;
+        if (filter.userId !== undefined)
+            query.userId = filter.userId;
+        if (filter.agentName !== undefined)
+            query.agentName = filter.agentName;
+        const doc = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.findOne(query).exec();
+        return toAgentConfigPreference(doc);
+    }
+    async create(entity) {
+        const doc = new AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel({
+            ...entity,
+            tools: entity.tools ?? {},
+        });
+        const saved = await doc.save();
+        const result = toAgentConfigPreference(saved);
+        if (!result)
+            throw new Error('Failed to create agent config preference');
+        return result;
+    }
+    async update(id, updates) {
+        const doc = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.findByIdAndUpdate(id, { $set: updates }, { new: true, runValidators: true }).exec();
+        return toAgentConfigPreference(doc);
+    }
+    async delete(id) {
+        const result = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.findByIdAndDelete(id).exec();
+        return !!result;
+    }
+    async exists(id) {
+        const count = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.countDocuments({ _id: id }).exec();
+        return count > 0;
+    }
+    async count(filter) {
+        const query = {};
+        if (filter) {
+            if (filter.id)
+                query._id = filter.id;
+            if (filter.userId !== undefined)
+                query.userId = filter.userId;
+            if (filter.agentName !== undefined)
+                query.agentName = filter.agentName;
+        }
+        return AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.countDocuments(query).exec();
+    }
+    async findByUserId(userId) {
+        const docs = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.find({ userId }).exec();
+        return docs.map((doc) => toAgentConfigPreference(doc)).filter(Boolean);
+    }
+    async findByUserIdAndAgentName(userId, agentName) {
+        const doc = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.findOne({ userId, agentName }).exec();
+        return toAgentConfigPreference(doc);
+    }
+    async upsertByUserIdAndAgentName(userId, agentName, updates) {
+        const doc = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.findOneAndUpdate({ userId, agentName }, {
+            $set: {
+                tools: updates.tools ?? {},
+            },
+        }, {
+            new: true,
+            upsert: true,
+            runValidators: true,
+        }).exec();
+        const result = toAgentConfigPreference(doc);
+        if (!result)
+            throw new Error('Failed to upsert agent config preference');
+        return result;
+    }
+    async deleteByUserId(userId) {
+        const result = await AgentConfigPreferenceSchema_1.AgentConfigPreferenceModel.deleteMany({ userId }).exec();
+        return result.deletedCount || 0;
+    }
+}
+exports.MongoAgentConfigRepository = MongoAgentConfigRepository;
+//# sourceMappingURL=MongoAgentConfigRepository.js.map

package/dist/cjs/repositories/MongoAgentConfigRepository.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentConfigRepository.js","sourceRoot":"","sources":["../../../src/repositories/MongoAgentConfigRepository.ts"],"names":[],"mappings":";;;AAEA,oEAA4D;AAC5D,wFAGgD;AAEhD,SAAS,uBAAuB,CAAC,GAAyC;IACxE,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IAEtB,MAAM,KAAK,GAAI,GAAW,CAAC,QAAQ,CAAC,CAAC,CAAE,GAAW,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;IACpE,MAAM,UAAU,GAAG,KAAK,CAAC,KAAmE,CAAC;IAC7F,MAAM,KAAK,GACT,UAAU,YAAY,GAAG;QACvB,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;IACzB,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE;QACtB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,KAAK;QACL,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,WAAW,EAAE,EAAE,IAAI,KAAK,CAAC,SAAS;QAC9D,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,WAAW,EAAE,EAAE,IAAI,KAAK,CAAC,SAAS;KAC/D,CAAC;AACJ,CAAC;AAED,MAAa,0BAA0B;IACrC,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,GAAG,GAAG,MAAM,wDAA0B,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACjE,OAAO,uBAAuB,CAAC,GAAG,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,IAAI,CACR,MAAuC,EACvC,OAIC;QAED,MAAM,KAAK,GAA4B,EAAE,CAAC;QAC1C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE;gBAAE,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;YACrC,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;gBAAE,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YAC9D,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS;gBAAE,KAAK,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACzE,CAAC;QAED,IAAI,EAAE,GAAG,wDAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,IAAI,OAAO,EAAE,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,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;QAC3C,CAAC;QACD,IAAI,OAAO,EAAE,IAAI;YAAE,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,OAAO,EAAE,KAAK;YAAE,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAEjD,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,uBAAuB,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAsC;QAClD,MAAM,KAAK,GAA4B,EAAE,CAAC;QAC1C,IAAI,MAAM,CAAC,EAAE;YAAE,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;QACrC,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;YAAE,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC9D,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS;YAAE,KAAK,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAEvE,MAAM,GAAG,GAAG,MAAM,wDAA0B,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACnE,OAAO,uBAAuB,CAAC,GAAG,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAqE;QAChF,MAAM,GAAG,GAAG,IAAI,wDAA0B,CAAC;YACzC,GAAG,MAAM;YACT,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,EAAE;SAC1B,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QACzE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,MAAM,CACV,EAAU,EACV,OAAiE;QAEjE,MAAM,GAAG,GAAG,MAAM,wDAA0B,CAAC,iBAAiB,CAC5D,EAAE,EACF,EAAE,IAAI,EAAE,OAAO,EAAE,EACjB,EAAE,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CACnC,CAAC,IAAI,EAAE,CAAC;QACT,OAAO,uBAAuB,CAAC,GAAG,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,MAAM,GAAG,MAAM,wDAA0B,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7E,OAAO,CAAC,CAAC,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,KAAK,GAAG,MAAM,wDAA0B,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAClF,OAAO,KAAK,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAuC;QACjD,MAAM,KAAK,GAA4B,EAAE,CAAC;QAC1C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,MAAM,CAAC,EAAE;gBAAE,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,EAAE,CAAC;YACrC,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;gBAAE,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YAC9D,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS;gBAAE,KAAK,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACzE,CAAC;QACD,OAAO,wDAA0B,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,IAAI,GAAG,MAAM,wDAA0B,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACtE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,uBAAuB,CAAC,GAAG,CAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,wBAAwB,CAAC,MAAc,EAAE,SAAiB;QAC9D,MAAM,GAAG,GAAG,MAAM,wDAA0B,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACnF,OAAO,uBAAuB,CAAC,GAAG,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,0BAA0B,CAC9B,MAAc,EACd,SAAiB,EACjB,OAA6C;QAE7C,MAAM,GAAG,GAAG,MAAM,wDAA0B,CAAC,gBAAgB,CAC3D,EAAE,MAAM,EAAE,SAAS,EAAE,EACrB;YACE,IAAI,EAAE;gBACJ,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,EAAE;aAC3B;SACF,EACD;YACE,GAAG,EAAE,IAAI;YACT,MAAM,EAAE,IAAI;YACZ,aAAa,EAAE,IAAI;SACpB,CACF,CAAC,IAAI,EAAE,CAAC;QAET,MAAM,MAAM,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QACzE,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAAc;QACjC,MAAM,MAAM,GAAG,MAAM,wDAA0B,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC9E,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;CACF;AA7HD,gEA6HC"}

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

@@ -0,0 +1,25 @@
+import type { AgentConfigPreference } from '@multiplayer-app/ai-agent-types';
+import type { AgentConfigRepository } from '@multiplayer-app/ai-agent-db';
+import { SortOrder } from '@multiplayer-app/ai-agent-types';
+export declare class MongoAgentConfigRepository implements AgentConfigRepository {
+    findById(id: string): Promise<AgentConfigPreference | null>;
+    find(filter?: Partial<AgentConfigPreference>, options?: {
+        sort?: {
+            field: string;
+            order: SortOrder;
+        };
+        skip?: number;
+        limit?: number;
+    }): Promise<AgentConfigPreference[]>;
+    findOne(filter: Partial<AgentConfigPreference>): Promise<AgentConfigPreference | null>;
+    create(entity: Omit<AgentConfigPreference, 'id' | 'createdAt' | 'updatedAt'>): Promise<AgentConfigPreference>;
+    update(id: string, updates: Partial<Omit<AgentConfigPreference, 'id' | 'createdAt'>>): Promise<AgentConfigPreference | null>;
+    delete(id: string): Promise<boolean>;
+    exists(id: string): Promise<boolean>;
+    count(filter?: Partial<AgentConfigPreference>): Promise<number>;
+    findByUserId(userId: string): Promise<AgentConfigPreference[]>;
+    findByUserIdAndAgentName(userId: string, agentName: string): Promise<AgentConfigPreference | null>;
+    upsertByUserIdAndAgentName(userId: string, agentName: string, updates: Pick<AgentConfigPreference, 'tools'>): Promise<AgentConfigPreference>;
+    deleteByUserId(userId: string): Promise<number>;
+}
+//# sourceMappingURL=MongoAgentConfigRepository.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"MongoAgentConfigRepository.d.ts","sourceRoot":"","sources":["../../../src/repositories/MongoAgentConfigRepository.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAC7E,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAC1E,OAAO,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAyB5D,qBAAa,0BAA2B,YAAW,qBAAqB;IAChE,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC;IAK3D,IAAI,CACR,MAAM,CAAC,EAAE,OAAO,CAAC,qBAAqB,CAAC,EACvC,OAAO,CAAC,EAAE;QACR,IAAI,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,SAAS,CAAA;SAAE,CAAC;QAC3C,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GACA,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAqB7B,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,qBAAqB,CAAC,GAAG,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC;IAUtF,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,qBAAqB,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAW7G,MAAM,CACV,EAAE,EAAE,MAAM,EACV,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAChE,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC;IASlC,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,qBAAqB,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAU/D,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAK9D,wBAAwB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC;IAKlG,0BAA0B,CAC9B,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,IAAI,CAAC,qBAAqB,EAAE,OAAO,CAAC,GAC5C,OAAO,CAAC,qBAAqB,CAAC;IAoB3B,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAItD"}

package/dist/cjs/repositories/MongoAgentMessageRepository.cjs

@@ -14,14 +14,16 @@ function toAgentMessage(doc) {
         id: doc._id.toString(),
         chat: doc.chat.toString(),
         role: message.role,
-        content: message.content,
+        content: message.content || '',
         createdAt: message.createdAt,
+        activity: message.activity,
         reasoning: message.reasoning,
         toolCalls: message.toolCalls || [],
         attachments: message.attachments || [],
         annotations: message.annotations,
         tokens: message.tokens,
         agentName: message.agentName,
+        userId: message.userId,
     };
 }
 /**
@@ -35,22 +37,22 @@ class MongoAgentMessageRepository {
     async find(filter) {
         const query = {};
         if (filter) {
-            if (filter.id) {
-                query._id = filter.id;
-                delete filter.id;
+            const { id, ...rest } = filter;
+            if (id) {
+                query._id = id;
             }
-            Object.assign(query, filter);
+            Object.assign(query, rest);
         }
         const docs = await AgentMessageSchema_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;
+        const { id, ...rest } = filter;
+        if (id) {
+            query._id = id;
         }
-        Object.assign(query, filter);
+        Object.assign(query, rest);
         const doc = await AgentMessageSchema_1.AgentMessageModel.findOne(query).exec();
         return toAgentMessage(doc);
     }
@@ -69,7 +71,29 @@ class MongoAgentMessageRepository {
         return result;
     }
     async update(id, updates) {
-        const doc = await AgentMessageSchema_1.AgentMessageModel.findByIdAndUpdate(id, { $set: updates }, { new: true }).exec();
+        const doc = await AgentMessageSchema_1.AgentMessageModel.findByIdAndUpdate(id, { $set: updates }, { new: true, runValidators: true }).exec();
+        return toAgentMessage(doc);
+    }
+    async updateToolCall(messageId, toolCallId, updates) {
+        const setUpdates = {};
+        if (updates.input !== undefined) {
+            setUpdates['toolCalls.$.input'] = updates.input;
+        }
+        if (updates.output !== undefined) {
+            setUpdates['toolCalls.$.output'] = updates.output;
+        }
+        if (updates.status !== undefined) {
+            setUpdates['toolCalls.$.status'] = updates.status;
+        }
+        if (Object.keys(setUpdates).length === 0) {
+            return this.findById(messageId);
+        }
+        const doc = await AgentMessageSchema_1.AgentMessageModel.findOneAndUpdate({
+            _id: messageId,
+            'toolCalls.id': toolCallId,
+        }, {
+            $set: setUpdates,
+        }, { new: true, runValidators: true }).exec();
         return toAgentMessage(doc);
     }
     async delete(id) {
@@ -83,11 +107,11 @@ class MongoAgentMessageRepository {
     async count(filter) {
         const query = {};
         if (filter) {
-            if (filter.id) {
-                query._id = filter.id;
-                delete filter.id;
+            const { id, ...rest } = filter;
+            if (id) {
+                query._id = id;
             }
-            Object.assign(query, filter);
+            Object.assign(query, rest);
         }
         return AgentMessageSchema_1.AgentMessageModel.countDocuments(query).exec();
     }
@@ -95,6 +119,34 @@ class MongoAgentMessageRepository {
         const docs = await AgentMessageSchema_1.AgentMessageModel.find({ chat: chatId }).exec();
         return docs.map(doc => toAgentMessage(doc)).filter(Boolean);
     }
+    async findByChatIdPaginated(chatId, options) {
+        const { limit, before } = options ?? {};
+        if (limit == null || limit <= 0) {
+            const messages = await this.findByChatId(chatId);
+            return { messages, hasMore: false };
+        }
+        const query = { chat: chatId };
+        if (before) {
+            const cursorMessage = await AgentMessageSchema_1.AgentMessageModel.findById(before).exec();
+            if (!cursorMessage || cursorMessage.chat.toString() !== chatId) {
+                throw new Error('Message not found or does not belong to this chat');
+            }
+            const cursorCreatedAt = cursorMessage.createdAt ?? cursorMessage.get?.('createdAt');
+            if (cursorCreatedAt) {
+                query.createdAt = { $lt: cursorCreatedAt };
+            }
+        }
+        const take = limit + 1;
+        const docs = await AgentMessageSchema_1.AgentMessageModel.find(query)
+            .sort({ createdAt: -1 })
+            .limit(take)
+            .exec();
+        const hasMore = docs.length > limit;
+        const slice = hasMore ? docs.slice(0, limit) : docs;
+        const messages = slice.map(doc => toAgentMessage(doc)).filter(Boolean);
+        messages.reverse();
+        return { messages, hasMore };
+    }
     async findByRole(role) {
         const docs = await AgentMessageSchema_1.AgentMessageModel.find({ role }).exec();
         return docs.map(doc => toAgentMessage(doc)).filter(Boolean);