pi-provider-litellm 4.0.1 → 4.1.0

package/README.md

@@ -156,7 +156,7 @@ Provider fields:
 | `baseUrl` | `LITELLM_BASE_URL` for `litellm`; required for aliases | LiteLLM proxy URL, with or without `/v1`. Must be a full `http`/`https` URL; a provider with no resolvable base URL exposes no models (see [Model host enforcement](#model-host-enforcement)) |
 | `apiKey` | `LITELLM_API_KEY_HELPER`/`LITELLM_API_KEY` for `litellm`; required for aliases | Pi config value for this provider's key. Use `$ENV_VAR`, `${ENV_VAR}`, `!command`, or a literal key. Escape a literal `$` as `$$`. |
 | `headers` | `$LITELLM_HEADERS` for `litellm`; unset for aliases | JSON string env reference or inline object of request headers |
-| `displayName` | provider name | Label shown in Pi UI |
+| `displayName` | `LITELLM_DISPLAY_NAME` / `"LiteLLM"` for `litellm`; alias name for aliases | Label shown in Pi UI |
 | `enabled` | `true` | Set `false` to skip an alias |
 | `oidc` | unset | `litellm` only. Sign in directly with an OpenID Connect identity provider instead of the LiteLLM-hosted flows; see [Direct OIDC login](#direct-oidc-login) |
 | `allowInsecureHttp` | `false` | Set `true` to permit plaintext HTTP for this provider, for example `http://host.docker.internal`. Credentials and request data will not be encrypted. Loopback HTTP works without this setting. |
@@ -214,11 +214,12 @@ Native Messages authenticates with `x-api-key`; every transport carries the `x-l
 | Variable | Default | Effect |
 |---|---|---|
 | `LITELLM_API_KEY_HELPER` | unset | Command that prints a fresh LiteLLM bearer token. Takes precedence over `LITELLM_API_KEY`. The extension runs it while resolving request auth, and Pi's per-request auth path is uncached, so rotating/short-lived tokens stay fresh. |
+| `LITELLM_DISPLAY_NAME` | `LiteLLM` | Custom display name for the default `litellm` provider in the Pi UI. |
 | `LITELLM_HEADERS` | unset | JSON object of extra headers sent to LiteLLM provider, discovery, MCP, and Skills Gateway requests. Provider aliases can use it with `"headers": "$LITELLM_HEADERS"`. |
 | `LITELLM_GCLOUD_TOKEN_AUTH` | unset | If set to a non-empty value other than `0`, use Google Application Default Credentials as the LiteLLM bearer token source. This takes precedence over `LITELLM_API_KEY_HELPER` and `LITELLM_API_KEY` when no stored `/login litellm` credential exists. |
 | `GOOGLE_APPLICATION_CREDENTIALS` | Google default ADC path | Optional path to an ADC JSON file used by `LITELLM_GCLOUD_TOKEN_AUTH`. If unset, the extension checks the default gcloud ADC locations. |
 | `LITELLM_OFFLINE` | unset | If `1`, disable all model and MCP discovery, including post-login discovery; use cached models only when their stored canonical proxy root exactly matches the active credential root, including any path prefix. URL-standard host casing and default ports are canonicalized, but paths remain case-sensitive. |
-| `PI_OFFLINE` | unset | Pi documents it as disabling automatic network activity, including model catalog refreshes. Through Pi 1.0.0, any set value (including `0` or an empty string) disables startup model discovery and `/model` network refreshes. Run `/litellm-refresh`, or unset it, to discover; cached models can still be used offline. |
+| `PI_OFFLINE` | unset | Pi documents it as disabling automatic network activity, including model catalog refreshes. Through Pi 1.0.3, any set value (including `0` or an empty string) disables startup model discovery and `/model` network refreshes. Run `/litellm-refresh`, or unset it, to discover; cached models can still be used offline. |
 | `LITELLM_DISCOVERY_TIMEOUT_MS` | `5000` | Background and explicit discovery fetch timeout in ms; `0` disables automatic discovery |
 | `LITELLM_CLI_JWT_EXPIRATION_HOURS` | `24` | CLI SSO token lifetime fallback for older proxies whose poll response omits `expires_in`; mirror a non-default proxy setting locally |
 | `LITELLM_VERBOSE_DISCOVERY` | unset | If `1`, enable progress messages during model discovery (login, refresh, startup), including startup skip reasons and defaulted `/model/info` context windows. Progress messages are off by default |

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-provider-litellm",
-  "version": "4.0.1",
+  "version": "4.1.0",
   "description": "LiteLLM proxy provider extension for Pi",
   "type": "module",
   "license": "MIT",
@@ -67,8 +67,8 @@
   },
   "devDependencies": {
     "@biomejs/biome": "^2.4.15",
-    "@earendil-works/pi-ai": "^1.0.0",
-    "@earendil-works/pi-coding-agent": "^1.0.0",
+    "@earendil-works/pi-ai": "^1.0.3",
+    "@earendil-works/pi-coding-agent": "^1.0.3",
     "@kitlangton/terminal-control": "1.2.1",
     "@types/node": "^26.0.0",
     "@typescript/native-preview": "7.0.0-dev.20260707.2",

package/src/backend-identity.ts

@@ -1,4 +1,4 @@
-export const LITELLM_DISCOVERY_VERSION = 5 as const;
+export const LITELLM_DISCOVERY_VERSION = 6 as const;
 
 export type BackendFamily = "claude" | "deepseek" | "gemini" | "kimi" | "openai";
 

package/src/discover.ts

@@ -26,7 +26,12 @@ import {
   wireString,
 } from "./model-groups.js";
 import { type ProxyVersion, type ProxyVersionFloor, probeProxyVersion, proxyVersionAtLeast } from "./proxy-version.js";
-import { loadPublicCatalog, type PublicCatalog, type PublicCatalogRecord } from "./public-catalog.js";
+import {
+  loadPublicCatalog,
+  PI_AZURE_PROVIDER,
+  type PublicCatalog,
+  type PublicCatalogRecord,
+} from "./public-catalog.js";
 import { intersectThinkingLevelMaps } from "./thinking-levels.js";
 import type {
   DiscoveredModel,
@@ -443,11 +448,11 @@ function nativeMessagesCatalog(
     : {};
 }
 
-const ADAPTER_CATALOG_PROVIDERS: Readonly<Record<string, BuiltinProvider>> = {
+const ADAPTER_CATALOG_PROVIDERS: Readonly<Record<string, BuiltinProvider | undefined>> = {
   anthropic: "anthropic",
   claude: "anthropic",
-  azure: "azure-openai-responses",
-  azure_ai: "azure-openai-responses",
+  azure: toKnownProvider(PI_AZURE_PROVIDER),
+  azure_ai: toKnownProvider(PI_AZURE_PROVIDER),
   bedrock: "amazon-bedrock",
   bedrock_converse: "amazon-bedrock",
   chatgpt: "openai-codex",

package/src/index.ts

@@ -37,6 +37,7 @@ const PROVIDER_NAME = "litellm";
 const SETTINGS_KEY = "litellm";
 const ENV_BASE_URL = "LITELLM_BASE_URL";
 const ENV_API_KEY = "LITELLM_API_KEY";
+const ENV_DISPLAY_NAME = "LITELLM_DISPLAY_NAME";
 const GCLOUD_ADC_SOURCE = "gcloud ADC";
 const ENV_API_KEY_HELPER = "LITELLM_API_KEY_HELPER";
 const ENV_HEADERS = "LITELLM_HEADERS";
@@ -496,6 +497,7 @@ function getProviderDefinitions(settings: Record<string, unknown> | undefined):
   const rawProviders = settings?.providers && typeof settings.providers === "object" ? settings.providers : undefined;
   const providerSettings = rawProviders as Record<string, unknown> | undefined;
   const defaultSettings = normalizeProviderSettings(providerSettings?.[PROVIDER_NAME]);
+  const defaultDisplayName = cleanConfig(process.env[ENV_DISPLAY_NAME]) ?? "LiteLLM";
 
   const makeDefinition = (
     name: string,
@@ -503,7 +505,7 @@ function getProviderDefinitions(settings: Record<string, unknown> | undefined):
     isDefault: boolean,
   ): ProviderDefinition => ({
     name,
-    displayName: stringSetting(raw?.displayName) ?? (isDefault ? "LiteLLM" : name),
+    displayName: stringSetting(raw?.displayName) ?? (isDefault ? defaultDisplayName : name),
     baseUrl: stringSetting(raw?.baseUrl),
     apiKeyConfig: stringSetting(raw?.apiKey),
     headers: raw?.headers ?? (isDefault ? `$${ENV_HEADERS}` : undefined),

package/src/public-catalog.ts

@@ -8,6 +8,8 @@ export const MODELS_DEV_URL = "https://models.dev/api.json";
 const MODELS_DEV_CACHE_TTL_MS = 28 * 24 * 60 * 60 * 1000;
 const DEFAULT_TIMEOUT_MS = 5000;
 const KNOWN_PROVIDERS = new Set<string>(getProviders());
+// Pi 1.0.3 renamed its Azure catalog from `azure-openai-responses` to `azure`.
+export const PI_AZURE_PROVIDER = KNOWN_PROVIDERS.has("azure") ? "azure" : "azure-openai-responses";
 
 export interface PublicCatalogRecord {
   source: "models.dev" | "pi-vendor" | "pi-adapter";
@@ -72,8 +74,8 @@ const PROVIDER_ALIASES: Readonly<Record<string, readonly string[]>> = {
 
 const PI_PROVIDER_ALIASES: Readonly<Record<string, readonly string[]>> = {
   "amazon-bedrock": ["amazon-bedrock"],
-  azure: ["azure-openai-responses"],
-  azure_ai: ["azure-openai-responses"],
+  azure: [PI_AZURE_PROVIDER],
+  azure_ai: [PI_AZURE_PROVIDER],
   chatgpt: ["openai-codex"],
   "fireworks-ai": ["fireworks"],
   openai: ["openai"],