update

swuecho · Mar 20, 2023 · 7767026 · 7767026
1 parent 3a3d6e0
commit 7767026
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/api/chat_main_handler.go b/api/chat_main_handler.go
@@ -150,10 +150,14 @@ func (h *ChatHandler) OpenAIChatCompletionAPIWithStreamHandler(w http.ResponseWr
 			http.Error(w, "Error: '"+err.Error()+"'", http.StatusBadRequest)
 			return
 		}
+		lastN := chat_session.MaxLength
+		if chat_session.MaxLength == 0 {
+			lastN = 10
+		}
 		msgs, err := h.chatService.q.GetLastNChatMessages(ctx,
 			sqlc_queries.GetLastNChatMessagesParams{
 				Uuid:  chatUuid,
-				Limit: chat_session.MaxLength,
+				Limit: lastN,
 			})
 		if err != nil {
 			http.Error(w, "Error: '"+err.Error()+"'", http.StatusBadRequest)