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

package/dist/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,UAAU;AACV,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAElE,eAAe;AACf,OAAO,EAAE,wBAAwB,EAAE,MAAM,yCAAyC,CAAC;AACnF,OAAO,EAAE,2BAA2B,EAAE,MAAM,4CAA4C,CAAC;AAEzF,QAAQ;AACR,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,MAAM,SAAS,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAoB,MAAM,gBAAgB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,UAAU;AACV,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAClE,OAAO,EAAE,2BAA2B,EAAE,MAAM,uCAAuC,CAAC;AACpF,OAAO,EAAE,wBAAwB,EAAE,MAAM,oCAAoC,CAAC;AAE9E,eAAe;AACf,OAAO,EAAE,wBAAwB,EAAE,MAAM,yCAAyC,CAAC;AACnF,OAAO,EAAE,2BAA2B,EAAE,MAAM,4CAA4C,CAAC;AACzF,OAAO,EAAE,uBAAuB,EAAE,MAAM,wCAAwC,CAAC;AACjF,OAAO,EAAE,0BAA0B,EAAE,MAAM,2CAA2C,CAAC;AACvF,OAAO,EAAE,iCAAiC,EAAE,MAAM,kDAAkD,CAAC;AAErG,QAAQ;AACR,OAAO,EAAE,OAAO,IAAI,KAAK,EAAE,MAAM,SAAS,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAoB,MAAM,gBAAgB,CAAC"}

package/dist/esm/repositories/MongoActivityRepository.d.ts

@@ -0,0 +1,64 @@
+import { Activity, ActivitySummary, ActivitySummaryParams, SortOrder } from '@multiplayer-app/ai-agent-types';
+import type { ActivityRepository } from '@multiplayer-app/ai-agent-db';
+/**
+ * MongoDB implementation of ActivityRepository
+ */
+export declare class MongoActivityRepository implements ActivityRepository {
+    getActivitySummary(params?: ActivitySummaryParams): Promise<ActivitySummary[]>;
+    findById(id: string): Promise<Activity | null>;
+    find(filter?: Partial<Activity>, options?: {
+        sort?: {
+            field: string;
+            order: SortOrder;
+        };
+        limit?: number;
+        skip?: number;
+    }): Promise<Activity[]>;
+    findOne(filter: Partial<Activity>): Promise<Activity | null>;
+    create(entity: Omit<Activity, 'id' | 'createdAt' | 'updatedAt'>, id?: string): Promise<Activity>;
+    update(id: string, updates: Partial<Omit<Activity, 'id' | 'createdAt'>>): Promise<Activity | null>;
+    updateMetadata(id: string, metadata: Partial<Activity['metadata']>): Promise<Activity | null>;
+    delete(id: string): Promise<boolean>;
+    exists(id: string): Promise<boolean>;
+    findByParentId(filter?: {
+        ownerId?: string;
+        groupId?: string;
+        parentId?: string;
+        tenants?: Record<string, string>;
+        period?: {
+            from?: string | Date;
+            to?: string | Date;
+        };
+    }, options?: {
+        sort?: {
+            field: string;
+            order: SortOrder;
+        };
+        limit?: number;
+        skip?: number;
+    }): Promise<Activity[]>;
+    countByParentId(filter?: {
+        ownerId?: string;
+        groupId?: string;
+        parentId?: string;
+        tenants?: Record<string, string>;
+        period?: {
+            from?: string | Date;
+            to?: string | Date;
+        };
+    }): Promise<number>;
+    count(filter?: Partial<Activity>): Promise<number>;
+    deleteByOwnerId(ownerId: string): Promise<number>;
+    deleteByGroupId(groupId: string): Promise<number>;
+    getGroupedMetadataByParentId(parentId: string): Promise<{
+        inputTokens: number;
+        outputTokens: number;
+        cost: number;
+        childCount: number;
+        totalTokens: number;
+        reasoningTokens: number;
+        cachedInputTokens: number;
+    }>;
+    calculateCost(activityData: Partial<Activity>): Promise<number>;
+}
+//# sourceMappingURL=MongoActivityRepository.d.ts.map

