fix mkdocs auto build

Author: binary-husky

.github/workflows/doc.yaml

@@ -29,15 +29,17 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -q -e ".[dev,doc]"
+        pip install -e ".[dev,docs]"
 
     - name: Build Cookbook
       run: |
+        which mkdocs
+        mkdocs --version
         mkdocs build
 
     - name: Deploy to GitHub Pages
       uses: peaceiris/actions-gh-pages@v4
       with:
         github_token: ${{ secrets.GITHUB_TOKEN }}
-        publish_dir: docs/_build/preview/_build/html
-        cname: doc.agentjet.github.io
+        publish_dir: site
+        cname: doc.agentjet.github.io

pyproject.toml

@@ -59,7 +59,7 @@ reward = [
 ]
 
 flash_attn = [
-    "flash-attn==2.8.1"
+    "flash-attn==2.8.3"
 ]
 
 docs = [