@cloudbase/agent-adapter-langgraph 1.0.1-alpha.10 → 1.0.1-alpha.11

package/dist/index.js

@@ -2042,7 +2042,7 @@ var langchain_exports = {};
 __export(langchain_exports, {
   CallbackHandler: () => CallbackHandler
 });
-var import_base, import_messages, CallbackHandler;
+var import_base, import_messages, import_agent_shared, CallbackHandler;
 var init_langchain = __esm({
   "../../observability/dist/langchain.mjs"() {
     "use strict";
@@ -2050,6 +2050,7 @@ var init_langchain = __esm({
     init_chunk_NFEGQTCC();
     import_base = require("@langchain/core/callbacks/base");
     import_messages = require("@langchain/core/messages");
+    import_agent_shared = require("@cloudbase/agent-shared");
     init_src();
     CallbackHandler = class extends import_base.BaseCallbackHandler {
       constructor(params) {
@@ -2068,17 +2069,17 @@ var init_langchain = __esm({
         __publicField(this, "externalParentSpanContext");
         // Adapter name for ROOT span prefix
         __publicField(this, "adapterName");
+        // Logger for debug output (defaults to noopLogger for silent operation)
+        __publicField(this, "logger");
         this.sessionId = params?.sessionId;
         this.userId = params?.userId;
         this.tags = params?.tags ?? [];
         this.traceMetadata = params?.traceMetadata;
         this.version = params?.version;
         this.adapterName = params?.adapterName;
+        this.logger = params?.logger ?? import_agent_shared.noopLogger;
         this.promptToParentRunMap = /* @__PURE__ */ new Map();
       }
-      get logger() {
-        return console;
-      }
       /**
        * Set external parent SpanContext from AG-UI.Server span.
        * This allows the CallbackHandler to link LangChain/LangGraph spans
@@ -2092,13 +2093,13 @@ var init_langchain = __esm({
       }
       async handleLLMNewToken(token, _idx, runId, _parentRunId, _tags, _fields) {
         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] = /* @__PURE__ */ new Date();
         }
       }
       async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name) {
         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";
           this.registerPromptInfo(parentRunId, metadata);
           let finalInput = inputs;
@@ -2135,12 +2136,12 @@ var init_langchain = __esm({
             });
           }
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleAgentAction(action, runId, parentRunId) {
         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,
@@ -2151,23 +2152,23 @@ var init_langchain = __esm({
             asType: "tool"
           });
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleAgentEnd(action, runId, _parentRunId) {
         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));
         }
       }
       async handleChainError(err, runId, _parentRunId) {
         try {
-          this.logger.debug(`Chain error: ${err} with ID: ${runId}`);
+          this.logger.debug?.(`Chain error: ${err} with ID: ${runId}`);
           this.handleObservationEnd({
             runId,
             attributes: {
@@ -2176,11 +2177,11 @@ var init_langchain = __esm({
             }
           });
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleGenerationStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
-        this.logger.debug(
+        this.logger.debug?.(
           `Generation start with ID: ${runId} and parentRunId ${parentRunId}`
         );
         const runName = name ?? llm.id.at(-1)?.toString() ?? "Langchain Generation";
@@ -2226,7 +2227,7 @@ var init_langchain = __esm({
       }
       async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
         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))
           );
@@ -2241,12 +2242,12 @@ var init_langchain = __esm({
             name
           );
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleChainEnd(outputs, runId, _parentRunId) {
         try {
-          this.logger.debug(`Chain end with ID: ${runId}`);
+          this.logger.debug?.(`Chain end with ID: ${runId}`);
           let finalOutput = outputs;
           if (typeof outputs === "object" && "output" in outputs && typeof outputs["output"] === "string") {
             finalOutput = outputs["output"];
@@ -2265,12 +2266,12 @@ var init_langchain = __esm({
           });
           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));
         }
       }
       async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
         try {
-          this.logger.debug(`LLM start with ID: ${runId}`);
+          this.logger.debug?.(`LLM start with ID: ${runId}`);
           this.handleGenerationStart(
             llm,
             prompts,
@@ -2282,12 +2283,12 @@ var init_langchain = __esm({
             name
           );
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleToolStart(tool, input, runId, parentRunId, tags, metadata, name) {
         try {
-          this.logger.debug(`Tool start with ID: ${runId}`);
+          this.logger.debug?.(`Tool start with ID: ${runId}`);
           this.startAndRegisterObservation({
             runId,
             parentRunId,
@@ -2300,12 +2301,12 @@ var init_langchain = __esm({
             asType: "tool"
           });
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
         try {
-          this.logger.debug(`Retriever start with ID: ${runId}`);
+          this.logger.debug?.(`Retriever start with ID: ${runId}`);
           this.startAndRegisterObservation({
             runId,
             parentRunId,
@@ -2318,12 +2319,12 @@ var init_langchain = __esm({
             asType: "span"
           });
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleRetrieverEnd(documents, runId, _parentRunId) {
         try {
-          this.logger.debug(`Retriever end with ID: ${runId}`);
+          this.logger.debug?.(`Retriever end with ID: ${runId}`);
           this.handleObservationEnd({
             runId,
             attributes: {
@@ -2331,12 +2332,12 @@ var init_langchain = __esm({
             }
           });
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleRetrieverError(err, runId, _parentRunId) {
         try {
-          this.logger.debug(`Retriever error: ${err} with ID: ${runId}`);
+          this.logger.debug?.(`Retriever error: ${err} with ID: ${runId}`);
           this.handleObservationEnd({
             runId,
             attributes: {
@@ -2345,23 +2346,23 @@ var init_langchain = __esm({
             }
           });
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleToolEnd(output, runId, _parentRunId) {
         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));
         }
       }
       async handleToolError(err, runId, _parentRunId) {
         try {
-          this.logger.debug(`Tool error ${err} with ID: ${runId}`);
+          this.logger.debug?.(`Tool error ${err} with ID: ${runId}`);
           this.handleObservationEnd({
             runId,
             attributes: {
@@ -2370,12 +2371,12 @@ var init_langchain = __esm({
             }
           });
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       async handleLLMEnd(output, runId, _parentRunId) {
         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][output.generations[output.generations.length - 1].length - 1];
           const llmUsage = this.extractUsageMetadata(lastResponse) ?? output.llmOutput?.["tokenUsage"];
           const modelName = this.extractModelNameFromMetadata(lastResponse);
@@ -2420,12 +2421,12 @@ var init_langchain = __esm({
             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));
         }
       }
       async handleLLMError(err, runId, _parentRunId) {
         try {
-          this.logger.debug(`LLM error ${err} with ID: ${runId}`);
+          this.logger.debug?.(`LLM error ${err} with ID: ${runId}`);
           this.handleObservationEnd({
             runId,
             attributes: {
@@ -2434,7 +2435,7 @@ var init_langchain = __esm({
             }
           });
         } catch (e) {
-          this.logger.debug(e instanceof Error ? e.message : String(e));
+          this.logger.debug?.(e instanceof Error ? e.message : String(e));
         }
       }
       registerPromptInfo(parentRunId, metadata) {
@@ -2479,7 +2480,7 @@ var init_langchain = __esm({
         const { runId, attributes = {} } = params;
         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;
         }
         observation.update(attributes).end();
@@ -2513,7 +2514,7 @@ var init_langchain = __esm({
           const usageMetadata = "message" in generation && (import_messages.AIMessage.isInstance(generation["message"]) || import_messages.AIMessageChunk.isInstance(generation["message"])) ? generation["message"].usage_metadata : void 0;
           return usageMetadata;
         } catch (err) {
-          this.logger.debug(`Error extracting usage metadata: ${err}`);
+          this.logger.debug?.(`Error extracting usage metadata: ${err}`);
           return;
         }
       }
@@ -2579,8 +2580,8 @@ __export(index_exports, {
   LanggraphAgent: () => LanggraphAgent,
   TDAISaver: () => TDAISaver,
   TDAIStore: () => TDAIStore,
-  createConsoleLogger: () => import_agent_shared2.createConsoleLogger,
-  noopLogger: () => import_agent_shared2.noopLogger
+  createConsoleLogger: () => import_agent_shared3.createConsoleLogger,
+  noopLogger: () => import_agent_shared3.noopLogger
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -2637,7 +2638,7 @@ function convertJsonSchemaToZodSchema(jsonSchema, required) {
 }
 
 // src/agent.ts
-var import_agent_shared = require("@cloudbase/agent-shared");
+var import_agent_shared2 = require("@cloudbase/agent-shared");
 var LangChainCallbackHandler;
 var observabilityLoadAttempted = false;
 var ClientPropertiesAnnotation = import_langgraph.Annotation.Root({
@@ -2652,7 +2653,7 @@ var LanggraphAgent = class extends import_client.AbstractAgent {
     super(agentConfig);
     this.compiledWorkflow = agentConfig.compiledWorkflow;
     this.adapterName = agentConfig.adapterName || "LangGraph";
-    const baseLogger = agentConfig.logger ?? import_agent_shared.noopLogger;
+    const baseLogger = agentConfig.logger ?? import_agent_shared2.noopLogger;
     this.logger = baseLogger.child?.({ component: "langgraph-agent" }) ?? baseLogger;
   }
   run(input) {
@@ -2672,7 +2673,9 @@ var LanggraphAgent = class extends import_client.AbstractAgent {
         LangChainCallbackHandler = obsModule.CallbackHandler;
         if (LangChainCallbackHandler) {
           this.observabilityCallback = new LangChainCallbackHandler({
-            adapterName: this.adapterName
+            adapterName: this.adapterName,
+            logger
+            // Reuse the run-specific logger with context
           });
           logger.debug?.("\u2713 Observability callback created");
         }
@@ -3142,7 +3145,7 @@ var LanggraphAgent = class extends import_client.AbstractAgent {
         { err: error, eventCount, toolCallCount, textChunkCount },
         "Error during stream processing"
       );
-      const errorCode = (0, import_agent_shared.isErrorWithCode)(error) ? error.code : "INTERNAL_ERROR";
+      const errorCode = (0, import_agent_shared2.isErrorWithCode)(error) ? error.code : "INTERNAL_ERROR";
       const errorMessage = error instanceof Error ? error.message : String(error);
       subscriber.next({
         type: import_client.EventType.RUN_ERROR,
@@ -3901,7 +3904,7 @@ var TDAIStore = class extends import_langgraph3.BaseStore {
 };
 
 // src/index.ts
-var import_agent_shared2 = require("@cloudbase/agent-shared");
+var import_agent_shared3 = require("@cloudbase/agent-shared");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ClientPropertiesAnnotation,