@google/gemini-cli-core 0.35.0-preview.2 → 0.36.0-nightly.20260321.fc03891a1

package/dist/src/agents/local-invocation.test.js

@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
-import { AgentTerminateMode, } from './types.js';
+import { AgentTerminateMode, SubagentActivityErrorType, SUBAGENT_REJECTED_ERROR_PREFIX, } from './types.js';
 import { LocalSubagentInvocation } from './local-invocation.js';
 import { LocalAgentExecutor } from './local-executor.js';
 import { makeFakeConfig } from '../test-utils/config.js';
@@ -126,8 +126,11 @@ describe('LocalSubagentInvocation', () => {
                     text: expect.stringContaining("Subagent 'MockAgent' finished.\nTermination Reason: GOAL\nResult:\nAnalysis complete."),
                 },
             ]);
-            expect(result.returnDisplay).toBe('Analysis complete.');
-            expect(result.returnDisplay).not.toContain('Termination Reason');
+            const display = result.returnDisplay;
+            expect(display.isSubagentProgress).toBe(true);
+            expect(display.state).toBe('completed');
+            expect(display.result).toBe('Analysis complete.');
+            expect(display.terminateReason).toBe(AgentTerminateMode.GOAL);
         });
         it('should show detailed UI for non-goal terminations (e.g., TIMEOUT)', async () => {
             const mockOutput = {
@@ -136,11 +139,13 @@ describe('LocalSubagentInvocation', () => {
             };
             mockExecutorInstance.run.mockResolvedValue(mockOutput);
             const result = await invocation.execute(signal, updateOutput);
-            expect(result.returnDisplay).toContain('### Subagent MockAgent Finished Early');
-            expect(result.returnDisplay).toContain('**Termination Reason:** TIMEOUT');
-            expect(result.returnDisplay).toContain('Partial progress...');
+            const display = result.returnDisplay;
+            expect(display.isSubagentProgress).toBe(true);
+            expect(display.state).toBe('completed');
+            expect(display.result).toBe('Partial progress...');
+            expect(display.terminateReason).toBe(AgentTerminateMode.TIMEOUT);
         });
-        it('should stream THOUGHT_CHUNK activities from the executor', async () => {
+        it('should stream THOUGHT_CHUNK activities from the executor, replacing the last running thought', async () => {
             mockExecutorInstance.run.mockImplementation(async () => {
                 const onActivity = MockLocalAgentExecutor.create.mock.calls[0][2];
                 if (onActivity) {
@@ -154,17 +159,48 @@ describe('LocalSubagentInvocation', () => {
                         isSubagentActivityEvent: true,
                         agentName: 'MockAgent',
                         type: 'THOUGHT_CHUNK',
-                        data: { text: ' Still thinking.' },
+                        data: { text: 'Thinking about next steps.' },
+                    });
+                }
+                return { result: 'Done', terminate_reason: AgentTerminateMode.GOAL };
+            });
+            await invocation.execute(signal, updateOutput);
+            expect(updateOutput).toHaveBeenCalledTimes(4); // Initial + 2 updates + Final completion
+            const lastCall = updateOutput.mock.calls[3][0];
+            expect(lastCall.recentActivity).toContainEqual(expect.objectContaining({
+                type: 'thought',
+                content: 'Thinking about next steps.',
+            }));
+            expect(lastCall.recentActivity).not.toContainEqual(expect.objectContaining({
+                type: 'thought',
+                content: 'Analyzing...',
+            }));
+        });
+        it('should overwrite the thought content with new THOUGHT_CHUNK activity', async () => {
+            mockExecutorInstance.run.mockImplementation(async () => {
+                const onActivity = MockLocalAgentExecutor.create.mock.calls[0][2];
+                if (onActivity) {
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'MockAgent',
+                        type: 'THOUGHT_CHUNK',
+                        data: { text: 'I am thinking.' },
+                    });
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'MockAgent',
+                        type: 'THOUGHT_CHUNK',
+                        data: { text: 'Now I will act.' },
                     });
                 }
                 return { result: 'Done', terminate_reason: AgentTerminateMode.GOAL };
             });
             await invocation.execute(signal, updateOutput);
-            expect(updateOutput).toHaveBeenCalledTimes(3); // Initial + 2 updates
-            const lastCall = updateOutput.mock.calls[2][0];
+            const calls = updateOutput.mock.calls;
+            const lastCall = calls[calls.length - 1][0];
             expect(lastCall.recentActivity).toContainEqual(expect.objectContaining({
                 type: 'thought',
-                content: 'Analyzing... Still thinking.',
+                content: 'Now I will act.',
             }));
         });
         it('should stream other activities (e.g., TOOL_CALL_START, ERROR)', async () => {
@@ -187,14 +223,50 @@ describe('LocalSubagentInvocation', () => {
                 return { result: 'Done', terminate_reason: AgentTerminateMode.GOAL };
             });
             await invocation.execute(signal, updateOutput);
-            expect(updateOutput).toHaveBeenCalledTimes(3);
-            const lastCall = updateOutput.mock.calls[2][0];
+            expect(updateOutput).toHaveBeenCalledTimes(4); // Initial + 2 updates + Final completion
+            const lastCall = updateOutput.mock.calls[3][0];
             expect(lastCall.recentActivity).toContainEqual(expect.objectContaining({
                 type: 'thought',
                 content: 'Error: Failed',
                 status: 'error',
             }));
         });
