Update @vercel/queue (#498)

Author: VaguelySerious

.changeset/wet-roses-happen.md

@@ -0,0 +1,5 @@
+---
+"@workflow/world-vercel": patch
+---
+
+Update `@vercel/queue` to use new QueueClient class to simplify custom header/path overwrites

packages/world-vercel/src/queue.ts

@@ -1,4 +1,4 @@
-import { handleCallback, send } from '@vercel/queue';
+import { Client } from '@vercel/queue';
 import {
   MessageId,
   type Queue,
@@ -18,21 +18,12 @@ const VERCEL_QUEUE_MAX_VISIBILITY = 82800; // 23 hours in seconds
 export function createQueue(config?: APIConfig): Queue {
   const { baseUrl, usingProxy } = getHttpUrl(config);
   const headers = getHeaders(config);
-  if (usingProxy) {
-    // If we're using a proxy for the Workflow API, we should also go
-    // through the proxy for the queues API.
-    process.env.VERCEL_QUEUE_BASE_URL = `${baseUrl}`;
-    process.env.VERCEL_QUEUE_BASE_PATH = '/queues/v2/messages';
-    if (config?.token) {
-      process.env.VERCEL_QUEUE_TOKEN = config.token;
-    }
-    if (headers) {
-      headers.forEach((value, key) => {
-        const sanitizedKey = key.replaceAll('-', '__');
-        process.env[`VERCEL_QUEUE_HEADER_${sanitizedKey}`] = value;
-      });
-    }
-  }
+  const queueClient = new Client({
+    baseUrl: usingProxy ? baseUrl : undefined,
+    basePath: usingProxy ? '/queues/v2/messages' : undefined,
+    token: usingProxy ? config?.token : undefined,
+    headers: Object.fromEntries(headers.entries()),
+  });
 
   const queue: Queue['queue'] = async (queueName, x, opts) => {
     // zod v3 doesn't have the `encode` method. We only support zod v4 officially,
@@ -51,12 +42,16 @@ export function createQueue(config?: APIConfig): Queue {
       queueName,
     });
     const sanitizedQueueName = queueName.replace(/[^A-Za-z0-9-_]/g, '-');
-    const { messageId } = await send(sanitizedQueueName, encoded, opts);
+    const { messageId } = await queueClient.send(
+      sanitizedQueueName,
+      encoded,
+      opts
+    );
     return { messageId: MessageId.parse(messageId) };
   };
 
   const createQueueHandler: Queue['createQueueHandler'] = (prefix, handler) => {
-    return handleCallback({
+    return queueClient.handleCallback({
       [`${prefix}*`]: {
         default: async (body, meta) => {
           const { payload, queueName } = MessageWrapper.parse(body);

pnpm-lock.yaml

@@ -10,7 +10,7 @@ catalog:
   "@types/node": 22.19.0
   "@vercel/functions": ^3.1.4
   "@vercel/oidc": 3.0.5
-  "@vercel/queue": 0.0.0-alpha.29
+  "@vercel/queue": 0.0.0-alpha.33
   "@vitest/coverage-v8": ^3.2.4
   ai: 5.0.104
   esbuild: ^0.25.11