fix: ignore markdown codeblock (#205)

nekomeowww · May 29, 2024 · 0ec307b · 0ec307b
1 parent f9810c4
commit 0ec307b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/internal/models/chathistories/recap.go b/internal/models/chathistories/recap.go
@@ -89,6 +89,9 @@ func (m *Model) summarizeChatHistoriesSlice(chatID int64, s string) ([]*openai.C
 	}
 
 	var outputs []*openai.ChatHistorySummarizationOutputs
+	resp.Choices[0].Message.Content = strings.TrimPrefix(resp.Choices[0].Message.Content, "```json")
+	resp.Choices[0].Message.Content = strings.TrimPrefix(resp.Choices[0].Message.Content, "```")
+	resp.Choices[0].Message.Content = strings.TrimSuffix(resp.Choices[0].Message.Content, "```")
 
 	err = json.Unmarshal([]byte(resp.Choices[0].Message.Content), &outputs)
 	if err != nil {