@ilya-lesikov/pi-pi 0.1.0

package/3p/pi-plannotator/packages/ai/index.ts

@@ -0,0 +1,105 @@
+/**
+ * @plannotator/ai — AI provider layer for Plannotator.
+ *
+ * This package provides the backbone for AI-powered features (inline chat,
+ * plan Q&A, code review assistance) across all Plannotator surfaces.
+ *
+ * Architecture:
+ *
+ *   ┌─────────────────┐     ┌──────────────┐
+ *   │  Plan Review UI  │────▶│              │
+ *   ├─────────────────┤     │  AI Endpoints │──▶ SSE stream
+ *   │  Code Review UI  │────▶│  (HTTP)      │
+ *   ├─────────────────┤     │              │
+ *   │  Annotate UI     │────▶└──────┬───────┘
+ *   └─────────────────┘            │
+ *                                  ▼
+ *                         ┌────────────────┐
+ *                         │ Session Manager │
+ *                         └────────┬───────┘
+ *                                  │
+ *                         ┌────────▼───────┐
+ *                         │  AIProvider     │ (abstract)
+ *                         └────────┬───────┘
+ *                                  │
+ *                    ┌─────────────┼──────────────┐
+ *                    ▼             ▼               ▼
+ *           ┌──────────────┐ ┌──────────┐  ┌───────────┐
+ *           │ Claude Agent │ │ OpenCode │  │  Future   │
+ *           │ SDK Provider │ │ Provider │  │ Providers │
+ *           └──────────────┘ └──────────┘  └───────────┘
+ *
+ * Quick start:
+ *
+ * ```ts
+ * import "@plannotator/ai/providers/claude-agent-sdk";
+ * import { ProviderRegistry, createProvider, createAIEndpoints, SessionManager } from "@plannotator/ai";
+ *
+ * // 1. Create a registry and provider
+ * const registry = new ProviderRegistry();
+ * const provider = await createProvider({ type: "claude-agent-sdk", cwd: process.cwd() });
+ * registry.register(provider);
+ *
+ * // 2. Create endpoints and session manager
+ * const sessionManager = new SessionManager();
+ * const aiEndpoints = createAIEndpoints({ registry, sessionManager });
+ *
+ * // 3. Mount endpoints in your Bun server
+ * // aiEndpoints["/api/ai/query"](request) → SSE Response
+ * ```
+ */
+
+// Types
+export type {
+  AIProvider,
+  AIProviderCapabilities,
+  AIProviderConfig,
+  AISession,
+  AIMessage,
+  AITextMessage,
+  AITextDeltaMessage,
+  AIToolUseMessage,
+  AIToolResultMessage,
+  AIErrorMessage,
+  AIResultMessage,
+  AIPermissionRequestMessage,
+  AIUnknownMessage,
+  AIContext,
+  AIContextMode,
+  PlanContext,
+  CodeReviewContext,
+  AnnotateContext,
+  ParentSession,
+  CreateSessionOptions,
+  ClaudeAgentSDKConfig,
+  CodexSDKConfig,
+  PiSDKConfig,
+  OpenCodeConfig,
+} from "./types.ts";
+
+// Provider registry
+export {
+  ProviderRegistry,
+  registerProviderFactory,
+  createProvider,
+} from "./provider.ts";
+
+// Context builders
+export { buildSystemPrompt, buildForkPreamble, buildEffectivePrompt } from "./context.ts";
+
+// Base session
+export { BaseSession } from "./base-session.ts";
+
+// Session manager
+export { SessionManager } from "./session-manager.ts";
+export type { SessionEntry, SessionManagerOptions } from "./session-manager.ts";
+
+// HTTP endpoints
+export { createAIEndpoints } from "./endpoints.ts";
+export type {
+  AIEndpoints,
+  AIEndpointDeps,
+  CreateSessionRequest,
+  QueryRequest,
+  AbortRequest,
+} from "./endpoints.ts";

package/3p/pi-plannotator/packages/ai/package.json

