From 44a4a9121e7f0488dc1d25d3134af31e10c9f7cc Mon Sep 17 00:00:00 2001
From: wangshangwen <shangwenwang20@gmail.com>
Date: Tue, 28 Mar 2023 12:00:12 +0800
Subject: [PATCH] fix: defer saving new message util got response from openAI

---
 src/chatgpt-api.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/chatgpt-api.ts b/src/chatgpt-api.ts
index b44cce2..d0d26eb 100644
--- a/src/chatgpt-api.ts
+++ b/src/chatgpt-api.ts
@@ -156,7 +156,8 @@ export class ChatGPTAPI {
       parentMessageId,
       text
     }
-    await this._upsertMessage(message)
+
+    const latestQuestion = message
 
     const { messages, maxTokens, numTokens } = await this._buildMessages(
       text,
@@ -287,7 +288,9 @@ export class ChatGPTAPI {
         }
       }
     ).then((message) => {
-      return this._upsertMessage(message).then(() => message)
+      return this._upsertMessage(latestQuestion).then(() =>
+        this._upsertMessage(message).then(() => message)
+      )
     })
 
     if (timeoutMs) {