package cmd import ( "fmt" "git.mlow.ca/mlow/lmcli/pkg/api" cmdutil "git.mlow.ca/mlow/lmcli/pkg/cmd/util" "git.mlow.ca/mlow/lmcli/pkg/lmcli" "github.com/spf13/cobra" ) func NewCmd(ctx *lmcli.Context) *cobra.Command { cmd := &cobra.Command{ Use: "new [message]", Short: "Start a new conversation", Long: `Start a new conversation with the Large Language Model.`, RunE: func(cmd *cobra.Command, args []string) error { err := validateGenerationFlags(ctx, cmd) if err != nil { return err } input := inputFromArgsOrEditor(args, "# Start a new conversation below\n", "") if input == "" { return fmt.Errorf("No message was provided.") } var messages []api.Message system := ctx.Config.GetSystemPrompt() if system != "" { messages = append(messages, api.Message{ Role: api.MessageRoleSystem, Content: system, }) } messages = append(messages, api.Message{ Role: api.MessageRoleUser, Content: input, }) conversation, messages, err := ctx.Store.StartConversation(messages...) if err != nil { return fmt.Errorf("Could not start a new conversation: %v", err) } cmdutil.HandleReply(ctx, &messages[len(messages)-1], true) title, err := cmdutil.GenerateTitle(ctx, messages) if err != nil { lmcli.Warn("Could not generate title for conversation %s: %v\n", conversation.ShortName.String, err) } conversation.Title = title err = ctx.Store.UpdateConversation(conversation) if err != nil { lmcli.Warn("Could not save conversation title: %v\n", err) } return nil }, } applyGenerationFlags(ctx, cmd) return cmd }