bashkit 0.5.0 → 0.5.2

package/dist/tools/index.d.ts

@@ -1,7 +1,7 @@
 import type { ToolSet } from "ai";
 import type { Sandbox } from "../sandbox/interface";
 import type { AgentConfig } from "../types";
-import { type BudgetTracker } from "../utils/budget-tracking";
+import { type BudgetTracker, type ModelInfo } from "../utils/budget-tracking";
 import { type PlanModeState } from "./enter-plan-mode";
 /**
  * Result from createAgentTools including tools and optional shared state.
@@ -13,6 +13,8 @@ export interface AgentToolsResult {
     planModeState?: PlanModeState;
     /** Budget tracker (only present when budget config is set) */
     budget?: BudgetTracker;
+    /** Model info from OpenRouter (only present when modelRegistry or budget pricingProvider is configured) */
+    openRouterModels?: Map<string, ModelInfo>;
 }
 /**
  * Creates agent tools for AI SDK's generateText/streamText.

package/dist/tools/task.d.ts

@@ -13,6 +13,9 @@ export interface TaskOutput {
 }
 export interface TaskError {
     error: string;
+    subagent?: string;
+    description?: string;
+    duration_ms?: number;
 }
 declare const taskInputSchema: z.ZodObject<{
     description: z.ZodString;

package/dist/tools/web-constants.d.ts

@@ -0,0 +1,5 @@
+/**
+ * HTTP status codes that indicate a retryable error.
+ * Shared between web-search and web-fetch tools.
+ */
+export declare const RETRYABLE_STATUS_CODES: number[];

package/dist/types.d.ts

