converted default_config_path and get_config_dir
This commit is contained in:
@@ -24,14 +24,10 @@ from typing import Annotated
|
||||
import httpx
|
||||
import typer
|
||||
|
||||
from pipelines.config import get_config_dir
|
||||
from pipelines.tools.bill_token_compression import compress_bill_text
|
||||
|
||||
_PROMPTS_PATH = (
|
||||
Path(__file__).resolve().parents[2]
|
||||
/ "config"
|
||||
/ "prompts"
|
||||
/ "summarization_prompts.toml"
|
||||
)
|
||||
_PROMPTS_PATH = get_config_dir() / "prompts" / "summarization_prompts.toml"
|
||||
_PROMPTS = tomllib.loads(_PROMPTS_PATH.read_text())["summarization"]
|
||||
SUMMARIZATION_SYSTEM_PROMPT: str = _PROMPTS["system_prompt"]
|
||||
SUMMARIZATION_USER_TEMPLATE: str = _PROMPTS["user_template"]
|
||||
|
||||
Reference in New Issue
Block a user