pi-provider-litellm 0.1.1

package/CHANGELOG.md

@@ -0,0 +1,31 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [Unreleased]
+
+## [0.1.1] - 2026-04-28
+
+### Added
+
+- `pi-package` keyword so the package surfaces in the [pi.dev/packages](https://pi.dev/packages) gallery.
+
+### Changed
+
+- Renamed package from `@balcsida/pi-provider-litellm` to unscoped `pi-provider-litellm` to match the `pi-provider-*` convention used by other extensions in the pi-mono ecosystem. The previous scoped name was published as 0.1.0 only.
+- README install instructions now use `pi install npm:pi-provider-litellm` (and `pi -e` for ephemeral trials), with the source-clone path kept as a fallback.
+
+## [0.1.0] - 2026-04-27
+
+### Added
+
+- Initial release.
+- Discovers models from a self-hosted LiteLLM proxy via `/model/info` with fallback to `/v1/models` on 401/403/404.
+- Sha256-fingerprinted disk cache with atomic writes.
+- `/login litellm` interactive login (prompts for base URL + API key).
+- `/litellm-refresh` slash command to force re-fetch.
+- `LITELLM_BASE_URL`, `LITELLM_API_KEY`, `LITELLM_OFFLINE`, `LITELLM_DISCOVERY_TIMEOUT_MS` environment variables.
+- `compat.cacheControlFormat: "anthropic"` for both `anthropic/*`-prefixed model IDs and bare Claude aliases (`claude-*`, `opus-*`, `sonnet-*`, `haiku-*`, case-insensitive). Common LiteLLM deployments alias Claude models without the `anthropic/` prefix; without this flag, prompt caching silently no-ops on Claude through the proxy.

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Dávid Balatoni
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,88 @@
+# pi-provider-litellm
+
+[![npm version](https://img.shields.io/npm/v/pi-provider-litellm.svg)](https://www.npmjs.com/package/pi-provider-litellm)
+
+LiteLLM proxy provider extension for [pi-mono](https://github.com/badlogic/pi-mono).
+
+Discovers models from a self-hosted LiteLLM proxy and registers them under the `litellm` provider. Supports `/login litellm` and `/litellm-refresh`. Tries `/model/info` first (admin endpoint with rich metadata), falls back to `/v1/models` (OpenAI-compatible) on 401/403/404.
+
+## Install
+
+```bash
+pi install npm:pi-provider-litellm
+```
+
+Pi fetches the package from npm and registers it. Add `-l` to install into project settings (`.pi/settings.json`) instead of global.
+
+To try it without installing (one-off, current run only):
+
+```bash
+pi -e npm:pi-provider-litellm
+```
+
+<details>
+<summary>Alternative: install from source</summary>
+
+```bash
+git clone https://github.com/balcsida/pi-provider-litellm.git ~/.pi/agent/extensions/pi-provider-litellm
+cd ~/.pi/agent/extensions/pi-provider-litellm
+npm install
+npm run build
+```
+
+</details>
+
+## Configure
+
+### Option A — environment variables
+
+```bash
+export LITELLM_BASE_URL="https://litellm.your-domain.com"
+export LITELLM_API_KEY="sk-..."
+```
+
+### Option B — interactive login
+
+Inside pi:
+
+```
+/login litellm
+```
+
+You'll be prompted for the base URL and API key. Credentials are persisted to `~/.pi/agent/auth.json`.
+
+## Use
+
+```
+/model litellm/anthropic/claude-3-5-sonnet
+```
+
+(Whatever model IDs your proxy exposes — see the discovered list at startup.)
+
+## Optional environment variables
+
+| Variable | Default | Effect |
+|---|---|---|
+| `LITELLM_OFFLINE` | unset | If `1`, skip discovery on this start; use cache only |
+| `LITELLM_DISCOVERY_TIMEOUT_MS` | `5000` | Discovery fetch timeout in ms; `0` to skip discovery |
+
+## Slash commands
+
+- `/litellm-refresh` — force re-fetch the model list, ignoring cache
+
+## Cache
+
+The model list is cached at `~/.pi/agent/litellm-models.json` with a sha256 fingerprint of the base URL + API key. Changing either invalidates the cache automatically.
+
+## Troubleshooting
+
+| Symptom | Likely cause |
+|---|---|
+| "no credentials" warning at startup | Env vars not set and no OAuth credential — run `/login litellm` |
+| "discovered no models" | Proxy returned an empty list — check pi's startup log for the actual response |
+| `/model/info` returning 401/403/404 | Expected behavior with virtual keys — extension falls back to `/v1/models` |
+| Discovery times out | Increase `LITELLM_DISCOVERY_TIMEOUT_MS` or set `LITELLM_OFFLINE=1` to fall back on cached models |
+
+## License
+
+MIT — see [LICENSE](./LICENSE).

package/dist/cache.d.ts

@@ -0,0 +1,6 @@
+import type { CacheFile } from "./types.js";
+export declare function fingerprint(apiKey: string): string;
+export declare function isCacheValid(cache: CacheFile | null, baseUrl: string, apiKey: string): boolean;
+export declare function readCache(path: string): Promise<CacheFile | null>;
+export declare function writeCache(path: string, cache: CacheFile): Promise<void>;
+//# sourceMappingURL=cache.d.ts.map

package/dist/cache.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache.d.ts","sourceRoot":"","sources":["../src/cache.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAE5C,wBAAgB,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAElD;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,SAAS,GAAG,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAG9F;AAED,wBAAsB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,CAevE;AAcD,wBAAsB,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAK9E"}

package/dist/cache.js

@@ -0,0 +1,47 @@
+import { createHash } from "node:crypto";
+import { mkdir, readFile, rename, writeFile } from "node:fs/promises";
+import { dirname } from "node:path";
+export function fingerprint(apiKey) {
+    return createHash("sha256").update(apiKey).digest("hex");
+}
+export function isCacheValid(cache, baseUrl, apiKey) {
+    if (!cache)
+        return false;
+    return cache.baseUrl === baseUrl && cache.apiKeyFingerprint === fingerprint(apiKey);
+}
+export async function readCache(path) {
+    let raw;
+    try {
+        raw = await readFile(path, "utf8");
+    }
+    catch {
+        return null;
+    }
+    let parsed;
+    try {
+        parsed = JSON.parse(raw);
+    }
+    catch {
+        return null;
+    }
+    if (!isCacheFileShape(parsed))
+        return null;
+    return parsed;
+}
+function isCacheFileShape(value) {
+    if (typeof value !== "object" || value === null)
+        return false;
+    const v = value;
+    return (typeof v.baseUrl === "string" &&
+        typeof v.apiKeyFingerprint === "string" &&
+        typeof v.fetchedAt === "number" &&
+        (v.source === "model_info" || v.source === "models_list") &&
+        Array.isArray(v.models));
+}
+export async function writeCache(path, cache) {
+    await mkdir(dirname(path), { recursive: true });
+    const tmp = `${path}.${process.pid}.${Date.now()}.tmp`;
+    await writeFile(tmp, JSON.stringify(cache, null, 2), "utf8");
+    await rename(tmp, path);
+}
+//# sourceMappingURL=cache.js.map

package/dist/cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache.js","sourceRoot":"","sources":["../src/cache.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAGpC,MAAM,UAAU,WAAW,CAAC,MAAc,EAAU;IAClD,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAAA,CAC1D;AAED,MAAM,UAAU,YAAY,CAAC,KAAuB,EAAE,OAAe,EAAE,MAAc,EAAW;IAC9F,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,OAAO,KAAK,CAAC,OAAO,KAAK,OAAO,IAAI,KAAK,CAAC,iBAAiB,KAAK,WAAW,CAAC,MAAM,CAAC,CAAC;AAAA,CACrF;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,IAAY,EAA6B;IACvE,IAAI,GAAW,CAAC;IAChB,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACrC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,MAAe,CAAC;IACpB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IAC3C,OAAO,MAAM,CAAC;AAAA,CACf;AAED,SAAS,gBAAgB,CAAC,KAAc,EAAsB;IAC5D,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC;IAC9D,MAAM,CAAC,GAAG,KAAgC,CAAC;IAC3C,OAAO,CACL,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ;QAC7B,OAAO,CAAC,CAAC,iBAAiB,KAAK,QAAQ;QACvC,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ;QAC/B,CAAC,CAAC,CAAC,MAAM,KAAK,YAAY,IAAI,CAAC,CAAC,MAAM,KAAK,aAAa,CAAC;QACzD,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CACxB,CAAC;AAAA,CACH;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,IAAY,EAAE,KAAgB,EAAiB;IAC9E,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAChD,MAAM,GAAG,GAAG,GAAG,IAAI,IAAI,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;IACvD,MAAM,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IAC7D,MAAM,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;AAAA,CACzB"}

package/dist/discover.d.ts

@@ -0,0 +1,6 @@
+import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
+import type { DiscoveryOptions, DiscoveryResult } from "./types.js";
+export declare function normalizeBaseUrl(input: string): string;
+export declare function buildCompat(modelId: string): ProviderModelConfig["compat"];
+export declare function discoverModels(baseUrl: string, apiKey: string, options?: DiscoveryOptions): Promise<DiscoveryResult>;
+//# sourceMappingURL=discover.d.ts.map

package/dist/discover.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"discover.d.ts","sourceRoot":"","sources":["../src/discover.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AACzE,OAAO,KAAK,EACV,gBAAgB,EAChB,eAAe,EAKhB,MAAM,YAAY,CAAC;AAMpB,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAEtD;AASD,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAK1E;AA4ED,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,gBAAqB,GAC7B,OAAO,CAAC,eAAe,CAAC,CAoB1B"}

package/dist/discover.js

@@ -0,0 +1,113 @@
+const DEFAULT_TIMEOUT_MS = 5000;
+const DEFAULT_CONTEXT_WINDOW = 128_000;
+const DEFAULT_MAX_TOKENS = 16_384;
+export function normalizeBaseUrl(input) {
+    return input.replace(/\/+$/, "").replace(/\/v1\/?$/i, "");
+}
+// Matches both the conventional `anthropic/...` prefix and bare aliases that
+// LiteLLM deployments commonly assign to Anthropic-backed routes (e.g.
+// `opus-4.7`, `sonnet-4.6`, `haiku-4.5`, `claude-3-5-sonnet`). Without the
+// `cacheControlFormat: "anthropic"` flag, pi never relays cache_control markers
+// through the proxy, so prompt caching silently no-ops on Claude models.
+const ANTHROPIC_MODEL_PATTERN = /^(anthropic\/|claude|opus|sonnet|haiku)/i;
+export function buildCompat(modelId) {
+    if (ANTHROPIC_MODEL_PATTERN.test(modelId)) {
+        return { supportsStore: false, cacheControlFormat: "anthropic" };
+    }
+    return { supportsStore: false };
+}
+function withTimeout(timeoutMs, signal) {
+    const controller = new AbortController();
+    const onAbort = () => controller.abort(signal?.reason);
+    if (signal) {
+        if (signal.aborted)
+            controller.abort(signal.reason);
+        else
+            signal.addEventListener("abort", onAbort, { once: true });
+    }
+    const timer = setTimeout(() => controller.abort(new Error(`Timed out after ${timeoutMs}ms`)), timeoutMs);
+    return {
+        signal: controller.signal,
+        cancel: () => {
+            clearTimeout(timer);
+            signal?.removeEventListener("abort", onAbort);
+        },
+    };
+}
+async function fetchJson(url, apiKey, options) {
+    const timeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+    const { signal, cancel } = withTimeout(timeoutMs, options.signal);
+    try {
+        const response = await fetch(url, {
+            headers: { Authorization: `Bearer ${apiKey}`, Accept: "application/json" },
+            signal,
+        });
+        if (!response.ok)
+            return { ok: false, status: response.status };
+        const data = (await response.json());
+        return { ok: true, data };
+    }
+    finally {
+        cancel();
+    }
+}
+function mapFromModelInfo(entry) {
+    const id = entry.model_name;
+    if (!id)
+        return undefined;
+    const info = entry.model_info ?? {};
+    if (info.mode && info.mode !== "chat")
+        return undefined;
+    return {
+        id,
+        name: id,
+        reasoning: info.supports_reasoning ?? false,
+        input: info.supports_vision ? ["text", "image"] : ["text"],
+        cost: {
+            input: (info.input_cost_per_token ?? 0) * 1_000_000,
+            output: (info.output_cost_per_token ?? 0) * 1_000_000,
+            cacheRead: (info.cache_read_input_token_cost ?? 0) * 1_000_000,
+            cacheWrite: (info.cache_creation_input_token_cost ?? 0) * 1_000_000,
+        },
+        contextWindow: info.max_input_tokens ?? DEFAULT_CONTEXT_WINDOW,
+        maxTokens: info.max_output_tokens ?? DEFAULT_MAX_TOKENS,
+        compat: buildCompat(id),
+    };
+}
+function mapFromModelsList(entry) {
+    const id = entry.id;
+    if (!id)
+        return undefined;
+    return {
+        id,
+        name: `${id} (no metadata)`,
+        reasoning: false,
+        input: ["text"],
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+        contextWindow: DEFAULT_CONTEXT_WINDOW,
+        maxTokens: DEFAULT_MAX_TOKENS,
+        compat: buildCompat(id),
+    };
+}
+export async function discoverModels(baseUrl, apiKey, options = {}) {
+    const base = normalizeBaseUrl(baseUrl);
+    const infoResult = await fetchJson(`${base}/model/info`, apiKey, options);
+    if (infoResult.ok) {
+        const models = (infoResult.data.data ?? [])
+            .map(mapFromModelInfo)
+            .filter((m) => m !== undefined);
+        return { source: "model_info", models };
+    }
+    if (![401, 403, 404].includes(infoResult.status)) {
+        throw new Error(`/model/info returned ${infoResult.status}`);
+    }
+    const listResult = await fetchJson(`${base}/v1/models`, apiKey, options);
+    if (!listResult.ok) {
+        throw new Error(`/v1/models returned ${listResult.status}`);
+    }
+    const models = (listResult.data.data ?? [])
+        .map(mapFromModelsList)
+        .filter((m) => m !== undefined);
+    return { source: "models_list", models };
+}
+//# sourceMappingURL=discover.js.map

package/dist/discover.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"discover.js","sourceRoot":"","sources":["../src/discover.ts"],"names":[],"mappings":"AAUA,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAChC,MAAM,sBAAsB,GAAG,OAAO,CAAC;AACvC,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAElC,MAAM,UAAU,gBAAgB,CAAC,KAAa,EAAU;IACtD,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;AAAA,CAC3D;AAED,6EAA6E;AAC7E,uEAAuE;AACvE,2EAA2E;AAC3E,gFAAgF;AAChF,yEAAyE;AACzE,MAAM,uBAAuB,GAAG,0CAA0C,CAAC;AAE3E,MAAM,UAAU,WAAW,CAAC,OAAe,EAAiC;IAC1E,IAAI,uBAAuB,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QAC1C,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,kBAAkB,EAAE,WAAW,EAAE,CAAC;IACnE,CAAC;IACD,OAAO,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;AAAA,CACjC;AAED,SAAS,WAAW,CAAC,SAAiB,EAAE,MAAoB,EAA+C;IACzG,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvD,IAAI,MAAM,EAAE,CAAC;QACX,IAAI,MAAM,CAAC,OAAO;YAAE,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;;YAC/C,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACjE,CAAC;IACD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,mBAAmB,SAAS,IAAI,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IACzG,OAAO;QACL,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,MAAM,EAAE,GAAG,EAAE,CAAC;YACZ,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAAA,CAC/C;KACF,CAAC;AAAA,CACH;AAED,KAAK,UAAU,SAAS,CACtB,GAAW,EACX,MAAc,EACd,OAAyB,EACuC;IAChE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,kBAAkB,CAAC;IAC1D,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,WAAW,CAAC,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;IAClE,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;YAC1E,MAAM;SACP,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,EAAE;YAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,CAAC;QAChE,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAM,CAAC;QAC1C,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IAC5B,CAAC;YAAS,CAAC;QACT,MAAM,EAAE,CAAC;IACX,CAAC;AAAA,CACF;AAED,SAAS,gBAAgB,CAAC,KAAqB,EAAmC;IAChF,MAAM,EAAE,GAAG,KAAK,CAAC,UAAU,CAAC;IAC5B,IAAI,CAAC,EAAE;QAAE,OAAO,SAAS,CAAC;IAC1B,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,IAAI,EAAE,CAAC;IACpC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;QAAE,OAAO,SAAS,CAAC;IACxD,OAAO;QACL,EAAE;QACF,IAAI,EAAE,EAAE;QACR,SAAS,EAAE,IAAI,CAAC,kBAAkB,IAAI,KAAK;QAC3C,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QAC1D,IAAI,EAAE;YACJ,KAAK,EAAE,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,CAAC,GAAG,SAAS;YACnD,MAAM,EAAE,CAAC,IAAI,CAAC,qBAAqB,IAAI,CAAC,CAAC,GAAG,SAAS;YACrD,SAAS,EAAE,CAAC,IAAI,CAAC,2BAA2B,IAAI,CAAC,CAAC,GAAG,SAAS;YAC9D,UAAU,EAAE,CAAC,IAAI,CAAC,+BAA+B,IAAI,CAAC,CAAC,GAAG,SAAS;SACpE;QACD,aAAa,EAAE,IAAI,CAAC,gBAAgB,IAAI,sBAAsB;QAC9D,SAAS,EAAE,IAAI,CAAC,iBAAiB,IAAI,kBAAkB;QACvD,MAAM,EAAE,WAAW,CAAC,EAAE,CAAC;KACxB,CAAC;AAAA,CACH;AAED,SAAS,iBAAiB,CAAC,KAAsB,EAAmC;IAClF,MAAM,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;IACpB,IAAI,CAAC,EAAE;QAAE,OAAO,SAAS,CAAC;IAC1B,OAAO;QACL,EAAE;QACF,IAAI,EAAE,GAAG,EAAE,gBAAgB;QAC3B,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,CAAC,MAAM,CAAC;QACf,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;QAC1D,aAAa,EAAE,sBAAsB;QACrC,SAAS,EAAE,kBAAkB;QAC7B,MAAM,EAAE,WAAW,CAAC,EAAE,CAAC;KACxB,CAAC;AAAA,CACH;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAe,EACf,MAAc,EACd,OAAO,GAAqB,EAAE,EACJ;IAC1B,MAAM,IAAI,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IACvC,MAAM,UAAU,GAAG,MAAM,SAAS,CAAoB,GAAG,IAAI,aAAa,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAC7F,IAAI,UAAU,CAAC,EAAE,EAAE,CAAC;QAClB,MAAM,MAAM,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;aACxC,GAAG,CAAC,gBAAgB,CAAC;aACrB,MAAM,CAAC,CAAC,CAAC,EAA4B,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;QAC5D,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;IAC1C,CAAC;IACD,IAAI,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,wBAAwB,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/D,CAAC;IACD,MAAM,UAAU,GAAG,MAAM,SAAS,CAAqB,GAAG,IAAI,YAAY,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAC7F,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,uBAAuB,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9D,CAAC;IACD,MAAM,MAAM,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;SACxC,GAAG,CAAC,iBAAiB,CAAC;SACtB,MAAM,CAAC,CAAC,CAAC,EAA4B,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;IAC5D,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;AAAA,CAC1C"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+export default function (pi: ExtensionAPI): Promise<void>;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,YAAY,EAAuB,MAAM,+BAA+B,CAAC;AA0HvF,yBAA+B,EAAE,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,CAoG9D"}

package/dist/index.js

@@ -0,0 +1,203 @@
+import { readFile } from "node:fs/promises";
+import { join } from "node:path";
+import { getAgentDir } from "@mariozechner/pi-coding-agent";
+import { fingerprint, readCache, writeCache } from "./cache.js";
+import { discoverModels, normalizeBaseUrl } from "./discover.js";
+const PROVIDER_NAME = "litellm";
+const ENV_BASE_URL = "LITELLM_BASE_URL";
+const ENV_API_KEY = "LITELLM_API_KEY";
+const ENV_TIMEOUT = "LITELLM_DISCOVERY_TIMEOUT_MS";
+const ENV_OFFLINE = "LITELLM_OFFLINE";
+const DEFAULT_TIMEOUT_MS = 5000;
+const LOGIN_TIMEOUT_MS = 10_000;
+const CACHE_FILENAME = "litellm-models.json";
+function getAuthPath() {
+    return join(getAgentDir(), "auth.json");
+}
+function getCachePath() {
+    return join(getAgentDir(), CACHE_FILENAME);
+}
+async function readAuthEntry() {
+    try {
+        const raw = await readFile(getAuthPath(), "utf8");
+        const parsed = JSON.parse(raw);
+        return parsed?.[PROVIDER_NAME];
+    }
+    catch {
+        return undefined;
+    }
+}
+async function resolveCredentials() {
+    const entry = await readAuthEntry();
+    const envBase = process.env[ENV_BASE_URL]?.trim();
+    const envKey = process.env[ENV_API_KEY]?.trim();
+    const authBase = entry?.type === "oauth" ? entry.baseUrl?.trim() : undefined;
+    const authKey = entry?.type === "oauth" ? entry.access?.trim() : entry?.type === "api_key" ? entry.key?.trim() : undefined;
+    const rawBase = envBase || authBase;
+    return {
+        baseUrl: rawBase ? normalizeBaseUrl(rawBase) : undefined,
+        apiKey: envKey || authKey || undefined,
+    };
+}
+function getDiscoveryTimeoutMs() {
+    const raw = process.env[ENV_TIMEOUT];
+    if (raw === undefined)
+        return DEFAULT_TIMEOUT_MS;
+    const parsed = Number.parseInt(raw, 10);
+    if (Number.isNaN(parsed) || parsed < 0)
+        return DEFAULT_TIMEOUT_MS;
+    return parsed;
+}
+function isOffline() {
+    return process.env[ENV_OFFLINE] === "1";
+}
+function isListModelsMode() {
+    return process.argv.includes("--list-models");
+}
+async function discoverWithFallback(baseUrl, apiKey, options) {
+    try {
+        return { result: await discoverModels(baseUrl, apiKey, options) };
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        return {
+            result: { models: [], source: "models_list" },
+            warning: message,
+        };
+    }
+}
+async function loginLiteLLM(callbacks) {
+    const rawBaseUrl = (await callbacks.onPrompt({
+        message: "Enter LiteLLM proxy URL (no trailing /v1):",
+        placeholder: "https://litellm.example.com",
+    })).trim();
+    const apiKey = (await callbacks.onPrompt({ message: "Enter API key:" })).trim();
+    if (!rawBaseUrl || !apiKey)
+        throw new Error("Both base URL and API key are required");
+    const baseUrl = normalizeBaseUrl(rawBaseUrl);
+    const { models, source } = await discoverModels(baseUrl, apiKey, {
+        timeoutMs: LOGIN_TIMEOUT_MS,
+        signal: callbacks.signal,
+    });
+    await writeCache(getCachePath(), {
+        baseUrl,
+        apiKeyFingerprint: fingerprint(apiKey),
+        fetchedAt: Date.now(),
+        source,
+        models,
+    });
+    callbacks.onProgress?.(`LiteLLM: ${models.length} models discovered (source: ${source})`);
+    return {
+        access: apiKey,
+        refresh: "",
+        expires: Number.MAX_SAFE_INTEGER,
+        baseUrl,
+    };
+}
+async function refreshLiteLLM(credentials) {
+    return credentials;
+}
+function modifyLiteLLMModels(models, cred) {
+    const baseUrl = cred.baseUrl;
+    if (!baseUrl)
+        return models;
+    return models.map((m) => (m.provider === PROVIDER_NAME ? { ...m, baseUrl: `${baseUrl}/v1` } : m));
+}
+export default async function (pi) {
+    const creds = await resolveCredentials();
+    const cache = await readCache(getCachePath());
+    const fp = creds.apiKey ? fingerprint(creds.apiKey) : undefined;
+    const cacheValid = cache !== null &&
+        creds.baseUrl !== undefined &&
+        fp !== undefined &&
+        cache.baseUrl === creds.baseUrl &&
+        cache.apiKeyFingerprint === fp;
+    let models = cacheValid && cache ? cache.models : [];
+    const haveCreds = creds.baseUrl !== undefined && creds.apiKey !== undefined && fp !== undefined;
+    const shouldFetch = haveCreds && !isOffline() && (!cacheValid || isListModelsMode());
+    if (shouldFetch && creds.baseUrl && creds.apiKey && fp) {
+        const timeoutMs = getDiscoveryTimeoutMs();
+        if (timeoutMs > 0) {
+            const { result, warning } = await discoverWithFallback(creds.baseUrl, creds.apiKey, {
+                timeoutMs,
+            });
+            if (warning) {
+                if (cacheValid && cache) {
+                    process.stderr.write(`LiteLLM: discovery failed (${warning}); using cached models.\n`);
+                    models = cache.models;
+                }
+                else {
+                    process.stderr.write(`LiteLLM: discovery failed (${warning}); registering provider with no models.\n`);
+                    models = [];
+                }
+            }
+            else {
+                models = result.models;
+                const next = {
+                    baseUrl: creds.baseUrl,
+                    apiKeyFingerprint: fp,
+                    fetchedAt: Date.now(),
+                    source: result.source,
+                    models: result.models,
+                };
+                await writeCache(getCachePath(), next);
+                if (isListModelsMode()) {
+                    process.stderr.write(`LiteLLM: ${result.models.length} models discovered (source: ${result.source}).\n`);
+                }
+            }
+        }
+    }
+    const oauth = {
+        name: "LiteLLM",
+        login: loginLiteLLM,
+        refreshToken: refreshLiteLLM,
+        getApiKey: (cred) => cred.access,
+        modifyModels: modifyLiteLLMModels,
+    };
+    pi.registerProvider(PROVIDER_NAME, {
+        baseUrl: creds.baseUrl ? `${creds.baseUrl}/v1` : "https://litellm.example.com/v1",
+        apiKey: ENV_API_KEY,
+        api: "openai-completions",
+        models,
+        oauth,
+    });
+    pi.registerCommand("litellm-refresh", {
+        description: "Re-discover models from the LiteLLM proxy.",
+        handler: async (_args, ctx) => {
+            if (isOffline()) {
+                ctx.ui.notify(`LiteLLM refresh disabled (${ENV_OFFLINE}=1)`, "warning");
+                return;
+            }
+            const fresh = await resolveCredentials();
+            const freshFp = fresh.apiKey ? fingerprint(fresh.apiKey) : undefined;
+            if (!fresh.baseUrl || !fresh.apiKey || !freshFp) {
+                ctx.ui.notify("LiteLLM refresh failed: no credentials. Run /login litellm or set env vars.", "error");
+                return;
+            }
+            const timeoutMs = getDiscoveryTimeoutMs() || DEFAULT_TIMEOUT_MS;
+            try {
+                const result = await discoverModels(fresh.baseUrl, fresh.apiKey, { timeoutMs });
+                await writeCache(getCachePath(), {
+                    baseUrl: fresh.baseUrl,
+                    apiKeyFingerprint: freshFp,
+                    fetchedAt: Date.now(),
+                    source: result.source,
+                    models: result.models,
+                });
+                pi.registerProvider(PROVIDER_NAME, {
+                    baseUrl: `${fresh.baseUrl}/v1`,
+                    apiKey: ENV_API_KEY,
+                    api: "openai-completions",
+                    models: result.models,
+                    oauth,
+                });
+                ctx.ui.notify(`LiteLLM: ${result.models.length} models refreshed (source: ${result.source})`, "info");
+            }
+            catch (error) {
+                const message = error instanceof Error ? error.message : String(error);
+                ctx.ui.notify(`LiteLLM refresh failed: ${message}`, "error");
+            }
+        },
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAGjC,OAAO,EAAE,WAAW,EAAE,MAAM,+BAA+B,CAAC;AAC5D,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAGjE,MAAM,aAAa,GAAG,SAAS,CAAC;AAChC,MAAM,YAAY,GAAG,kBAAkB,CAAC;AACxC,MAAM,WAAW,GAAG,iBAAiB,CAAC;AACtC,MAAM,WAAW,GAAG,8BAA8B,CAAC;AACnD,MAAM,WAAW,GAAG,iBAAiB,CAAC;AACtC,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAChC,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAChC,MAAM,cAAc,GAAG,qBAAqB,CAAC;AAE7C,SAAS,WAAW,GAAW;IAC7B,OAAO,IAAI,CAAC,WAAW,EAAE,EAAE,WAAW,CAAC,CAAC;AAAA,CACzC;AAED,SAAS,YAAY,GAAW;IAC9B,OAAO,IAAI,CAAC,WAAW,EAAE,EAAE,cAAc,CAAC,CAAC;AAAA,CAC5C;AAED,KAAK,UAAU,aAAa,GAAuC;IACjE,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,WAAW,EAAE,EAAE,MAAM,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAkC,CAAC;QAChE,OAAO,MAAM,EAAE,CAAC,aAAa,CAAC,CAAC;IACjC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AAAA,CACF;AAED,KAAK,UAAU,kBAAkB,GAAiC;IAChE,MAAM,KAAK,GAAG,MAAM,aAAa,EAAE,CAAC;IACpC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,CAAC;IAClD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,CAAC;IAChD,MAAM,QAAQ,GAAG,KAAK,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7E,MAAM,OAAO,GACX,KAAK,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7G,MAAM,OAAO,GAAG,OAAO,IAAI,QAAQ,CAAC;IACpC,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;QACxD,MAAM,EAAE,MAAM,IAAI,OAAO,IAAI,SAAS;KACvC,CAAC;AAAA,CACH;AAED,SAAS,qBAAqB,GAAW;IACvC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACrC,IAAI,GAAG,KAAK,SAAS;QAAE,OAAO,kBAAkB,CAAC;IACjD,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IACxC,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC;QAAE,OAAO,kBAAkB,CAAC;IAClE,OAAO,MAAM,CAAC;AAAA,CACf;AAED,SAAS,SAAS,GAAY;IAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,GAAG,CAAC;AAAA,CACzC;AAED,SAAS,gBAAgB,GAAY;IACnC,OAAO,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;AAAA,CAC/C;AAED,KAAK,UAAU,oBAAoB,CACjC,OAAe,EACf,MAAc,EACd,OAAyB,EAC+B;IACxD,IAAI,CAAC;QACH,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;IACpE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,OAAO;YACL,MAAM,EAAE,EAAE,MAAM,EAAE,EAA2B,EAAE,MAAM,EAAE,aAAa,EAAE;YACtE,OAAO,EAAE,OAAO;SACjB,CAAC;IACJ,CAAC;AAAA,CACF;AAED,KAAK,UAAU,YAAY,CAAC,SAA8B,EAA6B;IACrF,MAAM,UAAU,GAAG,CACjB,MAAM,SAAS,CAAC,QAAQ,CAAC;QACvB,OAAO,EAAE,4CAA4C;QACrD,WAAW,EAAE,6BAA6B;KAC3C,CAAC,CACH,CAAC,IAAI,EAAE,CAAC;IACT,MAAM,MAAM,GAAG,CAAC,MAAM,SAAS,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAChF,IAAI,CAAC,UAAU,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAEtF,MAAM,OAAO,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC;IAC7C,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE;QAC/D,SAAS,EAAE,gBAAgB;QAC3B,MAAM,EAAE,SAAS,CAAC,MAAM;KACzB,CAAC,CAAC;IAEH,MAAM,UAAU,CAAC,YAAY,EAAE,EAAE;QAC/B,OAAO;QACP,iBAAiB,EAAE,WAAW,CAAC,MAAM,CAAC;QACtC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,MAAM;QACN,MAAM;KACP,CAAC,CAAC;IACH,SAAS,CAAC,UAAU,EAAE,CAAC,YAAY,MAAM,CAAC,MAAM,+BAA+B,MAAM,GAAG,CAAC,CAAC;IAE1F,OAAO;QACL,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,MAAM,CAAC,gBAAgB;QAChC,OAAO;KACkC,CAAC;AAAA,CAC7C;AAED,KAAK,UAAU,cAAc,CAAC,WAA6B,EAA6B;IACtF,OAAO,WAAW,CAAC;AAAA,CACpB;AAED,SAAS,mBAAmB,CAAC,MAAoB,EAAE,IAAsB,EAAgB;IACvF,MAAM,OAAO,GAAI,IAA6B,CAAC,OAAO,CAAC;IACvD,IAAI,CAAC,OAAO;QAAE,OAAO,MAAM,CAAC;IAC5B,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,KAAK,aAAa,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAAA,CACnG;AAED,MAAM,CAAC,OAAO,CAAC,KAAK,WAAW,EAAgB,EAAiB;IAC9D,MAAM,KAAK,GAAG,MAAM,kBAAkB,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC;IAC9C,MAAM,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAEhE,MAAM,UAAU,GACd,KAAK,KAAK,IAAI;QACd,KAAK,CAAC,OAAO,KAAK,SAAS;QAC3B,EAAE,KAAK,SAAS;QAChB,KAAK,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO;QAC/B,KAAK,CAAC,iBAAiB,KAAK,EAAE,CAAC;IAEjC,IAAI,MAAM,GAA0B,UAAU,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IAC5E,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,EAAE,KAAK,SAAS,CAAC;IAChG,MAAM,WAAW,GAAG,SAAS,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,UAAU,IAAI,gBAAgB,EAAE,CAAC,CAAC;IAErF,IAAI,WAAW,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;QACvD,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;QAC1C,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,oBAAoB,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,EAAE;gBAClF,SAAS;aACV,CAAC,CAAC;YACH,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;oBACxB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,OAAO,2BAA2B,CAAC,CAAC;oBACvF,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;gBACxB,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,OAAO,2CAA2C,CAAC,CAAC;oBACvG,MAAM,GAAG,EAAE,CAAC;gBACd,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;gBACvB,MAAM,IAAI,GAAc;oBACtB,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,iBAAiB,EAAE,EAAE;oBACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,MAAM,EAAE,MAAM,CAAC,MAAM;iBACtB,CAAC;gBACF,MAAM,UAAU,CAAC,YAAY,EAAE,EAAE,IAAI,CAAC,CAAC;gBACvC,IAAI,gBAAgB,EAAE,EAAE,CAAC;oBACvB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,MAAM,CAAC,MAAM,CAAC,MAAM,+BAA+B,MAAM,CAAC,MAAM,MAAM,CAAC,CAAC;gBAC3G,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAG;QACZ,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,YAAY;QACnB,YAAY,EAAE,cAAc;QAC5B,SAAS,EAAE,CAAC,IAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM;QAClD,YAAY,EAAE,mBAAmB;KAClC,CAAC;IAEF,EAAE,CAAC,gBAAgB,CAAC,aAAa,EAAE;QACjC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,OAAO,KAAK,CAAC,CAAC,CAAC,gCAAgC;QACjF,MAAM,EAAE,WAAW;QACnB,GAAG,EAAE,oBAAoB;QACzB,MAAM;QACN,KAAK;KACN,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,CAAC,iBAAiB,EAAE;QACpC,WAAW,EAAE,4CAA4C;QACzD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;YAC7B,IAAI,SAAS,EAAE,EAAE,CAAC;gBAChB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,6BAA6B,WAAW,KAAK,EAAE,SAAS,CAAC,CAAC;gBACxE,OAAO;YACT,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,kBAAkB,EAAE,CAAC;YACzC,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACrE,IAAI,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;gBAChD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,6EAA6E,EAAE,OAAO,CAAC,CAAC;gBACtG,OAAO;YACT,CAAC;YACD,MAAM,SAAS,GAAG,qBAAqB,EAAE,IAAI,kBAAkB,CAAC;YAChE,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;gBAChF,MAAM,UAAU,CAAC,YAAY,EAAE,EAAE;oBAC/B,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,iBAAiB,EAAE,OAAO;oBAC1B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,MAAM,EAAE,MAAM,CAAC,MAAM;iBACtB,CAAC,CAAC;gBACH,EAAE,CAAC,gBAAgB,CAAC,aAAa,EAAE;oBACjC,OAAO,EAAE,GAAG,KAAK,CAAC,OAAO,KAAK;oBAC9B,MAAM,EAAE,WAAW;oBACnB,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,KAAK;iBACN,CAAC,CAAC;gBACH,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,MAAM,CAAC,MAAM,CAAC,MAAM,8BAA8B,MAAM,CAAC,MAAM,GAAG,EAAE,MAAM,CAAC,CAAC;YACxG,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACvE,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,2BAA2B,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;YAC/D,CAAC;QAAA,CACF;KACF,CAAC,CAAC;AAAA,CACJ"}

package/dist/types.d.ts

@@ -0,0 +1,55 @@
+import type { ProviderModelConfig } from "@mariozechner/pi-coding-agent";
+export type DiscoverySource = "model_info" | "models_list";
+export interface CacheFile {
+    baseUrl: string;
+    apiKeyFingerprint: string;
+    fetchedAt: number;
+    source: DiscoverySource;
+    models: ProviderModelConfig[];
+}
+export interface DiscoveryResult {
+    models: ProviderModelConfig[];
+    source: DiscoverySource;
+}
+export interface DiscoveryOptions {
+    timeoutMs?: number;
+    signal?: AbortSignal;
+}
+export interface ModelInfoEntry {
+    model_name?: string;
+    model_info?: {
+        mode?: string;
+        input_cost_per_token?: number;
+        output_cost_per_token?: number;
+        cache_read_input_token_cost?: number;
+        cache_creation_input_token_cost?: number;
+        max_input_tokens?: number;
+        max_output_tokens?: number;
+        supports_reasoning?: boolean;
+        supports_vision?: boolean;
+    };
+}
+export interface ModelInfoResponse {
+    data?: ModelInfoEntry[];
+}
+export interface ModelsListEntry {
+    id?: string;
+}
+export interface ModelsListResponse {
+    data?: ModelsListEntry[];
+}
+export type AuthFileEntry = {
+    type: "oauth";
+    access: string;
+    refresh: string;
+    expires: number;
+    baseUrl?: string;
+} | {
+    type: "api_key";
+    key: string;
+};
+export interface ResolvedCredentials {
+    baseUrl?: string;
+    apiKey?: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AAEzE,MAAM,MAAM,eAAe,GAAG,YAAY,GAAG,aAAa,CAAC;AAE3D,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,eAAe,CAAC;IACxB,MAAM,EAAE,mBAAmB,EAAE,CAAC;CAC/B;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,mBAAmB,EAAE,CAAC;IAC9B,MAAM,EAAE,eAAe,CAAC;CACzB;AAED,MAAM,WAAW,gBAAgB;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED,MAAM,WAAW,cAAc;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE;QACX,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;QAC/B,2BAA2B,CAAC,EAAE,MAAM,CAAC;QACrC,+BAA+B,CAAC,EAAE,MAAM,CAAC;QACzC,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,kBAAkB,CAAC,EAAE,OAAO,CAAC;QAC7B,eAAe,CAAC,EAAE,OAAO,CAAC;KAC3B,CAAC;CACH;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,CAAC,EAAE,cAAc,EAAE,CAAC;CACzB;AAED,MAAM,WAAW,eAAe;IAC9B,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,CAAC,EAAE,eAAe,EAAE,CAAC;CAC1B;AAED,MAAM,MAAM,aAAa,GACrB;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAA;CAAE,GACrF;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAAE,CAAC;AAErC,MAAM,WAAW,mBAAmB;IAClC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB"}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -0,0 +1,71 @@
+{
+  "name": "pi-provider-litellm",
+  "version": "0.1.1",
+  "description": "LiteLLM proxy provider extension for pi-mono.",
+  "type": "module",
+  "license": "MIT",
+  "author": "Dávid Balatoni <balcsida@gmail.com>",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/balcsida/pi-provider-litellm.git"
+  },
+  "homepage": "https://github.com/balcsida/pi-provider-litellm",
+  "bugs": "https://github.com/balcsida/pi-provider-litellm/issues",
+  "keywords": [
+    "pi-mono",
+    "pi-package",
+    "litellm",
+    "llm",
+    "extension",
+    "provider",
+    "proxy"
+  ],
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE",
+    "CHANGELOG.md"
+  ],
+  "pi": {
+    "extensions": [
+      "./dist/index.js"
+    ]
+  },
+  "scripts": {
+    "clean": "shx rm -rf dist",
+    "build": "tsgo -p tsconfig.build.json",
+    "dev": "tsgo -p tsconfig.build.json --watch --preserveWatchOutput",
+    "typecheck": "tsgo --noEmit",
+    "lint": "biome check --error-on-warnings .",
+    "format": "biome check --write .",
+    "test": "vitest --run",
+    "check": "npm run lint && npm run typecheck && npm test",
+    "prepublishOnly": "npm run check && npm run clean && npm run build"
+  },
+  "engines": {
+    "node": ">=20.0.0"
+  },
+  "peerDependencies": {
+    "@mariozechner/pi-ai": "^0.70.0",
+    "@mariozechner/pi-coding-agent": "^0.70.0"
+  },
+  "devDependencies": {
+    "@biomejs/biome": "^1.9.4",
+    "@mariozechner/pi-ai": "^0.70.0",
+    "@mariozechner/pi-coding-agent": "^0.70.0",
+    "@types/node": "^24.3.0",
+    "@typescript/native-preview": "7.0.0-dev.20260120.1",
+    "shx": "^0.4.0",
+    "tsx": "^4.21.0",
+    "typescript": "^5.7.3",
+    "vitest": "^3.2.4"
+  }
+}