fix: show custom BYOK models in selector (#1321)

Co-authored-by: Letta Code <noreply@letta.com>
Co-authored-by: cpacker <packercharles@gmail.com>
This commit is contained in:
Sarah Wooders
2026-03-10 14:17:51 -07:00
committed by GitHub
parent 4c9f63c4e2
commit 4451e50281
2 changed files with 132 additions and 26 deletions

View File

@@ -0,0 +1,44 @@
import { describe, expect, test } from "bun:test";
import {
buildByokProviderAliases,
isByokHandleForSelector,
} from "../../cli/components/ModelSelector";
describe("ModelSelector custom BYOK provider detection", () => {
test("treats connected custom OpenAI providers as BYOK", () => {
const aliases = buildByokProviderAliases([
{
name: "openai-sarah",
provider_type: "openai",
},
]);
expect(aliases["openai-sarah"]).toBe("openai");
expect(isByokHandleForSelector("openai-sarah/gpt-5.4-fast", aliases)).toBe(
true,
);
});
test("maps custom OpenAI provider handles back to base openai handles", () => {
const aliases = buildByokProviderAliases([
{
name: "openai-sarah",
provider_type: "openai",
},
]);
const provider = "openai-sarah";
const model = "gpt-5.4-fast";
const baseProvider = aliases[provider];
expect(`${baseProvider}/${model}`).toBe("openai/gpt-5.4-fast");
});
test("preserves existing lc-* aliases", () => {
const aliases = buildByokProviderAliases([]);
expect(isByokHandleForSelector("lc-openai/gpt-5.4-fast", aliases)).toBe(
true,
);
});
});