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

# =============================
# 聊天相关数据库配置
# =============================
CHAT_DB_HOST = 'localhost'
陈正乐 committed
14 15 16 17
CHAT_DB_DBNAME = 'laechat'
CHAT_DB_USER = 'postgres'
CHAT_DB_PASSWORD = 'chenzl'
CHAT_DB_PORT = '5432'
18 19 20 21

# =============================
# 向量化模型路径配置
# =============================
22
EMBEEDING_MODEL_PATH = 'C:\\Users\\15663\\AI\\models\\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向量库文件存储路径配置
# =============================
38
FAISS_STORE_PATH = 'C:\\Users\\15663\\code\\dkjj-llm\\LAE\\faiss'
39 40 41 42 43
INDEX_NAME = 'know'

# =============================
# 知识相关资料配置
# =============================
44
KNOWLEDGE_PATH = 'C:\\Users\\15663\\Desktop\\低空经济数据库'
陈正乐 committed
45 46 47 48 49

# =============================
# gradio服务相关配置
# =============================
GR_SERVER_NAME = 'localhost'
50 51 52 53 54 55 56 57 58
GR_PORT = 8888

# =============================
# 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 69

# =============================
# ICON配置
# =============================
ICON_PATH = 'C:\\Users\\15663\\code\\dkjj-llm\\LAE\\LAE\\test\\icon'