agentdog 1.1.0

package/dist/ai-gateway/passthrough.js

@@ -0,0 +1,81 @@
+import { sendUpstream } from './upstream/provider-client.js';
+import { anthropicError, mapUpstreamStatus } from './anthropic-errors.js';
+export const UPSTREAM_HEADER_TIMEOUT_MS = 30_000;
+export const STREAM_WATCHDOG_MS = 5 * 60_000;
+export async function passthroughMessages(req, res, provider, model) {
+    const body = { ...req.body, model };
+    const headers = {
+        'content-type': 'application/json',
+        'x-api-key': provider.apiKey,
+        'anthropic-version': req.headers['anthropic-version'] || '2023-06-01',
+        ...(provider.headers || {}),
+    };
+    if (req.headers['anthropic-beta']) {
+        headers['anthropic-beta'] = req.headers['anthropic-beta'];
+    }
+    const abort = new AbortController();
+    res.on('close', () => abort.abort());
+    const url = trimTrailingSlash(provider.baseUrl) + '/v1/messages';
+    let upstream;
+    try {
+        upstream = await sendUpstream({
+            url,
+            method: 'POST',
+            headers,
+            body: JSON.stringify(body),
+            timeoutMs: UPSTREAM_HEADER_TIMEOUT_MS,
+            signal: abort.signal,
+        });
+    }
+    catch (error) {
+        if (abort.signal.aborted) {
+            return; // 客户端已断连
+        }
+        const message = error.message;
+        const isTimeout = /timeout/i.test(message);
+        anthropicError(res, isTimeout ? 504 : 502, 'api_error', `upstream request failed: ${message}`);
+        return;
+    }
+    // 直通通道例外：上游响应头一到即转发（含错误响应，格式原生一致）
+    const contentType = upstream.headers['content-type'] || '';
+    if (!contentType.includes('text/event-stream')) {
+        const text = await upstream.text();
+        if (res.writableEnded) {
+            return;
+        }
+        res.status(upstream.status).set('content-type', contentType || 'application/json').send(text);
+        return;
+    }
+    // 流式：SSE 原样 pipe 回传 + 无字节 watchdog
+    res.status(200).set({
+        'content-type': 'text/event-stream',
+        'cache-control': 'no-cache',
+        connection: 'keep-alive',
+    });
+    let lastByteAt = Date.now();
+    const watchdog = setInterval(() => {
+        if (Date.now() - lastByteAt > STREAM_WATCHDOG_MS) {
+            clearInterval(watchdog);
+            res.write(`event: error\ndata: ${JSON.stringify({ type: 'error', error: { type: 'api_error', message: 'upstream stream stalled' } })}\n\n`);
+            res.end();
+            abort.abort();
+        }
+    }, 30_000);
+    try {
+        await upstream.stream(line => {
+            lastByteAt = Date.now();
+            res.write(line + '\n');
+        });
+    }
+    finally {
+        clearInterval(watchdog);
+        if (!res.writableEnded) {
+            res.end();
+        }
+    }
+}
+export function trimTrailingSlash(s) {
+    return s.endsWith('/') ? s.slice(0, -1) : s;
+}
+export { mapUpstreamStatus };
+//# sourceMappingURL=passthrough.js.map

package/dist/ai-gateway/passthrough.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"passthrough.js","sourceRoot":"","sources":["../../src/ai-gateway/passthrough.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAE1E,MAAM,CAAC,MAAM,0BAA0B,GAAG,MAAM,CAAC;AACjD,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,GAAG,MAAM,CAAC;AAE7C,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,GAAY,EACZ,GAAa,EACb,QAA0B,EAC1B,KAAa;IAEb,MAAM,IAAI,GAAG,EAAE,GAAI,GAAG,CAAC,IAAgC,EAAE,KAAK,EAAE,CAAC;IAEjE,MAAM,OAAO,GAA2B;QACtC,cAAc,EAAE,kBAAkB;QAClC,WAAW,EAAE,QAAQ,CAAC,MAAM;QAC5B,mBAAmB,EAAG,GAAG,CAAC,OAAO,CAAC,mBAAmB,CAAY,IAAI,YAAY;QACjF,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,EAAE,CAAC;KAC5B,CAAC;IACF,IAAI,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;QAClC,OAAO,CAAC,gBAAgB,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAW,CAAC;IACtE,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,eAAe,EAAE,CAAC;IACpC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;IAErC,MAAM,GAAG,GAAG,iBAAiB,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,cAAc,CAAC;IAEjE,IAAI,QAAQ,CAAC;IACb,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,YAAY,CAAC;YAC5B,GAAG;YACH,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1B,SAAS,EAAE,0BAA0B;YACrC,MAAM,EAAE,KAAK,CAAC,MAAM;SACrB,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO,CAAC,SAAS;QACnB,CAAC;QACD,MAAM,OAAO,GAAI,KAAe,CAAC,OAAO,CAAC;QACzC,MAAM,SAAS,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3C,cAAc,CACZ,GAAG,EACH,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EACrB,WAAW,EACX,4BAA4B,OAAO,EAAE,CACtC,CAAC;QACF,OAAO;IACT,CAAC;IAED,kCAAkC;IAClC,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;IAE3D,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;YACtB,OAAO;QACT,CAAC;QACD,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE,WAAW,IAAI,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9F,OAAO;IACT,CAAC;IAED,mCAAmC;IACnC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC;QAClB,cAAc,EAAE,mBAAmB;QACnC,eAAe,EAAE,UAAU;QAC3B,UAAU,EAAE,YAAY;KACzB,CAAC,CAAC;IACH,IAAI,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC5B,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE;QAChC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,GAAG,kBAAkB,EAAE,CAAC;YACjD,aAAa,CAAC,QAAQ,CAAC,CAAC;YACxB,GAAG,CAAC,KAAK,CAAC,uBAAuB,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,yBAAyB,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;YAC5I,GAAG,CAAC,GAAG,EAAE,CAAC;YACV,KAAK,CAAC,KAAK,EAAE,CAAC;QAChB,CAAC;IACH,CAAC,EAAE,MAAM,CAAC,CAAC;IAEX,IAAI,CAAC;QACH,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAC3B,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACxB,GAAG,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,aAAa,CAAC,QAAQ,CAAC,CAAC;QACxB,IAAI,CAAC,GAAG,CAAC,aAAa,EAAE,CAAC;YACvB,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,CAAS;IACzC,OAAO,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9C,CAAC;AAED,OAAO,EAAE,iBAAiB,EAAE,CAAC"}

