Create proxy.go

proxy.go

@@ -0,0 +1,75 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"os"
+	"os/exec"
+)
+
+func main() {
+	port := os.Getenv("PORT")
+	if port == "" {
+		port = "7860"
+	}
+
+	// 启动原始服务器
+	cmd := exec.Command("./server", "--port", port)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	if err := cmd.Start(); err != nil {
+		log.Fatalf("Failed to start server: %v", err)
+	}
+
+	// 设置代理处理程序
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path == "/proxies/v1/api/generate" {
+			r.URL.Path = "/proxies/v1/chat/completions"
+		}
+		proxyRequest(w, r)
+	})
+
+	// 启动代理服务器
+	fmt.Printf("Proxy server listening on :%s\n", port)
+	log.Fatal(http.ListenAndServe(":"+port, nil))
+}
+
+func proxyRequest(w http.ResponseWriter, r *http.Request) {
+	// 创建新的请求
+	proxyReq, err := http.NewRequest(r.Method, "http://localhost:"+os.Getenv("PORT")+r.URL.String(), r.Body)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	// 复制原始请求的 header
+	for name, values := range r.Header {
+		for _, value := range values {
+			proxyReq.Header.Add(name, value)
+		}
+	}
+
+	// 发送请求
+	client := &http.Client{}
+	resp, err := client.Do(proxyReq)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadGateway)
+		return
+	}
+	defer resp.Body.Close()
+
+	// 复制响应 header
+	for name, values := range resp.Header {
+		for _, value := range values {
+			w.Header().Add(name, value)
+		}
+	}
+
+	// 设置状态码
+	w.WriteHeader(resp.StatusCode)
+
+	// 复制响应体
+	io.Copy(w, resp.Body)
+}