@mariozechner/pi-coding-agent 0.49.3 → 0.50.0

package/examples/extensions/custom-provider-gitlab-duo/index.ts

@@ -0,0 +1,349 @@
+/**
+ * GitLab Duo Provider Extension
+ *
+ * Provides access to GitLab Duo AI models (Claude and GPT) through GitLab's AI Gateway.
+ * Delegates to pi-ai's built-in Anthropic and OpenAI streaming implementations.
+ *
+ * Usage:
+ *   pi -e ./packages/coding-agent/examples/extensions/custom-provider-gitlab-duo
+ *   # Then /login gitlab-duo, or set GITLAB_TOKEN=glpat-...
+ */
+
+import {
+	type Api,
+	type AssistantMessageEventStream,
+	type Context,
+	createAssistantMessageEventStream,
+	type Model,
+	type OAuthCredentials,
+	type OAuthLoginCallbacks,
+	type SimpleStreamOptions,
+	streamSimpleAnthropic,
+	streamSimpleOpenAIResponses,
+} from "@mariozechner/pi-ai";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+
+// =============================================================================
+// Constants
+// =============================================================================
+
+const GITLAB_COM_URL = "https://gitlab.com";
+const AI_GATEWAY_URL = "https://cloud.gitlab.com";
+const ANTHROPIC_PROXY_URL = `${AI_GATEWAY_URL}/ai/v1/proxy/anthropic/`;
+const OPENAI_PROXY_URL = `${AI_GATEWAY_URL}/ai/v1/proxy/openai/v1`;
+
+const BUNDLED_CLIENT_ID = "da4edff2e6ebd2bc3208611e2768bc1c1dd7be791dc5ff26ca34ca9ee44f7d4b";
+const OAUTH_SCOPES = ["api"];
+const REDIRECT_URI = "http://127.0.0.1:8080/callback";
+const DIRECT_ACCESS_TTL = 25 * 60 * 1000;
+
+// =============================================================================
+// Models - exported for use by tests
+// =============================================================================
+
+type Backend = "anthropic" | "openai";
+
+interface GitLabModel {
+	id: string;
+	name: string;
+	backend: Backend;
+	baseUrl: string;
+	reasoning: boolean;
+	input: ("text" | "image")[];
+	cost: { input: number; output: number; cacheRead: number; cacheWrite: number };
+	contextWindow: number;
+	maxTokens: number;
+}
+
+export const MODELS: GitLabModel[] = [
+	// Anthropic
+	{
+		id: "claude-opus-4-5-20251101",
+		name: "Claude Opus 4.5",
+		backend: "anthropic",
+		baseUrl: ANTHROPIC_PROXY_URL,
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 15, output: 75, cacheRead: 1.5, cacheWrite: 18.75 },
+		contextWindow: 200000,
+		maxTokens: 32000,
+	},
+	{
+		id: "claude-sonnet-4-5-20250929",
+		name: "Claude Sonnet 4.5",
+		backend: "anthropic",
+		baseUrl: ANTHROPIC_PROXY_URL,
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+		contextWindow: 200000,
+		maxTokens: 16384,
+	},
+	{
+		id: "claude-haiku-4-5-20251001",
+		name: "Claude Haiku 4.5",
+		backend: "anthropic",
+		baseUrl: ANTHROPIC_PROXY_URL,
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 1, output: 5, cacheRead: 0.1, cacheWrite: 1.25 },
+		contextWindow: 200000,
+		maxTokens: 8192,
+	},
+	// OpenAI (all use Responses API)
+	{
+		id: "gpt-5.1-2025-11-13",
+		name: "GPT-5.1",
+		backend: "openai",
+		baseUrl: OPENAI_PROXY_URL,
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 2.5, output: 10, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 128000,
+		maxTokens: 16384,
+	},
+	{
+		id: "gpt-5-mini-2025-08-07",
+		name: "GPT-5 Mini",
+		backend: "openai",
+		baseUrl: OPENAI_PROXY_URL,
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0.15, output: 0.6, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 128000,
+		maxTokens: 16384,
+	},
+	{
+		id: "gpt-5-codex",
+		name: "GPT-5 Codex",
+		backend: "openai",
+		baseUrl: OPENAI_PROXY_URL,
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 2.5, output: 10, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 128000,
+		maxTokens: 16384,
+	},
+];
+
+const MODEL_MAP = new Map(MODELS.map((m) => [m.id, m]));
+
+// =============================================================================
+// Direct Access Token Cache
+// =============================================================================
+
+interface DirectAccessToken {
+	token: string;
+	headers: Record<string, string>;
+	expiresAt: number;
+}
+
+let cachedDirectAccess: DirectAccessToken | null = null;
+
+async function getDirectAccessToken(gitlabAccessToken: string): Promise<DirectAccessToken> {
+	const now = Date.now();
+	if (cachedDirectAccess && cachedDirectAccess.expiresAt > now) {
+		return cachedDirectAccess;
+	}
+
+	const response = await fetch(`${GITLAB_COM_URL}/api/v4/ai/third_party_agents/direct_access`, {
+		method: "POST",
+		headers: { Authorization: `Bearer ${gitlabAccessToken}`, "Content-Type": "application/json" },
+		body: JSON.stringify({ feature_flags: { DuoAgentPlatformNext: true } }),
+	});
+
+	if (!response.ok) {
+		const errorText = await response.text();
+		if (response.status === 403) {
+			throw new Error(
+				`GitLab Duo access denied. Ensure GitLab Duo is enabled for your account. Error: ${errorText}`,
+			);
+		}
+		throw new Error(`Failed to get direct access token: ${response.status} ${errorText}`);
+	}
+
+	const data = (await response.json()) as { token: string; headers: Record<string, string> };
+	cachedDirectAccess = { token: data.token, headers: data.headers, expiresAt: now + DIRECT_ACCESS_TTL };
+	return cachedDirectAccess;
+}
+
+function invalidateDirectAccessToken() {
+	cachedDirectAccess = null;
+}
+
+// =============================================================================
+// OAuth
+// =============================================================================
+
+async function generatePKCE(): Promise<{ verifier: string; challenge: string }> {
+	const array = new Uint8Array(32);
+	crypto.getRandomValues(array);
+	const verifier = btoa(String.fromCharCode(...array))
+		.replace(/\+/g, "-")
+		.replace(/\//g, "_")
+		.replace(/=+$/, "");
+	const hash = await crypto.subtle.digest("SHA-256", new TextEncoder().encode(verifier));
+	const challenge = btoa(String.fromCharCode(...new Uint8Array(hash)))
+		.replace(/\+/g, "-")
+		.replace(/\//g, "_")
+		.replace(/=+$/, "");
+	return { verifier, challenge };
+}
+
+async function loginGitLab(callbacks: OAuthLoginCallbacks): Promise<OAuthCredentials> {
+	const { verifier, challenge } = await generatePKCE();
+	const authParams = new URLSearchParams({
+		client_id: BUNDLED_CLIENT_ID,
+		redirect_uri: REDIRECT_URI,
+		response_type: "code",
+		scope: OAUTH_SCOPES.join(" "),
+		code_challenge: challenge,
+		code_challenge_method: "S256",
+		state: crypto.randomUUID(),
+	});
+
+	callbacks.onAuth({ url: `${GITLAB_COM_URL}/oauth/authorize?${authParams.toString()}` });
+	const callbackUrl = await callbacks.onPrompt({ message: "Paste the callback URL:" });
+	const code = new URL(callbackUrl).searchParams.get("code");
+	if (!code) throw new Error("No authorization code found in callback URL");
+
+	const tokenResponse = await fetch(`${GITLAB_COM_URL}/oauth/token`, {
+		method: "POST",
+		headers: { "Content-Type": "application/x-www-form-urlencoded" },
+		body: new URLSearchParams({
+			client_id: BUNDLED_CLIENT_ID,
+			grant_type: "authorization_code",
+			code,
+			code_verifier: verifier,
+			redirect_uri: REDIRECT_URI,
+		}).toString(),
+	});
+
+	if (!tokenResponse.ok) throw new Error(`Token exchange failed: ${await tokenResponse.text()}`);
+	const data = (await tokenResponse.json()) as {
+		access_token: string;
+		refresh_token: string;
+		expires_in: number;
+		created_at: number;
+	};
+	invalidateDirectAccessToken();
+	return {
+		refresh: data.refresh_token,
+		access: data.access_token,
+		expires: (data.created_at + data.expires_in) * 1000 - 5 * 60 * 1000,
+	};
+}
+
+async function refreshGitLabToken(credentials: OAuthCredentials): Promise<OAuthCredentials> {
+	const response = await fetch(`${GITLAB_COM_URL}/oauth/token`, {
+		method: "POST",
+		headers: { "Content-Type": "application/x-www-form-urlencoded" },
+		body: new URLSearchParams({
+			client_id: BUNDLED_CLIENT_ID,
+			grant_type: "refresh_token",
+			refresh_token: credentials.refresh,
+		}).toString(),
+	});
+	if (!response.ok) throw new Error(`Token refresh failed: ${await response.text()}`);
+	const data = (await response.json()) as {
+		access_token: string;
+		refresh_token: string;
+		expires_in: number;
+		created_at: number;
+	};
+	invalidateDirectAccessToken();
+	return {
+		refresh: data.refresh_token,
+		access: data.access_token,
+		expires: (data.created_at + data.expires_in) * 1000 - 5 * 60 * 1000,
+	};
+}
+
+// =============================================================================
+// Stream Function
+// =============================================================================
+
+export function streamGitLabDuo(
+	model: Model<Api>,
+	context: Context,
+	options?: SimpleStreamOptions,
+): AssistantMessageEventStream {
+	const stream = createAssistantMessageEventStream();
+
+	(async () => {
+		try {
+			const gitlabAccessToken = options?.apiKey;
+			if (!gitlabAccessToken) throw new Error("No GitLab access token. Run /login gitlab-duo or set GITLAB_TOKEN");
+
+			const cfg = MODEL_MAP.get(model.id);
+			if (!cfg) throw new Error(`Unknown model: ${model.id}`);
+
+			const directAccess = await getDirectAccessToken(gitlabAccessToken);
+			const modelWithBaseUrl = { ...model, baseUrl: cfg.baseUrl };
+			const headers = { ...directAccess.headers, Authorization: `Bearer ${directAccess.token}` };
+			const streamOptions = { ...options, apiKey: "gitlab-duo", headers };
+
+			const innerStream =
+				cfg.backend === "anthropic"
+					? streamSimpleAnthropic(modelWithBaseUrl as Model<"anthropic-messages">, context, streamOptions)
+					: streamSimpleOpenAIResponses(modelWithBaseUrl as Model<"openai-responses">, context, streamOptions);
+
+			for await (const event of innerStream) stream.push(event);
+			stream.end();
+		} catch (error) {
+			stream.push({
+				type: "error",
+				reason: "error",
+				error: {
+					role: "assistant",
+					content: [],
+					api: model.api,
+					provider: model.provider,
+					model: model.id,
+					usage: {
+						input: 0,
+						output: 0,
+						cacheRead: 0,
+						cacheWrite: 0,
+						totalTokens: 0,
+						cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+					},
+					stopReason: "error",
+					errorMessage: error instanceof Error ? error.message : String(error),
+					timestamp: Date.now(),
+				},
+			});
+			stream.end();
+		}
+	})();
+
+	return stream;
+}
+
+// =============================================================================
+// Extension Entry Point
+// =============================================================================
+
+export default function (pi: ExtensionAPI) {
+	pi.registerProvider("gitlab-duo", {
+		baseUrl: AI_GATEWAY_URL,
+		apiKey: "GITLAB_TOKEN",
+		api: "gitlab-duo-api",
+		models: MODELS.map(({ id, name, reasoning, input, cost, contextWindow, maxTokens }) => ({
+			id,
+			name,
+			reasoning,
+			input,
+			cost,
+			contextWindow,
+			maxTokens,
+		})),
+		oauth: {
+			name: "GitLab Duo",
+			login: loginGitLab,
+			refreshToken: refreshGitLabToken,
+			getApiKey: (cred) => cred.access,
+		},
+		streamSimple: streamGitLabDuo,
+	});
+}

package/examples/extensions/custom-provider-gitlab-duo/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "pi-extension-custom-provider-gitlab-duo",
+  "private": true,
+  "version": "1.1.0",
+  "type": "module",
+  "scripts": {
+    "clean": "echo 'nothing to clean'",
+    "build": "echo 'nothing to build'",
+    "check": "echo 'nothing to check'"
+  },
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  }
+}

