@cloudbase/agent-adapter-langgraph 1.0.1-alpha.26 → 1.0.1-alpha.28

package/dist/index.d.mts

@@ -82,6 +82,12 @@ declare class LanggraphAgent extends AbstractAgent {
     }>;
     _run(subscriber: Subscriber<BaseEvent>, input: RunAgentInput): Promise<void>;
     clone(): LanggraphAgent;
+    /**
+     * Setup observability for agent execution.
+     * Lazy loads observability callback, restores server context from forwardedProps,
+     * and configures the callback for graph execution.
+     */
+    private setupObservability;
 }
 
 type PendingWrite = [string, any];

package/dist/index.d.ts

@@ -82,6 +82,12 @@ declare class LanggraphAgent extends AbstractAgent {
     }>;
     _run(subscriber: Subscriber<BaseEvent>, input: RunAgentInput): Promise<void>;
     clone(): LanggraphAgent;
+    /**
+     * Setup observability for agent execution.
+     * Lazy loads observability callback, restores server context from forwardedProps,
+     * and configures the callback for graph execution.
+     */
+    private setupObservability;
 }
 
 type PendingWrite = [string, any];

package/dist/index.js

@@ -120,46 +120,7 @@ var LanggraphAgent = class extends import_client.AbstractAgent {
     const { messages, runId, threadId } = input;
     const logger = this.logger.child?.({ runId, threadId }) ?? this.logger;
     logger.info?.("Run started");
-    if (!LangChainCallbackHandler) {
-      try {
-        logger.debug?.("Attempting to load observability...");
-        const obsModule = await import("@cloudbase/agent-observability/langchain");
-        LangChainCallbackHandler = obsModule.CallbackHandler;
-        logger.debug?.("\u2713 Observability handler loaded");
-      } catch (e) {
-        logger.debug?.(
-          "\u2717 Observability not available:",
-          e instanceof Error ? e.message : String(e)
-        );
-      }
-    }
-    this.observabilityCallback = LangChainCallbackHandler ? new LangChainCallbackHandler({
-      adapterName: this.adapterName,
-      logger
-    }) : void 0;
-    if (this.observabilityCallback && input.forwardedProps?.__agui_server_context) {
-      try {
-        const serverContextData = input.forwardedProps.__agui_server_context;
-        const serverSpanContext = {
-          traceId: serverContextData.traceId,
-          spanId: serverContextData.spanId,
-          traceFlags: serverContextData.traceFlags,
-          isRemote: false
-        };
-        this.observabilityCallback.setExternalParentContext(serverSpanContext, {
-          threadId: input.threadId,
-          runId: input.runId
-        });
-        logger.debug?.("\u2713 Server context restored:", {
-          traceId: serverSpanContext.traceId,
-          spanId: serverSpanContext.spanId,
-          threadId: input.threadId,
-          runId: input.runId
-        });
-      } catch (e) {
-        logger.debug?.("Failed to restore server context:", e);
-      }
-    }
+    await this.setupObservability(input, logger);
     const runStartedEvent = {
       type: import_client.EventType.RUN_STARTED,
       threadId,
@@ -585,6 +546,53 @@ var LanggraphAgent = class extends import_client.AbstractAgent {
     cloned.logger = logger;
     return cloned;
   }
+  /**
+   * Setup observability for agent execution.
+   * Lazy loads observability callback, restores server context from forwardedProps,
+   * and configures the callback for graph execution.
+   */
+  async setupObservability(input, logger) {
+    if (!LangChainCallbackHandler) {
+      try {
+        logger.debug?.("Attempting to load observability...");
+        const obsModule = await import("@cloudbase/agent-observability/langchain");
+        LangChainCallbackHandler = obsModule.CallbackHandler;
+        logger.debug?.("\u2713 Observability handler loaded");
+      } catch (e) {
+        logger.debug?.(
+          "\u2717 Observability not available:",
+          e instanceof Error ? e.message : String(e)
+        );
+      }
+    }
+    this.observabilityCallback = LangChainCallbackHandler ? new LangChainCallbackHandler({
+      adapterName: this.adapterName,
+      logger
+    }) : void 0;
+    if (this.observabilityCallback && input.forwardedProps?.__agui_server_context) {
+      try {
+        const serverContextData = input.forwardedProps.__agui_server_context;
+        const serverSpanContext = {
+          traceId: serverContextData.traceId,
+          spanId: serverContextData.spanId,
+          traceFlags: serverContextData.traceFlags,
+          isRemote: false
+        };
+        this.observabilityCallback.setExternalParentContext(serverSpanContext, {
+          threadId: input.threadId,
+          runId: input.runId
+        });
+        logger.debug?.("\u2713 Server context restored:", {
+          traceId: serverSpanContext.traceId,
+          spanId: serverSpanContext.spanId,
+          threadId: input.threadId,
+          runId: input.runId
+        });
+      } catch (e) {
+        logger.debug?.("Failed to restore server context:", e);
+      }
+    }
+  }
 };
 function aguiMessagesToLangChain(messages) {
   return messages.map((message) => {