fix(core): prevent redundant remote agent loading on model switch (#23576)

This commit is contained in:
Adam Weidman
2026-03-23 15:45:59 -04:00
committed by GitHub
parent 139ef0d5bd
commit d5b78dbeea
2 changed files with 35 additions and 5 deletions
+26
View File
@@ -1206,6 +1206,32 @@ describe('AgentRegistry', () => {
});
describe('inheritance and refresh', () => {
it('should skip remote agents when refreshing on model change', async () => {
const remoteAgent: AgentDefinition = {
kind: 'remote',
name: 'RemoteAgent',
description: 'A remote agent',
agentCardUrl: 'https://example.com/card',
inputConfig: { inputSchema: { type: 'object' } },
};
const loadAgentSpy = vi.fn().mockResolvedValue({ name: 'RemoteAgent' });
vi.spyOn(mockConfig, 'getA2AClientManager').mockReturnValue({
loadAgent: loadAgentSpy,
clearCache: vi.fn(),
} as unknown as A2AClientManager);
await registry.testRegisterAgent(remoteAgent);
expect(loadAgentSpy).toHaveBeenCalledTimes(1);
coreEvents.emitModelChanged('new-model');
await new Promise((resolve) => setTimeout(resolve, 0));
expect(loadAgentSpy).toHaveBeenCalledTimes(1);
});
it('should resolve "inherit" to the current model from configuration', async () => {
const config = makeMockedConfig({ model: 'current-model' });
const registry = new TestableAgentRegistry(config);