diff --git a/README.md b/README.md index 4da5d77..7d6d359 100644 --- a/README.md +++ b/README.md @@ -69,15 +69,12 @@ graph TD ├── scripts/ │ └── pipelines/ │ ├── flows/ # Processing pipelines -│ ├── tasks/ # Modules used by the flows -│ └── inputs/ # Inputs used by the flows +│ └── tasks/ # Modules used by the flows └── src/ - ├── __init__.py ├── config.py # Environment variables configuration file └── utils/ ├── emb_factory.py # Embedding model factory - ├── llm_factory.py # LLM model factory - └── __init__.py + └── llm_factory.py # LLM model factory ``` --- diff --git a/changelog b/changelog index 9a1e763..93d4daa 100644 --- a/changelog +++ b/changelog @@ -9,7 +9,7 @@ All notable changes to the **Brunix Assistance Engine** will be documented in th ### Added - IMPLEMENTED: - `scripts/pipelines/flows/translate_mbpp.py`: pipeline to generate synthethic dataset from mbpp dataset. - - `scripts/input/prompts.py`: module containing prompts for pipelines. + - `scripts/tasks/prompts.py`: module containing prompts for pipelines. - `scripts/tasks/chunk.py`: module containing functions related to chunk management. - `synthethic_datasets`: folder containing generated synthethic datasets. - `src/config.py`: environment variables configuration file. diff --git a/scripts/pipelines/flows/translate_mbpp.py b/scripts/pipelines/flows/translate_mbpp.py index 3f42963..f89497c 100644 --- a/scripts/pipelines/flows/translate_mbpp.py +++ b/scripts/pipelines/flows/translate_mbpp.py @@ -8,8 +8,7 @@ from botocore.config import Config from pathlib import Path from langchain_core.messages import SystemMessage, HumanMessage from src.utils.llm_factory import create_chat_model -from src.config import RAW_DIR, INTERIM_DIR -from scripts.pipelines.input.prompts import get_prompt_mbpp +from scripts.pipelines.tasks.prompts import get_prompt_mbpp app = typer.Typer() diff --git a/scripts/pipelines/inputs/prompts.py b/scripts/pipelines/tasks/prompts.py similarity index 100% rename from scripts/pipelines/inputs/prompts.py rename to scripts/pipelines/tasks/prompts.py