+        it('should reflect tool rejections in the activity stream as cancelled but not abort the agent', async () => {
+            mockExecutorInstance.run.mockImplementation(async () => {
+                const onActivity = MockLocalAgentExecutor.create.mock.calls[0][2];
+                if (onActivity) {
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'MockAgent',
+                        type: 'TOOL_CALL_START',
+                        data: { name: 'ls', args: {}, callId: 'call1' },
+                    });
+                    onActivity({
+                        isSubagentActivityEvent: true,
+                        agentName: 'MockAgent',
+                        type: 'ERROR',
+                        data: {
+                            name: 'ls',
+                            callId: 'call1',
+                            error: `${SUBAGENT_REJECTED_ERROR_PREFIX} Please acknowledge this, rethink your strategy, and try a different approach. If you cannot proceed without the rejected operation, summarize the issue and use \`complete_task\` to report your findings and the blocker.`,
+                            errorType: SubagentActivityErrorType.REJECTED,
+                        },
+                    });
+                }
+                return {
+                    result: 'Rethinking...',
+                    terminate_reason: AgentTerminateMode.GOAL,
+                };
+            });
+            await invocation.execute(signal, updateOutput);
+            expect(updateOutput).toHaveBeenCalledTimes(4);
+            const lastCall = updateOutput.mock.calls[3][0];
+            expect(lastCall.recentActivity).toContainEqual(expect.objectContaining({
+                type: 'tool_call',
+                content: 'ls',
+                status: 'cancelled',
+            }));
+        });
         it('should run successfully without an updateOutput callback', async () => {
             mockExecutorInstance.run.mockImplementation(async () => {
                 const onActivity = MockLocalAgentExecutor.create.mock.calls[0][2];
@@ -212,7 +284,10 @@ describe('LocalSubagentInvocation', () => {
             // Execute without the optional callback
             const result = await invocation.execute(signal);
             expect(result.error).toBeUndefined();
-            expect(result.returnDisplay).toBe('Done');
+            const display = result.returnDisplay;
+            expect(display.isSubagentProgress).toBe(true);
+            expect(display.state).toBe('completed');
+            expect(display.result).toBe('Done');
         });
         it('should handle executor run failure', async () => {
             const error = new Error('Model failed during execution.');

package/dist/src/agents/local-invocation.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"local-invocation.test.js","sourceRoot":"","sources":["../../../src/agents/local-invocation.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EACL,kBAAkB,GAKnB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAGzD,OAAO,EAAU,MAAM,KAAK,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAEzE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,MAAM,sBAAsB,GAAG,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;AAE7D,IAAI,UAAkB,CAAC;AAEvB,MAAM,cAAc,GAAuC;IACzD,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,WAAW;IACjB,WAAW,EAAE,YAAY;IACzB,WAAW,EAAE,eAAe;IAC5B,WAAW,EAAE;QACX,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE;gBAC7C,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE;aAClD;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB;KACF;IACD,WAAW,EAAE;QACX,KAAK,EAAE,MAAM;QACb,qBAAqB,EAAE;YACrB,WAAW,EAAE,CAAC;YACd,IAAI,EAAE,CAAC;SACR;KACF;IACD,SAAS,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE;IAChC,YAAY,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE;CACvC,CAAC;AAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,oBAA8D,CAAC;IACnE,IAAI,cAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,UAAU,GAAG,cAAc,EAAE,CAAC;QAC9B,kEAAkE;QAClE,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,UAAU,EAAE;YAC5C,GAAG,EAAE,GAAG,EAAE,CAAC,gBAAgB;YAC3B,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,cAAc,GAAG,oBAAoB,EAAE,CAAC;QAExC,oBAAoB,GAAG;YACrB,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;YACZ,UAAU,EAAE,cAAc;SAC4B,CAAC;QAEzD,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAC7C,oBAAmE,CACpE,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QACxC,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,cAAc,EACd,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;QAEF,6DAA6D;QAC7D,8DAA8D;QAC9D,MAAM,CAAE,UAAkB,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;YACrD,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,cAAc,EACd,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CACtB,+EAA+E,CAChF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjC,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAClC,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,cAAc,EACd,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,yCAAyC;YACzC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CACtB,qDAAqD,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CACxE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,wEAAwE;YACxE,MAAM,WAAW,GAAyB;gBACxC,GAAG,cAAc;gBACjB,IAAI,EAAE,mCAAmC;aAC1C,CAAC;YACF,MAAM,MAAM,GAAgB,EAAE,CAAC;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC;YACpC,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,WAAW,EACX,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,wCAAwC;YACxC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrC,MAAM,CACJ,WAAW,CAAC,UAAU,CACpB,sDAAsD,CACvD,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,IAAI,MAAmB,CAAC;QACxB,IAAI,YAAsC,CAAC;QAC3C,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QACxC,IAAI,UAAmC,CAAC;QAExC,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;YACtC,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACvB,UAAU,GAAG,IAAI,uBAAuB,CACtC,cAAc,EACd,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,UAAU,GAAG;gBACjB,MAAM,EAAE,oBAAoB;gBAC5B,gBAAgB,EAAE,kBAAkB,CAAC,IAAI;aAC1C,CAAC;YACF,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACxD,cAAc,EACd,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,kBAAkB,EAAE,IAAI;gBACxB,SAAS,EAAE,WAAW;aACvB,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;gBAChC;oBACE,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAC3B,uFAAuF,CACxF;iBACF;aACF,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,UAAU,GAAG;gBACjB,MAAM,EAAE,qBAAqB;gBAC7B,gBAAgB,EAAE,kBAAkB,CAAC,OAAO;aAC7C,CAAC;YACF,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CACpC,uCAAuC,CACxC,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,iCAAiC,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAElE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE;qBACN,CAAC,CAAC;oBAC5B,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE;qBACV,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE/C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB;YACrE,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAqB,CAAC;YACnE,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,cAAc,CAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,8BAA8B;aACxC,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAElE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,iBAAiB;wBACvB,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE;qBACN,CAAC,CAAC;oBAC5B,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;qBACD,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE/C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAqB,CAAC;YACnE,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,cAAc,CAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,eAAe;gBACxB,MAAM,EAAE,OAAO;aAChB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClE,IAAI,UAAU,EAAE,CAAC;oBACf,uEAAuE;oBACvE,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE;qBACZ,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,wCAAwC;YACxC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;YAC1D,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAC5B,uCAAuC,KAAK,CAAC,OAAO,EAAE,CACvD,CAAC;YACF,MAAM,OAAO,GAAG,MAAM,CAAC,aAAiC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,cAAc,CAC3C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,UAAU,KAAK,CAAC,OAAO,EAAE;gBAClC,MAAM,EAAE,OAAO;aAChB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YAC/D,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAE/D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAE3D,MAAM,OAAO,GAAG,MAAM,CAAC,aAAiC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,cAAc,CAC3C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,UAAU,aAAa,CAAC,OAAO,EAAE;gBAC1C,MAAM,EAAE,OAAO;aAChB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,UAAU,GAAG,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;YACxC,UAAU,CAAC,IAAI,GAAG,YAAY,CAAC;YAC/B,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,cAAc,GAAG,UAAU,CAAC,OAAO,CACvC,UAAU,CAAC,MAAM,EACjB,YAAY,CACb,CAAC;YACF,UAAU,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAExD,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACnD,MAAM,EACN,UAAU,CAAC,MAAM,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,MAAM,UAAU,GAAG;gBACjB,MAAM,EAAE,mBAAmB;gBAC3B,gBAAgB,EAAE,kBAAkB,CAAC,OAAO;aAC7C,CAAC;YACF,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACpE,6BAA6B,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"local-invocation.test.js","sourceRoot":"","sources":["../../../src/agents/local-invocation.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EACL,kBAAkB,EAKlB,yBAAyB,EACzB,8BAA8B,GAC/B,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAGzD,OAAO,EAAU,MAAM,KAAK,CAAC;AAC7B,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAEzE,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,MAAM,sBAAsB,GAAG,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;AAE7D,IAAI,UAAkB,CAAC;AAEvB,MAAM,cAAc,GAAuC;IACzD,IAAI,EAAE,OAAO;IACb,IAAI,EAAE,WAAW;IACjB,WAAW,EAAE,YAAY;IACzB,WAAW,EAAE,eAAe;IAC5B,WAAW,EAAE;QACX,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE;gBAC7C,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,EAAE;aAClD;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB;KACF;IACD,WAAW,EAAE;QACX,KAAK,EAAE,MAAM;QACb,qBAAqB,EAAE;YACrB,WAAW,EAAE,CAAC;YACd,IAAI,EAAE,CAAC;SACR;KACF;IACD,SAAS,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE;IAChC,YAAY,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE;CACvC,CAAC;AAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,oBAA8D,CAAC;IACnE,IAAI,cAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,UAAU,GAAG,cAAc,EAAE,CAAC;QAC9B,kEAAkE;QAClE,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,UAAU,EAAE;YAC5C,GAAG,EAAE,GAAG,EAAE,CAAC,gBAAgB;YAC3B,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,cAAc,GAAG,oBAAoB,EAAE,CAAC;QAExC,oBAAoB,GAAG;YACrB,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;YACZ,UAAU,EAAE,cAAc;SAC4B,CAAC;QAEzD,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAC7C,oBAAmE,CACpE,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QACxC,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,cAAc,EACd,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;QAEF,6DAA6D;QAC7D,8DAA8D;QAC9D,MAAM,CAAE,UAAkB,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;YACrD,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,cAAc,EACd,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CACtB,+EAA+E,CAChF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjC,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAClC,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,cAAc,EACd,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,yCAAyC;YACzC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CACtB,qDAAqD,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CACxE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,wEAAwE;YACxE,MAAM,WAAW,GAAyB;gBACxC,GAAG,cAAc;gBACjB,IAAI,EAAE,mCAAmC;aAC1C,CAAC;YACF,MAAM,MAAM,GAAgB,EAAE,CAAC;YAC/B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC;YACpC,CAAC;YACD,MAAM,UAAU,GAAG,IAAI,uBAAuB,CAC5C,WAAW,EACX,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;YACF,MAAM,WAAW,GAAG,UAAU,CAAC,cAAc,EAAE,CAAC;YAChD,wCAAwC;YACxC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrC,MAAM,CACJ,WAAW,CAAC,UAAU,CACpB,sDAAsD,CACvD,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,IAAI,MAAmB,CAAC;QACxB,IAAI,YAAsC,CAAC;QAC3C,MAAM,MAAM,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QACxC,IAAI,UAAmC,CAAC;QAExC,UAAU,CAAC,GAAG,EAAE;YACd,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;YACtC,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACvB,UAAU,GAAG,IAAI,uBAAuB,CACtC,cAAc,EACd,UAAU,EACV,MAAM,EACN,cAAc,CACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,UAAU,GAAG;gBACjB,MAAM,EAAE,oBAAoB;gBAC5B,gBAAgB,EAAE,kBAAkB,CAAC,IAAI;aAC1C,CAAC;YACF,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACxD,cAAc,EACd,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;YACF,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,MAAM,CAAC,gBAAgB,CAAC;gBACtB,kBAAkB,EAAE,IAAI;gBACxB,SAAS,EAAE,WAAW;aACvB,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;gBAChC;oBACE,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAC3B,uFAAuF,CACxF;iBACF;aACF,CAAC,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,CAAC,aAAiC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YAClD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,UAAU,GAAG;gBACjB,MAAM,EAAE,qBAAqB;gBAC7B,gBAAgB,EAAE,kBAAkB,CAAC,OAAO;aAC7C,CAAC;YACF,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,OAAO,GAAG,MAAM,CAAC,aAAiC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACnD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8FAA8F,EAAE,KAAK,IAAI,EAAE;YAC5G,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAElE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE;qBACN,CAAC,CAAC;oBAC5B,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,4BAA4B,EAAE;qBACpB,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE/C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC;YACxF,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAqB,CAAC;YACnE,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,cAAc,CAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,4BAA4B;aACtC,CAAC,CACH,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,cAAc,CAChD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,cAAc;aACxB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAElE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE;qBACR,CAAC,CAAC;oBAC5B,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE;qBACT,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE/C,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC;YACtC,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAqB,CAAC;YAChE,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,cAAc,CAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,iBAAiB;aAC3B,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAElE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,iBAAiB;wBACvB,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE;qBACN,CAAC,CAAC;oBAC5B,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE;qBACD,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE/C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,yCAAyC;YACxF,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAqB,CAAC;YACnE,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,cAAc,CAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,eAAe;gBACxB,MAAM,EAAE,OAAO;aAChB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4FAA4F,EAAE,KAAK,IAAI,EAAE;YAC1G,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAElE,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,iBAAiB;wBACvB,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;qBACvB,CAAC,CAAC;oBAC5B,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE;4BACJ,IAAI,EAAE,IAAI;4BACV,MAAM,EAAE,OAAO;4BACf,KAAK,EAAE,GAAG,8BAA8B,6NAA6N;4BACrQ,SAAS,EAAE,yBAAyB,CAAC,QAAQ;yBAC9C;qBACuB,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO;oBACL,MAAM,EAAE,eAAe;oBACvB,gBAAgB,EAAE,kBAAkB,CAAC,IAAI;iBAC1C,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE/C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAqB,CAAC;YACnE,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,cAAc,CAC5C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,WAAW;aACpB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,oBAAoB,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACrD,MAAM,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClE,IAAI,UAAU,EAAE,CAAC;oBACf,uEAAuE;oBACvE,UAAU,CAAC;wBACT,uBAAuB,EAAE,IAAI;wBAC7B,SAAS,EAAE,WAAW;wBACtB,IAAI,EAAE,eAAe;wBACrB,IAAI,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE;qBACZ,CAAC,CAAC;gBAC9B,CAAC;gBACD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,wCAAwC;YACxC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACrC,MAAM,OAAO,GAAG,MAAM,CAAC,aAAiC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;YAC1D,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAC5B,uCAAuC,KAAK,CAAC,OAAO,EAAE,CACvD,CAAC;YACF,MAAM,OAAO,GAAG,MAAM,CAAC,aAAiC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,cAAc,CAC3C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,UAAU,KAAK,CAAC,OAAO,EAAE;gBAClC,MAAM,EAAE,OAAO;aAChB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,aAAa,GAAG,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YAC/D,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAE/D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAE9D,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YAE3D,MAAM,OAAO,GAAG,MAAM,CAAC,aAAiC,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,cAAc,CAC3C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,UAAU,aAAa,CAAC,OAAO,EAAE;gBAC1C,MAAM,EAAE,OAAO;aAChB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,UAAU,GAAG,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;YACxC,UAAU,CAAC,IAAI,GAAG,YAAY,CAAC;YAC/B,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,cAAc,GAAG,UAAU,CAAC,OAAO,CACvC,UAAU,CAAC,MAAM,EACjB,YAAY,CACb,CAAC;YACF,UAAU,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAExD,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACnD,MAAM,EACN,UAAU,CAAC,MAAM,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,MAAM,UAAU,GAAG;gBACjB,MAAM,EAAE,mBAAmB;gBAC3B,gBAAgB,EAAE,kBAAkB,CAAC,OAAO;aAC7C,CAAC;YACF,oBAAoB,CAAC,GAAG,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YAEvD,MAAM,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACpE,6BAA6B,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/agents/memory-manager-agent.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { z } from 'zod';
+import type { LocalAgentDefinition } from './types.js';
+import type { Config } from '../config/config.js';
+declare const MemoryManagerSchema: z.ZodObject<{
+    response: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    response: string;
+}, {
+    response: string;
+}>;
+/**
+ * A memory management agent that replaces the built-in save_memory tool.
+ * It provides richer memory operations: adding, removing, de-duplicating,
+ * and organizing memories in the global GEMINI.md file.
+ *
+ * Users can override this agent by placing a custom save_memory.md
+ * in ~/.gemini/agents/ or .gemini/agents/.
+ */
+export declare const MemoryManagerAgent: (config: Config) => LocalAgentDefinition<typeof MemoryManagerSchema>;
+export {};

package/dist/src/agents/memory-manager-agent.js

@@ -0,0 +1,137 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { z } from 'zod';
+import { ASK_USER_TOOL_NAME, EDIT_TOOL_NAME, GLOB_TOOL_NAME, GREP_TOOL_NAME, LS_TOOL_NAME, READ_FILE_TOOL_NAME, WRITE_FILE_TOOL_NAME, } from '../tools/tool-names.js';
+import { Storage } from '../config/storage.js';
+import { flattenMemory } from '../config/memory.js';
+import { GEMINI_MODEL_ALIAS_FLASH } from '../config/models.js';
+const MemoryManagerSchema = z.object({
+    response: z
+        .string()
+        .describe('A summary of the memory operations performed.'),
+});
+/**
+ * A memory management agent that replaces the built-in save_memory tool.
+ * It provides richer memory operations: adding, removing, de-duplicating,
+ * and organizing memories in the global GEMINI.md file.
+ *
+ * Users can override this agent by placing a custom save_memory.md
+ * in ~/.gemini/agents/ or .gemini/agents/.
+ */
+export const MemoryManagerAgent = (config) => {
+    const globalGeminiDir = Storage.getGlobalGeminiDir();
+    const getInitialContext = () => {
+        const memory = config.getUserMemory();
+        // Only include global and project memory — extension memory is read-only
+        // and not relevant to the memory manager.
+        const content = typeof memory === 'string'
+            ? memory
+            : flattenMemory({ global: memory.global, project: memory.project });
+        if (!content.trim())
+            return '';
+        return `\n# Initial Context\n\n${content}\n`;
+    };
+    const buildSystemPrompt = () => `
+You are a memory management agent maintaining user memories in GEMINI.md files.
+
+# Memory Hierarchy
+
+## Global (${globalGeminiDir})
+- \`${globalGeminiDir}/GEMINI.md\` — Cross-project user preferences, key personal info,
+  and habits that apply everywhere.
+
+## Project (./)
+- \`./GEMINI.md\` — **Table of Contents** for project-specific context:
+  architecture decisions, conventions, key contacts, and references to
+  subdirectory GEMINI.md files for detailed context.
+- Subdirectory GEMINI.md files (e.g. \`src/GEMINI.md\`, \`docs/GEMINI.md\`) —
+  detailed, domain-specific context for that part of the project. Reference
+  these from the root \`./GEMINI.md\`.
+
+## Routing
+
+When adding a memory, route it to the right store:
+- **Global**: User preferences, personal info, tool aliases, cross-project habits → **global**
+- **Project Root**: Project architecture, conventions, workflows, team info → **project root**
+- **Subdirectory**: Detailed context about a specific module or directory → **subdirectory
+  GEMINI.md**, with a reference added to the project root
+
+- **Ambiguity**: If a memory (like a coding preference or workflow) could be interpreted as either a global habit or a project-specific convention, you **MUST** use \`${ASK_USER_TOOL_NAME}\` to clarify the user's intent. Do NOT make a unilateral decision when ambiguity exists between Global and Project stores.
+
+# Operations
+
+1. **Adding** — Route to the correct store and file. Check for duplicates in your provided context first.
+2. **Removing stale entries** — Delete outdated or unwanted entries. Clean up
+   dangling references.
+3. **De-duplicating** — Semantically equivalent entries should be combined. Keep the most informative version.
+4. **Organizing** — Restructure for clarity. Update references between files.
+
+# Restrictions
+- Keep GEMINI.md files lean — they are loaded into context every session.
+- Keep entries concise.
+- Edit surgically — preserve existing structure and user-authored content.
+- NEVER write or read any files other than GEMINI.md files.
+
+# Efficiency & Performance
+- **Use as few turns as possible.** Execute independent reads and writes to different files in parallel by calling multiple tools in a single turn.
+- **Do not perform any exploration of the codebase.** Try to use the provided file context and only search additional GEMINI.md files as needed to accomplish your task.
+- **Be strategic with your thinking.** carefully decide where to route memories and how to de-duplicate memories, but be decisive with simple memory writes.
+- **Minimize file system operations.** You should typically only modify the GEMINI.md files that are already provided in your context. Only read or write to other files if explicitly directed or if you are following a specific reference from an existing memory file.
+- **Context Awareness.** If a file's content is already provided in the "Initial Context" section, you do not need to call \`read_file\` for it.
+
+# Insufficient context
+If you find that you have insufficient context to read or modify the memories as described,
+reply with what you need, and exit. Do not search the codebase for the missing context.
+`.trim();
+    return {
+        kind: 'local',
+        name: 'save_memory',
+        displayName: 'Memory Manager',
+        description: `Writes and reads memory, preferences or facts across ALL future sessions. Use this for recurring instructions like coding styles or tool aliases.`,
+        inputConfig: {
+            inputSchema: {
+                type: 'object',
+                properties: {
+                    request: {
+                        type: 'string',
+                        description: 'The memory operation to perform. Examples: "Remember that I prefer tabs over spaces", "Clean up stale memories", "De-duplicate my memories", "Organize my memories".',
+                    },
+                },
+                required: ['request'],
+            },
+        },
+        outputConfig: {
+            outputName: 'result',
+            description: 'A summary of the memory operations performed.',
+            schema: MemoryManagerSchema,
+        },
+        modelConfig: {
+            model: GEMINI_MODEL_ALIAS_FLASH,
+        },
+        toolConfig: {
+            tools: [
+                READ_FILE_TOOL_NAME,
+                EDIT_TOOL_NAME,
+                WRITE_FILE_TOOL_NAME,
+                LS_TOOL_NAME,
+                GLOB_TOOL_NAME,
+                GREP_TOOL_NAME,
+                ASK_USER_TOOL_NAME,
+            ],
+        },
+        get promptConfig() {
+            return {
+                systemPrompt: buildSystemPrompt(),
+                query: `${getInitialContext()}\${request}`,
+            };
+        },
+        runConfig: {
+            maxTimeMinutes: 5,
+            maxTurns: 10,
+        },
+    };
+};
+//# sourceMappingURL=memory-manager-agent.js.map

package/dist/src/agents/memory-manager-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-manager-agent.js","sourceRoot":"","sources":["../../../src/agents/memory-manager-agent.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EACL,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,cAAc,EACd,YAAY,EACZ,mBAAmB,EACnB,oBAAoB,GACrB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAG/D,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,QAAQ,EAAE,CAAC;SACR,MAAM,EAAE;SACR,QAAQ,CAAC,+CAA+C,CAAC;CAC7D,CAAC,CAAC;AAEH;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,MAAc,EACoC,EAAE;IACpD,MAAM,eAAe,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAErD,MAAM,iBAAiB,GAAG,GAAW,EAAE;QACrC,MAAM,MAAM,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;QACtC,yEAAyE;QACzE,0CAA0C;QAC1C,MAAM,OAAO,GACX,OAAO,MAAM,KAAK,QAAQ;YACxB,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QACxE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE;YAAE,OAAO,EAAE,CAAC;QAC/B,OAAO,0BAA0B,OAAO,IAAI,CAAC;IAC/C,CAAC,CAAC;IAEF,MAAM,iBAAiB,GAAG,GAAW,EAAE,CACrC;;;;;aAKS,eAAe;MACtB,eAAe;;;;;;;;;;;;;;;;;;;yKAmBoJ,kBAAkB;;;;;;;;;;;;;;;;;;;;;;;;;;CA0B1L,CAAC,IAAI,EAAE,CAAC;IAEP,OAAO;QACL,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,gBAAgB;QAC7B,WAAW,EAAE,mJAAmJ;QAChK,WAAW,EAAE;YACX,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,OAAO,EAAE;wBACP,IAAI,EAAE,QAAQ;wBACd,WAAW,EACT,sKAAsK;qBACzK;iBACF;gBACD,QAAQ,EAAE,CAAC,SAAS,CAAC;aACtB;SACF;QACD,YAAY,EAAE;YACZ,UAAU,EAAE,QAAQ;YACpB,WAAW,EAAE,+CAA+C;YAC5D,MAAM,EAAE,mBAAmB;SAC5B;QACD,WAAW,EAAE;YACX,KAAK,EAAE,wBAAwB;SAChC;QACD,UAAU,EAAE;YACV,KAAK,EAAE;gBACL,mBAAmB;gBACnB,cAAc;gBACd,oBAAoB;gBACpB,YAAY;gBACZ,cAAc;gBACd,cAAc;gBACd,kBAAkB;aACnB;SACF;QACD,IAAI,YAAY;YACd,OAAO;gBACL,YAAY,EAAE,iBAAiB,EAAE;gBACjC,KAAK,EAAE,GAAG,iBAAiB,EAAE,aAAa;aAC3C,CAAC;QACJ,CAAC;QACD,SAAS,EAAE;YACT,cAAc,EAAE,CAAC;YACjB,QAAQ,EAAE,EAAE;SACb;KACF,CAAC;AACJ,CAAC,CAAC"}

package/dist/src/agents/memory-manager-agent.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/agents/memory-manager-agent.test.js

@@ -0,0 +1,117 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { MemoryManagerAgent } from './memory-manager-agent.js';
+import { ASK_USER_TOOL_NAME, EDIT_TOOL_NAME, GLOB_TOOL_NAME, GREP_TOOL_NAME, LS_TOOL_NAME, READ_FILE_TOOL_NAME, WRITE_FILE_TOOL_NAME, } from '../tools/tool-names.js';
+import { Storage } from '../config/storage.js';
+function createMockConfig(memory = '') {
+    return {
+        getUserMemory: vi.fn().mockReturnValue(memory),
+    };
+}
+describe('MemoryManagerAgent', () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    it('should have the correct name "save_memory"', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        expect(agent.name).toBe('save_memory');
+    });
+    it('should be a local agent', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        expect(agent.kind).toBe('local');
+    });
+    it('should have a description', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        expect(agent.description).toBeTruthy();
+        expect(agent.description).toContain('memory');
+    });
+    it('should have a system prompt with memory management instructions', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        const prompt = agent.promptConfig.systemPrompt;
+        const globalGeminiDir = Storage.getGlobalGeminiDir();
+        expect(prompt).toContain(`Global (${globalGeminiDir}`);
+        expect(prompt).toContain('Project (./');
+        expect(prompt).toContain('Memory Hierarchy');
+        expect(prompt).toContain('De-duplicating');
+        expect(prompt).toContain('Adding');
+        expect(prompt).toContain('Removing stale entries');
+        expect(prompt).toContain('Organizing');
+        expect(prompt).toContain('Routing');
+    });
+    it('should have efficiency guidelines in the system prompt', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        const prompt = agent.promptConfig.systemPrompt;
+        expect(prompt).toContain('Efficiency & Performance');
+        expect(prompt).toContain('Use as few turns as possible');
+        expect(prompt).toContain('Do not perform any exploration');
+        expect(prompt).toContain('Be strategic with your thinking');
+        expect(prompt).toContain('Context Awareness');
+    });
+    it('should inject hierarchical memory into initial context', () => {
+        const config = createMockConfig({
+            global: '--- Context from: ../../.gemini/GEMINI.md ---\nglobal context\n--- End of Context from: ../../.gemini/GEMINI.md ---',
+            project: '--- Context from: .gemini/GEMINI.md ---\nproject context\n--- End of Context from: .gemini/GEMINI.md ---',
+        });
+        const agent = MemoryManagerAgent(config);
+        const query = agent.promptConfig.query;
+        expect(query).toContain('# Initial Context');
+        expect(query).toContain('global context');
+        expect(query).toContain('project context');
+    });
+    it('should inject flat string memory into initial context', () => {
+        const config = createMockConfig('flat memory content');
+        const agent = MemoryManagerAgent(config);
+        const query = agent.promptConfig.query;
+        expect(query).toContain('# Initial Context');
+        expect(query).toContain('flat memory content');
+    });
+    it('should exclude extension memory from initial context', () => {
+        const config = createMockConfig({
+            global: 'global context',
+            extension: 'extension context that should be excluded',
+            project: 'project context',
+        });
+        const agent = MemoryManagerAgent(config);
+        const query = agent.promptConfig.query;
+        expect(query).toContain('global context');
+        expect(query).toContain('project context');
+        expect(query).not.toContain('extension context');
+    });
+    it('should not include initial context when memory is empty', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        const query = agent.promptConfig.query;
+        expect(query).not.toContain('# Initial Context');
+    });
+    it('should have file-management and search tools', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        expect(agent.toolConfig).toBeDefined();
+        expect(agent.toolConfig.tools).toEqual(expect.arrayContaining([
+            READ_FILE_TOOL_NAME,
+            EDIT_TOOL_NAME,
+            WRITE_FILE_TOOL_NAME,
+            LS_TOOL_NAME,
+            GLOB_TOOL_NAME,
+            GREP_TOOL_NAME,
+            ASK_USER_TOOL_NAME,
+        ]));
+    });
+    it('should require a "request" input parameter', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        const schema = agent.inputConfig.inputSchema;
+        expect(schema).toBeDefined();
+        expect(schema['properties']).toHaveProperty('request');
+        expect(schema['required']).toContain('request');
+    });
+    it('should use a fast model', () => {
+        const agent = MemoryManagerAgent(createMockConfig());
+        expect(agent.modelConfig.model).toBe('flash');
+    });
+});
+//# sourceMappingURL=memory-manager-agent.test.js.map

