bashkit 0.1.1 → 0.1.2

package/AGENTS.md

@@ -122,7 +122,7 @@ await sandbox.destroy();
 
 ## Sub-agents with Task Tool
 
-The Task tool spawns new `generateText` calls for complex subtasks:
+The Task tool spawns new agents for complex subtasks:
 
 ```typescript
 import { createTaskTool } from "bashkit";
@@ -157,6 +157,40 @@ The parent agent calls Task like any other tool:
 }}
 ```
 
+### Streaming Sub-agent Activity to UI
+
+Pass a `streamWriter` to stream real-time sub-agent activity:
+
+```typescript
+import { createUIMessageStream } from "ai";
+
+const stream = createUIMessageStream({
+  execute: async ({ writer }) => {
+    const taskTool = createTaskTool({
+      model,
+      tools: sandboxTools,
+      streamWriter: writer, // Enable real-time streaming
+      subagentTypes: { ... },
+    });
+
+    const result = streamText({
+      model,
+      tools: { Task: taskTool },
+      ...
+    });
+
+    writer.merge(result.toUIMessageStream());
+  },
+});
+```
+
+When `streamWriter` is provided:
+- Uses `streamText` internally (instead of `generateText`)
+- Emits `data-subagent` events: `start`, `tool-call`, `done`, `complete`
+- Events appear in `message.parts` as `{ type: "data-subagent", data: SubagentEventData }`
+
+**Note:** TaskOutput does NOT include messages (to avoid context bloat). The UI accesses the full conversation via the streamed `complete` event.
+
 ## Prompt Caching
 
 Enable Anthropic prompt caching to reduce costs on repeated prefixes:

package/README.md

