@cloudflare/aig-pi-extension 2026.9.18

package/README.md

@@ -0,0 +1,25 @@
+# Cloudflare AI Gateway extension for Pi
+
+Adds the `cloudflare/auto` Smart Router model to Pi through Cloudflare's internal AI gateway. It records routing diagnostics silently and shows the latest decision on demand through `/aig-show`, without occupying persistent UI space. It registers an isolated provider without replacing Pi's built-in provider or model catalog.
+
+## Prerequisites
+
+- `cloudflared`
+- Cloudflare Access entitlement to `opencode.cloudflare.dev`
+- Pi 0.84.4
+
+Authentication uses the same pinned `cloudflared access login` flow as `pi-cf-internal-gateway`. Tokens remain in memory and are not written to Pi's credential store.
+
+## Install
+
+```bash
+pi install ./packages/aig-pi-extension
+```
+
+Select `cloudflare-auto/cloudflare/auto` with `/model` or start Pi with:
+
+```bash
+pi --model cloudflare-auto/cloudflare/auto
+```
+
+Pi estimates cost using the internal catalog's `cloudflare/auto` rate. AI Gateway billing and logs report the concrete routed model and actual cost.

package/package.json

@@ -0,0 +1,50 @@
+{
+  "name": "@cloudflare/aig-pi-extension",
+  "version": "2026.9.18",
+  "private": false,
+  "description": "Cloudflare AI Gateway Smart Router support and routing diagnostics for Pi",
+  "keywords": [
+    "pi-package",
+    "cloudflare",
+    "ai-gateway",
+    "smart-router"
+  ],
+  "sideEffects": true,
+  "type": "module",
+  "exports": {
+    ".": "./src/index.ts"
+  },
+  "main": "src/index.ts",
+  "files": [
+    "src",
+    "README.md",
+    "!**/*.spec.ts"
+  ],
+  "devDependencies": {
+    "@earendil-works/pi-ai": "0.84.4",
+    "@earendil-works/pi-coding-agent": "0.84.4",
+    "vitest": "4.1.10",
+    "@repo/tools": "0.0.0",
+    "@repo/eslint-config": "0.0.0",
+    "@repo/typescript-config": "0.0.0"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-ai": "0.84.4",
+    "@earendil-works/pi-coding-agent": "0.84.4"
+  },
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org/"
+  },
+  "pi": {
+    "extensions": [
+      "./src/index.ts"
+    ]
+  },
+  "scripts": {
+    "check:lint": "run-biome && run-eslint",
+    "check:types": "run-tsc",
+    "test": "run-vitest",
+    "test:ci": "run-vitest-ci"
+  }
+}

package/src/index.ts

@@ -0,0 +1,70 @@
+import type { ExtensionAPI } from '@earendil-works/pi-coding-agent'
+
+interface RoutingResult {
+  requestedModel: string
+  routedModel: string
+  reason: string
+  decisionId: string
+  requestId: string
+}
+
+export default function aigPiExtension(pi: ExtensionAPI): void {
+  let latest: RoutingResult | undefined
+
+  pi.registerProvider('cloudflare-auto', {
+    name: 'Cloudflare Auto (CF Internal Gateway)',
+    api: 'openai-completions',
+    baseUrl: 'https://gateway.opencode.cloudflare.dev/compat',
+    apiKey: 'cf-internal-gateway',
+    headers: {
+      'cf-access-token':
+        '!cloudflared access login --no-verbose -app=https://opencode.cloudflare.dev',
+      'X-Requested-With': 'xmlhttprequest',
+    },
+    models: [
+      {
+        id: 'cloudflare/auto',
+        name: 'Cloudflare Auto (dynamic pricing)',
+        reasoning: false,
+        input: ['text', 'image'],
+        cost: { input: 2, output: 12, cacheRead: 0.2, cacheWrite: 2.5 },
+        contextWindow: 1_000_000,
+        maxTokens: 128_000,
+      },
+    ],
+  })
+
+  pi.on('after_provider_response', (event, ctx) => {
+    const headers = event.headers
+    if (
+      !headers['cf-aig-routed-model'] &&
+      !headers['cf-aig-routing-reason'] &&
+      !headers['cf-aig-routing-decision-id']
+    )
+      return
+
+    latest = {
+      requestedModel: ctx.model?.id ?? 'unknown',
+      routedModel: headers['cf-aig-routed-model'] ?? 'unknown',
+      reason: headers['cf-aig-routing-reason'] ?? 'unknown',
+      decisionId: headers['cf-aig-routing-decision-id'] ?? 'unknown',
+      requestId: headers['cf-aig-request-id'] ?? 'unknown',
+    }
+  })
+
+  pi.registerCommand('aig-show', {
+    description: 'Show the latest AI Gateway Smart Router decision',
+    handler: async (_args, ctx) => {
+      const message = latest
+        ? [
+            `requested: ${latest.requestedModel}`,
+            `routed: ${latest.routedModel}`,
+            `reason: ${latest.reason}`,
+            `decision: ${latest.decisionId}`,
+            `request: ${latest.requestId}`,
+          ].join('\n')
+        : 'No AI Gateway routing result captured yet.'
+      ctx.ui.notify(message, 'info')
+    },
+  })
+}