Honor configured logger across the request lifecycle

packages/runtimeuse-client-python/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "runtimeuse-client"
-version = "0.15.0"
+version = "0.15.1"
 description = "Client library for AI agent runtime communication over WebSocket"
 readme = "README.md"
 license = {"text" = "FSL"}

packages/runtimeuse/package.json

@@ -1,6 +1,6 @@
 {
   "name": "runtimeuse",
-  "version": "0.15.0",
+  "version": "0.15.1",
   "description": "AI agent runtime with WebSocket protocol, artifact handling, and secret management",
   "license": "FSL",
   "type": "module",

packages/runtimeuse/src/index.ts

@@ -56,7 +56,12 @@ export type {
 
 // Utilities
 export { redactSecrets, sleep } from "./utils.js";
-export { createLogger, createRedactingLogger, defaultLogger } from "./logger.js";
+export {
+  createLogger,
+  createPrefixedLogger,
+  createRedactingLogger,
+  defaultLogger,
+} from "./logger.js";
 export type { Logger } from "./logger.js";
 
 // Constants

packages/runtimeuse/src/logger.ts

@@ -7,16 +7,20 @@ export interface Logger {
   debug(...args: unknown[]): void;
 }
 
-export function createLogger(sourceId: string): Logger {
-  const prefix = `[${sourceId}]`;
+export function createPrefixedLogger(inner: Logger, prefix: string): Logger {
+  const tag = `[${prefix}]`;
   return {
-    log: (...args: unknown[]) => console.log(prefix, ...args),
-    warn: (...args: unknown[]) => console.warn(prefix, ...args),
-    error: (...args: unknown[]) => console.error(prefix, ...args),
-    debug: (...args: unknown[]) => console.debug(prefix, ...args),
+    log: (...args: unknown[]) => inner.log(tag, ...args),
+    warn: (...args: unknown[]) => inner.warn(tag, ...args),
+    error: (...args: unknown[]) => inner.error(tag, ...args),
+    debug: (...args: unknown[]) => inner.debug(tag, ...args),
   };
 }
 
+export function createLogger(sourceId: string): Logger {
+  return createPrefixedLogger(defaultLogger, sourceId);
+}
+
 export function createRedactingLogger(
   inner: Logger,
   secrets: string[],

packages/runtimeuse/src/session.test.ts

@@ -296,6 +296,52 @@ describe("WebSocketSession", () => {
     });
   });
 
+  describe("logger", () => {
+    it("routes per-request logs through the configured logger", async () => {
+      const logger = {
+        log: vi.fn(),
+        warn: vi.fn(),
+        error: vi.fn(),
+        debug: vi.fn(),
+      };
+
+      const { session, ws } = createSession({ logger });
+      const done = session.run();
+      sendMessage(ws, INVOCATION_MSG);
+      await waitForTerminal(ws);
+      await endSession(ws, done);
+
+      const logged = logger.log.mock.calls.map((c) => c.join(" "));
+      expect(logged.some((l) => l.includes("[test-source-id] Handling new request"))).toBe(true);
+    });
+
+    it("redacts secrets when routing through the configured logger", async () => {
+      const logger = {
+        log: vi.fn(),
+        warn: vi.fn(),
+        error: vi.fn(),
+        debug: vi.fn(),
+      };
+
+      mockHandlerRun.mockImplementation(async (invocation) => {
+        invocation.logger.log("leak: secret123");
+        return { type: "text", text: "ok" } as AgentResult;
+      });
+
+      const { session, ws } = createSession({ logger });
+      const done = session.run();
+      sendMessage(ws, INVOCATION_MSG);
+      await waitForTerminal(ws);
+      await endSession(ws, done);
+
+      const logged = logger.log.mock.calls.map((c) => c.join(" "));
+      expect(logged.some((l) => l.includes("leak: [REDACTED]"))).toBe(true);
+      for (const line of logged) {
+        expect(line).not.toContain("secret123");
+      }
+    });
+  });
+
   describe("heartbeats", () => {
     it("sends heartbeat messages while a request is in flight", async () => {
       let resolveAgent!: () => void;

packages/runtimeuse/src/session.ts

@@ -15,7 +15,7 @@ import type {
 import { getErrorMessage, serializeErrorMetadata } from "./error-utils.js";
 import { redactSecrets, sleep } from "./utils.js";
 import {
-  createLogger,
+  createPrefixedLogger,
   createRedactingLogger,
   defaultLogger,
   type Logger,
@@ -145,7 +145,11 @@ export class WebSocketSession {
 
     const sourceId = message.source_id ?? crypto.randomUUID();
     this.secrets = message.secrets_to_redact ?? [];
-    this.logger = createRedactingLogger(createLogger(sourceId), this.secrets);
+    const baseLogger = this.config.logger ?? defaultLogger;
+    this.logger = createRedactingLogger(
+      createPrefixedLogger(baseLogger, sourceId),
+      this.secrets,
+    );
     this.config.uploadTracker.setLogger(this.logger);
     this.logger.log("Handling new request");