feat(cli,database): 更新数据库引擎创建模式

src/nyahome/cli/cli.py

@@ -7,10 +7,3 @@ console = Console()
 DATA_DIR = Path.cwd() / ".nyahome"
 ENV_PATH = DATA_DIR / ".env"
 LOGGING_YAML = DATA_DIR / "logging.yaml"
-
-db_driver_available = {
-    "sqlite": ["sqlite3"],
-    "mysql": ["pymysql"],
-    "postgresql": ["psycopg"],
-}
-db_type_allowlist = ["sqlite", "mysql", "postgresql"]

src/nyahome/cli/cli_aii.py

@@ -81,6 +81,14 @@ def add_model(
             help="该模型所属于的模型提供商 ID",
         ),
     ],
+    reasonable: Annotated[
+        bool,
+        typer.Option(
+            "--reasonable",
+            "-r",
+            help="支持思考",
+        ),
+    ] = False,
 ) -> None:
     """
     添加 AI 模型。在此之前需要先添加该模型的提供商。
@@ -92,7 +100,12 @@ def add_model(
     from nyahome.database import AiiModel, engine
 
     with Session(engine) as session:
-        am = AiiModel(model_name=model_name, max_context_length=max_context_length, aii_provider_id=provider_id)
+        am = AiiModel(
+            model_name=model_name,
+            max_context_length=max_context_length,
+            aii_provider_id=provider_id,
+            reasonable=reasonable,
+        )
         session.add(am)
         session.commit()
         session.refresh(am)

src/nyahome/cli/cli_check.py

@@ -4,7 +4,8 @@ from importlib.util import find_spec
 from pathlib import Path
 from typing import Mapping
 
-from nyahome.cli.cli import console, db_driver_available, db_type_allowlist
+from nyahome.cli.cli import console
+from nyahome.database.engine import db_driver_available, db_type_allowlist
 
 
 class CliWarning:
@@ -73,7 +74,7 @@ def check_database_type(environ: Mapping[str, str | None]) -> None:
         if not db_host:
             cw.warning("NYAHOME_DB_HOST 未设置，将使用 [cyan]localhost[/cyan] 作为默认值。")
         if not db_port:
-            cw.warning("NYAHOME_DB_PORT 未设置，将使用 [cyan]3006[/cyan] 作为默认值。")
+            cw.warning("NYAHOME_DB_PORT 未设置，将使用 [cyan]3306[/cyan] 作为默认值。")
         cw.info("自检未检查数据库状态是否可用。")
     else:
         cw.info("使用 sqlite 数据库，跳过数据库凭证检查。")

src/nyahome/cli/cli_env.py

@@ -42,8 +42,11 @@ def set_env(
 
     保存在 .nyahome 内的 .env 文件。
     """
-    set_key(ENV_PATH, f"NYAHOME_{key.upper()}", value)
-    console.print(f"[cyan]已设置环境变量 NYAHOME_{key}。[/cyan]")
+    key = key.upper()
+    if not key.startswith("NYAHOME_"):
+        key = f"NYAHOME_{key}"
+    set_key(ENV_PATH, key, value)
+    console.print(f"[cyan]已设置环境变量 {key}。[/cyan]")
 
 
 @env_app.command(name="unset")