AutoRAG 0.3.4__tar.gz → 0.3.5__tar.gz

{autorag-0.3.4 → autorag-0.3.5}/.dockerignore

@@ -124,6 +124,7 @@ celerybeat.pid
 .venv
 env/
 venv/
+venv*/
 ENV/
 env.bak/
 venv.bak/

{autorag-0.3.4 → autorag-0.3.5}/.github/workflows/docker-push.yml

@@ -36,7 +36,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        variant: [all] # [all, ko, dev, parsing]
+        variant: [api, parsing] # [ko, ja, dev, parsing, api]
     steps:
     - name: Checkout code
       uses: actions/checkout@v4.2.0
@@ -67,7 +67,7 @@ jobs:
       uses: docker/build-push-action@v6.9.0
       with:
         context: .
-        file: ./Dockerfile
+        file: ./docker/base/Dockerfile
         push: true
         # push: ${{ github.event_name != 'pull_request' && matrix.variant != 'test' }}
         tags: ${{ env.DOCKER_REPO }}:${{ env.VERSION }}-${{matrix.variant}}

autorag-0.3.5/.github/workflows/gpu-docker-push.yml

@@ -0,0 +1,91 @@
+name: Docker Push
+
+on:
+  push:
+    branches: [ "main" ]
+
+env:
+  DOCKER_REPO: "autoraghq/autorag"
+
+jobs:
+  check-version:
+    runs-on: ubuntu-latest
+    outputs:
+      changed: ${{ steps.version_changed.outputs.changed }}
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v4.2.0
+
+    - name: Get changed files
+      id: changed-files
+      uses: tj-actions/changed-files@v45.0.3
+
+    - name: Check for VERSION file change
+      id: version_changed
+      env:
+        ALL_CHANGED_FILES: ${{ steps.changed-files.outputs.all_changed_files }}
+      run: |
+        echo "changed=false" >> $GITHUB_OUTPUT
+        if echo "${ALL_CHANGED_FILES}" | grep -q 'VERSION'; then
+          echo "changed=true" >> $GITHUB_OUTPUT
+        fi
+
+  build-and-push:
+    needs: check-version
+    if: needs.check-version.outputs.changed == 'true'
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        variant: [gpu, gpu-parsing] # [ko, ja, dev, parsing, api]
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v4.2.0
+
+    - name: Read VERSION file
+      run: echo "VERSION=$(cat autorag/VERSION)" >> $GITHUB_ENV
+
+    - name: Set up Docker Buildx
+      uses: docker/setup-buildx-action@v3.7.1
+
+    - name: Login to Docker Hub
+      uses: docker/login-action@v3.3.0
+      with:
+        username: ${{ secrets.DOCKER_HUB_USERNAME }}
+        password: ${{ secrets.DOCKER_HUB_ACCESS_TOKEN }}
+
+    - name: Extract metadata (tags, labels) for Docker
+      id: meta
+      uses: docker/metadata-action@v5.5.1
+      with:
+        images: ${{ env.DOCKER_REPO }}
+        tags: |
+          type=raw,value=${{ env.VERSION }}-${{ matrix.variant }}
+          type=raw,value=${{ matrix.variant }},enable=${{ github.ref == format('refs/heads/{0}', 'main') }}
+          type=raw,value=latest-${{ matrix.variant }},enable=${{ github.ref == format('refs/heads/{0}', 'main') }}
+
+    - name: Build and push Docker image
+      uses: docker/build-push-action@v6.9.0
+      with:
+        context: .
+        file: ./docker/gpu/Dockerfile
+        push: true
+        # push: ${{ github.event_name != 'pull_request' && matrix.variant != 'test' }}
+        tags: ${{ env.DOCKER_REPO }}:${{ env.VERSION }}-${{matrix.variant}}
+        labels: ${{ steps.meta.outputs.labels }}
+        build-args: |
+          TARGET_STAGE=${{ matrix.variant }}
+
+    - name: Tag and push 'all' for production
+      run: |
+        docker pull ${{ env.DOCKER_REPO }}:${{ env.VERSION }}-${{matrix.variant}}
+        docker tag ${{ env.DOCKER_REPO }}:${{ env.VERSION }}-${{matrix.variant}} ${{ env.DOCKER_REPO }}:latest
+        docker push ${{ env.DOCKER_REPO }}:${{ env.VERSION }}-${{matrix.variant}}
+        docker push ${{ env.DOCKER_REPO }}:latest
+
+    # - name: Update Docker Hub description
+    #   uses: peter-evans/dockerhub-description@v3
+    #   with:
+    #     username: ${{ secrets.DOCKER_HUB_USERNAME }}
+    #     password: ${{ secrets.DOCKER_HUB_PASSWORD }}
+    #     repository: ${{ env.DOCKER_REPO }}
+    #     short-description: ${{ github.event.repository.description }}

