@openparachute/agent 0.1.2 → 0.2.0

package/container/agent-runner/src/providers/claude.ts

@@ -1,379 +0,0 @@
-import fs from 'fs';
-import path from 'path';
-
-import { query as sdkQuery, type HookCallback, type PreCompactHookInput } from '@anthropic-ai/claude-agent-sdk';
-
-import { appendActivity, clearContainerToolInFlight, setContainerToolInFlight } from '../db/connection.js';
-import { registerProvider } from './provider-registry.js';
-import type {
-  AgentProvider,
-  AgentQuery,
-  McpServerConfig,
-  ProviderEvent,
-  ProviderOptions,
-  QueryInput,
-} from './types.js';
-
-function log(msg: string): void {
-  console.error(`[claude-provider] ${msg}`);
-}
-
-// Deferred SDK builtins that either sidestep parachute-agent's own scheduling or
-// don't fit our async message-passing model (they're designed for Claude
-// Code's interactive UI and would hang here).
-//
-// - CronCreate / CronDelete / CronList / ScheduleWakeup: we have durable
-//   scheduling via mcp__parachute_agent__schedule_task.
-// - AskUserQuestion: SDK returns a placeholder instead of blocking on a
-//   real answer — we have mcp__parachute_agent__ask_user_question that persists
-//   the question and blocks on the real reply.
-// - EnterPlanMode / ExitPlanMode / EnterWorktree / ExitWorktree: Claude
-//   Code UI affordances; in a headless container they'd appear stuck.
-const SDK_DISALLOWED_TOOLS = [
-  'CronCreate',
-  'CronDelete',
-  'CronList',
-  'ScheduleWakeup',
-  'AskUserQuestion',
-  'EnterPlanMode',
-  'ExitPlanMode',
-  'EnterWorktree',
-  'ExitWorktree',
-];
-
-// Tool allowlist for parachute-agent containers
-const TOOL_ALLOWLIST = [
-  'Bash',
-  'Read',
-  'Write',
-  'Edit',
-  'Glob',
-  'Grep',
-  'WebSearch',
-  'WebFetch',
-  'Task',
-  'TaskOutput',
-  'TaskStop',
-  'TeamCreate',
-  'TeamDelete',
-  'SendMessage',
-  'TodoWrite',
-  'ToolSearch',
-  'Skill',
-  'NotebookEdit',
-  'mcp__parachute_agent__*',
-];
-
-interface SDKUserMessage {
-  type: 'user';
-  message: { role: 'user'; content: string };
-  parent_tool_use_id: null;
-  session_id: string;
-}
-
-/**
- * Push-based async iterable for streaming user messages to the Claude SDK.
- */
-class MessageStream {
-  private queue: SDKUserMessage[] = [];
-  private waiting: (() => void) | null = null;
-  private done = false;
-
-  push(text: string): void {
-    this.queue.push({
-      type: 'user',
-      message: { role: 'user', content: text },
-      parent_tool_use_id: null,
-      session_id: '',
-    });
-    this.waiting?.();
-  }
-
-  end(): void {
-    this.done = true;
-    this.waiting?.();
-  }
-
-  async *[Symbol.asyncIterator](): AsyncGenerator<SDKUserMessage> {
-    while (true) {
-      while (this.queue.length > 0) {
-        yield this.queue.shift()!;
-      }
-      if (this.done) return;
-      await new Promise<void>((r) => {
-        this.waiting = r;
-      });
-      this.waiting = null;
-    }
-  }
-}
-
-// ── Transcript archiving (PreCompact hook) ──
-
-interface ParsedMessage {
-  role: 'user' | 'assistant';
-  content: string;
-}
-
-function parseTranscript(content: string): ParsedMessage[] {
-  const messages: ParsedMessage[] = [];
-  for (const line of content.split('\n')) {
-    if (!line.trim()) continue;
-    try {
-      const entry = JSON.parse(line);
-      if (entry.type === 'user' && entry.message?.content) {
-        const text =
-          typeof entry.message.content === 'string'
-            ? entry.message.content
-            : entry.message.content.map((c: { text?: string }) => c.text || '').join('');
-        if (text) messages.push({ role: 'user', content: text });
-      } else if (entry.type === 'assistant' && entry.message?.content) {
-        const textParts = entry.message.content
-          .filter((c: { type: string }) => c.type === 'text')
-          .map((c: { text: string }) => c.text);
-        const text = textParts.join('');
-        if (text) messages.push({ role: 'assistant', content: text });
-      }
-    } catch {
-      /* skip unparseable lines */
-    }
-  }
-  return messages;
-}
-
-function formatTranscriptMarkdown(messages: ParsedMessage[], title?: string | null, assistantName?: string): string {
-  const now = new Date();
-  const dateStr = now.toLocaleString('en-US', {
-    month: 'short',
-    day: 'numeric',
-    hour: 'numeric',
-    minute: '2-digit',
-    hour12: true,
-  });
-  const lines = [`# ${title || 'Conversation'}`, '', `Archived: ${dateStr}`, '', '---', ''];
-  for (const msg of messages) {
-    const sender = msg.role === 'user' ? 'User' : assistantName || 'Assistant';
-    const content = msg.content.length > 2000 ? msg.content.slice(0, 2000) + '...' : msg.content;
-    lines.push(`**${sender}**: ${content}`, '');
-  }
-  return lines.join('\n');
-}
-
-/**
- * PreToolUse hook: record the current tool + its declared timeout so the host
- * sweep can widen its stuck tolerance while Bash is running a long-declared
- * script. Defense-in-depth: if SDK_DISALLOWED_TOOLS slips through somehow,
- * block the call here instead of letting the agent hang.
- */
-const preToolUseHook: HookCallback = async (input) => {
-  const i = input as { tool_name?: string; tool_input?: Record<string, unknown> };
-  const toolName = i.tool_name ?? '';
-  if (SDK_DISALLOWED_TOOLS.includes(toolName)) {
-    return {
-      decision: 'block',
-      stopReason: `Tool '${toolName}' is not available in this environment — use the parachute-agent equivalent.`,
-    } as unknown as ReturnType<HookCallback>;
-  }
-  // Bash exposes its timeout via the tool_input.timeout field (ms). Any other
-  // tool: no declared timeout.
-  const declaredTimeoutMs =
-    toolName === 'Bash' && typeof i.tool_input?.timeout === 'number' ? (i.tool_input.timeout as number) : null;
-  try {
-    setContainerToolInFlight(toolName, declaredTimeoutMs);
-  } catch (err) {
-    log(`PreToolUse: failed to record container_state: ${err instanceof Error ? err.message : String(err)}`);
-  }
-  // Activity ledger — append-only per-invocation row. Kind classifies; target
-  // names the tool. summary is left null for Bash because tool_input.command
-  // can carry env-injected secrets via argv. Other tools' inputs may also
-  // contain secrets, so we keep summary null across the board for now and
-  // tighten case-by-case as the surface stabilizes.
-  try {
-    let kind: string;
-    if (toolName === 'Bash') kind = 'cmd_exec';
-    else if (toolName.startsWith('mcp__')) kind = 'mcp_call';
-    else kind = 'tool_call';
-    appendActivity(kind, toolName || null, null);
-  } catch (err) {
-    log(`PreToolUse: failed to append activity: ${err instanceof Error ? err.message : String(err)}`);
-  }
-  return { continue: true };
-};
-
-/** Clear in-flight tool on PostToolUse / PostToolUseFailure. */
-const postToolUseHook: HookCallback = async () => {
-  try {
-    clearContainerToolInFlight();
-  } catch (err) {
-    log(`PostToolUse: failed to clear container_state: ${err instanceof Error ? err.message : String(err)}`);
-  }
-  return { continue: true };
-};
-
-function createPreCompactHook(assistantName?: string): HookCallback {
-  return async (input) => {
-    const preCompact = input as PreCompactHookInput;
-    const { transcript_path: transcriptPath, session_id: sessionId } = preCompact;
-
-    if (!transcriptPath || !fs.existsSync(transcriptPath)) {
-      log('No transcript found for archiving');
-      return {};
-    }
-
-    try {
-      const content = fs.readFileSync(transcriptPath, 'utf-8');
-      const messages = parseTranscript(content);
-      if (messages.length === 0) return {};
-
-      // Try to get summary from sessions index
-      let summary: string | undefined;
-      const indexPath = path.join(path.dirname(transcriptPath), 'sessions-index.json');
-      if (fs.existsSync(indexPath)) {
-        try {
-          const index = JSON.parse(fs.readFileSync(indexPath, 'utf-8'));
-          summary = index.entries?.find(
-            (e: { sessionId: string; summary?: string }) => e.sessionId === sessionId,
-          )?.summary;
-        } catch {
-          /* ignore */
-        }
-      }
-
-      const name = summary
-        ? summary
-            .toLowerCase()
-            .replace(/[^a-z0-9]+/g, '-')
-            .replace(/^-+|-+$/g, '')
-            .slice(0, 50)
-        : `conversation-${new Date().getHours().toString().padStart(2, '0')}${new Date().getMinutes().toString().padStart(2, '0')}`;
-
-      const conversationsDir = '/workspace/agent/conversations';
-      fs.mkdirSync(conversationsDir, { recursive: true });
-      const filename = `${new Date().toISOString().split('T')[0]}-${name}.md`;
-      fs.writeFileSync(
-        path.join(conversationsDir, filename),
-        formatTranscriptMarkdown(messages, summary, assistantName),
-      );
-      log(`Archived conversation to ${filename}`);
-    } catch (err) {
-      log(`Failed to archive transcript: ${err instanceof Error ? err.message : String(err)}`);
-    }
-    return {};
-  };
-}
-
-// ── Provider ──
-
-/**
- * Claude Code auto-compacts context at this window (tokens). Kept here so
- * the generic bootstrap doesn't need to know about Claude-specific env vars.
- */
-const CLAUDE_CODE_AUTO_COMPACT_WINDOW = '165000';
-
-/**
- * Stale-session detection. Matches Claude Code's error text when a
- * resumed session can't be found — missing transcript .jsonl, unknown
- * session ID, etc.
- */
-const STALE_SESSION_RE = /no conversation found|ENOENT.*\.jsonl|session.*not found/i;
-
-export class ClaudeProvider implements AgentProvider {
-  readonly supportsNativeSlashCommands = true;
-
-  private assistantName?: string;
-  private mcpServers: Record<string, McpServerConfig>;
-  private env: Record<string, string | undefined>;
-  private additionalDirectories?: string[];
-
-  constructor(options: ProviderOptions = {}) {
-    this.assistantName = options.assistantName;
-    this.mcpServers = options.mcpServers ?? {};
-    this.additionalDirectories = options.additionalDirectories;
-    this.env = {
-      ...(options.env ?? {}),
-      CLAUDE_CODE_AUTO_COMPACT_WINDOW,
-    };
-  }
-
-  isSessionInvalid(err: unknown): boolean {
-    const msg = err instanceof Error ? err.message : String(err);
-    return STALE_SESSION_RE.test(msg);
-  }
-
-  query(input: QueryInput): AgentQuery {
-    const stream = new MessageStream();
-    stream.push(input.prompt);
-
-    const instructions = input.systemContext?.instructions;
-
-    const sdkResult = sdkQuery({
-      prompt: stream,
-      options: {
-        cwd: input.cwd,
-        additionalDirectories: this.additionalDirectories,
-        resume: input.continuation,
-        pathToClaudeCodeExecutable: '/pnpm/claude',
-        systemPrompt: instructions
-          ? { type: 'preset' as const, preset: 'claude_code' as const, append: instructions }
-          : undefined,
-        allowedTools: TOOL_ALLOWLIST,
-        disallowedTools: SDK_DISALLOWED_TOOLS,
-        env: this.env,
-        permissionMode: 'bypassPermissions',
-        allowDangerouslySkipPermissions: true,
-        settingSources: ['project', 'user'],
-        mcpServers: this.mcpServers,
-        hooks: {
-          PreToolUse: [{ hooks: [preToolUseHook] }],
-          PostToolUse: [{ hooks: [postToolUseHook] }],
-          PostToolUseFailure: [{ hooks: [postToolUseHook] }],
-          PreCompact: [{ hooks: [createPreCompactHook(this.assistantName)] }],
-        },
-      },
-    });
-
-    let aborted = false;
-
-    async function* translateEvents(): AsyncGenerator<ProviderEvent> {
-      let messageCount = 0;
-      for await (const message of sdkResult) {
-        if (aborted) return;
-        messageCount++;
-
-        // Yield activity for every SDK event so the poll loop knows the agent is working
-        yield { type: 'activity' };
-
-        if (message.type === 'system' && message.subtype === 'init') {
-          yield { type: 'init', continuation: message.session_id };
-        } else if (message.type === 'result') {
-          const text = 'result' in message ? ((message as { result?: string }).result ?? null) : null;
-          yield { type: 'result', text };
-        } else if (message.type === 'system' && (message as { subtype?: string }).subtype === 'api_retry') {
-          yield { type: 'error', message: 'API retry', retryable: true };
-        } else if (message.type === 'system' && (message as { subtype?: string }).subtype === 'rate_limit_event') {
-          yield { type: 'error', message: 'Rate limit', retryable: false, classification: 'quota' };
-        } else if (message.type === 'system' && (message as { subtype?: string }).subtype === 'compact_boundary') {
-          const meta = (message as { compact_metadata?: { pre_tokens?: number } }).compact_metadata;
-          const detail = meta?.pre_tokens ? ` (${meta.pre_tokens.toLocaleString()} tokens compacted)` : '';
-          yield { type: 'result', text: `Context compacted${detail}.` };
-        } else if (message.type === 'system' && (message as { subtype?: string }).subtype === 'task_notification') {
-          const tn = message as { summary?: string };
-          yield { type: 'progress', message: tn.summary || 'Task notification' };
-        }
-      }
-      log(`Query completed after ${messageCount} SDK messages`);
-    }
-
-    return {
-      push: (msg) => stream.push(msg),
-      end: () => stream.end(),
-      events: translateEvents(),
-      abort: () => {
-        aborted = true;
-        stream.end();
-      },
-    };
-  }
-}
-
-registerProvider('claude', (opts) => new ClaudeProvider(opts));

