pi-yandex-bridge 0.2.5 → 0.2.9

package/README.md

@@ -4,13 +4,19 @@ 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         |
+All models available in your Yandex Cloud folder are fetched dynamically after authentication. Model names are displayed as `name{folder_last5/tag}` — e.g. `yandexgpt-5.1{cffev/l}`.
 
-Unknown models discovered via the API are displayed as `slug {last4ofFolderId}`.
+## Installation
+
+Add to `~/.pi/agent/settings.json`:
+
+```json
+{
+  "extensions": ["/path/to/pi-yandex-bridge/dist/index.js"]
+}
+```
+
+Then restart Pi and run `/yalogin`.
 
 ## Auth: OAuth (default)
 
@@ -20,9 +26,9 @@ In the [Yandex Cloud console](https://console.yandex.cloud), select your folder.
 
 ### 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.
+Run `/yalogin` in Pi. A browser window opens automatically — authorize the app, and the token is captured without any pasting. Pi then prompts for your folder ID. Available models are fetched immediately after login.
 
-**IAM tokens expire after 12 hours** and are refreshed automatically using the stored OAuth token.
+**IAM tokens expire after 12 hours** and are refreshed automatically using the stored OAuth token. The model list is re-fetched on each refresh.
 
 To skip the browser flow, set env vars before starting Pi:
 
@@ -40,11 +46,4 @@ 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..."
-```
+Models are fetched at startup using the API key. You can generate an API key in the [Yandex AI Studio](https://aistudio.yandex.ru) or in the Yandex Cloud console under **Service accounts → your account → API keys**.

package/index.ts

@@ -17,6 +17,7 @@ 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,
@@ -31,6 +32,7 @@ import type {
 // ─── constants ────────────────────────────────────────────────────────────────
 
 const IAM_TOKEN_URL = "https://iam.api.cloud.yandex.net/iam/v1/tokens";
+const AI_FETCH_URL = "https://llm.api.cloud.yandex.net/v1";
 const AI_BASE_URL = "https://ai.api.cloud.yandex.net/v1";
 const AUTH_PATH = join(homedir(), ".pi", "agent", "auth.json");
 
@@ -42,32 +44,6 @@ const OAUTH_URL =
 	`&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 {
@@ -75,15 +51,34 @@ interface IamTokenResponse {
 	expiresAt: string;
 }
 
-/** Turns `gpt://<folderId>/<slug>/latest` into a readable display name.
- *  Known slugs get their canonical name; unknown ones get `slug {last4ofFolderId}`. */
+// Only applies to yandex gpt:// URIs — leaves all other model IDs untouched.
 function prettyModelName(id: string): string {
-	const match = id.match(/^gpt:\/\/([^/]+)\/(.+?)(?:\/latest)?$/);
+	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)}}`;
+	const [, folderId, slug, hasLatest] = match;
+	const tag = hasLatest ? "/l" : "";
+	return `${slug}{${folderId.slice(-5)}${tag}}`;
+}
+
+function modelEntry(id: string, folderId: string) {
+	return {
+		id,
+		name: prettyModelName(id),
+		api: "openai-responses" as const,
+		provider: "yandex",
+		baseUrl: AI_BASE_URL,
+		reasoning: true,
+		input: ["text"] as ("text" | "image")[],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 128_000,
+		maxTokens: 8_192,
+		headers: { "OpenAI-Project": folderId },
+		compat: {
+			supportsDeveloperRole: false,
+			supportsReasoningEffort: false,
+			maxTokensField: "max_tokens" as const,
+		},
+	};
 }
 
 function openBrowser(url: string) {
@@ -104,12 +99,10 @@ async function exchangeOAuthForIam(
 		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,
@@ -117,25 +110,29 @@ async function exchangeOAuthForIam(
 	};
 }
 
-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,
-		},
-	}));
+// authHeader: 'Bearer <iam_token>' for OAuth, 'Api-Key <api_key>' for static key
+async function fetchModelIds(
+	folderId: string,
+	authHeader: string,
+): Promise<string[]> {
+	const ac = new AbortController();
+	const timer = setTimeout(() => ac.abort(), 5_000);
+	try {
+		const res = await fetch(`${AI_FETCH_URL}/models`, {
+			headers: {
+				Authorization: authHeader,
+				"OpenAI-Project": folderId,
+			},
+			signal: ac.signal,
+		});
+		clearTimeout(timer);
+		if (!res.ok) return [];
+		const payload = (await res.json()) as { data: Array<{ id: string }> };
+		return payload.data.map((m) => m.id);
+	} catch {
+		clearTimeout(timer);
+		return [];
+	}
 }
 
 function readAuthJson(): Record<string, unknown> {
@@ -194,11 +191,7 @@ function captureOAuthToken(): Promise<string> {
 
 		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.`,
