package cmd import ( "fmt" cmdutil "git.mlow.ca/mlow/lmcli/pkg/cmd/util" "git.mlow.ca/mlow/lmcli/pkg/lmcli" "git.mlow.ca/mlow/lmcli/pkg/tui" "github.com/spf13/cobra" ) func ChatCmd(ctx *lmcli.Context) *cobra.Command { cmd := &cobra.Command{ Use: "chat [conversation]", Short: "Open the chat interface", Long: `Open the chat interface, optionally on a given conversation.`, RunE: func(cmd *cobra.Command, args []string) error { shortname := "" if len(args) == 1 { shortname = args[0] } if shortname != ""{ _, err := cmdutil.LookupConversationE(ctx, shortname) if err != nil { return err } } err := tui.Launch(ctx, shortname) if err != nil { return fmt.Errorf("Error fetching LLM response: %v", err) } return nil }, ValidArgsFunction: func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) { compMode := cobra.ShellCompDirectiveNoFileComp if len(args) != 0 { return nil, compMode } return ctx.Store.ConversationShortNameCompletions(toComplete), compMode }, } applyPromptFlags(ctx, cmd) return cmd }