fix(auth): add ollama to env API key map

This commit is contained in:
Baek, JH 2026-01-26 21:45:38 -08:00
parent 9daa846457
commit ab6d4c529a
2 changed files with 16 additions and 0 deletions

View File

@ -285,6 +285,7 @@ export function resolveEnvApiKey(provider: string): EnvApiKeyResult | null {
venice: "VENICE_API_KEY",
mistral: "MISTRAL_API_KEY",
opencode: "OPENCODE_API_KEY",
ollama: "OLLAMA_API_KEY",
};
const envVar = envMap[normalized];
if (!envVar) return null;

View File

@ -1,5 +1,6 @@
import { describe, expect, it } from "vitest";
import { resolveImplicitProviders } from "./models-config.providers.js";
import { resolveEnvApiKey } from "./model-auth.js";
import { mkdtempSync } from "node:fs";
import { join } from "node:path";
import { tmpdir } from "node:os";
@ -12,4 +13,18 @@ describe("Ollama provider", () => {
// Ollama requires explicit configuration via OLLAMA_API_KEY env var or profile
expect(providers?.ollama).toBeUndefined();
});
it("should resolve OLLAMA_API_KEY env var for ollama provider", () => {
const prev = process.env.OLLAMA_API_KEY;
try {
process.env.OLLAMA_API_KEY = "ollama";
const result = resolveEnvApiKey("ollama");
expect(result).not.toBeNull();
expect(result!.apiKey).toBe("ollama");
expect(result!.source).toContain("OLLAMA_API_KEY");
} finally {
if (prev === undefined) delete process.env.OLLAMA_API_KEY;
else process.env.OLLAMA_API_KEY = prev;
}
});
});