refactor(backend): 启动环境初始化与资源项目化(零跨项目依赖)
- main.py:uv run main.py 一键启动——读取 backend/.env、TORCH_HOME/NLTK_DATA 指向项目内缓存、certifi SSL + NO_PROXY 网络直连、默认开启 s2s,启动横幅 - vendor/:补丁后 s2s 云化栈移入项目(backend/vendor/s2s-cloud),pyproject sources 改相对路径 - .gitignore:忽略 .torch-cache / nltk_data 缓存 - AGENTS.md:进程管理铁律(AI 不启动/重启服务,由用户操作)
This commit is contained in:
+70
-4
@@ -1,8 +1,74 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""开发入口:python main.py(或 uvicorn app.main:app)"""
|
||||
import uvicorn
|
||||
"""开发/部署入口:uv run main.py(或 python main.py)
|
||||
|
||||
启动前自动完成环境初始化(无需手动 export):
|
||||
1. 读取 backend/.env(阿里云密钥、端口、MQTT 等,见 app/config.py)
|
||||
2. 本地资源收进 backend/ 内,不依赖其他项目:
|
||||
TORCH_HOME -> backend/.torch-cache (silero VAD)
|
||||
NLTK_DATA -> backend/nltk_data (标点/分词数据)
|
||||
3. 网络直连:SSL_CERT_FILE(certifi)、NO_PROXY(绕过系统代理直达阿里云)
|
||||
4. 默认开启 s2s 实时语音栈(.env 设 DPM_S2S_ENABLED=0 可关闭)
|
||||
|
||||
对应 uvicorn 命令:uvicorn app.main:app --host 0.0.0.0 --port 10085
|
||||
"""
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
BACKEND_DIR = Path(__file__).resolve().parent
|
||||
|
||||
|
||||
def _load_dotenv(path: Path):
|
||||
"""极简 .env 解析:KEY=VALUE,支持 # 注释与引号(与 app/config.py 一致)"""
|
||||
if not path.exists():
|
||||
return
|
||||
for raw in path.read_text("utf-8").splitlines():
|
||||
line = raw.strip()
|
||||
if not line or line.startswith("#") or "=" not in line:
|
||||
continue
|
||||
key, _, value = line.partition("=")
|
||||
key = key.strip()
|
||||
value = value.strip().strip('"').strip("'")
|
||||
if key and key not in os.environ:
|
||||
os.environ[key] = value
|
||||
|
||||
|
||||
def prepare_env():
|
||||
"""进程环境初始化 —— 必须在 import app.config / torch 之前调用。"""
|
||||
# 1) 项目内 .env 优先(已有环境变量不覆盖)
|
||||
_load_dotenv(BACKEND_DIR / ".env")
|
||||
# 2) 本地模型/数据缓存全部收进 backend/(部署时随目录整体拷贝即可)
|
||||
os.environ.setdefault("TORCH_HOME", str(BACKEND_DIR / ".torch-cache"))
|
||||
os.environ.setdefault("NLTK_DATA", str(BACKEND_DIR / "nltk_data"))
|
||||
# 3) 网络直连(绕过 macOS 系统 SOCKS 代理;TLS 用 certifi 证书)
|
||||
os.environ.setdefault("NO_PROXY", "*")
|
||||
os.environ.setdefault("no_proxy", "*")
|
||||
try:
|
||||
import certifi
|
||||
os.environ.setdefault("SSL_CERT_FILE", certifi.where())
|
||||
except Exception:
|
||||
pass
|
||||
# 4) 默认开启 s2s 实时语音栈(.env 的 DPM_S2S_ENABLED=0 优先生效)
|
||||
os.environ.setdefault("DPM_S2S_ENABLED", "1")
|
||||
|
||||
|
||||
prepare_env()
|
||||
|
||||
import uvicorn # noqa: E402
|
||||
|
||||
from app.config import settings # noqa: E402
|
||||
|
||||
|
||||
def main():
|
||||
s2s = "on" if settings.S2S_ENABLED else "off"
|
||||
print("=" * 64)
|
||||
print(f" DPM 后端 : http://{settings.HOST}:{settings.PORT}")
|
||||
print(f" s2s 语音栈 : {s2s}" + (f" (ws://{settings.S2S_HOST}:{settings.S2S_PORT}/v1/realtime)" if settings.S2S_ENABLED else ""))
|
||||
print(f" 模型 : ASR={settings.S2S_STT_MODEL} LLM={settings.S2S_LLM_MODEL} TTS={settings.S2S_TTS_MODEL} ({settings.S2S_TTS_VOICE})")
|
||||
print(f" 本地资源 : TORCH_HOME={os.environ.get('TORCH_HOME')}")
|
||||
print(f" NLTK_DATA={os.environ.get('NLTK_DATA')}")
|
||||
print("=" * 64)
|
||||
uvicorn.run("app.main:app", host=settings.HOST, port=settings.PORT, reload=True)
|
||||
|
||||
from app.config import settings
|
||||
|
||||
if __name__ == "__main__":
|
||||
uvicorn.run("app.main:app", host=settings.HOST, port=settings.PORT, reload=True)
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user