fix panic bug

TsaoLun · Oct 7, 2021 · ec32a14 · ec32a14
1 parent 09d685a
commit ec32a14
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 11 deletions.
diff --git a/cli/ask/ask.go b/cli/ask/ask.go
@@ -45,17 +45,23 @@ func main() {
 
 		startTime := time.Now()
 		answers := chatbot.GetResponse(question)
+		if len(answers) == 0 {
+			fmt.Println("No answer!")
+			continue
+		}
+
 		if *tops == 1 {
 			fmt.Printf("A: %s\n", answers[0].Content)
-		} else {
-			for i, answer := range answers {
-				fmt.Printf("%d: %s\n", i+1, answer.Content)
-				if *verbose {
-					fmt.Printf("%d: %s\tConfidence: %.3f\t%s\n", i+1, answer.Content,
-						answer.Confidence, time.Since(startTime))
-				}
+			continue
+		}
+
+		for i, answer := range answers {
+			fmt.Printf("%d: %s\n", i+1, answer.Content)
+			if *verbose {
+				fmt.Printf("%d: %s\tConfidence: %.3f\t%s\n", i+1, answer.Content,
+					answer.Confidence, time.Since(startTime))
 			}
-			fmt.Println(time.Since(startTime))
 		}
+		fmt.Println(time.Since(startTime))
 	}
 }
diff --git a/cli/train/train.go b/cli/train/train.go
@@ -29,10 +29,13 @@ func main() {
 	var corporaFiles string
 	if len(files) > 0 {
 		corporaFiles = strings.Join(files, ",")
+		if len(*corpora) > 0 {
+			corporaFiles = strings.Join([]string{corporaFiles, *corpora}, ",")
+		}
+	} else {
+		corporaFiles = *corpora
 	}
-	if len(*corpora) > 0 {
-		corporaFiles = strings.Join([]string{corporaFiles, *corpora}, ",")
-	}
+
 	if len(corporaFiles) == 0 {
 		flag.Usage()
 		return