pkg/cli/cmd.go

@@ -35,23 +35,6 @@ func Execute() error {
 	return rootCmd.Execute()
 }
 
-// InputFromArgsOrEditor returns either the provided input from the args slice
-// (joined with spaces), or if len(args) is 0, opens an editor and returns
-// whatever input was provided there. placeholder is a string which populates
-// the editor and gets stripped from the final output.
-func InputFromArgsOrEditor(args []string, placeholder string) (message string) {
-	var err error
-	if len(args) == 0 {
-		message, err = InputFromEditor(placeholder, "message.*.md")
-		if err != nil {
-			Fatal("Failed to get input: %v\n", err)
-		}
-	} else {
-		message = strings.Trim(strings.Join(args, " "), " \t\n")
-	}
-	return
-}
-
 var rootCmd = &cobra.Command{
 	Use:   "lmcli",
 	Short: "Interact with Large Language Models",
@@ -166,7 +149,7 @@ var lsCmd = &cobra.Command{
 }
 
 var rmCmd = &cobra.Command{
-	Use:   "rm <conversation>",
+	Use:   "rm [conversation]",
 	Short: "Remove a conversation",
 	Long:  `Removes a conversation by its short name.`,
 	Args: func(cmd *cobra.Command, args []string) error {
@@ -200,7 +183,7 @@ var rmCmd = &cobra.Command{
 }
 
 var viewCmd = &cobra.Command{
-	Use:   "view <conversation>",
+	Use:   "view [conversation]",
 	Short: "View messages in a conversation",
 	Long:  `Finds a conversation by its short name and displays its contents.`,
 	Args: func(cmd *cobra.Command, args []string) error {
@@ -234,7 +217,7 @@ var viewCmd = &cobra.Command{
 }
 
 var replyCmd = &cobra.Command{
-	Use:   "reply <conversation> [message]",
+	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 {
@@ -256,7 +239,7 @@ var replyCmd = &cobra.Command{
 			Fatal("Could not retrieve messages for conversation: %s\n", conversation.Title)
 		}
 
-		messageContents := InputFromArgsOrEditor(args[1:], "# How would you like to reply?\n")
+		messageContents, err := InputFromEditor("# How would you like to reply?\n", "reply.*.md")
 		if messageContents == "" {
 			Fatal("No reply was provided.\n")
 		}
@@ -311,18 +294,22 @@ var replyCmd = &cobra.Command{
 }
 
 var newCmd = &cobra.Command{
-	Use:   "new [message]",
+	Use:   "new",
 	Short: "Start a new conversation",
 	Long:  `Start a new conversation with the Large Language Model.`,
 	Run: func(cmd *cobra.Command, args []string) {
-		messageContents := InputFromArgsOrEditor(args, "# What would you like to say?\n")
+		messageContents, err := InputFromEditor("# What would you like to say?\n", "message.*.md")
+		if err != nil {
+			Fatal("Failed to get input: %v\n", err)
+		}
+
 		if messageContents == "" {
 			Fatal("No message was provided.\n")
 		}
 
 		// TODO: set title if --title provided, otherwise defer for later(?)
 		conversation := Conversation{}
-		err := store.SaveConversation(&conversation)
+		err = store.SaveConversation(&conversation)
 		if err != nil {
 			Fatal("Could not save new conversation: %v\n", err)
 		}
@@ -378,7 +365,6 @@ var newCmd = &cobra.Command{
 		if err != nil {
 			Warn("Could not generate title for conversation: %v\n", err)
 		}
-
 		err = store.SaveConversation(&conversation)
 		if err != nil {
 			Warn("Could not save conversation after generating title: %v\n", err)
@@ -387,12 +373,12 @@ var newCmd = &cobra.Command{
 }
 
 var promptCmd = &cobra.Command{
-	Use:   "prompt [message]",
+	Use:   "prompt",
 	Short: "Do a one-shot prompt",
 	Long:  `Prompt the Large Language Model and get a response.`,
 	Run: func(cmd *cobra.Command, args []string) {
-		message := InputFromArgsOrEditor(args, "# What would you like to say?\n")
-		if message == "" {
+		message := strings.Join(args, " ")
+		if len(strings.Trim(message, " \t\n")) == 0 {
 			Fatal("No message was provided.\n")
 		}
 

pkg/cli/conversation.go

@@ -33,7 +33,7 @@ func (c *Conversation) GenerateTitle() error {
 	}
 
 	model := "gpt-3.5-turbo" // use cheap model to generate title
-	response, err := CreateChatCompletion(model, messages, 25)
+	response, err := CreateChatCompletion(model, messages, 10)
 	if err != nil {
 		return err
 	}