bugfix:chat context to long to send message

bighu630 · Nov 25, 2024 · 3834144 · 3834144
1 parent a0115a0
commit 3834144
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ai/gemini/gemini.go b/ai/gemini/gemini.go
@@ -93,6 +93,9 @@ func (g *gemini) Chat(chatId string, msg string) (string, error) {
 		cs = g.model.StartChat()
 		g.chats[chatId] = cs
 	}
+	if len(cs.History) > 30 {
+		cs.History = cs.History[len(cs.History)-30:]
+	}
 	if err := g.db.Add(models.NewChat(chatId, true, msg)); err != nil {
 		log.Error().Err(err).Msg("failed to add chat record")
 	}