feat: add API key and custom provider configuration (#4)

* feat: add API key and custom provider configuration

Previously, model setup only offered OAuth login. This adds:

- API key configuration for 17 built-in providers (OpenAI, Anthropic,
  Google, Mistral, Groq, xAI, OpenRouter, etc.)
- Custom provider setup via models.json (for Ollama, vLLM, LM Studio,
  proxies, or any OpenAI/Anthropic/Google-compatible endpoint)
- Interactive prompts with smart defaults and auto-detection of models
- Verification flow that probes endpoints and provides actionable tips
- Doctor diagnostics for models.json path and missing apiKey warnings
- Dev environment fallback for running without dist/ build artifacts
- Unified auth flow: `feynman model login` now offers both API key
  and OAuth options (OAuth-only when a specific provider is given)

New files:
- src/model/models-json.ts: Read/write models.json with proper merging
- src/model/registry.ts: Centralized ModelRegistry creation with modelsJsonPath
- tests/models-json.test.ts: Unit tests for provider config upsert

* fix: harden runtime env and custom provider auth

---------

Co-authored-by: Advait Paliwal <advaitspaliwal@gmail.com>
This commit is contained in:
Mochamad Chairulridjal
2026-03-27 07:09:38 +07:00
committed by GitHub
parent dbd89d8e3d
commit 30d07246d1
13 changed files with 745 additions and 23 deletions

91
src/model/models-json.ts Normal file
View File

@@ -0,0 +1,91 @@
import { chmodSync, existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
import { dirname } from "node:path";
type ModelsJson = {
providers?: Record<string, Record<string, unknown>>;
};
function readModelsJson(modelsJsonPath: string): { ok: true; value: ModelsJson } | { ok: false; error: string } {
if (!existsSync(modelsJsonPath)) {
return { ok: true, value: { providers: {} } };
}
try {
const raw = readFileSync(modelsJsonPath, "utf8").trim();
if (!raw) {
return { ok: true, value: { providers: {} } };
}
const parsed = JSON.parse(raw) as unknown;
if (!parsed || typeof parsed !== "object") {
return { ok: false, error: `Invalid models.json (expected an object): ${modelsJsonPath}` };
}
return { ok: true, value: parsed as ModelsJson };
} catch (error) {
return {
ok: false,
error: `Failed to read models.json: ${error instanceof Error ? error.message : String(error)}`,
};
}
}
export function upsertProviderBaseUrl(
modelsJsonPath: string,
providerId: string,
baseUrl: string,
): { ok: true } | { ok: false; error: string } {
return upsertProviderConfig(modelsJsonPath, providerId, { baseUrl });
}
export type ProviderConfigPatch = {
baseUrl?: string;
apiKey?: string;
api?: string;
authHeader?: boolean;
headers?: Record<string, string>;
models?: Array<{ id: string }>;
};
export function upsertProviderConfig(
modelsJsonPath: string,
providerId: string,
patch: ProviderConfigPatch,
): { ok: true } | { ok: false; error: string } {
const loaded = readModelsJson(modelsJsonPath);
if (!loaded.ok) {
return loaded;
}
const value: ModelsJson = loaded.value;
const providers: Record<string, Record<string, unknown>> = {
...(value.providers && typeof value.providers === "object" ? value.providers : {}),
};
const currentProvider =
providers[providerId] && typeof providers[providerId] === "object" ? providers[providerId] : {};
const nextProvider: Record<string, unknown> = { ...currentProvider };
if (patch.baseUrl !== undefined) nextProvider.baseUrl = patch.baseUrl;
if (patch.apiKey !== undefined) nextProvider.apiKey = patch.apiKey;
if (patch.api !== undefined) nextProvider.api = patch.api;
if (patch.authHeader !== undefined) nextProvider.authHeader = patch.authHeader;
if (patch.headers !== undefined) nextProvider.headers = patch.headers;
if (patch.models !== undefined) nextProvider.models = patch.models;
providers[providerId] = nextProvider;
const next: ModelsJson = { ...value, providers };
try {
mkdirSync(dirname(modelsJsonPath), { recursive: true });
writeFileSync(modelsJsonPath, JSON.stringify(next, null, 2) + "\n", "utf8");
// models.json can contain API keys/headers; default to user-only permissions.
try {
chmodSync(modelsJsonPath, 0o600);
} catch {
// ignore permission errors (best-effort)
}
return { ok: true };
} catch (error) {
return { ok: false, error: `Failed to write models.json: ${error instanceof Error ? error.message : String(error)}` };
}
}