feat: admin model info CRUD (provider/price/context/vision) (v0.6.4)

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "aitokenpool"
-version = "0.6.3"
+version = "0.6.4"
 edition = "2021"
 description = "AI Token 共享池 — 企业 key 池 + 公共共享市场"
 license = "MIT"

README.md

@@ -35,6 +35,7 @@ AITokenPool 是一个开源的 **AI Token 共享平台**：企业版（内部 ke
 - ✅ **v0.6.1**：**首次启动自动创建初始管理员（rant 2026-08-19T14:35:05）**——空库启动时创建 `admin@aitokenpool.local` + 随机 16 位密码（打印到启动日志，仅首次）+ quotas 账户（balance=0），幂等不重复；新增 `POST /api/auth/change-password` 改密端点（旧密码校验 + argon2 更新）；不再需要手工插库
 - ✅ **v0.6.2**：**用户自助注册 + 邮箱验证（rant 2026-08-19T14:36:19 方案 B）**——`POST /api/auth/register` + `verify` + `resend-code`；6 位数字验证码（10 分钟有效、5 次错误失效、60 秒重发限频）；未验证邮箱不可登录（403）；登录页注册表单 + 验证码页（中英 i18n）；SMTP 发信（`[mail]` 配置，未配置时 dev 模式验证码打日志/响应）
 - ✅ **v0.6.3**：**接入方式 URL 配置化（rant 2026-08-19T20:37:37）**——设置页「接入方式」端点不再硬编码域名：新增 `[server].public_url` 配置（缺省 `http://localhost:8080`）+ `GET /api/config` 下发；前端从配置拼接 `{public_url}/v1`、`{public_url}/anthropic`，取不到配置时回退同源 origin
+- ✅ **v0.6.4**：**管理员模型信息 CRUD（rant 2026-08-19T20:40:29）**——models 表补 context_length / max_output / vision / cache_hit_input_per_m（迁移 v7，幂等）+ seed 从 models.example.json 写入；新增 `GET|POST /api/admin/models` + `PATCH|DELETE /api/admin/models/:id`（admin 权限，唯一冲突 409，删除后按 0 计费）；管理视图「模型管理」tab（搜索/新增/编辑/删除，行内表单 + 二次确认，中英 i18n）；`GET /api/models` 市场列表补新字段（读图/上下文真实值）
 
 `ui/` 已由纯静态原型升级为**对接真实 API**（登录、钱包、市场、共享、交易、设置、管理、运营全部真实数据），由后端 `ServeDir` 静态托管，无需单独部署前端。
 
@@ -109,14 +110,15 @@ open http://localhost:8080/                        # 浏览器访问
 
 ## API 端点（Bearer 认证）
 
-- `GET /healthz` → `{"status":"ok","version":"0.6.3"}`
+- `GET /healthz` → `{"status":"ok","version":"0.6.4"}`
 - `POST /api/auth/login` → `{api_key}`；`POST /api/auth/change-password`（改密）；`POST /api/auth/register|verify|resend-code`（注册+邮箱验证）；`GET /api/me` → `{id,email,name,role}`；`GET /api/config` → `{public_url}`（接入端点 base，rant 2026-08-19T20:37:37）
 - `POST|GET /api/api-keys`（key 脱敏 `atk_live_****xxxx`）；`DELETE /api/api-keys/:id`（撤销）
 - `POST /v1/chat/completions` / `POST /anthropic/v1/messages` / `POST /v1/responses`（网关，三协议互转，非流式 + 流式 SSE 跨协议转换）；`GET /v1/models`（OpenAI 兼容模型列表，认证可选）
-- `GET /api/models`（模型市场）
+- `GET /api/models`（模型市场，含 context_length / max_output / vision / cache_hit_input_per_m）
 - `POST|GET /api/sharings` + `PATCH /api/sharings/:id`（上架 / 列表 / 暂停 / 恢复 / 删除）
 - `GET /api/wallet` / `GET /api/transactions?type=` / `GET /api/dashboard`（钱包 / 交易 / 仪表盘）
 - 管理员：`POST /api/admin/credits` / `GET /api/admin/users` + `PATCH /api/admin/users/:id` / `GET /api/admin/usage`
+- 模型管理（rant 2026-08-19T20:40:29）：`GET|POST /api/admin/models`（列表 / 新增）+ `PATCH|DELETE /api/admin/models/:id`（更新 / 删除；删除后该 model 调用按 0 计费）
 - 部门：`GET|POST /api/admin/departments` + `PATCH|DELETE /api/admin/departments/:id`
 - 加额：`POST|GET /api/raise-requests` + `POST /api/admin/raise-requests/:id/approve|reject`
 - 运营者：`GET /api/ops/runtime` / `POST /api/ops/credits` / `GET /api/ops/users`

src/dao.rs

