@cloudbase/agent-observability 1.0.1-alpha.10 → 1.0.1-alpha.11

package/src/langchain/CallbackHandler.ts

@@ -28,6 +28,7 @@ import {
   type ObservationAttributes,
 } from "../index.js";
 import type { SpanContext } from "@opentelemetry/api";
+import { type Logger, noopLogger } from "@cloudbase/agent-shared";
 
 /**
  * Constructor parameters for CallbackHandler.
@@ -41,6 +42,8 @@ type ConstructorParams = {
   version?: string;
   traceMetadata?: Record<string, unknown>;
   adapterName?: string;  // e.g., "LangGraph" or "LangChain"
+  /** Logger for debug output. Defaults to noopLogger (silent). */
+  logger?: Logger;
 };
 
 /**
@@ -104,6 +107,9 @@ export class CallbackHandler extends BaseCallbackHandler {
   // Adapter name for ROOT span prefix
   private adapterName?: string;
 
+  // Logger for debug output (defaults to noopLogger for silent operation)
+  private logger: Logger;
+
   constructor(params?: ConstructorParams) {
     super();
 
@@ -113,14 +119,11 @@ export class CallbackHandler extends BaseCallbackHandler {
     this.traceMetadata = params?.traceMetadata;
     this.version = params?.version;
     this.adapterName = params?.adapterName;
+    this.logger = params?.logger ?? noopLogger;
 
     this.promptToParentRunMap = new Map<string, PromptInfo>();
   }
 
-  get logger() {
-    return console;
-  }
-
   /**
    * Set external parent SpanContext from AG-UI.Server span.
    * This allows the CallbackHandler to link LangChain/LangGraph spans
@@ -142,7 +145,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     _fields?: any
   ): Promise<void> {
     if (runId && !(runId in this.completionStartTimes)) {
-      this.logger.debug(`LLM first streaming token: ${runId}`);
+      this.logger.debug?.(`LLM first streaming token: ${runId}`);
       this.completionStartTimes[runId] = new Date();
     }
   }
@@ -158,7 +161,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     name?: string
   ): Promise<void> {
     try {
-      this.logger.debug(`Chain start with Id: ${runId}`);
+      this.logger.debug?.(`Chain start with Id: ${runId}`);
 
       const runName = name ?? chain.id.at(-1)?.toString() ?? "Langchain Run";
 
@@ -225,7 +228,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         });
       }
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -235,7 +238,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     parentRunId?: string
   ): Promise<void> {
     try {
-      this.logger.debug(`Agent action ${action.tool} with ID: ${runId}`);
+      this.logger.debug?.(`Agent action ${action.tool} with ID: ${runId}`);
       this.startAndRegisterObservation({
         runId,
         parentRunId,
@@ -246,7 +249,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         asType: "tool",
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -256,13 +259,13 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string
   ): Promise<void> {
     try {
-      this.logger.debug(`Agent finish with ID: ${runId}`);
+      this.logger.debug?.(`Agent finish with ID: ${runId}`);
       this.handleObservationEnd({
         runId,
         attributes: { output: action },
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -272,7 +275,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string | undefined
   ): Promise<void> {
     try {
-      this.logger.debug(`Chain error: ${err} with ID: ${runId}`);
+      this.logger.debug?.(`Chain error: ${err} with ID: ${runId}`);
       this.handleObservationEnd({
         runId,
         attributes: {
@@ -281,7 +284,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         },
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -295,7 +298,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     metadata?: Record<string, unknown> | undefined,
     name?: string
   ): Promise<void> {
-    this.logger.debug(
+    this.logger.debug?.(
       `Generation start with ID: ${runId} and parentRunId ${parentRunId}`
     );
 
@@ -370,7 +373,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     name?: string
   ): Promise<void> {
     try {
-      this.logger.debug(`Chat model start with ID: ${runId}`);
+      this.logger.debug?.(`Chat model start with ID: ${runId}`);
 
       const prompts = messages.flatMap((message) =>
         message.map((m) => this.extractChatMessageContent(m))
@@ -387,7 +390,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         name
       );
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -397,7 +400,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string | undefined
   ): Promise<void> {
     try {
-      this.logger.debug(`Chain end with ID: ${runId}`);
+      this.logger.debug?.(`Chain end with ID: ${runId}`);
 
       let finalOutput: ChainValues | string = outputs;
       if (
@@ -427,7 +430,7 @@ export class CallbackHandler extends BaseCallbackHandler {
       });
       this.deregisterPromptInfo(runId);
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -442,7 +445,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     name?: string
   ): Promise<void> {
     try {
-      this.logger.debug(`LLM start with ID: ${runId}`);
+      this.logger.debug?.(`LLM start with ID: ${runId}`);
       this.handleGenerationStart(
         llm,
         prompts,
@@ -454,7 +457,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         name
       );
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -468,7 +471,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     name?: string
   ): Promise<void> {
     try {
-      this.logger.debug(`Tool start with ID: ${runId}`);
+      this.logger.debug?.(`Tool start with ID: ${runId}`);
       this.startAndRegisterObservation({
         runId,
         parentRunId,
@@ -481,7 +484,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         asType: "tool",
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -495,7 +498,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     name?: string
   ): Promise<void> {
     try {
-      this.logger.debug(`Retriever start with ID: ${runId}`);
+      this.logger.debug?.(`Retriever start with ID: ${runId}`);
       this.startAndRegisterObservation({
         runId,
         parentRunId,
@@ -508,7 +511,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         asType: "span",
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -518,7 +521,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string | undefined
   ): Promise<void> {
     try {
-      this.logger.debug(`Retriever end with ID: ${runId}`);
+      this.logger.debug?.(`Retriever end with ID: ${runId}`);
       this.handleObservationEnd({
         runId,
         attributes: {
@@ -526,7 +529,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         },
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -536,7 +539,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string | undefined
   ): Promise<void> {
     try {
-      this.logger.debug(`Retriever error: ${err} with ID: ${runId}`);
+      this.logger.debug?.(`Retriever error: ${err} with ID: ${runId}`);
       this.handleObservationEnd({
         runId,
         attributes: {
@@ -545,7 +548,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         },
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -555,13 +558,13 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string | undefined
   ): Promise<void> {
     try {
-      this.logger.debug(`Tool end with ID: ${runId}`);
+      this.logger.debug?.(`Tool end with ID: ${runId}`);
       this.handleObservationEnd({
         runId,
         attributes: { output },
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -571,7 +574,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string | undefined
   ): Promise<void> {
     try {
-      this.logger.debug(`Tool error ${err} with ID: ${runId}`);
+      this.logger.debug?.(`Tool error ${err} with ID: ${runId}`);
       this.handleObservationEnd({
         runId,
         attributes: {
@@ -580,7 +583,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         },
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -590,7 +593,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string | undefined
   ): Promise<void> {
     try {
-      this.logger.debug(`LLM end with ID: ${runId}`);
+      this.logger.debug?.(`LLM end with ID: ${runId}`);
 
       const lastResponse =
         output.generations[output.generations.length - 1][
@@ -661,7 +664,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         delete this.completionStartTimes[runId];
       }
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -671,7 +674,7 @@ export class CallbackHandler extends BaseCallbackHandler {
     _parentRunId?: string | undefined
   ): Promise<void> {
     try {
-      this.logger.debug(`LLM error ${err} with ID: ${runId}`);
+      this.logger.debug?.(`LLM error ${err} with ID: ${runId}`);
       this.handleObservationEnd({
         runId,
         attributes: {
@@ -680,7 +683,7 @@ export class CallbackHandler extends BaseCallbackHandler {
         },
       });
     } catch (e) {
-      this.logger.debug(e instanceof Error ? e.message : String(e));
+      this.logger.debug?.(e instanceof Error ? e.message : String(e));
     }
   }
 
@@ -758,7 +761,7 @@ export class CallbackHandler extends BaseCallbackHandler {
 
     const observation = this.runMap.get(runId);
     if (!observation) {
-      this.logger.warn("Observation not found in runMap. Skipping operation.");
+      this.logger.warn?.("Observation not found in runMap. Skipping operation.");
       return;
     }
 
@@ -813,7 +816,7 @@ export class CallbackHandler extends BaseCallbackHandler {
           : undefined;
       return usageMetadata;
     } catch (err) {
-      this.logger.debug(`Error extracting usage metadata: ${err}`);
+      this.logger.debug?.(`Error extracting usage metadata: ${err}`);
       return;
     }
   }