{autorag-0.3.4 → autorag-0.3.5}/.github/workflows/sphinx.yml

@@ -10,8 +10,11 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v4
+    - uses: actions/setup-python@v5
+      with:
+        python-version: '3.10'
     - name: Upgrade pip
-      run: | 
+      run: |
         python -m pip install --upgrade pip
     - name: Install gcc
       run: |

{autorag-0.3.4 → autorag-0.3.5}/.github/workflows/test.yml

@@ -16,6 +16,9 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.10'
       - uses: actions/setup-java@v4
         with:
           distribution: 'zulu'
@@ -23,6 +26,9 @@ jobs:
       - name: Upgrade pip
         run: |
           python3 -m pip install --upgrade pip
+      - name: Update apt-get
+        run: |
+          sudo apt-get update
       - name: Install gcc
         run: |
           sudo apt-get install gcc
@@ -34,7 +40,7 @@ jobs:
           sudo apt-get install tesseract-ocr
       - name: Install AutoRAG
         run: |
-          pip install -e '.[all]'
+          pip install -e '.[ko,dev,parse,ja]'
       - name: Install dependencies
         run: |
           pip install -r tests/requirements.txt

{autorag-0.3.4 → autorag-0.3.5}/AutoRAG.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: AutoRAG
-Version: 0.3.4
+Version: 0.3.5
 Summary: Automatically Evaluate RAG pipelines with your own data. Find optimal structure for new RAG product.
 Author-email: Marker-Inc <vkehfdl1@gmail.com>
 License:                                  Apache License
@@ -225,7 +225,6 @@ Requires-Dist: tqdm
 Requires-Dist: tiktoken>=0.7.0
 Requires-Dist: openai>=1.0.0
 Requires-Dist: rank_bm25
-Requires-Dist: transformers
 Requires-Dist: pyyaml
 Requires-Dist: pyarrow
 Requires-Dist: fastparquet
@@ -235,17 +234,9 @@ Requires-Dist: rouge_score
 Requires-Dist: rich
 Requires-Dist: chromadb>=0.5.0
 Requires-Dist: click
-Requires-Dist: torch
-Requires-Dist: sentencepiece
 Requires-Dist: cohere>=5.8.0
 Requires-Dist: tokenlog>=0.0.2
 Requires-Dist: aiohttp
-Requires-Dist: bert_score
-Requires-Dist: sentence-transformers
-Requires-Dist: FlagEmbedding
-Requires-Dist: llmlingua
-Requires-Dist: peft
-Requires-Dist: optimum[nncf,openvino]
 Requires-Dist: voyageai
 Requires-Dist: mixedbread-ai
 Requires-Dist: quart
@@ -253,11 +244,8 @@ Requires-Dist: llama-index>=0.11.0
 Requires-Dist: llama-index-core>=0.11.0
 Requires-Dist: llama-index-readers-file
 Requires-Dist: llama-index-embeddings-openai
