owen/opencode-proxy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

maybe fix

Browse Source
This commit is contained in:
owen
2026-07-22 00:11:29 -05:00
parent 8731cc0307
commit 4b6fbb4356
2 changed files with 47 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/proxy.test.js
+27
View File
@@ -69,12 +69,39 @@ describe('proxy', () => {
expect(JSON.parse(fetchImpl.mock.calls[0][1].body)).toEqual(body);
});
it('normalizes max_completion_tokens for OpenCode Go providers', async () => {
const fetchImpl = vi.fn().mockResolvedValue(response('{"id":"completion-1"}'));
const app = createProxyApp({ keys: ['key'], fetchImpl });
await request(app).post('/oai/v1/chat/completions').send({
model: 'kimi-k3', messages: [{ role: 'user', content: 'Hi' }],
max_completion_tokens: 4096, stream: true, stream_options: { include_usage: true },
}).expect(200);
expect(JSON.parse(fetchImpl.mock.calls[0][1].body)).toEqual({
model: 'kimi-k3', messages: [{ role: 'user', content: 'Hi' }],
max_tokens: 4096, stream: true, stream_options: { include_usage: true },
});
});
it('prefers max_tokens when both token limit field names are supplied', async () => {
const fetchImpl = vi.fn().mockResolvedValue(response('{"id":"completion-1"}'));
const app = createProxyApp({ keys: ['key'], fetchImpl });
await request(app).post('/oai/v1/chat/completions').send({
model: 'm', max_tokens: 2048, max_completion_tokens: 4096,
}).expect(200);
expect(JSON.parse(fetchImpl.mock.calls[0][1].body)).toEqual({ model: 'm', max_tokens: 2048 });
});
it('enforces the configured context and IP limits and records cost', async () => {
const fetchImpl = vi.fn()
.mockResolvedValueOnce(response(JSON.stringify({ id: 'c', cost: 0 })))
.mockResolvedValue(response(JSON.stringify({ id: 'c', cost: 10 })));
const app = createProxyApp({ keys: ['key'], fetchImpl, maxContextTokens: 100 });
await request(app).post('/oai/v1/chat/completions').set('X-Real-IP', '198.51.100.1').send({ model: 'm', max_tokens: 101 }).expect(400);
await request(app).post('/oai/v1/chat/completions').set('X-Real-IP', '198.51.100.1').send({ model: 'm', max_completion_tokens: 101 }).expect(400);
await request(app).post('/oai/v1/chat/completions').set('X-Real-IP', '198.51.100.1').send({ model: 'm', max_tokens: 1 }).expect(200);
await request(app).post('/oai/v1/chat/completions').set('X-Real-IP', '198.51.100.1').send({ model: 'm', max_tokens: 1 }).expect(200);
await request(app).post('/oai/v1/chat/completions').set('X-Real-IP', '198.51.100.1').send({ model: 'm', max_tokens: 1 }).expect(429);