Rulga commited on
Commit
e89a8a3
·
1 Parent(s): 74d5dc1

Add DATASET_ANNOTATIONS_PATH import and update chat evaluator initialization to use annotations directory

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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(DATASET_CHAT_HISTORY_PATH, 'evaluations'), exist_ok=True)
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))}")