feat: Support restart a new Session

fengxxc · Mar 21, 2023 · 9c87118 · 9c87118
1 parent 6383f6e
commit 9c87118
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -1,6 +1,6 @@
 BUILD_ENV := CGO_ENABLED=0
 APP=go-chatgpt
-VERSION=v0.2.0
+VERSION=v0.3.0
 
 # linux or mac 环境编译
 # make [cmd]

diff --git a/main.go b/main.go
@@ -51,7 +51,8 @@ func main() {
 
 	if content == "" {
 		fmt.Println("Welcome to ChatGpt CLI~ 😘 ")
-		fmt.Print("Start your show~ 😙 (Enter 'exit' to quit.)\n> ")
+		fmt.Println("(Enter 'exit' to quit; Enter 'new' to restart a new Session.)")
+		fmt.Print("> ")
 		chatOfSession(config)
 	}
 
@@ -67,7 +68,12 @@ func chatOfSession(config *chatgpt.GptConfig) {
 		scanner.Scan()
 		q = scanner.Text()
 		q = strings.TrimSpace(q)
-		if q == "exit" {
+		if q == "new" {
+			gptMessages = []*chatgpt.GptMessage{}
+			fmt.Println("-----Restart a new session-----")
+			fmt.Printf("\n> ")
+			continue
+		} else if q == "exit" {
 			fmt.Println("See you next time~ 🤗")
 			os.Exit(0)
 			return