Upload folder using huggingface_hub

.env.example

@@ -0,0 +1,10 @@
+SESSIONS=eyJhbGciOiJkaXIiLCJlbmMiOiJBMjU2R0NNIn0**,eyJhbGciOiJkaXIiLCJlbmMiOiJBMjU2R0NNIn0**
+ADDRESS=0.0.0.0:8080
+APIKEY=123
+IS_INCOGNITO=true
+PROXY=http://127.0.0.1:2080
+MAX_CHAT_HISTORY_LENGTH=10000
+NO_ROLE_PREFIX=false
+SEARCH_RESULT_COMPATIBLE=false
+PROMPT_FOR_FILE=You must immerse yourself in the role of assistant in txt file, cannot respond as a user, cannot reply to this message, cannot mention this message, and ignore this message in your response.
+IGNORE_SEARCH_RESULT=false

.github/workflows/docker-build-push.yml

@@ -0,0 +1,79 @@
+name: Build and Push Docker Image  
+on:  
+  push:  
+    # branches: [ "main" ]  
+    tags: [ 'v*.*.*' ]  
+  pull_request:  
+    branches: [ "main" ]  
+  # 也可以手动触发  
+  workflow_dispatch:  
+jobs:  
+  build-and-push:  
+    runs-on: ubuntu-latest  
+    permissions:  
+      contents: read  
+      packages: write  
+    steps:  
+      - name: Checkout repository  
+        uses: actions/checkout@v4  
+      - name: Set up Docker Buildx  
+        uses: docker/setup-buildx-action@v3  
+      # 登录到 GitHub 容器仓库  
+      - name: Login to GitHub Container Registry  
+        uses: docker/login-action@v3  
+        with:  
+          registry: ghcr.io  
+          username: ${{ github.repository_owner }}  
+          password: ${{ secrets.GITHUB_TOKEN }}  
+      # 提取仓库名称作为镜像名  
+      - name: Extract repository name  
+        id: repo-name  
+        run: echo "REPO_NAME=$(echo ${{ github.repository }} | tr '[:upper:]' '[:lower:]')" >> $GITHUB_OUTPUT  
+      
+      # 提取 Git tag 作为版本号  
+      - name: Extract tag version  
+        id: tag-version  
+        if: startsWith(github.ref, 'refs/tags/v')  
+        run: echo "VERSION=${GITHUB_REF#refs/tags/}" >> $GITHUB_OUTPUT  
+
+      # 设置默认版本标签  
+      - name: Set default version  
+        id: version  
+        run: |  
+          if [[ "${{ github.ref }}" == refs/tags/v* ]]; then  
+            echo "TAG=${{ steps.tag-version.outputs.VERSION }}" >> $GITHUB_OUTPUT  
+          else  
+            SHORT_SHA=$(echo ${GITHUB_SHA} | cut -c1-7)  
+            DATE=$(date +'%Y%m%d')  
+            echo "TAG=${DATE}-${SHORT_SHA}" >> $GITHUB_OUTPUT  
+          fi  
+
+      - name: Build and Push Images  
+        uses: docker/build-push-action@v5  
+        with:  
+          context: .  
+          push: true  
+          tags: |  
+            ghcr.io/${{ steps.repo-name.outputs.REPO_NAME }}:latest  
+            ghcr.io/${{ steps.repo-name.outputs.REPO_NAME }}:${{ steps.version.outputs.TAG }}  
+      
+      # 专门为 ARM64 架构构建并推送镜像  
+      - name: Build and Push ARM64 Image  
+        uses: docker/build-push-action@v5  
+        with:  
+          context: .  
+          push: true  
+          platforms: linux/arm64  
+          tags: |  
+            ghcr.io/${{ steps.repo-name.outputs.REPO_NAME }}:arm-latest  
+            ${{ github.ref_type == 'tag' && format('ghcr.io/{0}:arm-{1}', steps.repo-name.outputs.REPO_NAME, steps.version.outputs.TAG) || '' }}  
+      
+      # 输出镜像信息  
+      - name: Image digest  
+        run: |  
+          echo " image has been pushed to ghcr.io/${{ steps.repo-name.outputs.REPO_NAME }}:latest"  
+          echo " image has been pushed to ghcr.io/${{ steps.repo-name.outputs.REPO_NAME }}:${{ steps.version.outputs.TAG }}"  
+          echo "ARM64-only image has been pushed to ghcr.io/${{ steps.repo-name.outputs.REPO_NAME }}:arm-latest"  
+          if [[ "${{ github.ref }}" == refs/tags/v* ]]; then  
+            echo "ARM64-only image has been pushed to ghcr.io/${{ steps.repo-name.outputs.REPO_NAME }}:arm-${{ steps.version.outputs.TAG }}"  
+          fi  

.github/workflows/release.yml

@@ -0,0 +1,53 @@
+name: Publish Release  
+
+on:  
+  push:  
+    tags:  
+      - 'v*' # 当推送以 "v" 开头的标签时触发（如 v1.0.0, v2.1.0）  
+
+jobs:  
+  update-release-draft:  
+    permissions:  
+      contents: write  
+      pull-requests: write  
+    runs-on: ubuntu-latest  
+    steps:  
+      # Step 1: 检出代码库  
+      - name: Checkout code  
+        uses: actions/checkout@v3  
+
+      # Step 2: 获取项目名称  
+      - name: Get Project Name  
+        id: project_info  
+        run: |  
+          REPO_NAME=${GITHUB_REPOSITORY#*/}  
+          echo "repo_name=$REPO_NAME" >> $GITHUB_OUTPUT  
+          echo "Using repository name: $REPO_NAME"  
+
+      # Step 3: 自动生成 Release  
+      - name: Create Release  
+        id: create_release  
+        uses: actions/create-release@v1  
+        env:  
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}  
+        with:  
+          tag_name: ${{ github.ref_name }}  
+          release_name: ${{ github.ref_name }}  
+          draft: false  
+          prerelease: false  
+      
+      # Step 4: 构建zip文件  
+      - name: Create ZIP file  
+        run: |  
+          zip -r ${{ steps.project_info.outputs.repo_name }}.zip . -x "*.git*" "*.github*" "*.env*" "logs/*" "tests/*"  
+
+      # Step 5: 上传构建文件  
+      - name: Upload Release Asset  
+        uses: actions/upload-release-asset@v1  
+        env:  
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}  
+        with:  
+          upload_url: ${{ steps.create_release.outputs.upload_url }}  
+          asset_path: ./${{ steps.project_info.outputs.repo_name }}.zip  
+          asset_name: ${{ steps.project_info.outputs.repo_name }}.zip  
+          asset_content_type: application/zip  

Dockerfile

@@ -0,0 +1,27 @@
+# Start from the official Golang image  
+FROM golang:1.23-alpine AS build  
+
+# Set working directory  
+WORKDIR /app  
+
+# Copy go.mod and go.sum files first for better caching  
+COPY go.mod go.sum* ./  
+
+# Download dependencies  
+RUN go mod download  
+
+# Copy the source code  
+COPY . .  
+
+# Build the application  
+RUN CGO_ENABLED=0 GOOS=linux go build -o main ./main.go  
+
+# Create a minimal production image  
+FROM alpine:latest  
+
+# Create app directory and set permissions  
+WORKDIR /app  
+COPY --from=build /app/main .  
+
+# Command to run the executable  
+CMD ["./main"]  

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 yuxiao
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,10 +1,15 @@
 ---
-title: Newper
-emoji: 🐢
-colorFrom: green
-colorTo: purple
+title: "newper"
+emoji: "🚀"
+colorFrom: blue
+colorTo: green
 sdk: docker