package/dist/esm/repositories/MongoActivityRepository.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoActivityRepository.d.ts","sourceRoot":"","sources":["../../../src/repositories/MongoActivityRepository.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,qBAAqB,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAC9G,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAyDvE;;GAEG;AACH,qBAAa,uBAAwB,YAAW,kBAAkB;IAC1D,kBAAkB,CAAC,MAAM,CAAC,EAAE,qBAAqB,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IA2H9E,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC;IAK9C,IAAI,CACR,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,OAAO,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,SAAS,CAAA;SAAE,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,GACtF,OAAO,CAAC,QAAQ,EAAE,CAAC;IAqDhB,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC;IAa5D,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC;IAoBhG,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC;IAUlG,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,GAAG,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC;IAU7F,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpC,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAKpC,cAAc,CAAC,MAAM,CAAC,EAAE;QAC5B,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,EAAE;YAAE,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;YAAC,EAAE,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;SAAE,CAAA;KACtD,EAAE,OAAO,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,SAAS,CAAA;SAAE,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAsD1G,eAAe,CAAC,MAAM,CAAC,EAAE;QAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,EAAE;YAAE,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;YAAC,EAAE,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;SAAE,CAAA;KACtD,GAAG,OAAO,CAAC,MAAM,CAAC;IAqBb,KAAK,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAclD,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAKjD,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAKjD,4BAA4B,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC;QAC5D,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;QACpB,eAAe,EAAE,MAAM,CAAC;QACxB,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;IAsDI,aAAa,CAAC,YAAY,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;CAGtE"}

package/dist/esm/repositories/MongoActivityRepository.js