package/dist/ai-gateway/passthrough.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=passthrough.test.d.ts.map

package/dist/ai-gateway/passthrough.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"passthrough.test.d.ts","sourceRoot":"","sources":["../../src/ai-gateway/passthrough.test.ts"],"names":[],"mappings":""}

package/dist/ai-gateway/passthrough.test.js

@@ -0,0 +1,108 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+// mock 上游客户端
+vi.mock('./upstream/provider-client.js', () => ({
+    sendUpstream: vi.fn(),
+}));
+import { sendUpstream } from './upstream/provider-client.js';
+import { passthroughMessages } from './passthrough.js';
+const provider = {
+    id: '1', slug: 'anthropic-official', dialect: 'anthropic',
+    baseUrl: 'https://api.anthropic.com', apiKey: 'sk-upstream', enabled: true,
+};
+function mockReqRes(body, headers = {}) {
+    const writes = [];
+    const res = {
+        statusCode: 0,
+        headers: {},
+        writableEnded: false,
+        sentText: undefined,
+        on: vi.fn(),
+        status(c) { this.statusCode = c; return this; },
+        set(h) { Object.assign(this.headers, h); return this; },
+        send(t) { this.sentText = t; this.writableEnded = true; return this; },
+        json(b) { this.sentText = JSON.stringify(b); this.writableEnded = true; return this; },
+        write(chunk) { writes.push(chunk); return true; },
+        end() { this.writableEnded = true; return this; },
+    };
+    return { req: { body, headers }, res, writes };
+}
+beforeEach(() => {
+    vi.mocked(sendUpstream).mockReset();
+});
+describe('passthroughMessages', () => {
+    it('转发 body 剥离 slug 前缀、保留未知原生字段、带上游认证头', async () => {
+        vi.mocked(sendUpstream).mockResolvedValue({
+            status: 200,
+            headers: { 'content-type': 'application/json' },
+            text: async () => '{"id":"msg_1"}',
+            stream: async () => { },
+        });
+        const { req, res } = mockReqRes({
+            model: 'anthropic-official:claude-sonnet-4',
+            max_tokens: 100,
+            thinking: { type: 'enabled', budget_tokens: 50 },
+            metadata: { user_id: 'u1' },
+        }, { 'anthropic-version': '2023-06-01' });
+        await passthroughMessages(req, res, provider, 'claude-sonnet-4');
+        const call = vi.mocked(sendUpstream).mock.calls[0];
+        expect(call[0].url).toBe('https://api.anthropic.com/v1/messages');
+        expect(call[0].headers['x-api-key']).toBe('sk-upstream');
+        const sentBody = JSON.parse(call[0].body);
+        expect(sentBody.model).toBe('claude-sonnet-4');
+        expect(sentBody.thinking).toEqual({ type: 'enabled', budget_tokens: 50 });
+        expect(sentBody.metadata).toEqual({ user_id: 'u1' });
+        expect(res.statusCode).toBe(200);
+        expect(res.sentText).toBe('{"id":"msg_1"}');
+    });
+    it('上游 401 原样透传状态码与响应体', async () => {
+        vi.mocked(sendUpstream).mockResolvedValue({
+            status: 401,
+            headers: { 'content-type': 'application/json' },
+            text: async () => '{"type":"error","error":{"type":"authentication_error","message":"invalid x-api-key"}}',
+            stream: async () => { },
+        });
+        const { req, res } = mockReqRes({ model: 'x:y', max_tokens: 1 });
+        await passthroughMessages(req, res, provider, 'claude-sonnet-4');
+        expect(res.statusCode).toBe(401);
+        expect(JSON.parse(res.sentText).error.message).toBe('invalid x-api-key');
+    });
+    it('流式响应原样 pipe 回传 SSE 行', async () => {
+        const sseLines = ['event: message_start', 'data: {"type":"message_start"}', '', 'event: message_stop', ''];
+        vi.mocked(sendUpstream).mockResolvedValue({
+            status: 200,
+            headers: { 'content-type': 'text/event-stream' },
+            text: async () => { throw new Error('not called'); },
+            stream: async (onLine) => { sseLines.forEach(onLine); },
+        });
+        const { req, res, writes } = mockReqRes({ model: 'x:y', stream: true, max_tokens: 1 });
+        await passthroughMessages(req, res, provider, 'claude-sonnet-4');
+        expect(res.statusCode).toBe(200);
+        expect(res.headers['content-type']).toBe('text/event-stream');
+        expect(writes.join('')).toBe(sseLines.map(l => l + '\n').join(''));
+        expect(res.writableEnded).toBe(true);
+    });
+    it('baseUrl 带尾斜杠不产生双斜杠', async () => {
+        vi.mocked(sendUpstream).mockResolvedValue({
+            status: 200, headers: {}, text: async () => '{}', stream: async () => { },
+        });
+        const { req, res } = mockReqRes({ model: 'x:y', max_tokens: 1 });
+        await passthroughMessages(req, res, { ...provider, baseUrl: 'https://api.anthropic.com/' }, 'm');
+        expect(vi.mocked(sendUpstream).mock.calls[0][0].url).toBe('https://api.anthropic.com/v1/messages');
+    });
+    it('上游连接失败返回 502 Anthropic 错误体', async () => {
+        vi.mocked(sendUpstream).mockRejectedValue(new Error('ECONNREFUSED'));
+        const { req, res } = mockReqRes({ model: 'x:y', max_tokens: 1 });
+        await passthroughMessages(req, res, provider, 'm');
+        expect(res.statusCode).toBe(502);
+        expect(JSON.parse(res.sentText).error.message).toContain('ECONNREFUSED');
+    });
+    it('provider 自定义 headers 合并进上游请求', async () => {
+        vi.mocked(sendUpstream).mockResolvedValue({
+            status: 200, headers: {}, text: async () => '{}', stream: async () => { },
+        });
+        const { req, res } = mockReqRes({ model: 'x:y', max_tokens: 1 });
+        await passthroughMessages(req, res, { ...provider, headers: { 'x-custom': 'v' } }, 'm');
+        expect(vi.mocked(sendUpstream).mock.calls[0][0].headers['x-custom']).toBe('v');
+    });
+});
+//# sourceMappingURL=passthrough.test.js.map

