@positronic/spec 0.0.67 → 0.0.69

package/dist/src/api/brains.js

@@ -123,39 +123,43 @@ function _ts_generator(thisArg, body) {
     }
 }
 import { STATUS, BRAIN_EVENTS } from '@positronic/core';
+import { startBrainRun, readSseUntil } from './helpers.js';
 export var brains = {
     run: /**
    * Test POST /brains/runs - Create a new brain run
    */ function run(fetch, identifier, options) {
         return _async_to_generator(function() {
-            var body, request, response, data, error;
+            var request, response, data, error;
             return _ts_generator(this, function(_state) {
                 switch(_state.label){
                     case 0:
-                        _state.trys.push([
-                            0,
+                        if (!(options && Object.keys(options).length > 0)) return [
                             3,
+                            5
+                        ];
+                        _state.label = 1;
+                    case 1:
+                        _state.trys.push([
+                            1,
+                            4,
                             ,
-                            4
+                            5
                         ]);
-                        body = {
-                            identifier: identifier
-                        };
-                        if (options && Object.keys(options).length > 0) {
-                            body.options = options;
-                        }
                         request = new Request('http://example.com/brains/runs', {
                             method: 'POST',
                             headers: {
                                 'Content-Type': 'application/json'
                             },
-                            body: JSON.stringify(body)
+                            body: JSON.stringify({
+                                identifier: identifier,
+                                options: options
+                            })
                         });
                         return [
                             4,
                             fetch(request)
                         ];
-                    case 1:
+                    case 2:
                         response = _state.sent();
                         if (response.status !== 201) {
                             console.error("POST /brains/runs returned ".concat(response.status, ", expected 201"));
@@ -168,7 +172,7 @@ export var brains = {
                             4,
                             response.json()
                         ];
-                    case 2:
+                    case 3:
                         data = _state.sent();
                         if (!data.brainRunId || typeof data.brainRunId !== 'string') {
                             console.error("Expected brainRunId to be string, got ".concat(_type_of(data.brainRunId)));
@@ -181,16 +185,17 @@ export var brains = {
                             2,
                             data.brainRunId
                         ];
-                    case 3:
+                    case 4:
                         error = _state.sent();
                         console.error("Failed to test POST /brains/runs:", error);
                         return [
                             2,
                             null
                         ];
-                    case 4:
+                    case 5:
                         return [
-                            2
+                            2,
+                            startBrainRun(fetch, identifier)
                         ];
                 }
             });
@@ -1544,45 +1549,26 @@ export var brains = {
    * Requires a brain with a loop step that will pause on a webhook.
    */ function killSuspended(fetch, loopBrainIdentifier, webhookSlug, webhookPayload) {
         return _async_to_generator(function() {
-            var runRequest, runResponse, brainRunId, watchRequest, watchResponse, foundWebhookEvent, reader, decoder, buffer, _ref, value, done, eventEndIndex, message, event, killRequest, killResponse, getRunRequest, getRunResponse, runData, webhookRequest, webhookResponse, webhookResult, finalCheckRequest, finalCheckResponse, finalRunData, error;
+            var brainRunId, watchRequest, watchResponse, earlyTermination, events, foundWebhookEvent, killRequest, killResponse, getRunRequest, getRunResponse, runData, webhookRequest, webhookResponse, webhookResult, finalCheckRequest, finalCheckResponse, finalRunData, error;
             return _ts_generator(this, function(_state) {
                 switch(_state.label){
                     case 0:
                         _state.trys.push([
                             0,
-                            18,
+                            11,
                             ,
-                            19
+                            12
                         ]);
-                        // Step 1: Start the loop brain
-                        runRequest = new Request('http://example.com/brains/runs', {
-                            method: 'POST',
-                            headers: {
-                                'Content-Type': 'application/json'
-                            },
-                            body: JSON.stringify({
-                                identifier: loopBrainIdentifier
-                            })
-                        });
                         return [
                             4,
-                            fetch(runRequest)
+                            startBrainRun(fetch, loopBrainIdentifier)
                         ];
                     case 1:
-                        runResponse = _state.sent();
-                        if (runResponse.status !== 201) {
-                            console.error("POST /brains/runs returned ".concat(runResponse.status, ", expected 201"));
-                            return [
-                                2,
-                                false
-                            ];
-                        }
-                        return [
-                            4,
-                            runResponse.json()
+                        brainRunId = _state.sent();
+                        if (!brainRunId) return [
+                            2,
+                            false
                         ];
-                    case 2:
-                        brainRunId = _state.sent().brainRunId;
                         // Step 2: Watch until WEBHOOK event (brain pauses)
                         watchRequest = new Request("http://example.com/brains/runs/".concat(brainRunId, "/watch"), {
                             method: 'GET'
@@ -1591,7 +1577,7 @@ export var brains = {
                             4,
                             fetch(watchRequest)
                         ];
-                    case 3:
+                    case 2:
                         watchResponse = _state.sent();
                         if (!watchResponse.ok) {
                             console.error("GET /brains/runs/".concat(brainRunId, "/watch returned ").concat(watchResponse.status));
@@ -1600,84 +1586,35 @@ export var brains = {
                                 false
                             ];
                         }
-                        foundWebhookEvent = false;
-                        if (!watchResponse.body) return [
-                            3,
-                            10
-                        ];
-                        reader = watchResponse.body.getReader();
-                        decoder = new TextDecoder();
-                        buffer = '';
-                        _state.label = 4;
-                    case 4:
-                        _state.trys.push([
-                            4,
-                            ,
-                            8,
-                            10
-                        ]);
-                        _state.label = 5;
-                    case 5:
-                        if (!!foundWebhookEvent) return [
-                            3,
-                            7
-                        ];
-                        return [
-                            4,
-                            reader.read()
-                        ];
-                    case 6:
-                        _ref = _state.sent(), value = _ref.value, done = _ref.done;
-                        if (done) return [
-                            3,
-                            7
-                        ];
-                        buffer += decoder.decode(value, {
-                            stream: true
-                        });
-                        eventEndIndex = void 0;
-                        while((eventEndIndex = buffer.indexOf('\n\n')) !== -1){
-                            message = buffer.substring(0, eventEndIndex);
-                            buffer = buffer.substring(eventEndIndex + 2);
-                            if (message.startsWith('data: ')) {
-                                try {
-                                    event = JSON.parse(message.substring(6));
-                                    if (event.type === BRAIN_EVENTS.WEBHOOK) {
-                                        foundWebhookEvent = true;
-                                        break;
-                                    }
-                                    if (event.type === BRAIN_EVENTS.COMPLETE || event.type === BRAIN_EVENTS.ERROR) {
-                                        console.error("Brain completed/errored before WEBHOOK event: ".concat(event.type));
-                                        return [
-                                            2,
-                                            false
-                                        ];
-                                    }
-                                } catch (e) {
-                                // Ignore parse errors
-                                }
-                            }
+                        if (!watchResponse.body) {
+                            console.error('Watch response has no body');
+                            return [
+                                2,
+                                false
+                            ];
                         }
-                        return [
-                            3,
-                            5
-                        ];
-                    case 7:
-                        return [
-                            3,
-                            10
-                        ];
-                    case 8:
+                        earlyTermination = false;
                         return [
                             4,
-                            reader.cancel()
+                            readSseUntil(watchResponse.body, function(event) {
+                                if (event.type === BRAIN_EVENTS.WEBHOOK) return true;
+                                if (event.type === BRAIN_EVENTS.COMPLETE || event.type === BRAIN_EVENTS.ERROR) {
+                                    console.error("Brain completed/errored before WEBHOOK event: ".concat(event.type));
+                                    earlyTermination = true;
+                                    return true;
+                                }
+                                return false;
+                            })
                         ];
-                    case 9:
-                        _state.sent();
-                        return [
-                            7
+                    case 3:
+                        events = _state.sent();
+                        if (earlyTermination) return [
+                            2,
+                            false
                         ];
-                    case 10:
+                        foundWebhookEvent = events.some(function(e) {
+                            return e.type === BRAIN_EVENTS.WEBHOOK;
+                        });
                         if (!foundWebhookEvent) {
                             console.error('Brain did not emit WEBHOOK event');
                             return [
@@ -1693,7 +1630,7 @@ export var brains = {
                             4,
                             fetch(killRequest)
                         ];
-                    case 11:
+                    case 4:
                         killResponse = _state.sent();
                         if (killResponse.status !== 204) {
                             console.error("DELETE /brains/runs/".concat(brainRunId, " returned ").concat(killResponse.status, ", expected 204"));
@@ -1710,7 +1647,7 @@ export var brains = {
                             4,
                             fetch(getRunRequest)
                         ];
-                    case 12:
+                    case 5:
                         getRunResponse = _state.sent();
                         if (!getRunResponse.ok) {
                             console.error("GET /brains/runs/".concat(brainRunId, " returned ").concat(getRunResponse.status));
@@ -1723,7 +1660,7 @@ export var brains = {
                             4,
                             getRunResponse.json()
                         ];
-                    case 13:
+                    case 6:
                         runData = _state.sent();
                         if (runData.status !== STATUS.CANCELLED) {
                             console.error("Expected status to be '".concat(STATUS.CANCELLED, "', got '").concat(runData.status, "'"));
@@ -1745,7 +1682,7 @@ export var brains = {
                             4,
                             fetch(webhookRequest)
                         ];
-                    case 14:
+                    case 7:
                         webhookResponse = _state.sent();
                         // Accept 200/202 - the important thing is it doesn't resume the brain
                         if (!webhookResponse.ok) {
@@ -1759,7 +1696,7 @@ export var brains = {
                             4,
                             webhookResponse.json()
                         ];
-                    case 15:
+                    case 8:
                         webhookResult = _state.sent();
                         // The action should be 'no-match' since webhook registrations were cleared
                         if (webhookResult.action === 'resumed') {
@@ -1777,7 +1714,7 @@ export var brains = {
                             4,
                             fetch(finalCheckRequest)
                         ];
-                    case 16:
+                    case 9:
                         finalCheckResponse = _state.sent();
                         if (!finalCheckResponse.ok) {
                             console.error("Final GET /brains/runs/".concat(brainRunId, " returned ").concat(finalCheckResponse.status));
@@ -1790,7 +1727,7 @@ export var brains = {
                             4,
                             finalCheckResponse.json()
                         ];
-                    case 17:
+                    case 10:
                         finalRunData = _state.sent();
                         if (finalRunData.status !== STATUS.CANCELLED) {
                             console.error("Final status check: expected '".concat(STATUS.CANCELLED, "', got '").concat(finalRunData.status, "'"));
@@ -1803,14 +1740,14 @@ export var brains = {
                             2,
                             true
                         ];
-                    case 18:
+                    case 11:
                         error = _state.sent();
                         console.error("Failed to test kill suspended brain for ".concat(loopBrainIdentifier, ":"), error);
                         return [
                             2,
                             false
                         ];
-                    case 19:
+                    case 12:
                         return [
                             2
                         ];
@@ -1830,45 +1767,26 @@ export var brains = {
    * - WEBHOOK
    */ function watchAgentEvents(fetch, agentBrainIdentifier) {
         return _async_to_generator(function() {
-            var runRequest, runResponse, brainRunId, watchRequest, watchResponse, events, reader, decoder, buffer, done, _ref, value, streamDone, eventEndIndex, message, event, hasAgentStart, agentStartEvent, hasAgentIteration, hasAgentToolCall, webhookIndex, agentWebhookIndex, agentWebhookEvent, error;
+            var brainRunId, watchRequest, watchResponse, events, _tmp, hasAgentStart, agentStartEvent, hasAgentIteration, hasAgentToolCall, webhookIndex, agentWebhookIndex, agentWebhookEvent, error;
             return _ts_generator(this, function(_state) {
                 switch(_state.label){
                     case 0:
                         _state.trys.push([
                             0,
-                            11,
+                            6,
                             ,
-                            12
+                            7
                         ]);
-                        // Start the agent brain
-                        runRequest = new Request('http://example.com/brains/runs', {
-                            method: 'POST',
-                            headers: {
-                                'Content-Type': 'application/json'
-                            },
-                            body: JSON.stringify({
-                                identifier: agentBrainIdentifier
-                            })
-                        });
                         return [
                             4,
-                            fetch(runRequest)
+                            startBrainRun(fetch, agentBrainIdentifier)
                         ];
                     case 1:
-                        runResponse = _state.sent();
-                        if (runResponse.status !== 201) {
-                            console.error("POST /brains/runs returned ".concat(runResponse.status, ", expected 201"));
-                            return [
-                                2,
-                                false
-                            ];
-                        }
-                        return [
-                            4,
-                            runResponse.json()
+                        brainRunId = _state.sent();
+                        if (!brainRunId) return [
+                            2,
+                            false
                         ];
-                    case 2:
-                        brainRunId = _state.sent().brainRunId;
                         // Watch the brain run
                         watchRequest = new Request("http://example.com/brains/runs/".concat(brainRunId, "/watch"), {
                             method: 'GET'
@@ -1877,7 +1795,7 @@ export var brains = {
                             4,
                             fetch(watchRequest)
                         ];
-                    case 3:
+                    case 2:
                         watchResponse = _state.sent();
                         if (!watchResponse.ok) {
                             console.error("GET /brains/runs/".concat(brainRunId, "/watch returned ").concat(watchResponse.status));
@@ -1886,82 +1804,27 @@ export var brains = {
                                 false
                             ];
                         }
-                        // Read SSE events until we get WEBHOOK or COMPLETE/ERROR
-                        events = [];
                         if (!watchResponse.body) return [
                             3,
-                            10
-                        ];
-                        reader = watchResponse.body.getReader();
-                        decoder = new TextDecoder();
-                        buffer = '';
-                        done = false;
-                        _state.label = 4;
-                    case 4:
-                        _state.trys.push([
-                            4,
-                            ,
-                            8,
-                            10
-                        ]);
-                        _state.label = 5;
-                    case 5:
-                        if (!!done) return [
-                            3,
-                            7
+                            4
                         ];
                         return [
                             4,
-                            reader.read()
-                        ];
-                    case 6:
-                        _ref = _state.sent(), value = _ref.value, streamDone = _ref.done;
-                        if (streamDone) return [
-                            3,
-                            7
+                            readSseUntil(watchResponse.body, function(event) {
+                                return event.type === BRAIN_EVENTS.WEBHOOK || event.type === BRAIN_EVENTS.COMPLETE || event.type === BRAIN_EVENTS.ERROR;
+                            })
                         ];
-                        buffer += decoder.decode(value, {
-                            stream: true
-                        });
-                        // Process complete SSE messages
-                        eventEndIndex = void 0;
-                        while((eventEndIndex = buffer.indexOf('\n\n')) !== -1){
-                            message = buffer.substring(0, eventEndIndex);
-                            buffer = buffer.substring(eventEndIndex + 2);
-                            if (message.startsWith('data: ')) {
-                                try {
-                                    event = JSON.parse(message.substring(6));
-                                    events.push(event);
-                                    // Stop on terminal events
-                                    if (event.type === BRAIN_EVENTS.WEBHOOK || event.type === BRAIN_EVENTS.COMPLETE || event.type === BRAIN_EVENTS.ERROR) {
-                                        done = true;
-                                        break;
-                                    }
-                                } catch (e) {
-                                // Ignore parse errors
-                                }
-                            }
-                        }
+                    case 3:
+                        _tmp = _state.sent();
                         return [
                             3,
                             5
                         ];
-                    case 7:
-                        return [
-                            3,
-                            10
-                        ];
-                    case 8:
-                        return [
-                            4,
-                            reader.cancel()
-                        ];
-                    case 9:
-                        _state.sent();
-                        return [
-                            7
-                        ];
-                    case 10:
+                    case 4:
+                        _tmp = [];
+                        _state.label = 5;
+                    case 5:
+                        events = _tmp;
                         // Verify required agent events are present
                         hasAgentStart = events.some(function(e) {
                             return e.type === BRAIN_EVENTS.AGENT_START;
@@ -2040,14 +1903,14 @@ export var brains = {
                             2,
                             true
                         ];
-                    case 11:
+                    case 6:
                         error = _state.sent();
                         console.error("Failed to test agent events for ".concat(agentBrainIdentifier, ":"), error);
                         return [
                             2,
                             false
                         ];
-                    case 12:
+                    case 7:
                         return [
                             2
                         ];
@@ -2067,45 +1930,26 @@ export var brains = {
    * - The webhook slug and identifier to trigger
    */ function agentWebhookResume(fetch, agentBrainIdentifier, webhookSlug, webhookPayload) {
         return _async_to_generator(function() {
-            var runRequest, runResponse, brainRunId, watchRequest, watchResponse, foundWebhookEvent, reader, decoder, buffer, _ref, value, done, eventEndIndex, message, event, webhookRequest, webhookResponse, webhookResult, resumeWatchRequest, resumeWatchResponse, resumeEvents, reader1, decoder1, buffer1, done1, _ref1, value1, streamDone, eventEndIndex1, message1, event1, hasWebhookResponse, hasAgentToolResult, completeEvent, error;
+            var brainRunId, watchRequest, watchResponse, earlyTermination, watchEvents, foundWebhookEvent, webhookRequest, webhookResponse, webhookResult, resumeWatchRequest, resumeWatchResponse, resumeEvents, _tmp, hasWebhookResponse, hasAgentToolResult, completeEvent, error;
             return _ts_generator(this, function(_state) {
                 switch(_state.label){
                     case 0:
                         _state.trys.push([
                             0,
-                            21,
+                            10,
                             ,
-                            22
+                            11
                         ]);
-                        // Step 1: Start the agent brain
-                        runRequest = new Request('http://example.com/brains/runs', {
-                            method: 'POST',
-                            headers: {
-                                'Content-Type': 'application/json'
-                            },
-                            body: JSON.stringify({
-                                identifier: agentBrainIdentifier
-                            })
-                        });
                         return [
                             4,
-                            fetch(runRequest)
+                            startBrainRun(fetch, agentBrainIdentifier)
                         ];
                     case 1:
-                        runResponse = _state.sent();
-                        if (runResponse.status !== 201) {
-                            console.error("POST /brains/runs returned ".concat(runResponse.status, ", expected 201"));
-                            return [
-                                2,
-                                false
-                            ];
-                        }
-                        return [
-                            4,
-                            runResponse.json()
+                        brainRunId = _state.sent();
+                        if (!brainRunId) return [
+                            2,
+                            false
                         ];
-                    case 2:
-                        brainRunId = _state.sent().brainRunId;
                         // Step 2: Watch until WEBHOOK event (brain pauses)
                         watchRequest = new Request("http://example.com/brains/runs/".concat(brainRunId, "/watch"), {
                             method: 'GET'
@@ -2114,7 +1958,7 @@ export var brains = {
                             4,
                             fetch(watchRequest)
                         ];
-                    case 3:
+                    case 2:
                         watchResponse = _state.sent();
                         if (!watchResponse.ok) {
                             console.error("GET /brains/runs/".concat(brainRunId, "/watch returned ").concat(watchResponse.status));
@@ -2123,84 +1967,35 @@ export var brains = {
                                 false
                             ];
                         }
-                        foundWebhookEvent = false;
-                        if (!watchResponse.body) return [
-                            3,
-                            10
-                        ];
-                        reader = watchResponse.body.getReader();
-                        decoder = new TextDecoder();
-                        buffer = '';
-                        _state.label = 4;
-                    case 4:
-                        _state.trys.push([
-                            4,
-                            ,
-                            8,
-                            10
-                        ]);
-                        _state.label = 5;
-                    case 5:
-                        if (!!foundWebhookEvent) return [
-                            3,
-                            7
-                        ];
-                        return [
-                            4,
-                            reader.read()
-                        ];
-                    case 6:
-                        _ref = _state.sent(), value = _ref.value, done = _ref.done;
-                        if (done) return [
-                            3,
-                            7
-                        ];
-                        buffer += decoder.decode(value, {
-                            stream: true
-                        });
-                        eventEndIndex = void 0;
-                        while((eventEndIndex = buffer.indexOf('\n\n')) !== -1){
-                            message = buffer.substring(0, eventEndIndex);
-                            buffer = buffer.substring(eventEndIndex + 2);
-                            if (message.startsWith('data: ')) {
-                                try {
-                                    event = JSON.parse(message.substring(6));
-                                    if (event.type === BRAIN_EVENTS.WEBHOOK) {
-                                        foundWebhookEvent = true;
-                                        break;
-                                    }
-                                    if (event.type === BRAIN_EVENTS.COMPLETE || event.type === BRAIN_EVENTS.ERROR) {
-                                        console.error("Brain completed/errored before WEBHOOK event: ".concat(event.type));
-                                        return [
-                                            2,
-                                            false
-                                        ];
-                                    }
-                                } catch (e) {
-                                // Ignore parse errors
-                                }
-                            }
+                        if (!watchResponse.body) {
+                            console.error('Watch response has no body');
+                            return [
+                                2,
+                                false
+                            ];
                         }
-                        return [
-                            3,
-                            5
-                        ];
-                    case 7:
-                        return [
-                            3,
-                            10
-                        ];
-                    case 8:
+                        earlyTermination = false;
                         return [
                             4,
-                            reader.cancel()
+                            readSseUntil(watchResponse.body, function(event) {
+                                if (event.type === BRAIN_EVENTS.WEBHOOK) return true;
+                                if (event.type === BRAIN_EVENTS.COMPLETE || event.type === BRAIN_EVENTS.ERROR) {
+                                    console.error("Brain completed/errored before WEBHOOK event: ".concat(event.type));
+                                    earlyTermination = true;
+                                    return true;
+                                }
+                                return false;
+                            })
                         ];
-                    case 9:
-                        _state.sent();
-                        return [
-                            7
+                    case 3:
+                        watchEvents = _state.sent();
+                        if (earlyTermination) return [
+                            2,
+                            false
                         ];
-                    case 10:
+                        foundWebhookEvent = watchEvents.some(function(e) {
+                            return e.type === BRAIN_EVENTS.WEBHOOK;
+                        });
                         if (!foundWebhookEvent) {
                             console.error('Brain did not emit WEBHOOK event');
                             return [
@@ -2220,7 +2015,7 @@ export var brains = {
                             4,
                             fetch(webhookRequest)
                         ];
-                    case 11:
+                    case 4:
                         webhookResponse = _state.sent();
                         if (!webhookResponse.ok) {
                             console.error("POST /webhooks/".concat(webhookSlug, " returned ").concat(webhookResponse.status));
@@ -2233,7 +2028,7 @@ export var brains = {
                             4,
                             webhookResponse.json()
                         ];
-                    case 12:
+                    case 5:
                         webhookResult = _state.sent();
                         if (!webhookResult.received) {
                             console.error('Webhook was not received');
@@ -2257,7 +2052,7 @@ export var brains = {
                             4,
                             fetch(resumeWatchRequest)
                         ];
-                    case 13:
+                    case 6:
                         resumeWatchResponse = _state.sent();
                         if (!resumeWatchResponse.ok) {
                             console.error("GET /brains/runs/".concat(brainRunId, "/watch (resume) returned ").concat(resumeWatchResponse.status));
@@ -2266,79 +2061,27 @@ export var brains = {
                                 false
                             ];
                         }
-                        resumeEvents = [];
                         if (!resumeWatchResponse.body) return [
                             3,
-                            20
-                        ];
-                        reader1 = resumeWatchResponse.body.getReader();
-                        decoder1 = new TextDecoder();
-                        buffer1 = '';
-                        done1 = false;
-                        _state.label = 14;
-                    case 14:
-                        _state.trys.push([
-                            14,
-                            ,
-                            18,
-                            20
-                        ]);
-                        _state.label = 15;
-                    case 15:
-                        if (!!done1) return [
-                            3,
-                            17
+                            8
                         ];
                         return [
                             4,
-                            reader1.read()
-                        ];
-                    case 16:
-                        _ref1 = _state.sent(), value1 = _ref1.value, streamDone = _ref1.done;
-                        if (streamDone) return [
-                            3,
-                            17
-                        ];
-                        buffer1 += decoder1.decode(value1, {
-                            stream: true
-                        });
-                        eventEndIndex1 = void 0;
-                        while((eventEndIndex1 = buffer1.indexOf('\n\n')) !== -1){
-                            message1 = buffer1.substring(0, eventEndIndex1);
-                            buffer1 = buffer1.substring(eventEndIndex1 + 2);
-                            if (message1.startsWith('data: ')) {
-                                try {
-                                    event1 = JSON.parse(message1.substring(6));
-                                    resumeEvents.push(event1);
-                                    if (event1.type === BRAIN_EVENTS.COMPLETE || event1.type === BRAIN_EVENTS.ERROR) {
-                                        done1 = true;
-                                        break;
-                                    }
-                                } catch (e) {
-                                // Ignore parse errors
-                                }
-                            }
-                        }
-                        return [
-                            3,
-                            15
+                            readSseUntil(resumeWatchResponse.body, function(event) {
+                                return event.type === BRAIN_EVENTS.COMPLETE || event.type === BRAIN_EVENTS.ERROR;
+                            })
                         ];
-                    case 17:
+                    case 7:
+                        _tmp = _state.sent();
                         return [
                             3,
-                            20
+                            9
                         ];
-                    case 18:
-                        return [
-                            4,
-                            reader1.cancel()
-                        ];
-                    case 19:
-                        _state.sent();
-                        return [
-                            7
-                        ];
-                    case 20:
+                    case 8:
+                        _tmp = [];
+                        _state.label = 9;
+                    case 9:
+                        resumeEvents = _tmp;
                         // Verify WEBHOOK_RESPONSE event is present
                         hasWebhookResponse = resumeEvents.some(function(e) {
                             return e.type === BRAIN_EVENTS.WEBHOOK_RESPONSE;
@@ -2383,14 +2126,14 @@ export var brains = {
                             2,
                             true
                         ];
-                    case 21:
+                    case 10:
                         error = _state.sent();
                         console.error("Failed to test agent webhook resume for ".concat(agentBrainIdentifier, ":"), error);
                         return [
                             2,
                             false
                         ];
-                    case 22:
+                    case 11:
                         return [
                             2
                         ];
@@ -2413,45 +2156,26 @@ export var brains = {
    * 6. Assert final status is COMPLETE
    */ function innerBrainCompleteDoesNotAffectOuterStatus(fetch, outerBrainIdentifier, webhookSlug, webhookPayload) {
         return _async_to_generator(function() {
-            var runRequest, runResponse, brainRunId, watchRequest, watchResponse, foundOuterWebhook, innerCompleteCount, reader, decoder, buffer, _ref, value, done, eventEndIndex, message, event, historyRequest, historyResponse, historyData, ourRun, webhookRequest, webhookResponse, resumeWatchRequest, resumeWatchResponse, foundFinalComplete, depth, reader1, decoder1, buffer1, _ref1, value1, done1, eventEndIndex1, message1, event1, finalHistoryResponse, finalHistoryData, finalRun, error;
+            var brainRunId, watchRequest, watchResponse, innerCompleteCount, hitError, watchEvents, foundOuterWebhook, historyRequest, historyResponse, historyData, ourRun, webhookRequest, webhookResponse, resumeWatchRequest, resumeWatchResponse, depth, finalHistoryResponse, finalHistoryData, finalRun, error;
             return _ts_generator(this, function(_state) {
                 switch(_state.label){
                     case 0:
                         _state.trys.push([
                             0,
-                            24,
+                            12,
                             ,
-                            25
+                            13
                         ]);
-                        // Step 1: Start the outer brain
-                        runRequest = new Request('http://example.com/brains/runs', {
-                            method: 'POST',
-                            headers: {
-                                'Content-Type': 'application/json'
-                            },
-                            body: JSON.stringify({
-                                identifier: outerBrainIdentifier
-                            })
-                        });
                         return [
                             4,
-                            fetch(runRequest)
+                            startBrainRun(fetch, outerBrainIdentifier)
                         ];
                     case 1:
-                        runResponse = _state.sent();
-                        if (runResponse.status !== 201) {
-                            console.error("POST /brains/runs returned ".concat(runResponse.status, ", expected 201"));
-                            return [
-                                2,
-                                false
-                            ];
-                        }
-                        return [
-                            4,
-                            runResponse.json()
+                        brainRunId = _state.sent();
+                        if (!brainRunId) return [
+                            2,
+                            false
                         ];
-                    case 2:
-                        brainRunId = _state.sent().brainRunId;
                         // Step 2: Watch SSE until WEBHOOK event from outer brain (after inner completes)
                         watchRequest = new Request("http://example.com/brains/runs/".concat(brainRunId, "/watch"), {
                             method: 'GET'
@@ -2460,7 +2184,7 @@ export var brains = {
                             4,
                             fetch(watchRequest)
                         ];
-                    case 3:
+                    case 2:
                         watchResponse = _state.sent();
                         if (!watchResponse.ok) {
                             console.error("GET /brains/runs/".concat(brainRunId, "/watch returned ").concat(watchResponse.status));
@@ -2469,91 +2193,39 @@ export var brains = {
                                 false
                             ];
                         }
-                        foundOuterWebhook = false;
+                        if (!watchResponse.body) {
+                            console.error('Watch response has no body');
+                            return [
+                                2,
+                                false
+                            ];
+                        }
                         innerCompleteCount = 0;
-                        if (!watchResponse.body) return [
-                            3,
-                            10
-                        ];
-                        reader = watchResponse.body.getReader();
-                        decoder = new TextDecoder();
-                        buffer = '';
-                        _state.label = 4;
-                    case 4:
-                        _state.trys.push([
-                            4,
-                            ,
-                            8,
-                            10
-                        ]);
-                        _state.label = 5;
-                    case 5:
-                        if (!!foundOuterWebhook) return [
-                            3,
-                            7
-                        ];
+                        hitError = false;
                         return [
                             4,
-                            reader.read()
-                        ];
-                    case 6:
-                        _ref = _state.sent(), value = _ref.value, done = _ref.done;
-                        if (done) return [
-                            3,
-                            7
-                        ];
-                        buffer += decoder.decode(value, {
-                            stream: true
-                        });
-                        eventEndIndex = void 0;
-                        while((eventEndIndex = buffer.indexOf('\n\n')) !== -1){
-                            message = buffer.substring(0, eventEndIndex);
-                            buffer = buffer.substring(eventEndIndex + 2);
-                            if (message.startsWith('data: ')) {
-                                try {
-                                    event = JSON.parse(message.substring(6));
-                                    // Track inner brain completes
-                                    if (event.type === BRAIN_EVENTS.COMPLETE) {
-                                        innerCompleteCount++;
-                                    // First complete is inner brain, second would be outer
-                                    }
-                                    // Outer brain webhook (happens after inner brain completes)
-                                    if (event.type === BRAIN_EVENTS.WEBHOOK) {
-                                        foundOuterWebhook = true;
-                                        break;
-                                    }
-                                    if (event.type === BRAIN_EVENTS.ERROR) {
-                                        console.error("Brain errored: ".concat(JSON.stringify(event.error)));
-                                        return [
-                                            2,
-                                            false
-                                        ];
-                                    }
-                                } catch (e) {
-                                // Ignore parse errors
+                            readSseUntil(watchResponse.body, function(event) {
+                                if (event.type === BRAIN_EVENTS.COMPLETE) {
+                                    innerCompleteCount++;
                                 }
-                            }
-                        }
-                        return [
-                            3,
-                            5
-                        ];
-                    case 7:
-                        return [
-                            3,
-                            10
-                        ];
-                    case 8:
-                        return [
-                            4,
-                            reader.cancel()
+                                if (event.type === BRAIN_EVENTS.WEBHOOK) return true;
+                                if (event.type === BRAIN_EVENTS.ERROR) {
+                                    console.error("Brain errored: ".concat(JSON.stringify(event.error)));
+                                    hitError = true;
+                                    return true;
+                                }
+                                return false;
+                            })
                         ];
-                    case 9:
-                        _state.sent();
-                        return [
-                            7
+                    case 3:
+                        watchEvents = _state.sent();
+                        if (hitError) return [
+                            2,
+                            false
                         ];
-                    case 10:
+                        foundOuterWebhook = watchEvents.some(function(e) {
+                            return e.type === BRAIN_EVENTS.WEBHOOK;
+                        });
                         if (!foundOuterWebhook) {
                             console.error('Did not receive outer brain WEBHOOK event');
                             return [
@@ -2576,7 +2248,7 @@ export var brains = {
                             4,
                             fetch(historyRequest)
                         ];
-                    case 11:
+                    case 4:
                         historyResponse = _state.sent();
                         if (!historyResponse.ok) {
                             console.error("GET /brains/".concat(outerBrainIdentifier, "/history returned ").concat(historyResponse.status));
@@ -2589,7 +2261,7 @@ export var brains = {
                             4,
                             historyResponse.json()
                         ];
-                    case 12:
+                    case 5:
                         historyData = _state.sent();
                         ourRun = historyData.runs.find(function(r) {
                             return r.brainRunId === brainRunId;
@@ -2622,7 +2294,7 @@ export var brains = {
                             4,
                             fetch(webhookRequest)
                         ];
-                    case 13:
+                    case 6:
                         webhookResponse = _state.sent();
                         if (!webhookResponse.ok) {
                             console.error("POST /webhooks/".concat(webhookSlug, " returned ").concat(webhookResponse.status));
@@ -2639,7 +2311,7 @@ export var brains = {
                             4,
                             fetch(resumeWatchRequest)
                         ];
-                    case 14:
+                    case 7:
                         resumeWatchResponse = _state.sent();
                         if (!resumeWatchResponse.ok) {
                             console.error("Resume watch returned ".concat(resumeWatchResponse.status));
@@ -2648,93 +2320,32 @@ export var brains = {
                                 false
                             ];
                         }
-                        // Wait for the OUTER brain's COMPLETE event specifically (by tracking depth)
-                        // When resuming, the SSE stream includes historical events, so we need to
-                        // track START/COMPLETE events to know when the outer brain truly finishes
-                        foundFinalComplete = false;
-                        depth = 0;
                         if (!resumeWatchResponse.body) return [
                             3,
-                            21
-                        ];
-                        reader1 = resumeWatchResponse.body.getReader();
-                        decoder1 = new TextDecoder();
-                        buffer1 = '';
-                        _state.label = 15;
-                    case 15:
-                        _state.trys.push([
-                            15,
-                            ,
-                            19,
-                            21
-                        ]);
-                        _state.label = 16;
-                    case 16:
-                        if (!!foundFinalComplete) return [
-                            3,
-                            18
+                            9
                         ];
+                        depth = 0;
                         return [
                             4,
-                            reader1.read()
-                        ];
-                    case 17:
-                        _ref1 = _state.sent(), value1 = _ref1.value, done1 = _ref1.done;
-                        if (done1) return [
-                            3,
-                            18
-                        ];
-                        buffer1 += decoder1.decode(value1, {
-                            stream: true
-                        });
-                        eventEndIndex1 = void 0;
-                        while((eventEndIndex1 = buffer1.indexOf('\n\n')) !== -1){
-                            message1 = buffer1.substring(0, eventEndIndex1);
-                            buffer1 = buffer1.substring(eventEndIndex1 + 2);
-                            if (message1.startsWith('data: ')) {
-                                try {
-                                    event1 = JSON.parse(message1.substring(6));
-                                    // Track depth to find the outer brain's COMPLETE
-                                    if (event1.type === BRAIN_EVENTS.START) {
-                                        depth++;
-                                    } else if (event1.type === BRAIN_EVENTS.COMPLETE) {
-                                        depth--;
-                                        // When depth reaches 0, the outer brain has completed
-                                        if (depth <= 0) {
-                                            foundFinalComplete = true;
-                                            break;
-                                        }
-                                    }
-                                } catch (e) {
-                                // Ignore parse errors
+                            readSseUntil(resumeWatchResponse.body, function(event) {
+                                if (event.type === BRAIN_EVENTS.START) {
+                                    depth++;
+                                } else if (event.type === BRAIN_EVENTS.COMPLETE) {
+                                    depth--;
+                                    if (depth <= 0) return true;
                                 }
-                            }
-                        }
-                        return [
-                            3,
-                            16
-                        ];
-                    case 18:
-                        return [
-                            3,
-                            21
-                        ];
-                    case 19:
-                        return [
-                            4,
-                            reader1.cancel()
+                                return false;
+                            })
                         ];
-                    case 20:
+                    case 8:
                         _state.sent();
-                        return [
-                            7
-                        ];
-                    case 21:
+                        _state.label = 9;
+                    case 9:
                         return [
                             4,
                             fetch(historyRequest)
                         ];
-                    case 22:
+                    case 10:
                         finalHistoryResponse = _state.sent();
                         if (!finalHistoryResponse.ok) {
                             console.error('Final history query failed');
@@ -2747,7 +2358,7 @@ export var brains = {
                             4,
                             finalHistoryResponse.json()
                         ];
-                    case 23:
+                    case 11:
                         finalHistoryData = _state.sent();
                         finalRun = finalHistoryData.runs.find(function(r) {
                             return r.brainRunId === brainRunId;
@@ -2763,14 +2374,14 @@ export var brains = {
                             2,
                             true
                         ];
-                    case 24:
+                    case 12:
                         error = _state.sent();
                         console.error("Failed to test inner brain complete status for ".concat(outerBrainIdentifier, ":"), error);
                         return [
                             2,
                             false
                         ];
-                    case 25:
+                    case 13:
                         return [
                             2
                         ];