@@ -0,0 +1,398 @@
+import { SortOrder } from '@multiplayer-app/ai-agent-types';
+import { ObjectId } from 'mongodb';
+import { ActivityModel } from '../schemas/ActivitySchema';
+function flattenObject(obj, parentKey = '', result = {}) {
+    for (const key in obj) {
+        const newKey = parentKey ? `${parentKey}.${key}` : key;
+        if (obj[key] &&
+            typeof obj[key] === 'object' &&
+            !Array.isArray(obj[key]) &&
+            !(obj[key] instanceof Date)) {
+            flattenObject(obj[key], newKey, result);
+        }
+        else {
+            result[newKey] = obj[key];
+        }
+    }
+    return result;
+}
+function normalizeTenants(tenants) {
+    if (!tenants)
+        return {};
+    if (tenants instanceof Map) {
+        return Object.fromEntries(tenants.entries());
+    }
+    return tenants;
+}
+/**
+ * Helper function to convert Mongoose document to Activity type
+ */
+function toActivity(doc) {
+    if (!doc)
+        return null;
+    const activity = doc.toObject ? doc.toObject() : doc;
+    return {
+        id: doc._id.toString(),
+        ownerId: activity.ownerId,
+        groupId: activity.groupId,
+        name: activity.name,
+        description: activity.description,
+        parentId: activity.parentId,
+        hasChildren: activity.hasChildren,
+        tenants: normalizeTenants(activity.tenants),
+        sourceId: activity.sourceId || '',
+        sourceType: activity.sourceType || 'Unknown',
+        metadata: activity.metadata || {}
+    };
+}
+/**
+ * MongoDB implementation of ActivityRepository
+ */
+export class MongoActivityRepository {
+    async getActivitySummary(params) {
+        const baseMatch = {
+            parentId: { $exists: true, $ne: null },
+        };
+        const fromDate = params?.period?.from ? new Date(params.period.from) : undefined;
+        const toDate = params?.period?.to ? new Date(params.period.to) : undefined;
+        if (fromDate && !Number.isNaN(fromDate.getTime())) {
+            baseMatch.createdAt = { ...(baseMatch.createdAt || {}), $gte: fromDate };
+        }
+        if (toDate && !Number.isNaN(toDate.getTime())) {
+            baseMatch.createdAt = { ...(baseMatch.createdAt || {}), $lte: toDate };
+        }
+        if (params?.ownerId) {
+            baseMatch.ownerId = params.ownerId;
+        }
+        if (typeof params?.tenant === 'string') {
+            baseMatch.$expr = {
+                $in: [
+                    params.tenant,
+                    {
+                        $map: {
+                            input: { $objectToArray: { $ifNull: ['$tenants', {}] } },
+                            as: 'tenantEntry',
+                            in: '$$tenantEntry.v',
+                        },
+                    },
+                ],
+            };
+        }
+        else if (params?.tenant && typeof params.tenant === 'object') {
+            for (const [tenantKey, tenantValue] of Object.entries(params.tenant)) {
+                baseMatch[`tenants.${tenantKey}`] = tenantValue;
+            }
+        }
+        const effectiveGroupBy = params?.groupBy?.length
+            ? Array.from(new Set(params.groupBy))
+            : ['model'];
+        const includeModel = effectiveGroupBy.includes('model');
+        const includeUserId = effectiveGroupBy.includes('userId');
+        const includeGroupBy = effectiveGroupBy.filter(groupBy => groupBy !== 'model' && groupBy !== 'userId');
+        const includePeriodStart = Boolean(params?.timeBucket);
+        const groupId = {};
+        if (includeModel)
+            groupId.model = '$summaryModel';
+        if (includeUserId)
+            groupId.userId = '$summaryUserId';
+        if (includePeriodStart)
+            groupId.periodStart = '$summaryPeriodStart';
+        includeGroupBy.forEach(groupBy => {
+            groupId[groupBy] = `$${groupBy}`;
+        });
+        const pipeline = [
+            { $match: baseMatch },
+            {
+                $addFields: {
+                    summaryModel: { $ifNull: ['$metadata.modelId', 'unknown'] },
+                    summaryTenant: { $ifNull: ['$tenants', {}] },
+                    summaryUserId: '$ownerId',
+                    summaryTotalTokens: { $ifNull: ['$metadata.usage.totalTokens', 0] },
+                    summaryCost: { $ifNull: ['$metadata.usage.cost', 0] },
+                    ...(includePeriodStart
+                        ? {
+                            summaryPeriodStart: {
+                                $dateTrunc: {
+                                    date: '$createdAt',
+                                    unit: params?.timeBucket,
+                                    timezone: 'UTC',
+                                },
+                            },
+                        }
+                        : {}),
+                },
+            },
+        ];
+        if (params?.model) {
+            pipeline.push({ $match: { summaryModel: params.model } });
+        }
+        pipeline.push({
+            $group: {
+                _id: groupId,
+                totalRequests: { $sum: 1 },
+                totalTokens: { $sum: '$summaryTotalTokens' },
+                totalCost: { $sum: '$summaryCost' },
+            },
+        }, {
+            $project: {
+                _id: 0,
+                ...(includePeriodStart ? { periodStart: '$_id.periodStart' } : {}),
+                ...(includeModel ? { model: '$_id.model' } : {}),
+                ...(includeUserId ? { userId: '$_id.userId' } : {}),
+                ...(includeGroupBy.length > 0 ? Object.fromEntries(includeGroupBy.map(groupBy => [groupBy, `$_id.${groupBy}`])) : {}),
+                totalRequests: 1,
+                totalTokens: 1,
+                totalCost: 1,
+            },
+        });
+        const result = await ActivityModel.aggregate(pipeline).exec();
+        return result.map((item) => ({
+            ...(includePeriodStart
+                ? {
+                    periodStart: item.periodStart instanceof Date
+                        ? item.periodStart.toISOString()
+                        : item.periodStart
+                            ? new Date(item.periodStart).toISOString()
+                            : undefined,
+                }
+                : {}),
+            ...(includeModel ? { model: item.model } : {}),
+            ...(includeGroupBy.length > 0 ? Object.fromEntries(includeGroupBy.map(groupBy => [groupBy, item[groupBy]])) : {}),
+            ...(includeUserId ? { userId: item.userId } : {}),
+            totalRequests: item.totalRequests ?? 0,
+            totalTokens: item.totalTokens ?? 0,
+            totalCost: item.totalCost ?? 0,
+        }));
+    }
+    async findById(id) {
+        const doc = await ActivityModel.findById(id).exec();
+        return toActivity(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 = ActivityModel.find(query);
+        // Apply sorting
+        if (options?.sort) {
+            const sortOrder = options.sort.order === SortOrder.Asc ? 1 : -1;
+            queryBuilder = queryBuilder.sort({ [options.sort.field]: sortOrder });
+        }
+        // Apply limit
+        if (options?.limit) {
+            queryBuilder = queryBuilder.limit(options.limit);
+        }
+        if (options?.skip) {
+            queryBuilder = queryBuilder.skip(options.skip);
+        }
+        const docs = await queryBuilder.exec();
+        if (docs.length === 0) {
+            return [];
+        }
+        const activityIds = docs.map(doc => doc._id.toString());
+        const parentIdsWithChildren = await ActivityModel.distinct('parentId', {
+            parentId: { $in: activityIds },
+        }).exec();
+        const parentIdsWithChildrenSet = new Set(parentIdsWithChildren);
+        return docs.reduce((activities, doc) => {
+            const activity = toActivity(doc);
+            if (!activity) {
+                return activities;
+            }
+            activities.push({
+                ...activity,
+                hasChildren: parentIdsWithChildrenSet.has(activity.id),
+            });
+            return activities;
+        }, []);
+    }
+    async findOne(filter) {
+        const query = {};
+        if (filter.id) {
+            query._id = filter.id;
+            delete filter.id;
+        }
+        Object.assign(query, filter);
+        const doc = await ActivityModel.findOne(query).exec();
+        return toActivity(doc);
+    }
+    async create(entity, id) {
+        const doc = new ActivityModel({
+            ...entity,
+            ...(id ? { _id: id } : {})
+        });
+        const cost = await this.calculateCost(entity);
+        doc.systemCost = doc.metadata?.usage?.cost ?? 0;
+        doc.metadata ??= {};
+        doc.metadata.usage ??= {};
+        doc.metadata.usage.cost = cost;
+        const saved = await doc.save();
+        const result = toActivity(saved);
+        if (!result) {
+            throw new Error('Failed to create activity');
+        }
+        return result;
+    }
+    async update(id, updates) {
+        const doc = await ActivityModel.findByIdAndUpdate(id, { $set: updates }, { new: true, runValidators: true }).exec();
+        return toActivity(doc);
+    }
+    async updateMetadata(id, metadata) {
+        const flattenedUpdates = flattenObject(metadata, 'metadata');
+        const doc = await ActivityModel.findByIdAndUpdate(id, { $set: flattenedUpdates }, { new: true, runValidators: true }).exec();
+        return toActivity(doc);
+    }
+    async delete(id) {
+        const result = await ActivityModel.findByIdAndDelete(id).exec();
+        return !!result;
+    }
+    async exists(id) {
+        const count = await ActivityModel.countDocuments({ _id: id }).exec();
+        return count > 0;
+    }
+    async findByParentId(filter, options) {
+        const query = {
+            parentId: filter?.parentId ? filter?.parentId : { $exists: false },
+            ...(filter?.ownerId ? { ownerId: filter?.ownerId } : {}),
+            ...(filter?.groupId ? { groupId: filter?.groupId } : {}),
+            ...(filter?.tenants && typeof filter.tenants === 'object' && Object.keys(filter.tenants).length > 0
+                ? Object.fromEntries(Object.entries(filter.tenants).map(([k, v]) => [`tenants.${k}`, v]))
+                : {}),
+        };
+        const fromDate = filter?.period?.from ? new Date(filter.period.from) : undefined;
+        const toDate = filter?.period?.to ? new Date(filter.period.to) : undefined;
+        if (fromDate && !Number.isNaN(fromDate.getTime())) {
+            query.createdAt = { ...(query.createdAt || {}), $gte: fromDate };
+        }
+        if (toDate && !Number.isNaN(toDate.getTime())) {
+            query.createdAt = { ...(query.createdAt || {}), $lte: toDate };
+        }
+        let queryBuilder = ActivityModel.find(query);
+        if (options?.sort) {
+            const sortOrder = options.sort.order === SortOrder.Asc ? 1 : -1;
+            queryBuilder = queryBuilder.sort({ [options.sort.field]: sortOrder });
+        }
+        if (options?.limit) {
+            queryBuilder = queryBuilder.limit(options.limit);
+        }
+        if (options?.skip) {
+            queryBuilder = queryBuilder.skip(options.skip);
+        }
+        const docs = await queryBuilder.exec();
+        const activityIds = docs.map(doc => doc._id.toString());
+        const parentIdsWithChildren = await ActivityModel.distinct('parentId', {
+            parentId: { $in: activityIds },
+        }).exec();
+        const parentIdsWithChildrenSet = new Set(parentIdsWithChildren);
+        return docs.reduce((activities, doc) => {
+            const activity = toActivity(doc);
+            if (!activity) {
+                return activities;
+            }
+            activities.push({
+                ...activity,
+                hasChildren: parentIdsWithChildrenSet.has(activity.id),
+            });
+            return activities;
+        }, []);
+    }
+    async countByParentId(filter) {
+        const query = {
+            parentId: filter?.parentId ? filter?.parentId : { $exists: false },
+            ...(filter?.ownerId ? { ownerId: filter?.ownerId } : {}),
+            ...(filter?.groupId ? { groupId: filter?.groupId } : {}),
+            ...(filter?.tenants && typeof filter.tenants === 'object' && Object.keys(filter.tenants).length > 0
+                ? Object.fromEntries(Object.entries(filter.tenants).map(([k, v]) => [`tenants.${k}`, v]))
+                : {}),
+        };
+        const fromDate = filter?.period?.from ? new Date(filter.period.from) : undefined;
+        const toDate = filter?.period?.to ? new Date(filter.period.to) : undefined;
+        if (fromDate && !Number.isNaN(fromDate.getTime())) {
+            query.createdAt = { ...(query.createdAt || {}), $gte: fromDate };
+        }
+        if (toDate && !Number.isNaN(toDate.getTime())) {
+            query.createdAt = { ...(query.createdAt || {}), $lte: toDate };
+        }
+        return ActivityModel.countDocuments(query).exec();
+    }
+    async count(filter) {
+        const query = {};
+        if (filter) {
+            if (filter.id) {
+                query._id = filter.id;
+                delete filter.id;
+            }
+            Object.assign(query, filter);
+        }
+        return ActivityModel.countDocuments(query).exec();
+    }
+    async deleteByOwnerId(ownerId) {
+        const result = await ActivityModel.deleteMany({ ownerId }).exec();
+        return result.deletedCount || 0;
+    }
+    async deleteByGroupId(groupId) {
+        const result = await ActivityModel.deleteMany({ groupId }).exec();
+        return result.deletedCount || 0;
+    }
+    async getGroupedMetadataByParentId(parentId) {
+        const result = await ActivityModel.aggregate([
+            {
+                $match: { _id: new ObjectId(parentId) }
+            },
+            {
+                $graphLookup: {
+                    from: "activities",
+                    startWith: { $toString: "$_id" },
+                    connectFromField: "_id",
+                    connectToField: "parentId",
+                    as: "descendants"
+                }
+            },
+            {
+                $project: {
+                    _id: 0,
+                    inputTokens: {
+                        $sum: "$descendants.metadata.usage.inputTokens"
+                    },
+                    outputTokens: {
+                        $sum: "$descendants.metadata.usage.outputTokens"
+                    },
+                    cost: {
+                        $sum: "$descendants.metadata.usage.cost"
+                    },
+                    totalTokens: {
+                        $sum: "$descendants.metadata.usage.totalTokens"
+                    },
+                    reasoningTokens: {
+                        $sum: "$descendants.metadata.usage.reasoningTokens"
+                    },
+                    cachedInputTokens: {
+                        $sum: "$descendants.metadata.usage.cachedInputTokens"
+                    },
+                    childCount: {
+                        $size: "$descendants"
+                    }
+                }
+            }
+        ]).exec();
+        return {
+            inputTokens: result[0]?.inputTokens ?? 0,
+            outputTokens: result[0]?.outputTokens ?? 0,
+            cost: result[0]?.cost ?? 0,
+            childCount: result[0]?.childCount ?? 0,
+            totalTokens: result[0]?.totalTokens ?? 0,
+            reasoningTokens: result[0]?.reasoningTokens ?? 0,
+            cachedInputTokens: result[0]?.cachedInputTokens ?? 0,
+        };
+    }
+    // override method for custom calculation 
+    async calculateCost(activityData) {
+        return activityData.metadata?.usage?.cost ?? 0;
+    }
+}
+//# sourceMappingURL=MongoActivityRepository.js.map

package/dist/esm/repositories/MongoActivityRepository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MongoActivityRepository.js","sourceRoot":"","sources":["../../../src/repositories/MongoActivityRepository.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoD,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAE9G,OAAO,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACnC,OAAO,EAAE,aAAa,EAAyB,MAAM,2BAA2B,CAAC;AAEjF,SAAS,aAAa,CACpB,GAAwB,EACxB,SAAS,GAAG,EAAE,EACd,SAA8B,EAAE;IAEhC,KAAK,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;QACtB,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS,IAAI,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;QAEvD,IACE,GAAG,CAAC,GAAG,CAAC;YACR,OAAO,GAAG,CAAC,GAAG,CAAC,KAAK,QAAQ;YAC5B,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACxB,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,EAC3B,CAAC;YACD,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAgB;IACxC,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAC;IACxB,IAAI,OAAO,YAAY,GAAG,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,OAAO,EAAE,CAA2B,CAAC;IACzE,CAAC;IACD,OAAO,OAAiC,CAAC;AAC3C,CAAC;AAED;;GAEG;AACH,SAAS,UAAU,CAAC,GAA4B;IAC9C,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IAEtB,MAAM,QAAQ,GAAI,GAAW,CAAC,QAAQ,CAAC,CAAC,CAAE,GAAW,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;IACvE,OAAO;QACL,EAAE,EAAE,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE;QACtB,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,OAAO,EAAE,gBAAgB,CAAC,QAAQ,CAAC,OAAO,CAAC;QAC3C,QAAQ,EAAE,QAAQ,CAAC,QAAQ,IAAI,EAAE;QACjC,UAAU,EAAE,QAAQ,CAAC,UAAU,IAAI,SAAS;QAC5C,QAAQ,EAAE,QAAQ,CAAC,QAAQ,IAAI,EAAE;KAClC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,uBAAuB;IAClC,KAAK,CAAC,kBAAkB,CAAC,MAA8B;QACrD,MAAM,SAAS,GAAwB;YACrC,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE;SACvC,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAE3E,IAAI,QAAQ,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAClD,SAAS,CAAC,SAAS,GAAG,EAAE,GAAG,CAAC,SAAS,CAAC,SAAS,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAC3E,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAC9C,SAAS,CAAC,SAAS,GAAG,EAAE,GAAG,CAAC,SAAS,CAAC,SAAS,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACzE,CAAC;QACD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QACrC,CAAC;QACD,IAAI,OAAO,MAAM,EAAE,MAAM,KAAK,QAAQ,EAAE,CAAC;YACvC,SAAS,CAAC,KAAK,GAAG;gBAChB,GAAG,EAAE;oBACH,MAAM,CAAC,MAAM;oBACb;wBACE,IAAI,EAAE;4BACJ,KAAK,EAAE,EAAE,cAAc,EAAE,EAAE,OAAO,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;4BACxD,EAAE,EAAE,aAAa;4BACjB,EAAE,EAAE,iBAAiB;yBACtB;qBACF;iBACF;aACF,CAAC;QACJ,CAAC;aAAM,IAAI,MAAM,EAAE,MAAM,IAAI,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC/D,KAAK,MAAM,CAAC,SAAS,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;gBACrE,SAAS,CAAC,WAAW,SAAS,EAAE,CAAC,GAAG,WAAW,CAAC;YAClD,CAAC;QACH,CAAC;QAED,MAAM,gBAAgB,GAAuC,MAAM,EAAE,OAAO,EAAE,MAAM;YAClF,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACrC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACd,MAAM,YAAY,GAAG,gBAAgB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACxD,MAAM,aAAa,GAAG,gBAAgB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC1D,MAAM,cAAc,GAAG,gBAAgB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,KAAK,OAAO,IAAI,OAAO,KAAK,QAAQ,CAAC,CAAC;QACvG,MAAM,kBAAkB,GAAG,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAEvD,MAAM,OAAO,GAAwB,EAAE,CAAC;QACxC,IAAI,YAAY;YAAE,OAAO,CAAC,KAAK,GAAG,eAAe,CAAC;QAClD,IAAI,aAAa;YAAE,OAAO,CAAC,MAAM,GAAG,gBAAgB,CAAC;QACrD,IAAI,kBAAkB;YAAE,OAAO,CAAC,WAAW,GAAG,qBAAqB,CAAC;QACpE,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAC/B,OAAO,CAAC,OAAO,CAAC,GAAG,IAAI,OAAO,EAAE,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAU;YACtB,EAAE,MAAM,EAAE,SAAS,EAAE;YACrB;gBACE,UAAU,EAAE;oBACV,YAAY,EAAE,EAAE,OAAO,EAAE,CAAC,mBAAmB,EAAE,SAAS,CAAC,EAAE;oBAC3D,aAAa,EAAE,EAAE,OAAO,EAAE,CAAC,UAAU,EAAE,EAAE,CAAC,EAAE;oBAC5C,aAAa,EAAE,UAAU;oBACzB,kBAAkB,EAAE,EAAE,OAAO,EAAE,CAAC,6BAA6B,EAAE,CAAC,CAAC,EAAE;oBACnE,WAAW,EAAE,EAAE,OAAO,EAAE,CAAC,sBAAsB,EAAE,CAAC,CAAC,EAAE;oBACrD,GAAG,CAAC,kBAAkB;wBACpB,CAAC,CAAC;4BACE,kBAAkB,EAAE;gCAClB,UAAU,EAAE;oCACV,IAAI,EAAE,YAAY;oCAClB,IAAI,EAAE,MAAM,EAAE,UAA+C;oCAC7D,QAAQ,EAAE,KAAK;iCAChB;6BACF;yBACF;wBACH,CAAC,CAAC,EAAE,CAAC;iBACR;aACF;SACF,CAAC;QAEF,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;YAClB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,QAAQ,CAAC,IAAI,CACX;YACE,MAAM,EAAE;gBACN,GAAG,EAAE,OAAO;gBACZ,aAAa,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;gBAC1B,WAAW,EAAE,EAAE,IAAI,EAAE,qBAAqB,EAAE;gBAC5C,SAAS,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE;aACpC;SACF,EACD;YACE,QAAQ,EAAE;gBACR,GAAG,EAAE,CAAC;gBACN,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClE,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChD,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnD,GAAG,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,EAAE,QAAQ,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACrH,aAAa,EAAE,CAAC;gBAChB,WAAW,EAAE,CAAC;gBACd,SAAS,EAAE,CAAC;aACb;SACF,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QAC9D,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC;YAChC,GAAG,CAAC,kBAAkB;gBACpB,CAAC,CAAC;oBACE,WAAW,EACT,IAAI,CAAC,WAAW,YAAY,IAAI;wBAC9B,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE;wBAChC,CAAC,CAAC,IAAI,CAAC,WAAW;4BAChB,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE;4BAC1C,CAAC,CAAC,SAAS;iBAClB;gBACH,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9C,GAAG,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACjH,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACjD,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC;YACtC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,CAAC;YAClC,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,CAAC;SAC/B,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,EAAU;QACvB,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACpD,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,IAAI,CACR,MAA0B,EAC1B,OAAuF;QAEvF,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,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE7C,gBAAgB;QAChB,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,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;QACD,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;YAClB,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,CAAC;QACvC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxD,MAAM,qBAAqB,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,UAAU,EAAE;YACrE,QAAQ,EAAE,EAAE,GAAG,EAAE,WAAW,EAAE;SAC/B,CAAC,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,wBAAwB,GAAG,IAAI,GAAG,CAAC,qBAAqB,CAAC,CAAC;QAEhE,OAAO,IAAI,CAAC,MAAM,CAAa,CAAC,UAAU,EAAE,GAAG,EAAE,EAAE;YACjD,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,UAAU,CAAC;YACpB,CAAC;YAED,UAAU,CAAC,IAAI,CAAC;gBACd,GAAG,QAAQ;gBACX,WAAW,EAAE,wBAAwB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;aACvD,CAAC,CAAC;YAEH,OAAO,UAAU,CAAC;QACpB,CAAC,EAAE,EAAE,CAAC,CAAC;IACT,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAyB;QACrC,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,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACtD,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAwD,EAAE,EAAW;QAChF,MAAM,GAAG,GAAG,IAAI,aAAa,CAAC;YAC5B,GAAG,MAAM;YACT,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC3B,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAC9C,GAAG,CAAC,UAAU,GAAG,GAAG,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,IAAI,CAAC,CAAC;QAChD,GAAG,CAAC,QAAQ,KAAK,EAAE,CAAC;QACpB,GAAG,CAAC,QAAQ,CAAC,KAAK,KAAK,EAAE,CAAC;QAC1B,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;QAE/B,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC;QACjC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,OAAoD;QAC3E,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,iBAAiB,CAC/C,EAAE,EACF,EAAE,IAAI,EAAE,OAAO,EAAE,EACjB,EAAE,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CACnC,CAAC,IAAI,EAAE,CAAC;QAET,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAU,EAAE,QAAuC;QACtE,MAAM,gBAAgB,GAAG,aAAa,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC7D,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,iBAAiB,CAC/C,EAAE,EACF,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAC1B,EAAE,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CACnC,CAAC,IAAI,EAAE,CAAC;QACT,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAChE,OAAO,CAAC,CAAC,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,KAAK,GAAG,MAAM,aAAa,CAAC,cAAc,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACrE,OAAO,KAAK,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,MAMpB,EAAE,OAAuF;QACxF,MAAM,KAAK,GAAQ;YACjB,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE;YAClE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxD,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxD,GAAG,CAAC,MAAM,EAAE,OAAO,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC;gBACjG,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;gBACzF,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3E,IAAI,QAAQ,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAClD,KAAK,CAAC,SAAS,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QACnE,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAC9C,KAAK,CAAC,SAAS,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACjE,CAAC;QAED,IAAI,YAAY,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE7C,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,KAAK,SAAS,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,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;YACnB,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,OAAO,EAAE,IAAI,EAAE,CAAC;YAClB,YAAY,GAAG,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,CAAC;QACvC,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxD,MAAM,qBAAqB,GAAG,MAAM,aAAa,CAAC,QAAQ,CAAC,UAAU,EAAE;YACrE,QAAQ,EAAE,EAAE,GAAG,EAAE,WAAW,EAAE;SAC/B,CAAC,CAAC,IAAI,EAAE,CAAC;QACV,MAAM,wBAAwB,GAAG,IAAI,GAAG,CAAC,qBAAqB,CAAC,CAAC;QAEhE,OAAO,IAAI,CAAC,MAAM,CAAa,CAAC,UAAU,EAAE,GAAG,EAAE,EAAE;YACjD,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,UAAU,CAAC;YACpB,CAAC;YAED,UAAU,CAAC,IAAI,CAAC;gBACd,GAAG,QAAQ;gBACX,WAAW,EAAE,wBAAwB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;aACvD,CAAC,CAAC;YAEH,OAAO,UAAU,CAAC;QACpB,CAAC,EAAE,EAAE,CAAC,CAAC;IACT,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,MAMrB;QACC,MAAM,KAAK,GAAQ;YACjB,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE;YAClE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxD,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxD,GAAG,CAAC,MAAM,EAAE,OAAO,IAAI,OAAO,MAAM,CAAC,OAAO,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC;gBACjG,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;gBACzF,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QACF,MAAM,QAAQ,GAAG,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3E,IAAI,QAAQ,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAClD,KAAK,CAAC,SAAS,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QACnE,CAAC;QACD,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YAC9C,KAAK,CAAC,SAAS,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QACjE,CAAC;QAED,OAAO,aAAa,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAA0B;QACpC,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,aAAa,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAe;QACnC,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAClE,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAe;QACnC,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAClE,OAAO,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,4BAA4B,CAAC,QAAgB;QASjD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,SAAS,CAAC;YAC3C;gBACE,MAAM,EAAE,EAAE,GAAG,EAAE,IAAI,QAAQ,CAAC,QAAQ,CAAC,EAAE;aACxC;YACD;gBACE,YAAY,EAAE;oBACZ,IAAI,EAAE,YAAY;oBAClB,SAAS,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;oBAChC,gBAAgB,EAAE,KAAK;oBACvB,cAAc,EAAE,UAAU;oBAC1B,EAAE,EAAE,aAAa;iBAClB;aACF;YACD;gBACE,QAAQ,EAAE;oBACR,GAAG,EAAE,CAAC;oBACN,WAAW,EAAE;wBACX,IAAI,EAAE,yCAAyC;qBAChD;oBACD,YAAY,EAAE;wBACZ,IAAI,EAAE,0CAA0C;qBACjD;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,kCAAkC;qBACzC;oBACD,WAAW,EAAE;wBACX,IAAI,EAAE,yCAAyC;qBAChD;oBACD,eAAe,EAAE;wBACf,IAAI,EAAE,6CAA6C;qBACpD;oBACD,iBAAiB,EAAE;wBACjB,IAAI,EAAE,+CAA+C;qBACtD;oBACD,UAAU,EAAE;wBACV,KAAK,EAAE,cAAc;qBACtB;iBACF;aACF;SACF,CAAC,CAAC,IAAI,EAAE,CAAC;QAEV,OAAO;YACL,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,WAAW,IAAI,CAAC;YACxC,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,IAAI,CAAC;YAC1C,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,CAAC;YAC1B,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,IAAI,CAAC;YACtC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,WAAW,IAAI,CAAC;YACxC,eAAe,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,eAAe,IAAI,CAAC;YAChD,iBAAiB,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,iBAAiB,IAAI,CAAC;SACrD,CAAA;IACH,CAAC;IAED,0CAA0C;IAC1C,KAAK,CAAC,aAAa,CAAC,YAA+B;QACjD,OAAO,YAAY,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,IAAI,CAAC,CAAC;IACjD,CAAC;CACF"}

package/dist/esm/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/esm/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"}