feat: add tracing spans to all prompt management SDK methods

Author: ericallam

apps/webapp/app/services/logger.server.ts

@@ -48,7 +48,7 @@ function flattenArgs(args: Array<Record<string, unknown> | undefined>) {
 }
 export const logger = new Logger(
   "webapp",
-  (process.env.APP_LOG_LEVEL ?? "debug") as LogLevel,
+  (process.env.APP_LOG_LEVEL ?? "info") as LogLevel,
   ["examples", "output", "connectionString", "payload"],
   sensitiveDataReplacer,
   () => {
@@ -60,7 +60,7 @@ export const logger = new Logger(
 
 export const workerLogger = new Logger(
   "worker",
-  (process.env.APP_LOG_LEVEL ?? "debug") as LogLevel,
+  (process.env.APP_LOG_LEVEL ?? "info") as LogLevel,
   ["examples", "output", "connectionString"],
   sensitiveDataReplacer,
   () => {
@@ -71,7 +71,7 @@ export const workerLogger = new Logger(
 
 export const socketLogger = new Logger(
   "socket",
-  (process.env.APP_LOG_LEVEL ?? "debug") as LogLevel,
+  (process.env.APP_LOG_LEVEL ?? "info") as LogLevel,
   [],
   sensitiveDataReplacer,
   () => {

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

@@ -52,7 +52,7 @@ export class DynamicFlushScheduler<T> {
   private readonly isDroppableEvent?: (item: T) => boolean;
   private isLoadShedding: boolean = false;
 
-  private readonly logger: Logger = new Logger("EventRepo.DynamicFlushScheduler", "debug");
+  private readonly logger: Logger = new Logger("EventRepo.DynamicFlushScheduler", "info");
 
   constructor(config: DynamicFlushSchedulerConfig<T>) {
     this.batchQueue = [];
@@ -329,7 +329,7 @@ export class DynamicFlushScheduler<T> {
         droppedByKind[kind] = count;
       });
 
-      this.logger.info("DynamicFlushScheduler metrics", {
+      this.logger.debug("DynamicFlushScheduler metrics", {
         totalQueuedItems: this.totalQueuedItems,
         batchQueueLength: this.batchQueue.length,
         currentBatchLength: this.currentBatch.length,

internal-packages/run-engine/src/run-queue/index.ts

@@ -2492,7 +2492,7 @@ export class RunQueue {
       return;
     }
 
-    this.logger.info("Scanning concurrency sets for completed runs");
+    this.logger.debug("Scanning concurrency sets for completed runs");
 
     const stats = {
       streamCallbacks: 0,

packages/redis-worker/src/worker.ts

@@ -1024,7 +1024,7 @@ class Worker<TCatalog extends WorkerCatalog> {
     Promise.allSettled(enqueuePromises).then((results) => {
       results.forEach((result) => {
         if (result.status === "fulfilled") {
-          this.logger.info("Enqueued cron job", { result: result.value });
+          this.logger.debug("Enqueued cron job", { result: result.value });
         } else {
           this.logger.error("Failed to enqueue cron job", { reason: result.reason });
         }
@@ -1050,7 +1050,7 @@ class Worker<TCatalog extends WorkerCatalog> {
       availableAt,
     });
 
-    this.logger.info("Enqueued cron job", {
+    this.logger.debug("Enqueued cron job", {
       identifier,
       cron,
       job,

packages/trigger-sdk/src/v3/promptManagement.ts

@@ -1,15 +1,32 @@
 import {
+  accessoryAttributes,
   apiClientManager,
+  SemanticInternalAttributes,
   type ApiRequestOptions,
   type CreatePromptOverrideRequestBody,
   type ListPromptsResponseBody,
   type ListPromptVersionsResponseBody,
   type PromptOkResponseBody,
   type PromptOverrideCreatedResponseBody,
-  type ResolvePromptResponseBody,
   type UpdatePromptOverrideRequestBody,
 } from "@trigger.dev/core/v3";
 import type { ResolvedPrompt } from "./prompt.js";
+import { tracer } from "./tracer.js";
+
+function promptSpanOptions(name: string, slug: string) {
+  return {
+    tracer,
+    name,
+    icon: "tabler-file-text-ai",
+    attributes: {
+      [SemanticInternalAttributes.STYLE_ICON]: "tabler-file-text-ai",
+      ...accessoryAttributes({
+        items: [{ text: slug, variant: "normal" as const }],
+        style: "codepath",
+      }),
+    },
+  };
+}
 
 function makeToAISDKTelemetry(
   slug: string,
@@ -49,11 +66,36 @@ export async function resolvePrompt(
 ): Promise<ResolvedPrompt> {
   const apiClient = apiClientManager.clientOrThrow();
   const vars = variables ?? {};
-  const response = await apiClient.resolvePrompt(slug, {
-    variables: vars,
-    label: options?.label,
-    version: options?.version,
-  });
+  const response = await apiClient.resolvePrompt(
+    slug,
+    {
+      variables: vars,
+      label: options?.label,
+      version: options?.version,
+    },
+    {
+      ...promptSpanOptions("prompts.resolve()", slug),
+      attributes: {
+        [SemanticInternalAttributes.STYLE_ICON]: "tabler-file-text-ai",
+        [SemanticInternalAttributes.ENTITY_TYPE]: "prompt",
+        [SemanticInternalAttributes.ENTITY_ID]: slug,
+        ...accessoryAttributes({
+          items: [{ text: slug, variant: "normal" as const }],
+          style: "codepath",
+        }),
+      },
+      onResponseBody: (body, span) => {
+        span.setAttribute("prompt.version", body.data.version);
+        span.setAttribute("prompt.slug", body.data.slug);
+        span.setAttribute("prompt.labels", body.data.labels.join(", "));
+        if (body.data.model) span.setAttribute("prompt.model", body.data.model);
+        if (body.data.text) span.setAttribute("prompt.text", body.data.text);
+        if (vars && Object.keys(vars).length > 0) {
+          span.setAttribute("prompt.input", JSON.stringify(vars));
+        }
+      },
+    }
+  );
 
   const data = response.data;
   const inputJson = Object.keys(vars).length > 0 ? JSON.stringify(vars) : undefined;
@@ -81,7 +123,12 @@ export function listPrompts(
   requestOptions?: ApiRequestOptions
 ): Promise<ListPromptsResponseBody> {
   const apiClient = apiClientManager.clientOrThrow();
-  return apiClient.listPrompts(requestOptions);
+  return apiClient.listPrompts({
+    ...promptSpanOptions("prompts.list()", "all"),
+    onResponseBody: (body, span) => {
+      span.setAttribute("prompt.count", body.data.length);
+    },
+  });
 }
 
 /** List all versions for a prompt. */
@@ -90,7 +137,13 @@ export function listPromptVersions(
   requestOptions?: ApiRequestOptions
 ): Promise<ListPromptVersionsResponseBody> {
   const apiClient = apiClientManager.clientOrThrow();
-  return apiClient.listPromptVersions(slug, requestOptions);
+  return apiClient.listPromptVersions(slug, {
+    ...promptSpanOptions("prompts.versions()", slug),
+    onResponseBody: (body, span) => {
+      span.setAttribute("prompt.slug", slug);
+      span.setAttribute("prompt.versions.count", body.data.length);
+    },
+  });
 }
 
 /** Promote a code-deployed version to be the current version. */
@@ -100,7 +153,14 @@ export async function promotePromptVersion(
   requestOptions?: ApiRequestOptions
 ): Promise<PromptOkResponseBody> {
   const apiClient = apiClientManager.clientOrThrow();
-  return apiClient.promotePromptVersion(slug, { version }, requestOptions);
+  return apiClient.promotePromptVersion(slug, { version }, {
+    ...promptSpanOptions("prompts.promote()", slug),
+    attributes: {
+      ...promptSpanOptions("prompts.promote()", slug).attributes,
+      "prompt.slug": slug,
+      "prompt.version": version,
+    },
+  });
 }
 
 /** Create an override — a dashboard/API edit that takes priority over the deployed version. */
@@ -110,7 +170,17 @@ export async function createPromptOverride(
   requestOptions?: ApiRequestOptions
 ): Promise<PromptOverrideCreatedResponseBody> {
   const apiClient = apiClientManager.clientOrThrow();
-  return apiClient.createPromptOverride(slug, body, requestOptions);
+  return apiClient.createPromptOverride(slug, body, {
+    ...promptSpanOptions("prompts.createOverride()", slug),
+    attributes: {
+      ...promptSpanOptions("prompts.createOverride()", slug).attributes,
+      "prompt.slug": slug,
+      ...(body.model ? { "prompt.model": body.model } : {}),
+    },
+    onResponseBody: (body, span) => {
+      span.setAttribute("prompt.override.version", body.version);
+    },
+  });
 }
 
 /** Update the active override's content or model. */
@@ -120,7 +190,13 @@ export async function updatePromptOverride(
   requestOptions?: ApiRequestOptions
 ): Promise<PromptOkResponseBody> {
   const apiClient = apiClientManager.clientOrThrow();
-  return apiClient.updatePromptOverride(slug, body, requestOptions);
+  return apiClient.updatePromptOverride(slug, body, {
+    ...promptSpanOptions("prompts.updateOverride()", slug),
+    attributes: {
+      ...promptSpanOptions("prompts.updateOverride()", slug).attributes,
+      "prompt.slug": slug,
+    },
+  });
 }
 
 /** Remove the active override, reverting to the current deployed version. */
@@ -129,7 +205,13 @@ export async function removePromptOverride(
   requestOptions?: ApiRequestOptions
 ): Promise<PromptOkResponseBody> {
   const apiClient = apiClientManager.clientOrThrow();
-  return apiClient.removePromptOverride(slug, requestOptions);
+  return apiClient.removePromptOverride(slug, {
+    ...promptSpanOptions("prompts.removeOverride()", slug),
+    attributes: {
+      ...promptSpanOptions("prompts.removeOverride()", slug).attributes,
+      "prompt.slug": slug,
+    },
+  });
 }
 
 /** Reactivate a previously removed override version. */
@@ -139,5 +221,12 @@ export async function reactivatePromptOverride(
   requestOptions?: ApiRequestOptions
 ): Promise<PromptOkResponseBody> {
   const apiClient = apiClientManager.clientOrThrow();
-  return apiClient.reactivatePromptOverride(slug, { version }, requestOptions);
+  return apiClient.reactivatePromptOverride(slug, { version }, {
+    ...promptSpanOptions("prompts.reactivateOverride()", slug),
+    attributes: {
+      ...promptSpanOptions("prompts.reactivateOverride()", slug).attributes,
+      "prompt.slug": slug,
+      "prompt.version": version,
+    },
+  });
 }