Feat/doubao batch

.github/workflows/docker-image-amd64.yml

@@ -1,54 +1,40 @@
-name: Publish Docker image (amd64)
+name: Build NewAPI Docker Image
 
 on:
   push:
     tags:
-      - '*'
+      - "main"
   workflow_dispatch:
     inputs:
-      name:
-        description: 'reason'
-        required: false
+      git_branch:
+        description: 'Git branch'
+        required: true
+        default: 'main'
+      remote_repo:
+        description: 'Remote repository'
+        required: true
+        default: 'https://github.com/Furion-cn/new-api.git'
+
 jobs:
-  push_to_registries:
-    name: Push Docker image to multiple registries
-    runs-on: ubuntu-latest
-    permissions:
-      packages: write
-      contents: read
+  build-dev:
+    runs-on: ubuntu-22.04
     steps:
-      - name: Check out the repo
-        uses: actions/checkout@v3
-
-      - name: Save version info
-        run: |
-          git describe --tags > VERSION 
-
-      - name: Log in to Docker Hub
-        uses: docker/login-action@v2
-        with:
-          username: ${{ secrets.DOCKERHUB_USERNAME }}
-          password: ${{ secrets.DOCKERHUB_TOKEN }}
+      - name: Checkout repository
+        uses: actions/checkout@v4
 
-      - name: Log in to the Container registry
+      - name: Login to Tencent Cloud Registry
         uses: docker/login-action@v2
         with:
-          registry: ghcr.io
-          username: ${{ github.actor }}
-          password: ${{ secrets.GITHUB_TOKEN }}
-
-      - name: Extract metadata (tags, labels) for Docker
-        id: meta
-        uses: docker/metadata-action@v4
-        with:
-          images: |
-            calciumion/new-api
-            ghcr.io/${{ github.repository }}
+          registry: furion-sh.tencentcloudcr.com
+          username: ${{ secrets.TCR_USERNAME }}
+          password: ${{ secrets.TCR_PASSWORD }}
 
-      - name: Build and push Docker images
-        uses: docker/build-push-action@v3
-        with:
-          context: .
-          push: true
-          tags: ${{ steps.meta.outputs.tags }}
-          labels: ${{ steps.meta.outputs.labels }}
+      - name: Build and Push SGLang Image
+        run: |
+          DATETIME=$(date '+%Y%m%d_%H%M')
+          ORG=$(echo "${{ inputs.remote_repo }}" | awk -F'/' '{print $(NF-1)}' | tr '[:upper:]' '[:lower:]' | sed 's/[^a-z0-9]/-/g')
+          BRANCH=$(echo "${{ inputs.git_branch }}" | tr '[:upper:]' '[:lower:]' | sed 's/[^a-z0-9]/-/g')
+          VERSION="${DATETIME}_${ORG}_${BRANCH}"
+          git clone ${{ inputs.remote_repo }} && cd new-api
+          docker build . -t furion-sh.tencentcloudcr.com/furion/new-api:${VERSION}
+          docker push furion-sh.tencentcloudcr.com/furion/new-api:${VERSION}

.gitignore

@@ -9,4 +9,13 @@ logs
 web/dist
 .env
 one-api
-.DS_Store
+.DS_Store
+new-api
+script/
+.env_2
+*_test.go
+out.log
+out.log_2
+.env_3
+*.prof
+.env*

Dockerfile

@@ -1,4 +1,4 @@
-FROM oven/bun:latest AS builder
+FROM swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/oven/bun:latest AS builder
 
 WORKDIR /build
 COPY web/package.json .
@@ -7,29 +7,33 @@ COPY ./web .
 COPY ./VERSION .
 RUN DISABLE_ESLINT_PLUGIN='true' VITE_REACT_APP_VERSION=$(cat VERSION) bun run build
 
-FROM golang:alpine AS builder2
+FROM swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/golang:alpine3.21 AS builder2
 
 ENV GO111MODULE=on \
     CGO_ENABLED=0 \
-    GOOS=linux
+    GOOS=linux \
+    GOPROXY=https://goproxy.cn
 
 WORKDIR /build
 
 ADD go.mod go.sum ./
+RUN go mod tidy
 RUN go mod download
 
 COPY . .
 COPY --from=builder /build/dist ./web/dist
+RUN go mod tidy
 RUN go build -ldflags "-s -w -X 'one-api/common.Version=$(cat VERSION)'" -o one-api
 
