freedak
f7a720204a
- 移除 GovAI, nomifun-tauri, 算力盒子 的 submodule 引用 - 添加所有子项目的完整源代码 - 保留原始 .git 为 .git.bak 备份
134 lines
3.2 KiB
Go
134 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
func main() {
|
|
// 加载环境变量
|
|
_ = godotenv.Load("../../.env")
|
|
|
|
dbURL := os.Getenv("DATABASE_URL")
|
|
if dbURL == "" {
|
|
dbURL = "postgres://freedak:@localhost:5432/govai_portal?sslmode=disable"
|
|
}
|
|
|
|
pool, err := pgxpool.New(context.Background(), dbURL)
|
|
if err != nil {
|
|
log.Fatalf("无法连接数据库: %v", err)
|
|
}
|
|
defer pool.Close()
|
|
|
|
fmt.Println("=== 数据库中的模型提供商配置 ===")
|
|
|
|
rows, err := pool.Query(context.Background(), `
|
|
SELECT id, name, base_url,
|
|
SUBSTRING(api_key_encrypted, 1, 20) as api_key_prefix,
|
|
models, is_active, priority, config
|
|
FROM model_providers
|
|
ORDER BY priority DESC, created_at
|
|
`)
|
|
if err != nil {
|
|
log.Fatalf("查询失败: %v", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
count := 0
|
|
for rows.Next() {
|
|
var id, name, baseURL, apiKeyPrefix string
|
|
var modelsJSON, configJSON []byte
|
|
var isActive bool
|
|
var priority int
|
|
|
|
err := rows.Scan(&id, &name, &baseURL, &apiKeyPrefix, &modelsJSON, &isActive, &priority, &configJSON)
|
|
if err != nil {
|
|
log.Printf("扫描行失败: %v", err)
|
|
continue
|
|
}
|
|
|
|
count++
|
|
fmt.Printf("【提供商 %d】\n", count)
|
|
fmt.Printf("ID: %s\n", id)
|
|
fmt.Printf("名称: %s\n", name)
|
|
fmt.Printf("Base URL: %s\n", baseURL)
|
|
fmt.Printf("API Key: %s...\n", apiKeyPrefix)
|
|
fmt.Printf("激活状态: %v\n", isActive)
|
|
fmt.Printf("优先级: %d\n", priority)
|
|
|
|
// 解析 models
|
|
var models []map[string]interface{}
|
|
if err := json.Unmarshal(modelsJSON, &models); err == nil {
|
|
fmt.Printf("可用模型: ")
|
|
for i, m := range models {
|
|
if i > 0 {
|
|
fmt.Print(", ")
|
|
}
|
|
fmt.Printf("%v", m["id"])
|
|
}
|
|
fmt.Println()
|
|
}
|
|
|
|
// 解析 config
|
|
var config map[string]interface{}
|
|
if err := json.Unmarshal(configJSON, &config); err == nil {
|
|
fmt.Printf("Provider 类型: %v\n", config["provider"])
|
|
fmt.Printf("默认模型: %v\n", config["default_model"])
|
|
}
|
|
|
|
fmt.Println()
|
|
}
|
|
|
|
if count == 0 {
|
|
fmt.Println("❌ 数据库中没有任何模型提供商配置")
|
|
fmt.Println("提示: 运行种子数据脚本来初始化配置")
|
|
} else {
|
|
fmt.Printf("✅ 共找到 %d 个提供商配置\n", count)
|
|
}
|
|
|
|
// 查询应用配置中的模型使用情况
|
|
fmt.Println("\n=== 应用配置中的模型使用情况 ===")
|
|
|
|
appRows, err := pool.Query(context.Background(), `
|
|
SELECT name,
|
|
COALESCE(app_config->>'model', '未配置') as model,
|
|
status
|
|
FROM applications
|
|
WHERE status = 'approved'
|
|
ORDER BY created_at DESC
|
|
LIMIT 10
|
|
`)
|
|
if err != nil {
|
|
log.Printf("查询应用失败: %v", err)
|
|
return
|
|
}
|
|
defer appRows.Close()
|
|
|
|
appCount := 0
|
|
modelStats := make(map[string]int)
|
|
|
|
for appRows.Next() {
|
|
var name, model, status string
|
|
if err := appRows.Scan(&name, &model, &status); err != nil {
|
|
continue
|
|
}
|
|
appCount++
|
|
modelStats[model]++
|
|
fmt.Printf("应用: %-30s | 模型: %-15s | 状态: %s\n", name, model, status)
|
|
}
|
|
|
|
if appCount > 0 {
|
|
fmt.Printf("\n✅ 共查询到 %d 个已批准的应用\n\n", appCount)
|
|
fmt.Println("模型使用统计:")
|
|
for model, count := range modelStats {
|
|
fmt.Printf(" - %-15s: %d 个应用\n", model, count)
|
|
}
|
|
}
|
|
}
|