test: 测试GitHub构建docker

.github/workflows/docker-build.yml

@@ -0,0 +1,85 @@
+name: Build and Push Multi-Arch Docker Image
+
+on:
+  push:
+    branches:
+      - main
+      - master
+    tags:
+      - 'v*'
+  pull_request:
+    branches:
+      - main
+      - master
+  workflow_dispatch:  # 允许手动触发
+
+env:
+  DOCKER_IMAGE: mumujie/mumuainovel
+
+jobs:
+  build-and-push:
+    runs-on: ubuntu-latest
+    
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+        with:
+          platforms: linux/amd64,linux/arm64
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+        with:
+          driver-opts: |
+            image=moby/buildkit:master
+            network=host
+
+      - name: Login to Docker Hub
+        if: github.event_name != 'pull_request'
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+
+      - name: Extract metadata (tags, labels)
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: ${{ env.DOCKER_IMAGE }}
+          tags: |
+            # 对于main/master分支，打latest标签
+            type=raw,value=latest,enable=${{ github.ref == 'refs/heads/main' || github.ref == 'refs/heads/master' }}
+            # 对于tag，使用版本号
+            type=semver,pattern={{version}}
+            type=semver,pattern={{major}}.{{minor}}
+            type=semver,pattern={{major}}
+            # 使用分支名
+            type=ref,event=branch
+            # 使用commit SHA
+            type=sha,prefix=sha-
+
+      - name: Build and push Docker image
+        uses: docker/build-push-action@v5
+        with:
+          context: .
+          file: ./Dockerfile
+          platforms: linux/amd64,linux/arm64
+          push: ${{ github.event_name != 'pull_request' }}
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}
+          cache-from: type=gha
+          cache-to: type=gha,mode=max
+          build-args: |
+            BUILDKIT_INLINE_CACHE=1
+
+      - name: Update Docker Hub Description
+        if: github.event_name != 'pull_request' && (github.ref == 'refs/heads/main' || github.ref == 'refs/heads/master')
+        uses: peter-evans/dockerhub-description@v4
+        with:
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+          repository: ${{ env.DOCKER_IMAGE }}
+          readme-filepath: ./README.md
+        continue-on-error: true

Dockerfile

@@ -1,16 +1,25 @@
 # 多阶段构建 Dockerfile for AI Story Creator
+# 支持多架构构建: linux/amd64, linux/arm64
+
+# 构建参数
+ARG USE_CN_MIRROR=false
+
 # 阶段1: 构建前端
 FROM node:22-alpine AS frontend-builder
 
+ARG USE_CN_MIRROR
+
 WORKDIR /frontend
 
 # 复制前端依赖文件
 COPY frontend/package*.json ./
 
-# 使用国内npm镜像加速
-RUN npm config set registry https://registry.npmmirror.com
+# 根据参数决定是否使用国内npm镜像
+RUN if [ "$USE_CN_MIRROR" = "true" ]; then \
+        npm config set registry https://registry.npmmirror.com; \
+    fi
 
-# 删除 package-lock.json 以SC避免因镜像源不一致导致的 404 错误
+# 删除 package-lock.json 以避免因镜像源不一致导致的 404 错误
 RUN rm -f package-lock.json
 
 # 安装依赖
@@ -28,12 +37,18 @@ RUN npm run build
 # 阶段2: 构建最终镜像
 FROM python:3.11-slim
 
+ARG USE_CN_MIRROR
+ARG TARGETPLATFORM
+ARG TARGETARCH
+
 # 设置工作目录
 WORKDIR /app
 
-# 使用国内镜像源加速
-RUN sed -i 's/deb.debian.org/mirrors.aliyun.com/g' /etc/apt/sources.list.d/debian.sources \
-    && sed -i 's/security.debian.org/mirrors.aliyun.com/g' /etc/apt/sources.list.d/debian.sources
+# 根据参数决定是否使用国内镜像源
+RUN if [ "$USE_CN_MIRROR" = "true" ]; then \
+        sed -i 's/deb.debian.org/mirrors.aliyun.com/g' /etc/apt/sources.list.d/debian.sources && \
+        sed -i 's/security.debian.org/mirrors.aliyun.com/g' /etc/apt/sources.list.d/debian.sources; \
+    fi
 
 # 安装系统依赖（添加数据库工具）
 RUN apt-get update && apt-get install -y \
@@ -45,13 +60,40 @@ RUN apt-get update && apt-get install -y \
 # 复制后端依赖文件
 COPY backend/requirements.txt ./
 
-# 先从PyTorch官方源安装CPU版本的torch（避免GPU依赖）
-RUN pip install --no-cache-dir torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu
+# 根据架构安装PyTorch CPU版本
+# arm64架构使用pip直接安装，amd64使用PyTorch官方CPU源
+RUN if [ "$TARGETARCH" = "arm64" ]; then \
+        pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu || \
+        pip install --no-cache-dir torch; \
+    else \
+        pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu; \
+    fi
 
-# 再安装其他Python依赖（使用阿里云镜像加速）
-RUN pip install --no-cache-dir -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
+# 安装其他Python依赖
+RUN if [ "$USE_CN_MIRROR" = "true" ]; then \
+        pip install --no-cache-dir -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/; \
+    else \
+        pip install --no-cache-dir -r requirements.txt; \
+    fi
 
-# 复制后端代码（包含embedding模型）
+# 创建embedding目录
+RUN mkdir -p /app/embedding
+
+# 设置 Sentence-Transformers 缓存目录
+ENV SENTENCE_TRANSFORMERS_HOME=/app/embedding
+
+# 下载 embedding 模型（从 HuggingFace）
+# 使用 Python 脚本预下载模型，这样运行时不需要网络
+RUN python -c "\
+from sentence_transformers import SentenceTransformer; \
+import os; \
+os.environ['SENTENCE_TRANSFORMERS_HOME'] = '/app/embedding'; \
+print('Downloading sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2...'); \
+model = SentenceTransformer('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2'); \
+print('Model downloaded successfully!'); \
+"
+
+# 复制后端代码（不包含embedding，因为已经下载了）
 COPY backend/ ./
 
 # 从前端构建阶段复制构建好的静态文件
@@ -77,11 +119,10 @@ ENV PYTHONUNBUFFERED=1
 ENV APP_HOST=0.0.0.0
 ENV APP_PORT=8000
 
-# 设置Transformers和Sentence-Transformers离线模式
+# 设置运行时为离线模式（模型已在构建时下载）
 ENV TRANSFORMERS_OFFLINE=1
 ENV HF_DATASETS_OFFLINE=1
 ENV HF_HUB_OFFLINE=1
-ENV SENTENCE_TRANSFORMERS_HOME=/app/embedding
 
 # 健康检查
 HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \