Implement lmcli reply

pkg/cli/cmd.go

@@ -201,8 +201,75 @@ var replyCmd = &cobra.Command{
 	Use:   "reply",
 	Short: "Send a reply to a conversation",
 	Long:  `Sends a reply to conversation and writes the response to stdout.`,
+	Args: func(cmd *cobra.Command, args []string) error {
+		argCount := 1
+		if err := cobra.MinimumNArgs(argCount)(cmd, args); err != nil {
+			return err
+		}
+		return nil
+	},
 	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("Replying to a conversation...")
+		shortName := args[0]
+		conversation, err := store.ConversationByShortName(shortName)
+		if conversation.ID == 0 {
+			Fatal("Conversation not found with short name: %s\n", shortName)
+		}
+
+		messages, err := store.Messages(conversation)
+		if err != nil {
+			Fatal("Could not retrieve messages for conversation: %s\n", conversation.Title)
+		}
+
+		messageContents, err := InputFromEditor("# How would you like to reply?\n", "reply.*.md")
+
+		userReply := Message{
+			ConversationID: conversation.ID,
+			Role: "user",
+			OriginalContent: messageContents,
+		}
+
+		err = store.SaveMessage(&userReply)
+		if err != nil {
+			Warn("Could not save your reply: %v\n", err)
+		}
+
+		messages = append(messages, userReply)
+		for _, message := range messages {
+			message.RenderTTY(true)
+		}
+
+		assistantReply := Message{
+			ConversationID: conversation.ID,
+			Role:           "assistant",
+		}
+		assistantReply.RenderTTY(false)
+
+		receiver := make(chan string)
+		response := make(chan string)
+		go func() {
+			response <- HandleDelayedResponse(receiver)
+		}()
+
+		err = CreateChatCompletionStream(config.OpenAI.DefaultModel, messages, MAX_TOKENS, receiver)
+		if err != nil {
+			Fatal("%v\n", err)
+		}
+
+		assistantReply.OriginalContent = <-response
+
+		err = store.SaveMessage(&assistantReply)
+		if err != nil {
+			Fatal("Could not save assistant reply: %v\n", err)
+		}
+
+		fmt.Println()
+	},
+	ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		compMode := cobra.ShellCompDirectiveNoFileComp
+		if len(args) != 0 {
+			return nil, compMode
+		}
+		return store.ConversationShortNameCompletions(toComplete), compMode
 	},
 }
 
@@ -326,6 +393,7 @@ func NewRootCmd() *cobra.Command {
 		lsCmd,
 		newCmd,
 		promptCmd,
+		replyCmd,
 		rmCmd,
 		viewCmd,
 	)