From 7ce8b3dd4a835b6076479feb098efbe114a3b039 Mon Sep 17 00:00:00 2001
From: Fangyin Cheng <staneyffer@gmail.com>
Date: Mon, 3 Mar 2025 12:00:54 +0800
Subject: [PATCH] ci(docs): Fix build doc error

---
 .github/workflows/doc-image-publish.yml | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/doc-image-publish.yml b/.github/workflows/doc-image-publish.yml
index bf784a318..13c1812e0 100644
--- a/.github/workflows/doc-image-publish.yml
+++ b/.github/workflows/doc-image-publish.yml
@@ -1,5 +1,4 @@
 name: Build and push docs image
-
 on:
   push:
     branches:
@@ -14,15 +13,29 @@ on:
   release:
     types: [published]
   workflow_dispatch:
-
+    inputs:
+      branch:
+        description: 'Branch to build (default: main)'
+        required: true
+        default: 'main'
+        type: string
 permissions:
   contents: read
-
 jobs:
   build-image:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0 
+          ref: ${{ github.event_name == 'workflow_dispatch' && inputs.branch || github.ref }}
+      
+      - name: Debug git status 
+        run: |
+            echo "Current commit: $(git rev-parse HEAD)"
+            echo "Current branch: $(git rev-parse --abbrev-ref HEAD || echo 'detached HEAD')"
+            echo "Main branch commit: $(git rev-parse origin/main)"
+            echo "Using branch: ${{ github.event_name == 'workflow_dispatch' && inputs.branch || github.ref_name }}"
       
       - name: Set up QEMU
         uses: docker/setup-qemu-action@v2
@@ -47,6 +60,7 @@ jobs:
           file: ./docs/Dockerfile-deploy
           platforms: linux/amd64
           push: false
+      
       # Only run when it is not a PR, build and push the image
       - name: Build and push
         if: github.event_name != 'pull_request'
@@ -56,4 +70,4 @@ jobs:
           file: ./docs/Dockerfile-deploy
           platforms: linux/amd64
           push: true
-          tags: eosphorosai/dbgpt-docs:${{ github.ref_name }},eosphorosai/dbgpt-docs:latest
\ No newline at end of file
+          tags: eosphorosai/dbgpt-docs:${{ github.event_name == 'workflow_dispatch' && inputs.branch || github.ref_name }},eosphorosai/dbgpt-docs:latest
\ No newline at end of file