fix race

2 files changed, 3 insertions, 19 deletions

diff --git a/js/src/index.ts b/js/src/index.ts
index df6c0aee2..1f6c7ae26 100644
--- a/js/src/index.ts
+++ b/js/src/index.ts
@@ -101,22 +101,6 @@ cli
           },
         ],
       });
-      await Session.summarize({
-        sessionID: session.id,
-        providerID,
-        modelID,
-      });
-      await Session.chat({
-        sessionID: session.id,
-        providerID,
-        modelID,
-        parts: [
-          {
-            type: "text",
-            text: "This is a test message",
-          },
-        ],
-      });
 
       for (const part of result.parts) {
         if (part.type === "text") {
diff --git a/js/src/session/session.ts b/js/src/session/session.ts
index 7770f2759..e773ed297 100644
--- a/js/src/session/session.ts
+++ b/js/src/session/session.ts
@@ -110,7 +110,8 @@ export namespace Session {
     const result = [] as Message.Info[];
     const list = Storage.list("session/message/" + sessionID);
     for await (const p of list) {
-      const read = await Storage.readJSON<Message.Info>(p);
+      const read = await Storage.readJSON<Message.Info>(p).catch(() => {});
+      if (!read) continue;
       result.push(read);
     }
     result.sort((a, b) => (a.id > b.id ? 1 : -1));
@@ -250,6 +251,7 @@ export namespace Session {
         tool: {},
       },
     };
+    await updateMessage(next);
     const result = streamText({
       onStepFinish: async (step) => {
         const assistant = next.metadata!.assistant!;
@@ -266,8 +268,6 @@ export namespace Session {
       tools,
       model: model.instance,
     });
-
-    msgs.push(next);
     let text: Message.TextPart | undefined;
     const reader = result.toUIMessageStream().getReader();
     while (true) {