-FROM alpine
+FROM swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/library/alpine:latest
 
-RUN apk update \
+RUN sed -i 's/dl-cdn.alpinelinux.org/mirrors.aliyun.com/g' /etc/apk/repositories \
+    && apk update \
     && apk upgrade \
     && apk add --no-cache ca-certificates tzdata ffmpeg \
     && update-ca-certificates
 
 COPY --from=builder2 /build/one-api /
 EXPOSE 3000
 WORKDIR /data
-ENTRYPOINT ["/one-api"]
+ENTRYPOINT ["/one-api"]

bin/groups.sql

@@ -0,0 +1,7 @@
+CREATE TABLE IF NOT EXISTS `groups` (
+  `id` int(11) NOT NULL AUTO_INCREMENT,
+  `name` varchar(255) NOT NULL DEFAULT '',
+  `ratio` int(11) NOT NULL DEFAULT 0,
+  PRIMARY KEY (`id`),
+  UNIQUE KEY `idx_name` (`name`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;

bin/log_requestid.sql

@@ -0,0 +1 @@
+ALTER TABLE logs ADD COLUMN request_id VARCHAR(255) DEFAULT '' COMMENT '请求ID';

common/constants.go

@@ -26,12 +26,16 @@ var DataExportEnabled = true
 var DataExportInterval = 5         // unit: minute
 var DataExportDefaultTime = "hour" // unit: minute
 var DefaultCollapseSidebar = false // default value of collapse sidebar
+var MockResponseEnabled = false    // whether to enable mock response for test traffic
 
 // Any options with "Secret", "Token" in its key won't be return by GetOptions
 
 var SessionSecret = uuid.New().String()
 var CryptoSecret = uuid.New().String()
 
+// 需要先导入 model 包
+var Groups = map[string]int{}
+var GroupRWMutex sync.RWMutex
 var OptionMap map[string]string
 var OptionMapRWMutex sync.RWMutex
 
@@ -234,6 +238,8 @@ const (
 	ChannelTypeMokaAI         = 44
 	ChannelTypeVolcEngine     = 45
 	ChannelTypeBaiduV2        = 46
+	ChannelTypeXai            = 47
+	ChannelTypeDoubaoOffline  = 100
 	ChannelTypeDummy          // this one is only for count, do not add any channel after this
 
 )
@@ -286,4 +292,58 @@ var ChannelBaseURLs = []string{
 	"https://api.moka.ai",                       //44
 	"https://ark.cn-beijing.volces.com",         //45
 	"https://qianfan.baidubce.com",              //46
+	"",                                          //47
+	"https://api.x.ai",                          //48
+	"",                                          //49
+	"",                                          //50
+	"",                                          //51
+	"",                                          //52
+	"",                                          //53
+	"",                                          //54
+	"",                                          //55
+	"",                                          //56
+	"",                                          //57
+	"",                                          //58
+	"",                                          //59
+	"",                                          //60
+	"",                                          //61
+	"",                                          //62
+	"",                                          //63
+	"",                                          //64
+	"",                                          //65
+	"",                                          //66
+	"",                                          //67
+	"",                                          //68
+	"",                                          //69
+	"",                                          //70
+	"",                                          //71
+	"",                                          //72
+	"",                                          //73
+	"",                                          //74
+	"",                                          //75
+	"",                                          //76
+	"",                                          //77
+	"",                                          //78
+	"",                                          //79
+	"",                                          //80
+	"",                                          //81
+	"",                                          //82
+	"",                                          //83
+	"",                                          //84
+	"",                                          //85
+	"",                                          //86
+	"",                                          //87
+	"",                                          //88
+	"",                                          //89
+	"",                                          //90
+	"",                                          //91
+	"",                                          //92
+	"",                                          //93
+	"",                                          //94
+	"",                                          //95
+	"",                                          //96
+	"",                                          //97
+	"",                                          //98
+	"",                                          //99
+	"https://ark.cn-beijing.volces.com",         //100 - 豆包离线
 }

common/init.go

@@ -66,4 +66,8 @@ func LoadEnv() {
 			}
 		}
 	}
+
+	// Initialize mock response feature
+	MockResponseEnabled = os.Getenv("MOCK_RESPONSE_ENABLED") == "true"
+	SysLog(fmt.Sprintf("MockResponseEnabled: %v", MockResponseEnabled))
 }