增加v1结尾路径的支持

fruitbars · Jul 13, 2024 · 3f6d465 · 3f6d465
1 parent e1796a1
commit 3f6d465
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.go b/main.go
@@ -78,7 +78,7 @@ func main() {
 	{
 		// 中间件检查路径是否以 /v1/chat/completions 结尾
 		v1.POST("/*path", func(c *gin.Context) {
-			if strings.HasSuffix(c.Request.URL.Path, "/v1/chat/completions") {
+			if strings.HasSuffix(c.Request.URL.Path, "/v1/chat/completions") || strings.HasSuffix(c.Request.URL.Path, "/v1") {
 				handler.OpenAIHandler(c)
 				return
 			}