@standardagents/builder 0.24.4 → 0.24.6

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import { ModelDefinition, AgentDefinition } from './runtime.js';
 export { AgentBuilderEnv, AuthContext, AuthUser, BroadcastOptions, CodeExecutionBridge, DurableAgentBuilder, DurableThread, FlowStateSdk, FlowStateWithSdk, ImageContextConfig, InjectMessageOptions, LLMProviderInterface, NamespaceResolutionError, PromptContent, PromptDefinition, PromptIncludePart, PromptPart, PromptToolConfig, Provider, ProviderContentPart, ProviderFilePart, ProviderImagePart, ProviderRegistry, ProviderTextPart, SessionToolBinding, SessionToolConfig, SideConfig, SubagentToolConfig, SubpromptConfig, ThreadRegistryEntry, ToolConfig, UpdateThreadParams, User, authenticate, buildImageDescription, cat, createNamespaceContext, emitThreadEvent, enhanceFlowState, exists, find, forceTurn, generateImageDescription, getFileStats, getMessages, getMessagesToSummarize, getShortName, getThumbnail, getUnsummarizedImageAttachments, getVisibleAgentNames, getVisibleModelNames, getVisiblePromptNames, getVisibleToolNames, grep, hasImageAttachments, head, injectMessage, isQualifiedName, linkFile, mkdir, move, optimizeImageContext, parseQualifiedName, qualifyName, queueTool, readFile, readdir, reloadHistory, rename, replaceImagesWithDescriptions, requireAdmin, requireAuth, resolveAgent, resolveHook, resolveModel, resolvePrompt, resolveTool, rmdir, stat, tail, unlink, updateThread, writeFile, writeImage } from './runtime.js';
 export { AgentPluginOptions, agentbuilder, builder } from './plugin.js';
-export { A as Agent, m as AttachmentRef, f as BuilderThreadEndpointHandler, B as Controller, a as ControllerContext, E as Env, l as FileRecord, n as FileStats, j as FlowResult, F as FlowState, G as GrepResult, p as ImageContentPart, I as ImageMetadata, L as LLMResponse, M as Message, r as MessageContent, q as MultimodalContent, R as RequestContext, S as StorageBackend, k as TelemetryEvent, o as TextContentPart, b as ThreadEndpointContext, e as ThreadEnv, g as ThreadInstance, h as ThreadMetadata, T as ToolCall, i as ToolResult, c as createThreadEndpointHandler, d as defineController } from './index-jDKWlMKG.js';
+export { A as Agent, m as AttachmentRef, f as BuilderThreadEndpointHandler, B as Controller, a as ControllerContext, E as Env, l as FileRecord, n as FileStats, j as FlowResult, F as FlowState, G as GrepResult, p as ImageContentPart, I as ImageMetadata, L as LLMResponse, M as Message, r as MessageContent, q as MultimodalContent, R as RequestContext, S as StorageBackend, k as TelemetryEvent, o as TextContentPart, b as ThreadEndpointContext, e as ThreadEnv, g as ThreadInstance, h as ThreadMetadata, T as ToolCall, i as ToolResult, c as createThreadEndpointHandler, d as defineController } from './index-D3FNNQWC.js';
 export { AgentType, DefinitionLoader, GlobalNamespaceContext, HookSignatures, ImageContent, ModelCapabilities, ModelProvider, NamespaceContext, PackageSignature, PackedExports, PackedMeta, PackedMetadata, PackedNamespaceContext, PromptInput, PromptTextPart, ProviderAssistantMessage, ProviderConfigSlot, ProviderConfigSlots, ProviderConfigValueSource, ProviderDefinition, ProviderError, ProviderErrorCode, ProviderFactory, ProviderFactoryConfig, ProviderFinishReason, ProviderGeneratedImage, ProviderMessage, ProviderMessageContent, ProviderMethodOverrides, ModelCapabilities as ProviderModelCapabilities, ProviderOverride, ProviderOverrideContext, ProviderReasoningDetail, ProviderRequest, ProviderResponse, ProviderStreamChunk, ProviderSyncOverride, ProviderSystemMessage, ProviderTool, ProviderToolCallPart, ProviderToolMessage, ProviderToolResultContent, ProviderUsage, ProviderUserMessage, ReasoningConfig, StructuredPrompt, TextContent, Tool, ToolArgs, ToolArgsNode, ToolArgsRawShape, ToolContent, belongsToPackage, defineAgent, defineHook, defineModel, definePrompt, defineProvider, defineTool, isPacked, isVisibleInNamespace, mapReasoningLevel, providerConst, providerEnv, providerValue } from '@standardagents/spec';
 export { C as ConstituentItem, D as DiscoveredPackage, N as NamespacedRegistry, P as PackingAnalysis, a as PackingOptions, b as PackingResult, c as UnpackOptions, d as UnpackResult } from './types-Bpe7IANZ.js';
 export { D as DiscoveryConfig, P as PackageDiscoveryService, d as discoverPackages } from './discovery-DiMJWisl.js';

package/dist/index.js

@@ -217,6 +217,18 @@ function getFallbackPricing(providerName, modelId) {
   const pricing = getProviderModelPricing(providerName, modelId);
   return pricing ? toResolvedModelPricing(pricing) : null;
 }
+function resolveUsageReportedPricing(usage) {
+  const pricing = usage.pricing;
+  if (!pricing || typeof pricing.inputPerMillion !== "number" || typeof pricing.outputPerMillion !== "number") {
+    return null;
+  }
+  return {
+    inputPrice: pricing.inputPerMillion,
+    outputPrice: pricing.outputPerMillion,
+    cachedPrice: pricing.cachedInputPerMillion,
+    source: "provider"
+  };
+}
 function resolveModelPricing(modelDef, providerName) {
   if (typeof modelDef.inputPrice === "number" && typeof modelDef.outputPrice === "number") {
     return {
@@ -245,14 +257,14 @@ function calculateUsageCost(usage, pricing) {
   const inputPrice = useLongContext ? pricing.longInputPrice ?? pricing.inputPrice : pricing.inputPrice;
   const outputPrice = useLongContext ? pricing.longOutputPrice ?? pricing.outputPrice : pricing.outputPrice;
   const cachedTokenPrice = useLongContext ? pricing.longCachedPrice ?? pricing.cachedPrice ?? pricing.inputPrice : pricing.cachedPrice ?? pricing.inputPrice;
-  const costInput = roundCost(
-    (uncachedPromptTokens * inputPrice + cachedTokens * cachedTokenPrice) / TOKENS_PER_MILLION
-  );
+  const costInput = roundCost(uncachedPromptTokens * inputPrice / TOKENS_PER_MILLION);
+  const costCached = roundCost(cachedTokens * cachedTokenPrice / TOKENS_PER_MILLION);
   const costOutput = roundCost(billableOutputTokens * outputPrice / TOKENS_PER_MILLION);
   return {
     costInput,
+    costCached,
     costOutput,
-    costTotal: roundCost(costInput + costOutput)
+    costTotal: roundCost(costInput + costCached + costOutput)
   };
 }
 var TOKENS_PER_MILLION;
@@ -1358,7 +1370,8 @@ function convertUsage(usage) {
     prompt_tokens_details: usage.cachedTokens ? { cached_tokens: usage.cachedTokens } : void 0,
     completion_tokens_details: usage.reasoningTokens ? { reasoning_tokens: usage.reasoningTokens } : void 0,
     cost: usage.cost,
-    provider: usage.provider
+    provider: usage.provider,
+    pricing: usage.pricing
   };
 }
 function inferProviderFromModelId(model) {
@@ -2053,7 +2066,7 @@ var init_LLMRequest = __esm({
           const toolsCalled = response.tool_calls ? JSON.stringify(response.tool_calls.map((tc) => tc.function.name)) : null;
           const providerName = response._provider?.name;
           const standardAgentsRouterUsed = response._viaPlatform === true || providerName === "platform";
-          const resolvedPricing = resolveModelPricing(modelDef, providerName);
+          const resolvedPricing = resolveModelPricing(modelDef) ?? resolveUsageReportedPricing(response.usage) ?? resolveModelPricing(modelDef, providerName);
           const calculatedCost = calculateUsageCost(response.usage, resolvedPricing);
           const providerReportedCost = typeof response.usage.cost === "number" ? response.usage.cost : typeof response.usage.cost === "string" ? parseFloat(response.usage.cost) : null;
           const cost_total = providerReportedCost != null && !Number.isNaN(providerReportedCost) ? providerReportedCost : calculatedCost?.costTotal ?? null;
@@ -2120,6 +2133,7 @@ var init_LLMRequest = __esm({
             reasoning_tokens: response.usage.completion_tokens_details?.reasoning_tokens || 0,
             total_tokens: response.usage.total_tokens,
             cost_input: calculatedCost?.costInput,
+            cost_cached: calculatedCost?.costCached,
             cost_output: calculatedCost?.costOutput,
             latency_ms: Date.now() - startTime,
             time_to_first_token_ms: timing?.latencyMs ?? void 0,
@@ -2149,19 +2163,20 @@ var init_LLMRequest = __esm({
               reasoning_tokens = ?5,
               total_tokens = ?6,
               cost_input = ?7,
-              cost_output = ?8,
-              latency_ms = ?9,
-              finish_reason = ?10,
-              tools_called = ?11,
-              cost_total = ?12,
-              is_complete = ?13,
-              reasoning_content = ?14,
-              provider_tools = ?15,
-              actual_provider = ?16,
-              standard_agents_router_used = ?17,
-              time_to_first_token_ms = ?18,
-              generation_ms = ?19
-            WHERE id = ?20
+              cost_cached = ?8,
+              cost_output = ?9,
+              latency_ms = ?10,
+              finish_reason = ?11,
+              tools_called = ?12,
+              cost_total = ?13,
+              is_complete = ?14,
+              reasoning_content = ?15,
+              provider_tools = ?16,
+              actual_provider = ?17,
+              standard_agents_router_used = ?18,
+              time_to_first_token_ms = ?19,
+              generation_ms = ?20
+            WHERE id = ?21
           `,
                 logData.response_body,
                 logData.input_tokens,
@@ -2170,6 +2185,7 @@ var init_LLMRequest = __esm({
                 logData.reasoning_tokens,
                 logData.total_tokens,
                 logData.cost_input ?? null,
+                logData.cost_cached ?? null,
                 logData.cost_output ?? null,
                 logData.latency_ms,
                 logData.finish_reason,
@@ -22918,8 +22934,23 @@ var migration31 = {
   }
 };
 
+// src/durable-objects/migrations/032_add_cost_cached.ts
+var migration32 = {
+  version: 32,
+  name: "add_cost_cached",
+  async up(sql) {
+    const tableInfo = sql.exec(`PRAGMA table_info(logs)`).toArray();
+    const hasCostCached = tableInfo.some((col) => col.name === "cost_cached");
+    if (!hasCostCached) {
+      await sql.exec(`
+        ALTER TABLE logs ADD COLUMN cost_cached REAL;
+      `);
+    }
+  }
+};
+
 // src/durable-objects/migrations/index.ts
-var migrations = [migration, migration2, migration3, migration4, migration5, migration6, migration7, migration8, migration9, migration10, migration11, migration12, migration13, migration14, migration15, migration16, migration17, migration18, migration19, migration20, migration21, migration22, migration23, migration24, migration25, migration26, migration27, migration28, migration29, migration30, migration31];
+var migrations = [migration, migration2, migration3, migration4, migration5, migration6, migration7, migration8, migration9, migration10, migration11, migration12, migration13, migration14, migration15, migration16, migration17, migration18, migration19, migration20, migration21, migration22, migration23, migration24, migration25, migration26, migration27, migration28, migration29, migration30, migration31, migration32];
 var LATEST_SCHEMA_VERSION = migrations.length;
 
 // src/durable-objects/DurableThread.ts
@@ -24399,8 +24430,10 @@ var DurableThread = class _DurableThread extends DurableObject {
   async setValue(key, value) {
     await this.ensureMigrated();
     this.assertValidThreadValueKey(key);
+    const updatedAt = Date.now() * 1e3;
     if (value === null || value === void 0) {
       await this.ctx.storage.sql.exec(`DELETE FROM thread_values WHERE key = ?`, key);
+      this.broadcastLog({ type: "kv_data", key, deleted: true, updated_at: updatedAt });
       return;
     }
     const serialized = this.serializeThreadValue(value);
@@ -24414,8 +24447,9 @@ var DurableThread = class _DurableThread extends DurableObject {
     `,
       key,
       serialized,
-      Date.now() * 1e3
+      updatedAt
     );
+    this.broadcastLog({ type: "kv_data", key, deleted: false, updated_at: updatedAt });
   }
   async listValues(options = {}) {
     await this.ensureMigrated();
@@ -25035,9 +25069,9 @@ var DurableThread = class _DurableThread extends DurableObject {
    * Each migration is run in order, starting from the current version + 1.
    */
   async runMigrations(fromVersion) {
-    for (const migration42 of migrations) {
-      if (migration42.version > fromVersion) {
-        await migration42.up(this.ctx.storage.sql);
+    for (const migration43 of migrations) {
+      if (migration43.version > fromVersion) {
+        await migration43.up(this.ctx.storage.sql);
       }
     }
   }
@@ -26471,7 +26505,7 @@ ${resultContent}${attachmentSummary}`;
         SELECT
           id, message_id, provider, actual_provider, standard_agents_router_used, model, model_name, prompt_name,
           tools_called, queued_tools, provider_tools, parent_log_id, retry_of_log_id, error, cost_total,
-          input_tokens, output_tokens, total_tokens,
+          input_tokens, cached_tokens, output_tokens, total_tokens,
           is_complete, created_at, request_body
         FROM logs
         ORDER BY created_at ${order}
@@ -26497,6 +26531,7 @@ ${resultContent}${attachmentSummary}`;
         error: row.error,
         cost_total: row.cost_total,
         input_tokens: row.input_tokens,
+        cached_tokens: row.cached_tokens,
         output_tokens: row.output_tokens,
         total_tokens: row.total_tokens,
         is_complete: row.is_complete === 1,
@@ -26520,11 +26555,17 @@ ${resultContent}${attachmentSummary}`;
     await this.ensureMigrated();
     const byModel = {};
     let totalCost = 0;
+    let inputCost = 0;
+    let cachedCost = 0;
+    let outputCost = 0;
     let requestCount = 0;
     const modelRows = await this.ctx.storage.sql.exec(
       `
       SELECT COALESCE(model_name, model, 'unknown') AS model_key,
              SUM(cost_total) AS total_cost,
+             SUM(CASE WHEN cost_cached IS NOT NULL THEN cost_input END) AS input_cost,
+             SUM(cost_cached) AS cached_cost,
+             SUM(cost_output) AS output_cost,
              COUNT(*) AS request_count
       FROM logs
       WHERE cost_total IS NOT NULL
@@ -26532,15 +26573,32 @@ ${resultContent}${attachmentSummary}`;
       `
     );
     for (const row of modelRows.toArray()) {
-      byModel[row.model_key] = { totalCost: row.total_cost, count: row.request_count };
+      byModel[row.model_key] = {
+        totalCost: row.total_cost,
+        inputCost: row.input_cost ?? 0,
+        cachedCost: row.cached_cost ?? 0,
+        outputCost: row.output_cost ?? 0,
+        count: row.request_count
+      };
       totalCost += row.total_cost;
+      inputCost += row.input_cost ?? 0;
+      cachedCost += row.cached_cost ?? 0;
+      outputCost += row.output_cost ?? 0;
       requestCount += row.request_count;
     }
     const messageRow = await this.ctx.storage.sql.exec(
       `SELECT COUNT(DISTINCT message_id) AS message_count FROM logs WHERE cost_total IS NOT NULL`
     );
     const messageCount = messageRow.one().message_count || 0;
-    return { totalCost, requestCount, messageCount, byModel };
+    return {
+      totalCost,
+      inputCost,
+      cachedCost,
+      outputCost,
+      requestCount,
+      messageCount,
+      byModel
+    };
   }
   /**
    * Get specific logs, optionally including all descendant logs.
@@ -26753,7 +26811,7 @@ ${resultContent}${attachmentSummary}`;
           l.request_body, l.request_headers, l.response_body, l.response_headers,
           l.status_code, l.reasoning_content, l.input_tokens, l.cached_tokens, l.output_tokens,
           l.reasoning_tokens, l.total_tokens, l.latency_ms, l.time_to_first_token_ms, l.generation_ms,
-          l.finish_reason, l.error, l.error_type, l.cost_input, l.cost_output, l.cost_total,
+          l.finish_reason, l.error, l.error_type, l.cost_input, l.cost_cached, l.cost_output, l.cost_total,
           l.message_history_length, l.tools_available, l.prompt_name, l.tools_called,
           l.queued_tools, l.provider_tools, l.parent_log_id, l.tools_schema, l.system_prompt,
           l.errors, l.retry_of_log_id, l.tool_results, l.is_complete, l.created_at,
@@ -26843,6 +26901,7 @@ ${resultContent}${attachmentSummary}`;
         error: row.error,
         error_type: row.error_type,
         cost_input: row.cost_input,
+        cost_cached: row.cost_cached,
         cost_output: row.cost_output,
         cost_total: row.cost_total,
         message_history_length: row.message_history_length,
@@ -27342,6 +27401,7 @@ ${resultContent}${attachmentSummary}`;
    * Broadcast a log record to all connected WebSocket clients
    */
   broadcastLog(log) {
+    if (!this.logSockets || this.logSockets.size === 0) return;
     const payload = JSON.stringify(log);
     for (const ws of Array.from(this.logSockets)) {
       try {
@@ -28745,7 +28805,7 @@ ${resultContent}${attachmentSummary}`;
 };
 
 // src/durable-objects/agentbuilder-migrations/0001_initial.ts
-var migration32 = {
+var migration33 = {
   version: 1,
   async up(sql) {
     sql.exec(`
@@ -28844,7 +28904,7 @@ var migration32 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0002_add_tenvs_properties.ts
-var migration33 = {
+var migration34 = {
   version: 2,
   async up(sql) {
     sql.exec(`ALTER TABLE threads ADD COLUMN tenvs TEXT`);
@@ -28856,7 +28916,7 @@ var migration33 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0003_add_parent_and_terminated.ts
-var migration34 = {
+var migration35 = {
   version: 3,
   async up(sql) {
     sql.exec(`ALTER TABLE threads ADD COLUMN parent TEXT REFERENCES threads(id)`);
@@ -28869,7 +28929,7 @@ var migration34 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0004_add_env_tables.ts
-var migration35 = {
+var migration36 = {
   version: 4,
   async up(sql) {
     sql.exec(`ALTER TABLE threads ADD COLUMN env TEXT`);
@@ -28893,7 +28953,7 @@ var migration35 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0005_add_pending_subagent_env_requests.ts
-var migration36 = {
+var migration37 = {
   version: 5,
   async up(sql) {
     sql.exec(`
@@ -28919,7 +28979,7 @@ var migration36 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0006_add_env_value_types.ts
-var migration37 = {
+var migration38 = {
   version: 6,
   async up(sql) {
     sql.exec(
@@ -28934,7 +28994,7 @@ var migration37 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0007_platform_identity_replica.ts
-var migration38 = {
+var migration39 = {
   version: 7,
   async up(sql) {
     sql.exec(`ALTER TABLE users ADD COLUMN platform_user_id TEXT`);
@@ -28963,7 +29023,7 @@ var migration38 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0008_add_skills.ts
-var migration39 = {
+var migration40 = {
   version: 8,
   async up(sql) {
     sql.exec(`
@@ -28994,7 +29054,7 @@ var migration39 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0009_add_user_permissions.ts
-var migration40 = {
+var migration41 = {
   version: 9,
   async up(sql) {
     sql.exec(`ALTER TABLE users ADD COLUMN permissions TEXT`);
@@ -29005,7 +29065,7 @@ var migration40 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/0010_add_device_logins.ts
-var migration41 = {
+var migration42 = {
   version: 10,
   async up(sql) {
     sql.exec(`
@@ -29025,7 +29085,7 @@ var migration41 = {
 };
 
 // src/durable-objects/agentbuilder-migrations/index.ts
-var migrations2 = [migration32, migration33, migration34, migration35, migration36, migration37, migration38, migration39, migration40, migration41];
+var migrations2 = [migration33, migration34, migration35, migration36, migration37, migration38, migration39, migration40, migration41, migration42];
 var LATEST_SCHEMA_VERSION2 = 10;
 
 // src/utils/crypto.ts
@@ -29788,9 +29848,9 @@ ${result ?? error ?? "No result content."}${attachmentSummary}`;
     }
   }
   async runMigrations(fromVersion) {
-    for (const migration42 of migrations2) {
-      if (migration42.version > fromVersion) {
-        await migration42.up(this.ctx.storage.sql);
+    for (const migration43 of migrations2) {
+      if (migration43.version > fromVersion) {
+        await migration43.up(this.ctx.storage.sql);
       }
     }
   }