Add usage and cost to TaskEvent

Author: ericallam

apps/webapp/app/v3/otlpExporter.server.ts

@@ -307,9 +307,18 @@ function convertSpansToCreateableEvents(resourceSpan: ResourceSpans): Array<Crea
                 "."
               )
             ) ?? resourceProperties.attemptNumber,
-          usageDurationMs: extractDoubleAttribute(
+          usageDurationMs:
+            extractDoubleAttribute(
+              span.attributes ?? [],
+              SemanticInternalAttributes.USAGE_DURATION_MS
+            ) ??
+            extractNumberAttribute(
+              span.attributes ?? [],
+              SemanticInternalAttributes.USAGE_DURATION_MS
+            ),
+          usageCostInCents: extractDoubleAttribute(
             span.attributes ?? [],
-            SemanticInternalAttributes.USAGE_DURATION_MS
+            SemanticInternalAttributes.USAGE_COST_IN_CENTS
           ),
         };
       })

packages/core/src/v3/semanticInternalAttributes.ts

@@ -47,4 +47,5 @@ export const SemanticInternalAttributes = {
   LINK_TITLE: "$link.title",
   IDEMPOTENCY_KEY: "ctx.run.idempotencyKey",
   USAGE_DURATION_MS: "$usage.durationMs",
+  USAGE_COST_IN_CENTS: "$usage.costInCents",
 };

packages/core/src/v3/tracer.ts

@@ -12,6 +12,7 @@ import { Logger, logs } from "@opentelemetry/api-logs";
 import { SemanticInternalAttributes } from "./semanticInternalAttributes";
 import { clock } from "./clock-api";
 import { usage } from "./usage-api";
+import { taskContext } from "./task-context-api";
 
 export type TriggerTracerConfig =
   | {
@@ -100,9 +101,13 @@ export class TriggerTracer {
           throw e;
         } finally {
           const usageSample = usage.stop(usageMeasurement);
+          const machine = taskContext.ctx?.machine;
 
           span.setAttributes({
             [SemanticInternalAttributes.USAGE_DURATION_MS]: usageSample.cpuTime,
+            [SemanticInternalAttributes.USAGE_COST_IN_CENTS]: machine?.centsPerMs
+              ? usageSample.cpuTime * machine.centsPerMs
+              : 0,
           });
 
           span.end(clock.preciseNow());

packages/database/prisma/migrations/20240612175820_add_usage_cost_in_cents_to_task_event/migration.sql

@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE "TaskEvent" ADD COLUMN     "usageCostInCents" DOUBLE PRECISION NOT NULL DEFAULT 0;

packages/database/prisma/schema.prisma

@@ -1912,7 +1912,8 @@ model TaskEvent {
   createdAt DateTime @default(now())
 
   // This represents the amount of "usage time" the event took, e.g. the CPU time
-  usageDurationMs Int @default(0)
+  usageDurationMs  Int   @default(0)
+  usageCostInCents Float @default(0)
 
   machinePreset           String?
   machinePresetCpu        Float?

references/v3-catalog/src/trigger/usage.ts

@@ -12,7 +12,9 @@ export const usagePlayground = task({
 
     logger.info("Cost and duration", { cost: ctx.run.costInCents, duration: ctx.run.durationMs });
 
-    await new Promise((resolve) => setTimeout(resolve, payload.duration));
+    await logger.trace("Doing some work...", async () => {
+      await new Promise((resolve) => setTimeout(resolve, payload.duration));
+    });
 
     let currentUsage = usage.getCurrent();