fix(ai): qualified auth-gateway model ids

Advertised provider-qualified model IDs and removed duplicate resolver-map entries from the model catalog response.

Added regression coverage for shared IDs across multiple providers.

Fixes #6170
This commit is contained in:
roboomp
2026-07-21 21:57:05 +00:00
parent 89d6a8f6d1
commit 505b6fdc38
3 changed files with 55 additions and 7 deletions
+4
View File
@@ -2,6 +2,10 @@
## [Unreleased]
### Fixed
- Fixed `auth-gateway` model listings exposing duplicate, ambiguous bare IDs by advertising one provider-qualified routing ID per upstream model ([#6170](https://github.com/can1357/oh-my-pi/issues/6170)).
## [17.0.6] - 2026-07-20
### Fixed
+13 -7
View File
@@ -726,13 +726,19 @@ async function handleCredentialsCheck(storage: AuthStorage, signal: AbortSignal)
}
function handleModelsList(opts: AuthGatewayBootOptions): Response {
const list = opts.listModels ? Array.from(opts.listModels()) : [];
const data = list.map(model => ({
id: model.id,
object: "model" as const,
owned_by: model.provider,
api: model.api,
}));
const seen = new Set<string>();
const data: Array<{ id: string; object: "model"; owned_by: string; api: Api }> = [];
for (const model of opts.listModels?.() ?? []) {
const id = `${model.provider}/${model.id}`;
if (seen.has(id)) continue;
seen.add(id);
data.push({
id,
object: "model",
owned_by: model.provider,
api: model.api,
});
}
return json(200, { object: "list", data });
}
@@ -0,0 +1,38 @@
import { expect, test } from "bun:test";
import * as fs from "node:fs/promises";
import * as os from "node:os";
import * as path from "node:path";
import { startAuthGateway } from "@oh-my-pi/pi-ai/auth-gateway";
import { AuthStorage } from "@oh-my-pi/pi-ai/auth-storage";
import { createMockModel } from "@oh-my-pi/pi-ai/providers/mock";
test("model listing exposes one provider-qualified route per upstream model", async () => {
const dir = await fs.mkdtemp(path.join(os.tmpdir(), "gw-model-list-"));
const storage = await AuthStorage.create(path.join(dir, "auth.db"));
const anthropic = createMockModel({ provider: "anthropic", id: "shared-model" });
const devin = createMockModel({ provider: "devin", id: "shared-model" });
const handle = startAuthGateway({
bind: "127.0.0.1:0",
bearerTokens: [],
storage,
resolveModel: () => undefined,
listModels: () => [anthropic, anthropic, devin, devin],
version: "test",
});
try {
const response = await fetch(`${handle.url}/v1/models`);
expect(response.status).toBe(200);
expect(await response.json()).toEqual({
object: "list",
data: [
{ id: "anthropic/shared-model", object: "model", owned_by: "anthropic", api: "mock" },
{ id: "devin/shared-model", object: "model", owned_by: "devin", api: "mock" },
],
});
} finally {
await handle.close();
storage.close();
await fs.rm(dir, { recursive: true, force: true });
}
});