package/dist/src/agents/memory-manager-agent.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-manager-agent.test.js","sourceRoot":"","sources":["../../../src/agents/memory-manager-agent.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EACL,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,cAAc,EACd,YAAY,EACZ,mBAAmB,EACnB,oBAAoB,GACrB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAI/C,SAAS,gBAAgB,CAAC,SAAsC,EAAE;IAChE,OAAO;QACL,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC;KAC1B,CAAC;AACzB,CAAC;AAED,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC;QAC/C,MAAM,eAAe,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,WAAW,eAAe,EAAE,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iCAAiC,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,MAAM,GAAG,gBAAgB,CAAC;YAC9B,MAAM,EACJ,qHAAqH;YACvH,OAAO,EACL,0GAA0G;SAC7G,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC;QAEvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,MAAM,GAAG,gBAAgB,CAAC,qBAAqB,CAAC,CAAC;QAEvD,MAAM,KAAK,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC;QAEvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,gBAAgB,CAAC;YAC9B,MAAM,EAAE,gBAAgB;YACxB,SAAS,EAAE,2CAA2C;YACtD,OAAO,EAAE,iBAAiB;SAC3B,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC;QAEvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,KAAK,GAAG,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC;QAEvC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,UAAW,CAAC,KAAK,CAAC,CAAC,OAAO,CACrC,MAAM,CAAC,eAAe,CAAC;YACrB,mBAAmB;YACnB,cAAc;YACd,oBAAoB;YACpB,YAAY;YACZ,cAAc;YACd,cAAc;YACd,kBAAkB;SACnB,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,KAAK,CAAC,WAAW,CAAC,WAAsC,CAAC;QACxE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,KAAK,GAAG,kBAAkB,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/agents/registry.js

