@myrialabs/clopen 0.0.7 → 0.1.1

package/shared/types/messaging/tool.ts

@@ -1,401 +1,135 @@
 /**
- * Tool Input Types
+ * Tool Types
  *
- * Defines all tool input interfaces for UI components
+ * Defines all tool input/output interfaces for UI components.
+ * Input types AND output types are imported from the official SDK.
  */
 
-// Import tool input types statically
+// ============================================================
+// SDK Tool Input Types
+// ============================================================
+
 import type {
+  AgentInput,
+  AskUserQuestionInput,
   BashInput,
   TaskOutputInput,
-  FileEditInput,
+  ConfigInput,
+  EnterWorktreeInput,
   ExitPlanModeInput,
+  FileEditInput,
+  FileReadInput,
+  FileWriteInput,
   GlobInput,
   GrepInput,
-  KillShellInput,
   ListMcpResourcesInput,
+  McpInput,
   NotebookEditInput,
   ReadMcpResourceInput,
-  FileReadInput,
-  AgentInput,
+  TaskStopInput,
   TodoWriteInput,
   WebFetchInput,
-  WebSearchInput,
-  FileWriteInput
+  WebSearchInput
 } from '@anthropic-ai/claude-agent-sdk/sdk-tools';
 
-// Tool output types based on Claude Code SDK
-
-/**
- * Output from Task tool execution
- */
-export interface TaskOutput {
-  /** Final result message from the subagent */
-  result: string;
-  /** Token usage statistics */
-  usage?: {
-    /** Number of input tokens consumed */
-    input_tokens: number;
-    /** Number of output tokens generated */
-    output_tokens: number;
-    /** Number of cache creation input tokens */
-    cache_creation_input_tokens?: number;
-    /** Number of cache read input tokens */
-    cache_read_input_tokens?: number;
-  };
-  /** Total cost in USD for this task execution */
-  total_cost_usd?: number;
-  /** Execution duration in milliseconds */
-  duration_ms?: number;
-}
-
-/**
- * Output from Bash tool execution
- */
-export interface BashOutput {
-  /** Combined stdout and stderr output */
-  output: string;
-  /** Exit code of the command */
-  exitCode: number;
-  /** Whether the command was killed due to timeout */
-  killed?: boolean;
-  /** Shell ID for background processes */
-  shellId?: string;
-}
-
-/**
- * Output from BashOutput tool for monitoring background processes
- */
-export interface BashOutputToolOutput {
-  /** Output from the background process */
-  output: string;
-  /** Current status of the background process */
-  status: 'running' | 'completed' | 'failed';
-  /** Exit code if the process has completed */
-  exitCode?: number;
-}
-
-/**
- * Output from Edit tool for file modifications
- */
-export interface EditOutput {
-  /** Confirmation message about the edit operation */
-  message: string;
-  /** Number of replacements made */
-  replacements: number;
-  /** Path to the edited file */
-  file_path: string;
-}
-
-/**
- * Output when reading a text file
- */
-export interface TextFileOutput {
-  /** Text content of the file */
-  content: string;
-  /** Total number of lines in the file */
-  total_lines: number;
-  /** Number of lines returned (may be limited by offset/limit) */
-  lines_returned: number;
-}
-
-/**
- * Output when reading an image file
- */
-export interface ImageFileOutput {
-  /** Base64-encoded image data */
-  image: string;
-  /** MIME type of the image */
-  mime_type: string;
-  /** File size in bytes */
-  file_size: number;
-}
-
-/**
- * Output when reading a PDF file
- */
-export interface PDFFileOutput {
-  /** Array of pages with text and images */
-  pages: Array<{
-    /** Page number (1-indexed) */
-    page_number: number;
-    /** Extracted text from the page */
-    text?: string;
-    /** Images found on the page */
-    images?: Array<{
-      /** Base64-encoded image data */
-      image: string;
-      /** MIME type of the image */
-      mime_type: string;
-    }>;
-  }>;
-  /** Total number of pages in the PDF */
-  total_pages: number;
-}
-
-/**
- * Output when reading a Jupyter notebook file
- */
-export interface NotebookFileOutput {
-  /** Array of notebook cells */
-  cells: Array<{
-    /** Type of the cell */
-    cell_type: 'code' | 'markdown';
-    /** Source code or markdown content */
-    source: string;
-    /** Cell outputs (for code cells) */
-    outputs?: any[];
-    /** Execution count (for code cells) */
-    execution_count?: number;
-  }>;
-  /** Notebook metadata */
-  metadata?: Record<string, any>;
-}
-
-/**
- * Union type for all possible Read tool outputs
- */
-export type ReadOutput =
-  | TextFileOutput
-  | ImageFileOutput
-  | PDFFileOutput
-  | NotebookFileOutput;
-
-/**
- * Output from Glob tool for file pattern matching
- */
-export interface GlobOutput {
-  /** Array of file paths matching the pattern */
-  matches: string[];
-  /** Number of matches found */
-  count: number;
-  /** Search directory used */
-  search_path: string;
-}
-
-/**
- * Output from Grep tool for content searching (content mode)
- */
-export interface GrepContentOutput {
-  /** Matching lines with context */
-  matches: Array<{
-    /** File path where the match was found */
-    file: string;
-    /** Line number of the match (if available) */
-    line_number?: number;
-    /** Content of the matching line */
-    line: string;
-    /** Lines before the match (if -B or -C was used) */
-    before_context?: string[];
-    /** Lines after the match (if -A or -C was used) */
-    after_context?: string[];
-  }>;
-  /** Total number of matches found */
-  total_matches: number;
-}
-
-/**
- * Output from Grep tool for files mode
- */
-export interface GrepFilesOutput {
-  /** Files containing matches */
-  files: string[];
-  /** Number of files with matches */
-  count: number;
-}
-
-/**
- * Output from Grep tool for count mode
- */
-export interface GrepCountOutput {
-  /** Match counts per file */
-  counts: Array<{
-    /** File path */
-    file: string;
-    /** Number of matches in this file */
-    count: number;
-  }>;
-  /** Total matches across all files */
-  total: number;
-}
-
-/**
- * Union type for all possible Grep outputs based on output_mode
- */
-export type GrepOutput =
-  | GrepContentOutput
-  | GrepFilesOutput
-  | GrepCountOutput;
-
-/**
- * Output from WebFetch tool for web content retrieval
- */
-export interface WebFetchOutput {
-  /** AI model's response to the prompt */
-  response: string;
-  /** URL that was fetched */
-  url: string;
-  /** Final URL after redirects */
-  final_url?: string;
-  /** HTTP status code */
-  status_code?: number;
-}
-
-/**
- * Output from WebSearch tool for web searching
- */
-export interface WebSearchOutput {
-  /** Array of search results */
-  results: Array<{
-    /** Title of the search result */
-    title: string;
-    /** URL of the search result */
-    url: string;
-    /** Snippet/description of the search result */
-    snippet: string;
-  }>;
-  /** Total number of results found */
-  total_results: number;
-}
-
-/**
- * Output from Write tool for file creation
- */
-export interface WriteOutput {
-  /** Confirmation message about the write operation */
-  message: string;
-  /** Path to the written file */
-  file_path: string;
-  /** Number of bytes written */
-  bytes_written: number;
-}
-
-/**
- * Output from NotebookEdit tool for Jupyter notebook modifications
- */
-export interface NotebookEditOutput {
-  /** Confirmation message about the notebook edit */
-  message: string;
-  /** Type of edit performed */
-  edit_type: 'replaced' | 'inserted' | 'deleted';
-  /** ID of the edited cell (if applicable) */
-  cell_id?: string;
-  /** Total cells in notebook after edit */
-  total_cells: number;
-}
-
-/**
- * Output from TodoWrite tool for task management
- */
-export interface TodoWriteOutput {
-  /** Confirmation message about the todo operation */
-  message: string;
-  /** Current todo statistics */
-  stats: {
-    /** Total number of todos */
-    total: number;
-    /** Number of pending todos */
-    pending: number;
-    /** Number of in-progress todos */
-    in_progress: number;
-    /** Number of completed todos */
-    completed: number;
-  };
-}
+// ============================================================
+// SDK Tool Output Types
+// ============================================================
 
