feat(core): Fully migrate packages/core to AgentLoopContext. (#22115)

This commit is contained in:
joshualitt
2026-03-12 18:56:31 -07:00
committed by GitHub
parent 1d2585dba6
commit de656f01d7
53 changed files with 522 additions and 292 deletions

View File

@@ -98,6 +98,10 @@ describe('SimpleExtensionLoader', () => {
mockConfig = {
getMcpClientManager: () => mockMcpClientManager,
getEnableExtensionReloading: () => extensionReloadingEnabled,
geminiClient: {
isInitialized: () => true,
setTools: mockGeminiClientSetTools,
},
getGeminiClient: vi.fn(() => ({
isInitialized: () => true,
setTools: mockGeminiClientSetTools,

View File

@@ -140,7 +140,7 @@ export abstract class ExtensionLoader {
extension: GeminiCLIExtension,
): Promise<void> {
if (extension.excludeTools && extension.excludeTools.length > 0) {
const geminiClient = this.config?.getGeminiClient();
const geminiClient = this.config?.geminiClient;
if (geminiClient?.isInitialized()) {
await geminiClient.setTools();
}

View File

@@ -71,6 +71,10 @@ describe('checkNextSpeaker', () => {
generateContentConfig: {},
};
mockConfig = {
get config() {
return this;
},
promptId: 'test-session-id',
getProjectRoot: vi.fn().mockReturnValue('/test/project/root'),
getSessionId: vi.fn().mockReturnValue('test-session-id'),
getModel: () => 'test-model',