Update main.go

main.go

@@ -2,12 +2,10 @@ package main
 
 import (
 	"bufio"
-	"embed"
 	"encoding/json"
 	"errors"
 	"fmt"
 	"io"
-	"io/fs"
 	"log"
 	"net/http"
 	"os"
@@ -18,9 +16,6 @@ import (
 	"github.com/joho/godotenv"
 )
 
-//go:embed web/*
-var staticFiles embed.FS
-
 type Config struct {
 	APIPrefix     string
 	APIKey        string
@@ -85,32 +80,13 @@ func authMiddleware() gin.HandlerFunc {
 func main() {
 	r := gin.Default()
 	r.Use(corsMiddleware())
-	// 1. 映射 Web 目录到 /web 路由
-	//r.Static("/web", "./web") // 确保 ./web 文件夹存在，并包含 index.html
-	//subFS, err := fs.Sub(staticFiles, "web")
-	//if err != nil {
-	//	log.Fatal(err)
-	//}
-	//r.StaticFS("/web", http.FS(subFS))
-
-	//// 2. 根路径重定向到 /web
-	//r.GET("/", func(c *gin.Context) {
-	//	c.Redirect(http.StatusMovedPermanently, "/web")
-	//})
-	// r.GET("/", func(c *gin.Context) {
-	// 	c.JSON(http.StatusOK, gin.H{"message": "API 服务运行中~"})
-	// })
-	// 3. 健康检查
+
+	// 健康检查
 	r.GET("/ping", func(c *gin.Context) {
 		c.JSON(http.StatusOK, gin.H{"message": "pong"})
 	})
-	// 4. API 路由组
-	// authorized := r.Group("/")
-	// authorized.Use(authMiddleware())
-	// {
-	// 	authorized.GET("/hf/v1/models", handleModels)
-	// 	authorized.POST("/hf/v1/chat/completions", handleCompletion)
-	// }
+
+	// API 路由组
 	apiGroup := r.Group("/")
 	apiGroup.Use(authMiddleware()) // 可以选择性地提供 API 密钥
 	{
@@ -129,7 +105,8 @@ func main() {
 		// 新路径 /completions
 		apiGroup.POST("/completions", handleCompletion)
 	}
-	// 5. 从环境变量中读取端口号
+
+	// 从环境变量中读取端口号
 	port := os.Getenv("PORT")
 	if port == "" {
 		port = "7860"
@@ -164,7 +141,6 @@ func handleCompletion(c *gin.Context) {
 
 	model := convertModel(req.Model)
 	content := prepareMessages(req.Messages)
-	// log.Printf("messages: %v", content)
 
 	reqBody := map[string]interface{}{
 		"model": model,
@@ -238,7 +214,6 @@ func handleCompletion(c *gin.Context) {
 				line = strings.TrimSpace(line)
 				// 忽略非 JSON 数据块（例如特殊标记 [DONE]）
 				if line == "[DONE]" {
-					//log.Printf("响应行 DONE, 即将跳过")
 					break
 				}
 				var chunk map[string]interface{}
@@ -347,7 +322,6 @@ func handleCompletion(c *gin.Context) {
 	}
 }
 
-
 func requestToken() (string, error) {
 	url := "https://duckduckgo.com/duckchat/v1/status"
 	client := &http.Client{
@@ -493,4 +467,4 @@ func getIntEnv(key string, fallback int) int {
 
 func getDurationEnv(key string, fallback int) time.Duration {
 	return time.Duration(getIntEnv(key, fallback)) * time.Millisecond
-}
+}