@office-ai/aioncli-core 0.26.0 → 0.30.0

package/dist/src/agents/agentLoader.test.js

@@ -0,0 +1,458 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import * as fs from 'node:fs/promises';
+import * as path from 'node:path';
+import * as os from 'node:os';
+import { parseAgentMarkdown, markdownToAgentDefinition, loadAgentsFromDirectory, AgentLoadError, } from './agentLoader.js';
+import { GEMINI_MODEL_ALIAS_PRO } from '../config/models.js';
+import { DEFAULT_MAX_TIME_MINUTES, DEFAULT_MAX_TURNS } from './types.js';
+describe('loader', () => {
+    let tempDir;
+    beforeEach(async () => {
+        tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'agent-test-'));
+    });
+    afterEach(async () => {
+        if (tempDir) {
+            await fs.rm(tempDir, { recursive: true, force: true });
+        }
+    });
+    async function writeAgentMarkdown(content, fileName = 'test.md') {
+        const filePath = path.join(tempDir, fileName);
+        await fs.writeFile(filePath, content);
+        return filePath;
+    }
+    describe('parseAgentMarkdown', () => {
+        it('should parse a valid markdown agent file', async () => {
+            const filePath = await writeAgentMarkdown(`---
+name: test-agent-md
+description: A markdown agent
+---
+You are a markdown agent.`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toMatchObject({
+                name: 'test-agent-md',
+                description: 'A markdown agent',
+                kind: 'local',
+                system_prompt: 'You are a markdown agent.',
+            });
+        });
+        it('should parse frontmatter with tools and model config', async () => {
+            const filePath = await writeAgentMarkdown(`---
+name: complex-agent
+description: A complex markdown agent
+tools:
+  - run_shell_command
+model: gemini-pro
+temperature: 0.7
+---
+System prompt content.`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toMatchObject({
+                name: 'complex-agent',
+                description: 'A complex markdown agent',
+                tools: ['run_shell_command'],
+                model: 'gemini-pro',
+                temperature: 0.7,
+                system_prompt: 'System prompt content.',
+            });
+        });
+        it('should throw AgentLoadError if frontmatter is missing', async () => {
+            const filePath = await writeAgentMarkdown(`Just some markdown content.`);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow(AgentLoadError);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow('Missing mandatory YAML frontmatter');
+        });
+        it('should throw AgentLoadError if frontmatter is invalid YAML', async () => {
+            const filePath = await writeAgentMarkdown(`---
+name: [invalid yaml
+---
+Body`);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow(AgentLoadError);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow('YAML frontmatter parsing failed');
+        });
+        it('should throw AgentLoadError if validation fails (missing required field)', async () => {
+            const filePath = await writeAgentMarkdown(`---
+name: test-agent
+# missing description
+---
+Body`);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow(/Validation failed/);
+        });
+        it('should parse a valid remote agent markdown file', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: remote-agent
+description: A remote agent
+agent_card_url: https://example.com/card
+---
+`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toEqual({
+                kind: 'remote',
+                name: 'remote-agent',
+                description: 'A remote agent',
+                agent_card_url: 'https://example.com/card',
+            });
+        });
+        it('should infer remote agent kind from agent_card_url', async () => {
+            const filePath = await writeAgentMarkdown(`---
+name: inferred-remote
+description: Inferred
+agent_card_url: https://example.com/inferred
+---
+`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toEqual({
+                kind: 'remote',
+                name: 'inferred-remote',
+                description: 'Inferred',
+                agent_card_url: 'https://example.com/inferred',
+            });
+        });
+        it('should parse a remote agent with no body', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: no-body-remote
+agent_card_url: https://example.com/card
+---
+`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toEqual({
+                kind: 'remote',
+                name: 'no-body-remote',
+                agent_card_url: 'https://example.com/card',
+            });
+        });
+        it('should parse multiple remote agents in a list', async () => {
+            const filePath = await writeAgentMarkdown(`---
+- kind: remote
+  name: remote-1
+  agent_card_url: https://example.com/1
+- kind: remote
+  name: remote-2
+  agent_card_url: https://example.com/2
+---
+`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(2);
+            expect(result[0]).toEqual({
+                kind: 'remote',
+                name: 'remote-1',
+                agent_card_url: 'https://example.com/1',
+            });
+            expect(result[1]).toEqual({
+                kind: 'remote',
+                name: 'remote-2',
+                agent_card_url: 'https://example.com/2',
+            });
+        });
+        it('should parse frontmatter without a trailing newline', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: no-trailing-newline
+agent_card_url: https://example.com/card
+---`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toEqual({
+                kind: 'remote',
+                name: 'no-trailing-newline',
+                agent_card_url: 'https://example.com/card',
+            });
+        });
+        it('should throw AgentLoadError if agent name is not a valid slug', async () => {
+            const filePath = await writeAgentMarkdown(`---
+name: Invalid Name With Spaces
+description: Test
+---
+Body`);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow(/Name must be a valid slug/);
+        });
+    });
+    describe('markdownToAgentDefinition', () => {
+        it('should convert valid Markdown DTO to AgentDefinition with defaults', () => {
+            const markdown = {
+                kind: 'local',
+                name: 'test-agent',
+                description: 'A test agent',
+                system_prompt: 'You are a test agent.',
+            };
+            const result = markdownToAgentDefinition(markdown);
+            expect(result).toMatchObject({
+                name: 'test-agent',
+                description: 'A test agent',
+                promptConfig: {
+                    systemPrompt: 'You are a test agent.',
+                    query: '${query}',
+                },
+                modelConfig: {
+                    model: 'inherit',
+                    generateContentConfig: {
+                        topP: 0.95,
+                    },
+                },
+                runConfig: {
+                    maxTimeMinutes: DEFAULT_MAX_TIME_MINUTES,
+                    maxTurns: DEFAULT_MAX_TURNS,
+                },
+                inputConfig: {
+                    inputSchema: {
+                        type: 'object',
+                        properties: {
+                            query: {
+                                type: 'string',
+                                description: 'The task for the agent.',
+                            },
+                        },
+                        required: [],
+                    },
+                },
+            });
+        });
+        it('should pass through model aliases', () => {
+            const markdown = {
+                kind: 'local',
+                name: 'test-agent',
+                description: 'A test agent',
+                model: GEMINI_MODEL_ALIAS_PRO,
+                system_prompt: 'You are a test agent.',
+            };
+            const result = markdownToAgentDefinition(markdown);
+            expect(result.modelConfig.model).toBe(GEMINI_MODEL_ALIAS_PRO);
+        });
+        it('should pass through unknown model names (e.g. auto)', () => {
+            const markdown = {
+                kind: 'local',
+                name: 'test-agent',
+                description: 'A test agent',
+                model: 'auto',
+                system_prompt: 'You are a test agent.',
+            };
+            const result = markdownToAgentDefinition(markdown);
+            expect(result.modelConfig.model).toBe('auto');
+        });
+        it('should convert remote agent definition', () => {
+            const markdown = {
+                kind: 'remote',
+                name: 'remote-agent',
+                description: 'A remote agent',
+                agent_card_url: 'https://example.com/card',
+            };
+            const result = markdownToAgentDefinition(markdown);
+            expect(result).toEqual({
+                kind: 'remote',
+                name: 'remote-agent',
+                description: 'A remote agent',
+                displayName: undefined,
+                agentCardUrl: 'https://example.com/card',
+                inputConfig: {
+                    inputSchema: {
+                        type: 'object',
+                        properties: {
+                            query: {
+                                type: 'string',
+                                description: 'The task for the agent.',
+                            },
+                        },
+                        required: [],
+                    },
+                },
+            });
+        });
+    });
+    describe('loadAgentsFromDirectory', () => {
+        it('should load definitions from a directory (Markdown only)', async () => {
+            await writeAgentMarkdown(`---
+name: agent-1
+description: Agent 1
+---
+Prompt 1`, 'valid.md');
+            // Create a non-supported file
+            await fs.writeFile(path.join(tempDir, 'other.txt'), 'content');
+            // Create a hidden file
+            await writeAgentMarkdown(`---
+name: hidden
+description: Hidden
+---
+Hidden`, '_hidden.md');
+            const result = await loadAgentsFromDirectory(tempDir);
+            expect(result.agents).toHaveLength(1);
+            expect(result.agents[0].name).toBe('agent-1');
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should return empty result if directory does not exist', async () => {
+            const nonExistentDir = path.join(tempDir, 'does-not-exist');
+            const result = await loadAgentsFromDirectory(nonExistentDir);
+            expect(result.agents).toHaveLength(0);
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should capture errors for malformed individual files', async () => {
+            // Create a malformed Markdown file
+            await writeAgentMarkdown('invalid markdown', 'malformed.md');
+            const result = await loadAgentsFromDirectory(tempDir);
+            expect(result.agents).toHaveLength(0);
+            expect(result.errors).toHaveLength(1);
+        });
+    });
+    describe('remote agent auth configuration', () => {
+        it('should parse remote agent with apiKey auth', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: api-key-agent
+agent_card_url: https://example.com/card
+auth:
+  type: apiKey
+  key: $MY_API_KEY
+  in: header
+  name: X-Custom-Key
+---
+`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toMatchObject({
+                kind: 'remote',
+                name: 'api-key-agent',
+                auth: {
+                    type: 'apiKey',
+                    key: '$MY_API_KEY',
+                    in: 'header',
+                    name: 'X-Custom-Key',
+                },
+            });
+        });
+        it('should parse remote agent with http Bearer auth', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: bearer-agent
+agent_card_url: https://example.com/card
+auth:
+  type: http
+  scheme: Bearer
+  token: $BEARER_TOKEN
+---
+`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toMatchObject({
+                kind: 'remote',
+                name: 'bearer-agent',
+                auth: {
+                    type: 'http',
+                    scheme: 'Bearer',
+                    token: '$BEARER_TOKEN',
+                },
+            });
+        });
+        it('should parse remote agent with http Basic auth', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: basic-agent
+agent_card_url: https://example.com/card
+auth:
+  type: http
+  scheme: Basic
+  username: $AUTH_USER
+  password: $AUTH_PASS
+---
+`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result).toHaveLength(1);
+            expect(result[0]).toMatchObject({
+                kind: 'remote',
+                name: 'basic-agent',
+                auth: {
+                    type: 'http',
+                    scheme: 'Basic',
+                    username: '$AUTH_USER',
+                    password: '$AUTH_PASS',
+                },
+            });
+        });
+        it('should throw error for Bearer auth without token', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: invalid-bearer
+agent_card_url: https://example.com/card
+auth:
+  type: http
+  scheme: Bearer
+---
+`);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow(/Bearer scheme requires "token"/);
+        });
+        it('should throw error for Basic auth without credentials', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: invalid-basic
+agent_card_url: https://example.com/card
+auth:
+  type: http
+  scheme: Basic
+  username: user
+---
+`);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow(/Basic scheme requires "username" and "password"/);
+        });
+        it('should throw error for apiKey auth without key', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: invalid-apikey
+agent_card_url: https://example.com/card
+auth:
+  type: apiKey
+---
+`);
+            await expect(parseAgentMarkdown(filePath)).rejects.toThrow(/auth\.key.*Required/);
+        });
+        it('should convert auth config in markdownToAgentDefinition', () => {
+            const markdown = {
+                kind: 'remote',
+                name: 'auth-agent',
+                agent_card_url: 'https://example.com/card',
+                auth: {
+                    type: 'apiKey',
+                    key: '$API_KEY',
+                    in: 'header',
+                },
+            };
+            const result = markdownToAgentDefinition(markdown);
+            expect(result).toMatchObject({
+                kind: 'remote',
+                name: 'auth-agent',
+                auth: {
+                    type: 'apiKey',
+                    key: '$API_KEY',
+                    location: 'header',
+                },
+            });
+        });
+        it('should parse auth with agent_card_requires_auth flag', async () => {
+            const filePath = await writeAgentMarkdown(`---
+kind: remote
+name: protected-card-agent
+agent_card_url: https://example.com/card
+auth:
+  type: apiKey
+  key: $MY_API_KEY
+  agent_card_requires_auth: true
+---
+`);
+            const result = await parseAgentMarkdown(filePath);
+            expect(result[0]).toMatchObject({
+                auth: {
+                    type: 'apiKey',
+                    agent_card_requires_auth: true,
+                },
+            });
+        });
+    });
+});
+//# sourceMappingURL=agentLoader.test.js.map

package/dist/src/agents/agentLoader.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agentLoader.test.js","sourceRoot":"","sources":["../../../src/agents/agentLoader.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EACL,kBAAkB,EAClB,yBAAyB,EACzB,uBAAuB,EACvB,cAAc,GACf,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAE7D,OAAO,EAAE,wBAAwB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAEzE,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;IACtB,IAAI,OAAe,CAAC;IAEpB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,KAAK,UAAU,kBAAkB,CAAC,OAAe,EAAE,QAAQ,GAAG,SAAS;QACrE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC9C,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACtC,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;0BAItB,CAAC,CAAC;YAEtB,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC9B,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,kBAAkB;gBAC/B,IAAI,EAAE,OAAO;gBACb,aAAa,EAAE,2BAA2B;aAC3C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;;uBAQzB,CAAC,CAAC;YAEnB,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC9B,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,0BAA0B;gBACvC,KAAK,EAAE,CAAC,mBAAmB,CAAC;gBAC5B,KAAK,EAAE,YAAY;gBACnB,WAAW,EAAE,GAAG;gBAChB,aAAa,EAAE,wBAAwB;aACxC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,6BAA6B,CAAC,CAAC;YACzE,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,cAAc,CACf,CAAC;YACF,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,oCAAoC,CACrC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;KAG3C,CAAC,CAAC;YACD,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,cAAc,CACf,CAAC;YACF,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,iCAAiC,CAClC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;KAI3C,CAAC,CAAC;YACD,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,mBAAmB,CACpB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;CAM/C,CAAC,CAAC;YACG,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,gBAAgB;gBAC7B,cAAc,EAAE,0BAA0B;aAC3C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;CAK/C,CAAC,CAAC;YACG,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,iBAAiB;gBACvB,WAAW,EAAE,UAAU;gBACvB,cAAc,EAAE,8BAA8B;aAC/C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;CAK/C,CAAC,CAAC;YACG,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,gBAAgB;gBACtB,cAAc,EAAE,0BAA0B;aAC3C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;;CAQ/C,CAAC,CAAC;YACG,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,UAAU;gBAChB,cAAc,EAAE,uBAAuB;aACxC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,UAAU;gBAChB,cAAc,EAAE,uBAAuB;aACxC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;IAI5C,CAAC,CAAC;YACA,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,qBAAqB;gBAC3B,cAAc,EAAE,0BAA0B;aAC3C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;KAI3C,CAAC,CAAC;YACD,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,2BAA2B,CAC5B,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;YAC5E,MAAM,QAAQ,GAAG;gBACf,IAAI,EAAE,OAAgB;gBACtB,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,cAAc;gBAC3B,aAAa,EAAE,uBAAuB;aACvC,CAAC;YAEF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC;gBAC3B,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,cAAc;gBAC3B,YAAY,EAAE;oBACZ,YAAY,EAAE,uBAAuB;oBACrC,KAAK,EAAE,UAAU;iBAClB;gBACD,WAAW,EAAE;oBACX,KAAK,EAAE,SAAS;oBAChB,qBAAqB,EAAE;wBACrB,IAAI,EAAE,IAAI;qBACX;iBACF;gBACD,SAAS,EAAE;oBACT,cAAc,EAAE,wBAAwB;oBACxC,QAAQ,EAAE,iBAAiB;iBAC5B;gBACD,WAAW,EAAE;oBACX,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;gCACd,WAAW,EAAE,yBAAyB;6BACvC;yBACF;wBACD,QAAQ,EAAE,EAAE;qBACb;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,QAAQ,GAAG;gBACf,IAAI,EAAE,OAAgB;gBACtB,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,cAAc;gBAC3B,KAAK,EAAE,sBAAsB;gBAC7B,aAAa,EAAE,uBAAuB;aACvC,CAAC;YAEF,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,CACe,CAAC;YAC1B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG;gBACf,IAAI,EAAE,OAAgB;gBACtB,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,cAAc;gBAC3B,KAAK,EAAE,MAAM;gBACb,aAAa,EAAE,uBAAuB;aACvC,CAAC;YAEF,MAAM,MAAM,GAAG,yBAAyB,CACtC,QAAQ,CACe,CAAC;YAC1B,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,QAAQ,GAAG;gBACf,IAAI,EAAE,QAAiB;gBACvB,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,gBAAgB;gBAC7B,cAAc,EAAE,0BAA0B;aAC3C,CAAC;YAEF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,gBAAgB;gBAC7B,WAAW,EAAE,SAAS;gBACtB,YAAY,EAAE,0BAA0B;gBACxC,WAAW,EAAE;oBACX,WAAW,EAAE;wBACX,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;gCACd,WAAW,EAAE,yBAAyB;6BACvC;yBACF;wBACD,QAAQ,EAAE,EAAE;qBACb;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,kBAAkB,CACtB;;;;SAIC,EACD,UAAU,CACX,CAAC;YAEF,8BAA8B;YAC9B,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,SAAS,CAAC,CAAC;YAE/D,uBAAuB;YACvB,MAAM,kBAAkB,CACtB;;;;OAID,EACC,YAAY,CACb,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,cAAc,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,mCAAmC;YACnC,MAAM,kBAAkB,CAAC,kBAAkB,EAAE,cAAc,CAAC,CAAC;YAE7D,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;;;;CAU/C,CAAC,CAAC;YACG,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC9B,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,GAAG,EAAE,aAAa;oBAClB,EAAE,EAAE,QAAQ;oBACZ,IAAI,EAAE,cAAc;iBACrB;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;;;CAS/C,CAAC,CAAC;YACG,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC9B,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,cAAc;gBACpB,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,MAAM,EAAE,QAAQ;oBAChB,KAAK,EAAE,eAAe;iBACvB;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;;;;CAU/C,CAAC,CAAC;YACG,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC9B,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,IAAI,EAAE,MAAM;oBACZ,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,YAAY;oBACtB,QAAQ,EAAE,YAAY;iBACvB;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;;CAQ/C,CAAC,CAAC;YACG,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,gCAAgC,CACjC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;;;CAS/C,CAAC,CAAC;YACG,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,iDAAiD,CAClD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;CAO/C,CAAC,CAAC;YACG,MAAM,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACxD,qBAAqB,CACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,QAAQ,GAAG;gBACf,IAAI,EAAE,QAAiB;gBACvB,IAAI,EAAE,YAAY;gBAClB,cAAc,EAAE,0BAA0B;gBAC1C,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAiB;oBACvB,GAAG,EAAE,UAAU;oBACf,EAAE,EAAE,QAAiB;iBACtB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC;gBAC3B,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,YAAY;gBAClB,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,GAAG,EAAE,UAAU;oBACf,QAAQ,EAAE,QAAQ;iBACnB;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC;;;;;;;;;CAS/C,CAAC,CAAC;YACG,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC9B,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,wBAAwB,EAAE,IAAI;iBAC/B;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/agents/auth-provider/base-provider.d.ts

@@ -0,0 +1,40 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { HttpHeaders } from '@a2a-js/sdk/client';
+import type { A2AAuthProvider, A2AAuthProviderType } from './types.js';
+/**
+ * Abstract base class for A2A authentication providers.
+ * Provides default implementations for optional methods.
+ */
+export declare abstract class BaseA2AAuthProvider implements A2AAuthProvider {
+    /**
+     * The type of authentication provider.
+     */
+    abstract readonly type: A2AAuthProviderType;
+    /**
+     * Get the HTTP headers to include in requests.
+     * Subclasses must implement this method.
+     */
+    abstract headers(): Promise<HttpHeaders>;
+    private static readonly MAX_AUTH_RETRIES;
+    private authRetryCount;
+    /**
+     * Check if a request should be retried with new headers.
+     *
+     * The default implementation checks for 401/403 status codes and
+     * returns fresh headers for retry. Subclasses can override for
+     * custom retry logic.
+     *
+     * @param _req The original request init
+     * @param res The response from the server
+     * @returns New headers for retry, or undefined if no retry should be made
+     */
+    shouldRetryWithHeaders(_req: RequestInit, res: Response): Promise<HttpHeaders | undefined>;
+    /**
+     * Initialize the provider. Override in subclasses that need async setup.
+     */
+    initialize(): Promise<void>;
+}

package/dist/src/agents/auth-provider/base-provider.js

@@ -0,0 +1,43 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Abstract base class for A2A authentication providers.
+ * Provides default implementations for optional methods.
+ */
+export class BaseA2AAuthProvider {
+    static MAX_AUTH_RETRIES = 2;
+    authRetryCount = 0;
+    /**
+     * Check if a request should be retried with new headers.
+     *
+     * The default implementation checks for 401/403 status codes and
+     * returns fresh headers for retry. Subclasses can override for
+     * custom retry logic.
+     *
+     * @param _req The original request init
+     * @param res The response from the server
+     * @returns New headers for retry, or undefined if no retry should be made
+     */
+    async shouldRetryWithHeaders(_req, res) {
+        if (res.status === 401 || res.status === 403) {
+            if (this.authRetryCount >= BaseA2AAuthProvider.MAX_AUTH_RETRIES) {
+                return undefined; // Max retries exceeded
+            }
+            this.authRetryCount++;
+            return this.headers();
+        }
+        // Reset count if not an auth error
+        this.authRetryCount = 0;
+        return undefined;
+    }
+    /**
+     * Initialize the provider. Override in subclasses that need async setup.
+     */
+    async initialize() {
+        // Default: no-op
+    }
+}
+//# sourceMappingURL=base-provider.js.map

package/dist/src/agents/auth-provider/base-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base-provider.js","sourceRoot":"","sources":["../../../../src/agents/auth-provider/base-provider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH;;;GAGG;AACH,MAAM,OAAgB,mBAAmB;IAY/B,MAAM,CAAU,gBAAgB,GAAG,CAAC,CAAC;IACrC,cAAc,GAAG,CAAC,CAAC;IAE3B;;;;;;;;;;OAUG;IACH,KAAK,CAAC,sBAAsB,CAC1B,IAAiB,EACjB,GAAa;QAEb,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAC7C,IAAI,IAAI,CAAC,cAAc,IAAI,mBAAmB,CAAC,gBAAgB,EAAE,CAAC;gBAChE,OAAO,SAAS,CAAC,CAAC,uBAAuB;YAC3C,CAAC;YACD,IAAI,CAAC,cAAc,EAAE,CAAC;YACtB,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;QACD,mCAAmC;QACnC,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,iBAAiB;IACnB,CAAC"}

package/dist/src/agents/auth-provider/base-provider.test.d.ts

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

package/dist/src/agents/auth-provider/base-provider.test.js

@@ -0,0 +1,107 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { BaseA2AAuthProvider } from './base-provider.js';
+/**
+ * Concrete implementation of BaseA2AAuthProvider for testing.
+ */
+class TestAuthProvider extends BaseA2AAuthProvider {
+    type = 'apiKey';
+    testHeaders;
+    constructor(headers = { Authorization: 'test-token' }) {
+        super();
+        this.testHeaders = headers;
+    }
+    async headers() {
+        return this.testHeaders;
+    }
+    setHeaders(headers) {
+        this.testHeaders = headers;
+    }
+}
+describe('BaseA2AAuthProvider', () => {
+    describe('shouldRetryWithHeaders', () => {
+        it('should return headers for 401 response', async () => {
+            const provider = new TestAuthProvider({ Authorization: 'Bearer token' });
+            const response = new Response(null, { status: 401 });
+            const result = await provider.shouldRetryWithHeaders({}, response);
+            expect(result).toEqual({ Authorization: 'Bearer token' });
+        });
+        it('should return headers for 403 response', async () => {
+            const provider = new TestAuthProvider({ Authorization: 'Bearer token' });
+            const response = new Response(null, { status: 403 });
+            const result = await provider.shouldRetryWithHeaders({}, response);
+            expect(result).toEqual({ Authorization: 'Bearer token' });
+        });
+        it('should return undefined for 200 response', async () => {
+            const provider = new TestAuthProvider();
+            const response = new Response(null, { status: 200 });
+            const result = await provider.shouldRetryWithHeaders({}, response);
+            expect(result).toBeUndefined();
+        });
+        it('should return undefined for 500 response', async () => {
+            const provider = new TestAuthProvider();
+            const response = new Response(null, { status: 500 });
+            const result = await provider.shouldRetryWithHeaders({}, response);
+            expect(result).toBeUndefined();
+        });
+        it('should return undefined for 404 response', async () => {
+            const provider = new TestAuthProvider();
+            const response = new Response(null, { status: 404 });
+            const result = await provider.shouldRetryWithHeaders({}, response);
+            expect(result).toBeUndefined();
+        });
+        it('should call headers() to get fresh headers on retry', async () => {
+            const provider = new TestAuthProvider({ Authorization: 'old-token' });
+            const response = new Response(null, { status: 401 });
+            // Change headers before retry
+            provider.setHeaders({ Authorization: 'new-token' });
+            const result = await provider.shouldRetryWithHeaders({}, response);
+            expect(result).toEqual({ Authorization: 'new-token' });
+        });
+        it('should retry up to 2 times on 401/403', async () => {
+            const provider = new TestAuthProvider({ Authorization: 'Bearer token' });
+            const response401 = new Response(null, { status: 401 });
+            // First retry should succeed
+            const result1 = await provider.shouldRetryWithHeaders({}, response401);
+            expect(result1).toEqual({ Authorization: 'Bearer token' });
+            // Second retry should succeed
+            const result2 = await provider.shouldRetryWithHeaders({}, response401);
+            expect(result2).toEqual({ Authorization: 'Bearer token' });
+        });
+        it('should return undefined after max retries exceeded', async () => {
+            const provider = new TestAuthProvider({ Authorization: 'Bearer token' });
+            const response401 = new Response(null, { status: 401 });
+            // Exhaust retries
+            await provider.shouldRetryWithHeaders({}, response401); // retry 1
+            await provider.shouldRetryWithHeaders({}, response401); // retry 2
+            // Third attempt should return undefined
+            const result = await provider.shouldRetryWithHeaders({}, response401);
+            expect(result).toBeUndefined();
+        });
+        it('should reset retry count on successful response', async () => {
+            const provider = new TestAuthProvider({ Authorization: 'Bearer token' });
+            const response401 = new Response(null, { status: 401 });
+            const response200 = new Response(null, { status: 200 });
+            // Use up retries
+            await provider.shouldRetryWithHeaders({}, response401); // retry 1
+            await provider.shouldRetryWithHeaders({}, response401); // retry 2
+            // Success resets counter
+            await provider.shouldRetryWithHeaders({}, response200);
+            // Should be able to retry again
+            const result = await provider.shouldRetryWithHeaders({}, response401);
+            expect(result).toEqual({ Authorization: 'Bearer token' });
+        });
+    });
+    describe('initialize', () => {
+        it('should be a no-op by default', async () => {
+            const provider = new TestAuthProvider();
+            // Should not throw
+            await expect(provider.initialize()).resolves.toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=base-provider.test.js.map