@langgraph-js/pure-graph 1.0.0 → 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,16 +1,8 @@
 {
   "name": "@langgraph-js/pure-graph",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "",
   "main": "dist/index.js",
-  "exports": {
-    ".": {
-      "import": "./dist/index.js"
-    },
-    "/hono": {
-      "import": "./dist/adapter/hono/index.js"
-    }
-  },
   "keywords": [],
   "author": "",
   "license": "MIT",
@@ -34,6 +26,7 @@
   },
   "scripts": {
     "dev": "bun run test/hono.ts",
-    "build": "tsc"
+    "build": "rm -r dist && tsc",
+    "prepublish": "pnpm build"
   }
 }

package/src/adapter/hono/runs.ts

@@ -53,7 +53,6 @@ api.post(
             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); // 不知为何要等
         });
     },
 );

package/src/createEndpoint.ts

@@ -78,12 +78,15 @@ export const createEndpoint = (threads: BaseThreadsManager): ILangGraphClient =>
                     };
                 }
 
-                const run = threads.createRun(threadId, assistantId, payload);
-
-                for await (const data of streamState(threads, run, payload, {
-                    attempt: 0,
-                    getGraph,
-                })) {
+                for await (const data of streamState(
+                    threads,
+                    threads.createRun(threadId, assistantId, payload),
+                    payload,
+                    {
+                        attempt: 0,
+                        getGraph,
+                    },
+                )) {
                     yield data;
                 }
             },

package/src/graph/stream.ts

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

package/src/queue/event_message.ts

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

package/src/storage/memory/queue.ts

@@ -21,20 +21,26 @@ export class MemoryStreamQueue extends BaseStreamQueue implements BaseStreamQueu
      * 异步生成器：支持 for await...of 方式消费队列数据
      */
     async *onDataReceive(): AsyncGenerator<EventMessage, void, unknown> {
-        const queue: EventMessage[] = [];
+        let queue: EventMessage[] = [];
         let pendingResolve: (() => void) | null = null;
         let isStreamEnded = false;
         const handleData = async (item: EventMessage) => {
             const data = this.compressMessages ? ((await this.decodeData(item as any)) as EventMessage) : item;
             queue.push(data);
-
             // 检查是否为流结束或错误信号
             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();
                 }
@@ -45,14 +51,16 @@ export class MemoryStreamQueue extends BaseStreamQueue implements BaseStreamQueu
                 pendingResolve = null;
             }
         };
-
+        // todo 这个框架的事件监听的数据返回顺序有误
         this.on('dataChange', handleData as any);
 
         try {
             while (!isStreamEnded) {
                 if (queue.length > 0) {
-                    const item = queue.shift() as EventMessage;
-                    yield item;
+                    for (const item of queue) {
+                        yield item;
+                    }
+                    queue = [];
                 } else {
                     await new Promise((resolve) => {
                         pendingResolve = resolve as () => void;

package/test/test.ts

@@ -0,0 +1,10 @@
+import { registerGraph } from '../src/createEndpoint';
+import { graph } from '/Users/konghayao/code/ai/code-graph/agents/code/graph';
+import { Hono } from 'hono';
+import LangGraphApp from '../src/adapter/hono/index';
+registerGraph('code', graph);
+
+const app = new Hono();
+app.route('/', LangGraphApp);
+
+export default app;