package/examples/extensions/custom-provider-gitlab-duo/test.ts

@@ -0,0 +1,82 @@
+/**
+ * Test script for GitLab Duo extension
+ * Run: npx tsx test.ts [model-id] [--thinking]
+ *
+ * Examples:
+ *   npx tsx test.ts                              # Test default (claude-sonnet-4-5-20250929)
+ *   npx tsx test.ts gpt-5-codex                  # Test GPT-5 Codex
+ *   npx tsx test.ts claude-sonnet-4-5-20250929 --thinking
+ */
+
+import { type Api, type Context, type Model, registerApiProvider, streamSimple } from "@mariozechner/pi-ai";
+import { readFileSync } from "fs";
+import { homedir } from "os";
+import { join } from "path";
+import { MODELS, streamGitLabDuo } from "./index.js";
+
+const MODEL_MAP = new Map(MODELS.map((m) => [m.id, m]));
+
+async function main() {
+	const modelId = process.argv[2] || "claude-sonnet-4-5-20250929";
+	const useThinking = process.argv.includes("--thinking");
+
+	const cfg = MODEL_MAP.get(modelId);
+	if (!cfg) {
+		console.error(`Unknown model: ${modelId}`);
+		console.error("Available:", MODELS.map((m) => m.id).join(", "));
+		process.exit(1);
+	}
+
+	// Read auth
+	const authPath = join(homedir(), ".pi", "agent", "auth.json");
+	const authData = JSON.parse(readFileSync(authPath, "utf-8"));
+	const gitlabCred = authData["gitlab-duo"];
+	if (!gitlabCred?.access) {
+		console.error("No gitlab-duo credentials. Run /login gitlab-duo first.");
+		process.exit(1);
+	}
+
+	// Register provider
+	registerApiProvider({
+		api: "gitlab-duo-api" as Api,
+		stream: streamGitLabDuo,
+		streamSimple: streamGitLabDuo,
+	});
+
+	// Create model
+	const model: Model<Api> = {
+		id: cfg.id,
+		name: cfg.name,
+		api: "gitlab-duo-api" as Api,
+		provider: "gitlab-duo",
+		baseUrl: cfg.baseUrl,
+		reasoning: cfg.reasoning,
+		input: cfg.input,
+		cost: cfg.cost,
+		contextWindow: cfg.contextWindow,
+		maxTokens: cfg.maxTokens,
+	};
+
+	const context: Context = {
+		messages: [{ role: "user", content: "Say hello in exactly 3 words.", timestamp: Date.now() }],
+	};
+
+	console.log(`Model: ${model.id}, Backend: ${cfg.backend}, Thinking: ${useThinking}`);
+
+	const stream = streamSimple(model, context, {
+		apiKey: gitlabCred.access,
+		maxTokens: 100,
+		reasoning: useThinking ? "low" : undefined,
+	});
+
+	for await (const event of stream) {
+		if (event.type === "thinking_start") console.log("[Thinking]");
+		else if (event.type === "thinking_delta") process.stdout.write(event.delta);
+		else if (event.type === "thinking_end") console.log("\n[/Thinking]\n");
+		else if (event.type === "text_delta") process.stdout.write(event.delta);
+		else if (event.type === "error") console.error("\nError:", event.error.errorMessage);
+		else if (event.type === "done") console.log("\n\nDone!", event.reason, event.message.usage);
+	}
+}
+
+main().catch(console.error);

