@starlink-awaken/agentmesh 1.0.3 → 1.2.0

package/dist/tests/model-gateway/retry.test.js

@@ -0,0 +1,47 @@
+import { describe, test, expect } from 'bun:test';
+import { configureRetry, getRetryConfig, getRetryDelay, isRetryable, } from '../../src/model-gateway/retry.js';
+describe('retry config', () => {
+    test('default config values', () => {
+        const cfg = getRetryConfig();
+        expect(cfg.maxRetries).toBe(3);
+        expect(cfg.baseDelayMs).toBe(500);
+        expect(cfg.maxDelayMs).toBe(10000);
+    });
+    test('retryable status codes', () => {
+        expect(isRetryable(429)).toBe(true);
+        expect(isRetryable(500)).toBe(true);
+        expect(isRetryable(502)).toBe(true);
+        expect(isRetryable(503)).toBe(true);
+        expect(isRetryable(504)).toBe(true);
+        expect(isRetryable(400)).toBe(false);
+        expect(isRetryable(401)).toBe(false);
+        expect(isRetryable(404)).toBe(false);
+        expect(isRetryable(200)).toBe(false);
+    });
+    test('config overrides', () => {
+        configureRetry({ maxRetries: 5, baseDelayMs: 1000 });
+        const cfg = getRetryConfig();
+        expect(cfg.maxRetries).toBe(5);
+        expect(cfg.baseDelayMs).toBe(1000);
+        // Reset
+        configureRetry({ maxRetries: 3, baseDelayMs: 500 });
+    });
+});
+describe('retry delay', () => {
+    test('exponential backoff with base 500', () => {
+        const delay0 = getRetryDelay(0); // 500 * 2^0 = 500ms ± jitter
+        const delay1 = getRetryDelay(1); // 500 * 2^1 = 1000ms ± jitter
+        const delay2 = getRetryDelay(2); // 500 * 2^2 = 2000ms ± jitter
+        // 由于 jitter (+/-25%)，检查范围
+        expect(delay0).toBeGreaterThanOrEqual(375);
+        expect(delay0).toBeLessThanOrEqual(750);
+        expect(delay1).toBeGreaterThanOrEqual(750);
+        expect(delay1).toBeLessThanOrEqual(1500);
+        expect(delay2).toBeGreaterThanOrEqual(1500);
+        expect(delay2).toBeLessThanOrEqual(3000);
+    });
+    test('capped at maxDelayMs', () => {
+        const delay = getRetryDelay(10); // 500 * 2^10 = 512000ms, capped at 10000ms
+        expect(delay).toBeLessThanOrEqual(10000);
+    });
+});

package/dist/tests/model-gateway/router.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/tests/model-gateway/router.test.js

@@ -0,0 +1,108 @@
+import { describe, test, expect, beforeEach } from 'bun:test';
+import { initModelRouter, resolveProvider } from '../../src/model-gateway/router.js';
+const testConfig = {
+    default_model: 'deepseek-chat',
+    providers: {
+        deepseek: {
+            base_url: 'https://api.deepseek.com/v1',
+            api_key: 'sk-test-deepseek',
+            models: ['deepseek-chat', 'deepseek-v4-pro'],
+        },
+        openai: {
+            base_url: 'https://api.openai.com/v1',
+            api_key: 'sk-test-openai',
+            models: ['gpt-5.1'],
+        },
+        openrouter: {
+            base_url: 'https://openrouter.ai/api/v1',
+            api_key: 'sk-test-or',
+        },
+        ollama: {
+            base_url: 'http://127.0.0.1:11434/v1',
+            api_key: 'ollama',
+            models: ['qwen3:14b'],
+        },
+    },
+    fallback_chain: ['deepseek', 'openrouter', 'ollama'],
+    model_routing: {
+        'gpt-': ['openai', 'deepseek'],
+        deepseek: ['deepseek'],
+        claude: ['openrouter'],
+        qwen: ['ollama'],
+    },
+};
+describe('model router', () => {
+    beforeEach(() => {
+        initModelRouter(testConfig);
+    });
+    test('routes deepseek models to deepseek provider', () => {
+        const provider = resolveProvider('deepseek-chat');
+        expect(provider).toBeDefined();
+        expect(provider.name).toBe('deepseek');
+        expect(provider.base_url).toBe('https://api.deepseek.com/v1');
+        expect(provider.api_key).toBe('sk-test-deepseek');
+    });
+    test('routes gpt models to openai first', () => {
+        const provider = resolveProvider('gpt-5.1');
+        expect(provider).toBeDefined();
+        expect(provider.name).toBe('openai');
+    });
+    test('routes claude models to openrouter', () => {
+        const provider = resolveProvider('claude-sonnet-4-6');
+        expect(provider).toBeDefined();
+        expect(provider.name).toBe('openrouter');
+    });
+    test('routes qwen models to ollama', () => {
+        const provider = resolveProvider('qwen3:14b');
+        expect(provider).toBeDefined();
+        expect(provider.name).toBe('ollama');
+    });
+    test('unknown models fall through to fallback chain', () => {
+        const provider = resolveProvider('codestral:22b');
+        expect(provider).toBeDefined();
+        expect(provider.name).toBe('deepseek');
+    });
+    test('returns null when module has no config (fresh import)', () => {
+        // resolveProvider 返回 null 当 config 未被 initModelRouter 设置
+        // 但在 beforeEach 中已经 init，所以这个测试验证已知行为
+        const provider = resolveProvider('any-model');
+        expect(provider).toBeDefined();
+    });
+    test('skips provider without api key in fallback', () => {
+        const cfgWithEmpty = {
+            ...testConfig,
+            fallback_chain: ['empty_provider', 'deepseek'],
+            providers: {
+                ...testConfig.providers,
+                empty_provider: {
+                    base_url: 'https://nope.example.com/v1',
+                    api_key: '',
+                    models: [],
+                },
+            },
+        };
+        initModelRouter(cfgWithEmpty);
+        const provider = resolveProvider('unknown-model');
+        expect(provider.name).toBe('deepseek');
+    });
+    test('resolves api key from env var', () => {
+        Bun.env.TEST_PROVIDER_KEY = 'sk-from-env';
+        const cfgWithEnv = {
+            ...testConfig,
+            fallback_chain: ['env_provider'],
+            providers: {
+                ...testConfig.providers,
+                env_provider: {
+                    base_url: 'https://env.example.com/v1',
+                    api_key_env: 'TEST_PROVIDER_KEY',
+                    models: [],
+                },
+            },
+        };
+        initModelRouter(cfgWithEnv);
+        const provider = resolveProvider('unknown-model');
+        expect(provider).toBeDefined();
+        expect(provider.api_key).toBe('sk-from-env');
+        delete Bun.env.TEST_PROVIDER_KEY;
+    });
+});