@@ -0,0 +1,21 @@
+{
+  "name": "@plannotator/ai",
+  "version": "0.0.1",
+  "private": true,
+  "description": "AI provider layer for Plannotator — pluggable agent runtime backbone",
+  "exports": {
+    ".": "./index.ts",
+    "./types": "./types.ts",
+    "./context": "./context.ts",
+    "./provider": "./provider.ts",
+    "./session-manager": "./session-manager.ts",
+    "./endpoints": "./endpoints.ts",
+    "./providers/claude-agent-sdk": "./providers/claude-agent-sdk.ts",
+    "./providers/codex-sdk": "./providers/codex-sdk.ts",
+    "./providers/command-path": "./providers/command-path.ts",
+    "./providers/pi-sdk": "./providers/pi-sdk.ts",
+    "./providers/opencode-sdk": "./providers/opencode-sdk.ts",
+    "./providers/pi-sdk-node": "./providers/pi-sdk-node.ts"
+  },
+  "dependencies": {}
+}

package/3p/pi-plannotator/packages/ai/provider.ts

@@ -0,0 +1,103 @@
+/**
+ * Provider registry — manages AI provider instances.
+ *
+ * Supports multiple instances of the same provider type (e.g., two Claude
+ * Agent SDK providers with different configs) keyed by instance ID.
+ *
+ * Each server (plan review, code review, annotate) should create its own
+ * ProviderRegistry or share one — no module-level global state.
+ */
+
+import type { AIProvider, AIProviderConfig } from "./types.ts";
+
+// ---------------------------------------------------------------------------
+// Factory registry (global — factories are stateless type→constructor maps)
+// ---------------------------------------------------------------------------
+
+type ProviderFactory = (config: AIProviderConfig) => Promise<AIProvider>;
+const factories = new Map<string, ProviderFactory>();
+
+/** Register a factory function for a provider type. */
+export function registerProviderFactory(
+  type: string,
+  factory: ProviderFactory
+): void {
+  factories.set(type, factory);
+}
+
+/** Create a provider from config using a registered factory. Does NOT auto-register. */
+export async function createProvider(
+  config: AIProviderConfig
+): Promise<AIProvider> {
+  const factory = factories.get(config.type);
+  if (!factory) {
+    throw new Error(
+      `No AI provider factory registered for type "${config.type}". ` +
+        `Available: ${[...factories.keys()].join(", ") || "(none)"}`
+    );
+  }
+  return factory(config);
+}
+
+// ---------------------------------------------------------------------------
+// Registry
+// ---------------------------------------------------------------------------
+
+export class ProviderRegistry {
+  private instances = new Map<string, AIProvider>();
+
+  /**
+   * Register a provider instance under an ID.
+   * If no instanceId is provided, uses `provider.name`.
+   * Returns the instanceId used.
+   */
+  register(provider: AIProvider, instanceId?: string): string {
+    const id = instanceId ?? provider.name;
+    this.instances.set(id, provider);
+    return id;
+  }
+
+  /** Get a provider by instance ID. */
+  get(instanceId: string): AIProvider | undefined {
+    return this.instances.get(instanceId);
+  }
+
+  /** Get the first registered provider (convenience for single-provider setups). */
+  getDefault(): { id: string; provider: AIProvider } | undefined {
+    const first = this.instances.entries().next();
+    if (first.done) return undefined;
+    return { id: first.value[0], provider: first.value[1] };
+  }
+
+  /** Get all instances of a given provider type (by provider.name). */
+  getByType(typeName: string): AIProvider[] {
+    return [...this.instances.values()].filter((p) => p.name === typeName);
+  }
+
+  /** List all instance IDs. */
+  list(): string[] {
+    return [...this.instances.keys()];
+  }
+
+  /** Dispose and remove a single instance. No-op if not found. */
+  dispose(instanceId: string): void {
+    const provider = this.instances.get(instanceId);
+    if (provider) {
+      provider.dispose();
+      this.instances.delete(instanceId);
+    }
+  }
+
+  /** Dispose all providers and clear the registry. */
+  disposeAll(): void {
+    for (const provider of this.instances.values()) {
+      provider.dispose();
+    }
+    this.instances.clear();
+  }
+
+  /** Number of registered instances. */
+  get size(): number {
+    return this.instances.size;
+  }
+}

