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}
|
|
}
|