feat: 支持新会话

src/cli/commands/chat.ts

@@ -61,7 +61,7 @@ class ChatCommand extends BaseCommand {
       console.warn = () => {};
       process.stdout.write(
         this.getDisplayContent(PromptRole.SYSTEM) +
-          `If you want to end this conversation, please tell the AI directly\n`
+          'If you want to end this conversation, please tell the AI directly\n'
       );
       process.stdout.write(aiDisplay + lastMessage + '\n');
 
@@ -86,10 +86,18 @@ class ChatCommand extends BaseCommand {
           break;
         }
 
-        process.stdout.write('\n');
-
         await this.addMessage(PromptRole.USER, userMessage);
         await this.addMessage(PromptRole.AI, lastMessage);
+
+        if (lastMessage === '$command:clear$') {
+          process.stdout.write(ansiEscapes.eraseLine + ansiEscapes.cursorLeft);
+          process.stdout.write(this.getDisplayContent(PromptRole.SYSTEM) + 'This is a new conversation.\n');
+          process.stdout.write(aiDisplay + 'How can I help you today?');
+
+          await this.messages.clear();
+        }
+
+        process.stdout.write('\n');
       }
     } else {
       this.showHelp();

src/core/index.ts

@@ -35,7 +35,7 @@ export class AIGC {
     const platform = getPlatform(this.platform);
     const prompt = ChatPromptTemplate.fromMessages([
       SystemMessagePromptTemplate.fromTemplate(
-        'You are a helpful assistant. Answer all questions to the best of your ability.\nIf the user wants to end the conversation or exit the current chat command, directly reply to the user "$command:exit$"'
+        'You are a helpful assistant. Answer all questions to the best of your ability.\nIf the user wants to end the conversation or exit the current chat command, directly reply to the user "$command:exit$".\nIf the user wants to start a new conversation, directly reply to the user "$command:clear$".'
       ),
       new MessagesPlaceholder('messages'),
       HumanMessagePromptTemplate.fromTemplate('{question}')