@cloudbase/agent-agents 0.0.2 → 0.0.6

package/dist/index.mjs

@@ -1791,13 +1791,14 @@ init_events();
 import { randomUUID as randomUUID3 } from "crypto";
 import { z as z2 } from "zod/v4";
 var AgentExecutor = class {
-  constructor(config, modelProvider, toolExecutionHandler, messageBuilder, getBusinessState, addMessages) {
+  constructor(config, modelProvider, toolExecutionHandler, messageBuilder, getBusinessState, addMessages, syncState) {
     this.config = config;
     this.modelProvider = modelProvider;
     this.toolExecutionHandler = toolExecutionHandler;
     this.messageBuilder = messageBuilder;
     this.getBusinessState = getBusinessState;
     this.addMessages = addMessages;
+    this.syncState = syncState;
   }
   /**
    * Execute the agent core loop
@@ -2124,6 +2125,10 @@ var AgentExecutor = class {
           [result.messageToSave],
           this.getBusinessState()
         );
+        if (this.syncState && result.messageToSave.metadata?.state) {
+          const updatedState = result.messageToSave.metadata.state;
+          this.syncState(updatedState);
+        }
       }
       if (!result.success) {
         allToolsHandled = false;
@@ -2137,10 +2142,17 @@ var AgentExecutor = class {
   async checkControlFlow(customHandler, context, eventEmitter, startTime, allMessages, totalTokenUsage) {
     if (!customHandler) return null;
     try {
+      const latestUserMessage = allMessages.filter((m) => m.role === "user").pop()?.content || "";
+      const state = this.getBusinessState();
+      const userMessage = state?.userResponse || latestUserMessage;
       const controlDecision = await customHandler.handleNextStep(
         context,
-        this.getBusinessState()
+        state,
+        userMessage
       );
+      if (this.syncState) {
+        this.syncState(state);
+      }
       switch (controlDecision.action) {
         case "continue":
           return null;
@@ -4016,7 +4028,15 @@ var Agent = class {
       this.toolExecutionHandler,
       this.messageBuilder,
       () => this.state.businessState,
-      (conversationId, messages, state) => this.addMessages(conversationId, messages, state)
+      (conversationId, messages, state) => this.addMessages(conversationId, messages, state),
+      // State sync callback: merge updated state into businessState
+      (updatedState) => {
+        if (this.state.businessState && typeof this.state.businessState === "object" && typeof updatedState === "object") {
+          Object.assign(this.state.businessState, updatedState);
+        } else {
+          this.state.businessState = updatedState;
+        }
+      }
     );
   }
   /**