From 111bfc6a0d824b9626e69a3f9e4e69145cdd2558 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Tue, 9 Dec 2025 18:17:31 +0800
Subject: [PATCH] fix: test

---
 packages/service/common/middle/tracks/schema.ts       |  2 +-
 test/cases/service/common/vectorDB/controller.test.ts | 10 +++++-----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/packages/service/common/middle/tracks/schema.ts b/packages/service/common/middle/tracks/schema.ts
index bed24e7b4..97acafe55 100644
--- a/packages/service/common/middle/tracks/schema.ts
+++ b/packages/service/common/middle/tracks/schema.ts
@@ -23,7 +23,7 @@ TrackSchema.index(
 );
 // QPM index
 TrackSchema.index(
-  { event: 1, createTime: -1 },
+  { event: 1, createTime: -1, 'data.requestCount': 1 },
   {
     partialFilterExpression: {
       event: TrackEnum.teamChatQPM
diff --git a/test/cases/service/common/vectorDB/controller.test.ts b/test/cases/service/common/vectorDB/controller.test.ts
index aac7e6ca9..7582d71bb 100644
--- a/test/cases/service/common/vectorDB/controller.test.ts
+++ b/test/cases/service/common/vectorDB/controller.test.ts
@@ -121,27 +121,27 @@ describe('VectorDB Controller', () => {
       const result = await getVectorCountByTeamId('team_123');
 
       expect(result).toBe(150);
-      expect(mockGetVectorCountByTeamId).not.toHaveBeenCalled();
+      expect(mockGetVectorCount).not.toHaveBeenCalled();
     });
 
     it('should fetch from Vector and cache if no cache exists', async () => {
       mockGetRedisCache.mockResolvedValue(null);
-      mockGetVectorCountByTeamId.mockResolvedValue(200);
+      mockGetVectorCount.mockResolvedValue(200);
 
       const result = await getVectorCountByTeamId('team_456');
 
       expect(result).toBe(200);
-      expect(mockGetVectorCountByTeamId).toHaveBeenCalledWith('team_456');
+      expect(mockGetVectorCount).toHaveBeenCalledWith({ teamId: 'team_456' });
     });
 
     it('should handle undefined cache value', async () => {
       mockGetRedisCache.mockResolvedValue(undefined);
-      mockGetVectorCountByTeamId.mockResolvedValue(50);
+      mockGetVectorCount.mockResolvedValue(50);
 
       const result = await getVectorCountByTeamId('team_789');
 
       expect(result).toBe(50);
-      expect(mockGetVectorCountByTeamId).toHaveBeenCalled();
+      expect(mockGetVectorCount).toHaveBeenCalledWith({ teamId: 'team_789' });
     });
   });