package/examples/extensions/doom-overlay/doom/build.sh

@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/usr/bin/env bash
 # Build DOOM for pi-doom using doomgeneric and Emscripten
 
 set -e

package/examples/extensions/event-bus.ts

@@ -0,0 +1,43 @@
+/**
+ * Inter-extension event bus example.
+ *
+ * Shows pi.events for communication between extensions. One extension
+ * can emit events that other extensions listen to.
+ *
+ * Usage: /emit [event-name] [data] - emit an event on the bus
+ */
+
+import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+
+export default function (pi: ExtensionAPI) {
+	// Store ctx for use in event handler
+	let currentCtx: ExtensionContext | undefined;
+
+	pi.on("session_start", async (_event, ctx) => {
+		currentCtx = ctx;
+	});
+
+	// Listen for events from other extensions
+	pi.events.on("my:notification", (data) => {
+		const { message, from } = data as { message: string; from: string };
+		currentCtx?.ui.notify(`Event from ${from}: ${message}`, "info");
+	});
+
+	// Command to emit events (emits "my:notification" which the listener above receives)
+	pi.registerCommand("emit", {
+		description: "Emit my:notification event (usage: /emit message)",
+		handler: async (args, _ctx) => {
+			const message = args.trim() || "hello";
+			pi.events.emit("my:notification", { message, from: "/emit command" });
+			// Listener above will show the notification
+		},
+	});
+
+	// Example: emit on session start
+	pi.on("session_start", async () => {
+		pi.events.emit("my:notification", {
+			message: "Session started",
+			from: "event-bus-example",
+		});
+	});
+}

