consts.py

# =============================
# 资料存储数据库配置
# =============================
VEC_DB_HOST = '192.168.10.189'
VEC_DB_DBNAME = 'lae'
VEC_DB_USER = 'postgres'
VEC_DB_PASSWORD = '111111'
VEC_DB_PORT = '5433'

# =============================
# 聊天相关数据库配置
# =============================
CHAT_DB_HOST = '192.168.10.189'
CHAT_DB_DBNAME = 'lae'
CHAT_DB_USER = 'postgres'
CHAT_DB_PASSWORD = '111111'
CHAT_DB_PORT = '5433'

# =============================
# 向量化模型路径配置
# =============================
EMBEEDING_MODEL_PATH = 'D:\\work\\py\\LAE\\bge-large-zh-v1.5'

# =============================
# 重排序模型路径配置
# =============================
RERANK_MODEL_PATH = 'D:\\work\\py\\LAE\\bge-reranker-large'
# RERANK_MODEL_PATH = 'BAAI/bge-reranker-large'

# =============================
# 模型服务URL配置
# =============================
LLM_SERVER_URL = '192.168.10.102:8002'

# =============================
# FAISS相似性查找配置
# =============================
SIMILARITY_SHOW_NUMBER = 10
SIMILARITY_THRESHOLD = 0.8

# =============================
# FAISS向量库文件存储路径配置
# =============================
FAISS_STORE_PATH = 'D:\\work\\py\\LAE\\faiss'
INDEX_NAME = 'know'

# =============================
# 知识相关资料配置
# =============================
KNOWLEDGE_PATH = 'D:\\work\\py\\LAE\\testdoc'

# =============================
# gradio服务相关配置
# =============================
GR_SERVER_NAME = 'localhost'
GR_PORT = 8003

# =============================
# prompt 配置
# =============================
prompt1 = """'''
{context}
'''
请你根据上述已知资料回答下面的问题，问题如下：
{question}"""

prompt_enhancement_history_template = """{history}
上面是之前的对话，下面是可参考的内容,参考内容中如果和问题不符合，可以不用参考。
{context}
请结合上述内容回答以下问题，不要提无关内容：
{question}
"""
# =============================
# NLP_BERT模型路径配置
# =============================
NLP_BERT_PATH = 'C:\\Users\\15663\\AI\\models\\nlp_bert_document-segmentation_chinese-base'

# =============================
# ICON配置
# =============================
ICON_PATH = '/app/src/icon'