@standardagents/builder 0.22.1 → 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++) {
@@ -9342,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) {
@@ -9412,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 = {
@@ -69077,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) {
@@ -69093,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);
@@ -69604,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}`);
@@ -70639,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) {
@@ -71054,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;