feat: 支持通过对话退出

Author: crazyurus

package.json

@@ -1,6 +1,6 @@
 {
   "name": "aigc-detector",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "Detect if content is generated by AI",
   "keywords": [
     "aigc",

src/cli/commands/chat.ts

@@ -1,4 +1,5 @@
 import { AIMessage, HumanMessage, SystemMessage } from '@langchain/core/messages';
+import ansiEscapes from 'ansi-escapes';
 import chalk from 'chalk';
 import { ChatMessageHistory } from 'langchain/stores/message/in_memory';
 import readline from 'node:readline';
@@ -59,27 +60,32 @@ class ChatCommand extends BaseCommand {
 
       console.warn = () => {};
       process.stdout.write(
-        this.getDisplayContent(PromptRole.SYSTEM) + `Type ${chalk.cyan('exit')} to end this conversation\n`
+        this.getDisplayContent(PromptRole.SYSTEM) +
+          `If you want to end this conversation, please tell the AI directly\n`
       );
       process.stdout.write(aiDisplay + lastMessage + '\n');
 
       // eslint-disable-next-line no-constant-condition
       while (true) {
         const userMessage = await this.getUserMessage();
-
-        if (userMessage === 'exit') {
-          close();
-
-          break;
-        }
-
         const stream = detector.chat(userMessage, await this.messages.getMessages());
 
         process.stdout.write(aiDisplay);
         stream.pipe(process.stdout);
 
         lastMessage = await stream.getData();
 
+        if (lastMessage === '$command:exit$') {
+          process.stdout.write(ansiEscapes.eraseLine + ansiEscapes.cursorLeft);
+          process.stdout.write(aiDisplay + 'Nice to chat with you. Bye~\n');
+
+          await this.wait(1000);
+
+          close();
+
+          break;
+        }
+
         process.stdout.write('\n');
 
         await this.addMessage(PromptRole.USER, userMessage);

src/cli/extends/command.ts

@@ -63,6 +63,12 @@ abstract class BaseCommand extends Command {
 
     this.log(bang + ' ' + text);
   }
+
+  protected wait(timeout: number): Promise<void> {
+    return new Promise((resolve) => {
+      setTimeout(resolve, timeout);
+    });
+  }
 }
 
 export default BaseCommand;

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.'
+        '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$"'
       ),
       new MessagesPlaceholder('messages'),
       HumanMessagePromptTemplate.fromTemplate('{question}')