Spaces:
Running
Running
Add DATASET_ANNOTATIONS_PATH import and update chat evaluator initialization to use annotations directory
Browse files
app.py
CHANGED
|
@@ -35,6 +35,7 @@ from config.settings import (
|
|
| 35 |
DATASET_ID,
|
| 36 |
DATASET_PREFERENCES_PATH,
|
| 37 |
DATASET_VECTOR_STORE_PATH,
|
|
|
|
| 38 |
DEFAULT_MODEL,
|
| 39 |
EMBEDDING_MODEL,
|
| 40 |
HF_TOKEN,
|
|
@@ -974,14 +975,12 @@ def initialize_chat_evaluator():
|
|
| 974 |
try:
|
| 975 |
evaluator = ChatEvaluator(
|
| 976 |
hf_token=HF_TOKEN,
|
| 977 |
-
dataset_id=DATASET_ID
|
| 978 |
-
#chat_history_path=DATASET_CHAT_HISTORY_PATH,
|
| 979 |
-
annotations_dir=os.path.join(DATASET_CHAT_HISTORY_PATH, 'evaluations')
|
| 980 |
)
|
| 981 |
|
| 982 |
# Проверим наличие директорий
|
| 983 |
os.makedirs(DATASET_CHAT_HISTORY_PATH, exist_ok=True)
|
| 984 |
-
os.makedirs(os.path.join(
|
| 985 |
|
| 986 |
logger.debug(f"Chat history path: {DATASET_CHAT_HISTORY_PATH}")
|
| 987 |
logger.debug(f"Number of chat files: {len(os.listdir(DATASET_CHAT_HISTORY_PATH))}")
|
|
|
|
| 35 |
DATASET_ID,
|
| 36 |
DATASET_PREFERENCES_PATH,
|
| 37 |
DATASET_VECTOR_STORE_PATH,
|
| 38 |
+
DATASET_ANNOTATIONS_PATH, # Добавляем импорт
|
| 39 |
DEFAULT_MODEL,
|
| 40 |
EMBEDDING_MODEL,
|
| 41 |
HF_TOKEN,
|
|
|
|
| 975 |
try:
|
| 976 |
evaluator = ChatEvaluator(
|
| 977 |
hf_token=HF_TOKEN,
|
| 978 |
+
dataset_id=DATASET_ID
|
|
|
|
|
|
|
| 979 |
)
|
| 980 |
|
| 981 |
# Проверим наличие директорий
|
| 982 |
os.makedirs(DATASET_CHAT_HISTORY_PATH, exist_ok=True)
|
| 983 |
+
os.makedirs(os.path.join(DATASET_ANNOTATIONS_PATH), exist_ok=True)
|
| 984 |
|
| 985 |
logger.debug(f"Chat history path: {DATASET_CHAT_HISTORY_PATH}")
|
| 986 |
logger.debug(f"Number of chat files: {len(os.listdir(DATASET_CHAT_HISTORY_PATH))}")
|