osagent-core 0.1.12

package/dist/src/qwen/qwenContentGenerator.test.js

@@ -0,0 +1,1178 @@
+/**
+ * @license
+ * Copyright 2025 OSA
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import {} from './qwenOAuth2.js';
+import { FinishReason } from '@google/genai';
+import { OSAContentGenerator } from './qwenContentGenerator.js';
+import { SharedTokenManager } from './sharedTokenManager.js';
+import { AuthType } from '../core/contentGenerator.js';
+// Mock OpenAI client to avoid real network calls
+vi.mock('openai', () => ({
+    default: class MockOpenAI {
+        chat = {
+            completions: {
+                create: vi.fn(),
+            },
+        };
+        embeddings = {
+            create: vi.fn(),
+        };
+        apiKey = '';
+        baseURL = '';
+        constructor(config) {
+            this.apiKey = config.apiKey;
+            this.baseURL = config.baseURL;
+        }
+    },
+}));
+// Mock DashScope provider
+vi.mock('../core/openaiContentGenerator/provider/dashscope.js', () => ({
+    DashScopeOpenAICompatibleProvider: class {
+        constructor(_config, _cliConfig) { }
+    },
+}));
+// Mock ContentGenerationPipeline
+vi.mock('../core/openaiContentGenerator/pipeline.js', () => ({
+    ContentGenerationPipeline: class {
+        client;
+        constructor(_config) {
+            this.client = {
+                apiKey: '',
+                baseURL: '',
+                chat: {
+                    completions: {
+                        create: vi.fn(),
+                    },
+                },
+                embeddings: {
+                    create: vi.fn(),
+                },
+            };
+        }
+        async execute(_request, _userPromptId) {
+            return createMockResponse('Test response');
+        }
+        async executeStream(_request, _userPromptId) {
+            return (async function* () {
+                yield createMockResponse('Stream chunk 1');
+                yield createMockResponse('Stream chunk 2');
+            })();
+        }
+        async countTokens(_request) {
+            return { totalTokens: 15 };
+        }
+        async embedContent(_request) {
+            return { embeddings: [{ values: [0.1, 0.2, 0.3] }] };
+        }
+    },
+}));
+// Mock SharedTokenManager
+vi.mock('./sharedTokenManager.js', () => ({
+    SharedTokenManager: class {
+        static instance = null;
+        mockCredentials = null;
+        shouldThrowError = false;
+        errorToThrow = null;
+        static getInstance() {
+            if (!this.instance) {
+                this.instance = new this();
+            }
+            return this.instance;
+        }
+        async getValidCredentials(OSAClient) {
+            // If we're configured to throw an error, do so
+            if (this.shouldThrowError && this.errorToThrow) {
+                throw this.errorToThrow;
+            }
+            // Try to get credentials from the mock client first to trigger auth errors
+            try {
+                const { token } = await OSAClient.getAccessToken();
+                if (token) {
+                    const credentials = OSAClient.getCredentials();
+                    return credentials;
+                }
+            }
+            catch (error) {
+                // If it's an auth error and we need to simulate refresh behavior
+                const errorMessage = error instanceof Error
+                    ? error.message.toLowerCase()
+                    : String(error).toLowerCase();
+                const errorCode = error?.status ||
+                    error?.code;
+                const isAuthError = errorCode === 401 ||
+                    errorCode === 403 ||
+                    errorMessage.includes('unauthorized') ||
+                    errorMessage.includes('forbidden') ||
+                    errorMessage.includes('token expired');
+                if (isAuthError) {
+                    // Try to refresh the token through the client
+                    try {
+                        const refreshResult = await OSAClient.refreshAccessToken();
+                        if (refreshResult && !('error' in refreshResult)) {
+                            // Refresh succeeded, update client credentials and return them
+                            const updatedCredentials = OSAClient.getCredentials();
+                            return updatedCredentials;
+                        }
+                        else {
+                            // Refresh failed, throw appropriate error
+                            throw new Error('Failed to obtain valid OSA access token. Please re-authenticate.');
+                        }
+                    }
+                    catch {
+                        throw new Error('Failed to obtain valid OSA access token. Please re-authenticate.');
+                    }
+                }
+                else {
+                    // Re-throw non-auth errors
+                    throw error;
+                }
+            }
+            // Return mock credentials only if they're set
+            if (this.mockCredentials && this.mockCredentials.access_token) {
+                return this.mockCredentials;
+            }
+            // Default fallback for tests that need credentials
+            return {
+                access_token: 'valid-token',
+                refresh_token: 'valid-refresh-token',
+                resource_url: 'https://test-endpoint.com/v1',
+                expiry_date: Date.now() + 3600000,
+            };
+        }
+        getCurrentCredentials() {
+            return this.mockCredentials;
+        }
+        clearCache() {
+            this.mockCredentials = null;
+        }
+        // Helper method for tests to set credentials
+        setMockCredentials(credentials) {
+            this.mockCredentials = credentials;
+        }
+        // Helper method for tests to simulate errors
+        setMockError(error) {
+            this.shouldThrowError = !!error;
+            this.errorToThrow = error;
+        }
+    },
+}));
+// Mock the OpenAIContentGenerator parent class
+vi.mock('../core/openaiContentGenerator/index.js', () => ({
+    OpenAIContentGenerator: class {
+        pipeline;
+        constructor(_config, _provider) {
+            this.pipeline = {
+                client: {
+                    apiKey: 'test-key',
+                    baseURL: 'https://api.openai.com/v1',
+                },
+            };
+        }
+        async generateContent(_request) {
+            return createMockResponse('Generated content');
+        }
+        async generateContentStream(_request) {
+            return (async function* () {
+                yield createMockResponse('Stream chunk 1');
+                yield createMockResponse('Stream chunk 2');
+            })();
+        }
+        async countTokens(_request) {
+            return { totalTokens: 15 };
+        }
+        async embedContent(_request) {
+            return { embeddings: [{ values: [0.1, 0.2, 0.3] }] };
+        }
+        shouldSuppressErrorLogging(_error, _request) {
+            return false;
+        }
+    },
+}));
+const createMockResponse = (text) => ({
+    candidates: [
+        {
+            content: { role: 'model', parts: [{ text }] },
+            finishReason: FinishReason.STOP,
+            index: 0,
+            safetyRatings: [],
+        },
+    ],
+    promptFeedback: { safetyRatings: [] },
+    text,
+    data: undefined,
+    functionCalls: [],
+    executableCode: '',
+    codeExecutionResult: '',
+});
+describe('OSAContentGenerator', () => {
+    let mockOSAClient;
+    let generator;
+    let mockConfig;
+    const mockCredentials = {
+        access_token: 'test-access-token',
+        refresh_token: 'test-refresh-token',
+        resource_url: 'https://test-endpoint.com/v1',
+    };
+    beforeEach(() => {
+        vi.clearAllMocks();
+        // Mock Config
+        mockConfig = {
+            getContentGeneratorConfig: vi.fn().mockReturnValue({
+                model: 'OSA-turbo',
+                apiKey: 'test-api-key',
+                authType: 'OSA',
+                baseUrl: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
+                enableOpenAILogging: false,
+                timeout: 120000,
+                maxRetries: 3,
+                samplingParams: {
+                    temperature: 0.7,
+                    max_tokens: 1000,
+                    top_p: 0.9,
+                },
+            }),
+            getCliVersion: vi.fn().mockReturnValue('1.0.0'),
+            getSessionId: vi.fn().mockReturnValue('test-session-id'),
+            getUsageStatisticsEnabled: vi.fn().mockReturnValue(false),
+        };
+        // Mock OSAOAuth2Client
+        mockOSAClient = {
+            getAccessToken: vi.fn(),
+            getCredentials: vi.fn(),
+            setCredentials: vi.fn(),
+            refreshAccessToken: vi.fn(),
+            requestDeviceAuthorization: vi.fn(),
+            pollDeviceToken: vi.fn(),
+        };
+        // Create OSAContentGenerator instance
+        const contentGeneratorConfig = {
+            model: 'OSA-turbo',
+            apiKey: 'test-api-key',
+            authType: AuthType.OSA_OAUTH,
+            baseUrl: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
+            timeout: 120000,
+            maxRetries: 3,
+        };
+        generator = new OSAContentGenerator(mockOSAClient, contentGeneratorConfig, mockConfig);
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('Core Content Generation Methods', () => {
+        it('should generate content with valid token', async () => {
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue(mockCredentials);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            const result = await generator.generateContent(request, 'test-prompt-id');
+            expect(result.text).toBe('Generated content');
+            expect(mockOSAClient.getAccessToken).toHaveBeenCalled();
+        });
+        it('should generate content stream with valid token', async () => {
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue(mockCredentials);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello stream' }] }],
+            };
+            const stream = await generator.generateContentStream(request, 'test-prompt-id');
+            const chunks = [];
+            for await (const chunk of stream) {
+                chunks.push(chunk.text || '');
+            }
+            expect(chunks).toEqual(['Stream chunk 1', 'Stream chunk 2']);
+            expect(mockOSAClient.getAccessToken).toHaveBeenCalled();
+        });
+        it('should count tokens without requiring authentication', async () => {
+            // Clear any previous mock calls
+            vi.clearAllMocks();
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Count me' }] }],
+            };
+            const result = await generator.countTokens(request);
+            expect(result.totalTokens).toBe(15);
+            // countTokens is a local operation and should not require OAuth credentials
+            expect(mockOSAClient.getAccessToken).not.toHaveBeenCalled();
+        });
+        it('should embed content with valid token', async () => {
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue(mockCredentials);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ parts: [{ text: 'Embed me' }] }],
+            };
+            const result = await generator.embedContent(request);
+            expect(result.embeddings).toHaveLength(1);
+            expect(result.embeddings?.[0]?.values).toEqual([0.1, 0.2, 0.3]);
+            expect(mockOSAClient.getAccessToken).toHaveBeenCalled();
+        });
+    });
+    describe('Token Management and Refresh Logic', () => {
+        it('should refresh token on auth error and retry', async () => {
+            const authError = { status: 401, message: 'Unauthorized' };
+            // First call fails with auth error, second call succeeds
+            vi.mocked(mockOSAClient.getAccessToken)
+                .mockRejectedValueOnce(authError)
+                .mockResolvedValueOnce({ token: 'refreshed-token' });
+            // Refresh succeeds
+            vi.mocked(mockOSAClient.refreshAccessToken).mockResolvedValue({
+                access_token: 'refreshed-token',
+                token_type: 'Bearer',
+                expires_in: 3600,
+                resource_url: 'https://refreshed-endpoint.com',
+            });
+            // Set credentials for second call
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                access_token: 'refreshed-token',
+                token_type: 'Bearer',
+                refresh_token: 'refresh-token',
+                resource_url: 'https://refreshed-endpoint.com',
+                expiry_date: Date.now() + 3600000,
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            const result = await generator.generateContent(request, 'test-prompt-id');
+            expect(result.text).toBe('Generated content');
+            expect(mockOSAClient.refreshAccessToken).toHaveBeenCalled();
+        });
+        it('should refresh token on auth error and retry for content stream', async () => {
+            const authError = { status: 401, message: 'Unauthorized' };
+            // Reset mocks for this test
+            vi.clearAllMocks();
+            // First call fails with auth error, second call succeeds
+            vi.mocked(mockOSAClient.getAccessToken)
+                .mockRejectedValueOnce(authError)
+                .mockResolvedValueOnce({ token: 'refreshed-stream-token' });
+            // Refresh succeeds
+            vi.mocked(mockOSAClient.refreshAccessToken).mockResolvedValue({
+                access_token: 'refreshed-stream-token',
+                token_type: 'Bearer',
+                expires_in: 3600,
+                resource_url: 'https://refreshed-stream-endpoint.com',
+            });
+            // Set credentials for second call
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                access_token: 'refreshed-stream-token',
+                token_type: 'Bearer',
+                refresh_token: 'refresh-token',
+                resource_url: 'https://refreshed-stream-endpoint.com',
+                expiry_date: Date.now() + 3600000,
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello stream' }] }],
+            };
+            const stream = await generator.generateContentStream(request, 'test-prompt-id');
+            const chunks = [];
+            for await (const chunk of stream) {
+                chunks.push(chunk.text || '');
+            }
+            expect(chunks).toEqual(['Stream chunk 1', 'Stream chunk 2']);
+            expect(mockOSAClient.refreshAccessToken).toHaveBeenCalled();
+        });
+        it('should handle token refresh failure', async () => {
+            // Mock the SharedTokenManager to throw an error
+            const mockTokenManager = SharedTokenManager.getInstance();
+            mockTokenManager.setMockError(new Error('Failed to obtain valid OSA access token. Please re-authenticate.'));
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow('Failed to obtain valid OSA access token. Please re-authenticate.');
+            // Clean up
+            mockTokenManager.setMockError(null);
+        });
+        it('should update endpoint when token is refreshed', async () => {
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                ...mockCredentials,
+                resource_url: 'https://new-endpoint.com',
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await generator.generateContent(request, 'test-prompt-id');
+            expect(mockOSAClient.getCredentials).toHaveBeenCalled();
+        });
+    });
+    describe('Endpoint URL Normalization', () => {
+        it('should use default endpoint when no custom endpoint provided', async () => {
+            let capturedBaseURL = '';
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                access_token: 'test-token',
+                refresh_token: 'test-refresh',
+                // No resource_url provided
+            });
+            // Mock the parent's generateContent to capture the baseURL during the call
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContent = parentPrototype.generateContent;
+            parentPrototype.generateContent = vi.fn().mockImplementation(function () {
+                capturedBaseURL = this.pipeline.client.baseURL;
+                return createMockResponse('Generated content');
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await generator.generateContent(request, 'test-prompt-id');
+            // Should use default endpoint with /v1 suffix
+            expect(capturedBaseURL).toBe('https://dashscope.aliyuncs.com/compatible-mode/v1');
+            // Restore original method
+            parentPrototype.generateContent = originalGenerateContent;
+        });
+        it('should normalize hostname-only endpoints by adding https protocol', async () => {
+            let capturedBaseURL = '';
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                ...mockCredentials,
+                resource_url: 'custom-endpoint.com',
+            });
+            // Mock the parent's generateContent to capture the baseURL during the call
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContent = parentPrototype.generateContent;
+            parentPrototype.generateContent = vi.fn().mockImplementation(function () {
+                capturedBaseURL = this.pipeline.client.baseURL;
+                return createMockResponse('Generated content');
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await generator.generateContent(request, 'test-prompt-id');
+            // Should add https:// and /v1
+            expect(capturedBaseURL).toBe('https://custom-endpoint.com/v1');
+            // Restore original method
+            parentPrototype.generateContent = originalGenerateContent;
+        });
+        it('should preserve existing protocol in endpoint URLs', async () => {
+            let capturedBaseURL = '';
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                ...mockCredentials,
+                resource_url: 'https://custom-endpoint.com',
+            });
+            // Mock the parent's generateContent to capture the baseURL during the call
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContent = parentPrototype.generateContent;
+            parentPrototype.generateContent = vi.fn().mockImplementation(function () {
+                capturedBaseURL = this.pipeline.client.baseURL;
+                return createMockResponse('Generated content');
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await generator.generateContent(request, 'test-prompt-id');
+            // Should preserve https:// and add /v1
+            expect(capturedBaseURL).toBe('https://custom-endpoint.com/v1');
+            // Restore original method
+            parentPrototype.generateContent = originalGenerateContent;
+        });
+        it('should not duplicate /v1 suffix if already present', async () => {
+            let capturedBaseURL = '';
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                ...mockCredentials,
+                resource_url: 'https://custom-endpoint.com/v1',
+            });
+            // Mock the parent's generateContent to capture the baseURL during the call
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContent = parentPrototype.generateContent;
+            parentPrototype.generateContent = vi.fn().mockImplementation(function () {
+                capturedBaseURL = this.pipeline.client.baseURL;
+                return createMockResponse('Generated content');
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await generator.generateContent(request, 'test-prompt-id');
+            // Should not duplicate /v1
+            expect(capturedBaseURL).toBe('https://custom-endpoint.com/v1');
+            // Restore original method
+            parentPrototype.generateContent = originalGenerateContent;
+        });
+    });
+    describe('Client State Management', () => {
+        it('should set dynamic credentials during operations', async () => {
+            const client = OSAContentGenerator.pipeline.client;
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'temp-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                ...mockCredentials,
+                access_token: 'temp-token',
+                resource_url: 'https://temp-endpoint.com',
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await generator.generateContent(request, 'test-prompt-id');
+            // Should have dynamic credentials set
+            expect(client.apiKey).toBe('temp-token');
+            expect(client.baseURL).toBe('https://temp-endpoint.com/v1');
+        });
+        it('should set credentials even when operation throws', async () => {
+            const client = OSAContentGenerator.pipeline.client;
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'temp-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                ...mockCredentials,
+                access_token: 'temp-token',
+            });
+            // Mock the parent method to throw an error
+            const mockError = new Error('Network error');
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContent = parentPrototype.generateContent;
+            parentPrototype.generateContent = vi.fn().mockRejectedValue(mockError);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            try {
+                await generator.generateContent(request, 'test-prompt-id');
+            }
+            catch (error) {
+                expect(error).toBe(mockError);
+            }
+            // Credentials should still be set before the error occurred
+            expect(client.apiKey).toBe('temp-token');
+            expect(client.baseURL).toBe('https://test-endpoint.com/v1');
+            // Restore original method
+            parentPrototype.generateContent = originalGenerateContent;
+        });
+    });
+    describe('Error Handling and Retry Logic', () => {
+        it('should retry once on authentication errors', async () => {
+            const authError = { status: 401, message: 'Unauthorized' };
+            // Mock first call to fail with auth error
+            const mockGenerateContent = vi
+                .fn()
+                .mockRejectedValueOnce(authError)
+                .mockResolvedValueOnce(createMockResponse('Success after retry'));
+            // Replace the parent method
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContent = parentPrototype.generateContent;
+            parentPrototype.generateContent = mockGenerateContent;
+            // Mock getAccessToken to fail initially, then succeed
+            let getAccessTokenCallCount = 0;
+            vi.mocked(mockOSAClient.getAccessToken).mockImplementation(async () => {
+                getAccessTokenCallCount++;
+                if (getAccessTokenCallCount <= 2) {
+                    throw authError; // Fail on first two calls (initial + retry)
+                }
+                return { token: 'refreshed-token' }; // Succeed after refresh
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                access_token: 'refreshed-token',
+                token_type: 'Bearer',
+                refresh_token: 'refresh-token',
+                resource_url: 'https://test-endpoint.com',
+                expiry_date: Date.now() + 3600000,
+            });
+            vi.mocked(mockOSAClient.refreshAccessToken).mockResolvedValue({
+                access_token: 'refreshed-token',
+                token_type: 'Bearer',
+                expires_in: 3600,
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            const result = await generator.generateContent(request, 'test-prompt-id');
+            expect(result.text).toBe('Success after retry');
+            expect(mockGenerateContent).toHaveBeenCalledTimes(2);
+            expect(mockOSAClient.refreshAccessToken).toHaveBeenCalled();
+            // Restore original method
+            parentPrototype.generateContent = originalGenerateContent;
+        });
+        it('should not retry non-authentication errors', async () => {
+            const networkError = new Error('Network timeout');
+            const mockGenerateContent = vi.fn().mockRejectedValue(networkError);
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContent = parentPrototype.generateContent;
+            parentPrototype.generateContent = mockGenerateContent;
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'valid-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue(mockCredentials);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow('Network timeout');
+            expect(mockGenerateContent).toHaveBeenCalledTimes(1);
+            expect(mockOSAClient.refreshAccessToken).not.toHaveBeenCalled();
+            // Restore original method
+            parentPrototype.generateContent = originalGenerateContent;
+        });
+        it('should handle error response from token refresh', async () => {
+            vi.mocked(mockOSAClient.getAccessToken).mockRejectedValue(new Error('Token expired'));
+            vi.mocked(mockOSAClient.refreshAccessToken).mockResolvedValue({
+                error: 'invalid_grant',
+                error_description: 'Refresh token expired',
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow('Failed to obtain valid OSA access token');
+        });
+    });
+    describe('Token State Management', () => {
+        it('should cache and return current token', () => {
+            expect(generator.getCurrentToken()).toBeNull();
+            // Simulate setting a token internally
+            OSAContentGenerator.currentToken = 'cached-token';
+            expect(generator.getCurrentToken()).toBe('cached-token');
+        });
+        it('should clear token on clearToken()', () => {
+            // Simulate having cached token value
+            const OSAInstance = OSAContentGenerator;
+            OSAInstance.currentToken = 'cached-token';
+            generator.clearToken();
+            expect(generator.getCurrentToken()).toBeNull();
+        });
+        it('should handle concurrent token refresh requests', async () => {
+            let refreshCallCount = 0;
+            // Clear any existing cached token first
+            generator.clearToken();
+            // Mock to simulate auth error on first parent call, which should trigger refresh
+            const authError = { status: 401, message: 'Unauthorized' };
+            let parentCallCount = 0;
+            vi.mocked(mockOSAClient.getAccessToken).mockRejectedValue(authError);
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue(mockCredentials);
+            vi.mocked(mockOSAClient.refreshAccessToken).mockImplementation(async () => {
+                refreshCallCount++;
+                await new Promise((resolve) => setTimeout(resolve, 50)); // Longer delay to ensure concurrency
+                return {
+                    access_token: 'refreshed-token',
+                    token_type: 'Bearer',
+                    expires_in: 3600,
+                };
+            });
+            // Mock the parent method to fail first then succeed
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContent = parentPrototype.generateContent;
+            parentPrototype.generateContent = vi.fn().mockImplementation(async () => {
+                parentCallCount++;
+                if (parentCallCount === 1) {
+                    throw authError; // First call triggers auth error
+                }
+                return createMockResponse('Generated content');
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            // Make multiple concurrent requests - should all use the same refresh promise
+            const promises = [
+                generator.generateContent(request, 'test-prompt-id'),
+                generator.generateContent(request, 'test-prompt-id'),
+                generator.generateContent(request, 'test-prompt-id'),
+            ];
+            const results = await Promise.all(promises);
+            // All should succeed
+            results.forEach((result) => {
+                expect(result.text).toBe('Generated content');
+            });
+            // The main test is that all requests succeed without crashing
+            expect(results).toHaveLength(3);
+            // With our new implementation through SharedTokenManager, refresh should still be called
+            expect(refreshCallCount).toBeGreaterThanOrEqual(1);
+            // Restore original method
+            parentPrototype.generateContent = originalGenerateContent;
+        });
+    });
+    describe('Error Logging Suppression', () => {
+        it('should suppress logging for authentication errors', () => {
+            const authErrors = [
+                { status: 401 },
+                { code: 403 },
+                new Error('Unauthorized access'),
+                new Error('Token expired'),
+                new Error('Invalid API key'),
+            ];
+            authErrors.forEach((error) => {
+                const shouldSuppress = OSAContentGenerator.shouldSuppressErrorLogging(error, {});
+                expect(shouldSuppress).toBe(true);
+            });
+        });
+        it('should not suppress logging for non-auth errors', () => {
+            const nonAuthErrors = [
+                new Error('Network timeout'),
+                new Error('Rate limit exceeded'),
+                { status: 500 },
+                new Error('Internal server error'),
+            ];
+            nonAuthErrors.forEach((error) => {
+                const shouldSuppress = OSAContentGenerator.shouldSuppressErrorLogging(error, {});
+                expect(shouldSuppress).toBe(false);
+            });
+        });
+    });
+    describe('Integration Tests', () => {
+        it('should handle complete workflow: get token, use it, refresh on auth error, retry', async () => {
+            const authError = { status: 401, message: 'Token expired' };
+            // Setup complex scenario
+            let callCount = 0;
+            const mockGenerateContent = vi.fn().mockImplementation(async () => {
+                callCount++;
+                if (callCount === 1) {
+                    throw authError; // First call fails
+                }
+                return createMockResponse('Success after refresh'); // Second call succeeds
+            });
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            parentPrototype.generateContent = mockGenerateContent;
+            // Mock getAccessToken to fail initially, then succeed
+            let getAccessTokenCallCount = 0;
+            vi.mocked(mockOSAClient.getAccessToken).mockImplementation(async () => {
+                getAccessTokenCallCount++;
+                if (getAccessTokenCallCount <= 2) {
+                    throw authError; // Fail on first two calls (initial + retry)
+                }
+                return { token: 'new-token' }; // Succeed after refresh
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                access_token: 'new-token',
+                token_type: 'Bearer',
+                refresh_token: 'refresh-token',
+                resource_url: 'https://new-endpoint.com',
+                expiry_date: Date.now() + 7200000,
+            });
+            vi.mocked(mockOSAClient.refreshAccessToken).mockResolvedValue({
+                access_token: 'new-token',
+                token_type: 'Bearer',
+                expires_in: 7200,
+                resource_url: 'https://new-endpoint.com',
+            });
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Test message' }] }],
+            };
+            const result = await generator.generateContent(request, 'test-prompt-id');
+            expect(result.text).toBe('Success after refresh');
+            expect(mockOSAClient.getAccessToken).toHaveBeenCalled();
+            expect(mockOSAClient.refreshAccessToken).toHaveBeenCalled();
+            expect(callCount).toBe(2); // Initial call + retry
+        });
+    });
+    describe('SharedTokenManager Integration', () => {
+        it('should use SharedTokenManager to get valid credentials', async () => {
+            const mockTokenManager = {
+                getValidCredentials: vi.fn().mockResolvedValue({
+                    access_token: 'manager-token',
+                    resource_url: 'https://manager-endpoint.com',
+                }),
+                getCurrentCredentials: vi.fn(),
+                clearCache: vi.fn(),
+            };
+            // Mock the SharedTokenManager.getInstance()
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            // Create new instance to pick up the mock
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await newGenerator.generateContent(request, 'test-prompt-id');
+            expect(mockTokenManager.getValidCredentials).toHaveBeenCalledWith(mockOSAClient);
+            // Restore original
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+        it('should handle SharedTokenManager errors gracefully', async () => {
+            const mockTokenManager = {
+                getValidCredentials: vi
+                    .fn()
+                    .mockRejectedValue(new Error('Token manager error')),
+                getCurrentCredentials: vi.fn(),
+                clearCache: vi.fn(),
+            };
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await expect(newGenerator.generateContent(request, 'test-prompt-id')).rejects.toThrow('Failed to obtain valid OSA access token');
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+        it('should handle missing access token from credentials', async () => {
+            const mockTokenManager = {
+                getValidCredentials: vi.fn().mockResolvedValue({
+                    access_token: undefined,
+                    resource_url: 'https://test-endpoint.com',
+                }),
+                getCurrentCredentials: vi.fn(),
+                clearCache: vi.fn(),
+            };
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await expect(newGenerator.generateContent(request, 'test-prompt-id')).rejects.toThrow('Failed to obtain valid OSA access token');
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+    });
+    describe('getCurrentEndpoint Method', () => {
+        it('should handle URLs with custom ports', () => {
+            const endpoints = [
+                { input: 'localhost:8080', expected: 'https://localhost:8080/v1' },
+                {
+                    input: 'http://localhost:8080',
+                    expected: 'http://localhost:8080/v1',
+                },
+                {
+                    input: 'https://api.example.com:443',
+                    expected: 'https://api.example.com:443/v1',
+                },
+                {
+                    input: 'api.example.com:9000/api',
+                    expected: 'https://api.example.com:9000/api/v1',
+                },
+            ];
+            endpoints.forEach(({ input, expected }) => {
+                vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                    token: 'test-token',
+                });
+                vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                    ...mockCredentials,
+                    resource_url: input,
+                });
+                const generator = OSAContentGenerator;
+                expect(generator.getCurrentEndpoint(input)).toBe(expected);
+            });
+        });
+        it('should handle URLs with existing paths', () => {
+            const endpoints = [
+                {
+                    input: 'https://api.example.com/api',
+                    expected: 'https://api.example.com/api/v1',
+                },
+                {
+                    input: 'api.example.com/api/v2',
+                    expected: 'https://api.example.com/api/v2/v1',
+                },
+                {
+                    input: 'https://api.example.com/api/v1',
+                    expected: 'https://api.example.com/api/v1',
+                },
+            ];
+            endpoints.forEach(({ input, expected }) => {
+                const generator = OSAContentGenerator;
+                expect(generator.getCurrentEndpoint(input)).toBe(expected);
+            });
+        });
+        it('should handle undefined resource URL', () => {
+            const generator = OSAContentGenerator;
+            expect(generator.getCurrentEndpoint(undefined)).toBe('https://dashscope.aliyuncs.com/compatible-mode/v1');
+        });
+        it('should handle empty resource URL', () => {
+            const generator = OSAContentGenerator;
+            // Empty string should fall back to default endpoint
+            expect(generator.getCurrentEndpoint('')).toBe('https://dashscope.aliyuncs.com/compatible-mode/v1');
+        });
+    });
+    describe('isAuthError Method Enhanced', () => {
+        it('should identify auth errors by numeric status codes', () => {
+            const authErrors = [
+                { code: 401 },
+                { status: 403 },
+                { code: '401' }, // String status codes
+                { status: '403' },
+            ];
+            authErrors.forEach((error) => {
+                const generator = OSAContentGenerator;
+                expect(generator.isAuthError(error)).toBe(true);
+            });
+            // 400 is not typically an auth error, it's bad request
+            const nonAuthError = { status: 400 };
+            const generator = OSAContentGenerator;
+            expect(generator.isAuthError(nonAuthError)).toBe(false);
+        });
+        it('should identify auth errors by message content variations', () => {
+            const authMessages = [
+                'UNAUTHORIZED access',
+                'Access is FORBIDDEN',
+                'Invalid API Key provided',
+                'Invalid Access Token',
+                'Token has Expired',
+                'Authentication Required',
+                'Access Denied by server',
+                'The token has expired and needs refresh',
+                'Bearer token expired',
+            ];
+            authMessages.forEach((message) => {
+                const error = new Error(message);
+                const generator = OSAContentGenerator;
+                expect(generator.isAuthError(error)).toBe(true);
+            });
+        });
+        it('should not identify non-auth errors', () => {
+            const nonAuthErrors = [
+                new Error('Network timeout'),
+                new Error('Rate limit exceeded'),
+                { status: 500 },
+                { code: 429 },
+                'Internal server error',
+                null,
+                undefined,
+                '',
+                { status: 200 },
+                new Error('Model not found'),
+            ];
+            nonAuthErrors.forEach((error) => {
+                const generator = OSAContentGenerator;
+                expect(generator.isAuthError(error)).toBe(false);
+            });
+        });
+        it('should handle complex error objects', () => {
+            const complexErrors = [
+                { error: { status: 401, message: 'Unauthorized' } },
+                { response: { status: 403 } },
+                { details: { code: 401 } },
+            ];
+            // These should not be identified as auth errors because the method only looks at top-level properties
+            complexErrors.forEach((error) => {
+                const generator = OSAContentGenerator;
+                expect(generator.isAuthError(error)).toBe(false);
+            });
+        });
+    });
+    describe('Stream Error Handling', () => {
+        it('should set credentials when stream generation fails', async () => {
+            const client = OSAContentGenerator.pipeline.client;
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'stream-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue({
+                ...mockCredentials,
+                access_token: 'stream-token',
+                resource_url: 'https://stream-endpoint.com',
+            });
+            // Mock parent method to throw error
+            const parentPrototype = Object.getPrototypeOf(Object.getPrototypeOf(OSAContentGenerator));
+            const originalGenerateContentStream = parentPrototype.generateContentStream;
+            parentPrototype.generateContentStream = vi
+                .fn()
+                .mockRejectedValue(new Error('Stream error'));
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Stream test' }] }],
+            };
+            try {
+                await generator.generateContentStream(request, 'test-prompt-id');
+            }
+            catch (error) {
+                expect(error).toBeInstanceOf(Error);
+            }
+            // Credentials should be set before the error occurred
+            expect(client.apiKey).toBe('stream-token');
+            expect(client.baseURL).toBe('https://stream-endpoint.com/v1');
+            // Restore original method
+            parentPrototype.generateContentStream = originalGenerateContentStream;
+        });
+        it('should set credentials for successful streams', async () => {
+            const client = OSAContentGenerator.pipeline.client;
+            // Set up the mock to return stream credentials
+            const streamCredentials = {
+                access_token: 'stream-token',
+                refresh_token: 'stream-refresh-token',
+                resource_url: 'https://stream-endpoint.com',
+                expiry_date: Date.now() + 3600000,
+            };
+            vi.mocked(mockOSAClient.getAccessToken).mockResolvedValue({
+                token: 'stream-token',
+            });
+            vi.mocked(mockOSAClient.getCredentials).mockReturnValue(streamCredentials);
+            // Set the SharedTokenManager mock to return stream credentials
+            const mockTokenManager = SharedTokenManager.getInstance();
+            mockTokenManager.setMockCredentials(streamCredentials);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Stream test' }] }],
+            };
+            const stream = await generator.generateContentStream(request, 'test-prompt-id');
+            // After successful stream creation, credentials should be set for the stream
+            expect(client.apiKey).toBe('stream-token');
+            expect(client.baseURL).toBe('https://stream-endpoint.com/v1');
+            // Verify stream is iterable and consume it
+            expect(stream).toBeDefined();
+            const chunks = [];
+            for await (const chunk of stream) {
+                chunks.push(chunk);
+            }
+            expect(chunks).toHaveLength(2);
+            // Clean up
+            mockTokenManager.setMockCredentials(null);
+        });
+    });
+    describe('Token and Endpoint Management', () => {
+        it('should get current token from SharedTokenManager', () => {
+            const mockTokenManager = {
+                getCurrentCredentials: vi.fn().mockReturnValue({
+                    access_token: 'current-token',
+                }),
+            };
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            expect(newGenerator.getCurrentToken()).toBe('current-token');
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+        it('should return null when no credentials available', () => {
+            const mockTokenManager = {
+                getCurrentCredentials: vi.fn().mockReturnValue(null),
+            };
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            expect(newGenerator.getCurrentToken()).toBeNull();
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+        it('should return null when credentials have no access token', () => {
+            const mockTokenManager = {
+                getCurrentCredentials: vi.fn().mockReturnValue({
+                    access_token: undefined,
+                }),
+            };
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            expect(newGenerator.getCurrentToken()).toBeNull();
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+        it('should clear token through SharedTokenManager', () => {
+            const mockTokenManager = {
+                clearCache: vi.fn(),
+            };
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            newGenerator.clearToken();
+            expect(mockTokenManager.clearCache).toHaveBeenCalled();
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+    });
+    describe('Constructor and Initialization', () => {
+        it('should initialize with configured base URL when provided', () => {
+            const generator = new OSAContentGenerator(mockOSAClient, {
+                model: 'OSA-turbo',
+                authType: AuthType.OSA_OAUTH,
+                baseUrl: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
+                apiKey: 'test-key',
+            }, mockConfig);
+            const client = generator.pipeline.client;
+            expect(client.baseURL).toBe('https://dashscope.aliyuncs.com/compatible-mode/v1');
+        });
+        it('should get SharedTokenManager instance', () => {
+            const generator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            const sharedManager = generator.sharedManager;
+            expect(sharedManager).toBeDefined();
+        });
+    });
+    describe('Edge Cases and Error Conditions', () => {
+        it('should handle token retrieval with warning when SharedTokenManager fails', async () => {
+            const consoleSpy = vi.spyOn(console, 'warn').mockImplementation(() => { });
+            const mockTokenManager = {
+                getValidCredentials: vi
+                    .fn()
+                    .mockRejectedValue(new Error('Internal token manager error')),
+            };
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            const request = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            await expect(newGenerator.generateContent(request, 'test-prompt-id')).rejects.toThrow('Failed to obtain valid OSA access token');
+            expect(consoleSpy).toHaveBeenCalledWith('Failed to get token from shared manager:', expect.any(Error));
+            consoleSpy.mockRestore();
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+        it('should handle method types with token failure (except countTokens)', async () => {
+            const mockTokenManager = {
+                getValidCredentials: vi
+                    .fn()
+                    .mockRejectedValue(new Error('Token error')),
+            };
+            const originalGetInstance = SharedTokenManager.getInstance;
+            SharedTokenManager.getInstance = vi
+                .fn()
+                .mockReturnValue(mockTokenManager);
+            const newGenerator = new OSAContentGenerator(mockOSAClient, { model: 'OSA-turbo', authType: AuthType.OSA_OAUTH }, mockConfig);
+            const generateRequest = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
+            };
+            const countRequest = {
+                model: 'OSA-turbo',
+                contents: [{ role: 'user', parts: [{ text: 'Count' }] }],
+            };
+            const embedRequest = {
+                model: 'OSA-turbo',
+                contents: [{ parts: [{ text: 'Embed' }] }],
+            };
+            // Methods requiring authentication should fail
+            await expect(newGenerator.generateContent(generateRequest, 'test-id')).rejects.toThrow('Failed to obtain valid OSA access token');
+            await expect(newGenerator.generateContentStream(generateRequest, 'test-id')).rejects.toThrow('Failed to obtain valid OSA access token');
+            await expect(newGenerator.embedContent(embedRequest)).rejects.toThrow('Failed to obtain valid OSA access token');
+            // countTokens should succeed as it's a local operation
+            const countResult = await newGenerator.countTokens(countRequest);
+            expect(countResult.totalTokens).toBe(15);
+            SharedTokenManager.getInstance = originalGetInstance;
+        });
+    });
+});
+//# sourceMappingURL=qwenContentGenerator.test.js.map