renamed prompt_bench dir to tools

pipelines/tools/batch_bill_summarizer.py

@@ -23,7 +23,7 @@ import httpx
 import typer
 from tiktoken import Encoding, get_encoding
 
-from pipelines.prompt_bench.bill_token_compression import compress_bill_text
+from pipelines.tools.bill_token_compression import compress_bill_text
 
 _PROMPTS_PATH = (
     Path(__file__).resolve().parents[2]

pipelines/tools/compresion_test.py

@@ -24,7 +24,7 @@ from typing import Annotated
 import httpx
 import typer
 
-from pipelines.prompt_bench.bill_token_compression import compress_bill_text
+from pipelines.tools.bill_token_compression import compress_bill_text
 
 _PROMPTS_PATH = (
     Path(__file__).resolve().parents[2]

pipelines/tools/containers/finetune.py

@@ -9,7 +9,7 @@ from typing import Annotated
 
 import typer
 
-from pipelines.prompt_bench.containers.lib import check_gpu_free
+from pipelines.tools.containers.lib import check_gpu_free
 
 logger = logging.getLogger(__name__)
 

pipelines/tools/downloader.py

@@ -9,7 +9,7 @@ from typing import Annotated
 import typer
 from huggingface_hub import snapshot_download
 
-from pipelines.prompt_bench.models import BenchmarkConfig
+from pipelines.tools.models import BenchmarkConfig
 
 logger = logging.getLogger(__name__)
 

pipelines/tools/prompt_bench.py

@@ -11,11 +11,11 @@ from typing import Annotated
 
 import typer
 
-from pipelines.prompt_bench.containers.lib import check_gpu_free
-from pipelines.prompt_bench.containers.vllm import start_vllm, stop_vllm
-from pipelines.prompt_bench.downloader import is_model_present
-from pipelines.prompt_bench.models import BenchmarkConfig
-from pipelines.prompt_bench.vllm_client import VLLMClient
+from pipelines.tools.containers.lib import check_gpu_free
+from pipelines.tools.containers.vllm import start_vllm, stop_vllm
+from pipelines.tools.downloader import is_model_present
+from pipelines.tools.models import BenchmarkConfig
+from pipelines.tools.vllm_client import VLLMClient
 
 logger = logging.getLogger(__name__)