-Requires-Dist: llama-index-embeddings-huggingface
 Requires-Dist: llama-index-llms-openai>=0.2.7
-Requires-Dist: llama-index-llms-huggingface
 Requires-Dist: llama-index-llms-openai-like
-Requires-Dist: llama-index-llms-ollama
 Requires-Dist: llama-index-retrievers-bm25
 Requires-Dist: streamlit
 Requires-Dist: gradio
@@ -288,7 +276,22 @@ Requires-Dist: pdf2image; extra == "parse"
 Provides-Extra: ja
 Requires-Dist: sudachipy>=0.6.8; extra == "ja"
 Requires-Dist: sudachidict_core; extra == "ja"
+Provides-Extra: gpu
+Requires-Dist: torch; extra == "gpu"
+Requires-Dist: sentencepiece; extra == "gpu"
+Requires-Dist: bert_score; extra == "gpu"
+Requires-Dist: optimum[nncf,openvino]; extra == "gpu"
+Requires-Dist: peft; extra == "gpu"
+Requires-Dist: llmlingua; extra == "gpu"
+Requires-Dist: FlagEmbedding; extra == "gpu"
+Requires-Dist: sentence-transformers; extra == "gpu"
+Requires-Dist: transformers; extra == "gpu"
+Requires-Dist: llama-index-llms-ollama; extra == "gpu"
+Requires-Dist: llama-index-embeddings-huggingface; extra == "gpu"
+Requires-Dist: llama-index-llms-huggingface; extra == "gpu"
+Requires-Dist: onnxruntime; extra == "gpu"
 Provides-Extra: all
+Requires-Dist: AutoRAG[gpu]; extra == "all"
 Requires-Dist: AutoRAG[ko]; extra == "all"
 Requires-Dist: AutoRAG[dev]; extra == "all"
 Requires-Dist: AutoRAG[parse]; extra == "all"
@@ -358,6 +361,17 @@ We recommend using Python version 3.10 or higher for AutoRAG.
 pip install AutoRAG
 ```
 
+If you want to use the local models, you need to install gpu version.
+
+```bash
+pip install "AutoRAG[gpu]"
+```
+
+Or for parsing, you can use the parsing version.
+```bash
+pip install "AutoRAG[gpu,parse]"
+```
+
 # Data Creation
 
 ![image](https://github.com/user-attachments/assets/6079f696-207c-4221-8d28-5561a203dfe2)
@@ -490,7 +504,9 @@ This guide provides a quick overview of building and running the AutoRAG Docker
 
 ### 🚀 Building the Docker Image
 
-#### 1.Download dataset for [Turorial Step 1](https://colab.research.google.com/drive/19OEQXO_pHN6gnn2WdfPd4hjnS-4GurVd?usp=sharing)
+Tip: If you want to build an image for a gpu version, you can use `autoraghq/autorag:gpu` or `autoraghq/autorag:gpu-parsing`
+
+#### 1.Download dataset for [Tutorial Step 1](https://colab.research.google.com/drive/19OEQXO_pHN6gnn2WdfPd4hjnS-4GurVd?usp=sharing)
 ```bash
 python sample_dataset/eli5/load_eli5_dataset.py --save_path projects/tutorial_1
 ```
@@ -503,7 +519,7 @@ docker run --rm -it \
   -v ~/.cache/huggingface:/root/.cache/huggingface \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
-  autoraghq/autorag:all evaluate \
+  autoraghq/autorag:api evaluate \
   --config /usr/src/app/projects/tutorial_1/config.yaml \
   --qa_data_path /usr/src/app/projects/tutorial_1/qa_test.parquet \
   --corpus_data_path /usr/src/app/projects/tutorial_1/corpus.parquet \
@@ -517,7 +533,7 @@ docker run --rm -it \
   -v ~/.cache/huggingface:/root/.cache/huggingface \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
-  autoraghq/autorag:all validate \
+  autoraghq/autorag:api validate \
   --config /usr/src/app/projects/tutorial_1/config.yaml \
   --qa_data_path /usr/src/app/projects/tutorial_1/qa_test.parquet \
   --corpus_data_path /usr/src/app/projects/tutorial_1/corpus.parquet
@@ -531,7 +547,7 @@ docker run --rm -it \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
   -p 8502:8502 \
-  autoraghq/autorag:all dashboard \
+  autoraghq/autorag:api dashboard \
     --trial_dir /usr/src/app/projects/tutorial_1/0
 ```
 
