From 34b25a4eedc89274c74a2b6127fe828723b8d66c Mon Sep 17 00:00:00 2001 From: Hao Wu Date: Tue, 14 May 2024 07:59:48 +0800 Subject: [PATCH] Yi --- api/chat_main_handler.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/chat_main_handler.go b/api/chat_main_handler.go index 991cc4a5..6f257d22 100644 --- a/api/chat_main_handler.go +++ b/api/chat_main_handler.go @@ -267,7 +267,7 @@ func (h *ChatHandler) chooseChatStreamFn(chat_session sqlc_queries.ChatSession, isClaude = false isClaude3 = true } - isChatGPT := strings.HasPrefix(model, "gpt") || strings.HasPrefix(model, "deepseek") + isChatGPT := strings.HasPrefix(model, "gpt") || strings.HasPrefix(model, "deepseek") || strings.HasPrefix(model, "yi") isOllama := strings.HasPrefix(model, "ollama-") isGemini := strings.HasPrefix(model, "gemini")