Skip to content

Commit

Permalink
Yi
Browse files Browse the repository at this point in the history
  • Loading branch information
swuecho committed May 13, 2024
1 parent 5004127 commit 34b25a4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion api/chat_main_handler.go
Original file line number Diff line number Diff line change
Expand Up @@ -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")

Expand Down

0 comments on commit 34b25a4

Please sign in to comment.