@@ -543,7 +559,7 @@ docker run --rm -it \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
   -p 8501:8501 \
-  autoraghq/autorag:all run_web --trial_path ./projects/tutorial_1/0
+  autoraghq/autorag:api run_web --trial_path ./projects/tutorial_1/0
 ```
 
 #### Key Points :

{autorag-0.3.4 → autorag-0.3.5}/AutoRAG.egg-info/SOURCES.txt

@@ -3,7 +3,6 @@
 .pre-commit-config.yaml
 CODE_OF_CONDUCT.md
 CONTRIBUTING.md
-Dockerfile
 LICENSE
 README.md
 build_and_push.sh
@@ -15,6 +14,7 @@ requirements.txt
 .github/ISSUE_TEMPLATE/bug_report.md
 .github/ISSUE_TEMPLATE/feature_request.md
 .github/workflows/docker-push.yml
+.github/workflows/gpu-docker-push.yml
 .github/workflows/publish.yml
 .github/workflows/sphinx.yml
 .github/workflows/test.yml
@@ -137,6 +137,7 @@ autorag/nodes/passagereranker/cohere.py
 autorag/nodes/passagereranker/colbert.py
 autorag/nodes/passagereranker/flag_embedding.py
 autorag/nodes/passagereranker/flag_embedding_llm.py
+autorag/nodes/passagereranker/flashrank.py
 autorag/nodes/passagereranker/jina.py
 autorag/nodes/passagereranker/koreranker.py
 autorag/nodes/passagereranker/mixedbreadai.py
@@ -181,6 +182,8 @@ autorag/schema/node.py
 autorag/utils/__init__.py
 autorag/utils/preprocess.py
 autorag/utils/util.py
+docker/base/Dockerfile
+docker/gpu/Dockerfile
 docs/Makefile
 docs/make.bat
 docs/requirements.txt
@@ -299,6 +302,7 @@ docs/source/nodes/passage_reranker/cohere.md
 docs/source/nodes/passage_reranker/colbert.md
 docs/source/nodes/passage_reranker/flag_embedding_llm_reranker.md
 docs/source/nodes/passage_reranker/flag_embedding_reranker.md
+docs/source/nodes/passage_reranker/flashrank_reranker.md
 docs/source/nodes/passage_reranker/jina_reranker.md
 docs/source/nodes/passage_reranker/koreranker.md
 docs/source/nodes/passage_reranker/mixedbreadai_reranker.md
@@ -338,6 +342,7 @@ sample_config/parse/parse_hybird.yaml
 sample_config/parse/parse_ko.yaml
 sample_config/parse/parse_ocr.yaml
 sample_config/parse/simple_parse.yaml
+sample_config/rag/api_full.yaml
 sample_config/rag/config_korean.yaml
 sample_config/rag/extracted_sample.yaml
 sample_config/rag/full.yaml
@@ -430,6 +435,7 @@ tests/autorag/nodes/passagereranker/test_cohere_reranker.py
 tests/autorag/nodes/passagereranker/test_colbert_reranker.py
 tests/autorag/nodes/passagereranker/test_flag_embedding.py
 tests/autorag/nodes/passagereranker/test_flag_embedding_llm.py
+tests/autorag/nodes/passagereranker/test_flashrank_reranker.py
 tests/autorag/nodes/passagereranker/test_jina_reranker.py
 tests/autorag/nodes/passagereranker/test_koreranker.py
 tests/autorag/nodes/passagereranker/test_mixedbreadai_reranker.py

{autorag-0.3.4 → autorag-0.3.5}/AutoRAG.egg-info/requires.txt

@@ -4,7 +4,6 @@ tqdm
 tiktoken>=0.7.0
 openai>=1.0.0
 rank_bm25
-transformers
 pyyaml
 pyarrow
 fastparquet
@@ -14,17 +13,9 @@ rouge_score
 rich
 chromadb>=0.5.0
 click
-torch
-sentencepiece
 cohere>=5.8.0
 tokenlog>=0.0.2
 aiohttp
-bert_score
-sentence-transformers
-FlagEmbedding
-llmlingua
-peft
-optimum[nncf,openvino]
 voyageai
 mixedbread-ai
 quart
@@ -32,11 +23,8 @@ llama-index>=0.11.0
 llama-index-core>=0.11.0
 llama-index-readers-file
 llama-index-embeddings-openai
-llama-index-embeddings-huggingface
 llama-index-llms-openai>=0.2.7
-llama-index-llms-huggingface
 llama-index-llms-openai-like
-llama-index-llms-ollama
 llama-index-retrievers-bm25
 streamlit
 gradio
@@ -51,6 +39,7 @@ ipywidgets
 ipywidgets_bokeh
 
 [all]
+AutoRAG[gpu]
 AutoRAG[ko]
 AutoRAG[dev]
 AutoRAG[parse]
@@ -60,6 +49,21 @@ AutoRAG[ja]
 ruff
 pre-commit
 
+[gpu]
+torch
+sentencepiece
+bert_score
+optimum[nncf,openvino]
+peft
+llmlingua
+FlagEmbedding
+sentence-transformers
+transformers
+llama-index-llms-ollama
+llama-index-embeddings-huggingface
+llama-index-llms-huggingface
+onnxruntime
+
 [ja]
 sudachipy>=0.6.8
 sudachidict_core

{autorag-0.3.4 → autorag-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: AutoRAG
-Version: 0.3.4
+Version: 0.3.5
 Summary: Automatically Evaluate RAG pipelines with your own data. Find optimal structure for new RAG product.
 Author-email: Marker-Inc <vkehfdl1@gmail.com>
 License:                                  Apache License
@@ -225,7 +225,6 @@ Requires-Dist: tqdm
 Requires-Dist: tiktoken>=0.7.0
 Requires-Dist: openai>=1.0.0
 Requires-Dist: rank_bm25
-Requires-Dist: transformers
 Requires-Dist: pyyaml
 Requires-Dist: pyarrow
 Requires-Dist: fastparquet
@@ -235,17 +234,9 @@ Requires-Dist: rouge_score
 Requires-Dist: rich
 Requires-Dist: chromadb>=0.5.0
 Requires-Dist: click
-Requires-Dist: torch
-Requires-Dist: sentencepiece
 Requires-Dist: cohere>=5.8.0
 Requires-Dist: tokenlog>=0.0.2
 Requires-Dist: aiohttp
-Requires-Dist: bert_score
-Requires-Dist: sentence-transformers
-Requires-Dist: FlagEmbedding
-Requires-Dist: llmlingua
-Requires-Dist: peft
-Requires-Dist: optimum[nncf,openvino]
 Requires-Dist: voyageai
 Requires-Dist: mixedbread-ai
 Requires-Dist: quart
@@ -253,11 +244,8 @@ Requires-Dist: llama-index>=0.11.0
 Requires-Dist: llama-index-core>=0.11.0
 Requires-Dist: llama-index-readers-file
 Requires-Dist: llama-index-embeddings-openai
-Requires-Dist: llama-index-embeddings-huggingface
 Requires-Dist: llama-index-llms-openai>=0.2.7
-Requires-Dist: llama-index-llms-huggingface
 Requires-Dist: llama-index-llms-openai-like
-Requires-Dist: llama-index-llms-ollama
 Requires-Dist: llama-index-retrievers-bm25
 Requires-Dist: streamlit
 Requires-Dist: gradio
@@ -288,7 +276,22 @@ Requires-Dist: pdf2image; extra == "parse"
 Provides-Extra: ja
 Requires-Dist: sudachipy>=0.6.8; extra == "ja"
 Requires-Dist: sudachidict_core; extra == "ja"
+Provides-Extra: gpu
+Requires-Dist: torch; extra == "gpu"
+Requires-Dist: sentencepiece; extra == "gpu"
+Requires-Dist: bert_score; extra == "gpu"
+Requires-Dist: optimum[nncf,openvino]; extra == "gpu"
+Requires-Dist: peft; extra == "gpu"
+Requires-Dist: llmlingua; extra == "gpu"
+Requires-Dist: FlagEmbedding; extra == "gpu"
+Requires-Dist: sentence-transformers; extra == "gpu"
+Requires-Dist: transformers; extra == "gpu"
+Requires-Dist: llama-index-llms-ollama; extra == "gpu"
+Requires-Dist: llama-index-embeddings-huggingface; extra == "gpu"
+Requires-Dist: llama-index-llms-huggingface; extra == "gpu"
+Requires-Dist: onnxruntime; extra == "gpu"
 Provides-Extra: all
+Requires-Dist: AutoRAG[gpu]; extra == "all"
 Requires-Dist: AutoRAG[ko]; extra == "all"
 Requires-Dist: AutoRAG[dev]; extra == "all"
 Requires-Dist: AutoRAG[parse]; extra == "all"
@@ -358,6 +361,17 @@ We recommend using Python version 3.10 or higher for AutoRAG.
 pip install AutoRAG
 ```
 
