diff --git a/.github/workflows/pypi-publish.yml b/.github/workflows/pypi-publish.yml
index 78850d4d..45db4f7d 100644
--- a/.github/workflows/pypi-publish.yml
+++ b/.github/workflows/pypi-publish.yml
@@ -13,11 +13,27 @@ jobs:
 
     steps:
       - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0  # Fetch all history for tags
 
       - uses: actions/setup-python@v5
         with:
           python-version: "3.x"
 
+      - name: Get version from tag
+        id: get_version
+        run: |
+          TAG=$(git describe --tags --abbrev=0)
+          echo "Found tag: $TAG"
+          echo "Extracted version: $TAG"
+          echo "version=$VERSION" >> $GITHUB_OUTPUT
+
+      - name: Update version in __init__.py
+        run: |
+          sed -i "s/__version__ = \".*\"/__version__ = \"${{ steps.get_version.outputs.version }}\"/" lightrag/__init__.py
+          echo "Updated __init__.py with version ${{ steps.get_version.outputs.version }}"
+          cat lightrag/__init__.py | grep __version__
+
       - name: Build release distributions
         run: |
           python -m pip install build