lmcli/pkg/cmd/new.go

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 {
			input := inputFromArgsOrEditor(args, "# Start a new conversation below\n", "")
			if input == "" {
				return fmt.Errorf("No message was provided.")
			}

			var messages []api.Message

			// TODO: probably just make this part of the conversation
			system := ctx.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
		},
	}

	applyPromptFlags(ctx, cmd)
	return cmd
}