mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-19 21:37:20 +08:00
5d422ca9a1
改变 Embeddings 模型改为使用框架 API，不再手动加载，删除自定义 Embeddings Keyword 代码 修改依赖文件，移除 torch transformers 等重依赖 暂时移出对 loom 的集成 后续： 1、优化目录结构 2、检查合并中有无被覆盖的 0.2.10 内容
48 lines
1.3 KiB
Plaintext
48 lines
1.3 KiB
Plaintext
import logging
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import langchain
|
|
|
|
|
|
# 是否显示详细日志
|
|
log_verbose = False
|
|
langchain.verbose = False
|
|
|
|
# 通常情况下不需要更改以下内容
|
|
|
|
# 用户数据根目录
|
|
DATA_PATH = (Path(__file__).absolute().parent.parent) # / "data")
|
|
if not os.path.exists(DATA_PATH):
|
|
os.mkdir(DATA_PATH)
|
|
|
|
# nltk 模型存储路径
|
|
NLTK_DATA_PATH = os.path.join(DATA_PATH, "nltk_data")
|
|
import nltk
|
|
nltk.data.path = [NLTK_DATA_PATH] + nltk.data.path
|
|
|
|
# 日志格式
|
|
LOG_FORMAT = "%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s"
|
|
logger = logging.getLogger()
|
|
logger.setLevel(logging.INFO)
|
|
logging.basicConfig(format=LOG_FORMAT)
|
|
|
|
|
|
# 日志存储路径
|
|
LOG_PATH = os.path.join(DATA_PATH, "logs")
|
|
if not os.path.exists(LOG_PATH):
|
|
os.mkdir(LOG_PATH)
|
|
|
|
# 模型生成内容（图片、视频、音频等）保存位置
|
|
MEDIA_PATH = os.path.join(DATA_PATH, "media")
|
|
if not os.path.exists(MEDIA_PATH):
|
|
os.mkdir(MEDIA_PATH)
|
|
os.mkdir(os.path.join(MEDIA_PATH, "image"))
|
|
os.mkdir(os.path.join(MEDIA_PATH, "audio"))
|
|
os.mkdir(os.path.join(MEDIA_PATH, "video"))
|
|
|
|
# 临时文件目录，主要用于文件对话
|
|
BASE_TEMP_DIR = os.path.join(DATA_PATH, "temp")
|
|
if not os.path.exists(BASE_TEMP_DIR):
|
|
os.mkdir(BASE_TEMP_DIR)
|