package/examples/extensions/message-renderer.ts

@@ -0,0 +1,59 @@
+/**
+ * Custom message rendering example.
+ *
+ * Shows how to use registerMessageRenderer to control how custom messages
+ * appear in the TUI, with colors, formatting, and expandable details.
+ *
+ * Usage: /status [message] - sends a status message with custom rendering
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { Box, Text } from "@mariozechner/pi-tui";
+
+export default function (pi: ExtensionAPI) {
+	// Register custom renderer for "status-update" messages
+	pi.registerMessageRenderer("status-update", (message, { expanded }, theme) => {
+		const details = message.details as { level: string; timestamp: number } | undefined;
+		const level = details?.level ?? "info";
+
+		// Color based on level
+		const color = level === "error" ? "error" : level === "warn" ? "warning" : "success";
+		const prefix = theme.fg(color, `[${level.toUpperCase()}]`);
+
+		let text = `${prefix} ${message.content}`;
+
+		// Show timestamp when expanded
+		if (expanded && details?.timestamp) {
+			const time = new Date(details.timestamp).toLocaleTimeString();
+			text += `\n${theme.fg("dim", `  at ${time}`)}`;
+		}
+
+		// Use Box with customMessageBg for consistent styling
+		const box = new Box(1, 1, (t) => theme.bg("customMessageBg", t));
+		box.addChild(new Text(text, 0, 0));
+		return box;
+	});
+
+	// Command to send status messages
+	pi.registerCommand("status", {
+		description: "Send a status message (usage: /status [warn|error] message)",
+		handler: async (args, _ctx) => {
+			const parts = args.trim().split(/\s+/);
+			let level = "info";
+			let content = args.trim();
+
+			// Check for level prefix
+			if (parts[0] === "warn" || parts[0] === "error") {
+				level = parts[0];
+				content = parts.slice(1).join(" ") || "Status update";
+			}
+
+			pi.sendMessage({
+				customType: "status-update",
+				content,
+				display: true,
+				details: { level, timestamp: Date.now() },
+			});
+		},
+	});
+}

package/examples/extensions/session-name.ts

@@ -0,0 +1,27 @@
+/**
+ * Session naming example.
+ *
+ * Shows setSessionName/getSessionName to give sessions friendly names
+ * that appear in the session selector instead of the first message.
+ *
+ * Usage: /session-name [name] - set or show session name
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+
+export default function (pi: ExtensionAPI) {
+	pi.registerCommand("session-name", {
+		description: "Set or show session name (usage: /session-name [new name])",
+		handler: async (args, ctx) => {
+			const name = args.trim();
+
+			if (name) {
+				pi.setSessionName(name);
+				ctx.ui.notify(`Session named: ${name}`, "info");
+			} else {
+				const current = pi.getSessionName();
+				ctx.ui.notify(current ? `Session: ${current}` : "No session name set", "info");
+			}
+		},
+	});
+}

package/examples/extensions/with-deps/package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "pi-extension-with-deps",
-  "version": "1.13.3",
+  "version": "1.14.0",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "pi-extension-with-deps",
-      "version": "1.13.3",
+      "version": "1.14.0",
       "dependencies": {
         "ms": "^2.1.3"
       },

package/examples/extensions/with-deps/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-extension-with-deps",
   "private": true,
-  "version": "1.13.3",
+  "version": "1.14.0",
   "type": "module",
   "scripts": {
     "clean": "echo 'nothing to clean'",

package/examples/sdk/02-custom-model.ts

@@ -5,11 +5,11 @@
  */
 
 import { getModel } from "@mariozechner/pi-ai";