-pinned: false
+app_port: 8080
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+### 🚀 一键部署
+[![Deploy with HFSpaceDeploy](https://img.shields.io/badge/Deploy_with-HFSpaceDeploy-green?style=social&logo=rocket)](https://github.com/kfcx/HFSpaceDeploy)
+
+本项目由[HFSpaceDeploy](https://github.com/kfcx/HFSpaceDeploy)一键部署
+
+

config/config.go

@@ -0,0 +1,156 @@
+package config
+
+import (
+	"fmt"
+	"math/rand"
+	"os"
+	"pplx2api/logger"
+	"strconv"
+	"strings"
+	"sync"
+	"time"
+
+	"github.com/joho/godotenv"
+)
+
+type SessionInfo struct {
+	SessionKey string
+}
+
+type SessionRagen struct {
+	Index int
+	Mutex sync.Mutex
+}
+
+type Config struct {
+	Sessions               []SessionInfo
+	Address                string
+	APIKey                 string
+	Proxy                  string
+	IsIncognito            bool
+	MaxChatHistoryLength   int
+	RetryCount             int
+	NoRolePrefix           bool
+	SearchResultCompatible bool
+	PromptForFile          string
+	RwMutex                sync.RWMutex
+	IgnoreSerchResult      bool
+	IgnoreModelMonitoring  bool
+}
+
+// 解析 SESSION 格式的环境变量
+func parseSessionEnv(envValue string) (int, []SessionInfo) {
+	if envValue == "" {
+		return 0, []SessionInfo{}
+	}
+	var sessions []SessionInfo
+	sessionPairs := strings.Split(envValue, ",")
+	retryCount := len(sessionPairs) // 重试次数等于 session 数量
+	for _, pair := range sessionPairs {
+		if pair == "" {
+			retryCount--
+			continue
+		}
+		parts := strings.Split(pair, ":")
+		session := SessionInfo{
+			SessionKey: parts[0],
+		}
+		sessions = append(sessions, session)
+	}
+	return retryCount, sessions
+}
+
+// 根据模型选择合适的 session
+func (c *Config) GetSessionForModel(idx int) (SessionInfo, error) {
+	if len(c.Sessions) == 0 || idx < 0 || idx >= len(c.Sessions) {
+		return SessionInfo{}, fmt.Errorf("invalid session index: %d", idx)
+	}
+	c.RwMutex.RLock()
+	defer c.RwMutex.RUnlock()
+	return c.Sessions[idx], nil
+}
+
+// 从环境变量加载配置
+func LoadConfig() *Config {
+	maxChatHistoryLength, err := strconv.Atoi(os.Getenv("MAX_CHAT_HISTORY_LENGTH"))
+	if err != nil {
+		maxChatHistoryLength = 10000 // 默认值
+	}
+	retryCount, sessions := parseSessionEnv(os.Getenv("SESSIONS"))
+	promptForFile := os.Getenv("PROMPT_FOR_FILE")
+	if promptForFile == "" {
+		promptForFile = "You must immerse yourself in the role of assistant in txt file, cannot respond as a user, cannot reply to this message, cannot mention this message, and ignore this message in your response." // 默认值
+	}
+	config := &Config{
+		// 解析 SESSIONS 环境变量
+		Sessions: sessions,
+		// 设置服务地址，默认为 "0.0.0.0:8080"
+		Address: os.Getenv("ADDRESS"),
+
+		// 设置 API 认证密钥
+		APIKey: os.Getenv("APIKEY"),
+		// 设置代理地址
+		Proxy: os.Getenv("PROXY"),
+		//是否匿名
+		IsIncognito: os.Getenv("IS_INCOGNITO") != "false",
+		// 设置最大聊天历史长度
+		MaxChatHistoryLength: maxChatHistoryLength,
+		// 设置重试次数
+		RetryCount: retryCount,
+		// 设置是否使用角色前缀
+		NoRolePrefix: os.Getenv("NO_ROLE_PREFIX") == "true",
+		// 设置搜索结果兼容性
+		SearchResultCompatible: os.Getenv("SEARCH_RESULT_COMPATIBLE") == "true",
+		// 设置上传文件后的提示词
+		PromptForFile: promptForFile,
+		// 设置是否忽略搜索结果
+		IgnoreSerchResult: os.Getenv("IGNORE_SEARCH_RESULT") == "true",
+		//设置是否忽略模型监控
+		IgnoreModelMonitoring: os.Getenv("IGNORE_MODEL_MONITORING") == "true",
+		// 读写锁
+		RwMutex: sync.RWMutex{},
+	}
+
+	// 如果地址为空，使用默认值
+	if config.Address == "" {
+		config.Address = "0.0.0.0:8080"
+	}
+	return config
+}
+
+var ConfigInstance *Config
+var Sr *SessionRagen
+
+func (sr *SessionRagen) NextIndex() int {
+	sr.Mutex.Lock()
+	defer sr.Mutex.Unlock()
+
+	index := sr.Index
+	sr.Index = (index + 1) % len(ConfigInstance.Sessions)
+	return index
+}
+func init() {
+	rand.Seed(time.Now().UnixNano())
+	// 加载环境变量
+	_ = godotenv.Load()
+	Sr = &SessionRagen{
+		Index: 0,
+		Mutex: sync.Mutex{},
+	}
+	ConfigInstance = LoadConfig()
+	logger.Info("Loaded config:")
+	logger.Info(fmt.Sprintf("Sessions count: %d", ConfigInstance.RetryCount))
+	for _, session := range ConfigInstance.Sessions {
+		logger.Info(fmt.Sprintf("Session: %s", session.SessionKey))
+	}
+	logger.Info(fmt.Sprintf("Address: %s", ConfigInstance.Address))
+	logger.Info(fmt.Sprintf("APIKey: %s", ConfigInstance.APIKey))
+	logger.Info(fmt.Sprintf("Proxy: %s", ConfigInstance.Proxy))
+	logger.Info(fmt.Sprintf("IsIncognito: %t", ConfigInstance.IsIncognito))
+	logger.Info(fmt.Sprintf("MaxChatHistoryLength: %d", ConfigInstance.MaxChatHistoryLength))
+	logger.Info(fmt.Sprintf("NoRolePrefix: %t", ConfigInstance.NoRolePrefix))
+	logger.Info(fmt.Sprintf("SearchResultCompatible: %t", ConfigInstance.SearchResultCompatible))
+	logger.Info(fmt.Sprintf("PromptForFile: %s", ConfigInstance.PromptForFile))
+	logger.Info(fmt.Sprintf("IgnoreSerchResult: %t", ConfigInstance.IgnoreSerchResult))
+	logger.Info(fmt.Sprintf("IgnoreModelMonitoring: %t", ConfigInstance.IgnoreModelMonitoring))
+}

config/model.go

@@ -0,0 +1,50 @@
+package config
+
+var ModelReverseMap = map[string]string{}
+var ModelMap = map[string]string{
+	"claude-4.0-sonnet":       "claude2",
+	"claude-4.0-sonnet-think": "claude37sonnetthinking",
+	"deepseek-r1":             "r1",
+	"o4-mini":                 "o4mini",
+	"gpt-4o":                  "gpt4o",
+	"gemini-2.5-pro-06-05":    "gemini2flash",
+	"grok4":                   "grok4",
+	"gpt-4.1":                 "gpt41",
+	// "claude-4.0-opus":         "claude40opus",
+	"claude-4.0-opus-think": "claude40opusthinking",
+	"o3":                    "o3",
+	"o3-pro":                "o3pro",
+}
+
+// Get returns the value for the given key from the ModelMap.
+// If the key doesn't exist, it returns the provided default value.
+func ModelMapGet(key string, defaultValue string) string {
+	if value, exists := ModelMap[key]; exists {
+		return value
+	}
+	return defaultValue
+}
+
+// GetReverse returns the value for the given key from the ModelReverseMap.
+// If the key doesn't exist, it returns the provided default value.
+func ModelReverseMapGet(key string, defaultValue string) string {
+	if value, exists := ModelReverseMap[key]; exists {
+		return value
+	}
+	return defaultValue
+}
+
+var ResponseModles []map[string]string
+
+func init() {
+	for k, v := range ModelMap {
+		ModelReverseMap[v] = k
+		model := map[string]string{
+			"id": k,
+		}
+		modelSearch := map[string]string{
+			"id": k + "-search",
+		}
+		ResponseModles = append(ResponseModles, model, modelSearch)
+	}
+}

core/api.go

@@ -0,0 +1,610 @@
+package core
+
+import (
+	"bufio"
+	"bytes"
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"io"
+	"mime/multipart"
+	"net/http"
+	"pplx2api/config"
+	"pplx2api/logger"
+	"pplx2api/model"
+	"pplx2api/utils"
+	"strings"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+	"github.com/imroc/req/v3"
+)
+
+// Client represents a Perplexity API client
+type Client struct {
+	sessionToken string
+	client       *req.Client
+	Model        string
+	Attachments  []string
+	OpenSerch    bool
+}
+
+// Perplexity API structures
+type PerplexityRequest struct {
+	Params   PerplexityParams `json:"params"`
+	QueryStr string           `json:"query_str"`
+}
+
+type PerplexityParams struct {
+	Attachments              []string      `json:"attachments"`
+	Language                 string        `json:"language"`
+	Timezone                 string        `json:"timezone"`
+	SearchFocus              string        `json:"search_focus"`
+	Sources                  []string      `json:"sources"`
+	SearchRecencyFilter      interface{}   `json:"search_recency_filter"`
+	FrontendUUID             string        `json:"frontend_uuid"`
+	Mode                     string        `json:"mode"`
+	ModelPreference          string        `json:"model_preference"`
+	IsRelatedQuery           bool          `json:"is_related_query"`
+	IsSponsored              bool          `json:"is_sponsored"`
+	VisitorID                string        `json:"visitor_id"`
+	UserNextauthID           string        `json:"user_nextauth_id"`
+	FrontendContextUUID      string        `json:"frontend_context_uuid"`
+	PromptSource             string        `json:"prompt_source"`
+	QuerySource              string        `json:"query_source"`
+	BrowserHistorySummary    []interface{} `json:"browser_history_summary"`
+	IsIncognito              bool          `json:"is_incognito"`
+	UseSchematizedAPI        bool          `json:"use_schematized_api"`
+	SendBackTextInStreaming  bool          `json:"send_back_text_in_streaming_api"`
+	SupportedBlockUseCases   []string      `json:"supported_block_use_cases"`
+	ClientCoordinates        interface{}   `json:"client_coordinates"`
+	IsNavSuggestionsDisabled bool          `json:"is_nav_suggestions_disabled"`
+	Version                  string        `json:"version"`
+}
+
+// Response structures
+type PerplexityResponse struct {
+	Blocks       []Block `json:"blocks"`
+	Status       string  `json:"status"`
+	DisplayModel string  `json:"display_model"`
+}
+
+type Block struct {
+	MarkdownBlock      *MarkdownBlock      `json:"markdown_block,omitempty"`
+	ReasoningPlanBlock *ReasoningPlanBlock `json:"reasoning_plan_block,omitempty"`
+	WebResultBlock     *WebResultBlock     `json:"web_result_block,omitempty"`
+	ImageModeBlock     *ImageModeBlock     `json:"image_mode_block,omitempty"`
+}
+
+type MarkdownBlock struct {
+	Chunks []string `json:"chunks"`
+}
+
+type ReasoningPlanBlock struct {
+	Goals []Goal `json:"goals"`
+}
+
+type Goal struct {
+	Description string `json:"description"`
+}
+
+type WebResultBlock struct {
+	WebResults []WebResult `json:"web_results"`
+}
+
+type WebResult struct {
+	Name    string `json:"name"`
+	Snippet string `json:"snippet"`
+	URL     string `json:"url"`
+}
+
+type ImageModeBlock struct {
+	AnswerModeType string `json:"answer_mode_type"`
+	Progress       string `json:"progress"`
+	MediaItems     []struct {
+		Medium    string `json:"medium"`
+		Image     string `json:"image"`
+		URL       string `json:"url"`
+		Name      string `json:"name"`
+		Source    string `json:"source"`
+		Thumbnail string `json:"thumbnail"`
+	} `json:"media_items"`
+}
+
+// NewClient creates a new Perplexity API client
+func NewClient(sessionToken string, proxy string, model string, openSerch bool) *Client {
+	client := req.C().ImpersonateChrome().SetTimeout(time.Minute * 10)
+	client.Transport.SetResponseHeaderTimeout(time.Second * 10)
+	if proxy != "" {
+		client.SetProxyURL(proxy)
+	}
+
+	// Set common headers
+	headers := map[string]string{
+		"accept-language": "en-US,en;q=0.9,zh-CN;q=0.8,zh;q=0.7,zh-TW;q=0.6",
+		"cache-control":   "no-cache",
+		"origin":          "https://www.perplexity.ai",
+		"pragma":          "no-cache",
+		"priority":        "u=1, i",
+		"referer":         "https://www.perplexity.ai/",
+	}
+
+	for key, value := range headers {
+		client.SetCommonHeader(key, value)
+	}
+
+	// Set cookies
+	if sessionToken != "" {
+		client.SetCommonCookies(&http.Cookie{
+			Name:  "__Secure-next-auth.session-token",
+			Value: sessionToken,
+		})
+	}
+
+	// Create client with visitor ID
+	c := &Client{
+		sessionToken: sessionToken,
+		client:       client,
+		Model:        model,
+		Attachments:  []string{},
+		OpenSerch:    openSerch,
+	}
+
+	return c
+}
+
+// SendMessage sends a message to Perplexity and returns the status and response
+func (c *Client) SendMessage(message string, stream bool, is_incognito bool, gc *gin.Context) (int, error) {
+	// Create request body
+	requestBody := PerplexityRequest{
+		Params: PerplexityParams{
+			Attachments: c.Attachments,
+			Language:    "en-US",
+			Timezone:    "America/New_York",
+			SearchFocus: "writing",
+			Sources:     []string{},
+			// SearchFocus:             "internet",
+			// Sources:                 []string{"web"},
+			SearchRecencyFilter:     nil,
+			FrontendUUID:            uuid.New().String(),
+			Mode:                    "copilot",
+			ModelPreference:         c.Model,
+			IsRelatedQuery:          false,
+			IsSponsored:             false,
+			VisitorID:               uuid.New().String(),
+			UserNextauthID:          uuid.New().String(),
+			FrontendContextUUID:     uuid.New().String(),
+			PromptSource:            "user",
+			QuerySource:             "home",
+			BrowserHistorySummary:   []interface{}{},
+			IsIncognito:             is_incognito,
+			UseSchematizedAPI:       true,
+			SendBackTextInStreaming: false,
+			SupportedBlockUseCases: []string{
+				"answer_modes",
+				"media_items",
+				"knowledge_cards",
+				"inline_entity_cards",
+				"place_widgets",
+				"finance_widgets",
+				"sports_widgets",
+				"shopping_widgets",
+				"jobs_widgets",
+				"search_result_widgets",
+				"entity_list_answer",
+				"todo_list",
+			},
+			ClientCoordinates:        nil,
+			IsNavSuggestionsDisabled: false,
+			Version:                  "2.18",
+		},
+		QueryStr: message,
+	}
+	if c.OpenSerch {
+		requestBody.Params.SearchFocus = "internet"
+		requestBody.Params.Sources = append(requestBody.Params.Sources, "web")
+	}
+	logger.Info(fmt.Sprintf("Perplexity request body: %v", requestBody))
+	// Make the request
+	resp, err := c.client.R().DisableAutoReadResponse().
+		SetBody(requestBody).
+		Post("https://www.perplexity.ai/rest/sse/perplexity_ask")
+
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error sending request: %v", err))
+		return 500, fmt.Errorf("request failed: %w", err)
+	}
+
+	logger.Info(fmt.Sprintf("Perplexity response status code: %d", resp.StatusCode))
+
+	if resp.StatusCode == http.StatusTooManyRequests {
+		resp.Body.Close()
+		return http.StatusTooManyRequests, fmt.Errorf("rate limit exceeded")
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		logger.Error(fmt.Sprintf("Unexpected return data: %s", resp.String()))
+		resp.Body.Close()
+		return resp.StatusCode, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
+	}
+
+	return 200, c.HandleResponse(resp.Body, stream, gc)
+}
+
+func (c *Client) HandleResponse(body io.ReadCloser, stream bool, gc *gin.Context) error {
+	defer body.Close()
+	// Set headers for streaming
+	if stream {
+		gc.Writer.Header().Set("Content-Type", "text/event-stream")
+		gc.Writer.Header().Set("Cache-Control", "no-cache")
+		gc.Writer.Header().Set("Connection", "keep-alive")
+		gc.Writer.WriteHeader(http.StatusOK)
+		gc.Writer.Flush()
+	}
+	scanner := bufio.NewScanner(body)
+	clientDone := gc.Request.Context().Done()
+	// 增大缓冲区大小
+	scanner.Buffer(make([]byte, 1024*1024), 1024*1024)
+	full_text := ""
+	inThinking := false
+	thinkShown := false
+	final := false
+	for scanner.Scan() {
+		select {
+		case <-clientDone:
+			logger.Info("Client connection closed")
+			return nil
+		default:
+		}
+
+		line := scanner.Text()
+		// Skip empty lines
+		if line == "" {
+			continue
+		}
+		if !strings.HasPrefix(line, "data: ") {
+			continue
+		}
+		data := line[6:]
+		// logger.Info(fmt.Sprintf("Received data: %s", data))
+		var response PerplexityResponse
+		if err := json.Unmarshal([]byte(data), &response); err != nil {
+			logger.Error(fmt.Sprintf("Error parsing JSON: %v", err))
+			continue
+		}
+		// Check for completion and web results
+		if response.Status == "COMPLETED" {
+			final = true
+			for _, block := range response.Blocks {
+				if block.ImageModeBlock != nil && block.ImageModeBlock.Progress == "DONE" && len(block.ImageModeBlock.MediaItems) > 0 {
+					imageResultsText := ""
+					imageModelList := []string{}
+					for i, result := range block.ImageModeBlock.MediaItems {
+						imageResultsText += utils.ImageShow(i, result.Name, result.Image)
+						imageModelList = append(imageModelList, result.Name)
+
+					}
+					if len(imageModelList) > 0 {
+						imageResultsText = imageResultsText + "\n\n---\n" + strings.Join(imageModelList, ", ")
+					}
+					full_text += imageResultsText
+
+					if stream {
+						model.ReturnOpenAIResponse(imageResultsText, stream, gc)
+					}
+				}
+			}
+			for _, block := range response.Blocks {
+				if !config.ConfigInstance.IgnoreSerchResult && block.WebResultBlock != nil && len(block.WebResultBlock.WebResults) > 0 {
+					webResultsText := "\n\n---\n"
+					for i, result := range block.WebResultBlock.WebResults {
+						webResultsText += "\n\n" + utils.SearchShow(i, result.Name, result.URL, result.Snippet)
+					}
+					full_text += webResultsText
+
+					if stream {
+						model.ReturnOpenAIResponse(webResultsText, stream, gc)
+					}
+				}
+
+			}
+
+			if !config.ConfigInstance.IgnoreModelMonitoring && response.DisplayModel != c.Model {
+				res_text := "\n\n---\n"
+				res_text += fmt.Sprintf("Display Model: %s\n", config.ModelReverseMapGet(response.DisplayModel, response.DisplayModel))
+				full_text += res_text
+				if !stream {
+					break
+				}
+				model.ReturnOpenAIResponse(res_text, stream, gc)
+			}
+		}
+		if final {
+			break
+		}
+		// Process each block in the response
+		for _, block := range response.Blocks {
+			// Handle reasoning plan blocks (thinking)
+			if block.ReasoningPlanBlock != nil && len(block.ReasoningPlanBlock.Goals) > 0 {
+
+				res_text := ""
+				if !inThinking && !thinkShown {
+					res_text += "<think>"
+					inThinking = true
+				}
+
+				for _, goal := range block.ReasoningPlanBlock.Goals {
+					if goal.Description != "" && goal.Description != "Beginning analysis" && goal.Description != "Wrapping up analysis" {
+						res_text += goal.Description
+					}
+				}
+				full_text += res_text
+				if !stream {
+					continue
+				}
+				model.ReturnOpenAIResponse(res_text, stream, gc)
+			}
+		}
+		for _, block := range response.Blocks {
+			if block.MarkdownBlock != nil && len(block.MarkdownBlock.Chunks) > 0 {
+				res_text := ""
+				if inThinking {
+					res_text += "</think>\n\n"
+					inThinking = false
+					thinkShown = true
+				}
+				for _, chunk := range block.MarkdownBlock.Chunks {
+					if chunk != "" {
+						res_text += chunk
+					}
+				}
+				full_text += res_text
+				if !stream {
+					continue
+				}
+				model.ReturnOpenAIResponse(res_text, stream, gc)
+			}
+		}
+
+	}
+
+	if err := scanner.Err(); err != nil {
+		return fmt.Errorf("error reading response: %w", err)
+	}
+
+	if !stream {
+		model.ReturnOpenAIResponse(full_text, stream, gc)
+	} else {
+		// Send end marker for streaming mode
+		gc.Writer.Write([]byte("data: [DONE]\n\n"))
+		gc.Writer.Flush()
+	}
+
+	return nil
+}
+
+// UploadURLResponse represents the response from the create_upload_url endpoint
+type UploadURLResponse struct {
+	S3BucketURL string               `json:"s3_bucket_url"`
+	S3ObjectURL string               `json:"s3_object_url"`
+	Fields      CloudinaryUploadInfo `json:"fields"`
+	RateLimited bool                 `json:"rate_limited"`
+}
+
+type CloudinaryUploadInfo struct {
+	Timestamp         int    `json:"timestamp"`
+	UniqueFilename    string `json:"unique_filename"`
+	Folder            string `json:"folder"`
+	UseFilename       string `json:"use_filename"`
+	PublicID          string `json:"public_id"`
+	Transformation    string `json:"transformation"`
+	Moderation        string `json:"moderation"`
+	ResourceType      string `json:"resource_type"`
+	APIKey            string `json:"api_key"`
+	CloudName         string `json:"cloud_name"`
+	Signature         string `json:"signature"`
+	AWSAccessKeyId    string `json:"AWSAccessKeyId"`
+	Key               string `json:"key"`
+	Tagging           string `json:"tagging"`
+	Policy            string `json:"policy"`
+	Xamzsecuritytoken string `json:"x-amz-security-token"`
+	ACL               string `json:"acl"`
+}
+
+// UploadFile is a placeholder for file upload functionality
+func (c *Client) createUploadURL(filename string, contentType string) (*UploadURLResponse, error) {
+	requestBody := map[string]interface{}{
+		"filename":     filename,
+		"content_type": contentType,
+		"source":       "default",
+		"file_size":    12000,
+		"force_image":  false,
+	}
+	resp, err := c.client.R().
+		SetBody(requestBody).
+		Post("https://www.perplexity.ai/rest/uploads/create_upload_url?version=2.18&source=default")
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error creating upload URL: %v", err))
+		return nil, err
+	}
+	if resp.StatusCode != http.StatusOK {
+		logger.Error(fmt.Sprintf("Image Upload with status code %d: %s", resp.StatusCode, resp.String()))
+		return nil, fmt.Errorf("unexpected status code: %d", resp.StatusCode)
+	}
+	var uploadURLResponse UploadURLResponse
+	logger.Info(fmt.Sprintf("Create upload with status code %d: %s", resp.StatusCode, resp.String()))
+	if err := json.Unmarshal(resp.Bytes(), &uploadURLResponse); err != nil {
+		logger.Error(fmt.Sprintf("Error unmarshalling upload URL response: %v", err))
+		return nil, err
+	}
+	if uploadURLResponse.RateLimited {
+		logger.Error("Rate limit exceeded for upload URL")
+		return nil, fmt.Errorf("rate limit exceeded")
+	}
+	return &uploadURLResponse, nil
+
+}
+
+func (c *Client) UploadImage(img_list []string) error {
+	logger.Info(fmt.Sprintf("Uploading %d images to Cloudinary", len(img_list)))
+
+	// Upload images to Cloudinary
+	for _, img := range img_list {
+		filename := utils.RandomString(5) + ".jpg"
+		// Create upload URL
+		uploadURLResponse, err := c.createUploadURL(filename, "image/jpeg")
+		if err != nil {
+			logger.Error(fmt.Sprintf("Error creating upload URL: %v", err))
+			return err
+		}
+		logger.Info(fmt.Sprintf("Upload URL response: %v", uploadURLResponse))
+		// Upload image to Cloudinary
+		err = c.UloadFileToCloudinary(uploadURLResponse.Fields, "img", img, filename)
+		if err != nil {
+			logger.Error(fmt.Sprintf("Error uploading image: %v", err))
+			return err
+		}
+	}
+	return nil
+}
+
+func (c *Client) UloadFileToCloudinary(uploadInfo CloudinaryUploadInfo, contentType string, filedata string, filename string) error {
+	if len(filedata) > 100 {
+		logger.Info(fmt.Sprintf("filedata: %s ……", filedata[:50]))
+	}
+	// Add form fields
+	logger.Info(fmt.Sprintf("Uploading file %s to Cloudinary", filename))
+	var formFields map[string]string
+	if contentType == "img" {
+		formFields = map[string]string{
+			"timestamp":       fmt.Sprintf("%d", uploadInfo.Timestamp),
+			"unique_filename": uploadInfo.UniqueFilename,
+			"folder":          uploadInfo.Folder,
+			"use_filename":    uploadInfo.UseFilename,
+			"public_id":       uploadInfo.PublicID,
+			"transformation":  uploadInfo.Transformation,
+			"moderation":      uploadInfo.Moderation,
+			"resource_type":   uploadInfo.ResourceType,
+			"api_key":         uploadInfo.APIKey,
+			"cloud_name":      uploadInfo.CloudName,
+			"signature":       uploadInfo.Signature,
+			"type":            "private",
+		}
+	} else {
+		formFields = map[string]string{
+			"acl":                  uploadInfo.ACL,
+			"Content-Type":         "text/plain",
+			"tagging":              uploadInfo.Tagging,
+			"key":                  uploadInfo.Key,
+			"AWSAccessKeyId":       uploadInfo.AWSAccessKeyId,
+			"x-amz-security-token": uploadInfo.Xamzsecuritytoken,
+			"policy":               uploadInfo.Policy,
+			"signature":            uploadInfo.Signature,
+		}
+	}
+	var requestBody bytes.Buffer
+	writer := multipart.NewWriter(&requestBody)
+	for key, value := range formFields {
+		if err := writer.WriteField(key, value); err != nil {
+			logger.Error(fmt.Sprintf("Error writing form field %s: %v", key, err))
+			return err
+		}
+	}
+
+	// Add the file,filedata 是base64编码的字符串
+	decodedData, err := base64.StdEncoding.DecodeString(filedata)
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error decoding base64 data: %v", err))
+		return err
+	}
+
+	// 创建一个文件部分
+	part, err := writer.CreateFormFile("file", filename) // 替换 filename.ext 为实际文件名
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error creating form file: %v", err))
+		return err
+	}
+
+	// 将解码后的数据写入文件部分
+	if _, err := part.Write(decodedData); err != nil {
+		logger.Error(fmt.Sprintf("Error writing file data: %v", err))
+		return err
+	}
+	// Close the writer to finalize the form
+	if err := writer.Close(); err != nil {
+		logger.Error(fmt.Sprintf("Error closing writer: %v", err))
+		return err
+	}
+
+	// Create the upload request
+	var uploadURL string
+	if contentType == "img" {
+		uploadURL = fmt.Sprintf("https://api.cloudinary.com/v1_1/%s/image/upload", uploadInfo.CloudName)
+	} else {
+		uploadURL = "https://ppl-ai-file-upload.s3.amazonaws.com/"
+	}
+
+	resp, err := c.client.R().
+		SetHeader("Content-Type", writer.FormDataContentType()).
+		SetBodyBytes(requestBody.Bytes()).
+		Post(uploadURL)
+
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error uploading file: %v", err))
+		return err
+	}
+	logger.Info(fmt.Sprintf("Image Upload with status code %d: %s", resp.StatusCode, resp.String()))
+	if contentType == "img" {
+		var uploadResponse map[string]interface{}
+		if err := json.Unmarshal(resp.Bytes(), &uploadResponse); err != nil {
+			return err
+		}
+		imgUrl := uploadResponse["secure_url"].(string)
+		imgUrl = "https://pplx-res.cloudinary.com/image/private" + imgUrl[strings.Index(imgUrl, "/user_uploads"):]
+		c.Attachments = append(c.Attachments, imgUrl)
+	} else {
+		c.Attachments = append(c.Attachments, "https://ppl-ai-file-upload.s3.amazonaws.com/"+uploadInfo.Key)
+	}
+	return nil
+}
+
+// SetBigContext is a placeholder for setting context
+func (c *Client) UploadText(context string) error {
+	logger.Info("Uploading txt to Cloudinary")
+	filedata := base64.StdEncoding.EncodeToString([]byte(context))
+	filename := utils.RandomString(5) + ".txt"
+	// Upload images to Cloudinary
+	uploadURLResponse, err := c.createUploadURL(filename, "text/plain")
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error creating upload URL: %v", err))
+		return err
+	}
+	logger.Info(fmt.Sprintf("Upload URL response: %v", uploadURLResponse))
+	// Upload txt to Cloudinary
+	err = c.UloadFileToCloudinary(uploadURLResponse.Fields, "txt", filedata, filename)
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error uploading image: %v", err))
+		return err
+	}
+
+	return nil
+}
+
+func (c *Client) GetNewCookie() (string, error) {
+	resp, err := c.client.R().Get("https://www.perplexity.ai/api/auth/session")
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error getting session cookie: %v", err))
+		return "", err
+	}
+	if resp.StatusCode != http.StatusOK {
+		logger.Error(fmt.Sprintf("Error getting session cookie: %s", resp.String()))
+		return "", fmt.Errorf("unexpected status code: %d", resp.StatusCode)
+	}
+	for _, cookie := range resp.Cookies() {
+		if cookie.Name == "__Secure-next-auth.session-token" {
+			return cookie.Value, nil
+		}
+	}
+	return "", fmt.Errorf("session cookie not found")
+}