package/dist/ai-gateway/passthrough.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"passthrough.test.js","sourceRoot":"","sources":["../../src/ai-gateway/passthrough.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAE9D,aAAa;AACb,EAAE,CAAC,IAAI,CAAC,+BAA+B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9C,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AACJ,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAGvD,MAAM,QAAQ,GAAqB;IACjC,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,oBAAoB,EAAE,OAAO,EAAE,WAAW;IACzD,OAAO,EAAE,2BAA2B,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,IAAI;CAC3E,CAAC;AAEF,SAAS,UAAU,CAAC,IAAS,EAAE,UAAkC,EAAE;IACjE,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,GAAG,GAAQ;QACf,UAAU,EAAE,CAAC;QACb,OAAO,EAAE,EAA4B;QACrC,aAAa,EAAE,KAAK;QACpB,QAAQ,EAAE,SAA+B;QACzC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE;QACX,MAAM,CAAC,CAAS,IAAI,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QACvD,GAAG,CAAC,CAAM,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,CAAS,IAAI,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QAC9E,IAAI,CAAC,CAAM,IAAI,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QAC3F,KAAK,CAAC,KAAa,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QACzD,GAAG,KAAK,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;KAClD,CAAC;IACF,OAAO,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,OAAO,EAAS,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;AACxD,CAAC;AAED,UAAU,CAAC,GAAG,EAAE;IACd,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,EAAE,CAAC;AACtC,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,MAAM,EAAE,GAAG;YACX,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,gBAAgB;YAClC,MAAM,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;SAChB,CAAC,CAAC;QACV,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC;YAC9B,KAAK,EAAE,oCAAoC;YAC3C,UAAU,EAAE,GAAG;YACf,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,aAAa,EAAE,EAAE,EAAE;YAChD,QAAQ,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SAC5B,EAAE,EAAE,mBAAmB,EAAE,YAAY,EAAE,CAAC,CAAC;QAE1C,MAAM,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAEjE,MAAM,IAAI,GAAG,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;QACpD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;QAClE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAK,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC/C,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,CAAC;QAC1E,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,MAAM,EAAE,GAAG;YACX,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,wFAAwF;YAC1G,MAAM,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;SAChB,CAAC,CAAC;QACV,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QAEjE,MAAM,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAEjE,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAS,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,QAAQ,GAAG,CAAC,sBAAsB,EAAE,gCAAgC,EAAE,EAAE,EAAE,qBAAqB,EAAE,EAAE,CAAC,CAAC;QAC3G,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,MAAM,EAAE,GAAG;YACX,OAAO,EAAE,EAAE,cAAc,EAAE,mBAAmB,EAAE;YAChD,IAAI,EAAE,KAAK,IAAI,EAAE,GAAG,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACpD,MAAM,EAAE,KAAK,EAAE,MAA2B,EAAE,EAAE,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;SACtE,CAAC,CAAC;QACV,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QAEvF,MAAM,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;QAEjE,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACnE,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;SAClE,CAAC,CAAC;QACV,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QACjE,MAAM,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,4BAA4B,EAAE,EAAE,GAAG,CAAC,CAAC;QACjG,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uCAAuC,CAAC,CAAC;IACtG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;QACrE,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QACjE,MAAM,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,CAAC,CAAC;QACnD,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAS,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;SAClE,CAAC,CAAC;QACV,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QACjE,MAAM,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;QACxF,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/ai-gateway/upstream/model-lister.d.ts

@@ -0,0 +1,13 @@
+import { AIProviderConfig } from '../../types/index.js';
+export declare function openaiChatUrl(baseUrl: string): string;
+export declare function openaiModelsUrl(baseUrl: string): string;
+export declare function anthropicMessagesUrl(baseUrl: string): string;
+export declare function geminiGenerateUrl(baseUrl: string, model: string, stream: boolean): string;
+export declare function geminiModelsUrl(baseUrl: string): string;
+export declare function listUpstreamModels(provider: AIProviderConfig): Promise<{
+    ok: boolean;
+    status: number;
+    models: string[];
+    message: string;
+}>;
+//# sourceMappingURL=model-lister.d.ts.map

package/dist/ai-gateway/upstream/model-lister.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-lister.d.ts","sourceRoot":"","sources":["../../../src/ai-gateway/upstream/model-lister.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAaxD,wBAAgB,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAErD;AAED,wBAAgB,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAEvD;AAED,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAE5D;AAED,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,GAAG,MAAM,CAGzF;AAED,wBAAgB,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAEvD;AAED,wBAAsB,kBAAkB,CAAC,QAAQ,EAAE,gBAAgB,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,OAAO,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,EAAE,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC,CA+BhJ"}

package/dist/ai-gateway/upstream/model-lister.js

@@ -0,0 +1,61 @@
+import { sendUpstream } from './provider-client.js';
+function trimSlash(s) {
+    return s.endsWith('/') ? s.slice(0, -1) : s;
+}
+// openai 兼容：baseUrl 末尾已含 /v1 则直接拼，否则补 /v1
+function openaiBase(baseUrl) {
+    const base = trimSlash(baseUrl);
+    return /\/v\d+$/.test(base) ? base : base + '/v1';
+}
+export function openaiChatUrl(baseUrl) {
+    return openaiBase(baseUrl) + '/chat/completions';
+}
+export function openaiModelsUrl(baseUrl) {
+    return openaiBase(baseUrl) + '/models';
+}
+export function anthropicMessagesUrl(baseUrl) {
+    return trimSlash(baseUrl) + '/v1/messages';
+}
+export function geminiGenerateUrl(baseUrl, model, stream) {
+    const action = stream ? 'streamGenerateContent?alt=sse' : 'generateContent';
+    return `${trimSlash(baseUrl)}/v1beta/models/${encodeURIComponent(model)}:${action}`;
+}
+export function geminiModelsUrl(baseUrl) {
+    return `${trimSlash(baseUrl)}/v1beta/models`;
+}
+export async function listUpstreamModels(provider) {
+    try {
+        let url;
+        let headers;
+        if (provider.dialect === 'openai') {
+            url = openaiModelsUrl(provider.baseUrl);
+            headers = { authorization: `Bearer ${provider.apiKey}`, ...(provider.headers || {}) };
+        }
+        else if (provider.dialect === 'anthropic') {
+            url = `${trimSlash(provider.baseUrl)}/v1/models`;
+            headers = { 'x-api-key': provider.apiKey, 'anthropic-version': '2023-06-01', ...(provider.headers || {}) };
+        }
+        else {
+            url = geminiModelsUrl(provider.baseUrl);
+            headers = { 'x-goog-api-key': provider.apiKey, ...(provider.headers || {}) };
+        }
+        const resp = await sendUpstream({ url, method: 'GET', headers, timeoutMs: 15_000 });
+        const text = await resp.text();
+        if (resp.status !== 200) {
+            return { ok: false, status: resp.status, models: [], message: text.slice(0, 500) };
+        }
+        const json = JSON.parse(text);
+        let models = [];
+        if (Array.isArray(json.data)) {
+            models = json.data.map((m) => m.id).filter(Boolean); // openai
+        }
+        else if (Array.isArray(json.models)) {
+            models = json.models.map((m) => (m.name || '').replace(/^models\//, '')).filter(Boolean); // gemini
+        }
+        return { ok: true, status: 200, models, message: 'ok' };
+    }
+    catch (error) {
+        return { ok: false, status: 0, models: [], message: error.message };
+    }
+}
+//# sourceMappingURL=model-lister.js.map

package/dist/ai-gateway/upstream/model-lister.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-lister.js","sourceRoot":"","sources":["../../../src/ai-gateway/upstream/model-lister.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAEpD,SAAS,SAAS,CAAC,CAAS;IAC1B,OAAO,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAC9C,CAAC;AAED,0CAA0C;AAC1C,SAAS,UAAU,CAAC,OAAe;IACjC,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;IAChC,OAAO,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,KAAK,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,OAAe;IAC3C,OAAO,UAAU,CAAC,OAAO,CAAC,GAAG,mBAAmB,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,OAAe;IAC7C,OAAO,UAAU,CAAC,OAAO,CAAC,GAAG,SAAS,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,OAAe;IAClD,OAAO,SAAS,CAAC,OAAO,CAAC,GAAG,cAAc,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,OAAe,EAAE,KAAa,EAAE,MAAe;IAC/E,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC,iBAAiB,CAAC;IAC5E,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC,kBAAkB,kBAAkB,CAAC,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;AACtF,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,OAAe;IAC7C,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC,gBAAgB,CAAC;AAC/C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,QAA0B;IACjE,IAAI,CAAC;QACH,IAAI,GAAW,CAAC;QAChB,IAAI,OAA+B,CAAC;QACpC,IAAI,QAAQ,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAClC,GAAG,GAAG,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACxC,OAAO,GAAG,EAAE,aAAa,EAAE,UAAU,QAAQ,CAAC,MAAM,EAAE,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;QACxF,CAAC;aAAM,IAAI,QAAQ,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;YAC5C,GAAG,GAAG,GAAG,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC;YACjD,OAAO,GAAG,EAAE,WAAW,EAAE,QAAQ,CAAC,MAAM,EAAE,mBAAmB,EAAE,YAAY,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;QAC7G,CAAC;aAAM,CAAC;YACN,GAAG,GAAG,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACxC,OAAO,GAAG,EAAE,gBAAgB,EAAE,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;QAC/E,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;QACpF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAC/B,IAAI,IAAI,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACxB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;QACrF,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,MAAM,GAAa,EAAE,CAAC;QAC1B,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAQ,SAAS;QAC5E,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACtC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;QAC1G,CAAC;QACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC1D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,EAAG,KAAe,CAAC,OAAO,EAAE,CAAC;IACjF,CAAC;AACH,CAAC"}

package/dist/ai-gateway/upstream/provider-client.d.ts

@@ -0,0 +1,20 @@
+export interface UpstreamRequest {
+    url: string;
+    method: 'POST' | 'GET';
+    headers: Record<string, string>;
+    body?: string;
+    timeoutMs: number;
+    signal?: AbortSignal;
+}
+export interface UpstreamResponse {
+    status: number;
+    headers: Record<string, string>;
+    text(): Promise<string>;
+    stream(onLine: (line: string) => void): Promise<void>;
+}
+export declare class UpstreamTimeoutError extends Error {
+    constructor(url: string);
+}
+export declare function sendUpstream(req: UpstreamRequest): Promise<UpstreamResponse>;
+export declare function parseSSEStream(body: ReadableStream<Uint8Array>, onLine: (line: string) => void): Promise<void>;
+//# sourceMappingURL=provider-client.d.ts.map

package/dist/ai-gateway/upstream/provider-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-client.d.ts","sourceRoot":"","sources":["../../../src/ai-gateway/upstream/provider-client.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,eAAe;IAC9B,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,GAAG,KAAK,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChC,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;IACxB,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD;AAED,qBAAa,oBAAqB,SAAQ,KAAK;gBACjC,GAAG,EAAE,MAAM;CAIxB;AAED,wBAAsB,YAAY,CAAC,GAAG,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAyClF;AAGD,wBAAsB,cAAc,CAClC,IAAI,EAAE,cAAc,CAAC,UAAU,CAAC,EAChC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,GAC7B,OAAO,CAAC,IAAI,CAAC,CAyBf"}

package/dist/ai-gateway/upstream/provider-client.js

@@ -0,0 +1,76 @@
+export class UpstreamTimeoutError extends Error {
+    constructor(url) {
+        super(`upstream timeout: no response headers within limit from ${url}`);
+        this.name = 'UpstreamTimeoutError';
+    }
+}
+export async function sendUpstream(req) {
+    const timeoutController = new AbortController();
+    const timer = setTimeout(() => timeoutController.abort(), req.timeoutMs);
+    // 外部 abort 联动
+    const onExternalAbort = () => timeoutController.abort();
+    req.signal?.addEventListener('abort', onExternalAbort, { once: true });
+    let response;
+    try {
+        response = await fetch(req.url, {
+            method: req.method,
+            headers: req.headers,
+            body: req.body,
+            signal: timeoutController.signal,
+        });
+    }
+    catch (error) {
+        // 区分超时 abort 与外部 abort
+        if (timeoutController.signal.aborted && !req.signal?.aborted) {
+            throw new UpstreamTimeoutError(req.url);
+        }
+        throw error;
+    }
+    finally {
+        clearTimeout(timer);
+        req.signal?.removeEventListener('abort', onExternalAbort);
+    }
+    const headers = {};
+    response.headers.forEach((value, key) => { headers[key] = value; });
+    return {
+        status: response.status,
+        headers,
+        text: () => response.text(),
+        stream: async (onLine) => {
+            if (!response.body) {
+                return;
+            }
+            // 响应头已到，后续流读取不再受 timeoutMs 约束（无字节 watchdog 由调用方实现）
+            await parseSSEStream(response.body, onLine);
+        },
+    };
+}
+// 增量逐行解析：按 \n 分行缓冲，跨 chunk 行不撕裂
+export async function parseSSEStream(body, onLine) {
+    const reader = body.getReader();
+    const decoder = new TextDecoder();
+    let buffer = '';
+    try {
+        for (;;) {
+            const { done, value } = await reader.read();
+            if (done) {
+                break;
+            }
+            buffer += decoder.decode(value, { stream: true });
+            let newlineIndex;
+            while ((newlineIndex = buffer.indexOf('\n')) !== -1) {
+                const line = buffer.slice(0, newlineIndex).replace(/\r$/, '');
+                buffer = buffer.slice(newlineIndex + 1);
+                onLine(line);
+            }
+        }
+        buffer += decoder.decode();
+        if (buffer) {
+            onLine(buffer.replace(/\r$/, ''));
+        }
+    }
+    finally {
+        reader.releaseLock();
+    }
+}
+//# sourceMappingURL=provider-client.js.map

package/dist/ai-gateway/upstream/provider-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-client.js","sourceRoot":"","sources":["../../../src/ai-gateway/upstream/provider-client.ts"],"names":[],"mappings":"AAkBA,MAAM,OAAO,oBAAqB,SAAQ,KAAK;IAC7C,YAAY,GAAW;QACrB,KAAK,CAAC,2DAA2D,GAAG,EAAE,CAAC,CAAC;QACxE,IAAI,CAAC,IAAI,GAAG,sBAAsB,CAAC;IACrC,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,GAAoB;IACrD,MAAM,iBAAiB,GAAG,IAAI,eAAe,EAAE,CAAC;IAChD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,KAAK,EAAE,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;IACzE,cAAc;IACd,MAAM,eAAe,GAAG,GAAG,EAAE,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;IACxD,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,eAAe,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAEvE,IAAI,QAA6B,CAAC;IAClC,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE;YAC9B,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,MAAM,EAAE,iBAAiB,CAAC,MAAM;SACjC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,uBAAuB;QACvB,IAAI,iBAAiB,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;YAC7D,MAAM,IAAI,oBAAoB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC1C,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACpB,GAAG,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;IAC5D,CAAC;IAED,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAEpE,OAAO;QACL,MAAM,EAAE,QAAQ,CAAC,MAAM;QACvB,OAAO;QACP,IAAI,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE;QAC3B,MAAM,EAAE,KAAK,EAAE,MAA8B,EAAE,EAAE;YAC/C,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnB,OAAO;YACT,CAAC;YACD,mDAAmD;YACnD,MAAM,cAAc,CAAC,QAAQ,CAAC,IAAkC,EAAE,MAAM,CAAC,CAAC;QAC5E,CAAC;KACF,CAAC;AACJ,CAAC;AAED,gCAAgC;AAChC,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,IAAgC,EAChC,MAA8B;IAE9B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,CAAC;QACH,SAAS,CAAC;YACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM;YACR,CAAC;YACD,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,IAAI,YAAoB,CAAC;YACzB,OAAO,CAAC,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBACpD,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC9D,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;gBACxC,MAAM,CAAC,IAAI,CAAC,CAAC;YACf,CAAC;QACH,CAAC;QACD,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QAC3B,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;AACH,CAAC"}

package/dist/ai-gateway/upstream/provider-client.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=provider-client.test.d.ts.map

package/dist/ai-gateway/upstream/provider-client.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-client.test.d.ts","sourceRoot":"","sources":["../../../src/ai-gateway/upstream/provider-client.test.ts"],"names":[],"mappings":""}

package/dist/ai-gateway/upstream/provider-client.test.js

@@ -0,0 +1,69 @@
+import { describe, it, expect, vi, afterEach } from 'vitest';
+import { parseSSEStream, sendUpstream } from './provider-client.js';
+function sseChunks(lines) {
+    // 模拟跨 chunk 半行的字节流切分
+    return lines.map(l => new TextEncoder().encode(l));
+}
+describe('parseSSEStream', () => {
+    it('逐帧回调完整 data 行，跨 chunk 行不撕裂', async () => {
+        const lines = [];
+        const stream = new ReadableStream({
+            start(controller) {
+                controller.enqueue(new TextEncoder().encode('event: message\n'));
+                controller.enqueue(new TextEncoder().encode('data: {"a"')); // 半行撕裂点
+                controller.enqueue(new TextEncoder().encode(':1}\n\n'));
+                controller.enqueue(new TextEncoder().encode('data: {"b":2}\n\n'));
+                controller.close();
+            },
+        });
+        await parseSSEStream(stream, line => lines.push(line));
+        expect(lines).toEqual(['event: message', 'data: {"a":1}', '', 'data: {"b":2}', '']);
+    });
+    it('空行与注释行原样回调（由调用方按需忽略）', async () => {
+        const lines = [];
+        const stream = new ReadableStream({
+            start(controller) {
+                controller.enqueue(new TextEncoder().encode(': ping\n\ndata: x\n\n'));
+                controller.close();
+            },
+        });
+        await parseSSEStream(stream, line => lines.push(line));
+        expect(lines).toEqual([': ping', '', 'data: x', '']);
+    });
+});
+describe('sendUpstream', () => {
+    afterEach(() => { vi.unstubAllGlobals(); });
+    it('返回状态/头与 text()', async () => {
+        vi.stubGlobal('fetch', vi.fn().mockResolvedValue({
+            status: 200,
+            headers: new Headers({ 'content-type': 'application/json' }),
+            text: async () => '{"ok":true}',
+            body: null,
+        }));
+        const res = await sendUpstream({
+            url: 'https://up.test/v1/messages', method: 'POST',
+            headers: { 'x-api-key': 'k' }, body: '{}', timeoutMs: 5000,
+        });
+        expect(res.status).toBe(200);
+        expect(res.headers['content-type']).toContain('application/json');
+        expect(await res.text()).toBe('{"ok":true}');
+    });
+    it('响应头超时抛 TimeoutError 语义错误', async () => {
+        vi.stubGlobal('fetch', vi.fn().mockImplementation((_url, init) => new Promise((_resolve, reject) => {
+            init.signal.addEventListener('abort', () => reject(new Error('aborted')));
+        })));
+        await expect(sendUpstream({
+            url: 'https://up.test', method: 'POST', headers: {}, timeoutMs: 50,
+        })).rejects.toThrow(/timeout/i);
+    });
+    it('透传外部 signal abort', async () => {
+        const ac = new AbortController();
+        vi.stubGlobal('fetch', vi.fn().mockImplementation((_url, init) => new Promise((_resolve, reject) => {
+            init.signal.addEventListener('abort', () => reject(new Error('aborted')));
+        })));
+        const p = sendUpstream({ url: 'https://up.test', method: 'POST', headers: {}, timeoutMs: 5000, signal: ac.signal });
+        ac.abort();
+        await expect(p).rejects.toThrow();
+    });
+});
+//# sourceMappingURL=provider-client.test.js.map

package/dist/ai-gateway/upstream/provider-client.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-client.test.js","sourceRoot":"","sources":["../../../src/ai-gateway/upstream/provider-client.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAEpE,SAAS,SAAS,CAAC,KAAe;IAChC,qBAAqB;IACrB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,MAAM,MAAM,GAAG,IAAI,cAAc,CAAa;YAC5C,KAAK,CAAC,UAAU;gBACd,UAAU,CAAC,OAAO,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,CAAC;gBACjE,UAAU,CAAC,OAAO,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,QAAQ;gBACpE,UAAU,CAAC,OAAO,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;gBACxD,UAAU,CAAC,OAAO,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,CAAC;gBAClE,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC;SACF,CAAC,CAAC;QACH,MAAM,cAAc,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,gBAAgB,EAAE,eAAe,EAAE,EAAE,EAAE,eAAe,EAAE,EAAE,CAAC,CAAC,CAAC;IACtF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,MAAM,MAAM,GAAG,IAAI,cAAc,CAAa;YAC5C,KAAK,CAAC,UAAU;gBACd,UAAU,CAAC,OAAO,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,CAAC;gBACtE,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC;SACF,CAAC,CAAC;QACH,MAAM,cAAc,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAE5C,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;QAC9B,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAC/C,MAAM,EAAE,GAAG;YACX,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC;YAC5D,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC,aAAa;YAC/B,IAAI,EAAE,IAAI;SACX,CAAC,CAAC,CAAC;QACJ,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC;YAC7B,GAAG,EAAE,6BAA6B,EAAE,MAAM,EAAE,MAAM;YAClD,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI;SAC3D,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAC/C,CAAC,IAAY,EAAE,IAAS,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE;YAC5D,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC5E,CAAC,CAAC,CACH,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,YAAY,CAAC;YACxB,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE;SACnE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;QACjC,MAAM,EAAE,GAAG,IAAI,eAAe,EAAE,CAAC;QACjC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAC/C,CAAC,IAAY,EAAE,IAAS,EAAE,EAAE,CAAC,IAAI,OAAO,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE;YAC5D,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC5E,CAAC,CAAC,CACH,CAAC,CAAC;QACH,MAAM,CAAC,GAAG,YAAY,CAAC,EAAE,GAAG,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;QACpH,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/cli/cli-main.d.ts

@@ -0,0 +1,7 @@
+#!/usr/bin/env node
+/**
+ * MCPDog CLI - Main entry point for the command-line interface
+ * Supports server management, protocol detection, configuration optimization, etc.
+ */
+export {};
+//# sourceMappingURL=cli-main.d.ts.map

package/dist/cli/cli-main.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli-main.d.ts","sourceRoot":"","sources":["../../src/cli/cli-main.ts"],"names":[],"mappings":";AACA;;;GAGG"}

package/dist/cli/cli-main.js

@@ -0,0 +1,116 @@
+#!/usr/bin/env node
+/**
+ * MCPDog CLI - Main entry point for the command-line interface
+ * Supports server management, protocol detection, configuration optimization, etc.
+ */
+import { parseArgs } from 'util';
+import { CLICommandRouter } from './cli-router.js';
+import { CLIUtils } from './cli-utils.js';
+async function main() {
+    try {
+        // Parse command line arguments - only parse global options, let subcommands parse their specific options
+        const { values, positionals } = parseArgs({
+            args: process.argv.slice(2),
+            options: {
+                help: { type: 'boolean', short: 'h' },
+                version: { type: 'boolean', short: 'v' },
+                config: { type: 'string', short: 'c' },
+                verbose: { type: 'boolean' },
+                json: { type: 'boolean' },
+                'no-color': { type: 'boolean' },
+                'dashboard-port': { type: 'string' },
+                'mcp-http-port': { type: 'string' },
+                'web-port': { type: 'string' }, // deprecated, kept for backward compatibility
+                'gateway-port': { type: 'string' }, // AI API gateway port (daemon start)
+                'daemon-port': { type: 'string' },
+                'pid-file': { type: 'string' },
+                'stdio-only': { type: 'boolean' },
+                'http-only': { type: 'boolean' },
+                'no-dashboard': { type: 'boolean' },
+                port: { type: 'string', short: 'p' },
+                // Config command options
+                endpoint: { type: 'string' },
+                transport: { type: 'string' },
+                timeout: { type: 'string' },
+                retries: { type: 'string' },
+                description: { type: 'string' },
+                // Add command options
+                'auto-detect': { type: 'boolean' },
+                headers: { type: 'string' },
+                args: { type: 'string' },
+                yes: { type: 'boolean', short: 'y' },
+                // Audit command options
+                performance: { type: 'boolean' },
+                security: { type: 'boolean' },
+                compliance: { type: 'boolean' },
+                export: { type: 'string' },
+                // Detect command options
+                all: { type: 'boolean' },
+                detailed: { type: 'boolean' },
+                'no-add': { type: 'boolean' },
+                // Diagnose command options
+                'health-check': { type: 'boolean' },
+                fix: { type: 'boolean' },
+                // Optimize command options
+                apply: { type: 'boolean' },
+                preview: { type: 'boolean' },
+                // Daemon autostart command options
+                enable: { type: 'boolean' },
+                disable: { type: 'boolean' }
+            },
+            allowPositionals: true,
+            strict: false
+        });
+        // Set up CLI environment
+        CLIUtils.setupEnvironment({
+            verbose: Boolean(values.verbose),
+            noColor: Boolean(values['no-color']),
+            json: Boolean(values.json)
+        });
+        // Handle global options
+        if (values.version) {
+            CLIUtils.showVersion();
+            process.exit(0);
+        }
+        if (values.help && positionals.length === 0) {
+            CLIUtils.showGlobalHelp();
+            process.exit(0);
+        }
+        // Get command and arguments
+        const [command, ...args] = positionals;
+        // If no command is provided, default to 'proxy' (auto-start mode)
+        const finalCommand = command || 'proxy';
+        const finalArgs = command ? args : [];
+        // In auto mode (no command specified), suppress all output to avoid MCP client errors
+        if (!command) {
+            // Don't output anything in auto mode to avoid MCP client parsing errors
+            // The proxy command will handle all the necessary startup logic silently
+        }
+        if (!command && values.help) {
+            CLIUtils.showGlobalHelp();
+            process.exit(0);
+        }
+        // Create router and execute command
+        const router = new CLICommandRouter(values.config);
+        await router.executeCommand(finalCommand, finalArgs, values);
+    }
+    catch (error) {
+        CLIUtils.error('CLI execution failed:', error.message);
+        if (CLIUtils.isVerbose()) {
+            console.error(error.stack);
+        }
+        process.exit(1);
+    }
+}
+// Handle uncaught exceptions
+process.on('uncaughtException', (error) => {
+    CLIUtils.error('Uncaught exception:', error.message);
+    process.exit(1);
+});
+process.on('unhandledRejection', (reason, promise) => {
+    CLIUtils.error('Unhandled rejection at:', promise, 'reason:', reason);
+    process.exit(1);
+});
+// Start the CLI
+main();
+//# sourceMappingURL=cli-main.js.map