fix: support request-scoped rollout logging config (#449)

Allow TypeScript rollout loggers to use per-request gateway URLs and API keys so remote rollout servers can log status to the same tracing environment used for model calls.

Made-with: Cursor

Author: xzrderek

typescript/logging/fireworks-transport.ts

@@ -45,6 +45,7 @@ export class FireworksTransport extends Transport {
 
   constructor(opts: {
     gatewayBaseUrl?: string;
+    apiKey?: string;
     rolloutIdEnv?: string;
     waitUntil?: (promise: Promise<any>) => void;
   } = {}) {
@@ -56,7 +57,7 @@ export class FireworksTransport extends Transport {
       'https://tracing.fireworks.ai';
 
     this.rolloutIdEnv = opts.rolloutIdEnv || 'EP_ROLLOUT_ID';
-    this.apiKey = process.env.FIREWORKS_API_KEY;
+    this.apiKey = opts.apiKey || process.env.FIREWORKS_API_KEY;
     this.waitUntil = opts.waitUntil;
   }
 

typescript/logging/logger.ts

@@ -5,6 +5,12 @@
 import winston from 'winston';
 import { FireworksTransport } from './fireworks-transport.js';
 
+type RolloutLoggerOptions = {
+  gatewayBaseUrl?: string;
+  apiKey?: string;
+  name?: string;
+};
+
 // Global reference to waitUntil function
 let globalWaitUntil: ((promise: Promise<any>) => void) | undefined;
 
@@ -36,9 +42,41 @@ export const logger = winston.createLogger({
 /**
  * Create a child logger with rollout_id context.
  */
-export function createRolloutLogger(rolloutId: string, name: string = 'init'): winston.Logger {
-  return logger.child({
+export function createRolloutLogger(
+  rolloutId: string,
+  nameOrOptions: string | RolloutLoggerOptions = 'init'
+): winston.Logger {
+  const options = typeof nameOrOptions === 'string' ? { name: nameOrOptions } : nameOrOptions;
+  const name = options.name || 'init';
+  const defaultMeta = {
     rollout_id: rolloutId,
     logger_name: `${name}.${rolloutId}`
-  });
+  };
+
+  if (options.gatewayBaseUrl || options.apiKey) {
+    return winston.createLogger({
+      level: 'info',
+      format: winston.format.combine(
+        winston.format.timestamp(),
+        winston.format.errors({ stack: true }),
+        winston.format.json()
+      ),
+      defaultMeta,
+      transports: [
+        new winston.transports.Console({
+          format: winston.format.combine(
+            winston.format.colorize(),
+            winston.format.simple()
+          )
+        }),
+        new FireworksTransport({
+          gatewayBaseUrl: options.gatewayBaseUrl,
+          apiKey: options.apiKey,
+          waitUntil: (promise: Promise<any>) => globalWaitUntil?.(promise)
+        })
+      ]
+    });
+  }
+
+  return logger.child(defaultMeta);
 }