huggingface 支持

api/main.go

@@ -56,6 +56,18 @@ type OpenAIChoice struct {
 	FinishReason string  `json:"finish_reason"`
 }
 
+type ModelResponse struct {
+	Object string        `json:"object"`
+	Data   []ModelDetail `json:"data"`
+}
+
+type ModelDetail struct {
+	ID      string `json:"id"`
+	Object  string `json:"object"`
+	Created int64  `json:"created"`
+	OwnedBy string `json:"owned_by"`
+}
+
 var modelMap = map[string]string{
 	"deepseek-reasoner":  "deepseek_r1",
 	"deepseek-chat":      "deepseek_v3",
@@ -108,7 +120,38 @@ func reverseMapModelName(youModel string) string {
 var originalModel string
 
 func Handler(w http.ResponseWriter, r *http.Request) {
-	if r.URL.Path != "/v1/chat/completions" {
+	if r.URL.Path == "/v1/models" || r.URL.Path == "/api/v1/models" {
+		w.Header().Set("Content-Type", "application/json")
+		w.Header().Set("Access-Control-Allow-Origin", "*")
+		w.Header().Set("Access-Control-Allow-Methods", "GET, OPTIONS")
+		w.Header().Set("Access-Control-Allow-Headers", "*")
+
+		if r.Method == "OPTIONS" {
+			w.WriteHeader(http.StatusOK)
+			return
+		}
+
+		models := make([]ModelDetail, 0, len(modelMap))
+		created := time.Now().Unix()
+		for modelID := range modelMap {
+			models = append(models, ModelDetail{
+				ID:      modelID,
+				Object:  "model",
+				Created: created,
+				OwnedBy: "organization-owner",
+			})
+		}
+
+		response := ModelResponse{
+			Object: "list",
+			Data:   models,
+		}
+
+		json.NewEncoder(w).Encode(response)
+		return
+	}
+
+	if r.URL.Path != "/v1/chat/completions" && r.URL.Path != "/api/v1/chat/completions" {
 		w.Header().Set("Content-Type", "application/json")
 		json.NewEncoder(w).Encode(map[string]string{
 			"status":  "You2Api Service Running...",

huggingDockerfile

@@ -0,0 +1,30 @@
+# 使用最新的 Go 1.22 版本作为基础镜像
+FROM golang:1.22-alpine
+
+# 安装必要的系统依赖
+RUN apk --no-cache add ca-certificates git
+
+# 设置工作目录
+WORKDIR /app
+
+# 克隆项目代码
+RUN git clone https://github.com/bohesocool/you2api.git .
+
+# 下载依赖
+RUN go mod download
+
+# 构建应用
+RUN CGO_ENABLED=0 GOOS=linux go build -o main .
+
+# 设置环境变量
+ENV PORT=8080
+ENV ENABLE_PROXY=false
+ENV PROXY_URL=""
+ENV PROXY_TIMEOUT_MS=5000
+ENV LOG_LEVEL=info
+
+# 暴露端口
+EXPOSE 8080
+
+# 运行应用
+CMD ["./main"]

proxy/proxy.go

@@ -8,9 +8,9 @@ import (
 )
 
 type Proxy struct {
-	target     *url.URL
-	proxy      *httputil.ReverseProxy
-	timeoutMS  int
+	target    *url.URL
+	proxy     *httputil.ReverseProxy
+	timeoutMS int
 }
 
 func NewProxy(targetURL string, timeoutMS int) (*Proxy, error) {
@@ -27,7 +27,7 @@ func NewProxy(targetURL string, timeoutMS int) (*Proxy, error) {
 	proxy.proxy = &httputil.ReverseProxy{
 		Director: proxy.director,
 		Transport: &http.Transport{
-			Proxy: http.ProxyURL(target),
+			Proxy:                 http.ProxyURL(target),
 			ResponseHeaderTimeout: time.Duration(timeoutMS) * time.Millisecond,
 		},
 	}
@@ -43,4 +43,4 @@ func (p *Proxy) director(req *http.Request) {
 
 func (p *Proxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	p.proxy.ServeHTTP(w, r)
-} 
+}