+If you want to use the local models, you need to install gpu version.
+
+```bash
+pip install "AutoRAG[gpu]"
+```
+
+Or for parsing, you can use the parsing version.
+```bash
+pip install "AutoRAG[gpu,parse]"
+```
+
 # Data Creation
 
 ![image](https://github.com/user-attachments/assets/6079f696-207c-4221-8d28-5561a203dfe2)
@@ -490,7 +504,9 @@ This guide provides a quick overview of building and running the AutoRAG Docker
 
 ### 🚀 Building the Docker Image
 
-#### 1.Download dataset for [Turorial Step 1](https://colab.research.google.com/drive/19OEQXO_pHN6gnn2WdfPd4hjnS-4GurVd?usp=sharing)
+Tip: If you want to build an image for a gpu version, you can use `autoraghq/autorag:gpu` or `autoraghq/autorag:gpu-parsing`
+
+#### 1.Download dataset for [Tutorial Step 1](https://colab.research.google.com/drive/19OEQXO_pHN6gnn2WdfPd4hjnS-4GurVd?usp=sharing)
 ```bash
 python sample_dataset/eli5/load_eli5_dataset.py --save_path projects/tutorial_1
 ```
@@ -503,7 +519,7 @@ docker run --rm -it \
   -v ~/.cache/huggingface:/root/.cache/huggingface \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
-  autoraghq/autorag:all evaluate \
+  autoraghq/autorag:api evaluate \
   --config /usr/src/app/projects/tutorial_1/config.yaml \
   --qa_data_path /usr/src/app/projects/tutorial_1/qa_test.parquet \
   --corpus_data_path /usr/src/app/projects/tutorial_1/corpus.parquet \
@@ -517,7 +533,7 @@ docker run --rm -it \
   -v ~/.cache/huggingface:/root/.cache/huggingface \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
-  autoraghq/autorag:all validate \
+  autoraghq/autorag:api validate \
   --config /usr/src/app/projects/tutorial_1/config.yaml \
   --qa_data_path /usr/src/app/projects/tutorial_1/qa_test.parquet \
   --corpus_data_path /usr/src/app/projects/tutorial_1/corpus.parquet
@@ -531,7 +547,7 @@ docker run --rm -it \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
   -p 8502:8502 \
-  autoraghq/autorag:all dashboard \
+  autoraghq/autorag:api dashboard \
     --trial_dir /usr/src/app/projects/tutorial_1/0
 ```
 
@@ -543,7 +559,7 @@ docker run --rm -it \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
   -p 8501:8501 \
-  autoraghq/autorag:all run_web --trial_path ./projects/tutorial_1/0
+  autoraghq/autorag:api run_web --trial_path ./projects/tutorial_1/0
 ```
 
 #### Key Points :

{autorag-0.3.4 → autorag-0.3.5}/README.md

@@ -62,6 +62,17 @@ We recommend using Python version 3.10 or higher for AutoRAG.
 pip install AutoRAG
 ```
 
+If you want to use the local models, you need to install gpu version.
+
+```bash
+pip install "AutoRAG[gpu]"
+```
+
+Or for parsing, you can use the parsing version.
+```bash
+pip install "AutoRAG[gpu,parse]"
+```
+
 # Data Creation
 
 ![image](https://github.com/user-attachments/assets/6079f696-207c-4221-8d28-5561a203dfe2)
@@ -194,7 +205,9 @@ This guide provides a quick overview of building and running the AutoRAG Docker
 
 ### 🚀 Building the Docker Image
 
-#### 1.Download dataset for [Turorial Step 1](https://colab.research.google.com/drive/19OEQXO_pHN6gnn2WdfPd4hjnS-4GurVd?usp=sharing)
+Tip: If you want to build an image for a gpu version, you can use `autoraghq/autorag:gpu` or `autoraghq/autorag:gpu-parsing`
+
+#### 1.Download dataset for [Tutorial Step 1](https://colab.research.google.com/drive/19OEQXO_pHN6gnn2WdfPd4hjnS-4GurVd?usp=sharing)
 ```bash
 python sample_dataset/eli5/load_eli5_dataset.py --save_path projects/tutorial_1
 ```
@@ -207,7 +220,7 @@ docker run --rm -it \
   -v ~/.cache/huggingface:/root/.cache/huggingface \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
-  autoraghq/autorag:all evaluate \
+  autoraghq/autorag:api evaluate \
   --config /usr/src/app/projects/tutorial_1/config.yaml \
   --qa_data_path /usr/src/app/projects/tutorial_1/qa_test.parquet \
   --corpus_data_path /usr/src/app/projects/tutorial_1/corpus.parquet \
@@ -221,7 +234,7 @@ docker run --rm -it \
   -v ~/.cache/huggingface:/root/.cache/huggingface \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
-  autoraghq/autorag:all validate \
+  autoraghq/autorag:api validate \
   --config /usr/src/app/projects/tutorial_1/config.yaml \
   --qa_data_path /usr/src/app/projects/tutorial_1/qa_test.parquet \
   --corpus_data_path /usr/src/app/projects/tutorial_1/corpus.parquet
@@ -235,7 +248,7 @@ docker run --rm -it \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
   -p 8502:8502 \
-  autoraghq/autorag:all dashboard \
+  autoraghq/autorag:api dashboard \
     --trial_dir /usr/src/app/projects/tutorial_1/0
 ```
 
@@ -247,7 +260,7 @@ docker run --rm -it \
   -v $(pwd)/projects:/usr/src/app/projects \
   -e OPENAI_API_KEY=${OPENAI_API_KEY} \
   -p 8501:8501 \
-  autoraghq/autorag:all run_web --trial_path ./projects/tutorial_1/0
+  autoraghq/autorag:api run_web --trial_path ./projects/tutorial_1/0
 ```
 
 #### Key Points :

autorag-0.3.5/autorag/VERSION

@@ -0,0 +1 @@
+0.3.5