@carter-mcalister/pi-auto-name 0.1.1

package/README.md

@@ -0,0 +1,48 @@
+# `@carter-mcalister/pi-auto-name`
+
+English-only automatic session naming for [Pi Coding Agent](https://github.com/badlogic/pi-mono).
+
+This package replaces `@ryan_nookpi/pi-extension-auto-name` with the same basic behavior, but it forces generated session titles to be in English.
+
+## What it does
+
+- Watches the first user prompt in a session
+- Generates a short session title with the current model
+- Forces the generated title to be English-only
+- Applies the title through `pi.setSessionName()`
+- Mirrors the title into the Pi status area and terminal title
+- Skips subagent sessions
+
+## Why this exists
+
+The original package was prompting the model in Korean, which caused auto-generated session names to show up in Korean in the Pi session list and tree.
+
+This replacement keeps the auto-naming workflow while switching the prompt and context text to English.
+
+## Install
+
+```bash
+pi install /Users/carter/Developer/repos/pi-packages/packages/pi-auto-name
+```
+
+If Pi is already running, use `/reload` after installing the extension.
+
+## Remove the original package
+
+```bash
+pi remove npm:@ryan_nookpi/pi-extension-auto-name
+```
+
+## Notes
+
+- This affects session display names, not Pi compaction summaries.
+- If the model or auth is unavailable when a session starts, the session will simply remain unnamed.
+- Titles are intentionally short and clipped to 30 characters.
+
+## Development
+
+```bash
+mise install
+bun install
+mise run check
+```

package/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "@carter-mcalister/pi-auto-name",
+  "version": "0.1.1",
+  "description": "English-only automatic session naming for Pi",
+  "author": "Carter McAlister",
+  "type": "module",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/CarterMcAlister/pi-packages.git",
+    "directory": "packages/pi-auto-name"
+  },
+  "bugs": {
+    "url": "https://github.com/CarterMcAlister/pi-packages/issues"
+  },
+  "homepage": "https://github.com/CarterMcAlister/pi-packages/tree/main/packages/pi-auto-name#readme",
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "src",
+    "README.md"
+  ],
+  "keywords": [
+    "pi-package",
+    "pi-extension",
+    "pi",
+    "session-name",
+    "auto-name"
+  ],
+  "scripts": {
+    "format": "bunx @biomejs/biome check --write .",
+    "lint": "bunx @biomejs/biome check .",
+    "typecheck": "bunx tsc -p tsconfig.json --noEmit",
+    "test": "bun test",
+    "check": "bun run lint && bun run typecheck && bun run test"
+  },
+  "pi": {
+    "extensions": [
+      "./src/index.ts"
+    ]
+  },
+  "peerDependencies": {
+    "@mariozechner/pi-ai": "*",
+    "@mariozechner/pi-coding-agent": "*"
+  },
+  "devDependencies": {
+    "@biomejs/biome": "2.3.5",
+    "@mariozechner/pi-ai": "*",
+    "@mariozechner/pi-coding-agent": "*",
+    "bun-types": "latest",
+    "typescript": "^5.9.3"
+  },
+  "license": "MIT"
+}

package/src/auto-name-utils.ts