go.mod

@@ -0,0 +1,57 @@
+module pplx2api
+
+go 1.22.2
+
+require (
+	github.com/fatih/color v1.18.0
+	github.com/gin-gonic/gin v1.10.0
+	github.com/google/uuid v1.6.0
+	github.com/imroc/req/v3 v3.50.0
+	github.com/joho/godotenv v1.5.1
+)
+
+require (
+	github.com/andybalholm/brotli v1.1.1 // indirect
+	github.com/bytedance/sonic v1.11.6 // indirect
+	github.com/bytedance/sonic/loader v0.1.1 // indirect
+	github.com/cloudflare/circl v1.5.0 // indirect
+	github.com/cloudwego/base64x v0.1.4 // indirect
+	github.com/cloudwego/iasm v0.2.0 // indirect
+	github.com/gabriel-vasile/mimetype v1.4.3 // indirect
+	github.com/gin-contrib/sse v0.1.0 // indirect
+	github.com/go-playground/locales v0.14.1 // indirect
+	github.com/go-playground/universal-translator v0.18.1 // indirect
+	github.com/go-playground/validator/v10 v10.20.0 // indirect
+	github.com/go-task/slim-sprig/v3 v3.0.0 // indirect
+	github.com/goccy/go-json v0.10.2 // indirect
+	github.com/google/pprof v0.0.0-20241210010833-40e02aabc2ad // indirect
+	github.com/hashicorp/errwrap v1.1.0 // indirect
+	github.com/hashicorp/go-multierror v1.1.1 // indirect
+	github.com/json-iterator/go v1.1.12 // indirect
+	github.com/klauspost/compress v1.17.11 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.7 // indirect
+	github.com/leodido/go-urn v1.4.0 // indirect
+	github.com/mattn/go-colorable v0.1.13 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
+	github.com/modern-go/reflect2 v1.0.2 // indirect
+	github.com/onsi/ginkgo/v2 v2.22.0 // indirect
+	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
+	github.com/quic-go/qpack v0.5.1 // indirect
+	github.com/quic-go/quic-go v0.48.2 // indirect
+	github.com/refraction-networking/utls v1.6.7 // indirect
+	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
+	github.com/ugorji/go/codec v1.2.12 // indirect
+	go.uber.org/mock v0.5.0 // indirect
+	golang.org/x/arch v0.8.0 // indirect
+	golang.org/x/crypto v0.31.0 // indirect
+	golang.org/x/exp v0.0.0-20241215155358-4a5509556b9e // indirect
+	golang.org/x/mod v0.22.0 // indirect
+	golang.org/x/net v0.33.0 // indirect
+	golang.org/x/sync v0.10.0 // indirect
+	golang.org/x/sys v0.28.0 // indirect
+	golang.org/x/text v0.21.0 // indirect
+	golang.org/x/tools v0.28.0 // indirect
+	google.golang.org/protobuf v1.34.1 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)

