@cloudbase/agent-adapter-langgraph 1.0.1-alpha.27 → 1.0.1-alpha.29

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,
@@ -178,7 +139,7 @@ var LanggraphAgent = class extends import_client.AbstractAgent {
       "Preparing stream input"
     );
     logger.trace?.({ messages, tools: input.tools }, "Full input messages");
-    const langChainMessages = aguiMessagesToLangChain(messages);
+    const langChainMessages = aguiMessagesToLangChain(messages, logger);
     logger.trace?.(
       {
         langChainMessages,
@@ -585,8 +546,55 @@ 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: true
+        };
+        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) {
+function aguiMessagesToLangChain(messages, logger) {
   return messages.map((message) => {
     switch (message.role) {
       case "user":
@@ -598,10 +606,34 @@ function aguiMessagesToLangChain(messages) {
             type: "human"
           };
         } else {
+          const langChainContent = [];
+          for (const part of message.content) {
+            if (part.type === "text") {
+              langChainContent.push({ type: "text", text: part.text });
+            } else if (part.type === "binary") {
+              const imageUrl = convertBinaryToImageUrl(
+                part,
+                message.id,
+                logger
+              );
+              if (imageUrl) {
+                langChainContent.push({
+                  type: "image_url",
+                  image_url: { url: imageUrl }
+                });
+              }
+            }
+          }
+          if (langChainContent.length === 0) {
+            logger?.warn?.(
+              { messageId: message.id },
+              "User message content array resulted in empty LangChain content"
+            );
+          }
           return {
             id: message.id,
             role: message.role,
-            content: message.content.filter((m) => m.type === "text"),
+            content: langChainContent,
             type: "human"
           };
         }
@@ -646,6 +678,22 @@ function isValidJson(json) {
     return false;
   }
 }
+function convertBinaryToImageUrl(part, messageId, logger) {
+  if (!part.mimeType.startsWith("image/")) {
+    logger?.debug?.(
+      { mimeType: part.mimeType, messageId },
+      "Skipping non-image binary content"
+    );
+    return void 0;
+  }
+  if (part.url) {
+    return part.url;
+  }
+  if (part.data) {
+    return `data:${part.mimeType};base64,${part.data}`;
+  }
+  return void 0;
+}
 
 // src/checkpoint.ts
 var import_langgraph2 = require("@langchain/langgraph");