diff --git a/src/types/index.ts b/src/types/index.ts
index 60d3460..8bd9b18 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -27,6 +27,31 @@ export interface Message {
   role: 'user' | 'assistant';
   content: string;
   timestamp: Date;
+  /** 工具调用记录 */
+  toolCalls?: ToolCall[];
+  /** 工具调用结果 */
+  toolResults?: ToolResult[];
+}
+
+// 工具调用记录
+export interface ToolCall {
+  id: string;
+  name: string;
+  input: Record<string, unknown>;
+}
+
+// 工具调用结果
+export interface ToolResult {
+  toolUseId: string;
+  toolName: string;
+  content: string;
+  isError?: boolean;
+  /** 代码执行产生的图片（Base64） */
+  images?: string[];
+  /** 执行引擎 */
+  engine?: 'pyodide' | 'piston';
+  /** 执行时间 (ms) */
+  executionTime?: number;
 }
 
 // 用户类型