package/container/agent-runner/src/providers/factory.test.ts

@@ -1,19 +0,0 @@
-import { describe, it, expect } from 'bun:test';
-
-import { createProvider, type ProviderName } from './factory.js';
-import { ClaudeProvider } from './claude.js';
-import { MockProvider } from './mock.js';
-
-describe('createProvider', () => {
-  it('returns ClaudeProvider for claude', () => {
-    expect(createProvider('claude')).toBeInstanceOf(ClaudeProvider);
-  });
-
-  it('returns MockProvider for mock', () => {
-    expect(createProvider('mock')).toBeInstanceOf(MockProvider);
-  });
-
-  it('throws for unknown name', () => {
-    expect(() => createProvider('bogus' as ProviderName)).toThrow(/Unknown provider/);
-  });
-});

package/container/agent-runner/src/providers/factory.ts

@@ -1,13 +0,0 @@
-import type { AgentProvider, ProviderOptions } from './types.js';
-import { getProviderFactory } from './provider-registry.js';
-
-/**
- * Any registered provider name. Kept as a named alias for readability; the
- * set of valid names is open and determined at runtime by whichever provider
- * modules the `providers/index.ts` barrel imports.
- */
-export type ProviderName = string;
-
-export function createProvider(name: ProviderName, options: ProviderOptions = {}): AgentProvider {
-  return getProviderFactory(name)(options);
-}

