@oh-my-pi/pi-coding-agent 3.20.0 → 3.21.0

package/docs/hooks.md

@@ -72,7 +72,7 @@ Additional paths via `settings.json`:
 | --------------------------------- | --------------------------------------------- |
 | `@oh-my-pi/pi-coding-agent/hooks` | Hook types (`HookAPI`, `HookContext`, events) |
 | `@oh-my-pi/pi-coding-agent`       | Additional types if needed                    |
-| `@oh-my-pi/pi-ai`                 | AI utilities                                  |
+| `@mariozechner/pi-ai`             | AI utilities                                  |
 | `@oh-my-pi/pi-tui`                | TUI components                                |
 
 Node.js built-ins (`node:fs`, `node:path`, etc.) are also available.
@@ -548,7 +548,7 @@ Current model, or `undefined` if none selected yet. Use for LLM calls in hooks:
 ```typescript
 if (ctx.model) {
 	const apiKey = await ctx.modelRegistry.getApiKey(ctx.model);
-	// Use with @oh-my-pi/pi-ai complete()
+	// Use with @mariozechner/pi-ai complete()
 }
 ```
 

package/docs/sdk.md

@@ -248,7 +248,7 @@ const { session } = await createAgentSession({
 ### Model
 
 ```typescript
-import { getModel } from "@oh-my-pi/pi-ai";
+import { getModel } from "@mariozechner/pi-ai";
 import { discoverAuthStorage, discoverModels } from "@oh-my-pi/pi-coding-agent";
 
 const authStorage = discoverAuthStorage();
@@ -733,7 +733,7 @@ Project overrides global. Nested objects merge keys. Setters only modify global
 All discovery functions accept optional `cwd` and `agentDir` parameters.
 
 ```typescript
-import { getModel } from "@oh-my-pi/pi-ai";
+import { getModel } from "@mariozechner/pi-ai";
 import {
 	AuthStorage,
 	ModelRegistry,
@@ -806,7 +806,7 @@ interface CreateAgentSessionResult {
 ## Complete Example
 
 ```typescript
-import { getModel } from "@oh-my-pi/pi-ai";
+import { getModel } from "@mariozechner/pi-ai";
 import { Type } from "@sinclair/typebox";
 import {
 	AuthStorage,

package/examples/custom-tools/README.md

@@ -63,7 +63,7 @@ See [docs/custom-tools.md](../../docs/custom-tools.md) for full documentation.
 
 ```typescript
 import { Type } from "@sinclair/typebox";
-import { StringEnum } from "@oh-my-pi/pi-ai";
+import { StringEnum } from "@mariozechner/pi-ai";
 import { Text } from "@oh-my-pi/pi-tui";
 import type { CustomToolFactory } from "@oh-my-pi/pi-coding-agent";
 
@@ -114,7 +114,7 @@ renderResult(result, { expanded, isPartial }, theme) {
 **Use StringEnum for string parameters** (required for Google API compatibility):
 
 ```typescript
-import { StringEnum } from "@oh-my-pi/pi-ai";
+import { StringEnum } from "@mariozechner/pi-ai";
 
 // Good
 action: StringEnum(["list", "add"] as const);

package/examples/custom-tools/subagent/index.ts

@@ -15,8 +15,8 @@
 import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
+import type { Message } from "@mariozechner/pi-ai";
 import type { AgentToolResult } from "@oh-my-pi/pi-agent-core";
-import type { Message } from "@oh-my-pi/pi-ai";
 import type { CustomTool, CustomToolAPI, CustomToolFactory } from "@oh-my-pi/pi-coding-agent";
 import { type AgentConfig, type AgentScope, discoverAgents, formatAgentList } from "./agents";
 

package/examples/extensions/README.md

@@ -16,54 +16,54 @@ cp permission-gate.ts ~/.omp/agent/extensions/
 
 ### Lifecycle & Safety
 
-| Extension | Description |
-|-----------|-------------|
-| `permission-gate.ts` | Prompts for confirmation before dangerous bash commands (rm -rf, sudo, etc.) |
-| `protected-paths.ts` | Blocks writes to protected paths (.env, .git/, node_modules/) |
-| `confirm-destructive.ts` | Confirms before destructive session actions (clear, switch, branch) |
-| `dirty-repo-guard.ts` | Prevents session changes with uncommitted git changes |
+| Extension                | Description                                                                  |
+| ------------------------ | ---------------------------------------------------------------------------- |
+| `permission-gate.ts`     | Prompts for confirmation before dangerous bash commands (rm -rf, sudo, etc.) |
+| `protected-paths.ts`     | Blocks writes to protected paths (.env, .git/, node_modules/)                |
+| `confirm-destructive.ts` | Confirms before destructive session actions (clear, switch, branch)          |
+| `dirty-repo-guard.ts`    | Prevents session changes with uncommitted git changes                        |
 
 ### Custom Tools
 
-| Extension | Description |
-|-----------|-------------|
-| `todo.ts` | Todo list tool + `/todos` command with custom rendering and state persistence |
-| `hello.ts` | Minimal custom tool example |
-| `question.ts` | Demonstrates `ctx.ui.select()` for asking the user questions |
-| `subagent/` | Delegate tasks to specialized subagents with isolated context windows |
+| Extension     | Description                                                                   |
+| ------------- | ----------------------------------------------------------------------------- |
+| `todo.ts`     | Todo list tool + `/todos` command with custom rendering and state persistence |
+| `hello.ts`    | Minimal custom tool example                                                   |
+| `question.ts` | Demonstrates `ctx.ui.select()` for asking the user questions                  |
+| `subagent/`   | Delegate tasks to specialized subagents with isolated context windows         |
 
 ### Commands & UI
 
-| Extension | Description |
-|-----------|-------------|
-| `plan-mode.ts` | Claude Code-style plan mode for read-only exploration with `/plan` command |
-| `tools.ts` | Interactive `/tools` command to enable/disable tools with session persistence |
-| `handoff.ts` | Transfer context to a new focused session via `/handoff <goal>` |
-| `qna.ts` | Extracts questions from last response into editor via `ctx.ui.setEditorText()` |
-| `status-line.ts` | Shows turn progress in footer via `ctx.ui.setStatus()` with themed colors |
-| `snake.ts` | Snake game with custom UI, keyboard handling, and session persistence |
+| Extension        | Description                                                                    |
+| ---------------- | ------------------------------------------------------------------------------ |
+| `plan-mode.ts`   | Claude Code-style plan mode for read-only exploration with `/plan` command     |
+| `tools.ts`       | Interactive `/tools` command to enable/disable tools with session persistence  |
+| `handoff.ts`     | Transfer context to a new focused session via `/handoff <goal>`                |
+| `qna.ts`         | Extracts questions from last response into editor via `ctx.ui.setEditorText()` |
+| `status-line.ts` | Shows turn progress in footer via `ctx.ui.setStatus()` with themed colors      |
+| `snake.ts`       | Snake game with custom UI, keyboard handling, and session persistence          |
 
 ### Git Integration
 
-| Extension | Description |
-|-----------|-------------|
-| `git-checkpoint.ts` | Creates git stash checkpoints at each turn for code restoration on branch |
-| `auto-commit-on-exit.ts` | Auto-commits on exit using last assistant message for commit message |
+| Extension                | Description                                                               |
+| ------------------------ | ------------------------------------------------------------------------- |
+| `git-checkpoint.ts`      | Creates git stash checkpoints at each turn for code restoration on branch |
+| `auto-commit-on-exit.ts` | Auto-commits on exit using last assistant message for commit message      |
 
 ### System Prompt & Compaction
 
-| Extension | Description |
-|-----------|-------------|
-| `pirate.ts` | Demonstrates `systemPromptAppend` to dynamically modify system prompt |
-| `custom-compaction.ts` | Custom compaction that summarizes entire conversation |
+| Extension              | Description                                                           |
+| ---------------------- | --------------------------------------------------------------------- |
+| `pirate.ts`            | Demonstrates `systemPromptAppend` to dynamically modify system prompt |
+| `custom-compaction.ts` | Custom compaction that summarizes entire conversation                 |
 
 ### External Dependencies
 
-| Extension | Description |
-|-----------|-------------|
+| Extension         | Description                                                               |
+| ----------------- | ------------------------------------------------------------------------- |
 | `chalk-logger.ts` | Uses chalk from parent node_modules (demonstrates jiti module resolution) |
-| `with-deps/` | Extension with its own package.json and dependencies |
-| `file-trigger.ts` | Watches a trigger file and injects contents into conversation |
+| `with-deps/`      | Extension with its own package.json and dependencies                      |
+| `file-trigger.ts` | Watches a trigger file and injects contents into conversation             |
 
 ## Writing Extensions
 
@@ -74,68 +74,70 @@ import type { ExtensionAPI } from "@oh-my-pi/pi-coding-agent";
 import { Type } from "@sinclair/typebox";
 
 export default function (pi: ExtensionAPI) {
-  // Subscribe to lifecycle events
-  pi.on("tool_call", async (event, ctx) => {
-    if (event.toolName === "bash" && event.input.command?.includes("rm -rf")) {
-      const ok = await ctx.ui.confirm("Dangerous!", "Allow rm -rf?");
-      if (!ok) return { block: true, reason: "Blocked by user" };
-    }
-  });
-
-  // Register custom tools
-  pi.registerTool({
-    name: "greet",
-    label: "Greeting",
-    description: "Generate a greeting",
-    parameters: Type.Object({
-      name: Type.String({ description: "Name to greet" }),
-    }),
-    async execute(toolCallId, params, onUpdate, ctx, signal) {
-      return {
-        content: [{ type: "text", text: `Hello, ${params.name}!` }],
-        details: {},
-      };
-    },
-  });
-
-  // Register commands
-  pi.registerCommand("hello", {
-    description: "Say hello",
-    handler: async (args, ctx) => {
-      ctx.ui.notify("Hello!", "info");
-    },
-  });
+	// Subscribe to lifecycle events
+	pi.on("tool_call", async (event, ctx) => {
+		if (event.toolName === "bash" && event.input.command?.includes("rm -rf")) {
+			const ok = await ctx.ui.confirm("Dangerous!", "Allow rm -rf?");
+			if (!ok) return { block: true, reason: "Blocked by user" };
+		}
+	});
+
+	// Register custom tools
+	pi.registerTool({
+		name: "greet",
+		label: "Greeting",
+		description: "Generate a greeting",
+		parameters: Type.Object({
+			name: Type.String({ description: "Name to greet" }),
+		}),
+		async execute(toolCallId, params, onUpdate, ctx, signal) {
+			return {
+				content: [{ type: "text", text: `Hello, ${params.name}!` }],
+				details: {},
+			};
+		},
+	});
+
+	// Register commands
+	pi.registerCommand("hello", {
+		description: "Say hello",
+		handler: async (args, ctx) => {
+			ctx.ui.notify("Hello!", "info");
+		},
+	});
 }
 ```
 
 ## Key Patterns
 
 **Use StringEnum for string parameters** (required for Google API compatibility):
+
 ```typescript
-import { StringEnum } from "@oh-my-pi/pi-ai";
+import { StringEnum } from "@mariozechner/pi-ai";
 
 // Good
-action: StringEnum(["list", "add"] as const)
+action: StringEnum(["list", "add"] as const);
 
 // Bad - doesn't work with Google
-action: Type.Union([Type.Literal("list"), Type.Literal("add")])
+action: Type.Union([Type.Literal("list"), Type.Literal("add")]);
 ```
 
 **State persistence via details:**
+
 ```typescript
 // Store state in tool result details for proper branching support
 return {
-  content: [{ type: "text", text: "Done" }],
-  details: { todos: [...todos], nextId },  // Persisted in session
+	content: [{ type: "text", text: "Done" }],
+	details: { todos: [...todos], nextId }, // Persisted in session
 };
 
 // Reconstruct on session events
 pi.on("session_start", async (_event, ctx) => {
-  for (const entry of ctx.sessionManager.getBranch()) {
-    if (entry.type === "message" && entry.message.toolName === "my_tool") {
-      const details = entry.message.details;
-      // Reconstruct state from details
-    }
-  }
+	for (const entry of ctx.sessionManager.getBranch()) {
+		if (entry.type === "message" && entry.message.toolName === "my_tool") {
+			const details = entry.message.details;
+			// Reconstruct state from details
+		}
+	}
 });
 ```

package/examples/extensions/todo.ts

@@ -10,7 +10,7 @@
  * correct for that point in history.
  */
 
-import { StringEnum } from "@oh-my-pi/pi-ai";
+import { StringEnum } from "@mariozechner/pi-ai";
 import type { ExtensionAPI, ExtensionContext, Theme } from "@oh-my-pi/pi-coding-agent";
 import { matchesKey, Text, truncateToWidth } from "@oh-my-pi/pi-tui";
 import { Type } from "@sinclair/typebox";
@@ -261,10 +261,7 @@ export default function (pi: ExtensionAPI) {
 				case "add": {
 					const added = todoList[todoList.length - 1];
 					return new Text(
-						theme.fg("success", "✓ Added ") +
-							theme.fg("accent", `#${added.id}`) +
-							" " +
-							theme.fg("muted", added.text),
+						`${theme.fg("success", "✓ Added ") + theme.fg("accent", `#${added.id}`)} ${theme.fg("muted", added.text)}`,
 						0,
 						0,
 					);

package/examples/hooks/custom-compaction.ts

@@ -13,7 +13,7 @@
  *   omp --hook examples/hooks/custom-compaction.ts
  */
 
-import { complete, getModel } from "@oh-my-pi/pi-ai";
+import { complete, getModel } from "@mariozechner/pi-ai";
 import type { HookAPI } from "@oh-my-pi/pi-coding-agent";
 import { convertToLlm, serializeConversation } from "@oh-my-pi/pi-coding-agent";
 

package/examples/hooks/handoff.ts

@@ -12,7 +12,7 @@
  * The generated prompt appears as a draft in the editor for review/editing.
  */
 
-import { complete, type Message } from "@oh-my-pi/pi-ai";
+import { complete, type Message } from "@mariozechner/pi-ai";
 import type { HookAPI, SessionEntry } from "@oh-my-pi/pi-coding-agent";
 import { BorderedLoader, convertToLlm, serializeConversation } from "@oh-my-pi/pi-coding-agent";
 

package/examples/hooks/qna.ts

@@ -7,7 +7,7 @@
  * 3. Loads the result into the editor for user to fill in answers
  */
 
-import { complete, type UserMessage } from "@oh-my-pi/pi-ai";
+import { complete, type UserMessage } from "@mariozechner/pi-ai";
 import type { HookAPI } from "@oh-my-pi/pi-coding-agent";
 import { BorderedLoader } from "@oh-my-pi/pi-coding-agent";
 

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

@@ -4,7 +4,7 @@
  * Shows how to select a specific model and thinking level.
  */
 
-import { getModel } from "@oh-my-pi/pi-ai";
+import { getModel } from "@mariozechner/pi-ai";
 import { createAgentSession, discoverAuthStorage, discoverModels } from "@oh-my-pi/pi-coding-agent";
 
 // Set up auth storage and model registry

package/examples/sdk/12-full-control.ts

@@ -8,7 +8,7 @@
  * paths relative to your cwd.
  */
 
-import { getModel } from "@oh-my-pi/pi-ai";
+import { getModel } from "@mariozechner/pi-ai";
 import {
 	AuthStorage,
 	type CustomTool,

package/examples/sdk/README.md

@@ -29,7 +29,7 @@ npx tsx examples/sdk/01-minimal.ts
 ## Quick Reference
 
 ```typescript
-import { getModel } from "@oh-my-pi/pi-ai";
+import { getModel } from "@mariozechner/pi-ai";
 import {
 	AuthStorage,
 	createAgentSession,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-coding-agent",
-	"version": "3.20.0",
+	"version": "3.21.0",
 	"description": "Coding agent CLI with read, bash, edit, write tools and session management",
 	"type": "module",
 	"ompConfig": {
@@ -39,10 +39,10 @@
 		"prepublishOnly": "bun run generate-template && bun run clean && bun run build"
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-agent-core": "3.20.0",
-		"@oh-my-pi/pi-ai": "3.20.0",
-		"@oh-my-pi/pi-git-tool": "3.20.0",
-		"@oh-my-pi/pi-tui": "3.20.0",
+		"@mariozechner/pi-ai": "^0.37.4",
+		"@oh-my-pi/pi-agent-core": "3.21.0",
+		"@oh-my-pi/pi-git-tool": "3.21.0",
+		"@oh-my-pi/pi-tui": "3.21.0",
 		"@openai/agents": "^0.3.7",
 		"@sinclair/typebox": "^0.34.46",
 		"ajv": "^8.17.1",

package/src/cli/file-processor.ts

@@ -4,7 +4,7 @@
 
 import { existsSync, readFileSync, statSync } from "node:fs";
 import { resolve } from "node:path";
-import type { ImageContent } from "@oh-my-pi/pi-ai";
+import type { ImageContent } from "@mariozechner/pi-ai";
 import chalk from "chalk";
 import { resolveReadPath } from "../core/tools/path-utils";
 import { formatDimensionNote, resizeImage } from "../utils/image-resize";

package/src/cli/list-models.ts

@@ -2,7 +2,7 @@
  * List available models with optional fuzzy search
  */
 
-import type { Api, Model } from "@oh-my-pi/pi-ai";
+import type { Api, Model } from "@mariozechner/pi-ai";
 import type { ModelRegistry } from "../core/model-registry";
 import { fuzzyFilter } from "../utils/fuzzy";
 

package/src/core/agent-session.ts

@@ -13,9 +13,9 @@
  * Modes use this class and add their own I/O layer on top.
  */
 
+import type { AssistantMessage, ImageContent, Message, Model, TextContent, Usage } from "@mariozechner/pi-ai";
+import { isContextOverflow, modelsAreEqual, supportsXhigh } from "@mariozechner/pi-ai";
 import type { Agent, AgentEvent, AgentMessage, AgentState, AgentTool, ThinkingLevel } from "@oh-my-pi/pi-agent-core";
-import type { AssistantMessage, ImageContent, Message, Model, TextContent, Usage } from "@oh-my-pi/pi-ai";
-import { isContextOverflow, modelsAreEqual, supportsXhigh } from "@oh-my-pi/pi-ai";
 import type { Rule } from "../capability/rule";
 import { getAuthPath } from "../config";
 import { theme } from "../modes/interactive/theme/theme";
@@ -51,6 +51,7 @@ import { parseModelString } from "./model-resolver";
 import { expandPromptTemplate, type PromptTemplate, parseCommandArgs } from "./prompt-templates";
 import type { BranchSummaryEntry, CompactionEntry, NewSessionOptions, SessionManager } from "./session-manager";
 import type { SettingsManager, SkillsSettings } from "./settings-manager";
+import { expandSlashCommand, type FileSlashCommand } from "./slash-commands";
 import type { TtsrManager } from "./ttsr";
 
 /** Session-specific events that extend the core AgentEvent */
@@ -77,6 +78,8 @@ export interface AgentSessionConfig {
 	scopedModels?: Array<{ model: Model<any>; thinkingLevel: ThinkingLevel }>;
 	/** Prompt templates for expansion */
 	promptTemplates?: PromptTemplate[];
+	/** File-based slash commands for expansion */
+	slashCommands?: FileSlashCommand[];
 	/** Extension runner (created in main.ts with wrapped tools) */
 	extensionRunner?: ExtensionRunner;
 	/** Custom commands (TypeScript slash commands) */
@@ -175,6 +178,7 @@ export class AgentSession {
 
 	private _scopedModels: Array<{ model: Model<any>; thinkingLevel: ThinkingLevel }>;
 	private _promptTemplates: PromptTemplate[];
+	private _slashCommands: FileSlashCommand[];
 
 	// Event subscription state
 	private _unsubscribeAgent?: () => void;
@@ -232,6 +236,7 @@ export class AgentSession {
 		this.settingsManager = config.settingsManager;
 		this._scopedModels = config.scopedModels ?? [];
 		this._promptTemplates = config.promptTemplates ?? [];
+		this._slashCommands = config.slashCommands ?? [];
 		this._extensionRunner = config.extensionRunner;
 		this._customCommands = config.customCommands ?? [];
 		this._skillsSettings = config.skillsSettings;
@@ -684,6 +689,12 @@ export class AgentSession {
 				}
 				text = customResult;
 			}
+
+			// Try file-based slash commands (markdown files from commands/ directories)
+			// Only if text still starts with "/" (wasn't transformed by custom command)
+			if (text.startsWith("/")) {
+				text = expandSlashCommand(text, this._slashCommands);
+			}
 		}
 
 		// Expand file-based prompt templates if requested

package/src/core/auth-storage.ts

@@ -15,7 +15,7 @@ import {
 	loginOpenAICodex,
 	type OAuthCredentials,
 	type OAuthProvider,
-} from "@oh-my-pi/pi-ai";
+} from "@mariozechner/pi-ai";
 import { logger } from "./logger";
 
 export type ApiKeyCredential = {

package/src/core/compaction/branch-summarization.ts

@@ -5,9 +5,9 @@
  * a summary of the branch being left so context isn't lost.
  */
 
+import type { Model } from "@mariozechner/pi-ai";
+import { completeSimple } from "@mariozechner/pi-ai";
 import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { Model } from "@oh-my-pi/pi-ai";
-import { completeSimple } from "@oh-my-pi/pi-ai";
 import branchSummaryPrompt from "../../prompts/branch-summary.md" with { type: "text" };
 import branchSummaryPreamble from "../../prompts/branch-summary-preamble.md" with { type: "text" };
 import {

package/src/core/compaction/compaction.ts

@@ -5,9 +5,9 @@
  * and after compaction the session is reloaded.
  */
 
+import type { AssistantMessage, Model, Usage } from "@mariozechner/pi-ai";
+import { complete, completeSimple } from "@mariozechner/pi-ai";
 import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { AssistantMessage, Model, Usage } from "@oh-my-pi/pi-ai";
-import { complete, completeSimple } from "@oh-my-pi/pi-ai";
 import compactionSummaryPrompt from "../../prompts/compaction-summary.md" with { type: "text" };
 import compactionTurnPrefixPrompt from "../../prompts/compaction-turn-prefix.md" with { type: "text" };
 import compactionUpdateSummaryPrompt from "../../prompts/compaction-update-summary.md" with { type: "text" };

package/src/core/compaction/utils.ts

@@ -2,8 +2,8 @@
  * Shared utilities for compaction and branch summarization.
  */
 
+import type { Message } from "@mariozechner/pi-ai";
 import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { Message } from "@oh-my-pi/pi-ai";
 import summarizationSystemPrompt from "../../prompts/summarization-system.md" with { type: "text" };
 
 // ============================================================================

package/src/core/custom-tools/types.ts

@@ -5,8 +5,8 @@
  * They can provide custom rendering for tool calls and results in the TUI.
  */
 
+import type { Model } from "@mariozechner/pi-ai";
 import type { AgentToolResult, AgentToolUpdateCallback } from "@oh-my-pi/pi-agent-core";
-import type { Model } from "@oh-my-pi/pi-ai";
 import type { Component } from "@oh-my-pi/pi-tui";
 import type { Static, TSchema } from "@sinclair/typebox";
 import type { Theme } from "../../modes/interactive/theme/theme";

package/src/core/extensions/runner.ts

@@ -2,8 +2,8 @@
  * Extension runner - executes extensions and manages their lifecycle.
  */
 
+import type { ImageContent, Model } from "@mariozechner/pi-ai";
 import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { ImageContent, Model } from "@oh-my-pi/pi-ai";
 import type { KeyId } from "@oh-my-pi/pi-tui";
 import { theme } from "../../modes/interactive/theme/theme";
 import type { ModelRegistry } from "../model-registry";

package/src/core/extensions/types.ts

@@ -8,8 +8,8 @@
  * - Interact with the user via UI primitives
  */
 
+import type { ImageContent, Model, TextContent, ToolResultMessage } from "@mariozechner/pi-ai";
 import type { AgentMessage, AgentToolResult, AgentToolUpdateCallback } from "@oh-my-pi/pi-agent-core";
-import type { ImageContent, Model, TextContent, ToolResultMessage } from "@oh-my-pi/pi-ai";
 import type { Component, KeyId, TUI } from "@oh-my-pi/pi-tui";
 import type { Static, TSchema } from "@sinclair/typebox";
 import type { Theme } from "../../modes/interactive/theme/theme";

package/src/core/extensions/wrapper.ts

@@ -2,8 +2,8 @@
  * Tool wrappers for extensions.
  */
 
+import type { ImageContent, TextContent } from "@mariozechner/pi-ai";
 import type { AgentTool, AgentToolContext, AgentToolUpdateCallback } from "@oh-my-pi/pi-agent-core";
-import type { ImageContent, TextContent } from "@oh-my-pi/pi-ai";
 import type { Theme } from "../../modes/interactive/theme/theme";
 import type { ExtensionRunner } from "./runner";
 import type { ExtensionContext, RegisteredTool, ToolCallEventResult, ToolResultEventResult } from "./types";

package/src/core/hooks/runner.ts

@@ -2,8 +2,8 @@
  * Hook runner - executes hooks and manages their lifecycle.
  */
 
+import type { Model } from "@mariozechner/pi-ai";
 import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { Model } from "@oh-my-pi/pi-ai";
 import { theme } from "../../modes/interactive/theme/theme";
 import type { ModelRegistry } from "../model-registry";
 import type { SessionManager } from "../session-manager";
@@ -400,7 +400,7 @@ export class HookRunner {
 	 */
 	async emitBeforeAgentStart(
 		prompt: string,
-		images?: import("@oh-my-pi/pi-ai").ImageContent[],
+		images?: import("@mariozechner/pi-ai").ImageContent[],
 	): Promise<BeforeAgentStartEventResult | undefined> {
 		const ctx = this.createContext();
 		let result: BeforeAgentStartEventResult | undefined;

package/src/core/hooks/types.ts

@@ -5,8 +5,8 @@
  * and interact with the user via UI primitives.
  */
 
+import type { ImageContent, Message, Model, TextContent, ToolResultMessage } from "@mariozechner/pi-ai";
 import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { ImageContent, Message, Model, TextContent, ToolResultMessage } from "@oh-my-pi/pi-ai";
 import type { Component, TUI } from "@oh-my-pi/pi-tui";
 import type { Theme } from "../../modes/interactive/theme/theme";
 import type { CompactionPreparation, CompactionResult } from "../compaction/index";

package/src/core/messages.ts

@@ -5,8 +5,8 @@
  * and provides a transformer to convert them to LLM-compatible messages.
  */
 
+import type { ImageContent, Message, TextContent } from "@mariozechner/pi-ai";
 import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
-import type { ImageContent, Message, TextContent } from "@oh-my-pi/pi-ai";
 
 export const COMPACTION_SUMMARY_PREFIX = `The conversation history before this point was compacted into the following summary:
 

package/src/core/model-registry.ts

@@ -11,7 +11,7 @@ import {
 	type KnownProvider,
 	type Model,
 	normalizeDomain,
-} from "@oh-my-pi/pi-ai";
+} from "@mariozechner/pi-ai";
 import { type Static, Type } from "@sinclair/typebox";
 import AjvModule from "ajv";
 import type { AuthStorage } from "./auth-storage";

package/src/core/model-resolver.ts

@@ -2,8 +2,8 @@
  * Model resolution, scoping, and initial selection
  */
 
+import { type Api, type KnownProvider, type Model, modelsAreEqual } from "@mariozechner/pi-ai";
 import type { ThinkingLevel } from "@oh-my-pi/pi-agent-core";
-import { type Api, type KnownProvider, type Model, modelsAreEqual } from "@oh-my-pi/pi-ai";
 import chalk from "chalk";
 import { minimatch } from "minimatch";
 import { isValidThinkingLevel } from "../cli/args";