M4 收口:神识用户面与计量闭环——BYOK/配额三段式/usage 落库/Anthropic+Gemini 适配器/神识阁
- 神识 Go:store 双实现(PG 自动建表+60s 就绪重试 / 内存回退)承载 usage_logs·byok_keys·quota_balances;BYOK 密钥库(AES-GCM 加密、API 脱敏、撤回即失效、主密钥未配置 fail-closed);配额三段式(预扣 max_tokens 钳位 1M→结算实际 token→全败幂等退款,HJ-2004);计量归属(IR user_id→usage_logs,/me/usage 用户维度聚合+配额总览,/admin/usage/sample 对账抽样);Anthropic/Gemini 出站适配器(SSE 解析、system 提升、用量映射);stub 端点补名修复计量空 endpoint - IR 契约:user_id/byok_key_id 双端同步(Go types.go + Py types.py) - 灵台:runs.owner_id(迁移 005,server_default 兼容存量)经 kernel 全链透传为 IR user_id;shenshi_client 扩展用户面方法(X-Hunjing-User-Id 代理,Echo 桩带内存实现);/me/byok-keys 登记·列表·撤回与 /me/usage 端点(4 用例) - 前端神识阁(/models 重建):偏好(生成参数表单)/自带密钥(登记加密保存、脱敏列表、撤回二次确认)/用量(聚合表+配额进度条)三区;me API 库+mock+5 用例;侧栏恢复入口 - 配置与部署:.env.example 增 SHENSHI_DATABASE_URL/SHENSHI_BYOK_MASTER_KEY/SHENSHI_USER_BUDGET_TOKENS;compose 神识容器接自有库(init-shenshi.sql 预建 hunjing_shenshi,与灵台不共库共表) - 证据:Go 34 用例(BYOK 独占/配额拒绝/结算退款/归属聚合)、后端 532、前端 210、BFF 4 全绿;跨进程 E2E 四场景(灵台 HttpShenshiClient→真 Go 神识:登记脱敏/BYOK 独占不可达整体失败/用量归属聚合/撤回即失效)通过;rewrite-plan M4 标记完成、能力矩阵神识行升已验证
This commit is contained in:
1 parent
462d570751
commit
144415247e
51 files changed
+3533
-90
No files matched your search
+39
-9
@@ -1,14 +1,19 @@
|
||||
// 神识(Shenshi)入口:魂镜模型网关,rewrite/1.0 M0 骨架。
|
||||
// 数据面 POST /v1/chat/completions(IR 入站,SSE 出站);健康面 /healthz /readyz。
|
||||
// 神识(Shenshi)入口:魂镜模型网关(rewrite/1.0)。
|
||||
// 数据面 POST /v1/chat/completions(IR 入站,SSE 出站);健康面 /healthz /readyz;
|
||||
// 计量面 /me/usage、对账抽样 /admin/usage/sample;BYOK 面 /me/byok-keys。
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
|
||||
"hunjing/shenshi/internal/byok"
|
||||
"hunjing/shenshi/internal/quota"
|
||||
"hunjing/shenshi/internal/route"
|
||||
"hunjing/shenshi/internal/server"
|
||||
"hunjing/shenshi/internal/store"
|
||||
)
|
||||
|
||||
func envOr(key, fallback string) string {
|
||||
@@ -19,13 +24,38 @@ func envOr(key, fallback string) string {
|
||||
}
|
||||
|
||||
func main() {
|
||||
port := envOr("SHENSHI_PORT", "8080")
|
||||
srv := server.New(
|
||||
server.Config{ServiceToken: os.Getenv("SHENSHI_SERVICE_TOKEN")},
|
||||
route.New(),
|
||||
)
|
||||
addr := ":" + port
|
||||
log.Printf("神识就绪:http://0.0.0.0%s(stub 端点,等待 M4 接入真实 Provider)", addr)
|
||||
st, err := store.Open(os.Getenv("SHENSHI_DATABASE_URL"))
|
||||
if err != nil {
|
||||
log.Fatalf("神识存储初始化失败: %v", err)
|
||||
}
|
||||
|
||||
cfg := server.Config{
|
||||
ServiceToken: os.Getenv("SHENSHI_SERVICE_TOKEN"),
|
||||
Store: st,
|
||||
}
|
||||
|
||||
// BYOK:主密钥未配置时整体禁用(fail-closed)
|
||||
if master := os.Getenv("SHENSHI_BYOK_MASTER_KEY"); master != "" {
|
||||
vault, err := byok.NewVault(master, st)
|
||||
if err != nil {
|
||||
log.Fatalf("BYOK 主密钥配置无效: %v", err)
|
||||
}
|
||||
cfg.Vault = vault
|
||||
log.Printf("BYOK 密钥库就绪")
|
||||
}
|
||||
|
||||
// 配额:用户级软配额(token 口径;0 = 不限制)
|
||||
var budget int64
|
||||
if raw := os.Getenv("SHENSHI_USER_BUDGET_TOKENS"); raw != "" {
|
||||
if _, err := fmt.Sscanf(raw, "%d", &budget); err != nil {
|
||||
log.Fatalf("SHENSHI_USER_BUDGET_TOKENS 须为整数: %v", err)
|
||||
}
|
||||
cfg.Quota = quota.New(st, budget)
|
||||
}
|
||||
|
||||
srv := server.New(cfg, route.New())
|
||||
addr := ":" + envOr("SHENSHI_PORT", "8080")
|
||||
log.Printf("神识就绪:http://0.0.0.0%s(端点目录 SHENSHI_ENDPOINTS,未配置时回声桩兜底)", addr)
|
||||
if err := http.ListenAndServe(addr, srv); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user