@@ -219,7 +219,7 @@ const { tools, planModeState } = createAgentTools(sandbox, {
 
 ## Sub-agents with Task Tool
 
-The Task tool spawns new `generateText` calls for complex subtasks:
+The Task tool spawns new agents for complex subtasks:
 
 ```typescript
 import { createTaskTool } from 'bashkit';
@@ -254,6 +254,46 @@ The parent agent calls Task like any other tool:
 }}
 ```
 
+### Streaming Sub-agent Activity to UI
+
+Pass a `streamWriter` to stream real-time sub-agent activity to the UI:
+
+```typescript
+import { createUIMessageStream } from 'ai';
+
+const stream = createUIMessageStream({
+  execute: async ({ writer }) => {
+    const taskTool = createTaskTool({
+      model,
+      tools: sandboxTools,
+      streamWriter: writer, // Enable real-time streaming
+      subagentTypes: { ... },
+    });
+
+    // Use with streamText
+    const result = streamText({
+      model,
+      tools: { Task: taskTool },
+      ...
+    });
+
+    writer.merge(result.toUIMessageStream());
+  },
+});
+```
+
+When `streamWriter` is provided:
+- Uses `streamText` internally (instead of `generateText`)
+- Emits `data-subagent` events to the UI stream:
+  - `start` - Sub-agent begins work
+  - `tool-call` - Each tool the sub-agent uses (with args)
+  - `done` - Sub-agent finished
+  - `complete` - Full messages array for UI access
+
+These appear in `message.parts` on the client as `{ type: "data-subagent", data: SubagentEventData }`.
+
+**Important:** The TaskOutput returned to the lead agent does NOT include messages (to avoid context bloat). The UI accesses the full conversation via the streamed `complete` event.
+
 ## Context Management
 
 ### Conversation Compaction

package/dist/index.d.ts

@@ -1,8 +1,9 @@
+export type { UIMessageStreamWriter } from "ai";
 export { anthropicPromptCacheMiddleware } from "./middleware";
 export type { E2BSandboxConfig, LocalSandboxConfig, VercelSandboxConfig, } from "./sandbox";
 export { createE2BSandbox, createLocalSandbox, createVercelSandbox, } from "./sandbox";
 export type { ExecOptions, ExecResult, Sandbox } from "./sandbox/interface";
-export type { AgentToolsResult, AskUserError, AskUserOutput, AskUserResponseHandler, BashError, BashOutput, EditError, EditOutput, EnterPlanModeError, EnterPlanModeOutput, ExitPlanModeError, ExitPlanModeOutput, PlanModeState, GlobError, GlobOutput, GrepContentOutput, GrepCountOutput, GrepError, GrepFilesOutput, GrepMatch, GrepOutput, ReadDirectoryOutput, ReadError, ReadOutput, ReadTextOutput, SkillError, SkillOutput, SkillToolConfig, SubagentStepEvent, SubagentTypeConfig, TaskError, TaskOutput, TaskToolConfig, TodoItem, TodoState, TodoWriteError, TodoWriteOutput, WebFetchError, WebFetchOutput, WebFetchToolConfig, WebSearchError, WebSearchOutput, WebSearchResult, WebSearchToolConfig, WriteError, WriteOutput, } from "./tools";
+export type { AgentToolsResult, AskUserError, AskUserOutput, AskUserResponseHandler, BashError, BashOutput, EditError, EditOutput, EnterPlanModeError, EnterPlanModeOutput, ExitPlanModeError, ExitPlanModeOutput, PlanModeState, GlobError, GlobOutput, GrepContentOutput, GrepCountOutput, GrepError, GrepFilesOutput, GrepMatch, GrepOutput, ReadDirectoryOutput, ReadError, ReadOutput, ReadTextOutput, SkillError, SkillOutput, SkillToolConfig, SubagentEventData, SubagentStepEvent, SubagentTypeConfig, TaskError, TaskOutput, TaskToolConfig, TodoItem, TodoState, TodoWriteError, TodoWriteOutput, WebFetchError, WebFetchOutput, WebFetchToolConfig, WebSearchError, WebSearchOutput, WebSearchResult, WebSearchToolConfig, WriteError, WriteOutput, } from "./tools";
 export { createAgentTools, createAskUserTool, createBashTool, createEditTool, createEnterPlanModeTool, createExitPlanModeTool, createGlobTool, createGrepTool, createReadTool, createSkillTool, createTaskTool, createTodoWriteTool, createWebFetchTool, createWebSearchTool, createWriteTool, } from "./tools";
 export type { AgentConfig, AskUserConfig, SkillConfig, ToolConfig, WebFetchConfig, WebSearchConfig, } from "./types";
 export { DEFAULT_CONFIG } from "./types";

package/dist/index.js

@@ -1139,6 +1139,7 @@ function createWriteTool(sandbox, config) {
 // src/tools/task.ts
 import {
   generateText as generateText2,
+  streamText,
   stepCountIs,
   tool as tool13,
   zodSchema as zodSchema13
@@ -1149,6 +1150,10 @@ var taskInputSchema = z13.object({
   prompt: z13.string().describe("The task for the agent to perform"),
   subagent_type: z13.string().describe("The type of specialized agent to use for this task")
 });
+var eventCounter = 0;
+function generateEventId() {
+  return `subagent-${Date.now()}-${++eventCounter}`;
+}
 function filterTools(allTools, allowedTools) {
   if (!allowedTools)
     return allTools;
@@ -1165,10 +1170,9 @@ function createTaskTool(config) {
     model: defaultModel,
     tools: allTools,
     subagentTypes = {},
-    costPerInputToken = 0.000003,
-    costPerOutputToken = 0.000015,
     defaultStopWhen,
-    defaultOnStepFinish
+    defaultOnStepFinish,
+    streamWriter
   } = config;
   return tool13({
     description: "Launches a new agent to handle complex, multi-step tasks autonomously. Use this for tasks that require multiple steps, research, or specialized expertise.",
@@ -1184,13 +1188,88 @@ function createTaskTool(config) {
         const model = typeConfig.model || defaultModel;
         const tools = filterTools(allTools, typeConfig.tools);
         const systemPrompt = typeConfig.systemPrompt;
-        const result = await generateText2({
+        const commonOptions = {
           model,
           tools,
           system: systemPrompt,
           prompt,
           stopWhen: typeConfig.stopWhen ?? defaultStopWhen ?? stepCountIs(15),
-          prepareStep: typeConfig.prepareStep,
+          prepareStep: typeConfig.prepareStep
+        };
+        if (streamWriter) {
+          const startId = generateEventId();
+          streamWriter.write({
+            type: "data-subagent",
+            id: startId,
+            data: {
+              event: "start",
+              subagent: subagent_type,
+              description
+            }
+          });
+          const result2 = streamText({
+            ...commonOptions,
+            onStepFinish: async (step) => {
+              if (step.toolCalls?.length) {
+                for (const tc of step.toolCalls) {
+                  const eventId = generateEventId();
+                  streamWriter.write({
+                    type: "data-subagent",
+                    id: eventId,
+                    data: {
+                      event: "tool-call",
+                      subagent: subagent_type,
+                      description,
+                      toolName: tc.toolName,
+                      args: tc.input
+                    }
+                  });
+                }
+              }
+              await typeConfig.onStepFinish?.(step);
+              await defaultOnStepFinish?.({
+                subagentType: subagent_type,
+                description,
+                step
+              });
+            }
+          });
+          const text = await result2.text;
+          const usage2 = await result2.usage;
+          const response = await result2.response;
+          streamWriter.write({
+            type: "data-subagent",
+            id: generateEventId(),
+            data: {
+              event: "done",
+              subagent: subagent_type,
+              description
+            }
+          });
+          streamWriter.write({
+            type: "data-subagent",
+            id: generateEventId(),
+            data: {
+              event: "complete",
+              subagent: subagent_type,
+              description,
+              messages: response.messages
+            }
+          });
+          const durationMs2 = Math.round(performance.now() - startTime);
+          return {
+            result: text,
+            usage: usage2.inputTokens !== undefined && usage2.outputTokens !== undefined ? {
+              input_tokens: usage2.inputTokens,
+              output_tokens: usage2.outputTokens
+            } : undefined,
+            duration_ms: durationMs2,
+            subagent: subagent_type,
+            description
+          };
+        }
+        const result = await generateText2({
+          ...commonOptions,
           onStepFinish: async (step) => {
             await typeConfig.onStepFinish?.(step);
             await defaultOnStepFinish?.({
@@ -1205,20 +1284,16 @@ function createTaskTool(config) {
           input_tokens: result.usage.inputTokens,
           output_tokens: result.usage.outputTokens
         } : undefined;
-        let totalCostUsd;
-        if (usage) {
-          totalCostUsd = usage.input_tokens * costPerInputToken + usage.output_tokens * costPerOutputToken;
-        }
         return {
           result: result.text,
           usage,
-          total_cost_usd: totalCostUsd,
-          duration_ms: durationMs
+          duration_ms: durationMs,
+          subagent: subagent_type,
+          description
         };
       } catch (error) {
-        return {
-          error: error instanceof Error ? error.message : "Unknown error"
-        };
+        const errorMessage = error instanceof Error ? error.message : "Unknown error";
+        return { error: errorMessage };
       }
     }
   });

package/dist/tools/index.d.ts

@@ -65,7 +65,7 @@ export type { ReadDirectoryOutput, ReadError, ReadOutput, ReadTextOutput, } from
 export { createReadTool } from "./read";
 export type { SkillError, SkillOutput, SkillToolConfig } from "./skill";
 export { createSkillTool } from "./skill";
-export type { SubagentStepEvent, SubagentTypeConfig, TaskError, TaskOutput, TaskToolConfig, } from "./task";
+export type { SubagentEventData, SubagentStepEvent, SubagentTypeConfig, TaskError, TaskOutput, TaskToolConfig, } from "./task";
 export { createTaskTool } from "./task";
 export type { TodoItem, TodoState, TodoWriteError, TodoWriteOutput, } from "./todo-write";
 export { createTodoWriteTool } from "./todo-write";

package/dist/tools/task.d.ts

@@ -1,12 +1,13 @@
-import { type LanguageModel, type PrepareStepFunction, type StepResult, type StopCondition, type Tool, type ToolSet } from "ai";
+import { type ModelMessage, type LanguageModel, type PrepareStepFunction, type StepResult, type StopCondition, type UIMessageStreamWriter, type Tool, type ToolSet } from "ai";
 export interface TaskOutput {
     result: string;
     usage?: {
         input_tokens: number;
         output_tokens: number;
     };
-    total_cost_usd?: number;
     duration_ms?: number;
+    subagent?: string;
+    description?: string;
 }
 export interface TaskError {
     error: string;
@@ -17,6 +18,15 @@ export interface SubagentStepEvent {
     description: string;
     step: StepResult<ToolSet>;
 }
+/** Data format for streamed subagent events (appears in message.parts as type: "data-subagent") */
+export interface SubagentEventData {
+    event: "start" | "tool-call" | "done" | "complete";
+    subagent: string;
+    description: string;
+    toolName?: string;
+    args?: Record<string, unknown>;
+    messages?: ModelMessage[];
+}
 export interface SubagentTypeConfig {
     /** Model to use for this subagent type */
     model?: LanguageModel;
@@ -38,13 +48,11 @@ export interface TaskToolConfig {
     tools: ToolSet;
     /** Configuration for each subagent type */
     subagentTypes?: Record<string, SubagentTypeConfig>;
-    /** Cost per input token for usage tracking */
-    costPerInputToken?: number;
-    /** Cost per output token for usage tracking */
-    costPerOutputToken?: number;
     /** Default stop condition for subagents (default: stepCountIs(15)) */
     defaultStopWhen?: StopCondition<ToolSet>;
     /** Default callback for each step any subagent takes */
     defaultOnStepFinish?: (event: SubagentStepEvent) => void | Promise<void>;
+    /** Optional stream writer for real-time subagent activity (uses streamText instead of generateText) */
+    streamWriter?: UIMessageStreamWriter;
 }
 export declare function createTaskTool(config: TaskToolConfig): Tool;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bashkit",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Agentic coding tools for the Vercel AI SDK",
   "type": "module",
   "main": "./dist/index.js",

package/dist/tools/lsp.d.ts

@@ -1,94 +0,0 @@
-/**
- * A diagnostic (error/warning) from the language server.
- */
-export interface LspDiagnostic {
-    file: string;
-    line: number;
-    column: number;
-    endLine?: number;
-    endColumn?: number;
-    severity: "error" | "warning" | "info" | "hint";
-    message: string;
-    source?: string;
-    code?: string | number;
-}
-/**
- * A location in the codebase (for go-to-definition, references, etc.)
- */
-export interface LspLocation {
-    file: string;
-    line: number;
-    column: number;
-    endLine?: number;
-    endColumn?: number;
-    preview?: string;
-}
-/**
- * Hover information for a symbol.
- */
-export interface LspHoverInfo {
-    contents: string;
-    range?: {
-        startLine: number;
-        startColumn: number;
-        endLine: number;
-        endColumn: number;
-    };
-}
-/**
- * Interface that LSP providers must implement.
- * This abstracts over different LSP implementations (vscode, standalone servers, etc.)
- */
-export interface LspProvider {
-    /** Get diagnostics for a file or the entire workspace */
-    getDiagnostics(file?: string): Promise<LspDiagnostic[]>;
-    /** Go to definition of symbol at position */
-    getDefinition(file: string, line: number, column: number): Promise<LspLocation[]>;
-    /** Find all references to symbol at position */
-    getReferences(file: string, line: number, column: number): Promise<LspLocation[]>;
-    /** Get hover information for symbol at position */
-    getHover(file: string, line: number, column: number): Promise<LspHoverInfo | null>;
-    /** Get available actions at position (optional) */
-    getCodeActions?(file: string, line: number, column: number): Promise<Array<{
-        title: string;
-        kind?: string;
-    }>>;
-}
-export interface LspDiagnosticsOutput {
-    type: "diagnostics";
-    diagnostics: LspDiagnostic[];
-    counts: {
-        errors: number;
-        warnings: number;
-        info: number;
-        hints: number;
-    };
-}
-export interface LspDefinitionOutput {
-    type: "definition";
-    locations: LspLocation[];
-}
-export interface LspReferencesOutput {
-    type: "references";
-    locations: LspLocation[];
-    count: number;
-}
-export interface LspHoverOutput {
-    type: "hover";
-    info: LspHoverInfo | null;
-}
-export interface LspError {
-    error: string;
-}
-export type LspOutput = LspDiagnosticsOutput | LspDefinitionOutput | LspReferencesOutput | LspHoverOutput | LspError;
-/**
- * Creates an LSP tool for code intelligence operations.
- *
- * @param provider - An implementation of LspProvider that interfaces with language servers
- */
-export declare function createLspTool(provider: LspProvider): import("ai").Tool<{
-    action: "diagnostics" | "definition" | "references" | "hover";
-    file?: string | undefined;
-    line?: number | undefined;
-    column?: number | undefined;
-}, LspOutput>;