go.sum

@@ -0,0 +1,140 @@
+github.com/andybalholm/brotli v1.1.1 h1:PR2pgnyFznKEugtsUo0xLdDop5SKXd5Qf5ysW+7XdTA=
+github.com/andybalholm/brotli v1.1.1/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
+github.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0=
+github.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4=
+github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM=
+github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
+github.com/cloudflare/circl v1.5.0 h1:hxIWksrX6XN5a1L2TI/h53AGPhNHoUBo+TD1ms9+pys=
+github.com/cloudflare/circl v1.5.0/go.mod h1:uddAzsPgqdMAYatqJ0lsjX1oECcQLIlRpzZh3pJrofs=
+github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
+github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
+github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
+github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
+github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
+github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0=
+github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk=
+github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
+github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
+github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
+github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
+github.com/go-logr/logr v1.4.2 h1:6pFjapn8bFcIbiKo3XT4j/BhANplGihG6tvd+8rYgrY=
+github.com/go-logr/logr v1.4.2/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
+github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
+github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
+github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
+github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
+github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
+github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
+github.com/go-playground/validator/v10 v10.20.0 h1:K9ISHbSaI0lyB2eWMPJo+kOS/FBExVwjEviJTixqxL8=
+github.com/go-playground/validator/v10 v10.20.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
+github.com/go-task/slim-sprig/v3 v3.0.0 h1:sUs3vkvUymDpBKi3qH1YSqBQk9+9D/8M2mN1vB6EwHI=
+github.com/go-task/slim-sprig/v3 v3.0.0/go.mod h1:W848ghGpv3Qj3dhTPRyJypKRiqCdHZiAzKg9hl15HA8=
+github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
+github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
+github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
+github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
+github.com/google/pprof v0.0.0-20241210010833-40e02aabc2ad h1:a6HEuzUHeKH6hwfN/ZoQgRgVIWFJljSWa/zetS2WTvg=
+github.com/google/pprof v0.0.0-20241210010833-40e02aabc2ad/go.mod h1:vavhavw2zAxS5dIdcRluK6cSGGPlZynqzFM8NdvU144=
+github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
+github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
+github.com/hashicorp/errwrap v1.1.0 h1:OxrOeh75EUXMY8TBjag2fzXGZ40LB6IKw45YeGUDY2I=
+github.com/hashicorp/errwrap v1.1.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
+github.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=
+github.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM=
+github.com/imroc/req/v3 v3.50.0 h1:n3BVnZiTRpvkN5T1IB79LC/THhFU9iXksNRMH4ZNVaY=
+github.com/imroc/req/v3 v3.50.0/go.mod h1:tsOk8K7zI6cU4xu/VWCZVtq9Djw9IWm4MslKzme5woU=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
+github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
+github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
+github.com/klauspost/compress v1.17.11 h1:In6xLpyWOi1+C7tXUUWv2ot1QvBjxevKAaI6IXrJmUc=
+github.com/klauspost/compress v1.17.11/go.mod h1:pMDklpSncoRMuLFrf1W9Ss9KT+0rH90U12bZKk7uwG0=
+github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
+github.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM=
+github.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=
+github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=
+github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
+github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
+github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
+github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
+github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
+github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
+github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
+github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=
+github.com/onsi/ginkgo/v2 v2.22.0 h1:Yed107/8DjTr0lKCNt7Dn8yQ6ybuDRQoMGrNFKzMfHg=
+github.com/onsi/ginkgo/v2 v2.22.0/go.mod h1:7Du3c42kxCUegi0IImZ1wUQzMBVecgIHjR1C+NkhLQo=
+github.com/onsi/gomega v1.34.2 h1:pNCwDkzrsv7MS9kpaQvVb1aVLahQXyJ/Tv5oAZMI3i8=
+github.com/onsi/gomega v1.34.2/go.mod h1:v1xfxRgk0KIsG+QOdm7p8UosrOzPYRo60fd3B/1Dukc=
+github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
+github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/quic-go/qpack v0.5.1 h1:giqksBPnT/HDtZ6VhtFKgoLOWmlyo9Ei6u9PqzIMbhI=
+github.com/quic-go/qpack v0.5.1/go.mod h1:+PC4XFrEskIVkcLzpEkbLqq1uCoxPhQuvK5rH1ZgaEg=
+github.com/quic-go/quic-go v0.48.2 h1:wsKXZPeGWpMpCGSWqOcqpW2wZYic/8T3aqiOID0/KWE=
+github.com/quic-go/quic-go v0.48.2/go.mod h1:yBgs3rWBOADpga7F+jJsb6Ybg1LSYiQvwWlLX+/6HMs=
+github.com/refraction-networking/utls v1.6.7 h1:zVJ7sP1dJx/WtVuITug3qYUq034cDq9B2MR1K67ULZM=
+github.com/refraction-networking/utls v1.6.7/go.mod h1:BC3O4vQzye5hqpmDTWUqi4P5DDhzJfkV1tdqtawQIH0=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
+github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
+github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
+github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
+github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
+github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
+github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=
+github.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=
+github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
+github.com/xyproto/randomstring v1.0.5 h1:YtlWPoRdgMu3NZtP45drfy1GKoojuR7hmRcnhZqKjWU=
+github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
+go.uber.org/mock v0.5.0 h1:KAMbZvZPyBPWgD14IrIQ38QCyjwpvVVV6K/bHl1IwQU=
+go.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=
+golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
+golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc=
+golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
+golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
+golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
+golang.org/x/exp v0.0.0-20241215155358-4a5509556b9e h1:4qufH0hlUYs6AO6XmZC3GqfDPGSXHVXUFR6OND+iJX4=
+golang.org/x/exp v0.0.0-20241215155358-4a5509556b9e/go.mod h1:qj5a5QZpwLU2NLQudwIN5koi3beDhSAlJwa67PuM98c=
+golang.org/x/mod v0.22.0 h1:D4nJWe9zXqHOmWqj4VMOJhvzj7bEZg4wEYa759z1pH4=
+golang.org/x/mod v0.22.0/go.mod h1:6SkKJ3Xj0I0BrPOZoBy3bdMptDDU9oJrpohJ3eWZ1fY=
+golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I=
+golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4=
+golang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ=
+golang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
+golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
+golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
+golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
+golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=
+golang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
+golang.org/x/tools v0.28.0 h1:WuB6qZ4RPCQo5aP3WdKZS7i595EdWqWR8vqJTlwTVK8=
+golang.org/x/tools v0.28.0/go.mod h1:dcIOrVd3mfQKTgrDVQHqCPMWy6lnhfhtX3hLXYVLfRw=
+google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg=
+google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=
+rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=

