fix(cli): fix reset cmd

KeJunMao · Feb 18, 2023 · 44987aa · 44987aa
1 parent d4a70f0
commit 44987aa
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/src/ChatHub.ts b/src/ChatHub.ts
@@ -3,7 +3,7 @@ import { WebSocket } from "ws";
 import { DELIMITER } from "./constant";
 import { createRequest } from "./request";
 import { Conversation } from "./types";
-import { appendIdentifier, createHeaders, logger } from "./utils";
+import { appendIdentifier, createHeaders } from "./utils";
 import TypedEmitter from "typed-emitter";
 
 type ChatHubEvents = {

diff --git a/src/commands/run.ts b/src/commands/run.ts
@@ -3,7 +3,6 @@ import { EdgeGPTConfig } from "../types";
 
 import prompts, { Choice } from "prompts";
 import { ChatBot } from "../ChatBot";
-import { logger } from "../utils";
 import chalk from "chalk";
 
 export const run = async (options: Partial<EdgeGPTConfig>) => {
@@ -43,8 +42,8 @@ export const run = async (options: Partial<EdgeGPTConfig>) => {
       break;
     }
     if (cmd.prompt === "!reset") {
-      chatBot.reset();
-      break;
+      await chatBot.reset();
+      continue;
     }
     let wrote = 0;
     let response: any;
@@ -64,13 +63,13 @@ export const run = async (options: Partial<EdgeGPTConfig>) => {
         );
       }
       try {
-        choices = response["item"]["messages"][1]["suggestedResponses"].map(
-          (v: any) => {
+        choices = response["item"]["messages"][1]["suggestedResponses"]
+          .map((v: any) => {
             return {
               title: v.text,
             };
-          }
-        );
+          })
+          .filter((v: Choice) => typeof v.title === "string");
       } catch (error) {}
     }
   }

diff --git a/src/utils.ts b/src/utils.ts
@@ -3,6 +3,7 @@ import { v4 as uuidv4 } from "uuid";
 import _ from "lodash";
 import { DELIMITER } from "./constant";
 import { Cookie } from "./types";
+import { createWriteStream } from "fs";
 
 export const logger = consola.create({
   defaults: {
@@ -50,3 +51,5 @@ export const stringifyCookie = (cookie: Cookie) =>
     pre += `${cur.name}=${cur.value};`;
     return pre;
   }, "");
+
+export const loggerFile = createWriteStream("./edgegpt.log");