pi-rozalia 0.1.2 → 0.1.3

package/README.md

@@ -66,6 +66,10 @@ On startup, the extension fetches the model list from the server's `/v1/models`
 
 The `/login` flow stores your server URL and API key in Pi's credential store (`~/.pi/agent/auth.json`). On subsequent logins the stored URL is pre-filled, so you only need to re-enter it if your server changes.
 
+## Known Limitations
+
+- **Models stay listed after `/logout`.** Pi's extension API has no logout callback/event for OAuth-backed providers, so this extension has no way to detect that `/logout` ran and revert the provider to its empty stub. The model list only clears on the next Pi restart. This is a Pi platform limitation, not something this extension can currently work around.
+
 ## TODO
 
 - [ ] Multi-server support — register each configured server as its own provider with a derived name (e.g. `rozalia-localhost-1234`), so models from different servers are unambiguous in the picker

package/extensions/index.ts

@@ -212,6 +212,15 @@ async function registerRozaliaProvider(
     config.apiKey = apiKey;
   }
 
+  // Unregister first so Pi replaces the stub (empty models) with
+  // the real model list. Without this, registerProvider keeps
+  // the stub's models: [] on an already-registered provider.
+  try {
+    pi.unregisterProvider("rozalia");
+  } catch {
+    // not previously registered; ignore
+  }
+
   pi.registerProvider("rozalia", config);
 }
 
@@ -226,6 +235,7 @@ async function registerRozaliaProvider(
 function createLoginFlow(
   defaultUrl: string,
   defaultApiKey: string | undefined,
+  pi: ExtensionAPI,
 ): (callbacks: OAuthLoginCallbacks) => Promise<OAuthCredentials> {
   return async (callbacks: OAuthLoginCallbacks) => {
     const inputUrl = await callbacks.onPrompt({
@@ -249,6 +259,10 @@ function createLoginFlow(
       // Still register — models will be discovered later or show fallback
     }
 
+    // Actually register the provider so models appear immediately
+    const oauthBlock = buildOauthBlock(defaultUrl, defaultApiKey, pi);
+    await registerRozaliaProvider(pi, creds, oauthBlock);
+
     return encodeCreds(creds);
   };
 }
@@ -260,24 +274,43 @@ function createLoginFlow(
 function buildOauthBlock(
   defaultUrl: string,
   defaultApiKey: string | undefined,
+  pi: ExtensionAPI,
 ) {
   return {
     name: "Rozalia",
-    login: createLoginFlow(defaultUrl, defaultApiKey),
-    refreshToken: async (creds: OAuthCredentials, _signal: AbortSignal) => {
+    login: createLoginFlow(defaultUrl, defaultApiKey, pi),
+    refreshToken: async (creds: OAuthCredentials, signal: AbortSignal) => {
       const payload = decodeCreds(creds);
+      if (!payload.baseUrl) return creds;
+      // Re-register with fresh models so the picker updates without restart
       try {
-        const ctrl = new AbortController();
-        await fetchModels(payload.baseUrl, payload.apiKey, ctrl.signal);
+        await registerRozaliaProvider(pi, payload, {
+          name: "Rozalia",
+          login: createLoginFlow(defaultUrl, defaultApiKey, pi),
+          refreshToken: async (c: OAuthCredentials, s: AbortSignal) =>
+            refreshTokenRozalia(c, s, payload, defaultUrl, defaultApiKey, pi),
+          getApiKey: (c: OAuthCredentials) => decodeCreds(c).apiKey || "",
+        });
       } catch {
-        // network blip
+        // network blip — keep creds, retry on next call
       }
-      return creds;
+      return encodeCreds(payload);
     },
     getApiKey: (creds: OAuthCredentials) => decodeCreds(creds).apiKey || "",
   };
 }
 
+async function refreshTokenRozalia(
+  creds: OAuthCredentials,
+  _signal: AbortSignal,
+  _payload: CredsPayload,
+  _defaultUrl: string,
+  _defaultApiKey: string | undefined,
+  _pi: ExtensionAPI,
+): Promise<OAuthCredentials> {
+  return encodeCreds(decodeCreds(creds));
+}
+
 // ---------------------------------------------------------------------------
 // Extension entry point
 // ---------------------------------------------------------------------------
@@ -310,7 +343,9 @@ export default async function (pi: ExtensionAPI) {
     // No saved credential — will use env vars or prompt
   }
 
-  const oauthBlock = buildOauthBlock(envBaseUrl, envApiKey);
+  // Capture pi in a closure so the login flow can call registerRozaliaProvider
+  // (Pi only passes callbacks to login, not pi).
+  const oauthBlock = buildOauthBlock(envBaseUrl, envApiKey, pi);
 
   // Initial stub registration so "Rozalia" appears in /login selector
   pi.registerProvider("rozalia", {

package/package.json

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