@standardagents/builder 0.22.0 → 0.23.0

package/dist/built-in-routes.js

@@ -5755,7 +5755,7 @@ ${attachmentPaths}`;
             const attachmentId = crypto.randomUUID();
             const timestamp = Date.now();
             const ext = attachment.mimeType === "image/png" ? "png" : attachment.mimeType === "image/jpeg" ? "jpg" : attachment.mimeType === "image/gif" ? "gif" : attachment.mimeType === "image/webp" ? "webp" : attachment.name.split(".").pop() || "bin";
-            const path9 = `/attachments/${timestamp}-${attachmentId}.${ext}`;
+            const path9 = `/attachments/${timestamp}-${attachmentId.slice(0, 8)}.${ext}`;
             const binaryString = atob(attachment.data);
             const dataBuffer = new Uint8Array(binaryString.length);
             for (let i = 0; i < binaryString.length; i++) {
@@ -5989,9 +5989,10 @@ var init_context = __esm({
 // src/agents/FlowEngine.ts
 var FlowEngine_exports = {};
 __export(FlowEngine_exports, {
-  FlowEngine: () => FlowEngine
+  FlowEngine: () => FlowEngine,
+  HookBlockedExecutionError: () => HookBlockedExecutionError
 });
-var IMMEDIATE_BOOTSTRAP_TOOL_NAME, FlowEngine;
+var IMMEDIATE_BOOTSTRAP_TOOL_NAME, HookBlockedExecutionError, FlowEngine;
 var init_FlowEngine = __esm({
   "src/agents/FlowEngine.ts"() {
     init_types();
@@ -6002,6 +6003,12 @@ var init_FlowEngine = __esm({
     init_sessionTools();
     init_ThreadStateImpl();
     IMMEDIATE_BOOTSTRAP_TOOL_NAME = "bootstrap_immediate_subagent";
+    HookBlockedExecutionError = class extends Error {
+      constructor(message) {
+        super(message);
+        this.name = "HookBlockedExecutionError";
+      }
+    };
     FlowEngine = class _FlowEngine {
       static IMMEDIATE_SCOPED_ENV_CONTEXT_KEY = "__immediate_scoped_env_by_call_id";
       /**
@@ -6549,7 +6556,26 @@ This usually points to something I can't fix from here \u2014 for example the wo
         state.messageHistory = await this.loadMessageHistory(state);
         const context = await this.assembleContext(state);
         state.allowedTools = context.tools;
-        context.messages = await this.runPrefilterLLMHistoryHook(state, context.messages);
+        try {
+          context.messages = await this.runPrefilterLLMHistoryHook(state, context.messages);
+        } catch (error) {
+          if (error instanceof HookBlockedExecutionError) {
+            const threadState = ThreadStateImpl.fromFlowState(state);
+            await threadState.injectMessage({ role: "assistant", content: error.message });
+            state.stopped = true;
+            state.stoppedBy = state.currentSide;
+            state.stopReason = error.message;
+            state.stopReasonCode = "hook_blocked";
+            state.emitTelemetry?.({
+              type: "stopped",
+              reason: state.stopReason,
+              side: state.currentSide,
+              timestamp: Date.now()
+            });
+            return;
+          }
+          throw error;
+        }
         const pendingMessageId = await this.createPendingMessage(state);
         state.pendingMessageId = pendingMessageId;
         let responseStatus = "completed";
@@ -8705,6 +8731,9 @@ ${lines.join("\n\n")}`
           try {
             result = await executor(threadState, result);
           } catch (error) {
+            if (error && typeof error === "object" && error.block === true) {
+              throw new HookBlockedExecutionError(error instanceof Error ? error.message : String(error));
+            }
             console.error("[Hooks] \u2717 Error running prefilter_llm_history hook:", error);
           }
         }
@@ -9313,7 +9342,7 @@ ${lines.join("\n\n")}`
           const base64Data = match2[2];
           const ext = mimeType.split("/")[1] || "png";
           const attachmentId = crypto.randomUUID();
-          const path9 = `/attachments/${Date.now()}-${attachmentId}.${ext}`;
+          const path9 = `/attachments/${Date.now()}-${attachmentId.slice(0, 8)}.${ext}`;
           try {
             await state.thread.instance.writeFile(path9, base64Data, mimeType);
             if (image.id) {
@@ -9383,7 +9412,7 @@ ${lines.join("\n\n")}`
           const base64Data = imageUrl.slice(base64Index + base64Marker.length);
           const ext = mimeType.split("/")[1] || "png";
           const attachmentId = crypto.randomUUID();
-          const path9 = `/attachments/${Date.now()}-${attachmentId}.${ext}`;
+          const path9 = `/attachments/${Date.now()}-${attachmentId.slice(0, 8)}.${ext}`;
           try {
             await state.thread.instance.writeFile(path9, base64Data, mimeType);
             const attachmentRef = {
@@ -69048,7 +69077,54 @@ var continue_post_default = defineController2(async ({ req, params, env: env2 })
   }
 });
 