package/container/agent-runner/src/providers/index.ts

@@ -1,6 +0,0 @@
-// Provider self-registration barrel.
-// Each import triggers the provider module's registerProvider() call at top
-// level. Skills add a new provider by appending one import line below.
-
-import './claude.js';
-import './mock.js';

package/container/agent-runner/src/providers/mock.ts

@@ -1,77 +0,0 @@
-import { registerProvider } from './provider-registry.js';
-import type { AgentProvider, AgentQuery, ProviderEvent, ProviderOptions, QueryInput } from './types.js';
-
-/**
- * Mock provider for testing. Returns canned responses.
- * Supports push() — queued messages produce additional results.
- */
-export class MockProvider implements AgentProvider {
-  readonly supportsNativeSlashCommands = false;
-
-  private responseFactory: (prompt: string) => string;
-
-  constructor(_options: ProviderOptions = {}, responseFactory?: (prompt: string) => string) {
-    this.responseFactory = responseFactory ?? ((prompt) => `Mock response to: ${prompt.slice(0, 100)}`);
-  }
-
-  isSessionInvalid(_err: unknown): boolean {
-    return false;
-  }
-
-  query(input: QueryInput): AgentQuery {
-    const pending: string[] = [];
-    let waiting: (() => void) | null = null;
-    let ended = false;
-    let aborted = false;
-    const responseFactory = this.responseFactory;
-
-    const events: AsyncIterable<ProviderEvent> = {
-      async *[Symbol.asyncIterator]() {
-        yield { type: 'activity' };
-        yield { type: 'init', continuation: `mock-session-${Date.now()}` };
-
-        // Process initial prompt
-        yield { type: 'activity' };
-        yield { type: 'result', text: responseFactory(input.prompt) };
-
-        // Process any pushed follow-ups
-        while (!ended && !aborted) {
-          if (pending.length > 0) {
-            const msg = pending.shift()!;
-            yield { type: 'result', text: responseFactory(msg) };
-            continue;
-          }
-          // Wait for push() or end()
-          await new Promise<void>((resolve) => {
-            waiting = resolve;
-          });
-          waiting = null;
-        }
-
-        // Drain remaining
-        while (pending.length > 0) {
-          const msg = pending.shift()!;
-          yield { type: 'result', text: responseFactory(msg) };
-        }
-      },
-    };
-
-    return {
-      push(message: string) {
-        pending.push(message);
-        waiting?.();
-      },
-      end() {
-        ended = true;
-        waiting?.();
-      },
-      events,
-      abort() {
-        aborted = true;
-        waiting?.();
-      },
-    };
-  }
-}
-
-registerProvider('mock', (opts) => new MockProvider(opts));

