@cloudbase/agent-adapter-langgraph 1.0.1-alpha.7 → 1.0.1-alpha.9

package/dist/index.mjs

@@ -1,3 +1,5 @@
+import "./chunk-NSSMTXJJ.mjs";
+
 // src/agent.ts
 import {
   EventType,
@@ -58,6 +60,9 @@ function convertJsonSchemaToZodSchema(jsonSchema, required) {
 }
 
 // src/agent.ts
+import { noopLogger, isErrorWithCode } from "@cloudbase/agent-shared";
+var LangChainCallbackHandler;
+var observabilityLoadAttempted = false;
 var ClientPropertiesAnnotation = Annotation.Root({
   tools: Annotation
 });
@@ -69,6 +74,9 @@ var LanggraphAgent = class extends AbstractAgent {
   constructor(agentConfig) {
     super(agentConfig);
     this.compiledWorkflow = agentConfig.compiledWorkflow;
+    this.adapterName = agentConfig.adapterName || "LangGraph";
+    const baseLogger = agentConfig.logger ?? noopLogger;
+    this.logger = baseLogger.child?.({ component: "langgraph-agent" }) ?? baseLogger;
   }
   run(input) {
     return new Observable((subscriber) => {
@@ -77,15 +85,85 @@ var LanggraphAgent = class extends AbstractAgent {
   }
   async _run(subscriber, input) {
     const { messages, runId, threadId } = input;
-    subscriber.next({
+    const logger = this.logger.child?.({ runId, threadId }) ?? this.logger;
+    logger.info?.("Run started");
+    if (!observabilityLoadAttempted && !this.observabilityCallback) {
+      observabilityLoadAttempted = true;
+      try {
+        logger.debug?.("Attempting to load observability...");
+        const obsModule = await import("./langchain-NZNM23PL.mjs");
+        LangChainCallbackHandler = obsModule.CallbackHandler;
+        if (LangChainCallbackHandler) {
+          this.observabilityCallback = new LangChainCallbackHandler({
+            adapterName: this.adapterName
+          });
+          logger.debug?.("\u2713 Observability callback created");
+        }
+      } catch (e) {
+        logger.debug?.(
+          "\u2717 Observability not available:",
+          e instanceof Error ? e.message : String(e)
+        );
+      }
+    }
+    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);
+        logger.debug?.("\u2713 Server context restored:", {
+          traceId: serverSpanContext.traceId,
+          spanId: serverSpanContext.spanId
+        });
+      } catch (e) {
+        logger.debug?.("Failed to restore server context:", e);
+      }
+    }
+    const runStartedEvent = {
       type: EventType.RUN_STARTED,
       threadId,
       runId
-    });
-    const streamEventInput = input.forwardedProps?.resume ? new Command({
+    };
+    logger.trace?.({ aguiEvent: runStartedEvent }, "Emitting AGUI event");
+    subscriber.next(runStartedEvent);
+    const isResume = !!input.forwardedProps?.resume;
+    const lastUserMessage = messages.filter((m) => m.role === "user").pop();
+    logger.debug?.(
+      {
+        isResume,
+        messageCount: messages.length,
+        toolCount: input.tools?.length ?? 0,
+        tools: input.tools?.map((t) => t.name),
+        lastUserMessage: typeof lastUserMessage?.content === "string" ? lastUserMessage.content.slice(0, 200) : void 0
+      },
+      "Preparing stream input"
+    );
+    logger.trace?.({ messages, tools: input.tools }, "Full input messages");
+    const langChainMessages = isResume ? void 0 : aguiMessagesToLangChain(messages);
+    if (langChainMessages) {
+      logger.trace?.(
+        {
+          langChainMessages,
+          messageCount: langChainMessages.length,
+          messageTypes: langChainMessages.map((m) => ({
+            type: m.type,
+            id: m.id,
+            hasToolCalls: "tool_calls" in m && Array.isArray(m.tool_calls) && m.tool_calls.length > 0,
+            toolCallId: "tool_call_id" in m ? m.tool_call_id : void 0
+          }))
+        },
+        "Converted LangGraph messages"
+      );
+    }
+    const streamEventInput = isResume ? new Command({
       resume: JSON.stringify(input.forwardedProps?.resume?.payload)
     }) : {
-      messages: aguiMessagesToLangChain(messages),
+      messages: langChainMessages,
       client: {
         tools: convertActionsToDynamicStructuredTools(
           input.tools.map((x) => ({
@@ -97,11 +175,13 @@ var LanggraphAgent = class extends AbstractAgent {
     };
     const stream = this.compiledWorkflow.streamEvents(streamEventInput, {
       version: "v2",
+      ...this.observabilityCallback ? { callbacks: [this.observabilityCallback] } : {},
       runId,
       configurable: {
         thread_id: threadId
       }
     });
+    logger.debug?.("Stream created, starting event processing");
     const chatModelRuns = [];
     const handledToolCallIds = /* @__PURE__ */ new Set();
     for (const msg of messages) {
@@ -109,14 +189,32 @@ var LanggraphAgent = class extends AbstractAgent {
         handledToolCallIds.add(msg.toolCallId);
       }
     }
+    if (handledToolCallIds.size > 0) {
+      logger.debug?.(
+        { count: handledToolCallIds.size },
+        "Pre-populated handled tool call IDs from input messages"
+      );
+    }
     let interrupt;
     let currentToolCall = null;
+    let eventCount = 0;
+    let toolCallCount = 0;
+    let textChunkCount = 0;
     try {
       for await (const event of stream) {
+        eventCount++;
+        logger.trace?.(
+          { eventType: event.event, eventCount, langGraphEvent: event },
+          "Processing stream event"
+        );
         if (event.event.startsWith("ChannelWrite<")) {
           continue;
         }
         if (event.event === "on_chat_model_start") {
+          logger.debug?.(
+            { chatModelRunId: event.run_id },
+            "Chat model started"
+          );
           chatModelRuns.push({ runId: event.run_id });
           continue;
         }
@@ -125,8 +223,13 @@ var LanggraphAgent = class extends AbstractAgent {
             (run) => run.runId === event.run_id
           );
           if (!chatModelRun) {
+            logger.warn?.(
+              { chatModelRunId: event.run_id },
+              "Received message from unknown chat model run"
+            );
             subscriber.next({
               type: EventType.RUN_ERROR,
+              code: "INTERNAL_ERROR",
               message: `Received a message from an unknown chat model run. Run Id: ${event.run_id}`
             });
             continue;
@@ -134,14 +237,29 @@ var LanggraphAgent = class extends AbstractAgent {
           const chunkId = event.data.chunk.id;
           if (!chatModelRun.messageId) {
             chatModelRun.messageId = chunkId;
-            subscriber.next({
+            const textStartEvent = {
               messageId: chunkId,
               type: EventType.TEXT_MESSAGE_START,
               role: "assistant"
-            });
+            };
+            logger.debug?.({ messageId: chunkId }, "Text message started");
+            logger.trace?.(
+              { aguiEvent: textStartEvent },
+              "Emitting AGUI event"
+            );
+            subscriber.next(textStartEvent);
           } else if (chatModelRun.messageId !== chunkId) {
+            logger.warn?.(
+              {
+                expectedMessageId: chatModelRun.messageId,
+                receivedMessageId: chunkId,
+                chatModelRunId: event.run_id
+              },
+              "Received message with unexpected ID"
+            );
             subscriber.next({
               type: EventType.RUN_ERROR,
+              code: "INTERNAL_ERROR",
               message: `Received a message of unknown message id from current run. Run Id: ${event.run_id} Message Id from current run: ${chatModelRun.messageId} Message Id from received message: ${chunkId}`
             });
             continue;
@@ -154,29 +272,68 @@ var LanggraphAgent = class extends AbstractAgent {
             })).forEach((toolCall) => {
               if (currentToolCall) {
                 if (toolCall.id && currentToolCall.id !== toolCall.id) {
-                  subscriber.next({
+                  const toolEndEvent = {
                     toolCallId: currentToolCall.id,
                     type: EventType.TOOL_CALL_END
-                  });
+                  };
+                  logger.debug?.(
+                    {
+                      toolCallId: currentToolCall.id,
+                      toolCallName: currentToolCall.name
+                    },
+                    "Tool call ended"
+                  );
+                  logger.trace?.(
+                    { aguiEvent: toolEndEvent },
+                    "Emitting AGUI event"
+                  );
+                  subscriber.next(toolEndEvent);
                   if (toolCall.name && toolCall.id) {
                     currentToolCall = toolCall;
-                    subscriber.next({
+                    toolCallCount++;
+                    const toolStartEvent = {
                       toolCallId: currentToolCall.id,
                       toolCallName: currentToolCall.name,
                       parentMessageId,
                       type: EventType.TOOL_CALL_START
-                    });
+                    };
+                    logger.debug?.(
+                      {
+                        toolCallId: toolCall.id,
+                        toolCallName: toolCall.name
+                      },
+                      "Tool call started"
+                    );
+                    logger.trace?.(
+                      { aguiEvent: toolStartEvent },
+                      "Emitting AGUI event"
+                    );
+                    subscriber.next(toolStartEvent);
                     if (currentToolCall.args) {
-                      subscriber.next({
+                      const toolArgsEvent = {
                         toolCallId: currentToolCall.id,
                         delta: currentToolCall.args,
                         type: EventType.TOOL_CALL_ARGS
-                      });
+                      };
+                      logger.trace?.(
+                        { aguiEvent: toolArgsEvent },
+                        "Emitting AGUI event"
+                      );
+                      subscriber.next(toolArgsEvent);
                       if (isValidJson(currentToolCall.args)) {
-                        subscriber.next({
+                        const toolEndEvent2 = {
                           toolCallId: currentToolCall.id,
                           type: EventType.TOOL_CALL_END
-                        });
+                        };
+                        logger.debug?.(
+                          { toolCallId: currentToolCall.id },
+                          "Tool call ended (args complete)"
+                        );
+                        logger.trace?.(
+                          { aguiEvent: toolEndEvent2 },
+                          "Emitting AGUI event"
+                        );
+                        subscriber.next(toolEndEvent2);
                         currentToolCall = null;
                       }
                     }
@@ -184,16 +341,30 @@ var LanggraphAgent = class extends AbstractAgent {
                 } else {
                   if (toolCall.args) {
                     currentToolCall.args += toolCall.args;
-                    subscriber.next({
+                    const toolArgsEvent = {
                       toolCallId: currentToolCall.id,
                       delta: toolCall.args,
                       type: EventType.TOOL_CALL_ARGS
-                    });
+                    };
+                    logger.trace?.(
+                      { aguiEvent: toolArgsEvent },
+                      "Emitting AGUI event"
+                    );
+                    subscriber.next(toolArgsEvent);
                     if (isValidJson(currentToolCall.args)) {
-                      subscriber.next({
+                      const toolEndEvent = {
                         toolCallId: currentToolCall.id,
                         type: EventType.TOOL_CALL_END
-                      });
+                      };
+                      logger.debug?.(
+                        { toolCallId: currentToolCall.id },
+                        "Tool call ended (args complete)"
+                      );
+                      logger.trace?.(
+                        { aguiEvent: toolEndEvent },
+                        "Emitting AGUI event"
+                      );
+                      subscriber.next(toolEndEvent);
                       currentToolCall = null;
                     }
                   }
@@ -201,23 +372,47 @@ var LanggraphAgent = class extends AbstractAgent {
               } else {
                 if (toolCall.name && toolCall.id) {
                   currentToolCall = toolCall;
-                  subscriber.next({
+                  toolCallCount++;
+                  const toolStartEvent = {
                     toolCallId: toolCall.id,
                     toolCallName: toolCall.name,
                     parentMessageId,
                     type: EventType.TOOL_CALL_START
-                  });
+                  };
+                  logger.debug?.(
+                    { toolCallId: toolCall.id, toolCallName: toolCall.name },
+                    "Tool call started"
+                  );
+                  logger.trace?.(
+                    { aguiEvent: toolStartEvent },
+                    "Emitting AGUI event"
+                  );
+                  subscriber.next(toolStartEvent);
                   if (toolCall.args) {
-                    subscriber.next({
+                    const toolArgsEvent = {
                       toolCallId: toolCall.id,
                       delta: toolCall.args,
                       type: EventType.TOOL_CALL_ARGS
-                    });
+                    };
+                    logger.trace?.(
+                      { aguiEvent: toolArgsEvent },
+                      "Emitting AGUI event"
+                    );
+                    subscriber.next(toolArgsEvent);
                     if (isValidJson(toolCall.args)) {
-                      subscriber.next({
+                      const toolEndEvent = {
                         toolCallId: toolCall.id,
                         type: EventType.TOOL_CALL_END
-                      });
+                      };
+                      logger.debug?.(
+                        { toolCallId: toolCall.id },
+                        "Tool call ended (args complete)"
+                      );
+                      logger.trace?.(
+                        { aguiEvent: toolEndEvent },
+                        "Emitting AGUI event"
+                      );
+                      subscriber.next(toolEndEvent);
                       currentToolCall = null;
                     }
                   }
@@ -227,11 +422,17 @@ var LanggraphAgent = class extends AbstractAgent {
           }
           const delta = event.data.chunk.content;
           if (typeof delta === "string" && delta) {
-            subscriber.next({
+            textChunkCount++;
+            const textContentEvent = {
               messageId: chatModelRun.messageId,
               type: EventType.TEXT_MESSAGE_CONTENT,
               delta
-            });
+            };
+            logger.trace?.(
+              { aguiEvent: textContentEvent },
+              "Emitting AGUI event"
+            );
+            subscriber.next(textContentEvent);
           }
           continue;
         }
@@ -240,16 +441,27 @@ var LanggraphAgent = class extends AbstractAgent {
             (run) => run.runId === event.run_id
           );
           if (!chatModelRun) {
+            logger.warn?.(
+              { chatModelRunId: event.run_id },
+              "Received on_chat_model_end from unknown run"
+            );
             subscriber.next({
               type: EventType.RUN_ERROR,
+              code: "INTERNAL_ERROR",
               message: `Received a on_chat_model_end event from an unknown chat model run. Run Id: ${event.run_id}`
             });
             continue;
           }
-          subscriber.next({
+          const textEndEvent = {
             type: EventType.TEXT_MESSAGE_END,
             messageId: chatModelRun.messageId
-          });
+          };
+          logger.debug?.(
+            { messageId: chatModelRun.messageId },
+            "Text message ended"
+          );
+          logger.trace?.({ aguiEvent: textEndEvent }, "Emitting AGUI event");
+          subscriber.next(textEndEvent);
           continue;
         }
         if (event.event === "on_tool_end") {
@@ -262,57 +474,115 @@ var LanggraphAgent = class extends AbstractAgent {
                   toolMessage.lc_kwargs.id = toolMessage.id;
                 }
               }
-              subscriber.next({
+              const toolResultEvent = {
                 toolCallId: toolMessage.tool_call_id,
                 type: EventType.TOOL_CALL_RESULT,
                 content: typeof toolMessage.content === "string" ? toolMessage.content : JSON.stringify(toolMessage.content),
                 messageId: toolMessage.id
-              });
+              };
+              logger.debug?.(
+                {
+                  toolCallId: toolMessage.tool_call_id,
+                  messageId: toolMessage.id
+                },
+                "Tool call result received"
+              );
+              logger.trace?.(
+                { aguiEvent: toolResultEvent },
+                "Emitting AGUI event"
+              );
+              subscriber.next(toolResultEvent);
               handledToolCallIds.add(toolMessage.tool_call_id);
+            } else {
+              logger.trace?.(
+                { toolCallId: toolMessage.tool_call_id },
+                "Skipping duplicate tool call result"
+              );
             }
           }
           continue;
         }
-        if (event.event === "on_chain_stream" && event.data.chunk?.__interrupt__ && Array.isArray(event.data.chunk.__interrupt__) && event.data.chunk.__interrupt__.length > 0) {
-          const rawInterrupt = event.data.chunk.__interrupt__[0];
-          interrupt = {
-            id: rawInterrupt.id,
-            // TODO: replace with actual reason
-            reason: "agent requested interrupt",
-            payload: rawInterrupt.value
-          };
+        if (event.event === "on_chain_stream") {
+          const chunk = event.data.chunk;
+          if (chunk?.messages && Array.isArray(chunk.messages)) {
+            logger.trace?.(
+              {
+                nodeName: event.name,
+                stateMessages: chunk.messages.map((m) => ({
+                  type: m.type,
+                  id: m.id,
+                  contentPreview: typeof m.content === "string" ? m.content.slice(0, 100) : "[non-string content]",
+                  hasToolCalls: "tool_calls" in m && Array.isArray(m.tool_calls) && m.tool_calls.length > 0,
+                  toolCallId: "tool_call_id" in m ? m.tool_call_id : void 0
+                })),
+                messageCount: chunk.messages.length
+              },
+              "State update with messages"
+            );
+          }
+          if (chunk?.__interrupt__ && Array.isArray(chunk.__interrupt__) && chunk.__interrupt__.length > 0) {
+            const rawInterrupt = chunk.__interrupt__[0];
+            logger.debug?.(
+              { interruptId: rawInterrupt.id },
+              "Interrupt received"
+            );
+            interrupt = {
+              id: rawInterrupt.id,
+              // TODO: replace with actual reason
+              reason: "agent requested interrupt",
+              payload: rawInterrupt.value
+            };
+          }
         }
       }
+      const stats = { eventCount, toolCallCount, textChunkCount };
       if (interrupt) {
-        subscriber.next({
+        const runFinishedEvent = {
           type: EventType.RUN_FINISHED,
           threadId,
           runId,
           outcome: "interrupt",
           interrupt
-        });
+        };
+        logger.info?.(
+          { outcome: "interrupt", interruptId: interrupt.id, ...stats },
+          "Run finished with interrupt"
+        );
+        logger.trace?.({ aguiEvent: runFinishedEvent }, "Emitting AGUI event");
+        subscriber.next(runFinishedEvent);
       } else {
-        subscriber.next({
+        const runFinishedEvent = {
           type: EventType.RUN_FINISHED,
           threadId,
           runId
-        });
+        };
+        logger.info?.({ outcome: "complete", ...stats }, "Run finished");
+        logger.trace?.({ aguiEvent: runFinishedEvent }, "Emitting AGUI event");
+        subscriber.next(runFinishedEvent);
       }
     } catch (error) {
-      console.error("[LanggraphAgent] Error during stream processing:", error);
+      logger.error?.(
+        { err: error, eventCount, toolCallCount, textChunkCount },
+        "Error during stream processing"
+      );
+      const errorCode = isErrorWithCode(error) ? error.code : "INTERNAL_ERROR";
+      const errorMessage = error instanceof Error ? error.message : String(error);
       subscriber.next({
         type: EventType.RUN_ERROR,
-        message: error instanceof Error ? error.message : String(error)
+        code: errorCode,
+        message: errorMessage
       });
     }
     subscriber.complete();
   }
   clone() {
     const workflow = this.compiledWorkflow;
+    const logger = this.logger;
     this.compiledWorkflow = void 0;
     const cloned = super.clone();
     this.compiledWorkflow = workflow;
     cloned.compiledWorkflow = workflow;
+    cloned.logger = logger;
     return cloned;
   }
 };
@@ -364,8 +634,7 @@ function aguiMessagesToLangChain(messages) {
           id: message.id
         };
       default:
-        console.error(`Message role ${message.role} is not implemented`);
-        throw new Error("message role is not supported.");
+        throw new Error(`Message role ${message.role} is not supported.`);
     }
   });
 }
@@ -1060,11 +1329,16 @@ var TDAIStore = class extends BaseStore {
     }
   }
 };
+
+// src/index.ts
+import { noopLogger as noopLogger2, createConsoleLogger } from "@cloudbase/agent-shared";
 export {
   ClientPropertiesAnnotation,
   ClientStateAnnotation,
   LanggraphAgent,
   TDAISaver,
-  TDAIStore
+  TDAIStore,
+  createConsoleLogger,
+  noopLogger2 as noopLogger
 };
 //# sourceMappingURL=index.mjs.map