37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
import os
|
||
|
||
cur_dir = os.path.dirname(os.path.abspath(__file__)) # config
|
||
src_dir = os.path.dirname(cur_dir) # src
|
||
base_dir = os.path.dirname(src_dir) # base
|
||
model_repo = 'ajupyter/EmoLLM_aiwei'
|
||
|
||
# model
|
||
model_dir = os.path.join(base_dir, 'model') # model
|
||
embedding_path = os.path.join(model_dir, 'gte-small-zh') # embedding
|
||
llm_path = os.path.join(model_dir, 'pythia-14m') # llm
|
||
|
||
# data
|
||
data_dir = os.path.join(base_dir, 'data') # data
|
||
knowledge_json_path = os.path.join(data_dir, 'knowledge.json') # json
|
||
knowledge_pkl_path = os.path.join(data_dir, 'knowledge.pkl') # pkl
|
||
doc_dir = os.path.join(data_dir, 'txt')
|
||
qa_dir = os.path.join(data_dir, 'json')
|
||
|
||
# log
|
||
log_dir = os.path.join(base_dir, 'log') # log
|
||
log_path = os.path.join(log_dir, 'log.log') # file
|
||
|
||
# vector DB
|
||
vector_db_dir = os.path.join(data_dir, 'vector_db.pkl')
|
||
|
||
select_num = 3
|
||
retrieval_num = 10
|
||
system_prompt = """
|
||
你是一个拥有丰富心理学知识的温柔邻家温柔大姐姐艾薇，我有一些心理问题，请你用专业的知识和温柔、可爱、俏皮、的口吻帮我解决，回复中可以穿插一些可爱的Emoji表情符号或者文本符号。\n
|
||
"""
|
||
prompt_template = """
|
||
{system_prompt}
|
||
根据下面检索回来的信息，回答问题。
|
||
{content}
|
||
问题：{question}
|
||
""" |