package/3p/pi-plannotator/packages/ai/providers/claude-agent-sdk.ts

@@ -0,0 +1,447 @@
+/**
+ * Claude Agent SDK provider — the first concrete AIProvider implementation.
+ *
+ * Uses @anthropic-ai/claude-agent-sdk to create sessions that can:
+ * - Start fresh with Plannotator context as the system prompt
+ * - Fork from a parent Claude Code session (preserving full history)
+ * - Resume a previous Plannotator inline chat session
+ * - Stream text deltas back to the UI in real time
+ *
+ * Sessions are read-only by default (tools limited to Read, Glob, Grep)
+ * to keep inline chat safe and cost-bounded.
+ */
+
+import { buildSystemPrompt, buildForkPreamble, buildEffectivePrompt } from "../context.ts";
+import { BaseSession } from "../base-session.ts";
+import type {
+  AIProvider,
+  AIProviderCapabilities,
+  AISession,
+  AIMessage,
+  CreateSessionOptions,
+  ClaudeAgentSDKConfig,
+} from "../types.ts";
+
+// ---------------------------------------------------------------------------
+// Constants
+// ---------------------------------------------------------------------------
+
+const PROVIDER_NAME = "claude-agent-sdk";
+
+/** Default read-only tools for inline chat. */
+const DEFAULT_ALLOWED_TOOLS = ["Read", "Glob", "Grep", "WebSearch"];
+
+const DEFAULT_MAX_TURNS = 99;
+const DEFAULT_MODEL = "claude-sonnet-4-6";
+
+// ---------------------------------------------------------------------------
+// SDK query options — typed to catch typos at compile time
+// ---------------------------------------------------------------------------
+
+interface ClaudeSDKQueryOptions {
+  model: string;
+  maxTurns: number;
+  allowedTools: string[];
+  cwd: string;
+  abortController: AbortController;
+  includePartialMessages: boolean;
+  persistSession: boolean;
+  maxBudgetUsd?: number;
+  systemPrompt?: string | { type: "preset"; preset: string; append?: string };
+  resume?: string;
+  forkSession?: boolean;
+  permissionMode?: ClaudeAgentSDKConfig['permissionMode'];
+  allowDangerouslySkipPermissions?: boolean;
+  pathToClaudeCodeExecutable?: string;
+  settingSources?: string[];
+}
+
+// ---------------------------------------------------------------------------
+// Provider
+// ---------------------------------------------------------------------------
+
+export class ClaudeAgentSDKProvider implements AIProvider {
+  readonly name = PROVIDER_NAME;
+  readonly capabilities: AIProviderCapabilities = {
+    fork: true,
+    resume: true,
+    streaming: true,
+    tools: true,
+  };
+  readonly models = [
+    { id: 'claude-fable-5', label: 'Fable 5' },
+    { id: 'claude-opus-4-8', label: 'Opus 4.8' },
+    { id: 'claude-opus-4-8[1m]', label: 'Opus 4.8 (1M)' },
+    { id: 'claude-sonnet-4-6', label: 'Sonnet 4.6', default: true },
+    { id: 'claude-sonnet-4-6[1m]', label: 'Sonnet 4.6 (1M)' },
+    { id: 'claude-opus-4-7', label: 'Opus 4.7' },
+    { id: 'claude-opus-4-7[1m]', label: 'Opus 4.7 (1M)' },
+    { id: 'claude-opus-4-6', label: 'Opus 4.6' },
+    { id: 'claude-opus-4-6[1m]', label: 'Opus 4.6 (1M)' },
+    { id: 'claude-haiku-4-5', label: 'Haiku 4.5' },
+  ] as const;
+
+  private config: ClaudeAgentSDKConfig;
+
+  constructor(config: ClaudeAgentSDKConfig) {
+    this.config = config;
+  }
+
+  async createSession(options: CreateSessionOptions): Promise<AISession> {
+    return new ClaudeAgentSDKSession({
+      ...this.baseConfig(options),
+      systemPrompt: buildSystemPrompt(options.context),
+      cwd: options.cwd ?? this.config.cwd ?? process.cwd(),
+      parentSessionId: null,
+      forkFromSession: null,
+    });
+  }
+
+  async forkSession(options: CreateSessionOptions): Promise<AISession> {
+    const parent = options.context.parent;
+    if (!parent) {
+      throw new Error(
+        "Cannot fork: no parent session provided in context. " +
+          "Use createSession() for standalone sessions."
+      );
+    }
+
+    return new ClaudeAgentSDKSession({
+      ...this.baseConfig(options),
+      systemPrompt: null,
+      forkPreamble: buildForkPreamble(options.context),
+      cwd: parent.cwd,
+      parentSessionId: parent.sessionId,
+      forkFromSession: parent.sessionId,
+    });
+  }
+
+  async resumeSession(sessionId: string): Promise<AISession> {
+    return new ClaudeAgentSDKSession({
+      ...this.baseConfig(),
+      systemPrompt: null,
+      cwd: this.config.cwd ?? process.cwd(),
+      parentSessionId: null,
+      forkFromSession: null,
+      resumeSessionId: sessionId,
+    });
+  }
+
+  dispose(): void {
+    // No persistent resources to clean up
+  }
+
+  private baseConfig(options?: CreateSessionOptions) {
+    return {
+      model: options?.model ?? this.config.model ?? DEFAULT_MODEL,
+      maxTurns: options?.maxTurns ?? DEFAULT_MAX_TURNS,
+      maxBudgetUsd: options?.maxBudgetUsd,
+      allowedTools: this.config.allowedTools ?? DEFAULT_ALLOWED_TOOLS,
+      permissionMode: this.config.permissionMode ?? "default",
+      claudeExecutablePath: this.config.claudeExecutablePath,
+      settingSources: this.config.settingSources ?? ['user', 'project'],
+    };
+  }
+}
+
+// ---------------------------------------------------------------------------
+// SDK import cache — resolve once, reuse across all queries
+// ---------------------------------------------------------------------------
+
+// biome-ignore lint/suspicious/noExplicitAny: SDK types resolved at runtime via dynamic import
+let sdkQueryFn: ((...args: any[]) => any) | null = null;
+
+async function getSDKQuery() {
+  if (!sdkQueryFn) {
+    const sdk = await import("@anthropic-ai/claude-agent-sdk");
+    sdkQueryFn = sdk.query;
+  }
+  return sdkQueryFn!;
+}
+
+// ---------------------------------------------------------------------------
+// Session
+// ---------------------------------------------------------------------------
+
+interface SessionConfig {
+  systemPrompt: string | null;
+  forkPreamble?: string;
+  model: string;
+  maxTurns: number;
+  maxBudgetUsd?: number;
+  allowedTools: string[];
+  permissionMode: ClaudeAgentSDKConfig['permissionMode'];
+  cwd: string;
+  parentSessionId: string | null;
+  forkFromSession: string | null;
+  resumeSessionId?: string;
+  claudeExecutablePath?: string;
+  settingSources?: string[];
+}
+
+class ClaudeAgentSDKSession extends BaseSession {
+  private config: SessionConfig;
+  /** Active Query object — needed to send control responses (permission decisions) */
+  private _activeQuery: { streamInput: (iter: AsyncIterable<unknown>) => Promise<void> } | null = null;
+
+  constructor(config: SessionConfig) {
+    super({
+      parentSessionId: config.parentSessionId,
+      initialId: config.resumeSessionId,
+    });
+    this.config = config;
+  }
+
+  async *query(prompt: string): AsyncIterable<AIMessage> {
+    const started = this.startQuery();
+    if (!started) { yield BaseSession.BUSY_ERROR; return; }
+    const { gen } = started;
+
+    try {
+      const queryFn = await getSDKQuery();
+
+      const queryPrompt = buildEffectivePrompt(
+        prompt,
+        this.config.forkPreamble ?? null,
+        this._firstQuerySent,
+      );
+      const options = this.buildQueryOptions();
+
+      const stream = queryFn({ prompt: queryPrompt, options }) as
+        AsyncIterable<Record<string, unknown>> & { streamInput: (iter: AsyncIterable<unknown>) => Promise<void> };
+      this._activeQuery = stream;
+
+      this._firstQuerySent = true;
+
+      for await (const message of stream) {
+        const mapped = mapSDKMessage(message);
+
+        // Capture the real session ID from the init message
+        if (
+          !this._resolvedId &&
+          "session_id" in message &&
+          typeof message.session_id === "string" &&
+          message.session_id
+        ) {
+          this.resolveId(message.session_id);
+        }
+
+        for (const msg of mapped) {
+          yield msg;
+        }
+      }
+    } catch (err) {
+      yield {
+        type: "error",
+        error: err instanceof Error ? err.message : String(err),
+        code: "provider_error",
+      };
+    } finally {
+      this.endQuery(gen);
+      this._activeQuery = null;
+    }
+  }
+
+  abort(): void {
+    this._activeQuery = null;
+    super.abort();
+  }
+
+  respondToPermission(requestId: string, allow: boolean, message?: string): void {
+    if (!this._activeQuery || !this._activeQuery.streamInput) return;
+
+    const response = allow
+      ? { type: 'control_response', response: { subtype: 'success', request_id: requestId, response: { behavior: 'allow' } } }
+      : { type: 'control_response', response: { subtype: 'success', request_id: requestId, response: { behavior: 'deny', message: message ?? 'User denied this action' } } };
+
+    this._activeQuery.streamInput(
+      (async function* () { yield response; })()
+    ).catch(() => {});
+  }
+
+  // -------------------------------------------------------------------------
+  // Internal
+  // -------------------------------------------------------------------------
+
+  private buildQueryOptions(): ClaudeSDKQueryOptions {
+    const opts: ClaudeSDKQueryOptions = {
+      model: this.config.model,
+      maxTurns: this.config.maxTurns,
+      allowedTools: this.config.allowedTools,
+      cwd: this.config.cwd,
+      abortController: this._currentAbort!,
+      includePartialMessages: true,
+      persistSession: true,
+      ...(this.config.claudeExecutablePath && {
+        pathToClaudeCodeExecutable: this.config.claudeExecutablePath,
+      }),
+      ...(this.config.settingSources && {
+        settingSources: this.config.settingSources,
+      }),
+    };
+
+    if (this.config.maxBudgetUsd) {
+      opts.maxBudgetUsd = this.config.maxBudgetUsd;
+    }
+
+    // After the first query resolves a real session ID, all subsequent
+    // queries must resume that session to continue the conversation.
+    if (this._resolvedId) {
+      opts.resume = this._resolvedId;
+      return this.applyPermissionMode(opts);
+    }
+
+    // First query: use Claude Code's built-in prompt with our context appended
+    if (this.config.systemPrompt) {
+      opts.systemPrompt = {
+        type: "preset",
+        preset: "claude_code",
+        append: this.config.systemPrompt,
+      };
+    }
+
+    if (this.config.forkFromSession) {
+      opts.resume = this.config.forkFromSession;
+      opts.forkSession = true;
+    }
+
+    if (this.config.resumeSessionId) {
+      opts.resume = this.config.resumeSessionId;
+    }
+
+    return this.applyPermissionMode(opts);
+  }
+
+  private applyPermissionMode(opts: ClaudeSDKQueryOptions): ClaudeSDKQueryOptions {
+    if (this.config.permissionMode === "bypassPermissions") {
+      opts.permissionMode = "bypassPermissions";
+      opts.allowDangerouslySkipPermissions = true;
+    } else if (this.config.permissionMode === "plan") {
+      opts.permissionMode = "plan";
+    }
+    return opts;
+  }
+}
+
+// ---------------------------------------------------------------------------
+// Message mapping
+// ---------------------------------------------------------------------------
+
+/**
+ * Map an SDK message to one or more AIMessages.
+ *
+ * An SDK assistant message can contain both text and tool_use content blocks
+ * in a single response. We emit each block as a separate AIMessage so no
+ * content is dropped.
+ */
+function mapSDKMessage(msg: Record<string, unknown>): AIMessage[] {
+  const type = msg.type as string;
+
+  switch (type) {
+    case "assistant": {
+      const message = msg.message as Record<string, unknown> | undefined;
+      if (!message) return [{ type: "unknown", raw: msg }];
+      const content = message.content as Array<Record<string, unknown>>;
+      if (!content) return [{ type: "unknown", raw: msg }];
+
+      const messages: AIMessage[] = [];
+      const textParts: string[] = [];
+
+      for (const block of content) {
+        if (block.type === "text" && typeof block.text === "string") {
+          textParts.push(block.text);
+        } else if (block.type === "tool_use") {
+          // Flush accumulated text before the tool_use block
+          if (textParts.length > 0) {
+            messages.push({ type: "text", text: textParts.join("") });
+            textParts.length = 0;
+          }
+          messages.push({
+            type: "tool_use",
+            toolName: block.name as string,
+            toolInput: block.input as Record<string, unknown>,
+            toolUseId: block.id as string,
+          });
+        }
+      }
+
+      // Flush any remaining text after the last block
+      if (textParts.length > 0) {
+        messages.push({ type: "text", text: textParts.join("") });
+      }
+
+      return messages.length > 0 ? messages : [{ type: "unknown", raw: msg }];
+    }
+
+    case "stream_event": {
+      const event = msg.event as Record<string, unknown> | undefined;
+      if (!event) return [{ type: "unknown", raw: msg }];
+      const eventType = event.type as string;
+
+      if (eventType === "content_block_delta") {
+        const delta = event.delta as Record<string, unknown>;
+        if (delta?.type === "text_delta" && typeof delta.text === "string") {
+          return [{ type: "text_delta", delta: delta.text }];
+        }
+      }
+      return [{ type: "unknown", raw: msg }];
+    }
+
+    case "user": {
+      // SDK wraps tool results in SDKUserMessage (type: "user")
+      if (msg.tool_use_result != null) {
+        return [{
+          type: "tool_result",
+          result: typeof msg.tool_use_result === "string"
+            ? msg.tool_use_result
+            : JSON.stringify(msg.tool_use_result),
+        }];
+      }
+      return [{ type: "unknown", raw: msg }];
+    }
+
+    case "control_request": {
+      const request = msg.request as Record<string, unknown> | undefined;
+      if (request?.subtype === "can_use_tool") {
+        return [{
+          type: "permission_request",
+          requestId: msg.request_id as string,
+          toolName: request.tool_name as string,
+          toolInput: (request.input as Record<string, unknown>) ?? {},
+          title: request.title as string | undefined,
+          displayName: request.display_name as string | undefined,
+          description: request.description as string | undefined,
+          toolUseId: request.tool_use_id as string,
+        }];
+      }
+      return [{ type: "unknown", raw: msg }];
+    }
+
+    case "result": {
+      const sessionId = (msg.session_id as string) ?? "";
+      const subtype = msg.subtype as string;
+      return [{
+        type: "result",
+        sessionId,
+        success: subtype === "success",
+        result: (msg.result as string) ?? undefined,
+        costUsd: msg.total_cost_usd as number | undefined,
+        turns: msg.num_turns as number | undefined,
+      }];
+    }
+
+    default:
+      return [{ type: "unknown", raw: msg }];
+  }
+}
+
+// ---------------------------------------------------------------------------
+// Factory registration
+// ---------------------------------------------------------------------------
+
+import { registerProviderFactory } from "../provider.ts";
+
+registerProviderFactory(
+  PROVIDER_NAME,
+  async (config) => new ClaudeAgentSDKProvider(config as ClaudeAgentSDKConfig)
+);