-- FIX

- UPDATE LLM vendor
NoahAmethyst · Oct 29, 2024 · 4d22255 · 4d22255
1 parent 69b4f12
commit 4d22255
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/coolq/chat_assistant.go b/coolq/chat_assistant.go
@@ -267,8 +267,13 @@ func AskAssistant(assistant Assistant) {
 			return
 		case <-time.After(time.Second * 10):
 			vendor := "OpenAI"
-			if assistant.Model() == ai_util.BingCopilot {
+			switch assistant.Model() {
+			case ai_util.BingCopilot:
 				vendor = "BingCopilot"
+			case ai_util.Ernie:
+				vendor = "百度千帆"
+			default:
+				break
 			}
 			assistant.Reply(fmt.Sprintf("%s 正在响应，请稍后...", vendor))
 		}