package/dist/tests/model-gateway/routes.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/tests/model-gateway/routes.test.js

@@ -0,0 +1,83 @@
+import { describe, test, expect, beforeAll } from 'bun:test';
+import Fastify from 'fastify';
+import { modelGatewayRoutes } from '../../src/model-gateway/routes.js';
+import { initModelRouter } from '../../src/model-gateway/router.js';
+const testConfig = {
+    default_model: 'deepseek-chat',
+    providers: {
+        deepseek: {
+            base_url: 'https://api.deepseek.com/v1',
+            api_key: 'sk-test',
+            models: ['deepseek-chat', 'deepseek-v4-pro'],
+        },
+        ollama: {
+            base_url: 'http://127.0.0.1:11434/v1',
+            api_key: 'ollama',
+            models: ['qwen3:14b'],
+        },
+    },
+    fallback_chain: ['deepseek', 'ollama'],
+    model_routing: {
+        deepseek: ['deepseek'],
+        qwen: ['ollama'],
+    },
+};
+describe('model gateway routes', () => {
+    let fastify;
+    beforeAll(async () => {
+        initModelRouter(testConfig);
+        fastify = Fastify({ logger: false });
+        await fastify.register(modelGatewayRoutes);
+        await fastify.ready();
+    });
+    test('GET /v1/models returns model list', async () => {
+        const resp = await fastify.inject({ method: 'GET', url: '/v1/models' });
+        expect(resp.statusCode).toBe(200);
+        const body = JSON.parse(resp.body);
+        expect(body.object).toBe('list');
+        expect(Array.isArray(body.data)).toBe(true);
+        expect(body.data.length).toBeGreaterThanOrEqual(2);
+        const modelIds = body.data.map((m) => m.id);
+        expect(modelIds).toContain('deepseek-chat');
+        expect(modelIds).toContain('qwen3:14b');
+    });
+    // 这些端点依赖 codexbar（外部进程，15s+），单元测试 skip，集成测试单独运行
+    test.skip('GET /model-gateway/health returns status', async () => {
+        const resp = await fastify.inject({ method: 'GET', url: '/model-gateway/health' });
+        expect([200, 500]).toContain(resp.statusCode);
+    });
+    test.skip('GET /model-gateway/quota returns quota data', async () => {
+        const resp = await fastify.inject({ method: 'GET', url: '/model-gateway/quota' });
+        expect([200, 500]).toContain(resp.statusCode);
+    });
+    test('POST /v1/chat/completions requires messages', async () => {
+        const resp = await fastify.inject({
+            method: 'POST',
+            url: '/v1/chat/completions',
+            payload: { model: 'test' },
+        });
+        expect(resp.statusCode).toBe(400);
+        const body = JSON.parse(resp.body);
+        expect(body.error).toBeDefined();
+    });
+    test('POST /v1/responses requires input', async () => {
+        const resp = await fastify.inject({
+            method: 'POST',
+            url: '/v1/responses',
+            payload: { model: 'test' },
+        });
+        expect(resp.statusCode).toBe(400);
+    });
+    test('POST /v1/chat/completions resolves provider and attempts call', async () => {
+        const resp = await fastify.inject({
+            method: 'POST',
+            url: '/v1/chat/completions',
+            payload: {
+                model: 'deepseek-chat',
+                messages: [{ role: 'user', content: 'Hello' }],
+            },
+        });
+        // Any status that isn't a crash (500) is acceptable in test env
+        expect([200, 401, 502, 503]).toContain(resp.statusCode);
+    });
+});