-/**
- * Output from ExitPlanMode tool
- */
-export interface ExitPlanModeOutput {
-  /** Confirmation message about exiting plan mode */
-  message: string;
-}
-
-/**
- * Output from KillShell tool for terminating background processes
- */
-export interface KillShellOutput {
-  /** Confirmation message about killing the shell */
-  message: string;
-  /** ID of the killed shell */
-  shell_id: string;
-}
+import type {
+  AgentOutput as SDKAgentOutput,
+  AskUserQuestionOutput as SDKAskUserQuestionOutput,
+  BashOutput as SDKBashOutput,
+  ConfigOutput as SDKConfigOutput,
+  EnterWorktreeOutput as SDKEnterWorktreeOutput,
+  ExitPlanModeOutput as SDKExitPlanModeOutput,
+  FileEditOutput as SDKFileEditOutput,
+  FileReadOutput as SDKFileReadOutput,
+  FileWriteOutput as SDKFileWriteOutput,
+  GlobOutput as SDKGlobOutput,
+  GrepOutput as SDKGrepOutput,
+  ListMcpResourcesOutput as SDKListMcpResourcesOutput,
+  NotebookEditOutput as SDKNotebookEditOutput,
+  ReadMcpResourceOutput as SDKReadMcpResourceOutput,
+  TaskStopOutput as SDKTaskStopOutput,
+  TodoWriteOutput as SDKTodoWriteOutput,
+  WebFetchOutput as SDKWebFetchOutput,
+  WebSearchOutput as SDKWebSearchOutput
+} from '@anthropic-ai/claude-agent-sdk/sdk-tools';
 
