@kuralle-agents/cf-agent 0.3.20 → 0.4.0

package/dist/StreamAdapter.d.ts

@@ -15,4 +15,14 @@ import type { StreamAdapterConfig } from './types.js';
  * CF's applyChunkToParts() then builds UIMessage parts from these chunks.
  * CF handles persistence, broadcasting, and resumability.
  */
+type SSETextState = {
+    open: boolean;
+    canceledIds: Set<string>;
+};
 export declare function createSSEResponse(stream: AsyncGenerator<HarnessStreamPart>, config?: StreamAdapterConfig): Response;
+/**
+ * Convert a single HarnessStreamPart to SSE data lines.
+ * Returns 0+ formatted SSE lines ("data: {...}\n\n").
+ */
+export declare function convertToSSELines(part: HarnessStreamPart, config: StreamAdapterConfig, textState: SSETextState): string[];
+export {};

package/dist/StreamAdapter.js

@@ -1,38 +1,18 @@
 import { DEFAULT_STREAM_CONFIG } from './types.js';
-/**
- * Convert Kuralle's HarnessStreamPart generator into an SSE Response
- * that CF's AIChatAgent._reply() can parse.
- *
- * CF's _streamSSEReply reads lines in the format:
- *   data: {"type":"text-start"}\n\n
- *   data: {"type":"text-delta","delta":"Hello"}\n\n
- *   data: {"type":"tool-input-available","toolCallId":"...","toolName":"...","input":{...}}\n\n
- *   data: {"type":"tool-output-available","toolCallId":"...","output":{...}}\n\n
- *   data: {"type":"data-handoff","data":{...}}\n\n
- *   data: {"type":"text-end"}\n\n
- *
- * CF's applyChunkToParts() then builds UIMessage parts from these chunks.
- * CF handles persistence, broadcasting, and resumability.
- */
 export function createSSEResponse(stream, config = DEFAULT_STREAM_CONFIG) {
     const encoder = new TextEncoder();
-    let textStarted = false;
+    const textState = { open: false, canceledIds: new Set() };
     const body = new ReadableStream({
         async start(controller) {
             try {
                 for await (const part of stream) {
-                    const lines = convertToSSELines(part, config, textStarted);
+                    const lines = convertToSSELines(part, config, textState);
                     for (const line of lines) {
-                        // Track whether we've emitted text-start
-                        if (line.includes('"text-start"'))
-                            textStarted = true;
-                        if (line.includes('"text-end"'))
-                            textStarted = false;
                         controller.enqueue(encoder.encode(line));
                     }
                 }
                 // Ensure text is closed if still open
-                if (textStarted) {
+                if (textState.open) {
                     controller.enqueue(encoder.encode(`data: ${JSON.stringify({ type: 'text-end' })}\n\n`));
                 }
                 controller.close();
@@ -53,23 +33,39 @@ export function createSSEResponse(stream, config = DEFAULT_STREAM_CONFIG) {
  * Convert a single HarnessStreamPart to SSE data lines.
  * Returns 0+ formatted SSE lines ("data: {...}\n\n").
  */
-function convertToSSELines(part, config, textStarted) {
+export function convertToSSELines(part, config, textState) {
     const lines = [];
     const sse = (obj) => `data: ${JSON.stringify(obj)}\n\n`;
+    const closeTextSegment = () => {
+        if (!textState.open)
+            return;
+        lines.push(sse({ type: 'text-end' }));
+        textState.open = false;
+    };
     switch (part.type) {
+        case 'text-start':
+            break;
+        case 'text-end':
+            closeTextSegment();
+            break;
+        case 'text-cancel': {
+            textState.canceledIds.add(part.id);
+            closeTextSegment();
+            break;
+        }
         case 'text-delta': {
-            // Emit text-start on first text chunk
-            if (!textStarted) {
+            if (textState.canceledIds.has(part.id)) {
+                break;
+            }
+            if (!textState.open) {
                 lines.push(sse({ type: 'text-start' }));
+                textState.open = true;
             }
-            lines.push(sse({ type: 'text-delta', delta: part.text }));
+            lines.push(sse({ type: 'text-delta', delta: part.delta }));
             break;
         }
         case 'tool-call': {
-            // Close text before tool if open
-            if (textStarted) {
-                lines.push(sse({ type: 'text-end' }));
-            }
+            closeTextSegment();
             // CF expects tool-input-available with toolCallId, toolName, input
             lines.push(sse({
                 type: 'tool-input-available',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kuralle-agents/cf-agent",
-  "version": "0.3.20",
+  "version": "0.4.0",
   "description": "Kuralle agent integration for Cloudflare Workers with AIChatAgent",
   "type": "module",
   "main": "./dist/index.js",
@@ -22,18 +22,18 @@
     "@cloudflare/ai-chat": "^0.1.8",
     "ai": "^6.0.90",
     "zod-to-json-schema": "^3.24.0",
-    "@kuralle-agents/core": "0.3.20",
-    "@kuralle-agents/realtime-audio": "0.3.20"
+    "@kuralle-agents/core": "0.4.0",
+    "@kuralle-agents/realtime-audio": "0.4.0"
   },
   "peerDependencies": {
     "agents": "^0.11.5",
     "zod": "^3.0.0",
-    "@kuralle-agents/voice-protocol": "0.3.20"
+    "@kuralle-agents/voice-protocol": "0.4.0"
   },
   "devDependencies": {
     "agents": "^0.11.5",
     "typescript": "^5.7.0",
-    "@kuralle-agents/voice-protocol": "0.3.20"
+    "@kuralle-agents/voice-protocol": "0.4.0"
   },
   "engines": {
     "node": ">=20"
@@ -57,7 +57,7 @@
     "prebuild": "rm -rf dist",
     "build": "tsc",
     "dev": "tsc --watch",
-    "test": "bun test src/voice/__tests__",
+    "test": "bun test src/voice/__tests__ src/__tests__",
     "typecheck": "tsc --noEmit",
     "clean": "rm -rf dist"
   }