@pkent/aigateway 1.1.0 → 1.2.0

package/README.md

@@ -2,8 +2,8 @@
 
 A provider-neutral LLM client library. One class routes chat / vision /
 streaming requests to OpenAI, Anthropic, Qwen (Alibaba DashScope), GLM (Zhipu
-AI), or OpenRouter based on the model name, and normalizes every response into a
-single shape.
+AI), OpenRouter, or an OpenAI-compatible gateway (`aibroker`) based on the model
+name, and normalizes every response into a single shape.
 
 No server, no `.env` — all configuration is passed into the constructor.
 
@@ -49,7 +49,7 @@ new AIGateway(model, key, options?)
 
 | Option      | Type   | Applies to      | Default          | Description |
 |-------------|--------|-----------------|------------------|-------------|
-| `baseURL`   | string | all             | provider default | Override the upstream endpoint (regional endpoints, proxies, self-hosted). |
+| `baseURL`   | string | all             | provider default | Override the upstream endpoint (regional endpoints, proxies, self-hosted). **Required** for `aibroker` (no default). |
 | `maxTokens` | number | all             | unset (see below)| Default max output tokens; a per-call `maxTokens` overrides it. |
 | `timeout`   | number | all             | unset (SDK default) | Default per-call request timeout in ms; a per-call `timeout` overrides it. |
 | `referer`   | string | OpenRouter only | omitted          | Sent as the `HTTP-Referer` attribution header. |
@@ -71,7 +71,8 @@ AIGateway.providers()
 //   { id: 'anthropic',  prefix: 'anthropic/' },
 //   { id: 'qwen',       prefix: 'qwen/' },
 //   { id: 'glm',        prefix: 'glm/' },
-//   { id: 'openai',     prefix: 'openai/' }
+//   { id: 'openai',     prefix: 'openai/' },
+//   { id: 'aibroker',   prefix: 'aibroker/' }
 // ]
 ```
 
@@ -137,10 +138,25 @@ catch-all — an unknown or missing segment throws `AIGatewayError`
 | `qwen/<model>`                 | `qwen`       | `<model>`      | OpenAI-compatible. Default base URL `https://dashscope-intl.aliyuncs.com/compatible-mode/v1`. |
 | `glm/<model>`                  | `glm`        | `<model>`      | OpenAI-compatible. Default base URL `https://open.bigmodel.cn/api/paas/v4`. |
 | `openrouter/<vendor>/<model>`  | `openrouter` | `<vendor>/<model>` | Only the `openrouter/` segment is stripped, leaving OpenRouter's native id. Default base URL `https://openrouter.ai/api/v1`. |
+| `aibroker/<remainder>`         | `aibroker`   | `<remainder>`  | OpenAI-compatible gateway meta-provider. Only the `aibroker/` segment is stripped; the remainder is forwarded verbatim and the gateway does its own routing. **Requires** `baseURL` (no default) — the constructor throws `AIGatewayError` (`code: 'missing_base_url'`) without it. |
 
 OpenRouter is opt-in: send `openrouter/anthropic/claude-opus-4.8` to route
 through OpenRouter, or `anthropic/claude-opus-4.8` to hit Anthropic directly.
 
+`aibroker` is a gateway meta-provider: it forwards to *any* OpenAI-compatible
+gateway you point it at, so it embeds no host and **requires** a `baseURL`. Only
+the leading `aibroker/` segment is stripped — everything after it is sent
+upstream as the model id, letting the gateway do its own routing
+(`aibroker/openai/chatgpt-5.5` → `openai/chatgpt-5.5`,
+`aibroker/openrouter/openai/chatgpt-5.5` → `openrouter/openai/chatgpt-5.5`). The
+key is passed straight through as the Bearer for the gateway.
+
+```js
+const g = new AIGateway('aibroker/openai/chatgpt-5.5', GATEWAY_TOKEN, {
+  baseURL: 'https://your-gateway/v1',
+});
+```
+
 **Adding a provider:** drop a module into [src/providers/](src/providers/)
 exporting `{ id, prefix, matches, create }` (where `matches` tests the
 `<id>/` segment) and register it in

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pkent/aigateway",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "A provider-neutral LLM client library for OpenAI, Anthropic, Qwen, GLM, and OpenRouter",
   "main": "src/index.js",
   "type": "module",
@@ -9,7 +9,7 @@
   },
   "scripts": {
     "test": "node --test",
-    "prepublishOnly": "node --test test/aigateway.test.js test/cache-hints.test.js test/usage-normalization.test.js test/cancellation.test.js"
+    "prepublishOnly": "node --test test/aigateway.test.js test/cache-hints.test.js test/usage-normalization.test.js test/cancellation.test.js test/aibroker.test.js"
   },
   "files": [
     "src",

package/src/providers/aibroker.js

@@ -0,0 +1,22 @@
+import { createOpenAICompatibleProvider } from './openaiCompatible.js';
+import { AIGatewayError } from '../errors.js';
+
+// "aibroker/<provider>/<model>" routes to an OpenAI-compatible gateway. The
+// "aibroker/" segment is stripped (existing stripProviderPrefix) and the rest is
+// forwarded verbatim as the model — the gateway does its own routing. The
+// caller MUST supply the gateway URL via options.baseURL; there is no default.
+export default {
+  id: 'aibroker',
+  prefix: 'aibroker/',
+  matches: model => typeof model === 'string' && model.startsWith('aibroker/'),
+  create({ apiKey, baseURL, client }) {
+    if (!client && !baseURL) {
+      throw new AIGatewayError(
+        'The "aibroker" provider requires a baseURL (the gateway URL), e.g. ' +
+          'new AIGateway(model, key, { baseURL: "https://your-gateway/v1" })',
+        { code: 'missing_base_url' },
+      );
+    }
+    return createOpenAICompatibleProvider({ id: 'aibroker', apiKey, baseURL, client });
+  },
+};

package/src/providers/registry.js

@@ -3,10 +3,11 @@ import anthropic from './anthropic.js';
 import qwen from './qwen.js';
 import glm from './glm.js';
 import openai from './openai.js';
+import aibroker from './aibroker.js';
 
 // Every model is addressed as "<provider>/<model>". The provider segments are
 // disjoint, so resolution order does not matter; there is no catch-all.
-const ENTRIES = [openrouter, anthropic, qwen, glm, openai];
+const ENTRIES = [openrouter, anthropic, qwen, glm, openai, aibroker];
 
 export function resolveProviderEntry(model) {
   for (const entry of ENTRIES) {