feat: Optimize the output of bing

KeJunMao · Feb 18, 2023 · 1671b66 · 1671b66
1 parent 1001bdd
commit 1671b66
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -61,6 +61,7 @@
   },
   "dependencies": {
     "c12": "^1.1.2",
+    "chalk": "^5.2.0",
     "consola": "^2.15.3",
     "lodash": "^4.17.21",
     "prompts": "^2.4.2",

diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/src/commands/run.ts b/src/commands/run.ts
@@ -4,6 +4,7 @@ import { EdgeGPTConfig } from "../types";
 import prompts from "prompts";
 import { ChatBot } from "../ChatBot";
 import { logger } from "../utils";
+import chalk from "chalk";
 
 export const run = async (options: Partial<EdgeGPTConfig>) => {
   const config = await loadEdgeGPTConfig({
@@ -13,6 +14,7 @@ export const run = async (options: Partial<EdgeGPTConfig>) => {
 
   const chatBot = new ChatBot(config);
   await chatBot.create();
+  const prefix = [chalk.blue("?"), chalk.bold("Bing: ")].join(" ");
 
   while (true) {
     const cmd = await prompts([
@@ -23,7 +25,6 @@ export const run = async (options: Partial<EdgeGPTConfig>) => {
         validate: (value) => (!value ? "Prompt can not be empty" : true),
       },
     ]);
-    let wrote = 0;
     if (!cmd.prompt) {
       continue;
     }
@@ -34,15 +35,16 @@ export const run = async (options: Partial<EdgeGPTConfig>) => {
       chatBot.reset();
       break;
     }
+    let wrote = 0;
     if (cmd.prompt) {
+      process.stdout.write(prefix);
       if (config.stream) {
         await chatBot.ask(cmd.prompt, (msg) => {
           process.stdout.write(msg.slice(wrote));
           wrote = msg.length;
         });
-        process.stdout.write("\n");
       } else {
-        logger.log(await chatBot.askAsync(cmd.prompt));
+        process.stdout.write(prefix + (await chatBot.askAsync(cmd.prompt)));
       }
     }
   }