pi-yandex-bridge 0.1.0

package/README.md

@@ -0,0 +1,50 @@
+# pi-yandex-bridge
+
+Pi Coding Agent provider bridge for Yandex Cloud AI (YandexGPT).
+
+## Models
+
+| Model             | Context | Max output |
+| ----------------- | ------- | ---------- |
+| YandexGPT Pro 5.1 | 128k    | 8k         |
+| YandexGPT Pro     | 128k    | 8k         |
+| YandexGPT Lite    | 32k     | 4k         |
+
+Unknown models discovered via the API are displayed as `slug {last4ofFolderId}`.
+
+## Auth: OAuth (default)
+
+### 1. Find your Yandex Cloud folder ID
+
+In the [Yandex Cloud console](https://console.yandex.cloud), select your folder. The ID is shown in the URL and on the folder overview page (looks like `b1gk28...`).
+
+### 2. Log in
+
+Run `/yalogin` in Pi. A browser window will open automatically — authorize the app, and the token is captured without any pasting. Pi then prompts for your folder ID.
+
+**IAM tokens expire after 12 hours** and are refreshed automatically using the stored OAuth token.
+
+To skip the browser flow, set env vars before starting Pi:
+
+```sh
+export YANDEX_OAUTH_TOKEN="y0_AgAAAA..."
+export YANDEX_FOLDER_ID="b1g..."
+```
+
+## Auth: Static API key
+
+Set both env vars before starting Pi — no `auth.json` entry needed:
+
+```sh
+export YANDEX_API_KEY="your-api-key"
+export YANDEX_FOLDER_ID="your-folder-id"
+```
+
+You can generate an API key in the Yandex Cloud console under **Service accounts → your account → API keys**. The key is shown only once, so copy it immediately.
+
+Env vars also work for OAuth to skip interactive prompts:
+
+```sh
+export YANDEX_OAUTH_TOKEN="y0_AgAAAA..."
+export YANDEX_FOLDER_ID="b1g..."
+```

package/dist/index.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Pi Coding Agent — Yandex Provider Bridge
+ *
+ * Auth (OAuth — default):
+ *   Run /yalogin to authorize. A browser window opens automatically,
+ *   the token is captured via a local callback server on port 7890.
+ *
+ *   YANDEX_OAUTH_TOKEN env var skips the browser flow entirely.
+ *   YANDEX_FOLDER_ID   env var skips the folder ID prompt.
+ *
+ * Auth (static API key — opt-in):
+ *   Set both YANDEX_API_KEY and YANDEX_FOLDER_ID env vars.
+ */
+import type { ExtensionAPI } from "@earendil-works/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":["../index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAQH,OAAO,KAAK,EACX,YAAY,EAGZ,MAAM,iCAAiC,CAAC;AA4RzC,yBAA+B,EAAE,EAAE,YAAY,iBAwF9C"}

package/dist/index.js

@@ -0,0 +1,318 @@
+/**
+ * Pi Coding Agent — Yandex Provider Bridge
+ *
+ * Auth (OAuth — default):
+ *   Run /yalogin to authorize. A browser window opens automatically,
+ *   the token is captured via a local callback server on port 7890.
+ *
+ *   YANDEX_OAUTH_TOKEN env var skips the browser flow entirely.
+ *   YANDEX_FOLDER_ID   env var skips the folder ID prompt.
+ *
+ * Auth (static API key — opt-in):
+ *   Set both YANDEX_API_KEY and YANDEX_FOLDER_ID env vars.
+ */
+import { createServer } from "http";
+import { readFileSync, writeFileSync } from "fs";
+import { homedir } from "os";
+import { join } from "path";
+import { spawnSync } from "child_process";
+// ─── constants ────────────────────────────────────────────────────────────────
+const IAM_TOKEN_URL = "https://iam.api.cloud.yandex.net/iam/v1/tokens";
+const AI_BASE_URL = "https://ai.api.cloud.yandex.net/v1";
+const AUTH_PATH = join(homedir(), ".pi", "agent", "auth.json");
+const OAUTH_CLIENT_ID = "0414b7213b22435fa65051f64270584f";
+const OAUTH_CALLBACK_PORT = 7890;
+const OAUTH_REDIRECT_URI = `http://localhost:${OAUTH_CALLBACK_PORT}/callback`;
+const OAUTH_URL = `https://oauth.yandex.ru/authorize?response_type=token` +
+    `&client_id=${OAUTH_CLIENT_ID}` +
+    `&redirect_uri=${encodeURIComponent(OAUTH_REDIRECT_URI)}`;
+// ─── model catalogue ──────────────────────────────────────────────────────────
+const KNOWN_MODELS = [
+    {
+        slug: "yandexgpt-5.1",
+        name: "YandexGPT Pro 5.1",
+        contextWindow: 128_000,
+        maxTokens: 8_192,
+        cost: { input: 8.8, output: 8.8, cacheRead: 0, cacheWrite: 0 },
+    },
+    {
+        slug: "yandexgpt",
+        name: "YandexGPT Pro",
+        contextWindow: 128_000,
+        maxTokens: 8_192,
+        cost: { input: 8.8, output: 8.8, cacheRead: 0, cacheWrite: 0 },
+    },
+    {
+        slug: "yandexgpt-lite",
+        name: "YandexGPT Lite",
+        contextWindow: 32_000,
+        maxTokens: 4_096,
+        cost: { input: 2.2, output: 2.2, cacheRead: 0, cacheWrite: 0 },
+    },
+];
+/** Turns `gpt://<folderId>/<slug>/latest` into a readable display name.
+ *  Known slugs get their canonical name; unknown ones get `slug {last4ofFolderId}`. */
+function prettyModelName(id) {
+    const match = id.match(/^gpt:\/\/([^/]+)\/(.+?)(?:\/latest)?$/);
+    if (!match)
+        return id;
+    const [, folderId, slug] = match;
+    const known = KNOWN_MODELS.find((m) => m.slug === slug);
+    if (known)
+        return known.name;
+    return `${slug} {${folderId.slice(-4)}}`;
+}
+function openBrowser(url) {
+    const cmd = process.platform === "win32"
+        ? "start"
+        : process.platform === "darwin"
+            ? "open"
+            : "xdg-open";
+    spawnSync(cmd, [url], { stdio: "ignore" });
+}
+async function exchangeOAuthForIam(oauthToken) {
+    const res = await fetch(IAM_TOKEN_URL, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ yandexPassportOauthToken: oauthToken }),
+    });
+    if (!res.ok) {
+        const text = await res.text().catch(() => res.statusText);
+        throw new Error(`IAM token exchange failed (${res.status}): ${text}`);
+    }
+    const data = (await res.json());
+    return {
+        token: data.iamToken,
+        expiresAt: new Date(data.expiresAt).getTime(),
+    };
+}
+function buildModels(folderId) {
+    return KNOWN_MODELS.map((m) => ({
+        id: `gpt://${folderId}/${m.slug}/latest`,
+        name: m.name,
+        api: "openai-responses",
+        provider: "yandex",
+        baseUrl: AI_BASE_URL,
+        reasoning: false,
+        input: ["text"],
+        cost: m.cost,
+        contextWindow: m.contextWindow,
+        maxTokens: m.maxTokens,
+        headers: { "OpenAI-Project": folderId },
+        compat: {
+            supportsDeveloperRole: false,
+            supportsReasoningEffort: false,
+            maxTokensField: "max_tokens",
+        },
+    }));
+}
+function readAuthJson() {
+    try {
+        return JSON.parse(readFileSync(AUTH_PATH, "utf8"));
+    }
+    catch {
+        return {};
+    }
+}
+function writeAuthJson(data) {
+    writeFileSync(AUTH_PATH, JSON.stringify(data, null, 2), "utf8");
+}
+// ─── OAuth local callback server ──────────────────────────────────────────────
+const CALLBACK_HTML = `<!DOCTYPE html>
+<html><head><meta charset="utf-8"><title>pi-yandex-bridge</title></head>
+<body style="font-family:sans-serif;padding:2rem">
+<p>Authorizing pi-yandex-bridge…</p>
+<script>
+const p = new URLSearchParams(location.hash.slice(1));
+const t = p.get('access_token');
+if (t) {
+  fetch('/token', { method: 'POST', body: t })
+    .then(() => { document.body.innerHTML = '<p>✓ Authorized. You can close this tab.</p>'; });
+} else {
+  document.body.innerHTML = '<p>Error: no access_token in response.</p>';
+}
+</script>
+</body></html>`;
+function captureOAuthToken(onProgress) {
+    return new Promise((resolve, reject) => {
+        const server = createServer((req, res) => {
+            if (req.url?.startsWith("/callback")) {
+                res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
+                res.end(CALLBACK_HTML);
+            }
+            else if (req.url === "/token" && req.method === "POST") {
+                let body = "";
+                req.on("data", (chunk) => (body += chunk));
+                req.on("end", () => {
+                    res.writeHead(200);
+                    res.end();
+                    server.close();
+                    resolve(body.trim());
+                });
+            }
+            else {
+                res.writeHead(404);
+                res.end();
+            }
+        });
+        server.on("error", (err) => {
+            if (err.code === "EADDRINUSE") {
+                reject(new Error(`Port ${OAUTH_CALLBACK_PORT} is already in use. Stop the process using it and try again.`));
+            }
+            else {
+                reject(new Error(`Local auth server error: ${err.message}`));
+            }
+        });
+        const timeout = setTimeout(() => {
+            server.close();
+            reject(new Error("OAuth authorization timed out after 5 minutes."));
+        }, 5 * 60 * 1000);
+        server.listen(OAUTH_CALLBACK_PORT, "127.0.0.1", () => {
+            onProgress?.(`[yandex] Listening on port ${server.address().port}. Opening browser…`);
+            openBrowser(OAUTH_URL);
+        });
+        server.on("close", () => clearTimeout(timeout));
+    });
+}
+// ─── OAuth provider callbacks ─────────────────────────────────────────────────
+async function yandexLogin(callbacks) {
+    let oauthToken = process.env.YANDEX_OAUTH_TOKEN ?? "";
+    if (!oauthToken) {
+        oauthToken = await captureOAuthToken((msg) => callbacks.onProgress?.(msg));
+    }
+    else {
+        callbacks.onProgress?.("[yandex] Using YANDEX_OAUTH_TOKEN from environment.");
+    }
+    let folderId = process.env.YANDEX_FOLDER_ID ?? "";
+    if (!folderId) {
+        folderId = await callbacks.onPrompt({
+            message: "Enter your Yandex Cloud folder ID:",
+            placeholder: "b1g...",
+        });
+    }
+    callbacks.onProgress?.("[yandex] Exchanging OAuth token for IAM token…");
+    const iam = await exchangeOAuthForIam(oauthToken);
+    return {
+        refresh: oauthToken,
+        access: iam.token,
+        expires: iam.expiresAt,
+        folderId,
+    };
+}
+async function yandexRefreshToken(credentials) {
+    const iam = await exchangeOAuthForIam(credentials.refresh);
+    return { ...credentials, access: iam.token, expires: iam.expiresAt };
+}
+// ─── /yalogin command ─────────────────────────────────────────────────────────
+async function runYaLogin(ctx) {
+    try {
+        ctx.ui.notify("Opening browser for Yandex authorization…", "info");
+        const oauthToken = await captureOAuthToken();
+        let folderId = process.env.YANDEX_FOLDER_ID ?? "";
+        if (!folderId) {
+            folderId =
+                (await ctx.ui.input("Yandex Cloud folder ID", "b1g...")) ?? "";
+        }
+        if (!folderId) {
+            ctx.ui.notify("Login cancelled: folder ID is required.", "error");
+            return;
+        }
+        ctx.ui.notify("Exchanging OAuth token for IAM token…", "info");
+        const iam = await exchangeOAuthForIam(oauthToken);
+        const auth = readAuthJson();
+        auth.yandex = {
+            type: "oauth",
+            refresh: oauthToken,
+            access: iam.token,
+            expires: iam.expiresAt,
+            folderId,
+        };
+        writeAuthJson(auth);
+        ctx.ui.notify("✓ Yandex credentials saved. Restart Pi to activate the models.", "info");
+    }
+    catch (err) {
+        ctx.ui.notify(`Yandex login failed: ${err instanceof Error ? err.message : String(err)}`, "error");
+    }
+}
+// ─── extension entry point ────────────────────────────────────────────────────
+export default async function (pi) {
+    const apiKey = process.env.YANDEX_API_KEY;
+    const folderId = process.env.YANDEX_FOLDER_ID;
+    // Static API key path — both env vars must be present.
+    if (apiKey && folderId) {
+        const modelIds = new Set(KNOWN_MODELS.map((m) => `gpt://${folderId}/${m.slug}/latest`));
+        try {
+            const res = await fetch(`${AI_BASE_URL}/models`, {
+                headers: {
+                    Authorization: `Bearer ${apiKey}`,
+                    "OpenAI-Project": folderId,
+                },
+            });
+            if (res.ok) {
+                const payload = (await res.json());
+                for (const { id } of payload.data)
+                    modelIds.add(id);
+            }
+        }
+        catch {
+            /* static list is enough */
+        }
+        const models = [...modelIds].map((id) => {
+            const slug = id.replace(/^gpt:\/\/[^/]+\//, "").replace(/\/latest$/, "");
+            const known = KNOWN_MODELS.find((m) => m.slug === slug);
+            return {
+                id,
+                name: prettyModelName(id),
+                reasoning: false,
+                input: ["text"],
+                cost: known?.cost ?? { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+                contextWindow: known?.contextWindow ?? 128_000,
+                maxTokens: known?.maxTokens ?? 8_192,
+                headers: { "OpenAI-Project": folderId },
+                compat: {
+                    supportsDeveloperRole: false,
+                    supportsReasoningEffort: false,
+                    maxTokensField: "max_tokens",
+                },
+            };
+        });
+        pi.registerProvider("yandex", {
+            name: "Yandex Cloud",
+            baseUrl: AI_BASE_URL,
+            apiKey,
+            api: "openai-responses",
+            models,
+        });
+        console.log(`[yandex] Registered ${models.length} model(s) with static API key.`);
+    }
+    else {
+        // OAuth path — seed models from auth.json if folderId is already stored.
+        let storedFolderId;
+        try {
+            const auth = readAuthJson();
+            storedFolderId = auth.yandex?.folderId;
+        }
+        catch {
+            /* auth.json absent or unreadable */
+        }
+        pi.registerProvider("yandex", {
+            name: "Yandex Cloud",
+            baseUrl: AI_BASE_URL,
+            api: "openai-responses",
+            models: storedFolderId ? buildModels(storedFolderId) : [],
+            oauth: {
+                name: "Yandex Cloud (OAuth)",
+                login: yandexLogin,
+                refreshToken: yandexRefreshToken,
+                getApiKey: (credentials) => credentials.access,
+                modifyModels: (_, credentials) => buildModels(credentials.folderId),
+            },
+        });
+        console.log("[yandex] Registered with OAuth.");
+    }
+    pi.registerCommand("yalogin", {
+        description: "Authorize Yandex Cloud (opens browser, no pasting required)",
+        handler: async (_args, ctx) => runYaLogin(ctx),
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAY1C,iFAAiF;AAEjF,MAAM,aAAa,GAAG,gDAAgD,CAAC;AACvE,MAAM,WAAW,GAAG,oCAAoC,CAAC;AACzD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;AAE/D,MAAM,eAAe,GAAG,kCAAkC,CAAC;AAC3D,MAAM,mBAAmB,GAAG,IAAI,CAAC;AACjC,MAAM,kBAAkB,GAAG,oBAAoB,mBAAmB,WAAW,CAAC;AAC9E,MAAM,SAAS,GACd,uDAAuD;IACvD,cAAc,eAAe,EAAE;IAC/B,iBAAiB,kBAAkB,CAAC,kBAAkB,CAAC,EAAE,CAAC;AAE3D,iFAAiF;AAEjF,MAAM,YAAY,GAAG;IACpB;QACC,IAAI,EAAE,eAAe;QACrB,IAAI,EAAE,mBAAmB;QACzB,aAAa,EAAE,OAAO;QACtB,SAAS,EAAE,KAAK;QAChB,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;KAC9D;IACD;QACC,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,eAAe;QACrB,aAAa,EAAE,OAAO;QACtB,SAAS,EAAE,KAAK;QAChB,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;KAC9D;IACD;QACC,IAAI,EAAE,gBAAgB;QACtB,IAAI,EAAE,gBAAgB;QACtB,aAAa,EAAE,MAAM;QACrB,SAAS,EAAE,KAAK;QAChB,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;KAC9D;CACQ,CAAC;AASX;uFACuF;AACvF,SAAS,eAAe,CAAC,EAAU;IAClC,MAAM,KAAK,GAAG,EAAE,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAChE,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAC;IACtB,MAAM,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,KAAK,CAAC;IACjC,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IACxD,IAAI,KAAK;QAAE,OAAO,KAAK,CAAC,IAAI,CAAC;IAC7B,OAAO,GAAG,IAAI,KAAK,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;AAC1C,CAAC;AAED,SAAS,WAAW,CAAC,GAAW;IAC/B,MAAM,GAAG,GACR,OAAO,CAAC,QAAQ,KAAK,OAAO;QAC3B,CAAC,CAAC,OAAO;QACT,CAAC,CAAC,OAAO,CAAC,QAAQ,KAAK,QAAQ;YAC9B,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,UAAU,CAAC;IAChB,SAAS,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;AAC5C,CAAC;AAED,KAAK,UAAU,mBAAmB,CACjC,UAAkB;IAElB,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,aAAa,EAAE;QACtC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,wBAAwB,EAAE,UAAU,EAAE,CAAC;KAC9D,CAAC,CAAC;IAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACb,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1D,MAAM,IAAI,KAAK,CAAC,8BAA8B,GAAG,CAAC,MAAM,MAAM,IAAI,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAqB,CAAC;IACpD,OAAO;QACN,KAAK,EAAE,IAAI,CAAC,QAAQ;QACpB,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;KAC7C,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,QAAgB;IACpC,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC/B,EAAE,EAAE,SAAS,QAAQ,IAAI,CAAC,CAAC,IAAI,SAAS;QACxC,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,GAAG,EAAE,kBAA2B;QAChC,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,WAAW;QACpB,SAAS,EAAE,KAAK;QAChB,KAAK,EAAE,CAAC,MAAM,CAAyB;QACvC,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,aAAa,EAAE,CAAC,CAAC,aAAa;QAC9B,SAAS,EAAE,CAAC,CAAC,SAAS;QACtB,OAAO,EAAE,EAAE,gBAAgB,EAAE,QAAQ,EAAE;QACvC,MAAM,EAAE;YACP,qBAAqB,EAAE,KAAK;YAC5B,uBAAuB,EAAE,KAAK;YAC9B,cAAc,EAAE,YAAqB;SACrC;KACD,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,YAAY;IACpB,IAAI,CAAC;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAA4B,CAAC;IAC/E,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,EAAE,CAAC;IACX,CAAC;AACF,CAAC;AAED,SAAS,aAAa,CAAC,IAA6B;IACnD,aAAa,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;AACjE,CAAC;AAED,iFAAiF;AAEjF,MAAM,aAAa,GAAG;;;;;;;;;;;;;;eAcP,CAAC;AAEhB,SAAS,iBAAiB,CAAC,UAAkC;IAC5D,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACtC,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACxC,IAAI,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;gBACtC,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,0BAA0B,EAAE,CAAC,CAAC;gBACnE,GAAG,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;YACxB,CAAC;iBAAM,IAAI,GAAG,CAAC,GAAG,KAAK,QAAQ,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;gBAC1D,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC;gBAC3C,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBAClB,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;oBACnB,GAAG,CAAC,GAAG,EAAE,CAAC;oBACV,MAAM,CAAC,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBACtB,CAAC,CAAC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACP,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACnB,GAAG,CAAC,GAAG,EAAE,CAAC;YACX,CAAC;QACF,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAA0B,EAAE,EAAE;YACjD,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC/B,MAAM,CACL,IAAI,KAAK,CACR,QAAQ,mBAAmB,8DAA8D,CACzF,CACD,CAAC;YACH,CAAC;iBAAM,CAAC;gBACP,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC9D,CAAC;QACF,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC/B,MAAM,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC,CAAC;QACrE,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;QAElB,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,WAAW,EAAE,GAAG,EAAE;YACpD,UAAU,EAAE,CACX,8BAA+B,MAAM,CAAC,OAAO,EAAkB,CAAC,IAAI,oBAAoB,CACxF,CAAC;YACF,WAAW,CAAC,SAAS,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,iFAAiF;AAEjF,KAAK,UAAU,WAAW,CACzB,SAA8B;IAE9B,IAAI,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,EAAE,CAAC;IACtD,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,UAAU,GAAG,MAAM,iBAAiB,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAC5E,CAAC;SAAM,CAAC;QACP,SAAS,CAAC,UAAU,EAAE,CACrB,qDAAqD,CACrD,CAAC;IACH,CAAC;IAED,IAAI,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,EAAE,CAAC;IAClD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACf,QAAQ,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC;YACnC,OAAO,EAAE,oCAAoC;YAC7C,WAAW,EAAE,QAAQ;SACrB,CAAC,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,UAAU,EAAE,CAAC,gDAAgD,CAAC,CAAC;IACzE,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,UAAU,CAAC,CAAC;IAElD,OAAO;QACN,OAAO,EAAE,UAAU;QACnB,MAAM,EAAE,GAAG,CAAC,KAAK;QACjB,OAAO,EAAE,GAAG,CAAC,SAAS;QACtB,QAAQ;KACR,CAAC;AACH,CAAC;AAED,KAAK,UAAU,kBAAkB,CAChC,WAA6B;IAE7B,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC3D,OAAO,EAAE,GAAG,WAAW,EAAE,MAAM,EAAE,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC;AACtE,CAAC;AAED,iFAAiF;AAEjF,KAAK,UAAU,UAAU,CAAC,GAA4B;IACrD,IAAI,CAAC;QACJ,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,2CAA2C,EAAE,MAAM,CAAC,CAAC;QAEnE,MAAM,UAAU,GAAG,MAAM,iBAAiB,EAAE,CAAC;QAE7C,IAAI,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,EAAE,CAAC;QAClD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACf,QAAQ;gBACP,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,KAAK,CAClB,wBAAwB,EACxB,QAAQ,CACR,CAAC,IAAI,EAAE,CAAC;QACX,CAAC;QAED,IAAI,CAAC,QAAQ,EAAE,CAAC;YACf,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,yCAAyC,EAAE,OAAO,CAAC,CAAC;YAClE,OAAO;QACR,CAAC;QAED,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,uCAAuC,EAAE,MAAM,CAAC,CAAC;QAC/D,MAAM,GAAG,GAAG,MAAM,mBAAmB,CAAC,UAAU,CAAC,CAAC;QAElD,MAAM,IAAI,GAAG,YAAY,EAAE,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG;YACb,IAAI,EAAE,OAAO;YACb,OAAO,EAAE,UAAU;YACnB,MAAM,EAAE,GAAG,CAAC,KAAK;YACjB,OAAO,EAAE,GAAG,CAAC,SAAS;YACtB,QAAQ;SACR,CAAC;QACF,aAAa,CAAC,IAAI,CAAC,CAAC;QAEpB,GAAG,CAAC,EAAE,CAAC,MAAM,CACZ,gEAAgE,EAChE,MAAM,CACN,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACd,GAAG,CAAC,EAAE,CAAC,MAAM,CACZ,wBAAwB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAC1E,OAAO,CACP,CAAC;IACH,CAAC;AACF,CAAC;AAED,iFAAiF;AAEjF,MAAM,CAAC,OAAO,CAAC,KAAK,WAAW,EAAgB;IAC9C,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAC1C,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAE9C,uDAAuD;IACvD,IAAI,MAAM,IAAI,QAAQ,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,IAAI,GAAG,CACvB,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,QAAQ,IAAI,CAAC,CAAC,IAAI,SAAS,CAAC,CAC7D,CAAC;QAEF,IAAI,CAAC;YACJ,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,WAAW,SAAS,EAAE;gBAChD,OAAO,EAAE;oBACR,aAAa,EAAE,UAAU,MAAM,EAAE;oBACjC,gBAAgB,EAAE,QAAQ;iBAC1B;aACD,CAAC,CAAC;YACH,IAAI,GAAG,CAAC,EAAE,EAAE,CAAC;gBACZ,MAAM,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAoC,CAAC;gBACtE,KAAK,MAAM,EAAE,EAAE,EAAE,IAAI,OAAO,CAAC,IAAI;oBAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACrD,CAAC;QACF,CAAC;QAAC,MAAM,CAAC;YACR,2BAA2B;QAC5B,CAAC;QAED,MAAM,MAAM,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;YACvC,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YACzE,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YACxD,OAAO;gBACN,EAAE;gBACF,IAAI,EAAE,eAAe,CAAC,EAAE,CAAC;gBACzB,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,CAAC,MAAM,CAAyB;gBACvC,IAAI,EAAE,KAAK,EAAE,IAAI,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE;gBACzE,aAAa,EAAE,KAAK,EAAE,aAAa,IAAI,OAAO;gBAC9C,SAAS,EAAE,KAAK,EAAE,SAAS,IAAI,KAAK;gBACpC,OAAO,EAAE,EAAE,gBAAgB,EAAE,QAAQ,EAAE;gBACvC,MAAM,EAAE;oBACP,qBAAqB,EAAE,KAAK;oBAC5B,uBAAuB,EAAE,KAAK;oBAC9B,cAAc,EAAE,YAAqB;iBACrC;aACD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gBAAgB,CAAC,QAAQ,EAAE;YAC7B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,WAAW;YACpB,MAAM;YACN,GAAG,EAAE,kBAAkB;YACvB,MAAM;SACmB,CAAC,CAAC;QAE5B,OAAO,CAAC,GAAG,CACV,uBAAuB,MAAM,CAAC,MAAM,gCAAgC,CACpE,CAAC;IACH,CAAC;SAAM,CAAC;QACP,yEAAyE;QACzE,IAAI,cAAkC,CAAC;QACvC,IAAI,CAAC;YACJ,MAAM,IAAI,GAAG,YAAY,EAA2C,CAAC;YACrE,cAAc,GAAG,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC;QACxC,CAAC;QAAC,MAAM,CAAC;YACR,oCAAoC;QACrC,CAAC;QAED,EAAE,CAAC,gBAAgB,CAAC,QAAQ,EAAE;YAC7B,IAAI,EAAE,cAAc;YACpB,OAAO,EAAE,WAAW;YACpB,GAAG,EAAE,kBAAkB;YACvB,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,EAAE;YACzD,KAAK,EAAE;gBACN,IAAI,EAAE,sBAAsB;gBAC5B,KAAK,EAAE,WAAW;gBAClB,YAAY,EAAE,kBAAkB;gBAChC,SAAS,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM;gBAC9C,YAAY,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,CAChC,WAAW,CAAC,WAAW,CAAC,QAAkB,CAAC;aAC5C;SACwB,CAAC,CAAC;QAE5B,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IAChD,CAAC;IAED,EAAE,CAAC,eAAe,CAAC,SAAS,EAAE;QAC7B,WAAW,EAAE,6DAA6D;QAC1E,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;KAC9C,CAAC,CAAC;AACJ,CAAC"}

package/index.ts

@@ -0,0 +1,406 @@
+/**
+ * Pi Coding Agent — Yandex Provider Bridge
+ *
+ * Auth (OAuth — default):
+ *   Run /yalogin to authorize. A browser window opens automatically,
+ *   the token is captured via a local callback server on port 7890.
+ *
+ *   YANDEX_OAUTH_TOKEN env var skips the browser flow entirely.
+ *   YANDEX_FOLDER_ID   env var skips the folder ID prompt.
+ *
+ * Auth (static API key — opt-in):
+ *   Set both YANDEX_API_KEY and YANDEX_FOLDER_ID env vars.
+ */
+
+import { createServer } from "http";
+import { readFileSync, writeFileSync } from "fs";
+import { homedir } from "os";
+import { join } from "path";
+import { spawnSync } from "child_process";
+import type { AddressInfo } from "net";
+import type {
+	ExtensionAPI,
+	ExtensionCommandContext,
+	ProviderConfig,
+} from "@earendil-works/pi-coding-agent";
+import type {
+	OAuthCredentials,
+	OAuthLoginCallbacks,
+} from "@earendil-works/pi-ai";
+
+// ─── constants ────────────────────────────────────────────────────────────────
+
+const IAM_TOKEN_URL = "https://iam.api.cloud.yandex.net/iam/v1/tokens";
+const AI_BASE_URL = "https://ai.api.cloud.yandex.net/v1";
+const AUTH_PATH = join(homedir(), ".pi", "agent", "auth.json");
+
+const OAUTH_CLIENT_ID = "0414b7213b22435fa65051f64270584f";
+const OAUTH_CALLBACK_PORT = 7890;
+const OAUTH_REDIRECT_URI = `http://localhost:${OAUTH_CALLBACK_PORT}/callback`;
+const OAUTH_URL =
+	`https://oauth.yandex.ru/authorize?response_type=token` +
+	`&client_id=${OAUTH_CLIENT_ID}` +
+	`&redirect_uri=${encodeURIComponent(OAUTH_REDIRECT_URI)}`;
+
+// ─── model catalogue ──────────────────────────────────────────────────────────
+
+const KNOWN_MODELS = [
+	{
+		slug: "yandexgpt-5.1",
+		name: "YandexGPT Pro 5.1",
+		contextWindow: 128_000,
+		maxTokens: 8_192,
+		cost: { input: 8.8, output: 8.8, cacheRead: 0, cacheWrite: 0 },
+	},
+	{
+		slug: "yandexgpt",
+		name: "YandexGPT Pro",
+		contextWindow: 128_000,
+		maxTokens: 8_192,
+		cost: { input: 8.8, output: 8.8, cacheRead: 0, cacheWrite: 0 },
+	},
+	{
+		slug: "yandexgpt-lite",
+		name: "YandexGPT Lite",
+		contextWindow: 32_000,
+		maxTokens: 4_096,
+		cost: { input: 2.2, output: 2.2, cacheRead: 0, cacheWrite: 0 },
+	},
+] as const;
+
+// ─── helpers ──────────────────────────────────────────────────────────────────
+
+interface IamTokenResponse {
+	iamToken: string;
+	expiresAt: string;
+}
+
+/** Turns `gpt://<folderId>/<slug>/latest` into a readable display name.
+ *  Known slugs get their canonical name; unknown ones get `slug {last4ofFolderId}`. */
+function prettyModelName(id: string): string {
+	const match = id.match(/^gpt:\/\/([^/]+)\/(.+?)(?:\/latest)?$/);
+	if (!match) return id;
+	const [, folderId, slug] = match;
+	const known = KNOWN_MODELS.find((m) => m.slug === slug);
+	if (known) return known.name;
+	return `${slug} {${folderId.slice(-4)}}`;
+}
+
+function openBrowser(url: string) {
+	const cmd =
+		process.platform === "win32"
+			? "start"
+			: process.platform === "darwin"
+				? "open"
+				: "xdg-open";
+	spawnSync(cmd, [url], { stdio: "ignore" });
+}
+
+async function exchangeOAuthForIam(
+	oauthToken: string,
+): Promise<{ token: string; expiresAt: number }> {
+	const res = await fetch(IAM_TOKEN_URL, {
+		method: "POST",
+		headers: { "Content-Type": "application/json" },
+		body: JSON.stringify({ yandexPassportOauthToken: oauthToken }),
+	});
+
+	if (!res.ok) {
+		const text = await res.text().catch(() => res.statusText);
+		throw new Error(`IAM token exchange failed (${res.status}): ${text}`);
+	}
+
+	const data = (await res.json()) as IamTokenResponse;
+	return {
+		token: data.iamToken,
+		expiresAt: new Date(data.expiresAt).getTime(),
+	};
+}
+
+function buildModels(folderId: string) {
+	return KNOWN_MODELS.map((m) => ({
+		id: `gpt://${folderId}/${m.slug}/latest`,
+		name: m.name,
+		api: "openai-responses" as const,
+		provider: "yandex",
+		baseUrl: AI_BASE_URL,
+		reasoning: false,
+		input: ["text"] as ("text" | "image")[],
+		cost: m.cost,
+		contextWindow: m.contextWindow,
+		maxTokens: m.maxTokens,
+		headers: { "OpenAI-Project": folderId },
+		compat: {
+			supportsDeveloperRole: false,
+			supportsReasoningEffort: false,
+			maxTokensField: "max_tokens" as const,
+		},
+	}));
+}
+
+function readAuthJson(): Record<string, unknown> {
+	try {
+		return JSON.parse(readFileSync(AUTH_PATH, "utf8")) as Record<
+			string,
+			unknown
+		>;
+	} catch {
+		return {};
+	}
+}
+
+function writeAuthJson(data: Record<string, unknown>) {
+	writeFileSync(AUTH_PATH, JSON.stringify(data, null, 2), "utf8");
+}
+
+// ─── OAuth local callback server ──────────────────────────────────────────────
+
+const CALLBACK_HTML = `<!DOCTYPE html>
+<html><head><meta charset="utf-8"><title>pi-yandex-bridge</title></head>
+<body style="font-family:sans-serif;padding:2rem">
+<p>Authorizing pi-yandex-bridge…</p>
+<script>
+const p = new URLSearchParams(location.hash.slice(1));
+const t = p.get('access_token');
+if (t) {
+  fetch('/token', { method: 'POST', body: t })
+    .then(() => { document.body.innerHTML = '<p>✓ Authorized. You can close this tab.</p>'; });
+} else {
+  document.body.innerHTML = '<p>Error: no access_token in response.</p>';
+}
+</script>
+</body></html>`;
+
+function captureOAuthToken(
+	onProgress?: (msg: string) => void,
+): Promise<string> {
+	return new Promise((resolve, reject) => {
+		const server = createServer((req, res) => {
+			if (req.url?.startsWith("/callback")) {
+				res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
+				res.end(CALLBACK_HTML);
+			} else if (req.url === "/token" && req.method === "POST") {
+				let body = "";
+				req.on("data", (chunk) => (body += chunk));
+				req.on("end", () => {
+					res.writeHead(200);
+					res.end();
+					server.close();
+					resolve(body.trim());
+				});
+			} else {
+				res.writeHead(404);
+				res.end();
+			}
+		});
+
+		server.on("error", (err: NodeJS.ErrnoException) => {
+			if (err.code === "EADDRINUSE") {
+				reject(
+					new Error(
+						`Port ${OAUTH_CALLBACK_PORT} is already in use. Stop the process using it and try again.`,
+					),
+				);
+			} else {
+				reject(new Error(`Local auth server error: ${err.message}`));
+			}
+		});
+
+		const timeout = setTimeout(
+			() => {
+				server.close();
+				reject(new Error("OAuth authorization timed out after 5 minutes."));
+			},
+			5 * 60 * 1000,
+		);
+
+		server.listen(OAUTH_CALLBACK_PORT, "127.0.0.1", () => {
+			onProgress?.(
+				`[yandex] Listening on port ${(server.address() as AddressInfo).port}. Opening browser…`,
+			);
+			openBrowser(OAUTH_URL);
+		});
+
+		server.on("close", () => clearTimeout(timeout));
+	});
+}
+
+// ─── OAuth provider callbacks ─────────────────────────────────────────────────
+
+async function yandexLogin(
+	callbacks: OAuthLoginCallbacks,
+): Promise<OAuthCredentials> {
+	let oauthToken = process.env.YANDEX_OAUTH_TOKEN ?? "";
+	if (!oauthToken) {
+		oauthToken = await captureOAuthToken((msg) => callbacks.onProgress?.(msg));
+	} else {
+		callbacks.onProgress?.(
+			"[yandex] Using YANDEX_OAUTH_TOKEN from environment.",
+		);
+	}
+
+	let folderId = process.env.YANDEX_FOLDER_ID ?? "";
+	if (!folderId) {
+		folderId = await callbacks.onPrompt({
+			message: "Enter your Yandex Cloud folder ID:",
+			placeholder: "b1g...",
+		});
+	}
+
+	callbacks.onProgress?.("[yandex] Exchanging OAuth token for IAM token…");
+	const iam = await exchangeOAuthForIam(oauthToken);
+
+	return {
+		refresh: oauthToken,
+		access: iam.token,
+		expires: iam.expiresAt,
+		folderId,
+	};
+}
+
+async function yandexRefreshToken(
+	credentials: OAuthCredentials,
+): Promise<OAuthCredentials> {
+	const iam = await exchangeOAuthForIam(credentials.refresh);
+	return { ...credentials, access: iam.token, expires: iam.expiresAt };
+}
+
+// ─── /yalogin command ─────────────────────────────────────────────────────────
+
+async function runYaLogin(ctx: ExtensionCommandContext) {
+	try {
+		ctx.ui.notify("Opening browser for Yandex authorization…", "info");
+
+		const oauthToken = await captureOAuthToken();
+
+		let folderId = process.env.YANDEX_FOLDER_ID ?? "";
+		if (!folderId) {
+			folderId = (await ctx.ui.input("Yandex Cloud folder ID", "b1g...")) ?? "";
+		}
+
+		if (!folderId) {
+			ctx.ui.notify("Login cancelled: folder ID is required.", "error");
+			return;
+		}
+
+		ctx.ui.notify("Exchanging OAuth token for IAM token…", "info");
+		const iam = await exchangeOAuthForIam(oauthToken);
+
+		const auth = readAuthJson();
+		auth.yandex = {
+			type: "oauth",
+			refresh: oauthToken,
+			access: iam.token,
+			expires: iam.expiresAt,
+			folderId,
+		};
+		writeAuthJson(auth);
+
+		ctx.ui.notify(
+			"✓ Yandex credentials saved. Restart Pi to activate the models.",
+			"info",
+		);
+	} catch (err) {
+		ctx.ui.notify(
+			`Yandex login failed: ${err instanceof Error ? err.message : String(err)}`,
+			"error",
+		);
+	}
+}
+
+// ─── extension entry point ────────────────────────────────────────────────────
+
+export default async function (pi: ExtensionAPI) {
+	const apiKey = process.env.YANDEX_API_KEY;
+	const folderId = process.env.YANDEX_FOLDER_ID;
+
+	// Static API key path — both env vars must be present.
+	if (apiKey && folderId) {
+		const modelIds = new Set(
+			KNOWN_MODELS.map((m) => `gpt://${folderId}/${m.slug}/latest`),
+		);
+
+		try {
+			const res = await fetch(`${AI_BASE_URL}/models`, {
+				headers: {
+					Authorization: `Bearer ${apiKey}`,
+					"OpenAI-Project": folderId,
+				},
+			});
+			if (res.ok) {
+				const payload = (await res.json()) as { data: Array<{ id: string }> };
+				for (const { id } of payload.data) modelIds.add(id);
+			}
+		} catch {
+			/* static list is enough */
+		}
+
+		const models = [...modelIds].map((id) => {
+			const slug = id.replace(/^gpt:\/\/[^/]+\//, "").replace(/\/latest$/, "");
+			const known = KNOWN_MODELS.find((m) => m.slug === slug);
+			return {
+				id,
+				name: prettyModelName(id),
+				reasoning: false,
+				input: ["text"] as ("text" | "image")[],
+				cost: known?.cost ?? {
+					input: 0,
+					output: 0,
+					cacheRead: 0,
+					cacheWrite: 0,
+				},
+				contextWindow: known?.contextWindow ?? 128_000,
+				maxTokens: known?.maxTokens ?? 8_192,
+				headers: { "OpenAI-Project": folderId },
+				compat: {
+					supportsDeveloperRole: false,
+					supportsReasoningEffort: false,
+					maxTokensField: "max_tokens" as const,
+				},
+			};
+		});
+
+		pi.registerProvider("yandex", {
+			name: "Yandex Cloud",
+			baseUrl: AI_BASE_URL,
+			apiKey,
+			api: "openai-responses",
+			models,
+		} satisfies ProviderConfig);
+
+		console.log(
+			`[yandex] Registered ${models.length} model(s) with static API key.`,
+		);
+	} else {
+		// OAuth path — seed models from auth.json if folderId is already stored.
+		let storedFolderId: string | undefined;
+		try {
+			const auth = readAuthJson() as Record<string, { folderId?: string }>;
+			storedFolderId = auth.yandex?.folderId;
+		} catch {
+			/* auth.json absent or unreadable */
+		}
+
+		pi.registerProvider("yandex", {
+			name: "Yandex Cloud",
+			baseUrl: AI_BASE_URL,
+			api: "openai-responses",
+			models: storedFolderId ? buildModels(storedFolderId) : [],
+			oauth: {
+				name: "Yandex Cloud (OAuth)",
+				login: yandexLogin,
+				refreshToken: yandexRefreshToken,
+				getApiKey: (credentials) => credentials.access,
+				modifyModels: (_, credentials) =>
+					buildModels(credentials.folderId as string),
+			},
+		} satisfies ProviderConfig);
+
+		console.log("[yandex] Registered with OAuth.");
+	}
+
+	pi.registerCommand("yalogin", {
+		description: "Authorize Yandex Cloud (opens browser, no pasting required)",
+		handler: async (_args, ctx) => runYaLogin(ctx),
+	});
+}

package/package.json

@@ -0,0 +1,32 @@
+{
+  "name": "pi-yandex-bridge",
+  "version": "0.1.0",
+  "description": "Pi Coding Agent provider bridge for Yandex Cloud AI (YandexGPT)",
+  "main": "./dist/index.js",
+  "type": "module",
+  "pi": {
+    "extensions": ["./dist/index.js"],
+    "image": "https://upload.wikimedia.org/wikipedia/commons/5/5b/Yandex_cloud_logo_new.jpg"
+  },
+  "scripts": {
+    "build": "tsc"
+  },
+  "keywords": [
+    "pi-package",
+    "pi",
+    "pi-coding-agent",
+    "yandex",
+    "yandexgpt"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "devDependencies": {
+    "@earendil-works/pi-coding-agent": "latest",
+    "typescript": "^5"
+  },
+  "license": "MIT"
+}

package/tsconfig.json

@@ -0,0 +1,17 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "outDir": "./dist",
+    "rootDir": ".",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true
+  },
+  "include": ["index.ts"],
+  "exclude": ["node_modules", "dist"]
+}