From 1f5e1c221c98ba243a10dcae5d3d1fe5d90af638 Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Fri, 20 Dec 2024 03:49:30 +0800
Subject: [PATCH] fix: classify chatgpt-4o-latest

---
 router/chat.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/router/chat.go b/router/chat.go
index fa3d2b7..0637192 100644
--- a/router/chat.go
+++ b/router/chat.go
@@ -18,7 +18,7 @@ func ChatHandler(c *gin.Context) {
 		return
 	}
 
-	if strings.HasPrefix(chatreq.Model, "gpt") || strings.HasPrefix(chatreq.Model, "o1-") {
+	if strings.Contains(chatreq.Model, "gpt") || strings.HasPrefix(chatreq.Model, "o1-") {
 		openai.ChatProxy(c, &chatreq)
 		return
 	}