prise en charge des outils + weather

backend/cmd/dip.go

@@ -0,0 +1,105 @@
+package cmd
+
+import (
+	"time"
+
+	"trankilou.fr/lassistanoque/backend/internal/adapter/auth/password"
+	"trankilou.fr/lassistanoque/backend/internal/adapter/database"
+	"trankilou.fr/lassistanoque/backend/internal/adapter/file"
+	"trankilou.fr/lassistanoque/backend/internal/adapter/llm"
+	"trankilou.fr/lassistanoque/backend/internal/adapter/security"
+	"trankilou.fr/lassistanoque/backend/internal/adapter/tools/weather"
+	"trankilou.fr/lassistanoque/backend/internal/domain"
+	"trankilou.fr/lassistanoque/backend/internal/service/auth"
+	"trankilou.fr/lassistanoque/backend/internal/service/chat"
+	"trankilou.fr/lassistanoque/backend/internal/service/storage"
+)
+
+var db database.Database
+var store storage.StorageProvider
+var userRepository domain.UserRepository
+var providerRepository domain.ProviderRepository
+var agentRepository domain.AgentRepository
+var chatRepository domain.ChatRepository
+var settingsRepository domain.SettingsRepository
+var authenticators = make(map[string]auth.Authenticator)
+var tokenManager auth.TokenManager
+var llmengine domain.LLMEngine
+
+func init() {
+	var err error
+
+	db, err := database.GetDatabase()
+	if err != nil {
+		panic(err)
+	}
+
+	err = db.Migrate()
+	if err != nil {
+		panic(err)
+	}
+
+	// repositories
+	userRepository = db.UserRepository()
+	providerRepository = db.ProviderRepository()
+	agentRepository = db.AgentRepository()
+	chatRepository = db.ChatRepository()
+	settingsRepository = db.SettingsRepository()
+
+	// storage
+	store, err = file.GetStorageProvider(db.FileRepository())
+	if err != nil {
+		panic(err)
+	}
+
+	// authentication
+	tokenManager = security.NewJwtTokenManager(12*time.Hour, 7*24*time.Hour, "lassistanoque")
+	pwdAuth := password.NewPasswordAuthenticator(tokenManager, GetUserRepository())
+	authenticators["password"] = pwdAuth
+
+	// llm
+	llmengine = llm.NewAnyLLMEngine()
+
+	// llm tools
+	chat.RegisterTool(&weather.WeatherTool{})
+}
+
+func GetStorageProvider() storage.StorageProvider {
+	return store
+}
+
+func GetUserRepository() domain.UserRepository {
+	return userRepository
+}
+
+func GetProviderRepository() domain.ProviderRepository {
+	return providerRepository
+}
+
+func GetAgentRepository() domain.AgentRepository {
+	return agentRepository
+}
+
+func GetChatRepository() domain.ChatRepository {
+	return chatRepository
+}
+
+func GetSettingsRepository() domain.SettingsRepository {
+	return settingsRepository
+}
+
+func GetAuthenticators() map[string]auth.Authenticator {
+	return authenticators
+}
+
+func GetTokenManager() auth.TokenManager {
+	return tokenManager
+}
+
+func GetLLMEngine() domain.LLMEngine {
+	return llmengine
+}
+
+func Close() {
+	db.Close()
+}

backend/cmd/serve.go

@@ -11,11 +11,6 @@ import (
 	"time"
 
 	"github.com/spf13/cobra"
-	"trankilou.fr/lassistanoque/backend/internal/adapter/auth/password"
-	"trankilou.fr/lassistanoque/backend/internal/adapter/database"
-	"trankilou.fr/lassistanoque/backend/internal/adapter/file"
-	"trankilou.fr/lassistanoque/backend/internal/adapter/llm"
-	"trankilou.fr/lassistanoque/backend/internal/adapter/security"
 	"trankilou.fr/lassistanoque/backend/internal/brain"
 	"trankilou.fr/lassistanoque/backend/internal/http"
 	"trankilou.fr/lassistanoque/backend/internal/service/agent"
@@ -46,60 +41,30 @@ func newServeCmd() *cobra.Command {
 }
 
 func runServe() {
-	// database
-	db, err := database.GetDatabase()
-	if err != nil {
-		panic(err)
-	}
-	defer db.Close()
 
-	err = db.Migrate()
-	if err != nil {
-		panic(err)
-	}
-
-	// storage
-	store, err := file.GetStorageProvider(db.FileRepository())
-	if err != nil {
-		panic(err)
-	}
-
-	// Adapters
-	tokenManager := security.NewJwtTokenManager(12*time.Hour, 7*24*time.Hour, "lassistanoque")
-	pwdAuth := password.NewPasswordAuthenticator(tokenManager, db.UserRepository())
-	llmEngine := llm.NewAnyLLMEngine()
-	repoUser := db.UserRepository()
-	repoProvider := db.ProviderRepository()
-	repoAgent := db.AgentRepository()
-	repoChat := db.ChatRepository()
+	defer Close()
 
 	// services
-	authService := auth.NewService(
-		db.SettingsRepository(),
-		db.UserRepository(),
-		map[string]auth.Authenticator{
-			"password": pwdAuth,
-		},
-	)
-	userService := user.NewService(repoUser)
-	storageService := storage.NewService(store)
-	providerService := provider.NewService(repoProvider, repoUser, llmEngine)
-	agentService := agent.NewService(repoAgent, repoUser)
-	chatService := chat.NewService(repoUser, repoAgent, repoProvider, repoChat, llmEngine)
+	authService := auth.NewService(GetSettingsRepository(), GetUserRepository(), GetAuthenticators())
+	userService := user.NewService(GetUserRepository())
+	storageService := storage.NewService(GetStorageProvider())
+	providerService := provider.NewService(GetProviderRepository(), GetUserRepository(), GetLLMEngine())
+	agentService := agent.NewService(GetAgentRepository(), GetUserRepository())
+	chatService := chat.NewService(GetUserRepository(), GetAgentRepository(), GetProviderRepository(), GetChatRepository(), GetLLMEngine())
 
 	// http server
 	httpRouter := http.NewRouter(http.Dependencies{
 		StorageService:  storageService,
 		AuthService:     authService,
 		UserService:     userService,
-		TokenManager:    tokenManager,
 		ProviderService: providerService,
 		AgentService:    agentService,
 		ChatService:     chatService,
+		TokenManager:    GetTokenManager(),
 	})
 
 	brainRouter := brain.NewRouter(brain.Dependencies{
-		ChatRepository: db.ChatRepository(),
+		ChatRepository: GetChatRepository(),
 	})
 
 	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)