mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-16 16:30:15 +01:00
5b1a31ef4d
🔄 refactor: Optimize MCP Tool Initialization
fix: update tool caching to use separated mcp logic
refactor: Replace `req.user` with `userId` in MCP handling functions
refactor: Replace `req` parameter with `userId` in file search tool functions
fix: Update user connection parameter to use object format in reinitMCPServer
refactor: Simplify MCP tool creation logic and improve handling of tool configurations to avoid capturing too much in closures
refactor: ensure MCP available tools are fetched from cache only when needed
86 lines
2.5 KiB
JavaScript
86 lines
2.5 KiB
JavaScript
const { createFileSearchTool } = require('../../../../../app/clients/tools/util/fileSearch');
|
|
|
|
// Mock dependencies
|
|
jest.mock('../../../../../models', () => ({
|
|
Files: {
|
|
find: jest.fn(),
|
|
},
|
|
}));
|
|
|
|
jest.mock('../../../../../server/services/Files/VectorDB/crud', () => ({
|
|
queryVectors: jest.fn(),
|
|
}));
|
|
|
|
jest.mock('../../../../../config', () => ({
|
|
logger: {
|
|
warn: jest.fn(),
|
|
error: jest.fn(),
|
|
debug: jest.fn(),
|
|
},
|
|
}));
|
|
|
|
const { queryVectors } = require('../../../../../server/services/Files/VectorDB/crud');
|
|
|
|
describe('fileSearch.js - test only new file_id and page additions', () => {
|
|
beforeEach(() => {
|
|
jest.clearAllMocks();
|
|
});
|
|
|
|
// Test only the specific changes: file_id and page metadata additions
|
|
it('should add file_id and page to search result format', async () => {
|
|
const mockFiles = [{ file_id: 'test-file-123' }];
|
|
const mockResults = [
|
|
{
|
|
data: [
|
|
[
|
|
{
|
|
page_content: 'test content',
|
|
metadata: { source: 'test.pdf', page: 1 },
|
|
},
|
|
0.3,
|
|
],
|
|
],
|
|
},
|
|
];
|
|
|
|
queryVectors.mockResolvedValue(mockResults);
|
|
|
|
const fileSearchTool = await createFileSearchTool({
|
|
userId: 'user1',
|
|
files: mockFiles,
|
|
entity_id: 'agent-123',
|
|
});
|
|
|
|
// Mock the tool's function to return the formatted result
|
|
fileSearchTool.func = jest.fn().mockImplementation(async () => {
|
|
// Simulate the new format with file_id and page
|
|
const formattedResults = [
|
|
{
|
|
filename: 'test.pdf',
|
|
content: 'test content',
|
|
distance: 0.3,
|
|
file_id: 'test-file-123', // NEW: added file_id
|
|
page: 1, // NEW: added page
|
|
},
|
|
];
|
|
|
|
// NEW: Internal data section for processAgentResponse
|
|
const internalData = formattedResults
|
|
.map(
|
|
(result) =>
|
|
`File: ${result.filename}\nFile_ID: ${result.file_id}\nRelevance: ${(1.0 - result.distance).toFixed(4)}\nPage: ${result.page || 'N/A'}\nContent: ${result.content}\n`,
|
|
)
|
|
.join('\n---\n');
|
|
|
|
return `File: test.pdf\nRelevance: 0.7000\nContent: test content\n\n<!-- INTERNAL_DATA_START -->\n${internalData}\n<!-- INTERNAL_DATA_END -->`;
|
|
});
|
|
|
|
const result = await fileSearchTool.func('test');
|
|
|
|
// Verify the new additions
|
|
expect(result).toContain('File_ID: test-file-123');
|
|
expect(result).toContain('Page: 1');
|
|
expect(result).toContain('<!-- INTERNAL_DATA_START -->');
|
|
expect(result).toContain('<!-- INTERNAL_DATA_END -->');
|
|
});
|
|
});
|