@cloudbase/agent-adapter-langgraph 0.0.9 → 0.0.12

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @cloudbase/agent-adapter-langgraph
 
+## 1.0.1-alpha.6
+
+### Patch Changes
+
+- alpha release 0.1.2-alpha.1
+  - Update all public packages to version 0.1.2-alpha.1
+  - Trigger automated alpha release workflow
+  - Includes latest features and improvements
+
+- Updated dependencies
+  - @cloudbase/agent-agents@1.0.1-alpha.6
+
 ## 1.0.1-alpha.5
 
 ### Patch Changes

package/dist/index.js

@@ -144,37 +144,82 @@ var LanggraphAgent = class extends import_client.AbstractAgent {
     }
     let interrupt;
     let currentToolCall = null;
-    for await (const event of stream) {
-      if (event.event.startsWith("ChannelWrite<")) {
-        continue;
-      }
-      if (event.event === "on_chat_model_start") {
-        chatModelRuns.push({ runId: event.run_id });
-        continue;
-      }
-      if (event.event === "on_chat_model_stream") {
-        if (Array.isArray(event.data.chunk?.tool_call_chunks) && event.data.chunk?.tool_call_chunks?.length > 0) {
-          event.data.chunk.tool_call_chunks.map((x) => ({
-            ...x,
-            args: typeof x.args === "string" ? x.args : x.args ? JSON.stringify(x.args) : ""
-          })).forEach((toolCall) => {
-            if (currentToolCall) {
-              if (toolCall.id && currentToolCall.id !== toolCall.id) {
-                subscriber.next({
-                  toolCallId: currentToolCall.id,
-                  type: import_client.EventType.TOOL_CALL_END
-                });
-                if (toolCall.name && toolCall.id) {
-                  currentToolCall = toolCall;
+    try {
+      for await (const event of stream) {
+        if (event.event.startsWith("ChannelWrite<")) {
+          continue;
+        }
+        if (event.event === "on_chat_model_start") {
+          chatModelRuns.push({ runId: event.run_id });
+          continue;
+        }
+        if (event.event === "on_chat_model_stream") {
+          const chatModelRun = chatModelRuns.find(
+            (run) => run.runId === event.run_id
+          );
+          if (!chatModelRun) {
+            subscriber.next({
+              type: import_client.EventType.RUN_ERROR,
+              message: `Received a message from an unknown chat model run. Run Id: ${event.run_id}`
+            });
+            continue;
+          }
+          const chunkId = event.data.chunk.id;
+          if (!chatModelRun.messageId) {
+            chatModelRun.messageId = chunkId;
+            subscriber.next({
+              messageId: chunkId,
+              type: import_client.EventType.TEXT_MESSAGE_START,
+              role: "assistant"
+            });
+          } else if (chatModelRun.messageId !== chunkId) {
+            subscriber.next({
+              type: import_client.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}`
+            });
+            continue;
+          }
+          if (Array.isArray(event.data.chunk?.tool_call_chunks) && event.data.chunk?.tool_call_chunks?.length > 0) {
+            const parentMessageId = chatModelRun.messageId;
+            event.data.chunk.tool_call_chunks.map((x) => ({
+              ...x,
+              args: typeof x.args === "string" ? x.args : x.args ? JSON.stringify(x.args) : ""
+            })).forEach((toolCall) => {
+              if (currentToolCall) {
+                if (toolCall.id && currentToolCall.id !== toolCall.id) {
                   subscriber.next({
                     toolCallId: currentToolCall.id,
-                    toolCallName: currentToolCall.name,
-                    type: import_client.EventType.TOOL_CALL_START
+                    type: import_client.EventType.TOOL_CALL_END
                   });
-                  if (currentToolCall.args) {
+                  if (toolCall.name && toolCall.id) {
+                    currentToolCall = toolCall;
                     subscriber.next({
                       toolCallId: currentToolCall.id,
-                      delta: currentToolCall.args,
+                      toolCallName: currentToolCall.name,
+                      parentMessageId,
+                      type: import_client.EventType.TOOL_CALL_START
+                    });
+                    if (currentToolCall.args) {
+                      subscriber.next({
+                        toolCallId: currentToolCall.id,
+                        delta: currentToolCall.args,
+                        type: import_client.EventType.TOOL_CALL_ARGS
+                      });
+                      if (isValidJson(currentToolCall.args)) {
+                        subscriber.next({
+                          toolCallId: currentToolCall.id,
+                          type: import_client.EventType.TOOL_CALL_END
+                        });
+                        currentToolCall = null;
+                      }
+                    }
+                  }
+                } else {
+                  if (toolCall.args) {
+                    currentToolCall.args += toolCall.args;
+                    subscriber.next({
+                      toolCallId: currentToolCall.id,
+                      delta: toolCall.args,
                       type: import_client.EventType.TOOL_CALL_ARGS
                     });
                     if (isValidJson(currentToolCall.args)) {
@@ -187,152 +232,110 @@ var LanggraphAgent = class extends import_client.AbstractAgent {
                   }
                 }
               } else {
-                if (toolCall.args) {
-                  currentToolCall.args += toolCall.args;
-                  subscriber.next({
-                    toolCallId: currentToolCall.id,
-                    delta: toolCall.args,
-                    type: import_client.EventType.TOOL_CALL_ARGS
-                  });
-                  if (isValidJson(currentToolCall.args)) {
-                    subscriber.next({
-                      toolCallId: currentToolCall.id,
-                      type: import_client.EventType.TOOL_CALL_END
-                    });
-                    currentToolCall = null;
-                  }
-                }
-              }
-            } else {
-              if (toolCall.name && toolCall.id) {
-                currentToolCall = toolCall;
-                subscriber.next({
-                  toolCallId: toolCall.id,
-                  toolCallName: toolCall.name,
-                  type: import_client.EventType.TOOL_CALL_START
-                });
-                if (toolCall.args) {
+                if (toolCall.name && toolCall.id) {
+                  currentToolCall = toolCall;
                   subscriber.next({
                     toolCallId: toolCall.id,
-                    delta: toolCall.args,
-                    type: import_client.EventType.TOOL_CALL_ARGS
+                    toolCallName: toolCall.name,
+                    parentMessageId,
+                    type: import_client.EventType.TOOL_CALL_START
                   });
-                  if (isValidJson(toolCall.args)) {
+                  if (toolCall.args) {
                     subscriber.next({
                       toolCallId: toolCall.id,
-                      type: import_client.EventType.TOOL_CALL_END
+                      delta: toolCall.args,
+                      type: import_client.EventType.TOOL_CALL_ARGS
                     });
-                    currentToolCall = null;
+                    if (isValidJson(toolCall.args)) {
+                      subscriber.next({
+                        toolCallId: toolCall.id,
+                        type: import_client.EventType.TOOL_CALL_END
+                      });
+                      currentToolCall = null;
+                    }
                   }
                 }
               }
-            }
-          });
-        }
-        const chatModelRun = chatModelRuns.find(
-          (run) => run.runId === event.run_id
-        );
-        if (!chatModelRun) {
-          subscriber.next({
-            type: import_client.EventType.RUN_ERROR,
-            message: `Received a message from an unknown chat model run. Run Id: ${event.run_id}`
-          });
-          continue;
-        }
-        if (!chatModelRun.messageId) {
-          const messageId = event.data.chunk.id;
-          chatModelRun.messageId = messageId;
-          subscriber.next({
-            messageId,
-            type: import_client.EventType.TEXT_MESSAGE_START,
-            role: "assistant"
-          });
+            });
+          }
           const delta = event.data.chunk.content;
-          typeof delta === "string" && delta && subscriber.next({
-            messageId: chatModelRun.messageId,
-            type: import_client.EventType.TEXT_MESSAGE_CONTENT,
-            delta
-          });
+          if (typeof delta === "string" && delta) {
+            subscriber.next({
+              messageId: chatModelRun.messageId,
+              type: import_client.EventType.TEXT_MESSAGE_CONTENT,
+              delta
+            });
+          }
           continue;
-        } else {
-          if (chatModelRun.messageId !== event.data.chunk.id) {
+        }
+        if (event.event === "on_chat_model_end") {
+          const chatModelRun = chatModelRuns.find(
+            (run) => run.runId === event.run_id
+          );
+          if (!chatModelRun) {
             subscriber.next({
               type: import_client.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: ${event.data.chunk.id}`
+              message: `Received a on_chat_model_end event from an unknown chat model run. Run Id: ${event.run_id}`
             });
             continue;
           }
-          const delta = event.data.chunk.content;
-          typeof delta === "string" && delta && subscriber.next({
-            messageId: chatModelRun.messageId,
-            type: import_client.EventType.TEXT_MESSAGE_CONTENT,
-            delta
-          });
-          continue;
-        }
-      }
-      if (event.event === "on_chat_model_end") {
-        const chatModelRun = chatModelRuns.find(
-          (run) => run.runId === event.run_id
-        );
-        if (!chatModelRun) {
           subscriber.next({
-            type: import_client.EventType.RUN_ERROR,
-            message: `Received a on_chat_model_end event from an unknown chat model run. Run Id: ${event.run_id}`
+            type: import_client.EventType.TEXT_MESSAGE_END,
+            messageId: chatModelRun.messageId
           });
           continue;
         }
-        subscriber.next({
-          type: import_client.EventType.TEXT_MESSAGE_END,
-          messageId: chatModelRun.messageId
-        });
-        continue;
-      }
-      if (event.event === "on_chain_end") {
-        const messages2 = event.data.output?.messages;
-        if (Array.isArray(messages2)) {
-          const inputMessages = event.data.input?.messages;
-          const lastInputMessage = inputMessages?.[inputMessages?.length - 1];
-          const messageId = lastInputMessage?.id;
-          const toolCallMessages = messages2.filter((x) => x.id).filter((x) => x?.tool_call_id);
-          toolCallMessages.forEach((x) => {
-            if (handledToolCallIds.has(x.tool_call_id)) {
-              return;
+        if (event.event === "on_tool_end") {
+          const toolMessage = event.data.output;
+          if (toolMessage && toolMessage.tool_call_id) {
+            if (!handledToolCallIds.has(toolMessage.tool_call_id)) {
+              if (!toolMessage.id) {
+                toolMessage.id = crypto.randomUUID();
+                if (toolMessage.lc_kwargs) {
+                  toolMessage.lc_kwargs.id = toolMessage.id;
+                }
+              }
+              subscriber.next({
+                toolCallId: toolMessage.tool_call_id,
+                type: import_client.EventType.TOOL_CALL_RESULT,
+                content: typeof toolMessage.content === "string" ? toolMessage.content : JSON.stringify(toolMessage.content),
+                messageId: toolMessage.id
+              });
+              handledToolCallIds.add(toolMessage.tool_call_id);
             }
-            subscriber.next({
-              toolCallId: x.tool_call_id,
-              type: import_client.EventType.TOOL_CALL_RESULT,
-              content: x.content,
-              messageId
-            });
-            handledToolCallIds.add(x.tool_call_id);
-          });
+          }
           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" && 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 (interrupt) {
+        subscriber.next({
+          type: import_client.EventType.RUN_FINISHED,
+          threadId,
+          runId,
+          outcome: "interrupt",
+          interrupt
+        });
+      } else {
+        subscriber.next({
+          type: import_client.EventType.RUN_FINISHED,
+          threadId,
+          runId
+        });
       }
-    }
-    if (interrupt) {
-      subscriber.next({
-        type: import_client.EventType.RUN_FINISHED,
-        threadId,
-        runId,
-        outcome: "interrupt",
-        interrupt
-      });
-    } else {
+    } catch (error) {
+      console.error("[LanggraphAgent] Error during stream processing:", error);
       subscriber.next({
-        type: import_client.EventType.RUN_FINISHED,
-        threadId,
-        runId
+        type: import_client.EventType.RUN_ERROR,
+        message: error instanceof Error ? error.message : String(error)
       });
     }
     subscriber.complete();