Agent par défaut

backend/cmd/serve.go

@@ -1,7 +1,13 @@
 package cmd
 
 import (
+	"context"
+	"errors"
 	"log/slog"
+	nethttp "net/http"
+	"os"
+	"os/signal"
+	"syscall"
 	"time"
 
 	"github.com/spf13/cobra"
@@ -11,6 +17,7 @@ import (
 	"trankilou.fr/lassistanoque/backend/internal/adapter/llm"
 	"trankilou.fr/lassistanoque/backend/internal/adapter/security"
 	"trankilou.fr/lassistanoque/backend/internal/http"
+	"trankilou.fr/lassistanoque/backend/internal/llmserver"
 	"trankilou.fr/lassistanoque/backend/internal/service/agent"
 	"trankilou.fr/lassistanoque/backend/internal/service/auth"
 	"trankilou.fr/lassistanoque/backend/internal/service/provider"
@@ -84,7 +91,41 @@ func runServe() {
 		AgentService:    agentService,
 	})
 
-	if err := router.Start(); err != nil {
-		slog.Error("failed to start server", "error", err)
+	llms := llmserver.NewLLMServer(llmserver.Dependencies{})
+
+	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
+	defer stop()
+
+	// start LLM service
+	go func() {
+		if err := llms.Start(); err != nil && !errors.Is(err, nethttp.ErrServerClosed) {
+			slog.Error("failed to start llm server", "error", err)
+		}
+	}()
+
+	// start HTTP service
+	go func() {
+		if err := router.Start(); err != nil && !errors.Is(err, nethttp.ErrServerClosed) {
+			slog.Error("failed to start http server", "error", err)
+		}
+	}()
+
+	<-ctx.Done()
+
+	// stop HTTP service
+	ctx, cancel1 := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel1()
+	if err := router.Stop(ctx); err != nil {
+		slog.Error("failed to stop http server", "error", err)
 	}
+	slog.Info("HTTP server stopped gracefully")
+
+	// stop LLM service
+	ctx, cancel2 := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel2()
+	if err := llms.Stop(ctx); err != nil {
+		slog.Error("failed to stop llm server", "error", err)
+	}
+	slog.Info("LLM server stopped gracefully")
+
 }