-/**
- * Output from ListMcpResources tool for MCP resource listing
- */
-export interface ListMcpResourcesOutput {
-  /** Array of available MCP resources */
-  resources: Array<{
-    /** URI of the resource */
-    uri: string;
-    /** Name of the resource */
-    name: string;
-    /** Description of the resource */
-    description?: string;
-    /** MIME type of the resource */
-    mimeType?: string;
-    /** Server that provides this resource */
-    server: string;
-  }>;
-  /** Total number of resources */
-  total: number;
-}
+// Re-export SDK output types with consistent names
+export type TaskOutput = SDKAgentOutput;
+export type AskUserQuestionOutput = SDKAskUserQuestionOutput;
+export type BashOutput = SDKBashOutput;
+export type ConfigOutput = SDKConfigOutput;
+export type EnterWorktreeOutput = SDKEnterWorktreeOutput;
+export type ExitPlanModeOutput = SDKExitPlanModeOutput;
+export type EditOutput = SDKFileEditOutput;
+export type ReadOutput = SDKFileReadOutput;
+export type WriteOutput = SDKFileWriteOutput;
+export type GlobOutput = SDKGlobOutput;
+export type GrepOutput = SDKGrepOutput;
+export type ListMcpResourcesOutput = SDKListMcpResourcesOutput;
+export type NotebookEditOutput = SDKNotebookEditOutput;
+export type ReadMcpResourceOutput = SDKReadMcpResourceOutput;
+export type TaskStopOutput = SDKTaskStopOutput;
+export type TodoWriteOutput = SDKTodoWriteOutput;
+export type WebFetchOutput = SDKWebFetchOutput;
+export type WebSearchOutput = SDKWebSearchOutput;
 
 /**
- * Output from ReadMcpResource tool for MCP resource reading
- */
-export interface ReadMcpResourceOutput {
-  /** Resource contents (can be multiple if resource has multiple parts) */
-  contents: Array<{
-    /** URI of the resource */
-    uri: string;
-    /** MIME type of the resource */
-    mimeType?: string;
-    /** Text content (for text resources) */
-    text?: string;
-    /** Base64-encoded blob (for binary resources) */
-    blob?: string;
-  }>;
-  /** Server that provided the resource */
-  server: string;
-}
-
-/**
- * Union type for all possible tool outputs
+ * Union type for all possible tool outputs (SDK types)
  */
 export type ToolOutput =
   | TaskOutput
