File size: 772 Bytes
e2b8b61
 
 
 
 
 
 
 
 
edf8cae
 
e2b8b61
edf8cae
e2b8b61
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
from pathlib import Path
from src.model_registry import DEFAULT_MODEL_ALIAS


BASE_DIR = Path(__file__).resolve().parent.parent

DATA_DIR = BASE_DIR / "data"
RAW_DATA_PATH = DATA_DIR / "raw" / "dailydialog.json"

ENCODED_DIR = DATA_DIR / "full" / "encoded"
LOCALIZED_DIR = DATA_DIR / "full" / "localized"
REVIEWED_DIR = DATA_DIR / "reviewed"
DECODED_DIR = DATA_DIR / "full" / "decoded"
EVAL_DIR = DATA_DIR / "evaluation"

PROMPTS_DIR = BASE_DIR / "prompts"


# =========================
# MODEL SETTINGS
# =========================

DEFAULT_MODEL = DEFAULT_MODEL_ALIAS
TEMPERATURE = 0.2
MAX_TOKENS = 4096
MAX_TOKENS_THINKING = 16384  # Higher budget for models that output thinking text


# =========================
# PIPELINE
# =========================

MAX_RETRIES = 2