+// src/agents/cost-summary.ts
+function emptyCostSummary() {
+  return { totalCost: 0, requestCount: 0, messageCount: 0, byModel: {} };
+}
+function mergeByModel(summaries) {
+  const merged = {};
+  for (const summary of summaries) {
+    for (const [modelName, acc] of Object.entries(summary.byModel)) {
+      const entry = merged[modelName] ?? { totalCost: 0, count: 0 };
+      entry.totalCost += acc.totalCost;
+      entry.count += acc.count;
+      merged[modelName] = entry;
+    }
+  }
+  return merged;
+}
+function sumSummaries(summaries) {
+  let totalCost = 0;
+  let requestCount = 0;
+  let messageCount = 0;
+  for (const summary of summaries) {
+    totalCost += summary.totalCost;
+    requestCount += summary.requestCount;
+    messageCount += summary.messageCount;
+  }
+  return { totalCost, requestCount, messageCount };
+}
+function buildModelBreakdown(byModel) {
+  return Object.entries(byModel).map(([modelName, acc]) => ({
+    modelName,
+    totalCost: acc.totalCost,
+    requestCount: acc.count,
+    avgCostPerRequest: acc.count > 0 ? acc.totalCost / acc.count : 0
+  })).sort((a, b) => b.totalCost - a.totalCost);
+}
+function threadNameFromTags(tags) {
+  for (const tag of tags ?? []) {
+    if (typeof tag === "string" && tag.startsWith("name:")) {
+      const value = tag.slice("name:".length).trim();
+      if (value) return value;
+    }
+  }
+  return null;
+}
+
 // src/api/threads/[id]/cost.get.ts
