feat(agent-model): add per-agent model override with default-reset UX and runtime sync (#651)
This commit is contained in:
@@ -19,6 +19,8 @@ const mocks = vi.hoisted(() => ({
|
||||
setOpenClawDefaultModelWithOverride: vi.fn(),
|
||||
syncProviderConfigToOpenClaw: vi.fn(),
|
||||
updateAgentModelProvider: vi.fn(),
|
||||
updateSingleAgentModelProvider: vi.fn(),
|
||||
listAgentsSnapshot: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock('@electron/services/providers/provider-store', () => ({
|
||||
@@ -50,6 +52,11 @@ vi.mock('@electron/utils/openclaw-auth', () => ({
|
||||
setOpenClawDefaultModelWithOverride: mocks.setOpenClawDefaultModelWithOverride,
|
||||
syncProviderConfigToOpenClaw: mocks.syncProviderConfigToOpenClaw,
|
||||
updateAgentModelProvider: mocks.updateAgentModelProvider,
|
||||
updateSingleAgentModelProvider: mocks.updateSingleAgentModelProvider,
|
||||
}));
|
||||
|
||||
vi.mock('@electron/utils/agent-config', () => ({
|
||||
listAgentsSnapshot: mocks.listAgentsSnapshot,
|
||||
}));
|
||||
|
||||
vi.mock('@electron/utils/logger', () => ({
|
||||
@@ -62,6 +69,7 @@ vi.mock('@electron/utils/logger', () => ({
|
||||
}));
|
||||
|
||||
import {
|
||||
syncAgentModelOverrideToRuntime,
|
||||
syncDefaultProviderToRuntime,
|
||||
syncDeletedProviderToRuntime,
|
||||
syncSavedProviderToRuntime,
|
||||
@@ -109,6 +117,8 @@ describe('provider-runtime-sync refresh strategy', () => {
|
||||
mocks.saveProviderKeyToOpenClaw.mockResolvedValue(undefined);
|
||||
mocks.removeProviderFromOpenClaw.mockResolvedValue(undefined);
|
||||
mocks.updateAgentModelProvider.mockResolvedValue(undefined);
|
||||
mocks.updateSingleAgentModelProvider.mockResolvedValue(undefined);
|
||||
mocks.listAgentsSnapshot.mockResolvedValue({ agents: [] });
|
||||
});
|
||||
|
||||
it('uses debouncedReload after saving provider config', async () => {
|
||||
@@ -142,4 +152,48 @@ describe('provider-runtime-sync refresh strategy', () => {
|
||||
expect(gateway.debouncedReload).not.toHaveBeenCalled();
|
||||
expect(gateway.debouncedRestart).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('syncs a targeted agent model override to runtime provider registry', async () => {
|
||||
mocks.getAllProviders.mockResolvedValue([
|
||||
createProvider({
|
||||
id: 'ark',
|
||||
type: 'ark',
|
||||
model: 'doubao-pro',
|
||||
}),
|
||||
]);
|
||||
mocks.getProviderConfig.mockImplementation((providerType: string) => {
|
||||
if (providerType === 'ark') {
|
||||
return {
|
||||
api: 'openai-completions',
|
||||
baseUrl: 'https://ark.cn-beijing.volces.com/api/v3',
|
||||
apiKeyEnv: 'ARK_API_KEY',
|
||||
};
|
||||
}
|
||||
return {
|
||||
api: 'openai-completions',
|
||||
baseUrl: 'https://api.moonshot.cn/v1',
|
||||
apiKeyEnv: 'MOONSHOT_API_KEY',
|
||||
};
|
||||
});
|
||||
mocks.listAgentsSnapshot.mockResolvedValue({
|
||||
agents: [
|
||||
{
|
||||
id: 'coder',
|
||||
modelRef: 'ark/ark-code-latest',
|
||||
},
|
||||
],
|
||||
});
|
||||
|
||||
await syncAgentModelOverrideToRuntime('coder');
|
||||
|
||||
expect(mocks.updateSingleAgentModelProvider).toHaveBeenCalledWith(
|
||||
'coder',
|
||||
'ark',
|
||||
expect.objectContaining({
|
||||
baseUrl: 'https://ark.cn-beijing.volces.com/api/v3',
|
||||
api: 'openai-completions',
|
||||
models: [{ id: 'ark-code-latest', name: 'ark-code-latest' }],
|
||||
}),
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user