fix: classify chatgpt-4o-latest

README.md

@@ -77,9 +77,6 @@ wget https://github.com/mirrors2/opencatd-open/raw/main/docker/docker-compose.ym
   - [Fly.io](https://fly.io/)
   - 或者其他
 
-修改openai的endpoint地址？使用任意上游地址(套娃代理)
-  - 设置环境变量 openai_endpoint
-
 使用Nginx + Docker部署
   - [使用Nginx + Docker部署](./doc/deploy.md)
   
@@ -87,12 +84,13 @@ pandora for team
   - [pandora for team](./doc/pandora.md)
 
 如何自定义HOST地址? (仅OpenAI)
-  - 需修改环境变量，优先级递增
+  - 需修改环境变量，优先级递增(全局配置谨慎修改)
   - Cloudflare AI Gateway地址 `AIGateWay_Endpoint=https://gateway.ai.cloudflare.com/v1/123456789/xxxx/openai/chat/completions`
-  - 自定义的endpoint `$CUSTOM_ENDPOINT=true &&  $OpenAI_Endpoint=https://your.domain/v1/chat/completions`
+  - 自定义的endpoint `OpenAI_Endpoint=https://your.domain/v1/chat/completions`
   
 设置主页跳转地址?
   - 修改环境变量 `CUSTOM_REDIRECT=https://your.domain`
+  
 ## 获取更多信息
 [![TG](https://telegram.org/img/favicon.ico)](https://t.me/OpenTeamLLM)
 

pkg/claude/chat.go

@@ -143,51 +143,57 @@ func ChatMessages(c *gin.Context, chatReq *openai.ChatCompletionRequest) {
 		claudReq.Model = ""
 	}
 
-	var prompt string
-
 	var claudecontent []VisionContent
+	var prompt string
 	for _, msg := range chatReq.Messages {
-		if msg.Role == "system" {
-			claudReq.System = string(msg.Content)
-			continue
-		}
-
-		var oaivisioncontent []openai.VisionContent
-		if err := json.Unmarshal(msg.Content, &oaivisioncontent); err != nil {
-			prompt += "<" + msg.Role + ">: " + string(msg.Content) + "\n"
-
-			claudecontent = append(claudecontent, VisionContent{Type: "text", Text: msg.Role + ":" + string(msg.Content)})
-		} else {
-			if len(oaivisioncontent) > 0 {
-				for _, content := range oaivisioncontent {
-					if content.Type == "text" {
-						prompt += "<" + msg.Role + ">: " + content.Text + "\n"
-						claudecontent = append(claudecontent, VisionContent{Type: "text", Text: msg.Role + ":" + content.Text})
-					} else if content.Type == "image_url" {
-						if strings.HasPrefix(content.ImageURL.URL, "http") {
-							fmt.Println("链接:", content.ImageURL.URL)
-						} else if strings.HasPrefix(content.ImageURL.URL, "data:image") {
-							fmt.Println("base64:", content.ImageURL.URL[:20])
+		switch ct := msg.Content.(type) {
+		case string:
+			prompt += "<" + msg.Role + ">: " + msg.Content.(string) + "\n"
+			if msg.Role == "system" {
+				claudReq.System = msg.Content.(string)
+				continue
+			}
+			claudecontent = append(claudecontent, VisionContent{Type: "text", Text: msg.Role + ":" + msg.Content.(string)})
+		case []any:
+			for _, item := range ct {
+				if m, ok := item.(map[string]interface{}); ok {
+					if m["type"] == "text" {
+						prompt += "<" + msg.Role + ">: " + m["text"].(string) + "\n"
+						claudecontent = append(claudecontent, VisionContent{Type: "text", Text: msg.Role + ":" + m["text"].(string)})
+					} else if m["type"] == "image_url" {
+						if url, ok := m["image_url"].(map[string]interface{}); ok {
+							fmt.Printf("  URL: %v\n", url["url"])
+							if strings.HasPrefix(url["url"].(string), "http") {
+								fmt.Println("网络图片:", url["url"].(string))
+							} else if strings.HasPrefix(url["url"].(string), "data:image") {
+								fmt.Println("base64:", url["url"].(string)[:20])
+								var mediaType string
+								if strings.HasPrefix(url["url"].(string), "data:image/jpeg") {
+									mediaType = "image/jpeg"
+								}
+								if strings.HasPrefix(url["url"].(string), "data:image/png") {
+									mediaType = "image/png"
+								}
+								claudecontent = append(claudecontent, VisionContent{Type: "image", Source: &VisionSource{Type: "base64", MediaType: mediaType, Data: strings.Split(url["url"].(string), ",")[1]}})
+							}
 						}
-						// todo image tokens
-						var mediaType string
-						if strings.HasPrefix(content.ImageURL.URL, "data:image/jpeg") {
-							mediaType = "image/jpeg"
-						}
-						if strings.HasPrefix(content.ImageURL.URL, "data:image/png") {
-							mediaType = "image/png"
-						}
-						claudecontent = append(claudecontent, VisionContent{Type: "image", Source: &VisionSource{Type: "base64", MediaType: mediaType, Data: strings.Split(content.ImageURL.URL, ",")[1]}})
 					}
 				}
-
 			}
+		default:
+			c.JSON(http.StatusInternalServerError, gin.H{
+				"error": gin.H{
+					"message": "Invalid content type",
+				},
+			})
+			return
+		}
+		if len(chatReq.Tools) > 0 {
+			tooljson, _ := json.Marshal(chatReq.Tools)
+			prompt += "<tools>: " + string(tooljson) + "\n"
 		}
-		// if len(chatReq.Tools) > 0 {
-		// 	tooljson, _ := json.Marshal(chatReq.Tools)
-		// 	prompt += "<tools>: " + string(tooljson) + "\n"
-		// }
 	}
+
 	claudReq.Messages = []VisionMessages{{Role: "user", Content: claudecontent}}
 
 	usagelog.PromptCount = tokenizer.NumTokensFromStr(prompt, chatReq.Model)

pkg/google/chat.go

@@ -97,59 +97,55 @@ func ChatProxy(c *gin.Context, chatReq *openai.ChatCompletionRequest) {
 	var prompts []genai.Part
 	var prompt string
 	for _, msg := range chatReq.Messages {
-		var visioncontent []openai.VisionContent
-		if err := json.Unmarshal(msg.Content, &visioncontent); err != nil {
-			prompt += "<" + msg.Role + ">: " + string(msg.Content) + "\n"
-			prompts = append(prompts, genai.Text("<"+msg.Role+">: "+string(msg.Content)))
-		} else {
-			if len(visioncontent) > 0 {
-				for _, content := range visioncontent {
-					if content.Type == "text" {
-						prompt += "<" + msg.Role + ">: " + content.Text + "\n"
-						prompts = append(prompts, genai.Text("<"+msg.Role+">: "+content.Text))
-					} else if content.Type == "image_url" {
-						if strings.HasPrefix(content.ImageURL.URL, "http") {
-							fmt.Println("链接:", content.ImageURL.URL)
-						} else if strings.HasPrefix(content.ImageURL.URL, "data:image") {
-							fmt.Println("base64:", content.ImageURL.URL[:20])
-							if chatReq.Model != "gemini-pro-vision" {
-								chatReq.Model = "gemini-pro-vision"
+		switch ct := msg.Content.(type) {
+		case string:
+			prompt += "<" + msg.Role + ">: " + msg.Content.(string) + "\n"
+			prompts = append(prompts, genai.Text("<"+msg.Role+">: "+msg.Content.(string)))
+		case []any:
+			for _, item := range ct {
+				if m, ok := item.(map[string]interface{}); ok {
+					if m["type"] == "text" {
+						prompt += "<" + msg.Role + ">: " + m["text"].(string) + "\n"
+						prompts = append(prompts, genai.Text("<"+msg.Role+">: "+m["text"].(string)))
+					} else if m["type"] == "image_url" {
+						if url, ok := m["image_url"].(map[string]interface{}); ok {
+							if strings.HasPrefix(url["url"].(string), "http") {
+								fmt.Println("网络图片:", url["url"].(string))
+							} else if strings.HasPrefix(url["url"].(string), "data:image") {
+								fmt.Println("base64:", url["url"].(string)[:20])
+								var mime string
+								// openai 会以 data:image 开头，则去掉 data:image/png;base64, 和 data:image/jpeg;base64,
+								if strings.HasPrefix(url["url"].(string), "data:image/png") {
+									mime = "image/png"
+								} else if strings.HasPrefix(url["url"].(string), "data:image/jpeg") {
+									mime = "image/jpeg"
+								} else {
+									c.JSON(http.StatusInternalServerError, gin.H{"error": "Unsupported image format"})
+									return
+								}
+								imageString := strings.Split(url["url"].(string), ",")[1]
+								imageBytes, err := base64.StdEncoding.DecodeString(imageString)
+								if err != nil {
+									c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+									return
+								}
+								prompts = append(prompts, genai.Blob{MIMEType: mime, Data: imageBytes})
 							}
-
-							var mime string
-							// openai 会以 data:image 开头，则去掉 data:image/png;base64, 和 data:image/jpeg;base64,
-							if strings.HasPrefix(content.ImageURL.URL, "data:image/png") {
-								mime = "image/png"
-							} else if strings.HasPrefix(content.ImageURL.URL, "data:image/jpeg") {
-								mime = "image/jpeg"
-							} else {
-								c.JSON(http.StatusInternalServerError, gin.H{"error": "Unsupported image format"})
-								return
-							}
-							imageString := strings.Split(content.ImageURL.URL, ",")[1]
-							imageBytes, err := base64.StdEncoding.DecodeString(imageString)
-							if err != nil {
-								c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
-								return
-							}
-							prompts = append(prompts, genai.Blob{MIMEType: mime, Data: imageBytes})
 						}
-
-						// todo image tokens
 					}
-
 				}
-
 			}
+		default:
+			c.JSON(http.StatusInternalServerError, gin.H{
+				"error": gin.H{
+					"message": "Invalid content type",
+				},
+			})
+			return
 		}
 		if len(chatReq.Tools) > 0 {
 			tooljson, _ := json.Marshal(chatReq.Tools)
 			prompt += "<tools>: " + string(tooljson) + "\n"
-
-			// for _, tool := range chatReq.Tools {
-
-			// }
-
 		}
 	}
 
@@ -171,6 +167,7 @@ func ChatProxy(c *gin.Context, chatReq *openai.ChatCompletionRequest) {
 	defer client.Close()
 
 	model := client.GenerativeModel(chatReq.Model)
+	model.Tools = []*genai.Tool{}
 
 	iter := model.GenerateContentStream(ctx, prompts...)
 	datachan := make(chan string)

pkg/openai/chat.go

@@ -25,13 +25,13 @@ const (
 )
 
 var (
-	BaseURL            string // "https://api.openai.com"
-	AIGateWay_Endpoint = "https://gateway.ai.cloudflare.com/v1/431ba10f11200d544922fbca177aaa7f/openai/openai/chat/completions"
+	Custom_Endpoint    string
+	AIGateWay_Endpoint string // "https://gateway.ai.cloudflare.com/v1/431ba10f11200d544922fbca177aaa7f/openai/openai/chat/completions"
 )
 
 func init() {
 	if os.Getenv("OpenAI_Endpoint") != "" {
-		BaseURL = os.Getenv("OpenAI_Endpoint")
+		Custom_Endpoint = os.Getenv("OpenAI_Endpoint")
 	}
 	if os.Getenv("AIGateWay_Endpoint") != "" {
 		AIGateWay_Endpoint = os.Getenv("AIGateWay_Endpoint")
@@ -50,9 +50,9 @@ type VisionImageURL struct {
 }
 
 type ChatCompletionMessage struct {
-	Role    string          `json:"role"`
-	Content json.RawMessage `json:"content"`
-	Name    string          `json:"name,omitempty"`
+	Role    string `json:"role"`
+	Content any    `json:"content"`
+	Name    string `json:"name,omitempty"`
 	// MultiContent []VisionContent
 }
 
@@ -185,27 +185,31 @@ func ChatProxy(c *gin.Context, chatReq *ChatCompletionRequest) {
 
 	var prompt string
 	for _, msg := range chatReq.Messages {
-		// prompt += "<" + msg.Role + ">: " + msg.Content + "\n"
-		var visioncontent []VisionContent
-		if err := json.Unmarshal(msg.Content, &visioncontent); err != nil {
-			prompt += "<" + msg.Role + ">: " + string(msg.Content) + "\n"
-		} else {
-			if len(visioncontent) > 0 {
-				for _, content := range visioncontent {
-					if content.Type == "text" {
-						prompt += "<" + msg.Role + ">: " + content.Text + "\n"
-					} else if content.Type == "image_url" {
-						if strings.HasPrefix(content.ImageURL.URL, "http") {
-							fmt.Println("链接:", content.ImageURL.URL)
-						} else if strings.HasPrefix(content.ImageURL.URL, "data:image") {
-							fmt.Println("base64:", content.ImageURL.URL[:20])
+		switch ct := msg.Content.(type) {
+		case string:
+			prompt += "<" + msg.Role + ">: " + msg.Content.(string) + "\n"
+		case []any:
+			for _, item := range ct {
+				if m, ok := item.(map[string]interface{}); ok {
+					if m["type"] == "text" {
+						prompt += "<" + msg.Role + ">: " + m["text"].(string) + "\n"
+					} else if m["type"] == "image_url" {
+						if url, ok := m["image_url"].(map[string]interface{}); ok {
+							fmt.Printf("  URL: %v\n", url["url"])
+							if strings.HasPrefix(url["url"].(string), "http") {
+								fmt.Println("网络图片:", url["url"].(string))
+							}
 						}
-						// todo image tokens
 					}
-
 				}
-
 			}
+		default:
+			c.JSON(http.StatusInternalServerError, gin.H{
+				"error": gin.H{
+					"message": "Invalid content type",
+				},
+			})
+			return
 		}
 		if len(chatReq.Tools) > 0 {
 			tooljson, _ := json.Marshal(chatReq.Tools)
@@ -247,16 +251,16 @@ func ChatProxy(c *gin.Context, chatReq *ChatCompletionRequest) {
 		req.Header = c.Request.Header
 		req.Header.Set("api-key", onekey.Key)
 	default:
-		req, err = http.NewRequest(c.Request.Method, OpenAI_Endpoint, bytes.NewReader(chatReq.ToByteJson()))
-		if onekey.EndPoint != "" { // 优先key的endpoint
-			req, err = http.NewRequest(c.Request.Method, onekey.EndPoint+c.Request.RequestURI, bytes.NewReader(chatReq.ToByteJson()))
-		}
+		req, err = http.NewRequest(c.Request.Method, OpenAI_Endpoint, bytes.NewReader(chatReq.ToByteJson())) // default endpoint
+
 		if AIGateWay_Endpoint != "" { // cloudflare gateway的endpoint
 			req, err = http.NewRequest(c.Request.Method, AIGateWay_Endpoint, bytes.NewReader(chatReq.ToByteJson()))
 		}
-		customEndpoint := os.Getenv("CUSTOM_ENDPOINT") // 最后是用户自定义的endpoint CUSTOM_ENDPOINT=true OpenAI_Endpoint
-		if customEndpoint == "true" && OpenAI_Endpoint != "" {
-			req, err = http.NewRequest(c.Request.Method, BaseURL, bytes.NewReader(chatReq.ToByteJson()))
+		if Custom_Endpoint != "" { // 自定义endpoint
+			req, err = http.NewRequest(c.Request.Method, Custom_Endpoint, bytes.NewReader(chatReq.ToByteJson()))
+		}
+		if onekey.EndPoint != "" { // 优先key的endpoint
+			req, err = http.NewRequest(c.Request.Method, onekey.EndPoint+c.Request.RequestURI, bytes.NewReader(chatReq.ToByteJson()))
 		}
 
 		req.Header = c.Request.Header

pkg/tokenizer/tokenizer.go

@@ -183,6 +183,8 @@ func Cost(model string, promptCount, completionCount int) float64 {
 		cost = (0.00035/1000)*float64(prompt) + (0.00053/1000)*float64(completion)
 	case "gemini-2.0-flash-exp":
 		cost = (0.00035/1000)*float64(prompt) + (0.00053/1000)*float64(completion)
+	case "gemini-2.0-flash-thinking-exp-1219":
+		cost = (0.00035/1000)*float64(prompt) + (0.00053/1000)*float64(completion)
 	case "learnlm-1.5-pro-experimental", " gemini-exp-1114", "gemini-exp-1121", "gemini-exp-1206":
 		cost = (0.00035/1000)*float64(prompt) + (0.00053/1000)*float64(completion)
 

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
 	}

router/router.go

@@ -2,22 +2,18 @@ package router
 
 import (
 	"crypto/tls"
-	"fmt"
-	"log"
 	"net"
 	"net/http"
 	"net/http/httputil"
 	"opencatd-open/pkg/claude"
 	oai "opencatd-open/pkg/openai"
 	"opencatd-open/store"
-	"os"
 	"time"
 
 	"github.com/gin-gonic/gin"
 )
 
 var (
-	baseUrl       = "https://api.openai.com"
 	GPT3Dot5Turbo = "gpt-3.5-turbo"
 	GPT4          = "gpt-4"
 )
@@ -62,13 +58,6 @@ var (
 // 	} `json:"usage"`
 // }
 
-func init() {
-	if openai_endpoint := os.Getenv("openai_endpoint"); openai_endpoint != "" {
-		log.Println(fmt.Sprintf("replace %s to %s", baseUrl, openai_endpoint))
-		baseUrl = openai_endpoint
-	}
-}
-
 func HandleProxy(c *gin.Context) {
 	var (
 		localuser bool

store/cache.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 	"log"
 	"math/rand"
+	"os"
 	"strings"
 	"time"
 
@@ -76,6 +77,9 @@ func SelectKeyCache(apitype string) (Key, error) {
 
 func SelectKeyCacheByModel(model string) (Key, error) {
 	var keys []Key
+	if os.Getenv("OPENAI_API_KEY") != "" {
+		keys = append(keys, Key{ApiType: "openai", Key: os.Getenv("OPENAI_API_KEY")})
+	}
 	items := KeysCache.Items()
 	for _, item := range items {
 		if strings.Contains(model, "realtime") {
@@ -97,7 +101,7 @@ func SelectKeyCacheByModel(model string) (Key, error) {
 				keys = append(keys, item.Object.(Key))
 			}
 		}
-		if strings.HasPrefix(model, "o1-") || model == "chatgpt-4o-latest" {
+		if strings.HasPrefix(model, "o1-") || strings.HasPrefix(model, "chatgpt-") {
 			if item.Object.(Key).ApiType == "openai" {
 				keys = append(keys, item.Object.(Key))
 			}