job/cookie.go

@@ -0,0 +1,211 @@
+package job
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"log"
+	"os"
+	"sync"
+	"time"
+
+	"pplx2api/config"
+	"pplx2api/core"
+)
+
+const (
+	// ConfigFileName is the name of the file to store sessions
+	ConfigFileName = "sessions.json"
+)
+
+var (
+	sessionUpdaterInstance *SessionUpdater
+	sessionUpdaterOnce     sync.Once
+)
+
+// SessionConfig represents the structure to be saved to file
+type SessionConfig struct {
+	Sessions []config.SessionInfo `json:"sessions"`
+}
+
+// SessionUpdater 管理 Perplexity 会话的定时更新
+type SessionUpdater struct {
+	interval    time.Duration
+	stopChan    chan struct{}
+	isRunning   bool
+	runningLock sync.Mutex
+	configPath  string
+}
+
+// NewSessionUpdater 创建一个新的会话更新器
+// interval: 更新间隔时间
+func GetSessionUpdater(interval time.Duration) *SessionUpdater {
+	sessionUpdaterOnce.Do(func() {
+		// 使用当前文件夹下的配置文件
+		configPath := ConfigFileName
+
+		sessionUpdaterInstance = &SessionUpdater{
+			interval:   interval,
+			stopChan:   make(chan struct{}),
+			isRunning:  false,
+			configPath: configPath,
+		}
+		// 初始化时从文件加载会话
+		sessionUpdaterInstance.loadSessionsFromFile()
+	})
+	return sessionUpdaterInstance
+}
+
+// loadSessionsFromFile loads sessions from the config file if it exists
+func (su *SessionUpdater) loadSessionsFromFile() {
+	// Check if file exists
+	if _, err := os.Stat(su.configPath); os.IsNotExist(err) {
+		log.Println("No sessions config file found, will create on first update")
+		return
+	}
+
+	// Read the file
+	data, err := ioutil.ReadFile(su.configPath)
+	if err != nil {
+		log.Printf("Failed to read sessions config file: %v", err)
+		return
+	}
+
+	// Parse the JSON
+	var sessionConfig SessionConfig
+	if err := json.Unmarshal(data, &sessionConfig); err != nil {
+		log.Printf("Failed to parse sessions config file: %v", err)
+		return
+	}
+
+	// Update the config with loaded sessions
+	config.ConfigInstance.RwMutex.Lock()
+	config.ConfigInstance.Sessions = sessionConfig.Sessions
+	config.ConfigInstance.RwMutex.Unlock()
+
+	log.Printf("Loaded %d sessions from config file", len(sessionConfig.Sessions))
+}
+
+// saveSessionsToFile saves the current sessions to the config file
+func (su *SessionUpdater) saveSessionsToFile() error {
+	// Get current sessions
+	config.ConfigInstance.RwMutex.RLock()
+	sessionsCopy := make([]config.SessionInfo, len(config.ConfigInstance.Sessions))
+	copy(sessionsCopy, config.ConfigInstance.Sessions)
+	config.ConfigInstance.RwMutex.RUnlock()
+
+	// Create config structure
+	sessionConfig := SessionConfig{
+		Sessions: sessionsCopy,
+	}
+
+	// Convert to JSON
+	data, err := json.MarshalIndent(sessionConfig, "", "  ")
+	if err != nil {
+		return err
+	}
+
+	// Write to file
+	err = ioutil.WriteFile(su.configPath, data, 0644)
+	if err != nil {
+		return err
+	}
+
+	log.Printf("Saved %d sessions to sessions.json file", len(sessionsCopy))
+	return nil
+}
+
+// Start 启动定时更新任务
+func (su *SessionUpdater) Start() {
+	su.runningLock.Lock()
+	defer su.runningLock.Unlock()
+	if su.isRunning {
+		log.Println("Session updater is already running")
+		return
+	}
+	su.isRunning = true
+	su.stopChan = make(chan struct{})
+	go su.runUpdateLoop()
+	log.Println("Session updater started with interval:", su.interval)
+}
+
+// Stop 停止定时更新任务
+func (su *SessionUpdater) Stop() {
+	su.runningLock.Lock()
+	defer su.runningLock.Unlock()
+	if !su.isRunning {
+		log.Println("Session updater is not running")
+		return
+	}
+	close(su.stopChan)
+	su.isRunning = false
+	log.Println("Session updater stopped")
+}
+
+// runUpdateLoop 运行更新循环
+func (su *SessionUpdater) runUpdateLoop() {
+	ticker := time.NewTicker(su.interval)
+	defer ticker.Stop()
+	// 立即执行一次更新
+	// su.updateAllSessions()
+	for {
+		select {
+		case <-ticker.C:
+			su.updateAllSessions()
+		case <-su.stopChan:
+			log.Println("Update loop terminated")
+			return
+		}
+	}
+}
+
+// updateAllSessions 更新所有会话
+func (su *SessionUpdater) updateAllSessions() {
+	log.Println("Starting session update for all sessions...")
+	// 复制当前会话列表，避免长时间持有锁
+	config.ConfigInstance.RwMutex.RLock()
+	sessionsCopy := make([]config.SessionInfo, len(config.ConfigInstance.Sessions))
+	copy(sessionsCopy, config.ConfigInstance.Sessions)
+	proxy := config.ConfigInstance.Proxy
+	config.ConfigInstance.RwMutex.RUnlock()
+	// 如果没有会话需要更新，直接返回
+	if len(sessionsCopy) == 0 {
+		log.Println("No sessions to update")
+		return
+	}
+	// 创建更新后的会话切片
+	updatedSessions := make([]config.SessionInfo, len(sessionsCopy))
+	var wg sync.WaitGroup
+	// 对每个会话执行更新
+	for i, session := range sessionsCopy {
+		wg.Add(1)
+		go func(index int, origSession config.SessionInfo) {
+			defer wg.Done()
+			// 创建客户端��更新 cookie
+			// 写死 model 和 openSearch 参数
+			client := core.NewClient(origSession.SessionKey, proxy, "claude-3-opus-20240229", false)
+			newCookie, err := client.GetNewCookie()
+			if err != nil {
+				log.Printf("Failed to update session %d: %v", index, err)
+				// 如果更新失败，保留原始会话
+				updatedSessions[index] = origSession
+				return
+			}
+			// 创建更新后的会话对象
+			updatedSessions[index] = config.SessionInfo{
+				SessionKey: newCookie,
+			}
+		}(i, session)
+	}
+	// 等待所有更新完成
+	wg.Wait()
+	// 一次性替换所有会话
+	config.ConfigInstance.RwMutex.Lock()
+	config.ConfigInstance.Sessions = updatedSessions
+	config.ConfigInstance.RwMutex.Unlock()
+	log.Printf("All %d sessions have been updated", len(updatedSessions))
+
+	// 保存更新后的配置到文件
+	if err := su.saveSessionsToFile(); err != nil {
+		log.Printf("Failed to save updated config: %v", err)
+	}
+}

