feat: add context message count configuration option

src/chatgpt-api.ts

@@ -359,7 +359,7 @@ export class ChatGPTAPI {
   }
 
   protected async _buildMessages(text: string, opts: types.SendMessageOptions) {
-    const { systemMessage = this._systemMessage } = opts
+    let { systemMessage = this._systemMessage, messageCount } = opts
     let { parentMessageId } = opts
 
     const userLabel = USER_LABEL_DEFAULT
@@ -373,6 +373,7 @@ export class ChatGPTAPI {
         role: 'system',
         content: systemMessage
       })
+      if (typeof messageCount === 'number') messageCount++
     }
 
     const systemMessageOffset = messages.length
@@ -415,6 +416,10 @@ export class ChatGPTAPI {
         break
       }
 
+      if (typeof messageCount === 'number' && messages.length >= messageCount) {
+        break
+      }
+
       if (!parentMessageId) {
         break
       }

src/types.ts

@@ -48,6 +48,7 @@ export type SendMessageOptions = {
   completionParams?: Partial<
     Omit<openai.CreateChatCompletionRequest, 'messages' | 'n' | 'stream'>
   >
+  messageCount?: number
 }
 
 export type MessageActionType = 'next' | 'variant'