consts.py 1.73 KB
Newer Older
1 2 3
# =============================
# 资料存储数据库配置
# =============================
陈正乐 committed
4
VEC_DB_HOST = '192.168.10.93'
陈正乐 committed
5 6
VEC_DB_DBNAME = 'lae'
VEC_DB_USER = 'postgres'
陈正乐 committed
7 8
VEC_DB_PASSWORD = '111111'
VEC_DB_PORT = '5433'
9 10 11 12

# =============================
# 聊天相关数据库配置
# =============================
陈正乐 committed
13 14
CHAT_DB_HOST = '192.168.10.93'
CHAT_DB_DBNAME = 'lae_chat'
陈正乐 committed
15
CHAT_DB_USER = 'postgres'
陈正乐 committed
16 17
CHAT_DB_PASSWORD = '111111'
CHAT_DB_PORT = '5433'
18 19 20 21

# =============================
# 向量化模型路径配置
# =============================
陈正乐 committed
22
EMBEEDING_MODEL_PATH = '/app/bge-large-zh-v1.5'
23 24 25 26

# =============================
# 模型服务URL配置
# =============================
27
LLM_SERVER_URL = '192.168.10.102:8002'
28 29 30 31

# =============================
# FAISS相似性查找配置
# =============================
32 33
SIMILARITY_SHOW_NUMBER = 5
SIMILARITY_THRESHOLD = 0.8
34 35 36 37

# =============================
# FAISS向量库文件存储路径配置
# =============================
陈正乐 committed
38
FAISS_STORE_PATH = '/app/faiss'
39 40 41 42 43
INDEX_NAME = 'know'

# =============================
# 知识相关资料配置
# =============================
陈正乐 committed
44
KNOWLEDGE_PATH = '/app/lae_data'
陈正乐 committed
45 46 47 48 49

# =============================
# gradio服务相关配置
# =============================
GR_SERVER_NAME = 'localhost'
陈正乐 committed
50
GR_PORT = 8003
51 52 53 54 55 56 57 58

# =============================
# prompt 配置
# =============================
prompt1 = """'''
{context}
'''
请你根据上述已知资料回答下面的问题,问题如下:
59 60 61 62 63 64
{question}"""

# =============================
# NLP_BERT模型路径配置
# =============================
NLP_BERT_PATH = 'C:\\Users\\15663\\AI\\models\\nlp_bert_document-segmentation_chinese-base'
陈正乐 committed
65 66 67 68

# =============================
# ICON配置
# =============================
陈正乐 committed
69
ICON_PATH = '/app/test/icon'