logger/logger.go

@@ -0,0 +1,104 @@
+package logger
+
+import (
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/fatih/color"
+)
+
+// 日志级别
+const (
+	DEBUG = iota
+	INFO
+	WARN
+	ERROR
+	FATAL
+)
+
+var levelNames = map[int]string{
+	DEBUG: "DEBUG",
+	INFO:  "INFO",
+	WARN:  "WARN",
+	ERROR: "ERROR",
+	FATAL: "FATAL",
+}
+
+var levelColors = map[int]func(format string, a ...interface{}) string{
+	DEBUG: color.BlueString,
+	INFO:  color.GreenString,
+	WARN:  color.YellowString,
+	ERROR: color.RedString,
+	FATAL: color.New(color.FgHiRed, color.Bold).SprintfFunc(),
+}
+
+// 全局日志级别，默认为INFO
+var logLevel = INFO
+
+// SetLevel 设置日志级别
+func SetLevel(level int) {
+	if level >= DEBUG && level <= FATAL {
+		logLevel = level
+	}
+}
+
+// GetLevel 获取当前日志级别
+func GetLevel() int {
+	return logLevel
+}
+
+// GetLevelName 获取日志级别名称
+func GetLevelName(level int) string {
+	if name, ok := levelNames[level]; ok {
+		return name
+	}
+	return "UNKNOWN"
+}
+
+// 基础日志打印函数
+func log(level int, format string, args ...interface{}) {
+	if level < logLevel {
+		return
+	}
+
+	now := time.Now().Format("2006-01-02 15:04:05.000")
+	levelName := levelNames[level]
+	colorFunc := levelColors[level]
+
+	logContent := fmt.Sprintf(format, args...)
+	logPrefix := fmt.Sprintf("[%s] [%s] ", now, levelName)
+
+	// 使用颜色输出日志级别
+	fmt.Fprintf(os.Stdout, "%s%s\n", logPrefix, colorFunc(logContent))
+
+	// 如果是致命错误，则退出程序
+	if level == FATAL {
+		os.Exit(1)
+	}
+}
+
+// Debug 打印调试日志
+func Debug(format string, args ...interface{}) {
+	log(DEBUG, format, args...)
+}
+
+// Info 打印信息日志
+func Info(format string, args ...interface{}) {
+	log(INFO, format, args...)
+}
+
+// Warn 打印警告日志
+func Warn(format string, args ...interface{}) {
+	log(WARN, format, args...)
+}
+
+// Error 打印错误日志
+func Error(format string, args ...interface{}) {
+	log(ERROR, format, args...)
+}
+
+// Fatal 打印致命错误日志并退出程序
+func Fatal(format string, args ...interface{}) {
+	log(FATAL, format, args...)
+}

