File size: 4,764 Bytes
d8d694f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
from dataclasses import dataclass
import os
@dataclass
class WebConfigs:
"""
项目所有的配置
"""
# ==================================================================
# LLM 模型配置
# ==================================================================
if os.getenv("USING_4BIT") == "true":
LLM_MODEL_NAME: str = "HinGwenWoong/streamer-sales-lelemiao-7b-4bit"
else:
LLM_MODEL_NAME: str = "HinGwenWoong/streamer-sales-lelemiao-7b"
SALES_NAME: str = "智能医导小助手" # 启动的角色名
LLM_MODEL_DIR: str = r"./weights/llm_weights/"
# ==================================================================
# 组件配置
# ==================================================================
ENABLE_RAG: bool = True # True 启用 RAG 检索增强，False 不启用
ENABLE_TTS: bool = True # True 启动 tts，False 不启用
ENABLE_DIGITAL_HUMAN: bool = True # True 启动 数字人，False 不启用
ENABLE_AGENT: bool = os.environ.get("ENABLE_AGENT", "true") == "true" # True 启动 Agent，False 不启用
ENABLE_ASR: bool = os.environ.get("ENABLE_ASR", "true") == "true" # True 启动 语音转文字，False 不启用
DISABLE_UPLOAD: bool = os.getenv("DISABLE_UPLOAD") == "true"
CACHE_MAX_ENTRY_COUNT: float = float(
os.environ.get("KV_CACHE", 0.1)
) # KV cache 占比，如果部署出现 OOM 降低这个配置，反之可以加大
# ==================================================================
# 页面配置
# ==================================================================
PRODUCT_IMAGE_HEIGHT: int = 400 # 商品图片高度
EACH_CARD_OFFSET: int = 100 # 每个商品卡片比图片高度多出的距离
EACH_ROW_COL: int = 2 # 商品页显示多少列
# 定义用户和机器人头像路径
USER_AVATOR: str = "./assets/user.png"
ROBOT_AVATOR: str = "./assets/logo.png"
# ==================================================================
# 商品配置
# ==================================================================
PRODUCT_INSTRUCTION_DIR: str = r"./product_info/instructions"
PRODUCT_IMAGES_DIR: str = r"./product_info/images"
PRODUCT_INFO_YAML_PATH: str = r"./product_info/product_info.yaml"
PRODUCT_INFO_YAML_BACKUP_PATH: str = PRODUCT_INFO_YAML_PATH + ".bk"
# ==================================================================
# 配置文件路径
# ==================================================================
CONVERSATION_CFG_YAML_PATH: str = r"./configs/conversation_cfg.yaml"
# ==================================================================
# RAG 配置
# ==================================================================
RAG_CONFIG_PATH: str = r"./configs/rag_config.yaml"
RAG_VECTOR_DB_DIR: str = r"./work_dirs/instruction_db"
PRODUCT_INSTRUCTION_DIR_GEN_DB_TMP: str = r"./work_dirs/instructions_gen_db_tmp"
RAG_MODEL_DIR: str = r"./weights/rag_weights/"
# ==================================================================
# TTS 配置
# ==================================================================
TTS_WAV_GEN_PATH: str = r"./work_dirs/tts_wavs"
TTS_MODEL_DIR: str = r"./weights/gpt_sovits_weights/"
# ==================================================================
# 数字人 配置
# ==================================================================
DIGITAL_HUMAN_GEN_PATH: str = r"./work_dirs/digital_human"
DIGITAL_HUMAN_MODEL_DIR: str = r"./weights/digital_human_weights/"
DIGITAL_HUMAN_BBOX_SHIFT: int = 0
DIGITAL_HUMAN_VIDEO_PATH: str = r"./doc/digital_human/digital_human_video.mp4"
DIGITAL_HUMAN_FPS: str = 25
# ==================================================================
# Agent 配置
# ==================================================================
AGENT_WEATHER_API_KEY: str | None = os.environ.get("WEATHER_API_KEY", None) # 天气 API Key
AGENT_DELIVERY_TIME_API_KEY: str | None = os.environ.get("DELIVERY_TIME_API_KEY", None) # 快递查询 API Key
# ==================================================================
# ASR 配置
# ==================================================================
ASR_WAV_SAVE_PATH: str = r"./work_dirs/asr_wavs"
ASR_MODEL_DIR: str = r"./weights/asr_weights/"
# 实例化
WEB_CONFIGS = WebConfigs()
|