pi-provider-litellm 1.2.4 → 1.2.6

package/README.md

@@ -40,6 +40,8 @@ Inside pi:
 /login litellm
 ```
 
+You can also run `/login`, select `Use a subscription`, then select `LiteLLM`.
+
 You'll be prompted for the base URL and API key. Credentials are persisted to `~/.pi/agent/auth.json`.
 
 ### Option B — environment variables

package/dist/index.js

@@ -318,10 +318,13 @@ export default async function (pi) {
             }
         }
     }
+    let updateCosts = () => undefined;
     const oauth = {
         name: "LiteLLM",
         login: (callbacks) => loginLiteLLM(callbacks, (next) => {
             cacheFetchedAt = next.fetchedAt;
+            registerProvider(next.baseUrl, next.models);
+            updateCosts(next.models);
         }),
         refreshToken: refreshLiteLLM,
         getApiKey: getLiteLLMApiKey,
@@ -344,7 +347,7 @@ export default async function (pi) {
         });
     }
     registerProvider(creds.baseUrl, models);
-    const updateCosts = setupLiteLLMCostTracking(pi, models);
+    updateCosts = setupLiteLLMCostTracking(pi, models);
     let refreshInProgress = null;
     function discoveryDisabledReason() {
         if (isOffline())
@@ -379,6 +382,43 @@ export default async function (pi) {
         });
         return refreshInProgress;
     }
+    async function runLogin(ctx) {
+        const credential = await loginLiteLLM({
+            onAuth: () => undefined,
+            onDeviceCode: () => undefined,
+            onPrompt: async ({ message, placeholder }) => {
+                const value = await ctx.ui.input(message, placeholder);
+                if (value === undefined)
+                    throw new Error("Login cancelled");
+                return value;
+            },
+            onProgress: (message) => ctx.ui.notify(message, "info"),
+            onSelect: async () => undefined,
+            signal: ctx.signal,
+        }, (next) => {
+            cacheFetchedAt = next.fetchedAt;
+            registerProvider(next.baseUrl, next.models);
+            updateCosts(next.models);
+        });
+        ctx.modelRegistry.authStorage.set(PROVIDER_NAME, { type: "oauth", ...credential });
+        ctx.modelRegistry.refresh();
+        ctx.ui.notify(`Logged in to LiteLLM. Credentials saved to ${getAuthPath()}`, "info");
+    }
+    pi.on("input", async (event, ctx) => {
+        if (event.text.trim() !== `/login ${PROVIDER_NAME}`)
+            return { action: "continue" };
+        if (!ctx.hasUI)
+            return { action: "continue" };
+        try {
+            await runLogin(ctx);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            if (message !== "Login cancelled")
+                ctx.ui.notify(`LiteLLM login failed: ${message}`, "error");
+        }
+        return { action: "handled" };
+    });
     pi.registerCommand("litellm-refresh", {
         description: "Re-discover models from the LiteLLM proxy.",
         handler: async (_args, ctx) => {

package/dist/litellm.d.ts

@@ -1,18 +1 @@
-import type { OAuthCredentials } from "@earendil-works/pi-ai";
-import type { ResolvedCredentials } from "./types.js";
-export declare const PROVIDER_NAME = "litellm";
-export declare const ENV_BASE_URL = "LITELLM_BASE_URL";
-export declare const ENV_API_KEY = "LITELLM_API_KEY";
-export declare const ENV_TIMEOUT = "LITELLM_DISCOVERY_TIMEOUT_MS";
-export declare const ENV_OFFLINE = "LITELLM_OFFLINE";
-export declare const DEFAULT_TIMEOUT_MS = 5000;
-export declare const LOGIN_TIMEOUT_MS = 10000;
-export declare const CACHE_FILENAME = "litellm-models.json";
-export declare function getAuthPath(): string;
-export declare function getCachePath(): string;
-export declare function resolveCredentials(): Promise<ResolvedCredentials>;
-export declare function getDiscoveryTimeoutMs(): number;
-export declare function isOffline(): boolean;
-export declare function isListModelsMode(): boolean;
 export declare function getSessionIdFromFile(sessionFile?: string): string | undefined;
-export type { OAuthCredentials };

package/dist/litellm.js

@@ -1,62 +1,3 @@
-import { readFile } from "node:fs/promises";
-import { join } from "node:path";
-import { AuthStorage, getAgentDir } from "@earendil-works/pi-coding-agent";
-import { normalizeBaseUrl } from "./discover.js";
-export const PROVIDER_NAME = "litellm";
-export const ENV_BASE_URL = "LITELLM_BASE_URL";
-export const ENV_API_KEY = "LITELLM_API_KEY";
-export const ENV_TIMEOUT = "LITELLM_DISCOVERY_TIMEOUT_MS";
-export const ENV_OFFLINE = "LITELLM_OFFLINE";
-export const DEFAULT_TIMEOUT_MS = 5000;
-export const LOGIN_TIMEOUT_MS = 10_000;
-export const CACHE_FILENAME = "litellm-models.json";
-export function getAuthPath() {
-    return join(getAgentDir(), "auth.json");
-}
-export 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;
-    }
-}
-export 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"
-            ? (await AuthStorage.create(getAuthPath()).getApiKey(PROVIDER_NAME, { includeFallback: false }))?.trim()
-            : undefined;
-    const rawBase = authBase || envBase;
-    return {
-        baseUrl: rawBase ? normalizeBaseUrl(rawBase) : undefined,
-        apiKey: authKey || envKey || undefined,
-    };
-}
-export 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;
-}
-export function isOffline() {
-    return process.env[ENV_OFFLINE] === "1";
-}
-export function isListModelsMode() {
-    return process.argv.includes("--list-models");
-}
 export function getSessionIdFromFile(sessionFile) {
     if (!sessionFile)
         return undefined;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-provider-litellm",
-  "version": "1.2.4",
+  "version": "1.2.6",
   "description": "LiteLLM proxy provider extension for Pi",
   "type": "module",
   "license": "MIT",