main.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"pplx2api/config"
+	"pplx2api/job"
+	"pplx2api/router"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+func main() {
+	r := gin.Default()
+	// Load configuration
+
+	// Setup all routes
+	router.SetupRoutes(r)
+	// 创建会话更新器，设置更新间隔为24小时
+	sessionUpdater := job.GetSessionUpdater(24 * time.Hour)
+
+	// 启动会话更新器
+	sessionUpdater.Start()
+	defer sessionUpdater.Stop()
+
+	// Run the server on 0.0.0.0:8080
+	r.Run(config.ConfigInstance.Address)
+}

middleware/auth.go

@@ -0,0 +1,31 @@
+package middleware
+
+import (
+	"pplx2api/config"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+)
+
+// AuthMiddleware initializes the Claude client from the request header
+func AuthMiddleware() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		Key := c.GetHeader("Authorization")
+		if Key != "" {
+			Key = strings.TrimPrefix(Key, "Bearer ")
+			if Key != config.ConfigInstance.APIKey {
+				c.JSON(401, gin.H{
+					"error": "Invalid API key",
+				})
+				c.Abort()
+				return
+			}
+			c.Next()
+			return
+		}
+		c.JSON(401, gin.H{
+			"error": "Missing or invalid Authorization header",
+		})
+		c.Abort()
+	}
+}

middleware/cors.go

@@ -0,0 +1,17 @@
+package middleware
+
+import "github.com/gin-gonic/gin"
+
+// CORSMiddleware handles CORS headers
+func CORSMiddleware() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
+		c.Writer.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
+		c.Writer.Header().Set("Access-Control-Allow-Headers", "Content-Type, Content-Length, Accept-Encoding, Authorization")
+		if c.Request.Method == "OPTIONS" {
+			c.AbortWithStatus(204)
+			return
+		}
+		c.Next()
+	}
+}

model/openai.go

@@ -0,0 +1,130 @@
+package model
+
+import (
+	"encoding/json"
+	"fmt"
+	"pplx2api/logger"
+	"time"
+
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+)
+
+type ChatCompletionRequest struct {
+	Model    string                   `json:"model"`
+	Messages []map[string]interface{} `json:"messages"`
+	Stream   bool                     `json:"stream"`
+	Tools    []map[string]interface{} `json:"tools,omitempty"`
+}
+
+// OpenAISrteamResponse 定义 OpenAI 的流式响应结构
+type OpenAISrteamResponse struct {
+	ID      string         `json:"id"`
+	Object  string         `json:"object"`
+	Created int64          `json:"created"`
+	Model   string         `json:"model"`
+	Choices []StreamChoice `json:"choices"`
+}
+
+// Choice 结构表示 OpenAI 返回的单个选项
+type StreamChoice struct {
+	Index        int         `json:"index"`
+	Delta        Delta       `json:"delta"`
+	Logprobs     interface{} `json:"logprobs"`
+	FinishReason interface{} `json:"finish_reason"`
+}
+
+type NoStreamChoice struct {
+	Index        int         `json:"index"`
+	Message      Message     `json:"message"`
+	Logprobs     interface{} `json:"logprobs"`
+	FinishReason string      `json:"finish_reason"`
+}
+
+// Delta 结构用于存储返回的文本内容
+type Delta struct {
+	Content string `json:"content"`
+}
+type Message struct {
+	Role       string        `json:"role"`
+	Content    string        `json:"content"`
+	Refusal    interface{}   `json:"refusal"`
+	Annotation []interface{} `json:"annotation"`
+}
+
+type OpenAIResponse struct {
+	ID      string           `json:"id"`
+	Object  string           `json:"object"`
+	Created int64            `json:"created"`
+	Model   string           `json:"model"`
+	Choices []NoStreamChoice `json:"choices"`
+	Usage   Usage            `json:"usage"`
+}
+type Usage struct {
+	PromptTokens     int `json:"prompt_tokens"`
+	CompletionTokens int `json:"completion_tokens"`
+	TotalTokens      int `json:"total_tokens"`
+}
+
+func ReturnOpenAIResponse(text string, stream bool, gc *gin.Context) error {
+	if stream {
+		return streamRespose(text, gc)
+	} else {
+		return noStreamResponse(text, gc)
+	}
+}
+
+func streamRespose(text string, gc *gin.Context) error {
+	openAIResp := &OpenAISrteamResponse{
+		ID:      uuid.New().String(),
+		Object:  "chat.completion.chunk",
+		Created: time.Now().Unix(),
+		Model:   "claude-3-7-sonnet-20250219",
+		Choices: []StreamChoice{
+			{
+				Index: 0,
+				Delta: Delta{
+					Content: text,
+				},
+				Logprobs:     nil,
+				FinishReason: nil,
+			},
+		},
+	}
+
+	jsonBytes, err := json.Marshal(openAIResp)
+	jsonBytes = append([]byte("data: "), jsonBytes...)
+	jsonBytes = append(jsonBytes, []byte("\n\n")...)
+	if err != nil {
+		logger.Error(fmt.Sprintf("Error marshalling JSON: %v", err))
+		return err
+	}
+
+	// 发送数据
+	gc.Writer.Write(jsonBytes)
+	gc.Writer.Flush()
+	return nil
+}
+
+func noStreamResponse(text string, gc *gin.Context) error {
+	openAIResp := &OpenAIResponse{
+		ID:      uuid.New().String(),
+		Object:  "chat.completion",
+		Created: time.Now().Unix(),
+		Model:   "claude-3-7-sonnet-20250219",
+		Choices: []NoStreamChoice{
+			{
+				Index: 0,
+				Message: Message{
+					Role:    "assistant",
+					Content: text,
+				},
+				Logprobs:     nil,
+				FinishReason: "stop",
+			},
+		},
+	}
+
+	gc.JSON(200, openAIResp)
+	return nil
+}

