MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-05-13 05:12:55 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat(policy): Propagate Tool Annotations for MCP Servers (#20083)

Browse Source
This commit is contained in:
Jerop Kipruto
2026-02-24 09:20:11 -05:00
committed by GitHub
parent ee2e947435
commit 15f6c8b8da
19 changed files with 455 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/core/src/scheduler/policy.test.ts
+4 -1
View File
@@ -59,10 +59,11 @@ describe('policy.ts', () => {
expect(mockPolicyEngine.check).toHaveBeenCalledWith(
{ name: 'test-tool', args: {} },
undefined,
undefined,
);
});
it('should pass serverName for MCP tools', async () => {
it('should pass serverName and toolAnnotations for MCP tools', async () => {
const mockPolicyEngine = {
check: vi.fn().mockResolvedValue({ decision: PolicyDecision.ALLOW }),
} as unknown as Mocked<PolicyEngine>;
@@ -73,6 +74,7 @@ describe('policy.ts', () => {
const mcpTool = Object.create(DiscoveredMCPTool.prototype);
mcpTool.serverName = 'my-server';
mcpTool._toolAnnotations = { readOnlyHint: true };
const toolCall = {
request: { name: 'mcp-tool', args: {} },
@@ -83,6 +85,7 @@ describe('policy.ts', () => {
expect(mockPolicyEngine.check).toHaveBeenCalledWith(
{ name: 'mcp-tool', args: {} },
'my-server',
{ readOnlyHint: true },
);
});