@cloudbase/agent-adapter-langgraph 0.0.12 → 0.0.13

package/dist/index.mjs

@@ -58,6 +58,7 @@ function convertJsonSchemaToZodSchema(jsonSchema, required) {
 }
 
 // src/agent.ts
+import { noopLogger } from "@cloudbase/agent-shared";
 var ClientPropertiesAnnotation = Annotation.Root({
   tools: Annotation
 });
@@ -69,6 +70,8 @@ var LanggraphAgent = class extends AbstractAgent {
   constructor(agentConfig) {
     super(agentConfig);
     this.compiledWorkflow = agentConfig.compiledWorkflow;
+    const baseLogger = agentConfig.logger ?? noopLogger;
+    this.logger = baseLogger.child?.({ component: "langgraph-agent" }) ?? baseLogger;
   }
   run(input) {
     return new Observable((subscriber) => {
@@ -77,12 +80,29 @@ 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");
+    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 streamEventInput = isResume ? new Command({
       resume: JSON.stringify(input.forwardedProps?.resume?.payload)
     }) : {
       messages: aguiMessagesToLangChain(messages),
@@ -102,6 +122,7 @@ var LanggraphAgent = class extends AbstractAgent {
         thread_id: threadId
       }
     });
+    logger.debug?.("Stream created, starting event processing");
     const chatModelRuns = [];
     const handledToolCallIds = /* @__PURE__ */ new Set();
     for (const msg of messages) {
@@ -109,14 +130,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,6 +164,10 @@ 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,
               message: `Received a message from an unknown chat model run. Run Id: ${event.run_id}`
@@ -134,12 +177,26 @@ 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,
               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}`
@@ -154,29 +211,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 +280,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 +311,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 +361,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 +380,26 @@ 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,
               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,19 +412,40 @@ 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];
+          logger.debug?.(
+            { interruptId: rawInterrupt.id },
+            "Interrupt received"
+          );
           interrupt = {
             id: rawInterrupt.id,
             // TODO: replace with actual reason
@@ -283,23 +454,36 @@ var LanggraphAgent = class extends AbstractAgent {
           };
         }
       }
+      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"
+      );
       subscriber.next({
         type: EventType.RUN_ERROR,
         message: error instanceof Error ? error.message : String(error)
@@ -309,10 +493,12 @@ var LanggraphAgent = class extends AbstractAgent {
   }
   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 +550,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 +1245,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