package/container/agent-runner/src/providers/provider-registry.ts

@@ -1,33 +0,0 @@
-/**
- * Provider self-registration registry.
- *
- * Mirrors `src/channels/channel-registry.ts` on the host. Each provider module
- * calls `registerProvider()` at top level; the barrel (`providers/index.ts`)
- * imports every provider module for its side effect so registrations fire
- * before `createProvider()` is called.
- */
-import type { AgentProvider, ProviderOptions } from './types.js';
-
-export type ProviderFactory = (options: ProviderOptions) => AgentProvider;
-
-const registry = new Map<string, ProviderFactory>();
-
-export function registerProvider(name: string, factory: ProviderFactory): void {
-  if (registry.has(name)) {
-    throw new Error(`Provider already registered: ${name}`);
-  }
-  registry.set(name, factory);
-}
-
-export function getProviderFactory(name: string): ProviderFactory {
-  const factory = registry.get(name);
-  if (!factory) {
-    const known = [...registry.keys()].join(', ') || '(none)';
-    throw new Error(`Unknown provider: ${name}. Registered: ${known}`);
-  }
-  return factory;
-}
-
-export function listProviderNames(): string[] {
-  return [...registry.keys()];
-}

package/container/agent-runner/src/providers/types.ts

@@ -1,82 +0,0 @@
-export interface AgentProvider {
-  /**
-   * True if the provider's underlying SDK handles slash commands natively and
-   * wants them passed through as raw text. When false, the poll-loop formats
-   * slash commands like any other chat message.
-   */
-  readonly supportsNativeSlashCommands: boolean;
-
-  /** Start a new query. Returns a handle for streaming input and output. */
-  query(input: QueryInput): AgentQuery;
-
-  /**
-   * True if the given error indicates the stored continuation is invalid
-   * (missing transcript, unknown session, etc.) and should be cleared.
-   */
-  isSessionInvalid(err: unknown): boolean;
-}
-
-/**
- * Options passed to provider constructors. Fields are common to most
- * providers; individual providers may ignore any they don't need.
- */
-export interface ProviderOptions {
-  assistantName?: string;
-  mcpServers?: Record<string, McpServerConfig>;
-  env?: Record<string, string | undefined>;
-  additionalDirectories?: string[];
-}
-
-export interface QueryInput {
-  /** Initial prompt (already formatted by agent-runner). */
-  prompt: string;
-
-  /**
-   * Opaque continuation token from a previous query. The provider decides
-   * what this means (session ID, thread ID, nothing at all).
-   */
-  continuation?: string;
-
-  /** Working directory inside the container. */
-  cwd: string;
-
-  /**
-   * System context to inject. Providers translate this into whatever their
-   * SDK expects (preset append, full system prompt, per-turn injection…).
-   */
-  systemContext?: {
-    instructions?: string;
-  };
-}
-
-export interface McpServerConfig {
-  command: string;
-  args: string[];
-  env: Record<string, string>;
-}
-
-export interface AgentQuery {
-  /** Push a follow-up message into the active query. */
-  push(message: string): void;
-
-  /** Signal that no more input will be sent. */
-  end(): void;
-
-  /** Output event stream. */
-  events: AsyncIterable<ProviderEvent>;
-
-  /** Force-stop the query. */
-  abort(): void;
-}
-
-export type ProviderEvent =
-  | { type: 'init'; continuation: string }
-  | { type: 'result'; text: string | null }
-  | { type: 'error'; message: string; retryable: boolean; classification?: string }
-  | { type: 'progress'; message: string }
-  /**
-   * Liveness signal. Providers MUST yield this on every underlying SDK
-   * event (tool call, thinking, partial message, anything) so the
-   * poll-loop's idle timer stays honest during long tool runs.
-   */
-  | { type: 'activity' };