@@ -0,0 +1,90 @@
+import * as os from 'node:os'
+import * as path from 'node:path'
+
+export const SUBAGENT_SESSION_DIR = path.join(
+  os.homedir(),
+  '.pi',
+  'agent',
+  'sessions',
+  'subagents',
+)
+
+export const NAME_SYSTEM_PROMPT = [
+  "Read the user's first request and generate a very short session title.",
+  'Requirements:',
+  '- Output English only, even if the request is written in another language.',
+  '- Output only the title text.',
+  '- No quotes, markdown, prefixes, or explanations.',
+  '- Keep it concise: usually 2 to 6 words.',
+  '- Maximum 30 characters.',
+].join('\n')
+
+export const MAX_MESSAGE_LENGTH = 500
+export const MAX_NAME_LENGTH = 30
+export const MAX_STATUS_CHARS = 90
+export const SUCCESSFUL_STOP_REASON = 'stop'
+
+export function isSubagentSessionPath(
+  sessionFilePath: string | undefined,
+): boolean {
+  if (!sessionFilePath) return false
+  return (
+    sessionFilePath.startsWith(SUBAGENT_SESSION_DIR + path.sep) ||
+    sessionFilePath.startsWith(`${SUBAGENT_SESSION_DIR}/`)
+  )
+}
+
+export function extractSessionFilePath(
+  sessionManager: unknown,
+): string | undefined {
+  try {
+    if (
+      sessionManager &&
+      typeof sessionManager === 'object' &&
+      'getSessionFile' in sessionManager
+    ) {
+      const getSessionFile = (sessionManager as Record<string, unknown>)
+        .getSessionFile
+      if (typeof getSessionFile === 'function') {
+        const raw = String(getSessionFile() ?? '')
+        const cleaned = raw.replace(/[\r\n\t]+/g, '').trim()
+        return cleaned || undefined
+      }
+    }
+  } catch {
+    // Ignore errors and fall back to unnamed sessions.
+  }
+
+  return undefined
+}
+
+export function formatNameStatus(name: string): string {
+  const singleLine = name.replace(/\s+/g, ' ').trim()
+  return singleLine.length > MAX_STATUS_CHARS
+    ? `${singleLine.slice(0, MAX_STATUS_CHARS - 1)}…`
+    : singleLine
+}
+
+export function buildNameContext(userMessage: string): string {
+  return `User message: ${userMessage.slice(0, MAX_MESSAGE_LENGTH)}`
+}
+
+export function isSuccessfulResult(stopReason: string | undefined): boolean {
+  return stopReason === SUCCESSFUL_STOP_REASON
+}
+
+export function extractNameFromResult(
+  content: ReadonlyArray<{ type: string; text?: string }>,
+): string {
+  const text = content
+    .filter(
+      (part): part is { type: 'text'; text: string } =>
+        part.type === 'text' && typeof part.text === 'string',
+    )
+    .map((part) => part.text)
+    .join('')
+    .trim()
+    .replace(/^['"`]+|['"`]+$/g, '')
+
+  return text.slice(0, MAX_NAME_LENGTH)
+}

package/src/index.ts

@@ -0,0 +1,90 @@
+import * as path from 'node:path'
+import type {
+  ExtensionAPI,
+  ExtensionContext,
+} from '@mariozechner/pi-coding-agent'
+import {
+  buildNameContext,
+  extractNameFromResult,
+  extractSessionFilePath,
+  formatNameStatus,
+  isSubagentSessionPath,
+  NAME_SYSTEM_PROMPT,
+} from './auto-name-utils.ts'
+import { generateShortLabel } from './short-label.ts'
+
+const NAME_STATUS_KEY = 'name-footer'
+
+function isSubagentSession(ctx: ExtensionContext): boolean {
+  const sessionFilePath = extractSessionFilePath(ctx.sessionManager)
+  return isSubagentSessionPath(sessionFilePath)
+}
+
+async function detectNameFromMessage(
+  userMessage: string,
+  ctx: ExtensionContext,
+): Promise<string> {
+  return generateShortLabel(ctx, {
+    systemPrompt: NAME_SYSTEM_PROMPT,
+    prompt: buildNameContext(userMessage),
+    extractText: extractNameFromResult,
+  })
+}
+
+export default function autoSessionName(pi: ExtensionAPI) {
+  const updateTerminalTitle = (ctx: ExtensionContext) => {
+    if (!ctx.hasUI) return
+
+    const cwdBasename = path.basename(process.cwd())
+    const name = pi.getSessionName()
+    if (!name) return
+
+    ctx.ui.setTitle(`π - ${name} - ${cwdBasename}`)
+  }
+
+  const updateStatus = (ctx: ExtensionContext) => {
+    if (!ctx.hasUI) return
+
+    const name = pi.getSessionName()
+    if (!name) {
+      ctx.ui.setStatus(NAME_STATUS_KEY, undefined)
+      return
+    }
+
+    ctx.ui.setStatus(NAME_STATUS_KEY, formatNameStatus(name))
+    updateTerminalTitle(ctx)
+  }
+
+  pi.on('before_agent_start', async (event, ctx) => {
+    if (isSubagentSession(ctx)) return
+    if (pi.getSessionName()) return
+
+    const text = event.prompt.trim()
+    if (!text) return
+
+    void (async () => {
+      try {
+        const detected = await detectNameFromMessage(text, ctx)
+        if (detected && !pi.getSessionName()) {
+          pi.setSessionName(detected)
+          updateStatus(ctx)
+        }
+      } catch {
+        // Leave the session unnamed on failures.
+      }
+    })()
+  })
+
+  pi.on('session_start', async (_event, ctx) => {
+    updateStatus(ctx)
+  })
+
+  pi.on('session_tree', async (_event, ctx) => {
+    updateStatus(ctx)
+  })
+
+  pi.on('session_shutdown', async (_event, ctx) => {
+    if (!ctx.hasUI) return
+    ctx.ui.setStatus(NAME_STATUS_KEY, undefined)
+  })
+}

package/src/short-label.ts

@@ -0,0 +1,82 @@
+import { completeSimple } from '@mariozechner/pi-ai'
+
+type SummaryModel = Parameters<typeof completeSimple>[0]
+type SummaryResult = Awaited<ReturnType<typeof completeSimple>>
+
+type AuthResult = {
+  ok: boolean
+  apiKey?: string
+  headers?: Record<string, string>
+}
+
+export type ShortLabelContext = {
+  model?: SummaryModel
+  modelRegistry?: {
+    getApiKeyAndHeaders: (model: SummaryModel) => Promise<AuthResult>
+  }
+}
+
+export type GenerateShortLabelOptions = {
+  systemPrompt: string
+  prompt: string
+  maxTokens?: number
+  timeoutMs?: number
+  extractText?: (content: SummaryResult['content']) => string
+}
+
+function defaultExtractText(content: SummaryResult['content']): string {
+  return content
+    .filter(
+      (part): part is { type: 'text'; text: string } =>
+        part.type === 'text' && typeof part.text === 'string',
+    )
+    .map((part) => part.text)
+    .join('')
+    .trim()
+}
+
+export async function generateShortLabel(
+  ctx: ShortLabelContext,
+  options: GenerateShortLabelOptions,
+): Promise<string> {
+  const model = ctx.model
+  if (!model || !ctx.modelRegistry) return ''
+
+  const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model)
+  if (!auth.ok) return ''
+
+  const controller = new AbortController()
+  const timeoutMs = options.timeoutMs ?? 10000
+  const timer = setTimeout(() => controller.abort(), timeoutMs)
+
+  try {
+    const result = await completeSimple(
+      model,
+      {
+        systemPrompt: options.systemPrompt,
+        messages: [
+          {
+            role: 'user',
+            content: [{ type: 'text', text: options.prompt }],
+            timestamp: Date.now(),
+          },
+        ],
+      },
+      {
+        apiKey: auth.apiKey,
+        headers: auth.headers,
+        signal: controller.signal,
+        reasoning: 'minimal',
+        maxTokens: options.maxTokens ?? 60,
+      },
+    )
+
+    if (result.stopReason !== 'stop') return ''
+    const extractText = options.extractText ?? defaultExtractText
+    return extractText(result.content)
+  } catch {
+    return ''
+  } finally {
+    clearTimeout(timer)
+  }
+}