Merge pull request transitive-bullshit#504 from 189/fix/defer-save

mi-ann0613 · Apr 2, 2023 · dd25220 · dd25220
2 parents 34990a5 + 7db108d
commit dd25220
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/chatgpt-api.ts b/src/chatgpt-api.ts
@@ -160,7 +160,8 @@ export class ChatGPTAPI {
       parentMessageId,
       text
     }
-    await this._upsertMessage(message)
+
+    const latestQuestion = message
 
     const { messages, maxTokens, numTokens } = await this._buildMessages(
       text,
@@ -313,7 +314,7 @@ export class ChatGPTAPI {
         }
       }
 
-      return this._upsertMessage(message).then(() => message)
+      return Promise.all([ this._upsertMessage(latestQuestion), this._upsertMessage(message) ]).then(() => message)
     })
 
     if (timeoutMs) {