add deepseek direct via userToken

internal/handler/chat.go

@@ -49,6 +49,12 @@ func HandleChatCompletions(w http.ResponseWriter, r *http.Request) {
 	r.Body = io.NopCloser(bytes.NewReader(body))
 	var tempReq map[string]interface{}
 	json.Unmarshal(body, &tempReq)
+	if m, ok := tempReq["model"].(string); ok && IsDeepSeekModel(m) {
+		ok2, reason := CheckAndTrack(apiKey, 0)
+		if !ok2 { http.Error(w, reason, 429); return }
+		HandleDeepSeek(w, r, body, apiKey)
+		return
+	}
 	if m, ok := tempReq["model"].(string); ok && IsOpenRouterModel(m) {
 		ok2, reason := CheckAndTrack(apiKey, 0)
 		if !ok2 { http.Error(w, reason, 429); return }

internal/handler/deepseek.go

@@ -0,0 +1,58 @@
+package handler
+
+import (
+	"bytes"
+	"encoding/json"
+	"io"
+	"net/http"
+	"os"
+	"strings"
+)
+
+var deepseekToken = os.Getenv("DEEPSEEK_TOKEN")
+
+var deepseekModels = map[string]bool{
+	"deepseek":                   true,
+	"deepseek-expert":             true,
+	"deepseek-r1":                 true,
+	"deepseek-search":             true,
+	"deepseek-expert-r1":          true,
+	"deepseek-expert-search":      true,
+	"deepseek-r1-search":          true,
+	"deepseek-expert-r1-search":   true,
+	"deepseek-v4":                 true,
+	"deepseek-v4-flash":           true,
+	"deepseek-v4-pro":             true,
+}
+
+func IsDeepSeekModel(model string) bool {
+	_, ok := deepseekModels[strings.ToLower(model)]
+	return ok
+}
+
+func HandleDeepSeek(w http.ResponseWriter, r *http.Request, body []byte, apiKey string) {
+	var req map[string]interface{}
+	json.Unmarshal(body, &req)
+	newBody, _ := json.Marshal(req)
+	proxyReq, _ := http.NewRequest("POST", "https://chat.deepseek.com/api/v0/chat/completions", bytes.NewReader(newBody))
+	proxyReq.Header.Set("Authorization", "Bearer "+deepseekToken)
+	proxyReq.Header.Set("Content-Type", "application/json")
+	proxyReq.Header.Set("User-Agent", "Mozilla/5.0")
+	proxyReq.Header.Set("Referer", "https://chat.deepseek.com/")
+	client := &http.Client{}
+	resp, err := client.Do(proxyReq)
+	if err != nil { http.Error(w, "DeepSeek error", 500); return }
+	defer resp.Body.Close()
+	respBody, _ := io.ReadAll(resp.Body)
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(resp.StatusCode)
+	w.Write(respBody)
+	var result map[string]interface{}
+	if json.Unmarshal(respBody, &result) == nil {
+		if usage, ok := result["usage"].(map[string]interface{}); ok {
+			if total, ok := usage["total_tokens"].(float64); ok {
+				TrackUsage(apiKey, int(total))
+			}
+		}
+	}
+}

internal/model/mapping.go

@@ -48,6 +48,14 @@ func ResolveClaudeModel(model string, thinkingEnabled bool) (resolvedModel strin
 
 // v1/models 返回的模型列表（全部小写）
 var ModelList = []string{
+	// DEEPSEEK directo
+	"deepseek",
+	"deepseek-expert",
+	"deepseek-r1",
+	"deepseek-search",
+	"deepseek-expert-r1",
+	"deepseek-v4",
+	"deepseek-v4-pro",
 	// DEEPSEEK via OpenRouter
 	"deepseek-r1",
 	"deepseek-v3",