Update branch

.devcontainer/devcontainer.json

@@ -0,0 +1,19 @@
+{
+  "name": "dottxt-ai",
+  "image": "mcr.microsoft.com/devcontainers/python:3.12",
+  "runArgs": [
+    "--device=nvidia.com/gpu=all"
+  ],
+  "hostRequirements": {
+    "gpu": "optional"
+  },
+  "features": {
+    "ghcr.io/devcontainers/features/conda:1": {},
+    "ghcr.io/devcontainers/features/nvidia-cuda:1": {
+      "installCudnn": true,
+      "installToolkit": true,
+      "cudaVersion": "12.4"
+    },
+    "ghcr.io/devcontainers/features/rust:1": {}
+  }
+}

.editorconfig

@@ -0,0 +1,15 @@
+# EditorConfig is awesome: https://EditorConfig.org
+
+# top-most EditorConfig file
+root = true
+
+[*]
+indent_style = space
+indent_size = 4
+end_of_line = lf
+charset = utf-8
+trim_trailing_whitespace = true
+insert_final_newline = true
+
+[*.yaml]
+indent_size = 2

.github/ISSUE_TEMPLATE/bug_report.yml

@@ -63,9 +63,9 @@ body:
       label: "Outlines/Python version information:"
       description: |
           Please run the following code and paste the output here.
-          python -c "from outlines import _version; print(_version.__version__)"
-          python -c "import sys; print('Python', sys.version)"
-          pip freeze
+          python -c "from outlines import _version; print(_version.__version__)";
+          python -c "import sys; print('Python', sys.version)";
+          pip freeze;
       value: |
           Version information
           <details>

.github/ISSUE_TEMPLATE/config.yml

@@ -1,4 +1,4 @@
 contact_links:
   - name: 🤔 Questions & Help
-    url: https://github.com/outlines-dev/outlines/discussions/new
+    url: https://github.com/dottxt-ai/outlines/discussions/new
     about: "If you have a question about how to use Outlines, please start a discussion."

.github/PULL_REQUEST_TEMPLATE/pull_request_template.md

@@ -18,6 +18,6 @@ A few important guidelines and requirements before we can merge your PR:
 Consider opening a **Draft PR** if your work is still in progress but you would
 like some feedback from other contributors.
 
-[issues]: https://github.com/outlines-dev/outlines/issues
+[issues]: https://github.com/dottxt-ai/outlines/issues
 [git-guidelines]: https://tbaggery.com/2008/04/19/a-note-about-git-commit-messages.html
 [docstring-guidelines]: https://numpydoc.readthedocs.io/en/latest/format.html

.github/workflows/build_documentation.yml

@@ -3,6 +3,7 @@ name: Build the documentation
 on:
   pull_request:
     branches: [main]
+  workflow_dispatch:
 
 jobs:
   build:

.github/workflows/publish_documentation.yml

@@ -1,9 +1,13 @@
 name: Publish the documentation
 
 on:
+  workflow_dispatch:
   push:
     branches:
       - main
+  release:
+    types:
+      - created
 
 permissions:
   contents: write
@@ -13,6 +17,8 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
       - uses: actions/setup-python@v4
         with:
           python-version: 3.x
@@ -24,7 +30,24 @@ jobs:
           restore-keys: |
             mkdocs-material-
       - run: pip install -r requirements-doc.txt
-      - name: Build documentation
+      - run: mkdocs build
+
+      - name: Set up Git
+        run: |
+          git config user.name ${{ github.actor }}
+          git config user.email ${{ github.actor }}@users.noreply.github.com
+
+      - name: Publish Tag as latest
+        env:
+          GOOGLE_ANALYTICS_KEY: ${{ secrets.GOOGLE_ANALYTICS_KEY }}
+        if: github.event_name == 'release'
+        run: |
+          mike deploy --push --update-aliases ${{ github.ref_name }} latest
+          mike set-default --push latest
+
+      - name: Publish main as unstable
         env:
           GOOGLE_ANALYTICS_KEY: ${{ secrets.GOOGLE_ANALYTICS_KEY }}
-        run: mkdocs gh-deploy --force
+        if: github.event_name == 'push'
+        run: |
+          mike deploy --push --update-aliases ${{ github.ref_name }} unstable

.github/workflows/tests.yml

@@ -2,7 +2,7 @@ name: Tests
 
 on:
   pull_request:
-    branches: [main]
+    branches: [main,v1.0]
   push:
     branches: [main]
 
@@ -22,26 +22,45 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.10"]
+        python-version: ["3.10"]
     steps:
     - uses: actions/checkout@v3
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/setup-python@v4
       with:
         python-version: ${{ matrix.python-version }}
+    - name: Install Ollama
+      run: |
+        curl -fsSL https://ollama.com/install.sh | sh
+        ollama --version
+        ollama pull tinyllama
     - name: Set up test environment
       run: |
         python -m pip install --upgrade pip
-        pip install .[test]
+        pip install uv
+        uv sync --no-group test-gpu --extra test
+    - name: Create matrix id
+      id: matrix-id
+      env:
+        MATRIX_CONTEXT: ${{ toJson(matrix) }}
+      run: |
+        echo $MATRIX_CONTEXT
+        export MATRIX_ID=`echo $MATRIX_CONTEXT | md5sum | cut -c 1-32`
+        echo $MATRIX_ID
+        echo "::set-output name=id::$MATRIX_ID"
     - name: Run tests
       run: |
-        pytest --cov=outlines
+        rm -f .coverage*
+        uv run coverage erase
+        uv run python -m coverage run --branch --source=outlines --parallel-mode -m pytest -x -m 'not api_call'
     - name: Upload coverage data
-      uses: actions/upload-artifact@v3
+      uses: actions/upload-artifact@v4
       with:
         name: coverage-data
-        path: .coverage*
+        path: .coverage.*
         if-no-files-found: ignore
+        include-hidden-files: true
+        overwrite: true
 
   coverage:
     name: Combine & check coverage.
@@ -62,31 +81,23 @@ jobs:
         run: |
           pip install --upgrade "coverage[toml]>=5.1" diff-cover
 
-      - uses: actions/download-artifact@v3
+      - uses: actions/download-artifact@v4
         with:
           name: coverage-data
 
-      - name: Fetch master for coverage diff
-        run: |
-          git fetch --no-tags --prune origin main
-
       - name: Combine coverage & fail if it's <100%.
         run: |
-          # Combine coverage files (not needed now, but maybe later)
-          # python -m coverage combine
-
-          # Produce an html report with absolute coverage information
+          python -m coverage combine
           python -m coverage html --skip-covered --skip-empty
-
-          # Report relative coverage and write to the workflow's summary
           python -m coverage xml
-          diff-cover coverage.xml --markdown-report=coverage.md --fail-under=100 || (cat coverage.md >> $GITHUB_STEP_SUMMARY && exit 1)
+          python -m coverage report --fail-under=100 || (python -m coverage report && exit 1)
 
       - name: Upload HTML report if check failed.
-        uses: actions/upload-artifact@v3
+        uses: actions/upload-artifact@v4
         with:
           name: html-report
           path: htmlcov
+          overwrite: true
         if: ${{ failure() }}
 
   build-wheel: