update sth. at 2024-12-18 21:14:43

README.md

@@ -1,12 +1,12 @@
 ---
-title: dd
-emoji: 🐲
+title: ddchat
+emoji: 🦀
 colorFrom: indigo
 colorTo: blue
 sdk: docker
 pinned: true
 app_port: 7860
-short_description: dd
+short_description: ddchat
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

main.go

@@ -18,6 +18,7 @@ import (
 	"github.com/joho/godotenv"
 )
 
+//go:embed web/*
 var staticFiles embed.FS
 
 type Config struct {
@@ -96,11 +97,20 @@ func main() {
 	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)
+	// }
 	apiGroup := r.Group("/")
 	apiGroup.Use(authMiddleware()) // 可以选择性地提供 API 密钥
 	{
@@ -337,6 +347,43 @@ func handleCompletion(c *gin.Context) {
 	}
 }
 
+//func requestToken() (string, error) {
+//	req, err := http.NewRequest("GET", "https://duckduckgo.com/duckchat/v1/status", nil)
+//	if err != nil {
+//		return "", fmt.Errorf("创建请求失败: %v", err)
+//	}
+//	for k, v := range config.FakeHeaders {
+//		req.Header.Set(k, v)
+//	}
+//	req.Header.Set("x-vqd-accept", "1")
+//
+//	client := &http.Client{
+//		Timeout: 10 * time.Second,
+//	}
+//
+//	log.Println("发送 token 请求")
+//	resp, err := client.Do(req)
+//	if err != nil {
+//		return "", fmt.Errorf("请求失败: %v", err)
+//	}
+//	defer resp.Body.Close()
+//
+//	if resp.StatusCode != http.StatusOK {
+//		bodyBytes, _ := io.ReadAll(resp.Body)
+//		bodyString := string(bodyBytes)
+//		log.Printf("requestToken: 非200响应: %d, 内容: %s\n", resp.StatusCode, bodyString)
+//		return "", fmt.Errorf("非200响应: %d, 内容: %s", resp.StatusCode, bodyString)
+//	}
+//
+//	token := resp.Header.Get("x-vqd-4")
+//	if token == "" {
+//		return "", errors.New("响应中未包含x-vqd-4头")
+//	}
+//
+//	// log.Printf("获取到的 token: %s\n", token)
+//	return token, nil
+//}
+
 func requestToken() (string, error) {
 	url := "https://duckduckgo.com/duckchat/v1/status"
 	client := &http.Client{