-import { createAgentSession, discoverAuthStorage, discoverModels } from "@mariozechner/pi-coding-agent";
+import { AuthStorage, createAgentSession, ModelRegistry } from "@mariozechner/pi-coding-agent";
 
 // Set up auth storage and model registry
-const authStorage = discoverAuthStorage();
-const modelRegistry = discoverModels(authStorage);
+const authStorage = new AuthStorage();
+const modelRegistry = new ModelRegistry(authStorage);
 
 // Option 1: Find a specific built-in model by provider/id
 const opus = getModel("anthropic", "claude-opus-4-5");

package/examples/sdk/03-custom-prompt.ts

@@ -4,12 +4,19 @@
  * Shows how to replace or modify the default system prompt.
  */
 
-import { createAgentSession, SessionManager } from "@mariozechner/pi-coding-agent";
+import { createAgentSession, DefaultResourceLoader, SessionManager } from "@mariozechner/pi-coding-agent";
 
 // Option 1: Replace prompt entirely
-const { session: session1 } = await createAgentSession({
-	systemPrompt: `You are a helpful assistant that speaks like a pirate.
+const loader1 = new DefaultResourceLoader({
+	systemPromptOverride: () => `You are a helpful assistant that speaks like a pirate.
 Always end responses with "Arrr!"`,
+	// Needed to avoid DefaultResourceLoader appending APPEND_SYSTEM.md from ~/.pi/agent or <cwd>/.pi.
+	appendSystemPromptOverride: () => [],
+});
+await loader1.reload();
+
+const { session: session1 } = await createAgentSession({
+	resourceLoader: loader1,
 	sessionManager: SessionManager.inMemory(),
 });
 
@@ -23,13 +30,17 @@ console.log("=== Replace prompt ===");
 await session1.prompt("What is 2 + 2?");
 console.log("\n");
 
-// Option 2: Modify default prompt (receives default, returns modified)
-const { session: session2 } = await createAgentSession({
-	systemPrompt: (defaultPrompt) => `${defaultPrompt}
+// Option 2: Append instructions to the default prompt
+const loader2 = new DefaultResourceLoader({
+	appendSystemPromptOverride: (base) => [
+		...base,
+		"## Additional Instructions\n- Always be concise\n- Use bullet points when listing things",
+	],
+});
+await loader2.reload();
 
-## Additional Instructions
-- Always be concise
-- Use bullet points when listing things`,
+const { session: session2 } = await createAgentSession({
+	resourceLoader: loader2,
 	sessionManager: SessionManager.inMemory(),
 });