2024-03-07 17:56:07 +08:00
|
|
|
import os
|
|
|
|
|
|
|
|
"""
|
|
|
|
文件夹路径
|
|
|
|
"""
|
2024-03-07 23:59:10 +08:00
|
|
|
cur_dir = os.path.dirname(os.path.abspath(__file__)) # config
|
|
|
|
base_dir = os.path.dirname(cur_dir) # base
|
2024-03-07 17:56:07 +08:00
|
|
|
|
|
|
|
# model
|
2024-03-07 23:59:10 +08:00
|
|
|
model_dir = os.path.join(base_dir, 'model') # model
|
2024-03-07 17:56:07 +08:00
|
|
|
|
|
|
|
# data
|
2024-03-16 13:12:15 +08:00
|
|
|
data_dir = os.path.join(base_dir, 'data') # /Users/wangyoufang/Downloads/EmoLLM/scripts/qa_generation/data
|
|
|
|
clean_dir = os.path.join(data_dir, 'cleaned')
|
|
|
|
judge_dir = os.path.join(data_dir, '数据整合')
|
2024-03-07 23:59:10 +08:00
|
|
|
result_dir = os.path.join(data_dir, 'generated') # result
|
2024-03-07 17:56:07 +08:00
|
|
|
|
|
|
|
# log
|
2024-03-07 23:59:10 +08:00
|
|
|
log_dir = os.path.join(base_dir, 'log') # log
|
|
|
|
log_file_path = os.path.join(log_dir, 'log.log') # file
|
2024-03-07 17:56:07 +08:00
|
|
|
|
|
|
|
# system prompt
|
2024-03-16 13:12:15 +08:00
|
|
|
# Prompt内容
|
2024-03-07 23:59:10 +08:00
|
|
|
system_prompt_file_path = os.path.join(base_dir, 'system_prompt_v2.md') # system prompt
|
2024-03-16 13:12:15 +08:00
|
|
|
wash_prompt_file_path = os.path.join(base_dir, 'choose_prompt.md')
|
2024-03-07 17:56:07 +08:00
|
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
环境变量
|
|
|
|
"""
|
|
|
|
# api-keys
|
2024-03-08 14:25:34 +08:00
|
|
|
DASHSCOPE_API_KEY = ''
|
|
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
控制参数
|
|
|
|
"""
|
|
|
|
storage_interval = 10
|
|
|
|
window_size = 8
|
|
|
|
overlap_size = 2
|
2024-03-16 13:12:15 +08:00
|
|
|
multi_process_num = 1
|
|
|
|
|