- core/module: 命令模板、门控(feature flag/confirm/dry-run)、ecode 错误标签、ResultFormatter - runbook executor: http.get/http.post/assert.json(expect_success 严格断言)、目标主机管理、任务落库、最终 stdout 直接回传 - cf 模块: /cf dns list 支持域名自动解析 zone_id;dns list/zones/workers list 格式化回传;API success:false 判真失败 - 全部 CF/CPA runbook 接入严格断言,杜绝假阳性 success - 新增 cpa/cfdav/mail/sub 模块与 runbooks;QQ 订阅到期提醒调度 - Web 后台: 模块设置(CF/CPA/cfdav/AI)、OPS 任务、渠道管理、审计页
41 lines
814 B
Go
41 lines
814 B
Go
package ai
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"ops-assistant/models"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func LoadClient(db *gorm.DB) *Client {
|
|
if db == nil {
|
|
return nil
|
|
}
|
|
get := func(key string) string {
|
|
var sset models.AppSetting
|
|
if err := db.Where("key = ?", key).First(&sset).Error; err == nil {
|
|
return strings.TrimSpace(sset.Value)
|
|
}
|
|
return ""
|
|
}
|
|
if strings.ToLower(get("ai_enabled")) != "true" {
|
|
return nil
|
|
}
|
|
baseURL := get("ai_base_url")
|
|
apiKey := get("ai_api_key")
|
|
model := get("ai_model")
|
|
to := 15
|
|
if v := get("ai_timeout_seconds"); v != "" {
|
|
if n, err := strconv.Atoi(v); err == nil && n > 0 {
|
|
to = n
|
|
}
|
|
}
|
|
if baseURL == "" || apiKey == "" || model == "" {
|
|
return nil
|
|
}
|
|
return &Client{BaseURL: baseURL, APIKey: apiKey, Model: model, Timeout: time.Duration(to) * time.Second}
|
|
}
|