-					),
-				);
+				reject(new Error(`Port ${OAUTH_CALLBACK_PORT} is already in use.`));
 			} else {
 				reject(new Error(`Local auth server error: ${err.message}`));
 			}
@@ -213,6 +206,7 @@ function captureOAuthToken(): Promise<string> {
 		);
 
 		server.listen(OAUTH_CALLBACK_PORT, "127.0.0.1", () => {
+			void (server.address() as AddressInfo).port;
 			openBrowser(OAUTH_URL);
 		});
 
@@ -238,13 +232,22 @@ async function yandexLogin(
 			placeholder: "b1g...",
 		});
 	}
+
 	const iam = await exchangeOAuthForIam(oauthToken);
 
+	// Fetch available models while we have fresh credentials.
+	const modelIds = await fetchModelIds(
+		folderId as string,
+		`Bearer ${iam.token}`,
+	);
+
 	return {
 		refresh: oauthToken,
 		access: iam.token,
 		expires: iam.expiresAt,
-		folderId,
+		folderId: folderId as string,
+		// Store fetched model IDs so modifyModels can stay synchronous.
+		modelIds: JSON.stringify(modelIds),
 	};
 }
 
@@ -252,7 +255,17 @@ async function yandexRefreshToken(
 	credentials: OAuthCredentials,
 ): Promise<OAuthCredentials> {
 	const iam = await exchangeOAuthForIam(credentials.refresh);
-	return { ...credentials, access: iam.token, expires: iam.expiresAt };
+	// Re-fetch models on token refresh to pick up any new models.
+	const modelIds = await fetchModelIds(
+		credentials.folderId as string,
+		`Bearer ${iam.token}`,
+	);
+	return {
+		...credentials,
+		access: iam.token,
+		expires: iam.expiresAt,
+		modelIds: JSON.stringify(modelIds),
+	};
 }
 
 // ─── /yalogin command ─────────────────────────────────────────────────────────
@@ -260,7 +273,6 @@ async function yandexRefreshToken(
 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 ?? "";
@@ -276,6 +288,9 @@ async function runYaLogin(ctx: ExtensionCommandContext) {
 		ctx.ui.notify("Exchanging OAuth token for IAM token…", "info");
 		const iam = await exchangeOAuthForIam(oauthToken);
 
+		ctx.ui.notify("Fetching available models…", "info");
+		const modelIds = await fetchModelIds(folderId, `Bearer ${iam.token}`);
+
 		const auth = readAuthJson();
 		auth.yandex = {
 			type: "oauth",
@@ -283,11 +298,12 @@ async function runYaLogin(ctx: ExtensionCommandContext) {
 			access: iam.token,
 			expires: iam.expiresAt,
 			folderId,
+			modelIds: JSON.stringify(modelIds),
 		};
 		writeAuthJson(auth);
 
 		ctx.ui.notify(
-			"✓ Yandex credentials saved. Restart Pi to activate the models.",
+			`✓ Yandex credentials saved (${modelIds.length} models). Restart Pi to activate.`,
 			"info",
 		);
 	} catch (err) {
@@ -304,72 +320,32 @@ 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 ac = new AbortController();
-			const timer = setTimeout(() => ac.abort(), 5_000);
-			const res = await fetch(`${AI_BASE_URL}/models`, {
-				headers: {
-					Authorization: `Bearer ${apiKey}`,
-					"OpenAI-Project": folderId,
-				},
-				signal: ac.signal,
-			});
-			clearTimeout(timer);
-			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),
-				api: "openai-responses" as const,
-				provider: "yandex",
-				baseUrl: AI_BASE_URL,
-				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,
-				},
-			};
-		});
-
+		// Static API key — fetch models immediately, we have credentials.
+		const modelIds = await fetchModelIds(folderId, `Api-Key ${apiKey}`);
 		pi.registerProvider("yandex", {
 			name: "Yandex Cloud",
 			baseUrl: AI_BASE_URL,
 			apiKey,
 			api: "openai-responses",
-			models,
+			models: modelIds.map((id) => modelEntry(id, folderId)),
 		} satisfies ProviderConfig);
 	} else {
-		// OAuth path — seed models from auth.json if folderId is already stored.
-		let storedFolderId: string | undefined;
+		// OAuth path — seed from auth.json if credentials are already stored.
+		let seedModels: ReturnType<typeof modelEntry>[] = [];
 		try {
-			const auth = readAuthJson() as Record<string, { folderId?: string }>;
-			storedFolderId = auth.yandex?.folderId;
+			const auth = readAuthJson() as Record<
+				string,
+				{
+					folderId?: string;
+					modelIds?: string;
+				}
+			>;
+			const stored = auth.yandex;
+			if (stored?.folderId && stored?.modelIds) {
+				const ids = JSON.parse(stored.modelIds) as string[];
+				seedModels = ids.map((id) => modelEntry(id, stored.folderId!));
+			}
 		} catch {
 			/* auth.json absent or unreadable */
 		}
@@ -378,14 +354,22 @@ export default async function (pi: ExtensionAPI) {
 			name: "Yandex Cloud",
 			baseUrl: AI_BASE_URL,
 			api: "openai-responses",
-			models: storedFolderId ? buildModels(storedFolderId) : [],
+			models: seedModels,
 			oauth: {
 				name: "Yandex Cloud (OAuth)",
 				login: yandexLogin,
 				refreshToken: yandexRefreshToken,
 				getApiKey: (credentials) => credentials.access,
-				modifyModels: (_, credentials) =>
-					buildModels(credentials.folderId as string),
+				modifyModels: (models, credentials) => {
+					const fId = credentials.folderId as string;
+					const ids: string[] = credentials.modelIds
+						? (JSON.parse(credentials.modelIds as string) as string[])
+						: [];
+					return [
+						...models.filter((m) => m.provider !== "yandex"),
+						...ids.map((id) => modelEntry(id, fId)),
+					];
+				},
 			},
 		} satisfies ProviderConfig);
 	}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-yandex-bridge",
