notte-sdk 0.0.5 → 0.0.6

package/dist/index.js

@@ -1738,6 +1738,15 @@ function prepareAgentRequest(data) {
   }
   return { apiData, zodSchema: null };
 }
+function parseAgentStatusMessage(parsed, agentId) {
+  if (parsed?.status === "agent_stop" && parsed.agent && parsed.agent.agent_id === agentId && parsed.agent.task !== void 0 && parsed.agent.status !== void 0) {
+    return parsed.agent;
+  }
+  if (parsed?.agent_id === agentId && parsed.task !== void 0 && parsed.status !== void 0) {
+    return parsed;
+  }
+  return null;
+}
 function validatePersonaOption(persona, persona_id) {
   if (persona) {
     try {
@@ -1964,19 +1973,19 @@ var Agent = class {
           } catch {
             return;
           }
-          if (parsed.agent_id === agentId && parsed.task !== void 0 && parsed.status !== void 0) {
-            const response = parsed;
+          const finalStatus = parseAgentStatusMessage(parsed, agentId);
+          if (finalStatus) {
             if (updateHandler) {
               updateHandler({
                 type: "completion",
-                data: response,
+                data: finalStatus,
                 timestamp: (/* @__PURE__ */ new Date()).toISOString()
               });
-            } else if (!response.success) {
-              console.error(response.answer);
+            } else if (!finalStatus.success) {
+              console.error(finalStatus.answer);
             }
             this.disconnectWebSocket();
-            resolve(response);
+            resolve(finalStatus);
             return;
           }
           if (log) {
@@ -2625,7 +2634,7 @@ var NottePersona = class {
 };
 
 // src/version.ts
-var SDK_VERSION = true ? "0.0.5" : "dev";
+var SDK_VERSION = true ? "0.0.6" : "dev";
 
 // src/functions.ts
 var NotteFunction = class {