优化ai初始化逻辑

internal/infra/ai/gemini.go

@@ -16,25 +16,25 @@ import (
 // geminiImpl 是 Gemini AI 服务的实现。
 type geminiImpl struct {
 	client *genai.GenerativeModel
-	cfg    *config.Gemini
+	cfg    config.Gemini
 }
 
 // NewGeminiAI 创建一个新的 geminiImpl 实例。
-func NewGeminiAI(ctx context.Context, cfg *config.AIConfig) (AI, error) {
+func NewGeminiAI(ctx context.Context, cfg config.Gemini) (AI, error) {
 	// 检查 API Key 是否存在
-	if cfg.Gemini.APIKey == "" {
+	if cfg.APIKey == "" {
 		return nil, fmt.Errorf("Gemini API Key 未配置")
 	}
 
 	// 创建 Gemini 客户端
-	genaiClient, err := genai.NewClient(ctx, option.WithAPIKey(cfg.Gemini.APIKey))
+	genaiClient, err := genai.NewClient(ctx, option.WithAPIKey(cfg.APIKey))
 	if err != nil {
 		return nil, fmt.Errorf("创建 Gemini 客户端失败: %w", err)
 	}
 
 	return &geminiImpl{
-		client: genaiClient.GenerativeModel(cfg.Gemini.ModelName),
-		cfg:    &cfg.Gemini,
+		client: genaiClient.GenerativeModel(cfg.ModelName),
+		cfg:    cfg,
 	}, nil
 }
 

internal/infra/ai/no_ai.go

@@ -0,0 +1,31 @@
+package ai
+
+import (
+	"context"
+	"errors"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
+)
+
+var NoneAIError = errors.New("当前没有配置AI, 暂不支持此功能")
+
+type NoneAI struct {
+	ctx context.Context
+}
+
+func NewNoneAI(ctx context.Context) AI {
+	return &NoneAI{
+		ctx: ctx,
+	}
+}
+
+func (n *NoneAI) GenerateReview(ctx context.Context, prompt string) (string, error) {
+	logger := logs.TraceLogger(ctx, n.ctx, "GenerateReview")
+	logger.Warnf("当前没有配置AI, 无法处理AI请求, 消息: %s", prompt)
+	return "", NoneAIError
+}
+
+func (n *NoneAI) AIModel() models.AIModel {
+	return models.AI_MODEL_NONE
+}

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 服务的配置

internal/infra/models/models.go

@@ -15,6 +15,7 @@ import (
 type AIModel string
 
 const (
+	AI_MODEL_NONE   AIModel = "None"
 	AI_MODEL_GEMINI AIModel = "Gemini"
 )