+  | AskUserQuestionOutput
   | BashOutput
-  | BashOutputToolOutput
+  | ConfigOutput
+  | EnterWorktreeOutput
+  | ExitPlanModeOutput
   | EditOutput
   | ReadOutput
+  | WriteOutput
   | GlobOutput
   | GrepOutput
-  | WebFetchOutput
-  | WebSearchOutput
-  | WriteOutput
+  | ListMcpResourcesOutput
   | NotebookEditOutput
+  | ReadMcpResourceOutput
+  | TaskStopOutput
   | TodoWriteOutput
-  | ExitPlanModeOutput
-  | KillShellOutput
-  | ListMcpResourcesOutput
-  | ReadMcpResourceOutput;
+  | WebFetchOutput
+  | WebSearchOutput;
+
+// ============================================================
+// Tool Result Type (for embedding in tool_use)
+// ============================================================
 
-// Tool result type for embedding in tool_use
 export interface ToolResult {
   type: 'tool_result';
   tool_use_id: string;
   content: string;
 }
 
-// Tool input types for UI components
+// ============================================================
+// Tool Metadata (set by chat service on stream end)
+// ============================================================
+
+export interface ToolMetadata {
+  /** Tool was interrupted — stream ended (error/cancel/complete) before tool got its result */
+  interrupted?: boolean;
+}
+
+// ============================================================
+// Tool Input Types for UI Components
+// ============================================================
+
 export interface BashToolInput {
   type: 'tool_use';
   id: string;
   name: 'Bash';
   input: BashInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface BashOutputToolInput {
@@ -404,6 +138,7 @@ export interface BashOutputToolInput {
   name: 'TaskOutput';
   input: TaskOutputInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface EditToolInput {
@@ -412,6 +147,7 @@ export interface EditToolInput {
   name: 'Edit';
   input: FileEditInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface ExitPlanModeToolInput {
@@ -420,6 +156,7 @@ export interface ExitPlanModeToolInput {
   name: 'ExitPlanMode';
   input: ExitPlanModeInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface GlobToolInput {
@@ -428,6 +165,7 @@ export interface GlobToolInput {
   name: 'Glob';
   input: GlobInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface GrepToolInput {
@@ -436,14 +174,16 @@ export interface GrepToolInput {
   name: 'Grep';
   input: GrepInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
-export interface KillShellToolInput {
+export interface TaskStopToolInput {
   type: 'tool_use';
   id: string;
-  name: 'KillShell';
-  input: KillShellInput;
+  name: 'TaskStop';
+  input: TaskStopInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface ListMcpResourcesToolInput {
@@ -452,6 +192,7 @@ export interface ListMcpResourcesToolInput {
   name: 'ListMcpResources';
   input: ListMcpResourcesInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface NotebookEditToolInput {
@@ -460,6 +201,7 @@ export interface NotebookEditToolInput {
   name: 'NotebookEdit';
   input: NotebookEditInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface ReadMcpResourceToolInput {
@@ -468,6 +210,7 @@ export interface ReadMcpResourceToolInput {
   name: 'ReadMcpResource';
   input: ReadMcpResourceInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface ReadToolInput {
@@ -476,6 +219,7 @@ export interface ReadToolInput {
   name: 'Read';
   input: FileReadInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface TaskToolInput {
@@ -484,6 +228,7 @@ export interface TaskToolInput {
   name: 'Task';
   input: AgentInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface TodoWriteToolInput {
@@ -492,6 +237,7 @@ export interface TodoWriteToolInput {
   name: 'TodoWrite';
   input: TodoWriteInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface WebFetchToolInput {
@@ -500,6 +246,7 @@ export interface WebFetchToolInput {
   name: 'WebFetch';
   input: WebFetchInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface WebSearchToolInput {
@@ -508,6 +255,7 @@ export interface WebSearchToolInput {
   name: 'WebSearch';
   input: WebSearchInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
 export interface WriteToolInput {
@@ -516,11 +264,87 @@ export interface WriteToolInput {
   name: 'Write';
   input: FileWriteInput;
   $result?: ToolResult;
+  metadata?: ToolMetadata;
+}
+
+export interface AskUserQuestionToolInput {
+  type: 'tool_use';
+  id: string;
+  name: 'AskUserQuestion';
+  input: AskUserQuestionInput;
+  $result?: ToolResult;
+  metadata?: ToolMetadata;
 }
 
+export interface ConfigToolInput {
+  type: 'tool_use';
+  id: string;
+  name: 'Config';
+  input: ConfigInput;
+  $result?: ToolResult;
+  metadata?: ToolMetadata;
+}
+
+export interface EnterWorktreeToolInput {
+  type: 'tool_use';
+  id: string;
+  name: 'EnterWorktree';
+  input: EnterWorktreeInput;
+  $result?: ToolResult;
+  metadata?: ToolMetadata;
+}
+
+// Sub-agent activity item for Agent tool display
+export interface SubAgentActivity {
+  type: 'tool_use' | 'text';
+  toolName?: string;
+  toolInput?: any;
+  toolResult?: any;
+  text?: string;
+}
+
+export interface AgentToolInput {
+  type: 'tool_use';
+  id: string;
+  name: 'Agent';
+  input: AgentInput;
+  $result?: ToolResult;
+  $subMessages?: SubAgentActivity[];
+  metadata?: ToolMetadata;
+}
+
+export interface EnterPlanModeToolInput {
+  type: 'tool_use';
+  id: string;
+  name: 'EnterPlanMode';
+  input: Record<string, unknown>;
+  $result?: ToolResult;
+  metadata?: ToolMetadata;
+}
+
+// ============================================================
+// Union of All Tool Input Types
+// ============================================================
+
 export type ToolInput =
-  | BashToolInput | BashOutputToolInput | EditToolInput | ExitPlanModeToolInput
-  | GlobToolInput | GrepToolInput | KillShellToolInput | ListMcpResourcesToolInput
+  | BashToolInput
+  | BashOutputToolInput
+  | EditToolInput
+  | ExitPlanModeToolInput
+  | GlobToolInput
+  | GrepToolInput
+  | TaskStopToolInput
+  | ListMcpResourcesToolInput
   | NotebookEditToolInput
-  | ReadMcpResourceToolInput | ReadToolInput | TaskToolInput | TodoWriteToolInput
-  | WebFetchToolInput | WebSearchToolInput | WriteToolInput;
+  | ReadMcpResourceToolInput
+  | ReadToolInput
+  | TaskToolInput
+  | TodoWriteToolInput
+  | WebFetchToolInput
+  | WebSearchToolInput
+  | WriteToolInput
+  | AskUserQuestionToolInput
+  | ConfigToolInput
+  | EnterWorktreeToolInput
+  | AgentToolInput
+  | EnterPlanModeToolInput;

package/shared/utils/message-formatter.ts

@@ -30,6 +30,7 @@ export function formatDatabaseMessage(
 		metadata: {
 			...buildMetadataFromDb(msg, overrides),
 			...(sdkMessage.metadata?.reasoning && { reasoning: true }),
+			...(sdkMessage.metadata?.interrupted && { interrupted: true }),
 		}
 	};
 }