@@ -10,7 +10,7 @@ import { CodebaseInvestigatorAgent } from './codebase-investigator.js';
 import { CliHelpAgent } from './cli-help-agent.js';
 import { GeneralistAgent } from './generalist-agent.js';
 import { BrowserAgentDefinition } from './browser/browserAgentDefinition.js';
-import { A2AClientManager } from './a2a-client-manager.js';
+import { MemoryManagerAgent } from './memory-manager-agent.js';
 import { A2AAuthProviderFactory } from './auth-provider/factory.js';
 import {} from 'zod';
 import { debugLogger } from '../utils/debugLogger.js';
@@ -53,7 +53,7 @@ export class AgentRegistry {
      * Clears the current registry and re-scans for agents.
      */
     async reload() {
-        A2AClientManager.getInstance(this.config).clearCache();
+        this.config.getA2AClientManager()?.clearCache();
         await this.config.reloadAgents();
         this.agents.clear();
         this.allDefinitions.clear();
@@ -179,6 +179,20 @@ export class AgentRegistry {
         if (browserConfig.enabled) {
             this.registerLocalAgent(BrowserAgentDefinition(this.config));
         }
+        // Register the memory manager agent as a replacement for the save_memory tool.
+        if (this.config.isMemoryManagerEnabled()) {
+            this.registerLocalAgent(MemoryManagerAgent(this.config));
+            // Ensure the global .gemini directory is accessible to tools.
+            // This allows the save_memory agent to read and write to it.
+            // Access control is enforced by the Policy Engine (memory-manager.toml).
+            try {
+                const globalDir = Storage.getGlobalGeminiDir();
+                this.config.getWorkspaceContext().addDirectory(globalDir);
+            }
+            catch (e) {
+                debugLogger.warn(`[AgentRegistry] Could not add global .gemini directory to workspace:`, e);
+            }
+        }
     }
     async refreshAgents() {
         this.loadBuiltInAgents();
@@ -292,7 +306,11 @@ export class AgentRegistry {
         }
         // Load the remote A2A agent card and register.
         try {
-            const clientManager = A2AClientManager.getInstance(this.config);
+            const clientManager = this.config.getA2AClientManager();
+            if (!clientManager) {
+                debugLogger.warn(`[AgentRegistry] Skipping remote agent '${definition.name}': A2AClientManager is not available.`);
+                return;
+            }
             let authHandler;
             if (definition.auth) {
                 const provider = await A2AAuthProviderFactory.create({