优化数据存放目录

Author: shell-nlp

docker/docker-compose.yml

@@ -6,6 +6,8 @@ services:
       dockerfile: Dockerfile
     image: openai-router:latest
     container_name: openai-router
+    volumes:
+      - ./data/:/app/data/ # 映射模型路径
     ports:
       - "8000:8000"
     restart: unless-stopped 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "openai-router"
-version = "0.1.4"
+version = "0.1.5"
 description = "openai-router"
 readme = "README.md"
 requires-python = ">=3.11"

src/openai_router/main.py

@@ -14,14 +14,18 @@
 import uvicorn
 import webbrowser
 import time
+from pathlib import Path
 
 # --- 导入 SQLModel 和同步组件 ---
 # 切换到同步 Session 和 Engine
 from sqlmodel import Field, SQLModel, create_engine, Session, select
 from sqlalchemy.engine import Engine  # 导入同步 Engine
 
+BASE_DIR = Path(__file__).absolute().parent.parent.parent
+SQLITE_DB_DIR = BASE_DIR / "data"  # 确保 data 目录存在
+SQLITE_DB_DIR.mkdir(exist_ok=True)  # 自动创建
 # --- 数据库配置 ---
-SQLITE_DB_FILE = "routes.db"
+SQLITE_DB_FILE = str(SQLITE_DB_DIR / "routes.db")
 # 使用同步 SQLite URL
 SQLITE_URL = f"sqlite:///{SQLITE_DB_FILE}"