+var MAX_DESCENDANTS = 200;
+var SUMMARY_BATCH = 10;
 var cost_get_default = defineController2(async ({ req, params, env: env2 }) => {
   const threadId = params.id;
   if (!threadId) {
@@ -69064,52 +69140,70 @@ var cost_get_default = defineController2(async ({ req, params, env: env2 }) => {
         { status: 404 }
       );
     }
-    const durableId = env2.AGENT_BUILDER_THREAD.idFromName(threadId);
-    const stub = env2.AGENT_BUILDER_THREAD.get(durableId);
-    const logsResult = await stub.getLogs(1e4, 0, "ASC");
-    const logs = logsResult.logs || [];
-    let totalCost = 0;
-    const modelCosts = {};
-    const uniqueMessages = /* @__PURE__ */ new Set();
-    for (const log of logs) {
-      const cost = log.cost_total;
-      if (cost !== null && cost !== void 0 && !isNaN(cost)) {
-        totalCost += cost;
-        if (log.message_id) {
-          uniqueMessages.add(log.message_id);
-        }
-        const modelKey = log.model_name || log.model || "unknown";
-        if (!modelCosts[modelKey]) {
-          modelCosts[modelKey] = {
-            totalCost: 0,
-            count: 0
-          };
-        }
-        modelCosts[modelKey].totalCost += cost;
-        modelCosts[modelKey].count += 1;
+    const descendants = (await agentBuilder.getDescendantThreads(threadId) ?? []).slice(
+      0,
+      MAX_DESCENDANTS + 1
+    );
+    const truncated = descendants.length > MAX_DESCENDANTS;
+    if (truncated) descendants.length = MAX_DESCENDANTS;
+    const summarize = async (id) => {
+      try {
+        const durableId = env2.AGENT_BUILDER_THREAD.idFromName(id);
+        const stub = env2.AGENT_BUILDER_THREAD.get(durableId);
+        return await stub.getCostSummary() ?? emptyCostSummary();
+      } catch {
+        return emptyCostSummary();
       }
+    };
+    const ownSummary = await summarize(threadId);
+    const descendantSummaries = [];
+    for (let i = 0; i < descendants.length; i += SUMMARY_BATCH) {
+      const batch = descendants.slice(i, i + SUMMARY_BATCH);
+      descendantSummaries.push(
+        ...await Promise.all(batch.map((child) => summarize(child.id)))
+      );
     }
-    const messageCount = uniqueMessages.size;
-    const avgCostPerMessage = messageCount > 0 ? totalCost / messageCount : 0;
+    const allSummaries = [ownSummary, ...descendantSummaries];
+    const totals = sumSummaries(allSummaries);
+    const byModel = mergeByModel(allSummaries);
+    const modelBreakdown = buildModelBreakdown(byModel);
     const avgCostPerMessageByModel = {};
-    for (const [modelName, data] of Object.entries(modelCosts)) {
-      avgCostPerMessageByModel[modelName] = {
-        avgCost: data.count > 0 ? data.totalCost / data.count : 0,
-        messageCount: data.count
+    for (const entry of modelBreakdown) {
+      avgCostPerMessageByModel[entry.modelName] = {
+        avgCost: entry.avgCostPerRequest,
+        messageCount: entry.requestCount
       };
     }
-    const modelBreakdown = Object.entries(modelCosts).map(([modelName, data]) => ({
-      modelName,
-      totalCost: data.totalCost,
-      requestCount: data.count,
-      avgCostPerRequest: data.count > 0 ? data.totalCost / data.count : 0
-    }));
+    const threads = [
+      {
+        threadId,
+        agentName: thread.agent_name ?? null,
+        threadName: threadNameFromTags(thread.tags),
+        parent: null,
+        isRoot: true,
+        totalCost: ownSummary.totalCost,
+        requestCount: ownSummary.requestCount
+      },
+      ...descendants.map((child, index) => ({
+        threadId: child.id,
+        agentName: child.agent_name ?? null,
+        threadName: threadNameFromTags(child.tags),
+        parent: child.parent,
+        isRoot: false,
+        totalCost: descendantSummaries[index]?.totalCost ?? 0,
+        requestCount: descendantSummaries[index]?.requestCount ?? 0
+      })).filter((row) => row.requestCount > 0)
+    ];
     return Response.json({
-      totalCost,
-      messageCount,
-      avgCostPerMessage,
+      totalCost: totals.totalCost,
+      ownCost: ownSummary.totalCost,
+      requestCount: totals.requestCount,
+      messageCount: totals.messageCount,
+      avgCostPerMessage: totals.messageCount > 0 ? totals.totalCost / totals.messageCount : 0,
       modelBreakdown,
-      avgCostPerMessageByModel
+      avgCostPerMessageByModel,
+      threads,
+      truncated
     });
   } catch (error) {
     console.error(`Error calculating cost for thread ${threadId}:`, error);
@@ -69575,7 +69669,7 @@ var messages_post_default = defineController2(async ({ req, params, env: env2 })
           }
         }
         const ext = mimeType === "image/png" ? "png" : mimeType === "image/jpeg" ? "jpg" : attachment.name.split(".").pop() || "bin";
-        const path9 = `/attachments/${timestamp}-${attachmentId}.${ext}`;
+        const path9 = `/attachments/${timestamp}-${attachmentId.slice(0, 8)}.${ext}`;
         const base64Size = fileData.length;
         const binarySize = Math.ceil(base64Size * 3 / 4);
         console.log(`[writeFile] Storing ${attachment.name}: ${(binarySize / 1024 / 1024).toFixed(2)}MB binary (${(base64Size / 1024 / 1024).toFixed(2)}MB base64) to ${path9}`);
@@ -70189,7 +70283,13 @@ var approve_get_default = defineController2(async ({ req, env: env2, params }) =
   const code = params.code;
   if (!code) return page("Missing code", "This device link is malformed.", false);
   const authResult = await requireAuth(req, env2);
-  if (authResult instanceof Response) return authResult;
+  if (authResult instanceof Response) {
+    const url = new URL(req.url);
+    const returnTo = encodeURIComponent(url.pathname);
+    const hosted = Boolean(env2.STANDARD_AGENTS_HOSTED && String(env2.STANDARD_AGENTS_HOSTED) !== "0" && String(env2.STANDARD_AGENTS_HOSTED).toLowerCase() !== "false");
+    const target = hosted ? `/api/auth/sa/start?return_to=${returnTo}` : `/login?return_to=${returnTo}`;
+    return new Response(null, { status: 302, headers: { Location: `${url.origin}${target}` } });
+  }
   if (authResult.authType === "api_key") {
     return page("Sign in required", "Device logins must be approved from a signed-in browser session.", false);
   }
@@ -70604,6 +70704,20 @@ async function handleChunkedUploadComplete(stub, filePath, req) {
 
 // src/api/threads/[id]/logs/[logId].get.ts
 var logId_get_default = defineController2(async ({ req, params, env: env2 }) => {
+  if (isPlatformHosted(env2)) {
+    const authResult = await requirePermission(req, env2, "logs");
+    if (authResult instanceof Response) {
+      return authResult;
+    }
+  } else {
+    const authContext = await authenticate(req, env2);
+    if (authContext && !hasPermission(authContext.user, "logs")) {
+      return Response.json(
+        { error: "Forbidden: 'logs' access required" },
+        { status: 403 }
+      );
+    }
+  }
   const threadId = params.id;
   const logId = params.logId;
   if (!threadId) {
@@ -71019,6 +71133,20 @@ function createInspectProviderConfig(providerName, factory, env2) {
   };
 }
 var provider_get_default = defineController2(async ({ req, params, env: env2 }) => {
+  if (isPlatformHosted(env2)) {
+    const authResult = await requirePermission(req, env2, "logs");
+    if (authResult instanceof Response) {
+      return authResult;
+    }
+  } else {
+    const authContext = await authenticate(req, env2);
+    if (authContext && !hasPermission(authContext.user, "logs")) {
+      return Response.json(
+        { error: "Forbidden: 'logs' access required" },
+        { status: 403 }
+      );
+    }
+  }
   const threadId = params.id;
   const logId = params.logId;
   const providerName = params.provider;