fix: fix code command output

shellfly · Mar 29, 2023 · ade11ac · ade11ac
1 parent 1336ac3
commit ade11ac
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/command/code.go b/pkg/command/code.go
@@ -54,7 +54,10 @@ func (c *Code) Prompts(input string) []string {
 
 // Handle copy code in the reply to clipboard, and return the original reply
 func (c *Code) Handle(reply string) {
-	code := extractCode(reply)
+	code := reply
+	if strings.Contains(reply, "```") {
+		code = extractCode(reply)
+	}
 	fmt.Println(code)
 	fmt.Println()
 	if code != "" {