Skip to content
This repository has been archived by the owner on Jul 12, 2024. It is now read-only.

Commit

Permalink
fix: update the llama-index version and function
Browse files Browse the repository at this point in the history
  • Loading branch information
loyal812 committed Mar 28, 2024
1 parent edd6073 commit 3511866
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
4 changes: 2 additions & 2 deletions src/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ USER root
# Copy the requirements.txt file and install the Python dependencies.
COPY --chown=user:user ./requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt
# RUN pip install --upgrade llama-index
RUN pip install --upgrade llama-index

# Copy the project files into the container.
COPY --chown=user:user ./$PROJECT /home/user/$PROJECT
Expand All @@ -45,7 +45,7 @@ FROM base as test
# Copy the requirements.txt file and install the Python dependencies.
COPY --chown=user:user ./requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt
# RUN pip install --upgrade llama-index
RUN pip install --upgrade llama-index

USER root
# Copy the project files into the container.
Expand Down
10 changes: 5 additions & 5 deletions src/finetune/cl_fine_tuning.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,11 @@
import json
from itertools import cycle

from llama_index import SimpleDirectoryReader, ServiceContext, VectorStoreIndex
from llama_index.llms import OpenAI
from llama_index.evaluation import DatasetGenerator
from llama_index.callbacks import OpenAIFineTuningHandler
from llama_index.callbacks import CallbackManager
from llama_index.core import SimpleDirectoryReader, ServiceContext, VectorStoreIndex, Settings
from llama_index.llms.openai import OpenAI
from llama_index.core.evaluation import DatasetGenerator
from llama_index.finetuning.callbacks import OpenAIFineTuningHandler
from llama_index.core.callbacks import CallbackManager

from datasets import Dataset
from ragas import evaluate
Expand Down

0 comments on commit 3511866

Please sign in to comment.