@@ -158,6 +158,7 @@ pub fn get_model_price(
 pub fn list_models_with_availability(conn: &Connection) -> Result<Vec<serde_json::Value>> {
     let mut stmt = conn.prepare(
         "SELECT m.provider, m.model, m.currency, m.input_per_m, m.output_per_m, m.context_window, \
+                m.context_length, m.max_output, m.vision, m.cache_hit_input_per_m, \
                 (SELECT COUNT(*) FROM keys k WHERE k.model = m.model AND k.status = 'on') AS avail \
          FROM models m ORDER BY m.provider, m.model",
     )?;
@@ -169,7 +170,40 @@ pub fn list_models_with_availability(conn: &Connection) -> Result<Vec<serde_json
             "input_per_m": r.get::<_, f64>(3)?,
             "output_per_m": r.get::<_, f64>(4)?,
             "context_window": r.get::<_, i64>(5)?,
-            "available_keys": r.get::<_, i64>(6)?,
+            "context_length": r.get::<_, i64>(6)?,
+            "max_output": r.get::<_, i64>(7)?,
+            "vision": r.get::<_, i64>(8)?,
+            "cache_hit_input_per_m": r.get::<_, f64>(9)?,
+            "available_keys": r.get::<_, i64>(10)?,
+        }))
+    })?;
+    let mut out = Vec::new();
+    for r in rows {
+        out.push(r?);
+    }
+    Ok(out)
+}
+
+/// 管理员模型列表（rant 2026-08-19T20:40:29）：全部字段 + id，供 /api/admin/models
+pub fn list_all_models(conn: &Connection) -> Result<Vec<serde_json::Value>> {
+    let mut stmt = conn.prepare(
+        "SELECT id, provider, model, currency, input_per_m, output_per_m, \
+                context_length, max_output, vision, cache_hit_input_per_m, updated_at \
+         FROM models ORDER BY provider, model",
+    )?;
+    let rows = stmt.query_map([], |r| {
+        Ok(serde_json::json!({
+            "id": r.get::<_, i64>(0)?,
+            "provider": r.get::<_, String>(1)?,
+            "model": r.get::<_, String>(2)?,
+            "currency": r.get::<_, String>(3)?,
+            "input_per_m": r.get::<_, f64>(4)?,
+            "output_per_m": r.get::<_, f64>(5)?,
+            "context_length": r.get::<_, i64>(6)?,
+            "max_output": r.get::<_, i64>(7)?,
+            "vision": r.get::<_, i64>(8)?,
+            "cache_hit_input_per_m": r.get::<_, f64>(9)?,
+            "updated_at": r.get::<_, String>(10)?,
         }))
     })?;
     let mut out = Vec::new();

src/db.rs

@@ -10,7 +10,7 @@
 use anyhow::{Context, Result};
 use rusqlite::Connection;
 
-pub const SCHEMA_VERSION: i64 = 6;
+pub const SCHEMA_VERSION: i64 = 7;
 
 /// 打开（或创建）数据库并执行幂等迁移（生产标准：空库只建表，不种任何假数据）
 pub fn open(path: &str) -> Result<Connection> {
@@ -214,6 +214,33 @@ pub fn migrate(conn: &Connection) -> Result<()> {
             created_at TEXT NOT NULL DEFAULT (datetime('now'))
         );",
     )?;
+    // v7（rant 2026-08-19T20:40:29）：管理员模型信息 CRUD——
+    // models 补 context_length / max_output / vision / cache_hit_input_per_m
+    //（context_window 为 OpenAI 兼容 /v1/models 字段，与 context_length 并列保留）
+    ensure_column(
+        conn,
+        "models",
+        "context_length",
+        "context_length INTEGER NOT NULL DEFAULT 0",
+    )?;
+    ensure_column(
+        conn,
+        "models",
+        "max_output",
+        "max_output INTEGER NOT NULL DEFAULT 0",
+    )?;
+    ensure_column(
+        conn,
+        "models",
+        "vision",
+        "vision INTEGER NOT NULL DEFAULT 0",
+    )?;
+    ensure_column(
+        conn,
+        "models",
+        "cache_hit_input_per_m",
+        "cache_hit_input_per_m REAL NOT NULL DEFAULT 0",
+    )?;
     // schema_version：INSERT OR REPLACE 保证幂等
     let v: i64 = conn
         .query_row("SELECT version FROM schema_version", [], |r| r.get(0))
@@ -422,21 +449,29 @@ pub fn seed_models(conn: &Connection, cfg: &crate::config::Config) -> Result<()>
                 )
             });
         conn.execute(
-            "INSERT INTO models (provider, model, currency, input_per_m, output_per_m, context_window, updated_at) \
-             VALUES (?1, ?2, ?3, ?4, ?5, ?6, datetime('now')) \
+            "INSERT INTO models (provider, model, currency, input_per_m, output_per_m, context_window, context_length, max_output, vision, cache_hit_input_per_m, updated_at) \
+             VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7, ?8, ?9, ?10, datetime('now')) \
              ON CONFLICT(provider, model) DO UPDATE SET \
                currency = excluded.currency, \
                input_per_m = excluded.input_per_m, \
                output_per_m = excluded.output_per_m, \
                context_window = excluded.context_window, \
+               context_length = excluded.context_length, \
+               max_output = excluded.max_output, \
+               vision = excluded.vision, \
+               cache_hit_input_per_m = excluded.cache_hit_input_per_m, \
                updated_at = datetime('now')",
             rusqlite::params![
                 provider,
                 model,
                 currency,
                 input,
                 output,
-                m["context_length"].as_i64().unwrap_or(1048576)
+                m["context_length"].as_i64().unwrap_or(1048576),
+                m["context_length"].as_i64().unwrap_or(0),
+                m["max_output"].as_i64().unwrap_or(0),
+                m["vision"].as_i64().unwrap_or(0),
+                m["cache_hit_input_per_m"].as_f64().unwrap_or(0.0),
             ],
         )?;
         n += 1;