fix(core): 修复子 Agent 使用不同 provider 时配置错误的问题
当 Agent 通过 Global Defaults 指定不同的 provider 时, 之前错误地使用主 Agent 的 apiKey 和 baseUrl, 现在会从 ProviderRegistry 获取对应 provider 的配置
This commit is contained in:
@@ -37,10 +37,13 @@ vi.mock('../../../src/agent/permission-merger.js', () => ({
|
||||
|
||||
// Mock provider registry
|
||||
const mockGetModelFactory = vi.fn();
|
||||
const mockGetConfig = vi.fn();
|
||||
vi.mock('../../../src/provider/index.js', () => ({
|
||||
getProviderRegistry: () => ({
|
||||
getModelFactory: (...args: unknown[]) => mockGetModelFactory(...args),
|
||||
getConfig: (...args: unknown[]) => mockGetConfig(...args),
|
||||
}),
|
||||
resolveApiKey: (config: { apiKey?: string } | undefined) => config?.apiKey,
|
||||
}));
|
||||
|
||||
import { AgentExecutor } from '../../../src/agent/executor.js';
|
||||
@@ -113,6 +116,50 @@ describe('AgentExecutor - Agent 执行器扩展测试', () => {
|
||||
|
||||
expect(mockGetModelFactory).toHaveBeenCalledWith('openai', expect.any(Object));
|
||||
});
|
||||
|
||||
it('当 Agent 使用不同 provider 时,获取对应 provider 的配置', () => {
|
||||
// 设置 openai provider 的配置
|
||||
mockGetConfig.mockReturnValue({
|
||||
apiKey: 'openai-api-key',
|
||||
baseUrl: 'https://api.openai.com/v1',
|
||||
});
|
||||
|
||||
const agentWithDifferentProvider: AgentInfo = {
|
||||
...basicAgentInfo,
|
||||
model: { provider: 'openai', model: 'gpt-4' },
|
||||
};
|
||||
|
||||
new AgentExecutor(agentWithDifferentProvider, baseConfig, mockToolRegistry as any);
|
||||
|
||||
// 应该调用 getConfig 获取 openai 的配置
|
||||
expect(mockGetConfig).toHaveBeenCalledWith('openai');
|
||||
|
||||
// 应该使用 openai 的配置,而不是 baseConfig 的
|
||||
expect(mockGetModelFactory).toHaveBeenCalledWith('openai', {
|
||||
apiKey: 'openai-api-key',
|
||||
baseUrl: 'https://api.openai.com/v1',
|
||||
});
|
||||
});
|
||||
|
||||
it('当 Agent 使用相同 provider 时,使用 baseConfig 的配置', () => {
|
||||
mockGetConfig.mockClear();
|
||||
|
||||
const agentWithSameProvider: AgentInfo = {
|
||||
...basicAgentInfo,
|
||||
model: { provider: 'anthropic', model: 'claude-opus' },
|
||||
};
|
||||
|
||||
new AgentExecutor(agentWithSameProvider, baseConfig, mockToolRegistry as any);
|
||||
|
||||
// 不应该调用 getConfig,因为 provider 相同
|
||||
expect(mockGetConfig).not.toHaveBeenCalled();
|
||||
|
||||
// 应该使用 baseConfig 的配置
|
||||
expect(mockGetModelFactory).toHaveBeenCalledWith('anthropic', {
|
||||
apiKey: 'test-key',
|
||||
baseUrl: undefined,
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('execute - 执行任务', () => {
|
||||
|
||||
Reference in New Issue
Block a user