Added data to retrieve token

Author: matt-aitken

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

@@ -18,6 +18,7 @@ import {
   flattenAttributes,
   WaitpointListTokenItem,
   WaitpointTokenStatus,
+  WaitpointRetrieveTokenResponse,
 } from "@trigger.dev/core/v3";
 import { tracer } from "./tracer.js";
 import { conditionallyImportAndParsePacket } from "@trigger.dev/core/v3/utils/ioSerialization";
@@ -200,6 +201,24 @@ async function retrieveToken<T>(
           style: "codepath",
         }),
       },
+      onResponseBody: (body: WaitpointRetrieveTokenResponse, span) => {
+        span.setAttribute("id", body.id);
+        span.setAttribute("status", body.status);
+        if (body.completedAt) {
+          span.setAttribute("completedAt", body.completedAt.toISOString());
+        }
+        if (body.timeoutAt) {
+          span.setAttribute("timeoutAt", body.timeoutAt.toISOString());
+        }
+        if (body.idempotencyKey) {
+          span.setAttribute("idempotencyKey", body.idempotencyKey);
+        }
+        if (body.idempotencyKeyExpiresAt) {
+          span.setAttribute("idempotencyKeyExpiresAt", body.idempotencyKeyExpiresAt.toISOString());
+        }
+        span.setAttribute("tags", body.tags);
+        span.setAttribute("createdAt", body.createdAt.toISOString());
+      },
     },
     requestOptions
   );

references/hello-world/src/trigger/waits.ts

@@ -47,9 +47,11 @@ export const waitToken = task({
     }
 
     const tokens = await wait.listTokens();
-    for await (const token of tokens) {
-      logger.log("Token", token);
-    }
+    await logger.trace("Tokens", async () => {
+      for await (const token of tokens) {
+        logger.log("Token", token);
+      }
+    });
 
     const retrievedToken = await wait.retrieveToken(token.id);
     logger.log("Retrieved token", retrievedToken);
@@ -63,9 +65,11 @@ export const waitToken = task({
     }
 
     const tokens2 = await wait.listTokens({ tags, status: ["COMPLETED"] });
-    for await (const token of tokens2) {
-      logger.log("Token2", token);
-    }
+    await logger.trace("Tokens2", async () => {
+      for await (const token of tokens2) {
+        logger.log("Token2", token);
+      }
+    });
 
     const retrievedToken2 = await wait.retrieveToken(token.id);
     logger.log("Retrieved token2", retrievedToken2);