-  "version": "0.2.5",
+  "version": "0.2.9",
   "description": "Pi Coding Agent provider bridge for Yandex Cloud AI (YandexGPT)",
   "main": "./dist/index.js",
   "type": "module",

package/.pi-lens/cache/jscpd.json

@@ -1,7 +0,0 @@
-{
-  "success": true,
-  "clones": [],
-  "duplicatedLines": 0,
-  "totalLines": 400,
-  "percentage": 0
-}

package/.pi-lens/cache/jscpd.meta.json

@@ -1,4 +0,0 @@
-{
-  "timestamp": "2026-05-09T06:22:47.142Z",
-  "scanDurationMs": 1316
-}

package/.pi-lens/cache/knip.json

@@ -1,21 +0,0 @@
-{
-  "success": true,
-  "issues": [
-    {
-      "type": "file",
-      "name": "dist/index.d.ts",
-      "file": "dist/index.d.ts"
-    }
-  ],
-  "unusedExports": [],
-  "unusedFiles": [
-    {
-      "type": "file",
-      "name": "dist/index.d.ts",
-      "file": "dist/index.d.ts"
-    }
-  ],
-  "unusedDeps": [],
-  "unlistedDeps": [],
-  "summary": "Found 1 issues"
-}

package/.pi-lens/cache/knip.meta.json

@@ -1,3 +0,0 @@
-{
-  "timestamp": "2026-05-09T06:42:14.808Z"
-}

package/.pi-lens/cache/session-start-guidance.json

@@ -1 +0,0 @@
-null

package/.pi-lens/cache/session-start-guidance.meta.json

@@ -1,3 +0,0 @@
-{
-  "timestamp": "2026-05-09T06:23:50.558Z"
-}

package/.pi-lens/cache/todo-baseline.json

@@ -1,3 +0,0 @@
-{
-  "items": []
-}

package/.pi-lens/cache/todo-baseline.meta.json

@@ -1,3 +0,0 @@
-{
-  "timestamp": "2026-05-09T06:22:45.815Z"
-}

package/.pi-lens/cache/turn-end-findings-last.json

@@ -1,4 +0,0 @@
-{
-  "signature": "index.ts::🔴 New unresolved imports/deps in modified code (Knip):\n  index.ts:29 — unlisted: @earendil-works/pi-ai\n  First location: index.ts\n",
-  "sessionId": "lens-moxfkr7l-92597cdf"
-}

package/.pi-lens/cache/turn-end-findings-last.meta.json

@@ -1,3 +0,0 @@
-{
-  "timestamp": "2026-05-08T21:37:51.694Z"
-}

package/.pi-lens/cache/turn-end-findings.json

@@ -1 +0,0 @@
-null

package/.pi-lens/cache/turn-end-findings.meta.json

@@ -1,3 +0,0 @@
-{
-  "timestamp": "2026-05-08T21:37:55.621Z"
-}