feat: accept initial prompt

Author: 35C4n0r

cmd/server/server.go

@@ -112,6 +112,7 @@ func runServer(ctx context.Context, logger *slog.Logger, argsToPass []string) er
 		ChatBasePath:   viper.GetString(FlagChatBasePath),
 		AllowedHosts:   viper.GetStringSlice(FlagAllowedHosts),
 		AllowedOrigins: viper.GetStringSlice(FlagAllowedOrigins),
+		InitialPrompt:  viper.GetString(FlagInitialPrompt),
 	})
 	if err != nil {
 		return xerrors.Errorf("failed to create server: %w", err)
@@ -174,6 +175,7 @@ const (
 	FlagAllowedHosts   = "allowed-hosts"
 	FlagAllowedOrigins = "allowed-origins"
 	FlagExit           = "exit"
+	FlagInitialPrompt  = "initial-prompt"
 )
 
 func CreateServerCmd() *cobra.Command {
@@ -211,6 +213,7 @@ func CreateServerCmd() *cobra.Command {
 		{FlagAllowedHosts, "a", []string{"localhost", "127.0.0.1", "[::1]"}, "HTTP allowed hosts (hostnames only, no ports). Use '*' for all, comma-separated list via flag, space-separated list via AGENTAPI_ALLOWED_HOSTS env var", "stringSlice"},
 		// localhost:3284 is the default origin when you open the chat interface in your browser. localhost:3000 and 3001 are used during development.
 		{FlagAllowedOrigins, "o", []string{"http://localhost:3284", "http://localhost:3000", "http://localhost:3001"}, "HTTP allowed origins. Use '*' for all, comma-separated list via flag, space-separated list via AGENTAPI_ALLOWED_ORIGINS env var", "stringSlice"},
+		{FlagInitialPrompt, "I", "", "Initial prompt for the agent (recommended only if the agent doesn't support initial prompt in interaction mode)", "string"},
 	}
 
 	for _, spec := range flagSpecs {

lib/httpapi/server.go

@@ -29,17 +29,19 @@ import (
 
 // Server represents the HTTP server
 type Server struct {
-	router       chi.Router
-	api          huma.API
-	port         int
-	srv          *http.Server
-	mu           sync.RWMutex
-	logger       *slog.Logger
-	conversation *st.Conversation
-	agentio      *termexec.Process
-	agentType    mf.AgentType
-	emitter      *EventEmitter
-	chatBasePath string
+	router            chi.Router
+	api               huma.API
+	port              int
+	srv               *http.Server
+	mu                sync.RWMutex
+	logger            *slog.Logger
+	conversation      *st.Conversation
+	agentio           *termexec.Process
+	agentType         mf.AgentType
+	emitter           *EventEmitter
+	chatBasePath      string
+	initialPrompt     string
+	initialPromptSent bool
 }
 
 func (s *Server) NormalizeSchema(schema any) any {
@@ -95,6 +97,7 @@ type ServerConfig struct {
 	ChatBasePath   string
 	AllowedHosts   []string
 	AllowedOrigins []string
+	InitialPrompt  string
 }
 
 // Validate allowed hosts don't contain whitespace, commas, schemes, or ports.
@@ -233,15 +236,17 @@ func NewServer(ctx context.Context, config ServerConfig) (*Server, error) {
 	})
 	emitter := NewEventEmitter(1024)
 	s := &Server{
-		router:       router,
-		api:          api,
-		port:         config.Port,
-		conversation: conversation,
-		logger:       logger,
-		agentio:      config.Process,
-		agentType:    config.AgentType,
-		emitter:      emitter,
-		chatBasePath: strings.TrimSuffix(config.ChatBasePath, "/"),
+		router:            router,
+		api:               api,
+		port:              config.Port,
+		conversation:      conversation,
+		logger:            logger,
+		agentio:           config.Process,
+		agentType:         config.AgentType,
+		emitter:           emitter,
+		chatBasePath:      strings.TrimSuffix(config.ChatBasePath, "/"),
+		initialPrompt:     config.InitialPrompt,
+		initialPromptSent: len(config.InitialPrompt) == 0,
 	}
 
 	// Register API routes
@@ -306,7 +311,19 @@ func (s *Server) StartSnapshotLoop(ctx context.Context) {
 	s.conversation.StartSnapshotLoop(ctx)
 	go func() {
 		for {
-			s.emitter.UpdateStatusAndEmitChanges(s.conversation.Status())
+			currentStatus := s.conversation.Status()
+
+			// Send initial prompt when agent becomes stable for the first time
+			if !s.initialPromptSent && convertStatus(currentStatus) == AgentStatusStable {
+				if err := s.conversation.SendMessage(FormatMessage(s.agentType, s.initialPrompt)...); err != nil {
+					s.logger.Error("Failed to send initial prompt", "error", err)
+				} else {
+					s.initialPromptSent = true
+					currentStatus = st.ConversationStatusChanging
+					s.logger.Info("Initial prompt sent successfully")
+				}
+			}
+			s.emitter.UpdateStatusAndEmitChanges(currentStatus)
 			s.emitter.UpdateMessagesAndEmitChanges(s.conversation.Messages())
 			s.emitter.UpdateScreenAndEmitChanges(s.conversation.Screen())
 			time.Sleep(snapshotInterval)