Update main.go

main.go

@@ -143,6 +143,7 @@ func handleModels(c *gin.Context) {
 		{"id": "claude-3-haiku", "object": "model", "owned_by": "ddg"},
 		{"id": "llama-3.1-70b", "object": "model", "owned_by": "ddg"},
 		{"id": "mixtral-8x7b", "object": "model", "owned_by": "ddg"},
+        {"id": "o3-mini", "object": "model", "owned_by": "ddg"},
 	}
 	c.JSON(http.StatusOK, gin.H{"object": "list", "data": models})
 }
@@ -347,42 +348,7 @@ 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"
@@ -493,6 +459,8 @@ func convertModel(inputModel string) string {
 		return "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo"
 	case "mixtral-8x7b":
 		return "mistralai/Mixtral-8x7B-Instruct-v0.1"
+    	case "o3-mini":
+		return "o3-mini"
 	default:
 		return "gpt-4o-mini"
 	}