@astralform/js 0.1.0 → 0.1.2

package/dist/index.d.ts

@@ -29,11 +29,14 @@ interface ToolUseStartEvent {
     description?: string;
     arguments: Record<string, unknown>;
     is_client_tool: boolean;
+    tool_category?: string;
+    icon_url?: string;
 }
 interface ToolUseEndEvent {
     type: "tool_use_end";
     call_id: string;
     tool: string;
+    result?: string;
 }
 interface AgentStartEvent {
     type: "agent_start";
@@ -98,6 +101,13 @@ interface CapsuleOutputEvent {
     command?: string;
     output: string;
     duration_ms?: number;
+    call_id?: string;
+}
+interface CapsuleOutputChunkEvent {
+    type: "capsule_output_chunk";
+    call_id: string;
+    stream: "stdout" | "stderr";
+    chunk: string;
 }
 interface TodoUpdateEvent {
     type: "todo_update";
@@ -128,13 +138,29 @@ interface AssetCreatedEvent {
     media_type: string;
     size_bytes: number;
 }
+interface ActivitySSEEvent {
+    type: "activity";
+    activity_id: string;
+    status: "started" | "completed";
+    category: string;
+    title: string;
+    detail?: string;
+    sources?: Array<{
+        title: string;
+        url: string;
+        snippet?: string;
+    }>;
+    tool_name?: string;
+    agent_name?: string;
+    duration_ms?: number;
+}
 interface RetryEvent {
     type: "retry";
     attempt: number;
     max_attempts: number;
     delay_seconds: number;
 }
-type SSEEvent = MessageStartEvent | ContentBlockDeltaEvent | ToolUseStartEvent | ToolUseEndEvent | AgentStartEvent | AgentEndEvent | SubagentStartEvent | SubagentContentDeltaEvent | SubagentUpdateEvent | SubagentEndEvent | SubagentToolUseEvent | ThinkingDeltaEvent | ThinkingCompleteEvent | SourcesEvent | CapsuleOutputEvent | TodoUpdateEvent | MessageStopEvent | AssetCreatedEvent | RetryEvent | SSEErrorEvent;
+type SSEEvent = MessageStartEvent | ContentBlockDeltaEvent | ToolUseStartEvent | ToolUseEndEvent | AgentStartEvent | AgentEndEvent | SubagentStartEvent | SubagentContentDeltaEvent | SubagentUpdateEvent | SubagentEndEvent | SubagentToolUseEvent | ThinkingDeltaEvent | ThinkingCompleteEvent | SourcesEvent | CapsuleOutputEvent | CapsuleOutputChunkEvent | TodoUpdateEvent | MessageStopEvent | AssetCreatedEvent | ActivitySSEEvent | RetryEvent | SSEErrorEvent;
 type ChatEvent = {
     type: "connected";
 } | {
@@ -154,6 +180,7 @@ type ChatEvent = {
     type: "tool_end";
     callId: string;
     toolName: string;
+    result?: string;
 } | {
     type: "agent_start";
     agentName: string;
@@ -202,6 +229,12 @@ type ChatEvent = {
     command?: string;
     output: string;
     durationMs?: number;
+    callId?: string;
+} | {
+    type: "capsule_output_chunk";
+    callId: string;
+    stream: "stdout" | "stderr";
+    chunk: string;
 } | {
     type: "todo_update";
     todos: TodoItem[];
@@ -229,6 +262,21 @@ type ChatEvent = {
     attempt: number;
     maxAttempts: number;
     delaySeconds: number;
+} | {
+    type: "activity";
+    activityId: string;
+    status: "started" | "completed";
+    category: string;
+    title: string;
+    detail?: string;
+    sources?: Array<{
+        title: string;
+        url: string;
+        snippet?: string;
+    }>;
+    toolName?: string;
+    agentName?: string;
+    durationMs?: number;
 } | {
     type: "model_info";
     name: string;
@@ -299,6 +347,7 @@ interface CapsuleOutput {
     command?: string;
     output: string;
     durationMs?: number;
+    callId?: string;
 }
 interface Source {
     title: string;
@@ -324,6 +373,7 @@ interface ChatStreamRequest {
     resend_from?: string;
     upload_ids?: string[];
     agent_name?: string;
+    enable_search?: boolean;
 }
 interface ToolResultRequest {
     conversation_id: string;
@@ -348,6 +398,8 @@ interface ToolCallRequest {
     description?: string;
     arguments: Record<string, unknown>;
     isClientTool: boolean;
+    toolCategory?: string;
+    iconUrl?: string;
 }
 interface StreamJobSSEOptions {
     url: string;
@@ -359,11 +411,17 @@ interface ChatStreamEvent {
     event: string;
     data: string;
 }
+interface ConversationEvent {
+    seq: number;
+    event: string;
+    data: Record<string, unknown>;
+}
 interface SendOptions {
     conversationId?: string;
     enabledClientTools?: string[];
     uploadIds?: string[];
     agentName?: string;
+    enableSearch?: boolean;
 }
 interface ConversationAsset {
     id: string;
@@ -400,6 +458,7 @@ declare class AstralformClient {
     deleteConversation(id: string): Promise<void>;
     getAgents(): Promise<AgentInfo[]>;
     getSkills(): Promise<SkillInfo[]>;
+    getConversationEvents(conversationId: string): Promise<ConversationEvent[]>;
     submitToolResult(request: ToolResultRequest): Promise<void>;
     private mapAsset;
     uploadFile(conversationId: string, file: Blob, filename?: string): Promise<ConversationAsset>;
@@ -478,17 +537,28 @@ declare class ChatSession {
     connect(): Promise<void>;
     send(content: string, options?: SendOptions): Promise<void>;
     resendFromCheckpoint(messageId: string, newContent: string): Promise<void>;
+    private resetStreamingState;
     private processStream;
     /** Last received sequence number for resumable reconnection */
     private lastSeq;
     /** Current job ID for cancellation */
     private currentJobId;
     private consumeJobStream;
+    /**
+     * Apply a single SSE event to session state and notify consumers.
+     * Shared between live streaming and historical event replay.
+     */
+    private applyEvent;
     private executeClientTools;
     private completeStream;
     disconnect(): void;
     createNewConversation(): Promise<string>;
     switchConversation(id: string): Promise<void>;
+    /**
+     * Replay a single persisted event to reconstruct session state.
+     * Skips text deltas (final content is already in messages[]).
+     */
+    private replayEvent;
     deleteConversation(id: string): Promise<void>;
     toggleClientTool(name: string): boolean;
 }
@@ -523,4 +593,4 @@ declare function generateId(): string;
  */
 declare function streamJobSSE(options: StreamJobSSEOptions): AsyncGenerator<ChatStreamEvent>;
 
-export { type AgentEndEvent, type AgentInfo, type AgentStartEvent, AstralformClient, type AstralformConfig, AstralformError, AuthenticationError, type CapsuleOutput, type CapsuleOutputEvent, type ChatEvent, ChatSession, type ChatStorage, type ChatStreamEvent, type ChatStreamRequest, ConnectionError, type ContentBlockDeltaEvent, type Conversation, type ConversationAsset, InMemoryStorage, type JobCreateResponse, LLMNotConfiguredError, type Message, type MessageStartEvent, type MessageStopEvent, type ProjectStatus, RateLimitError, type RetryEvent, type SSEErrorEvent, type SSEEvent, type SendOptions, ServerError, type SkillInfo, type Source, type SourcesEvent, StreamAbortedError, type StreamJobSSEOptions, type SubagentContentDeltaEvent, type SubagentEndEvent, type SubagentStartEvent, type SubagentState, type SubagentToolUseEvent, type ThinkingCompleteEvent, type ThinkingDeltaEvent, type TodoItem, type TodoUpdateEvent, type ToolCallRequest, type ToolDefinition, type ToolHandler, ToolRegistry, type ToolResult, type ToolResultRequest, type ToolState, type ToolUseEndEvent, type ToolUseStartEvent, generateId, streamJobSSE };
+export { type AgentEndEvent, type AgentInfo, type AgentStartEvent, AstralformClient, type AstralformConfig, AstralformError, AuthenticationError, type CapsuleOutput, type CapsuleOutputChunkEvent, type CapsuleOutputEvent, type ChatEvent, ChatSession, type ChatStorage, type ChatStreamEvent, type ChatStreamRequest, ConnectionError, type ContentBlockDeltaEvent, type Conversation, type ConversationAsset, type ConversationEvent, InMemoryStorage, type JobCreateResponse, LLMNotConfiguredError, type Message, type MessageStartEvent, type MessageStopEvent, type ProjectStatus, RateLimitError, type RetryEvent, type SSEErrorEvent, type SSEEvent, type SendOptions, ServerError, type SkillInfo, type Source, type SourcesEvent, StreamAbortedError, type StreamJobSSEOptions, type SubagentContentDeltaEvent, type SubagentEndEvent, type SubagentStartEvent, type SubagentState, type SubagentToolUseEvent, type ThinkingCompleteEvent, type ThinkingDeltaEvent, type TodoItem, type TodoUpdateEvent, type ToolCallRequest, type ToolDefinition, type ToolHandler, ToolRegistry, type ToolResult, type ToolResultRequest, type ToolState, type ToolUseEndEvent, type ToolUseStartEvent, generateId, streamJobSSE };

package/dist/index.js

@@ -247,6 +247,11 @@ var AstralformClient = class {
       isEnabled: s.is_enabled
     }));
   }
+  async getConversationEvents(conversationId) {
+    return this.get(
+      `/v1/conversations/${encodeURIComponent(conversationId)}/events`
+    );
+  }
   async submitToolResult(request) {
     await this.post("/v1/tool-result", request);
   }
@@ -569,7 +574,8 @@ var ChatSession = class {
         options?.enabledClientTools ?? this.enabledClientTools
       ),
       upload_ids: options?.uploadIds,
-      agent_name: options?.agentName
+      agent_name: options?.agentName,
+      enable_search: options?.enableSearch
     };
     await this.processStream(request);
   }
@@ -584,8 +590,7 @@ var ChatSession = class {
     };
     await this.processStream(request);
   }
-  async processStream(request) {
-    this.isStreaming = true;
+  resetStreamingState() {
     this.streamingContent = "";
     this.thinkingContent = "";
     this.isThinking = false;
@@ -594,6 +599,10 @@ var ChatSession = class {
     this.capsuleOutputs = [];
     this.todos = [];
     this.activeTools.clear();
+  }
+  async processStream(request) {
+    this.isStreaming = true;
+    this.resetStreamingState();
     this.abortController = new AbortController();
     try {
       await this.consumeJobStream(request);
@@ -653,39 +662,26 @@ var ChatSession = class {
               name: parsed.model_display_name
             });
           }
-          if (parsed.agent_name) {
-            this.emit({
-              type: "agent_start",
-              agentName: parsed.agent_name,
-              agentDisplayName: parsed.agent_display_name
-            });
-          }
           break;
         case "content_block_delta":
           this.streamingContent += parsed.delta.text;
           this.emit({ type: "chunk", text: parsed.delta.text });
           break;
         case "tool_use_start": {
-          const toolCall = {
-            callId: parsed.call_id,
-            toolName: parsed.tool,
-            displayName: parsed.display_name,
-            description: parsed.description,
-            arguments: parsed.arguments,
-            isClientTool: parsed.is_client_tool
-          };
-          this.activeTools.set(parsed.call_id, {
-            toolName: parsed.tool,
-            displayName: parsed.display_name,
-            description: parsed.description,
-            arguments: parsed.arguments,
-            callId: parsed.call_id,
-            status: parsed.is_client_tool ? "calling" : "executing",
-            isClientTool: parsed.is_client_tool
-          });
-          this.emit({ type: "tool_call", request: toolCall });
+          this.applyEvent(parsed);
           if (parsed.is_client_tool) {
-            const results = await this.executeClientTools([toolCall]);
+            const results = await this.executeClientTools([
+              {
+                callId: parsed.call_id,
+                toolName: parsed.tool,
+                displayName: parsed.display_name,
+                description: parsed.description,
+                arguments: parsed.arguments,
+                isClientTool: parsed.is_client_tool,
+                toolCategory: parsed.tool_category,
+                iconUrl: parsed.icon_url
+              }
+            ]);
             await this.client.submitToolResult({
               conversation_id: conversationId,
               message_id: messageId,
@@ -694,61 +690,6 @@ var ChatSession = class {
           }
           break;
         }
-        case "tool_use_end": {
-          const toolState = this.activeTools.get(parsed.call_id);
-          if (toolState) {
-            toolState.status = "completed";
-          }
-          this.emit({
-            type: "tool_end",
-            callId: parsed.call_id,
-            toolName: parsed.tool
-          });
-          break;
-        }
-        case "agent_start":
-          this.emit({
-            type: "agent_start",
-            agentName: parsed.agent_name,
-            agentDisplayName: parsed.agent_display_name,
-            avatarUrl: parsed.avatar_url
-          });
-          break;
-        case "agent_end":
-          this.emit({ type: "agent_end", agentName: parsed.agent_name });
-          break;
-        case "subagent_start":
-          this.activeSubagents.set(parsed.tool_call_id, {
-            agentName: parsed.agent_name,
-            displayName: parsed.display_name,
-            avatarUrl: parsed.avatar_url,
-            description: parsed.description,
-            content: "",
-            isActive: true
-          });
-          this.emit({
-            type: "subagent_start",
-            agentName: parsed.agent_name,
-            displayName: parsed.display_name,
-            toolCallId: parsed.tool_call_id,
-            avatarUrl: parsed.avatar_url,
-            description: parsed.description
-          });
-          break;
-        case "subagent_update": {
-          const sub = this.activeSubagents.get(parsed.tool_call_id);
-          if (sub) {
-            sub.agentName = parsed.agent_name;
-            sub.displayName = parsed.display_name;
-          }
-          this.emit({
-            type: "subagent_update",
-            agentName: parsed.agent_name,
-            displayName: parsed.display_name,
-            toolCallId: parsed.tool_call_id
-          });
-          break;
-        }
         case "subagent_content_delta": {
           const subagent = this.activeSubagents.get(parsed.tool_call_id);
           if (subagent) {
@@ -762,19 +703,6 @@ var ChatSession = class {
           });
           break;
         }
-        case "subagent_end": {
-          const sub = this.activeSubagents.get(parsed.tool_call_id);
-          if (sub) {
-            sub.isActive = false;
-          }
-          this.emit({
-            type: "subagent_end",
-            agentName: parsed.agent_name,
-            displayName: parsed.display_name,
-            toolCallId: parsed.tool_call_id
-          });
-          break;
-        }
         case "thinking_delta":
           this.thinkingContent += parsed.delta.text;
           this.isThinking = true;
@@ -784,45 +712,6 @@ var ChatSession = class {
           this.isThinking = false;
           this.emit({ type: "thinking_complete" });
           break;
-        case "sources":
-          this.sources.push(...parsed.sources);
-          this.emit({ type: "sources", sources: parsed.sources });
-          break;
-        case "capsule_output": {
-          const capsule = {
-            toolName: parsed.tool_name,
-            agentName: parsed.agent_name,
-            command: parsed.command,
-            output: parsed.output,
-            durationMs: parsed.duration_ms
-          };
-          this.capsuleOutputs.push(capsule);
-          this.emit({ type: "capsule_output", ...capsule });
-          break;
-        }
-        case "todo_update":
-          this.todos = parsed.todos;
-          this.emit({ type: "todo_update", todos: parsed.todos });
-          break;
-        case "subagent_tool_use":
-          this.emit({
-            type: "subagent_tool_use",
-            agentName: parsed.agent_name,
-            toolName: parsed.tool,
-            toolCallId: parsed.tool_call_id,
-            result: parsed.result
-          });
-          break;
-        case "asset_created":
-          this.emit({
-            type: "asset_created",
-            assetId: parsed.asset_id,
-            name: parsed.name,
-            url: parsed.url,
-            mediaType: parsed.media_type,
-            sizeBytes: parsed.size_bytes
-          });
-          break;
         case "retry":
           this.emit({
             type: "retry",
@@ -840,11 +729,170 @@ var ChatSession = class {
             error: new AstralformError(parsed.message, parsed.code)
           });
           break;
+        default:
+          this.applyEvent(parsed);
       }
     }
     this.currentJobId = null;
     await this.completeStream(conversationId, messageId, stopTitle);
   }
+  /**
+   * Apply a single SSE event to session state and notify consumers.
+   * Shared between live streaming and historical event replay.
+   */
+  applyEvent(event) {
+    switch (event.type) {
+      case "tool_use_start": {
+        const request = {
+          callId: event.call_id,
+          toolName: event.tool,
+          displayName: event.display_name,
+          description: event.description,
+          arguments: event.arguments,
+          isClientTool: event.is_client_tool,
+          toolCategory: event.tool_category,
+          iconUrl: event.icon_url
+        };
+        this.activeTools.set(event.call_id, {
+          ...request,
+          status: event.is_client_tool ? "calling" : "executing"
+        });
+        this.emit({ type: "tool_call", request });
+        break;
+      }
+      case "tool_use_end": {
+        const toolState = this.activeTools.get(event.call_id);
+        if (toolState) {
+          toolState.status = "completed";
+        }
+        this.emit({
+          type: "tool_end",
+          callId: event.call_id,
+          toolName: event.tool,
+          result: event.result
+        });
+        break;
+      }
+      case "agent_start":
+        this.emit({
+          type: "agent_start",
+          agentName: event.agent_name,
+          agentDisplayName: event.agent_display_name,
+          avatarUrl: event.avatar_url
+        });
+        break;
+      case "agent_end":
+        this.emit({ type: "agent_end", agentName: event.agent_name });
+        break;
+      case "subagent_start":
+        this.activeSubagents.set(event.tool_call_id, {
+          agentName: event.agent_name,
+          displayName: event.display_name,
+          avatarUrl: event.avatar_url,
+          description: event.description,
+          content: "",
+          isActive: true
+        });
+        this.emit({
+          type: "subagent_start",
+          agentName: event.agent_name,
+          displayName: event.display_name,
+          toolCallId: event.tool_call_id,
+          avatarUrl: event.avatar_url,
+          description: event.description
+        });
+        break;
+      case "subagent_update": {
+        const sub = this.activeSubagents.get(event.tool_call_id);
+        if (sub) {
+          sub.agentName = event.agent_name;
+          sub.displayName = event.display_name;
+        }
+        this.emit({
+          type: "subagent_update",
+          agentName: event.agent_name,
+          displayName: event.display_name,
+          toolCallId: event.tool_call_id
+        });
+        break;
+      }
+      case "subagent_end": {
+        const sub = this.activeSubagents.get(event.tool_call_id);
+        if (sub) {
+          sub.isActive = false;
+        }
+        this.emit({
+          type: "subagent_end",
+          agentName: event.agent_name,
+          displayName: event.display_name,
+          toolCallId: event.tool_call_id
+        });
+        break;
+      }
+      case "subagent_tool_use":
+        this.emit({
+          type: "subagent_tool_use",
+          agentName: event.agent_name,
+          toolName: event.tool,
+          toolCallId: event.tool_call_id,
+          result: event.result
+        });
+        break;
+      case "sources":
+        this.sources.push(...event.sources);
+        this.emit({ type: "sources", sources: event.sources });
+        break;
+      case "capsule_output": {
+        const capsule = {
+          toolName: event.tool_name,
+          agentName: event.agent_name,
+          command: event.command,
+          output: event.output,
+          durationMs: event.duration_ms,
+          callId: event.call_id
+        };
+        this.capsuleOutputs.push(capsule);
+        this.emit({ type: "capsule_output", ...capsule });
+        break;
+      }
+      case "capsule_output_chunk":
+        this.emit({
+          type: "capsule_output_chunk",
+          callId: event.call_id,
+          stream: event.stream,
+          chunk: event.chunk
+        });
+        break;
+      case "todo_update":
+        this.todos = event.todos;
+        this.emit({ type: "todo_update", todos: event.todos });
+        break;
+      case "asset_created":
+        this.emit({
+          type: "asset_created",
+          assetId: event.asset_id,
+          name: event.name,
+          url: event.url,
+          mediaType: event.media_type,
+          sizeBytes: event.size_bytes
+        });
+        break;
+      case "activity":
+        this.emit({
+          type: "activity",
+          activityId: event.activity_id,
+          status: event.status,
+          category: event.category,
+          title: event.title,
+          detail: event.detail,
+          sources: event.sources,
+          toolName: event.tool_name,
+          agentName: event.agent_name,
+          durationMs: event.duration_ms
+        });
+        break;
+    }
+  }
   async executeClientTools(toolCalls) {
     const results = [];
     for (const call of toolCalls) {
@@ -921,12 +969,27 @@ var ChatSession = class {
   }
   async switchConversation(id) {
     this.conversationId = id;
-    try {
-      this.messages = await this.client.getMessages(id);
-    } catch {
-      this.messages = await this.storage.fetchMessages(id);
+    this.resetStreamingState();
+    const [messagesResult, eventsResult] = await Promise.allSettled([
+      this.client.getMessages(id).catch(() => this.storage.fetchMessages(id)),
+      this.client.getConversationEvents(id)
+    ]);
+    this.messages = messagesResult.status === "fulfilled" ? messagesResult.value : [];
+    if (eventsResult.status === "fulfilled") {
+      for (const ev of eventsResult.value) {
+        this.replayEvent(ev.event, ev.data);
+      }
     }
-    this.streamingContent = "";
+  }
+  /**
+   * Replay a single persisted event to reconstruct session state.
+   * Skips text deltas (final content is already in messages[]).
+   */
+  replayEvent(eventType, data) {
+    if (eventType === "content_block_delta" || eventType === "thinking_delta" || eventType === "subagent_content_delta" || eventType === "thinking_complete") {
+      return;
+    }
+    this.applyEvent({ type: eventType, ...data });
   }
   async deleteConversation(id) {
     try {