Merge branch 'main' into feat-file-upload

Author: 35C4n0r

CHANGELOG.md

@@ -1,5 +1,10 @@
 # Changelog
 
+## v0.9.0
+
+### Features
+- Add support for initial prompt via `-I` flag
+
 ## v0.8.0
 
 ### Features

chat/package.json

@@ -48,5 +48,5 @@
     "start": "next start",
     "storybook": "storybook dev -p 6006"
   },
-  "version": "0.8.0"
+  "version": "0.9.0"
 }

chat/src/components/message-list.tsx

@@ -20,12 +20,6 @@ interface MessageListProps {
 
 export default function MessageList({ messages }: MessageListProps) {
   const scrollAreaRef = useRef<HTMLDivElement>(null);
-  // Avoid the message list to change its height all the time. It causes some
-  // flickering in the screen because some messages, as the ones displaying
-  // progress statuses, are changing the content(the number of lines) and size
-  // constantily. To minimize it, we keep track of the biggest scroll height of
-  // the content, and use that as the min height of the scroll area.
-  const contentMinHeight = useRef(0);
 
   // Track if user is at bottom - default to true for initial scroll
   const isAtBottomRef = useRef(true);
@@ -67,11 +61,6 @@ export default function MessageList({ messages }: MessageListProps) {
     const isNewUserMessage =
       messages.length > 0 && messages[messages.length - 1].role === "user";
 
-    // Update content min height if needed
-    if (currentScrollHeight > contentMinHeight.current) {
-      contentMinHeight.current = currentScrollHeight;
-    }
-
     // Auto-scroll only if:
     // 1. It's the first render, OR
     // 2. There's new content AND user was at the bottom, OR
@@ -104,9 +93,7 @@ export default function MessageList({ messages }: MessageListProps) {
   return (
     <div className="overflow-y-auto flex-1" ref={scrollAreaRef}>
       <div
-        className="p-4 flex flex-col gap-4 max-w-4xl mx-auto"
-        style={{ minHeight: contentMinHeight.current }}
-      >
+        className="p-4 flex flex-col gap-4 max-w-4xl mx-auto transition-all duration-300 ease-in-out min-h-0">
         {messages.map((message) => (
           <div
             key={message.id ?? "draft"}

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 {

internal/version/version.go

@@ -2,4 +2,4 @@
 
 package version
 
-var Version = "0.8.0"
+var Version = "0.9.0"

lib/httpapi/server.go

@@ -101,6 +101,7 @@ type ServerConfig struct {
 	ChatBasePath   string
 	AllowedHosts   []string
 	AllowedOrigins []string
+	InitialPrompt  string
 }
 
 // Validate allowed hosts don't contain whitespace, commas, schemes, or ports.
@@ -236,7 +237,7 @@ func NewServer(ctx context.Context, config ServerConfig) (*Server, error) {
 		SnapshotInterval:      snapshotInterval,
 		ScreenStabilityLength: 2 * time.Second,
 		FormatMessage:         formatMessage,
-	})
+	}, config.InitialPrompt)
 	emitter := NewEventEmitter(1024)
 
 	// Create temporary directory for uploads
@@ -321,7 +322,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.conversation.InitialPromptSent && convertStatus(currentStatus) == AgentStatusStable {
+				if err := s.conversation.SendMessage(FormatMessage(s.agentType, s.conversation.InitialPrompt)...); err != nil {
+					s.logger.Error("Failed to send initial prompt", "error", err)
+				} else {
+					s.conversation.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)

lib/screentracker/conversation.go

@@ -74,6 +74,10 @@ type Conversation struct {
 	messages                    []ConversationMessage
 	screenBeforeLastUserMessage string
 	lock                        sync.Mutex
+	// InitialPrompt is the initial prompt passed to the agent
+	InitialPrompt string
+	// InitialPromptSent keeps track if the InitialPrompt has been successfully sent to the agents
+	InitialPromptSent bool
 }
 
 type ConversationStatus string
@@ -94,7 +98,7 @@ func getStableSnapshotsThreshold(cfg ConversationConfig) int {
 	return threshold + 1
 }
 
-func NewConversation(ctx context.Context, cfg ConversationConfig) *Conversation {
+func NewConversation(ctx context.Context, cfg ConversationConfig, initialPrompt string) *Conversation {
 	threshold := getStableSnapshotsThreshold(cfg)
 	c := &Conversation{
 		cfg:                      cfg,
@@ -107,6 +111,8 @@ func NewConversation(ctx context.Context, cfg ConversationConfig) *Conversation
 				Time:    cfg.GetTime(),
 			},
 		},
+		InitialPrompt:     initialPrompt,
+		InitialPromptSent: len(initialPrompt) == 0,
 	}
 	return c
 }

lib/screentracker/conversation_test.go

@@ -42,7 +42,7 @@ func statusTest(t *testing.T, params statusTestParams) {
 		if params.cfg.GetTime == nil {
 			params.cfg.GetTime = func() time.Time { return time.Now() }
 		}
-		c := st.NewConversation(ctx, params.cfg)
+		c := st.NewConversation(ctx, params.cfg, "")
 		assert.Equal(t, st.ConversationStatusInitializing, c.Status())
 
 		for i, step := range params.steps {
@@ -147,7 +147,7 @@ func TestMessages(t *testing.T) {
 		for _, opt := range opts {
 			opt(&cfg)
 		}
-		return st.NewConversation(context.Background(), cfg)
+		return st.NewConversation(context.Background(), cfg, "")
 	}
 
 	t.Run("messages are copied", func(t *testing.T) {

openapi.json

@@ -334,7 +334,7 @@
   "info": {
     "description": "HTTP API for Claude Code, Goose, and Aider.\n\nhttps://github.com/coder/agentapi",
     "title": "AgentAPI",
-    "version": "0.8.0"
+    "version": "0.9.0"
   },
   "openapi": "3.1.0",
   "paths": {