shumengya/sproutclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(coding-agent): add provider display names

Browse Source 
closes #3956
This commit is contained in:
Mario Zechner
2026-04-30 00:10:38 +02:00
parent b79e7f8058
commit cf5ec23240
10 changed files with 110 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
packages/coding-agent/test/model-registry.test.ts
View File

@@ -841,6 +841,56 @@ describe("ModelRegistry", () => {
});
describe("dynamic provider lifecycle", () => {
test("getProviderDisplayName resolves registered, OAuth, built-in, and fallback names", () => {
const registry = ModelRegistry.create(authStorage, modelsJsonPath);
expect(registry.getProviderDisplayName("openai")).toBe("OpenAI");
expect(registry.getProviderDisplayName("github-copilot")).toBe("GitHub Copilot");
expect(registry.getProviderDisplayName("unknown-provider")).toBe("unknown-provider");
registry.registerProvider("named-provider", {
name: "Named Provider",
baseUrl: "https://provider.test/v1",
apiKey: "TEST_KEY",
api: "openai-completions",
models: [
{
id: "demo-model",
name: "Demo Model",
reasoning: false,
input: ["text"],
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
contextWindow: 128000,
maxTokens: 4096,
},
],
});
expect(registry.getProviderDisplayName("named-provider")).toBe("Named Provider");
registry.registerProvider("oauth-provider", {
baseUrl: "https://provider.test/v1",
api: "openai-completions",
oauth: {
name: "OAuth Provider",
login: async () => ({ access: "access", refresh: "refresh", expires: Date.now() + 60_000 }),
refreshToken: async (credentials) => credentials,
getApiKey: (credentials) => credentials.access,
},
models: [
{
id: "demo-model",
name: "Demo Model",
reasoning: false,
input: ["text"],
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
contextWindow: 128000,
maxTokens: 4096,
},
],
});
expect(registry.getProviderDisplayName("oauth-provider")).toBe("OAuth Provider");
});
test("failed registerProvider does not persist invalid streamSimple config", () => {
const registry = ModelRegistry.create(authStorage, modelsJsonPath);

5
packages/coding-agent/test/oauth-selector.test.ts
View File

@@ -3,8 +3,9 @@ import stripAnsi from "strip-ansi";
import { afterEach, beforeAll, beforeEach, describe, expect, it } from "vitest";
import { AuthStorage } from "../src/core/auth-storage.js";
import { KeybindingsManager } from "../src/core/keybindings.js";
import { BUILT_IN_PROVIDER_DISPLAY_NAMES } from "../src/core/provider-display-names.js";
import { OAuthSelectorComponent } from "../src/modes/interactive/components/oauth-selector.js";
import { getApiKeyProviderDisplayName, isApiKeyLoginProvider } from "../src/modes/interactive/interactive-mode.js";
import { isApiKeyLoginProvider } from "../src/modes/interactive/interactive-mode.js";
import { initTheme } from "../src/modes/interactive/theme/theme.js";
const originalOpenAiApiKey = process.env.OPENAI_API_KEY;
@@ -31,7 +32,7 @@ describe("OAuthSelectorComponent", () => {
const builtInProviderIds = new Set(["anthropic", "github-copilot", "amazon-bedrock", "openai"]);
expect(isApiKeyLoginProvider("anthropic", oauthProviderIds, builtInProviderIds)).toBe(true);
expect(getApiKeyProviderDisplayName("anthropic")).toBe("Anthropic");
expect(BUILT_IN_PROVIDER_DISPLAY_NAMES.anthropic).toBe("Anthropic");
expect(isApiKeyLoginProvider("openai", oauthProviderIds, builtInProviderIds)).toBe(true);
expect(isApiKeyLoginProvider("github-copilot", oauthProviderIds, builtInProviderIds)).toBe(false);
expect(isApiKeyLoginProvider("amazon-bedrock", oauthProviderIds, builtInProviderIds)).toBe(true);