优化ai初始化逻辑

2025-12-02 16:34:14 +08:00
parent da2c296c05
commit 6764684fe7
7 changed files with 56 additions and 11 deletions
--- a/internal/infra/config/config.go
+++ b/internal/infra/config/config.go
@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"os"

+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
 	"gopkg.in/yaml.v2"
 )

@@ -236,7 +237,8 @@ type AlarmNotificationConfig struct {

 // AIConfig AI 服务配置
 type AIConfig struct {
-	Gemini Gemini `yaml:"gemini"`
+	Model  models.AIModel `yaml:"model"`
+	Gemini Gemini         `yaml:"gemini"`
 }

 // Gemini 代表 Gemini AI 服务的配置