@langgraph-js/pure-graph 1.0.1 → 1.0.2

package/dist/adapter/hono/runs.js

@@ -42,7 +42,6 @@ api.post('/threads/:thread_id/runs/stream', zValidator('param', z.object({ threa
         for await (const { event, data } of client.runs.stream(thread_id, payload.assistant_id, payload)) {
             await stream.writeSSE({ data: serialiseAsDict(data), event });
         }
-        // await stream.sleep(500); // 不知为何要等
     });
 });
 api.get('/threads/:thread_id/runs', zValidator('param', z.object({ thread_id: z.string().uuid() })), zValidator('query', z.object({

package/dist/createEndpoint.js

@@ -61,8 +61,7 @@ export const createEndpoint = (threads) => {
                         },
                     };
                 }
-                const run = threads.createRun(threadId, assistantId, payload);
-                for await (const data of streamState(threads, run, payload, {
+                for await (const data of streamState(threads, threads.createRun(threadId, assistantId, payload), payload, {
                     attempt: 0,
                     getGraph,
                 })) {

package/dist/graph/stream.js

@@ -180,7 +180,7 @@ export async function* streamState(threads, run, payload, options) {
         // throw error;
     }
     finally {
-        // 在完成后立即清理队列，因为消费者已经完成
+        // 在完成后清理队列
         await threads.set(threadId, { status: 'idle' });
         globalMessageQueue.removeQueue(queueId);
     }

package/dist/queue/event_message.js

@@ -14,7 +14,7 @@ export class CancelEventMessage extends EventMessage {
 }
 export class StreamEndEventMessage extends EventMessage {
     constructor() {
-        super('__stream_end__');
+        super('__stream_end__', 'stream end');
     }
 }
 export class StreamErrorEventMessage extends EventMessage {

package/dist/storage/memory/queue.js

@@ -18,7 +18,7 @@ export class MemoryStreamQueue extends BaseStreamQueue {
      * 异步生成器：支持 for await...of 方式消费队列数据
      */
     async *onDataReceive() {
-        const queue = [];
+        let queue = [];
         let pendingResolve = null;
         let isStreamEnded = false;
         const handleData = async (item) => {
@@ -28,7 +28,13 @@ export class MemoryStreamQueue extends BaseStreamQueue {
             if (data.event === '__stream_end__' ||
                 data.event === '__stream_error__' ||
                 data.event === '__stream_cancel__') {
-                isStreamEnded = true;
+                setTimeout(() => {
+                    isStreamEnded = true;
+                    if (pendingResolve) {
+                        pendingResolve();
+                        pendingResolve = null;
+                    }
+                }, 300);
                 if (data.event === '__stream_cancel__') {
                     this.cancel();
                 }
@@ -38,12 +44,15 @@ export class MemoryStreamQueue extends BaseStreamQueue {
                 pendingResolve = null;
             }
         };
+        // todo 这个框架的事件监听的数据返回顺序有误
         this.on('dataChange', handleData);
         try {
             while (!isStreamEnded) {
                 if (queue.length > 0) {
-                    const item = queue.shift();
-                    yield item;
+                    for (const item of queue) {
+                        yield item;
+                    }
+                    queue = [];
                 }
                 else {
                     await new Promise((resolve) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langgraph-js/pure-graph",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "",
   "main": "dist/index.js",
   "keywords": [],
@@ -26,6 +26,7 @@
   },
   "scripts": {
     "dev": "bun run test/hono.ts",
-    "build": "tsc"
+    "build": "rm -r dist && tsc",
+    "prepublish": "pnpm build"
   }
 }