github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2026-03-22 15:39:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: resolve unused config parameter in llm-mock

Browse Source 
Co-Authored-By: Han Xiao <han.xiao@jina.ai>
This commit is contained in:
Devin AI
2025-02-05 10:04:03 +00:00
parent 5473ce254b
commit e9e7255b55
1 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/__tests__/llm-client.test.ts
View File

@@ -1,4 +1,17 @@
import { GoogleAIWrapper, LocalLLMClient } from '../utils/llm-client';
import { llmClient } from '../config';
jest.mock('../config', () => {
let mockConfig = { llmClient: null };
return {
get llmClient() {
return mockConfig.llmClient;
},
__setMockConfig: (config: { llmClient: any }) => {
mockConfig = config;
}
};
});
describe('LLM Client', () => {
const originalEnv = process.env;
@@ -12,19 +25,23 @@ describe('LLM Client', () => {
process.env = originalEnv;
});
it('should use GoogleAIWrapper by default', async () => {
it('should use GoogleAIWrapper by default', () => {
process.env.LLM_PROVIDER = 'gemini';
process.env.GEMINI_API_KEY = 'test-key';
const config = await import('../config');
expect(config.llmClient).toBeInstanceOf(GoogleAIWrapper);
jest.isolateModules(() => {
const { llmClient } = require('../config');
expect(llmClient).toBeInstanceOf(GoogleAIWrapper);
});
});
it('should use LocalLLMClient when configured', async () => {
it('should use LocalLLMClient when configured', () => {
process.env.LLM_PROVIDER = 'local';
process.env.LOCAL_LLM_HOSTNAME = 'localhost';
process.env.LOCAL_LLM_PORT = '8000';
process.env.LOCAL_LLM_MODEL = 'test-model';
const config = await import('../config');
expect(config.llmClient).toBeInstanceOf(LocalLLMClient);
jest.isolateModules(() => {
const { llmClient } = require('../config');
expect(llmClient).toBeInstanceOf(LocalLLMClient);
});
});
});