add deepseek via openrouter

internal/handler/chat.go

@@ -6,6 +6,8 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"bytes"
+	"io"
 	"strings"
 	"time"
 
@@ -44,6 +46,16 @@ func HandleChatCompletions(w http.ResponseWriter, r *http.Request) {
 		}
 		token = anonymousToken
 	}
+	body, _ := io.ReadAll(r.Body)
+	r.Body = io.NopCloser(bytes.NewReader(body))
+	var tempReq map[string]interface{}
+	json.Unmarshal(body, &tempReq)
+	if m, ok := tempReq["model"].(string); ok && IsOpenRouterModel(m) {
+		ok2, reason := CheckAndTrack(apiKey, 0)
+		if !ok2 { http.Error(w, reason, 429); return }
+		HandleOpenRouter(w, r, body)
+		return
+	}
 	var req model.ChatRequest
 	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
 		http.Error(w, "Invalid request", http.StatusBadRequest)

internal/handler/openrouter.go

@@ -0,0 +1,47 @@
+package handler
+
+import (
+	"bytes"
+	"encoding/json"
+	"io"
+	"net/http"
+	"os"
+)
+
+var openRouterKey = os.Getenv("OP_Rowproxy_key")
+
+var openRouterModels = map[string]string{
+	"deepseek-r1":           "deepseek/deepseek-r1",
+	"deepseek-v3":           "deepseek/deepseek-v3-0324",
+	"deepseek-r1-0528":      "deepseek/deepseek-r1-0528",
+	"deepseek-v3-turbo":     "deepseek/deepseek-v3",
+}
+
+func IsOpenRouterModel(model string) bool {
+	_, ok := openRouterModels[model]
+	return ok
+}
+
+func HandleOpenRouter(w http.ResponseWriter, r *http.Request, body []byte) {
+	var req map[string]interface{}
+	json.Unmarshal(body, &req)
+	if m, ok := req["model"].(string); ok {
+		if mapped, found := openRouterModels[m]; found {
+			req["model"] = mapped
+		}
+	}
+	newBody, _ := json.Marshal(req)
+	proxyReq, _ := http.NewRequest("POST", "https://openrouter.ai/api/v1/chat/completions", bytes.NewReader(newBody))
+	proxyReq.Header.Set("Authorization", "Bearer "+openRouterKey)
+	proxyReq.Header.Set("Content-Type", "application/json")
+	client := &http.Client{}
+	resp, err := client.Do(proxyReq)
+	if err != nil {
+		http.Error(w, "OpenRouter error", 500)
+		return
+	}
+	defer resp.Body.Close()
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(resp.StatusCode)
+	io.Copy(w, resp.Body)
+}

internal/model/mapping.go

@@ -48,6 +48,11 @@ func ResolveClaudeModel(model string, thinkingEnabled bool) (resolvedModel strin
 
 // v1/models 返回的模型列表（全部小写）
 var ModelList = []string{
+	// DEEPSEEK via OpenRouter
+	"deepseek-r1",
+	"deepseek-v3",
+	"deepseek-r1-0528",
+	"deepseek-v3-turbo",
 // CLAUDE
 "claude-sonnet-4-6",
 "claude-sonnet-4-5-20241022",