update cicd

cmd/agent/main.go

@@ -0,0 +1,54 @@
+package main
+
+import (
+	"context"
+	"log"
+	"os"
+	"os/signal"
+	"runtime"
+	"syscall"
+
+	"stream.api/internal/workflow/agent"
+)
+
+func main() {
+	serverAddr := os.Getenv("AGENT_SERVER")
+	if serverAddr == "" {
+		serverAddr = "localhost:9000"
+	}
+
+	secret := os.Getenv("APP_INTERNAL_MARKER")
+	if secret == "" {
+		secret = os.Getenv("INTERNAL_MARKER")
+	}
+	if secret == "" {
+		log.Fatal("APP_INTERNAL_MARKER environment variable is required")
+	}
+
+	capacity := runtime.NumCPU()
+
+	log.Printf("Starting stream.api agent")
+	log.Printf("Server: %s", serverAddr)
+	log.Printf("Capacity: %d", capacity)
+
+	a, err := agent.New(serverAddr, secret, capacity)
+	if err != nil {
+		log.Fatalf("Failed to initialize agent: %v", err)
+	}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	sigCh := make(chan os.Signal, 1)
+	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
+
+	go func() {
+		<-sigCh
+		log.Println("Shutting down agent...")
+		cancel()
+	}()
+
+	if err := a.Run(ctx); err != nil {
+		log.Fatalf("Agent error: %v", err)
+	}
+}