router/router.go

@@ -0,0 +1,30 @@
+package router
+
+import (
+	"pplx2api/middleware"
+	"pplx2api/service"
+
+	"github.com/gin-gonic/gin"
+)
+
+func SetupRoutes(r *gin.Engine) {
+	// Apply middleware
+	r.Use(middleware.CORSMiddleware())
+	r.Use(middleware.AuthMiddleware())
+
+	// Health check endpoint
+	r.GET("/health", service.HealthCheckHandler)
+
+	// Chat completions endpoint (OpenAI-compatible)
+	r.POST("/v1/chat/completions", service.ChatCompletionsHandler)
+	r.GET("/v1/models", service.MoudlesHandler)
+	// HuggingFace compatible routes
+	hfRouter := r.Group("/hf")
+	{
+		v1Router := hfRouter.Group("/v1")
+		{
+			v1Router.POST("/chat/completions", service.ChatCompletionsHandler)
+			v1Router.GET("/models", service.MoudlesHandler)
+		}
+	}
+}

service/handle.go

@@ -0,0 +1,170 @@
+package service
+
+import (
+	"fmt"
+	"net/http"
+	"pplx2api/config"
+	"pplx2api/core"
+	"pplx2api/logger"
+	"pplx2api/utils"
+	"strings"
+
+	"github.com/gin-gonic/gin"
+)
+
+type ChatCompletionRequest struct {
+	Model    string                   `json:"model"`
+	Messages []map[string]interface{} `json:"messages"`
+	Stream   bool                     `json:"stream"`
+	Tools    []map[string]interface{} `json:"tools,omitempty"`
+}
+
+type ErrorResponse struct {
+	Error string `json:"error"`
+}
+
+// HealthCheckHandler handles the health check endpoint
+func HealthCheckHandler(c *gin.Context) {
+	c.JSON(http.StatusOK, gin.H{
+		"status": "ok",
+	})
+}
+
+// ChatCompletionsHandler handles the chat completions endpoint
+func ChatCompletionsHandler(c *gin.Context) {
+
+	// Parse request body
+	var req ChatCompletionRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, ErrorResponse{
+			Error: fmt.Sprintf("Invalid request: %v", err),
+		})
+		return
+	}
+	// logger.Info(fmt.Sprintf("Received request: %v", req))
+	// Validate request
+	if len(req.Messages) == 0 {
+		c.JSON(http.StatusBadRequest, ErrorResponse{
+			Error: "No messages provided",
+		})
+		return
+	}
+
+	// Get model or use default
+	model := req.Model
+	if model == "" {
+		model = "claude-3.7-sonnet"
+	}
+	openSearch := false
+	if strings.HasSuffix(model, "-search") {
+		openSearch = true
+		model = strings.TrimSuffix(model, "-search")
+	}
+	model = config.ModelMapGet(model, model) // 获取模型名称
+	var prompt strings.Builder
+	img_data_list := []string{}
+	// Format messages into a single prompt
+	for _, msg := range req.Messages {
+		role, roleOk := msg["role"].(string)
+		if !roleOk {
+			continue // 忽略无效格式
+		}
+
+		content, exists := msg["content"]
+		if !exists {
+			continue
+		}
+
+		prompt.WriteString(utils.GetRolePrefix(role)) // 获取角色前缀
+		switch v := content.(type) {
+		case string: // 如果 content 直接是 string
+			prompt.WriteString(v + "\n\n")
+		case []interface{}: // 如果 content 是 []interface{} 类型的数组
+			for _, item := range v {
+				if itemMap, ok := item.(map[string]interface{}); ok {
+					if itemType, ok := itemMap["type"].(string); ok {
+						if itemType == "text" {
+							if text, ok := itemMap["text"].(string); ok {
+								prompt.WriteString(text + "\n\n")
+							}
+						} else if itemType == "image_url" {
+							if imageUrl, ok := itemMap["image_url"].(map[string]interface{}); ok {
+								if url, ok := imageUrl["url"].(string); ok {
+									if len(url) > 50 {
+										logger.Info(fmt.Sprintf("Image URL: %s ……", url[:50]))
+									}
+									if strings.HasPrefix(url, "data:image/") {
+										// 保留 base64 编码的图片数据
+										url = strings.Split(url, ",")[1]
+									}
+									img_data_list = append(img_data_list, url) // 收集图片数据
+								}
+							}
+						}
+					}
+				}
+			}
+		}
+	}
+	fmt.Println(prompt.String())                             // 输出最终构造的内容
+	fmt.Println("img_data_list_length:", len(img_data_list)) // 输出图片数据列表长度
+	var rootPrompt strings.Builder
+	rootPrompt.WriteString(prompt.String())
+	// 切号重试机制
+	var pplxClient *core.Client
+	index := config.Sr.NextIndex()
+	for i := 0; i < config.ConfigInstance.RetryCount; i++ {
+		if i > 0 {
+			prompt.Reset()
+			prompt.WriteString(rootPrompt.String())
+		}
+		index = (index + 1) % len(config.ConfigInstance.Sessions)
+		session, err := config.ConfigInstance.GetSessionForModel(index)
+		logger.Info(fmt.Sprintf("Using session for model %s: %s", model, session.SessionKey))
+		if err != nil {
+			logger.Error(fmt.Sprintf("Failed to get session for model %s: %v", model, err))
+			logger.Info("Retrying another session")
+			continue
+		}
+		// Initialize the Claude client
+		pplxClient = core.NewClient(session.SessionKey, config.ConfigInstance.Proxy, model, openSearch)
+		if len(img_data_list) > 0 {
+			err := pplxClient.UploadImage(img_data_list)
+			if err != nil {
+				logger.Error(fmt.Sprintf("Failed to upload file: %v", err))
+				logger.Info("Retrying another session")
+
+				continue
+			}
+		}
+		if prompt.Len() > config.ConfigInstance.MaxChatHistoryLength {
+			err := pplxClient.UploadText(prompt.String())
+			if err != nil {
+				logger.Error(fmt.Sprintf("Failed to upload text: %v", err))
+				logger.Info("Retrying another session")
+
+				continue
+			}
+			prompt.Reset()
+			prompt.WriteString(config.ConfigInstance.PromptForFile)
+		}
+		if _, err := pplxClient.SendMessage(prompt.String(), req.Stream, config.ConfigInstance.IsIncognito, c); err != nil {
+			logger.Error(fmt.Sprintf("Failed to send message: %v", err))
+			logger.Info("Retrying another session")
+
+			continue // Retry on error
+		}
+
+		return
+
+	}
+	logger.Error("Failed for all retries")
+	c.JSON(http.StatusInternalServerError, ErrorResponse{
+		Error: "Failed to process request after multiple attempts"})
+}
+
+func MoudlesHandler(c *gin.Context) {
+	c.JSON(http.StatusOK, gin.H{
+		"data": config.ResponseModles,
+	})
+}

utils/imageShow.go

@@ -0,0 +1,8 @@
+package utils
+
+import "fmt"
+
+func ImageShow(index int, modelName, url string) string {
+	index++
+	return fmt.Sprintf("![%s](%s)", modelName, url)
+}

utils/random.go

@@ -0,0 +1,12 @@
+package utils
+
+import "math/rand"
+
+func RandomString(length int) string {
+	const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+	result := make([]byte, length)
+	for i := range result {
+		result[i] = charset[rand.Intn(len(charset))]
+	}
+	return string(result)
+}

utils/role.go

@@ -0,0 +1,22 @@
+package utils
+
+import (
+	"pplx2api/config"
+)
+
+// **获取角色前缀**
+func GetRolePrefix(role string) string {
+	if config.ConfigInstance.NoRolePrefix {
+		return ""
+	}
+	switch role {
+	case "system":
+		return "System: "
+	case "user":
+		return "Human: "
+	case "assistant":
+		return "Assistant: "
+	default:
+		return "Unknown: "
+	}
+}

utils/searchShow.go

@@ -0,0 +1,26 @@
+package utils
+
+import (
+	"fmt"
+	"pplx2api/config"
+)
+
+func searchShowDetails(index int, title, url, snippet string) string {
+	return fmt.Sprintf("<details>\n<summary>[%d] %s</summary>\n\n%s\n\n[Link](%s)\n\n</details>", index, title, snippet, url)
+}
+
+func searchShowCompatible(index int, title, url, snippet string) string {
+	return fmt.Sprintf("[%d] [%s](%s):\n%s\n", index, title, url, snippet)
+}
+
+func SearchShow(index int, title, url, snippet string) string {
+	index++
+	if len([]rune(snippet)) > 150 {
+		runeSnippet := []rune(snippet)
+		snippet = fmt.Sprintf("%s ……", string(runeSnippet[:150]))
+	}
+	if config.ConfigInstance.SearchResultCompatible {
+		return searchShowCompatible(index, title, url, snippet)
+	}
+	return searchShowDetails(index, title, url, snippet)
+}