pi-rozalia 0.1.4 → 0.2.1

package/README.md

@@ -1,6 +1,6 @@
 # pi-rozalia
 
-An extension for the [Pi coding agent](https://github.com/microsoft/pi-coding-agent) that connects to a [Rozalia AI](https://ai.zygoon.pl) server via the OpenAI-compatible Chat Completions API.
+An extension for the [Pi coding agent](https://github.com/earendil-works/pi) that connects to a [Rozalia AI](https://ai.zygoon.pl) server via the OpenAI-compatible Chat Completions API.
 
 ## Prerequisites
 
@@ -34,7 +34,6 @@ The server URL and API key are stored in `~/.pi/agent/auth.json` and reused on s
 | Variable | Description | Default |
 |----------|-------------|---------|
 | `ROZALIA_BASE_URL` | Rozalia AI server base URL | `https://ai.zygoon.pl/v1` |
-| `ROZALIA_BASE_URL` | Rozalia AI server base URL | `https://ai.zygoon.pl/v1` |
 | `ROZALIA_API_KEY` | API key for authentication | *(none)* |
 | `ROZALIA_TIMEOUT` | Model discovery timeout in ms | `5000` |
 

package/extensions/index.ts

@@ -29,6 +29,7 @@
 
 import type { ExtensionAPI, ProviderModelConfig } from "@earendil-works/pi-coding-agent";
 import type { OAuthCredentials, OAuthLoginCallbacks } from "@earendil-works/pi-ai";
+import type { RefreshModelsContext } from "@earendil-works/pi-ai";
 
 // ---------------------------------------------------------------------------
 // Credential payload (stored in OAuth refresh field).
@@ -198,6 +199,78 @@ function logDiscoveryError(baseUrl: string, error: unknown): void {
   );
 }
 
+// ---------------------------------------------------------------------------
+// TTL cache for model discovery
+// ---------------------------------------------------------------------------
+
+let cachedModels: ProviderModelConfig[] = [];
+let lastFetchedAt = 0;
+const MODEL_CACHE_TTL_MS = 5 * 1000; // 5 seconds — debounce only; /v1/models is cheap on a self-hosted server and the coordinator already dedupes concurrent calls
+
+/**
+ * Resolve baseUrl and apiKey from Pi's credential context.
+ *
+ * Priority: oauth credential > env vars.
+ * This is called by Pi during its refresh cycle — the credential is the
+ * effective configured auth (OAuth stored in ~/.pi/agent/auth.json or
+ * env-based API key).
+ */
+function resolveAuthFromContext(context: RefreshModelsContext): { baseUrl: string; apiKey: string } {
+  let baseUrl = getEnvBaseUrl();
+  let apiKey = getEnvApiKey();
+
+  const cred = context.credential;
+  if (cred?.type === "oauth") {
+    const p = decodeCreds(cred as OAuthCredentials);
+    if (p.baseUrl) baseUrl = p.baseUrl;
+    if (p.apiKey) apiKey = p.apiKey;
+  } else if (cred?.type === "api_key" && typeof cred.key === "string") {
+    apiKey = apiKey || cred.key;
+  }
+
+  return { baseUrl, apiKey };
+}
+
+/**
+ * The `refreshModels` hook — called by Pi on every refresh cycle (startup,
+ * explicit model refresh, etc.). Implements a TTL cache to avoid hammering
+ * the server while still responding to real changes within 5 minutes.
+ */
+async function refreshRozaliaModels(context: RefreshModelsContext): Promise<ProviderModelConfig[]> {
+  // TTL cache: skip fetch if we have a fresh list and caller didn't force
+  if (context.allowNetwork && !context.force && cachedModels.length > 0) {
+    const age = Date.now() - lastFetchedAt;
+    if (age < MODEL_CACHE_TTL_MS) {
+      return cachedModels;
+    }
+  }
+
+  const { baseUrl, apiKey } = resolveAuthFromContext(context);
+  if (!baseUrl) {
+    return cachedModels.length > 0 ? cachedModels : getFallbackModels();
+  }
+
+  // No API key means the user is logged out — skip fetch, return stub.
+  if (!apiKey) {
+    return cachedModels.length > 0 ? cachedModels : getFallbackModels();
+  }
+
+  try {
+    const controller = new AbortController();
+    const fetched = await fetchModels(baseUrl, apiKey, controller.signal);
+    if (fetched.length > 0) {
+      cachedModels = fetched;
+      lastFetchedAt = Date.now();
+      return fetched;
+    }
+    return getFallbackModels();
+  } catch (error) {
+    logDiscoveryError(baseUrl, error);
+    // Return cached models if available (don't replace with fallback on transient failure)
+    return cachedModels.length > 0 ? cachedModels : getFallbackModels();
+  }
+}
+
 // ---------------------------------------------------------------------------
 // Provider registration helper
 // ---------------------------------------------------------------------------
@@ -215,6 +288,9 @@ async function registerRozaliaProvider(
   const { baseUrl, apiKey } = creds;
   if (!baseUrl) return;
 
+  // No API key — user is logged out, keep the stub.
+  if (!apiKey) return;
+
   let models: ProviderModelConfig[];
   try {
     const controller = new AbortController();
@@ -313,7 +389,7 @@ function buildOauthBlock(
     refreshToken: async (creds: OAuthCredentials) => {
       const payload = decodeCreds(creds);
       if (!payload.baseUrl) return creds;
-      // Re-register with fresh models so the picker updates without restart.
+      // Refresh the model list so the picker updates without restart.
       // Reuse this same oauthBlock instead of reconstructing it — only the
       // model list changes, not the auth callbacks.
       try {
@@ -340,25 +416,27 @@ export default async function (pi: ExtensionAPI) {
   // (Pi only passes callbacks to login, not pi).
   const oauthBlock = buildOauthBlock(envBaseUrl, envApiKey, pi);
 
-  // Initial stub registration so "Rozalia" appears in /login selector.
-  // Credentials are resolved by Pi's OAuth machinery at request time — we no
-  // longer read ~/.pi/agent/auth.json directly (see commit history for why).
+  // Register the provider with refreshModels so Pi populates models on startup.
+  // The refreshModels hook is called by Pi during its refresh cycle (startup,
+  // explicit model refresh, etc.) with the resolved credential — this is the
+  // canonical Pi mechanism for dynamic model discovery.
   pi.registerProvider("rozalia", {
     name: "Rozalia",
     baseUrl: envBaseUrl,
     api: "openai-completions",
     authHeader: true,
-    models: [],
+    models: [], // Stub — refreshModels will populate the real list
+    refreshModels: refreshRozaliaModels,
     oauth: oauthBlock,
   });
 
   // Best-effort: if the API key is provided via env var, register eagerly so
-  // models appear without waiting for /login.
+  // models appear immediately without waiting for Pi's refresh cycle.
   if (envApiKey) {
     try {
       await registerRozaliaProvider(pi, { baseUrl: envBaseUrl, apiKey: envApiKey }, oauthBlock);
     } catch {
-      // ignore — will retry on next call
+      // ignore — will retry on next Pi refresh cycle
     }
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-rozalia",
-  "version": "0.1.4",
+  "version": "0.2.1",
   "description": "OpenAI-compatible provider extension for the Pi coding agent — connects to Rozalia AI",
   "keywords": [
     "pi-package",