lmcli/pkg/cli/cmd.go

137 lines
3.4 KiB
Go
Raw Normal View History

package cli
import (
"fmt"
"strings"
2023-11-04 16:56:22 -06:00
"github.com/spf13/cobra"
)
var rootCmd = &cobra.Command{
2023-11-04 16:37:18 -06:00
Use: "lmcli",
Short: "Interact with Large Language Models",
2023-11-04 16:37:18 -06:00
Long: `lmcli is a CLI tool to interact with Large Language Models.`,
Run: func(cmd *cobra.Command, args []string) {
// execute `lm ls` by default
},
}
var lsCmd = &cobra.Command{
Use: "ls",
Short: "List existing conversations",
Long: `List all existing conversations in descending order of recent activity.`,
Run: func(cmd *cobra.Command, args []string) {
fmt.Println("Listing conversations...")
// Example output, asterisk to indicate current converation
// $ lm ls
// last hour:
// 98sg - 12 minutes ago - Project discussion
// last day:
// tj3l - 10 hours ago - Deep learning concepts
// last week:
// bwfm - 2 days ago - Machine learning study
// * 8n3h - 3 days ago - Weekend plans
// f3n7 - 6 days ago - CLI development
// last month:
// 5hn2 - 8 days ago - Book club discussion
// b7ze - 20 days ago - Gardening tips and tricks
// last 6 months:
// 3jn2 - 30 days ago - Web development best practices
// 43jk - 2 months ago - Longboard maintenance
// g8d9 - 3 months ago - History book club
// 4lk3 - 4 months ago - Local events and meetups
// 43jn - 6 months ago - Mobile photography techniques
},
}
var viewCmd = &cobra.Command{
Use: "view",
Short: "View messages in a conversation",
Long: `Displays all the messages in a coversation.`,
Run: func(cmd *cobra.Command, args []string) {
fmt.Println("Displaying conversation messages...")
},
}
2023-11-04 16:37:18 -06:00
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.`,
Run: func(cmd *cobra.Command, args []string) {
fmt.Println("Replying to a conversation...")
},
}
var newCmd = &cobra.Command{
Use: "new",
Short: "Start a new conversation",
2023-11-04 16:56:22 -06:00
Long: `Start a new conversation with the Large Language Model.`,
Run: func(cmd *cobra.Command, args []string) {
2023-11-04 16:53:09 -06:00
messageContents, err := InputFromEditor("# What would you like to say?\n", "message.*.md")
if err != nil {
Fatal("Failed to get input: %v\n", err)
return
}
if messageContents == "" {
Fatal("No message was provided.\n")
return
}
fmt.Printf("> %s\n", messageContents)
messages := []Message{
{
2023-11-04 16:56:22 -06:00
Role: "user",
OriginalContent: messageContents,
},
}
receiver := make(chan string)
go HandleDelayedResponse(receiver)
err = CreateChatCompletionStream("You are a helpful assistant.", messages, receiver)
if err != nil {
Fatal("%v\n", err)
return
}
fmt.Println()
},
}
var promptCmd = &cobra.Command{
2023-11-04 16:56:22 -06:00
Use: "prompt",
Short: "Do a one-shot prompt",
2023-11-04 16:56:22 -06:00
Long: `Prompt the Large Language Model and get a response.`,
Run: func(cmd *cobra.Command, args []string) {
2023-11-04 16:53:09 -06:00
message := strings.Join(args, " ")
if len(strings.Trim(message, " \t\n")) == 0 {
Fatal("No message was provided.\n")
return
}
messages := []Message{
{
2023-11-04 16:56:22 -06:00
Role: "user",
2023-11-04 16:53:09 -06:00
OriginalContent: message,
},
}
receiver := make(chan string)
go HandleDelayedResponse(receiver)
err := CreateChatCompletionStream("You are a helpful assistant.", messages, receiver)
if err != nil {
Fatal("%v\n", err)
return
}
fmt.Println()
},
}
func NewRootCmd() *cobra.Command {
rootCmd.AddCommand(newCmd, promptCmd)
2023-11-04 16:56:22 -06:00
return rootCmd
}