实现ai

internal/core/component_initializers.go

@@ -15,6 +15,7 @@ import (
 	"git.huangwc.com/pig/pig-farm-controller/internal/domain/plan"
 	"git.huangwc.com/pig/pig-farm-controller/internal/domain/recipe"
 	"git.huangwc.com/pig/pig-farm-controller/internal/domain/task"
+	infra_ai "git.huangwc.com/pig/pig-farm-controller/internal/infra/ai"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/config"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/database"
 	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
@@ -34,6 +35,7 @@ type Infrastructure struct {
 	storage        database.Storage
 	repos          *Repositories
 	lora           *LoraComponents
+	aiManager      infra_ai.AI
 	tokenGenerator token.Generator
 }
 
@@ -53,10 +55,17 @@ func initInfrastructure(ctx context.Context, cfg *config.Config) (*Infrastructur
 
 	tokenGenerator := token.NewTokenGenerator([]byte(cfg.App.JWTSecret))
 
+	// 初始化 AI
+	aiManager, err := infra_ai.NewGeminiAI(logs.AddCompName(ctx, "GeminiAI"), &cfg.AI)
+	if err != nil {
+		return nil, fmt.Errorf("初始化 AI 管理器失败: %w", err)
+	}
+
 	return &Infrastructure{
 		storage:        storage,
 		repos:          repos,
 		lora:           lora,
+		aiManager:      aiManager,
 		tokenGenerator: tokenGenerator,
 	}, nil
 }
@@ -238,6 +247,8 @@ func initDomainServices(ctx context.Context, cfg *config.Config, infra *Infrastr
 		pigTypeService,
 		recipeCoreService,
 		recipeGenerateManager,
+		infra.repos.recipeRepo,
+		infra.aiManager,
 	)
 
 	return &DomainServices{