@@ -94,16 +94,37 @@ export type CacheConfig = boolean | CacheStore | {
  * Supported pricing providers for automatic model cost lookup.
  */
 export type PricingProvider = "openRouter";
+/**
+ * Supported model registry providers.
+ * Add new providers here as union types (e.g., 'openRouter' | 'together')
+ */
+export type ModelRegistryProvider = "openRouter";
+/**
+ * Configuration for fetching model info (pricing + context lengths) from a provider.
+ * Used by budget tracking and compaction.
+ */
+export type ModelRegistryConfig = {
+    /** Provider to fetch model info from */
+    provider: ModelRegistryProvider;
+    /** API key for the provider (optional, passed as bearer token) */
+    apiKey?: string;
+};
 /**
  * Budget tracking configuration.
- * At least one of `pricingProvider` or `modelPricing` must be provided.
+ * At least one of `pricingProvider`, `modelPricing`, or top-level `modelRegistry` must be provided.
  */
 export type BudgetConfig = {
     /** Maximum budget in USD (must be positive) */
     maxUsd: number;
-    /** Pricing provider for automatic model cost lookup. Omit to skip fetching. */
+    /**
+     * Pricing provider for automatic model cost lookup. Omit to skip fetching.
+     * @deprecated Use top-level `modelRegistry` instead.
+     */
     pricingProvider?: PricingProvider;
-    /** API key for the pricing provider. Passed as bearer token. */
+    /**
+     * API key for the pricing provider. Passed as bearer token.
+     * @deprecated Use top-level `modelRegistry` instead.
+     */
     apiKey?: string;
     /** Per-model pricing overrides (always highest priority over provider data) */
     modelPricing?: Record<string, ModelPricing>;
@@ -129,6 +150,9 @@ export type AgentConfig = {
     webFetch?: WebFetchConfig;
     /** Enable tool result caching */
     cache?: CacheConfig;
+    /** Fetch model info (pricing + context lengths) from a provider.
+     *  Used by budget tracking and compaction. */
+    modelRegistry?: ModelRegistryConfig;
     /** Budget tracking configuration */
     budget?: BudgetConfig;
     defaultTimeout?: number;

package/dist/utils/budget-tracking.d.ts

@@ -12,6 +12,10 @@ export interface ModelPricing {
     cacheReadPerToken?: number;
     cacheWritePerToken?: number;
 }
+export interface ModelInfo {
+    pricing: ModelPricing;
+    contextLength: number;
+}
 export interface BudgetStatus {
     totalCostUsd: number;
     maxUsd: number;
@@ -37,6 +41,14 @@ export interface BudgetTracker {
  * On failure, throws an error (callers decide whether to rethrow or fall back).
  */
 export declare function fetchOpenRouterPricing(apiKey?: string): Promise<Map<string, ModelPricing>>;
+/**
+ * Fetches model info (pricing + context length) from OpenRouter's public API.
+ * Results are cached at module level. Concurrent calls are deduplicated.
+ * Shares the same fetch and cache as `fetchOpenRouterPricing`.
+ *
+ * On failure, throws an error (callers decide whether to rethrow or fall back).
+ */
+export declare function fetchOpenRouterModels(apiKey?: string): Promise<Map<string, ModelInfo>>;
 /**
  * Reset the OpenRouter cache. Primarily for testing.
  * @internal
@@ -55,6 +67,13 @@ export declare function getModelMatchVariants(model: string): string[];
  * 3. Reverse containment: cost entry variant *contains* response model variant
  */
 export declare function searchModelInCosts(model: string, costsMap: Map<string, ModelPricing>): ModelPricing | undefined;
+/**
+ * Looks up a model's context length from the models map using the same
+ * 3-tier matching strategy as pricing lookup (exact, contained, reverse).
+ *
+ * Returns `undefined` if the model is not found or has no context length data.
+ */
+export declare function getModelContextLength(model: string, modelsMap: Map<string, ModelInfo>): number | undefined;
 /**
  * Finds pricing for a model, checking user overrides first then OpenRouter cache.
  * Pass a `warnedModels` set to suppress duplicate warnings per tracker instance.

package/dist/utils/compact-conversation.d.ts

@@ -1,4 +1,8 @@
-import { type LanguageModel, type ModelMessage } from "ai";
+import { type LanguageModel, type ModelMessage, type PrepareStepFunction, type ToolSet } from "ai";
+import { type ModelInfo } from "./budget-tracking";
+export declare class CompactionError extends Error {
+    readonly name = "CompactionError";
+}
 export interface CompactConversationConfig {
     /** Model's context limit (e.g., 200000 for Claude) */
     maxTokens: number;
@@ -10,6 +14,8 @@ export interface CompactConversationConfig {
     summarizerModel: LanguageModel;
     /** The original task/goal the agent is working on - helps preserve context */
     taskContext?: string;
+    /** Extra instructions for the summarizer (e.g., "Focus on database schema decisions") */
+    summaryInstructions?: string;
 }
 export interface CompactConversationState {
     /** Accumulated summary from previous compactions */
@@ -63,9 +69,42 @@ export interface CompactConversationResult {
  * ```
  */
 export declare function compactConversation(messages: ModelMessage[], config: CompactConversationConfig, state?: CompactConversationState): Promise<CompactConversationResult>;
+/**
+ * Create an auto-compaction helper that integrates with the AI SDK's `prepareStep` hook.
+ *
+ * Returns a `prepareStep` function that monitors context usage and automatically
+ * compacts the conversation when the threshold is exceeded.
+ *
+ * @example
+ * ```typescript
+ * import { createAutoCompaction } from 'bashkit';
+ *
+ * const compaction = createAutoCompaction({
+ *   maxTokens: 200_000,
+ *   summarizerModel: anthropic('claude-haiku-4'),
+ *   taskContext: 'Building a REST API',
+ * });
+ *
+ * const result = await generateText({
+ *   model: anthropic('claude-sonnet-4-5'),
+ *   tools,
+ *   messages,
+ *   prepareStep: compaction.prepareStep,
+ *   stopWhen: stepCountIs(20),
+ * });
+ * ```
+ */
+export declare function createAutoCompaction(config: CompactConversationConfig): {
+    prepareStep: PrepareStepFunction<ToolSet>;
+    getState: () => Readonly<CompactConversationState>;
+};
 /**
  * Pre-configured token limits for common models.
  * Use these with compactConversation config.
+ *
+ * @deprecated Use `fetchOpenRouterModels()` + `createCompactConfigFromModels()` instead.
+ * This static table goes stale as models evolve. The OpenRouter-based approach
+ * fetches up-to-date context lengths automatically.
  */
 export declare const MODEL_CONTEXT_LIMITS: {
     readonly "claude-opus-4-5": 200000;
@@ -78,8 +117,39 @@ export declare const MODEL_CONTEXT_LIMITS: {
     readonly "gemini-2.5-pro": 1000000;
     readonly "gemini-2.5-flash": 1000000;
 };
+/**
+ * @deprecated Use `fetchOpenRouterModels()` + `createCompactConfigFromModels()` instead.
+ */
 export type ModelContextLimit = keyof typeof MODEL_CONTEXT_LIMITS;
 /**
- * Helper to create config with model preset
+ * Helper to create config with model preset.
+ *
+ * @deprecated Use `createCompactConfigFromModels()` instead, which looks up
+ * context lengths dynamically from OpenRouter data.
  */
 export declare function createCompactConfig(modelId: ModelContextLimit, summarizerModel: LanguageModel, overrides?: Partial<Omit<CompactConversationConfig, "summarizerModel">>): CompactConversationConfig;
+/**
+ * Create a compaction config by looking up the model's context length from
+ * an OpenRouter models map. This replaces the static `MODEL_CONTEXT_LIMITS`
+ * table with dynamic, always-up-to-date values.
+ *
+ * @param modelId - Model ID to look up (e.g., "claude-sonnet-4-5", "anthropic/claude-sonnet-4-5")
+ * @param summarizerModel - Language model to use for summarization
+ * @param modelsMap - Models map from `fetchOpenRouterModels()`
+ * @param overrides - Optional config overrides
+ * @returns CompactConversationConfig
+ * @throws If the model's context length cannot be found in the models map
+ *
+ * @example
+ * ```typescript
+ * import { fetchOpenRouterModels, createCompactConfigFromModels } from 'bashkit';
+ *
+ * const models = await fetchOpenRouterModels();
+ * const config = createCompactConfigFromModels(
+ *   'claude-sonnet-4-5',
+ *   anthropic('claude-haiku-4'),
+ *   models,
+ * );
+ * ```
+ */
+export declare function createCompactConfigFromModels(modelId: string, summarizerModel: LanguageModel, modelsMap: Map<string, ModelInfo>, overrides?: Partial<Omit<CompactConversationConfig, "summarizerModel">>): CompactConversationConfig;

package/dist/utils/context-status.d.ts

@@ -29,6 +29,8 @@ export interface ContextStatusConfig {
     highGuidance?: string | ((metrics: ContextMetrics) => string);
     /** Custom guidance message for 'critical' status */
     criticalGuidance?: string | ((metrics: ContextMetrics) => string);
+    /** Pre-computed token count to skip re-estimation */
+    knownTokenCount?: number;
 }
 /**
  * Get the current context window status for a conversation.

package/dist/utils/debug.d.ts

@@ -59,13 +59,12 @@ export declare function debugEnd(id: string, tool: string, options: {
  */
 export declare function debugError(id: string, tool: string, error: string | Error): void;
 /**
- * Push a parent context for nested tool calls (e.g., when Task starts a subagent).
+ * Run a function with a debug parent context.
+ * All tool calls within `fn` (including async continuations) will have
+ * their `parent` field set to `parentId`. Safe for parallel execution —
+ * each call gets its own isolated async context via AsyncLocalStorage.
  */
-export declare function pushParent(id: string): void;
-/**
- * Pop the current parent context.
- */
-export declare function popParent(): void;
+export declare function runWithDebugParent<T>(parentId: string, fn: () => T): T;
 /**
  * Get all debug logs (memory mode only).
  * @returns Array of debug events, or empty array if not in memory mode

package/dist/utils/helpers.d.ts

@@ -0,0 +1,21 @@
+import type { ToolCallPart, ToolResultPart } from "ai";
+/**
+ * Type guard for tool call content parts in AI SDK messages.
+ * Checks the `type` discriminant to avoid false positives from objects
+ * that happen to have `toolName` and `input` properties.
+ */
+export declare function isToolCallPart(part: unknown): part is ToolCallPart;
+/**
+ * Type guard for tool result content parts in AI SDK messages.
+ * Checks the `type` discriminant to avoid false positives from objects
+ * that happen to have `toolName` and `output` properties.
+ */
+export declare function isToolResultPart(part: unknown): part is ToolResultPart;
+/**
+ * Middle-truncates text, keeping the first half and last half with a
+ * marker in between. Preserves both the beginning context and the
+ * actionable end (error summaries, test failures).
+ *
+ * Inspired by OpenAI Codex's truncate.rs — 50/50 head/tail split.
+ */
+export declare function middleTruncate(text: string, maxLength: number): string;

package/dist/utils/index.d.ts

@@ -1,5 +1,6 @@
-export { type BudgetStatus, type BudgetTracker, type ModelPricing, createBudgetTracker, } from "./budget-tracking";
-export { type CompactConversationConfig, type CompactConversationResult, type CompactConversationState, compactConversation, createCompactConfig, MODEL_CONTEXT_LIMITS, type ModelContextLimit, } from "./compact-conversation";
+export { type BudgetStatus, type BudgetTracker, type ModelInfo, type ModelPricing, createBudgetTracker, fetchOpenRouterModels, getModelContextLength, } from "./budget-tracking";
+export { type CompactConversationConfig, type CompactConversationResult, type CompactConversationState, CompactionError, compactConversation, createAutoCompaction, createCompactConfig, createCompactConfigFromModels, MODEL_CONTEXT_LIMITS, type ModelContextLimit, } from "./compact-conversation";
 export { type ContextMetrics, type ContextStatus, type ContextStatusConfig, type ContextStatusLevel, contextNeedsAttention, contextNeedsCompaction, getContextStatus, } from "./context-status";
 export { type DebugEvent, clearDebugLogs, getDebugLogs, isDebugEnabled, reinitDebugMode, } from "./debug";
+export { middleTruncate } from "./helpers";
 export { estimateMessagesTokens, estimateMessageTokens, estimateTokens, type PruneMessagesConfig, pruneMessagesByTokens, } from "./prune-messages";

package/package.json

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

package/dist/cloudflare/index.d.ts

@@ -1,20 +0,0 @@
-/**
- * bashkit/cloudflare - Cloudflare Workers compatible exports
- *
- * This module exports only the parts of bashkit that work in Cloudflare Workers.
- * It excludes local sandbox (uses Bun/Node APIs) and bundled ripgrep.
- *
- * @example
- * ```typescript
- * import { createE2BSandbox, createBashTool } from 'bashkit/cloudflare';
- * ```
- */
-export { createE2BSandbox, type E2BSandboxConfig } from "../sandbox/e2b";
-export type { ExecOptions, ExecResult, Sandbox } from "../sandbox/interface";
-export { createBashTool } from "../tools/bash";
-export { createReadTool } from "../tools/read";
-export { createWriteTool } from "../tools/write";
-export { createEditTool } from "../tools/edit";
-export { createGlobTool } from "../tools/glob";
-export { createGrepTool } from "../tools/grep";
-export type { BashOutput, BashError, ReadOutput, ReadError, WriteOutput, WriteError, EditOutput, EditError, GlobOutput, GlobError, GrepOutput, GrepError, } from "../tools";