diff --git a/chat/main.go b/chat/main.go new file mode 100644 index 0000000..265784a --- /dev/null +++ b/chat/main.go @@ -0,0 +1,23 @@ +package chat + +import ( + "github.com/nemoden/chat/config" + gogpt "github.com/sashabaranov/go-gpt3" +) + +type Client struct { + client *gogpt.Client +} + +func NewClient(c *config.Config) (*Client, error) { + apiKey, _, err := config.LoadApiKey() + if err != nil { + return nil, err + } + client := gogpt.NewClient(apiKey) + return &Client{client}, nil +} + +func (c *Client) GptClient() *gogpt.Client { + return c.client +} diff --git a/cmd/init.go b/cmd/init.go new file mode 100644 index 0000000..b33c0c7 --- /dev/null +++ b/cmd/init.go @@ -0,0 +1,46 @@ +/* +Copyright © 2023 NAME HERE + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +*/ +package cmd + +import ( + "fmt" + + "github.com/spf13/cobra" +) + +// initCmd represents the init command +var initCmd = &cobra.Command{ + Use: "init", + Short: "Initialises chat", + Long: "", + Run: func(cmd *cobra.Command, args []string) { + fmt.Println("init called") + }, +} + +func init() { + rootCmd.AddCommand(initCmd) + + // Here you will define your flags and configuration settings. + + // Cobra supports Persistent Flags which will work for this command + // and all subcommands, e.g.: + // initCmd.PersistentFlags().String("foo", "", "A help for foo") + + // Cobra supports local flags which will only run when this command + // is called directly, e.g.: + // initCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle") +} diff --git a/cmd/main.go b/cmd/main.go new file mode 100644 index 0000000..62bc654 --- /dev/null +++ b/cmd/main.go @@ -0,0 +1,17 @@ +package cmd + +import "github.com/spf13/cobra" + +func GetRootCmd() *cobra.Command { + return rootCmd +} + +func GetSubcommandsMap() map[string]bool { + commands := rootCmd.Commands() + commandsLen := len(commands) + subcommands := make(map[string]bool, commandsLen) + for _, c := range commands { + subcommands[c.Use] = true + } + return subcommands +} diff --git a/cmd/root.go b/cmd/root.go new file mode 100644 index 0000000..955a4fe --- /dev/null +++ b/cmd/root.go @@ -0,0 +1,110 @@ +/* +Copyright © 2023 NAME HERE +*/ +package cmd + +import ( + "bufio" + "context" + "errors" + "fmt" + "strings" + + "os" + + "github.com/nemoden/chat/chat" + "github.com/nemoden/chat/config" + gogpt "github.com/sashabaranov/go-gpt3" + + "github.com/spf13/cobra" +) + +// rootCmd represents the base command when called without any subcommands +var rootCmd = &cobra.Command{ + Use: "chat", + Short: "ChatGPT CLI and REPL", + Long: `Use ChatGPT not leaving your terminal`, + FParseErrWhitelist: cobra.FParseErrWhitelist{ + UnknownFlags: true, + }, + Run: func(cmd *cobra.Command, args []string) { + reader := bufio.NewReader(os.Stdin) + + options := getOptions(os.Args[1:]) + c := config.LoadConfig(options) + client, err := chat.NewClient(c) + + if errors.Is(err, config.ErrApiTokenFileDoesntExist) { + fmt.Printf(config.TokenFileDoesntExistPrompt) + os.Exit(0) + } + if errors.Is(err, config.ErrCantOpenApiTokenFileForReading) { + fmt.Printf(config.TokenFileNotReadablePrompt) + os.Exit(0) + } + + ctx := context.Background() + + for { + fmt.Printf("You: ") + input, _ := reader.ReadString('\n') + input = strings.TrimSpace(input) + + if input == "exit" { + break + } + + prompt := "" + if c.Format == "markdown" { + prompt += "Return response in markdown format. Prompt on a new line:\n" + } + prompt += input + req := gogpt.CompletionRequest{ + Model: gogpt.GPT3TextDavinci003, + Prompt: prompt, + MaxTokens: 1000, + Temperature: 0.5, + } + + fmt.Printf("ChatGPT: ") + + stream, _ := client.GptClient().CreateCompletionStream(ctx, req) + + defer stream.Close() + + c.Renderer.Render(stream) + } + }, +} + +// Execute adds all child commands to the root command and sets flags appropriately. +// This is called by main.main(). It only needs to happen once to the rootCmd. +func Execute() { + err := rootCmd.Execute() + if err != nil { + os.Exit(1) + } +} + +func init() { + // Here you will define your flags and configuration settings. + // Cobra supports persistent flags, which, if defined here, + // will be global for your application. + + // rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.chat.yaml)") + + // Cobra also supports local flags, which will only run + // when this action is called directly. + rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle") +} + +// @TODO DRY, duplication in ./main.go +func getOptions(args []string) []string { + var opts []string + for _, arg := range args { + if strings.HasPrefix(arg, "--") { + opts = append(opts, arg) + } + } + return opts +} diff --git a/config/main.go b/config/main.go new file mode 100644 index 0000000..ced9710 --- /dev/null +++ b/config/main.go @@ -0,0 +1,155 @@ +package config + +import ( + "bufio" + "errors" + "fmt" + "os" + "path" + "runtime" + + "github.com/nemoden/chat/renderer" +) + +var ( + ConfigDir string = getConfigDir() + CacheDir string = getCacheDir() + ApiKeyFilePath string = getApiKeyFilePath() + InitPrompt string = `Please initialise chat using the init command: + +$ chat init +` + // @TODO revisit instructions + TokenFileDoesntExistPrompt string = fmt.Sprintf(`Oops. Looks like your token file %s doesn't exist. + +Please obtain the openai API key from their website https://platform.openai.com/account/api-keys + +Once you have the API key, you can either add it manually to %s: + +$ echo "" > %s + +Or if you didn't run chat init, it's adviced that you do this instead: + +$ chat init + +Alternatively, you can provide a token using environment variable %s. How to set it depends on your shell, i.e. + +- for bash shell, add export %s= to your ~/.bash_profile +- for zsh shell, add export %s= to your ~/.zsh_profile +- for fish shell, add set -x %s to your ~/.config/fish/config.fish +`, ApiKeyFilePath, ApiKeyFilePath, ApiKeyFilePath, API_KEY_ENV_VAR_NAME, API_KEY_ENV_VAR_NAME, API_KEY_ENV_VAR_NAME, API_KEY_ENV_VAR_NAME) + + TokenFileNotReadablePrompt string = fmt.Sprintf(`Oops. Looks like your token file %s is not readable. + +chat stores the api token in that file. + +Another option is to store the API key in the environment variable. How to set it depends on your shell, i.e. + +- for bash shell, add export %s= to your ~/.bash_profile +- for zsh shell, add export %s= to your ~/.zsh_profile +- for fish shell, add set -x %s to your ~/.config/fish/config.fish +`, ApiKeyFilePath, API_KEY_ENV_VAR_NAME, API_KEY_ENV_VAR_NAME, API_KEY_ENV_VAR_NAME) +) + +var ( + ErrApiTokenFileDoesntExist = errors.New(fmt.Sprintf("API key file %s doesn't exist", ApiKeyFilePath)) + ErrCantOpenApiTokenFileForReading = errors.New(fmt.Sprintf("Can not open API key file %s for reading", ApiKeyFilePath)) +) + +const ( + APP_NAME = "chat" + API_KEY_ENV_VAR_NAME = "CHAT_GPT_API_TOKEN" + API_KEY_SOURCE_ENV = "env" + API_KEY_SOURCE_FILE = "file" +) + +type Config struct { + Renderer renderer.Renderer + Format string +} + +func inSlice(what string, slice []string) bool { + for _, i := range slice { + if what == i { + return true + } + } + return false +} +func LoadConfig(optionsOverride []string) *Config { + var r renderer.Renderer + var format string + if inSlice("--md", optionsOverride) { + r = renderer.NewMarkdownRenderer(os.Stdout, "ChatGPT: ") + format = "markdown" + } else { + r = renderer.NewPassthruRenderer(os.Stdout, "ChatGPT: ") + format = "" + } + return &Config{ + Renderer: r, + Format: format, + } +} + +func LoadApiKey() (string, string, error) { + apiKey := os.Getenv(API_KEY_ENV_VAR_NAME) + if apiKey != "" { + return apiKey, API_KEY_SOURCE_ENV, nil + } + _, err := os.Stat(ApiKeyFilePath) + if err != nil { + return "", "", ErrApiTokenFileDoesntExist + } + file, err := os.Open(ApiKeyFilePath) + if err != nil { + return "", "", ErrCantOpenApiTokenFileForReading + } + defer file.Close() + + scanner := bufio.NewScanner(file) + scanner.Split(bufio.ScanWords) + scanner.Scan() + return scanner.Text(), API_KEY_SOURCE_FILE, nil +} + +func getCacheDir() string { + var dir string + switch runtime.GOOS { + case "darwin", "linux", "freebsd", "openbsd": + dir = os.Getenv("XDG_CACHE_HOME") + if dir != "" { + return dir + } + home, _ := os.UserHomeDir() + return path.Join(home, ".cache", APP_NAME) + default: + dir, _ := os.UserConfigDir() + return dir + } +} + +func getConfigDir() string { + var dir string + switch runtime.GOOS { + case "darwin", "linux", "freebsd", "openbsd": + dir = os.Getenv("XDG_CONFIG_HOME") + if dir != "" { + return dir + } + home, _ := os.UserHomeDir() + return path.Join(home, ".config", APP_NAME) + default: + dir, _ := os.UserConfigDir() + return dir + } +} + +func getApiKeyFilePath() string { + home, _ := os.UserHomeDir() + return path.Join(home, "."+APP_NAME) +} + +func StoreApiToken(token string) bool { + return true +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..1791bd9 --- /dev/null +++ b/go.mod @@ -0,0 +1,29 @@ +module github.com/nemoden/chat + +go 1.19 + +require ( + github.com/alecthomas/chroma v0.10.0 // indirect + github.com/aymanbagabas/go-osc52 v1.0.3 // indirect + github.com/aymerick/douceur v0.2.0 // indirect + github.com/charmbracelet/glamour v0.6.0 // indirect + github.com/dlclark/regexp2 v1.4.0 // indirect + github.com/gorilla/css v1.0.0 // indirect + github.com/gosuri/uilive v0.0.4 // indirect + github.com/inconshreveable/mousetrap v1.0.1 // indirect + github.com/lucasb-eyer/go-colorful v1.2.0 // indirect + github.com/mattn/go-isatty v0.0.16 // indirect + github.com/mattn/go-runewidth v0.0.14 // indirect + github.com/microcosm-cc/bluemonday v1.0.21 // indirect + github.com/muesli/reflow v0.3.0 // indirect + github.com/muesli/termenv v0.13.0 // indirect + github.com/olekukonko/tablewriter v0.0.5 // indirect + github.com/rivo/uniseg v0.2.0 // indirect + github.com/sashabaranov/go-gpt3 v1.0.0 // indirect + github.com/spf13/cobra v1.6.1 // indirect + github.com/spf13/pflag v1.0.5 // indirect + github.com/yuin/goldmark v1.5.2 // indirect + github.com/yuin/goldmark-emoji v1.0.1 // indirect + golang.org/x/net v0.0.0-20221002022538-bcab6841153b // indirect + golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..1a2e040 --- /dev/null +++ b/go.sum @@ -0,0 +1,65 @@ +github.com/alecthomas/chroma v0.10.0 h1:7XDcGkCQopCNKjZHfYrNLraA+M7e0fMiJ/Mfikbfjek= +github.com/alecthomas/chroma v0.10.0/go.mod h1:jtJATyUxlIORhUOFNA9NZDWGAQ8wpxQQqNSB4rjA/1s= +github.com/aymanbagabas/go-osc52 v1.0.3 h1:DTwqENW7X9arYimJrPeGZcV0ln14sGMt3pHZspWD+Mg= +github.com/aymanbagabas/go-osc52 v1.0.3/go.mod h1:zT8H+Rk4VSabYN90pWyugflM3ZhpTZNC7cASDfUCdT4= +github.com/aymerick/douceur v0.2.0 h1:Mv+mAeH1Q+n9Fr+oyamOlAkUNPWPlA8PPGR0QAaYuPk= +github.com/aymerick/douceur v0.2.0/go.mod h1:wlT5vV2O3h55X9m7iVYN0TBM0NH/MmbLnd30/FjWUq4= +github.com/charmbracelet/glamour v0.6.0 h1:wi8fse3Y7nfcabbbDuwolqTqMQPMnVPeZhDM273bISc= +github.com/charmbracelet/glamour v0.6.0/go.mod h1:taqWV4swIMMbWALc0m7AfE9JkPSU8om2538k9ITBxOc= +github.com/cpuguy83/go-md2man/v2 v2.0.2/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dlclark/regexp2 v1.4.0 h1:F1rxgk7p4uKjwIQxBs9oAXe5CqrXlCduYEJvrF4u93E= +github.com/dlclark/regexp2 v1.4.0/go.mod h1:2pZnwuY/m+8K6iRw6wQdMtk+rH5tNGR1i55kozfMjCc= +github.com/gorilla/css v1.0.0 h1:BQqNyPTi50JCFMTw/b67hByjMVXZRwGha6wxVGkeihY= +github.com/gorilla/css v1.0.0/go.mod h1:Dn721qIggHpt4+EFCcTLTU/vk5ySda2ReITrtgBl60c= +github.com/gosuri/uilive v0.0.4 h1:hUEBpQDj8D8jXgtCdBu7sWsy5sbW/5GhuO8KBwJ2jyY= +github.com/gosuri/uilive v0.0.4/go.mod h1:V/epo5LjjlDE5RJUcqx8dbw+zc93y5Ya3yg8tfZ74VI= +github.com/inconshreveable/mousetrap v1.0.1 h1:U3uMjPSQEBMNp1lFxmllqCPM6P5u/Xq7Pgzkat/bFNc= +github.com/inconshreveable/mousetrap v1.0.1/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw= +github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY= +github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0= +github.com/mattn/go-isatty v0.0.16 h1:bq3VjFmv/sOjHtdEhmkEV4x1AJtvUvOJ2PFAZ5+peKQ= +github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +github.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI= +github.com/mattn/go-runewidth v0.0.12/go.mod h1:RAqKPSqVFrSLVXbA8x7dzmKdmGzieGRCM46jaSJTDAk= +github.com/mattn/go-runewidth v0.0.14 h1:+xnbZSEeDbOIg5/mE6JF0w6n9duR1l3/WmbinWVwUuU= +github.com/mattn/go-runewidth v0.0.14/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w= +github.com/microcosm-cc/bluemonday v1.0.21 h1:dNH3e4PSyE4vNX+KlRGHT5KrSvjeUkoNPwEORjffHJg= +github.com/microcosm-cc/bluemonday v1.0.21/go.mod h1:ytNkv4RrDrLJ2pqlsSI46O6IVXmZOBBD4SaJyDwwTkM= +github.com/muesli/reflow v0.3.0 h1:IFsN6K9NfGtjeggFP+68I4chLZV2yIKsXJFNZ+eWh6s= +github.com/muesli/reflow v0.3.0/go.mod h1:pbwTDkVPibjO2kyvBQRBxTWEEGDGq0FlB1BIKtnHY/8= +github.com/muesli/termenv v0.13.0 h1:wK20DRpJdDX8b7Ek2QfhvqhRQFZ237RGRO0RQ/Iqdy0= +github.com/muesli/termenv v0.13.0/go.mod h1:sP1+uffeLaEYpyOTb8pLCUctGcGLnoFjSn4YJK5e2bc= +github.com/olekukonko/tablewriter v0.0.5 h1:P2Ga83D34wi1o9J6Wh1mRuqd4mF/x/lgBS7N7AbDhec= +github.com/olekukonko/tablewriter v0.0.5/go.mod h1:hPp6KlRPjbx+hW8ykQs1w3UBbZlj6HuIJcUGPhkA7kY= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rivo/uniseg v0.1.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY= +github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= +github.com/sashabaranov/go-gpt3 v1.0.0 h1:+wOQUqVpX/JVkNwPu2Il5VbCNefnn5FVKlhFIZq3aAU= +github.com/sashabaranov/go-gpt3 v1.0.0/go.mod h1:BIZdbwdzxZbCrcKGMGH6u2eyGe1xFuX9Anmh3tCP8lQ= +github.com/spf13/cobra v1.6.1 h1:o94oiPyS4KD1mPy2fmcYYHHfCxLqYjJOhGsCHFZtEzA= +github.com/spf13/cobra v1.6.1/go.mod h1:IOw/AERYS7UzyrGinqmz6HLUo219MORXGxhbaJUqzrY= +github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA= +github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.5.2 h1:ALmeCk/px5FSm1MAcFBAsVKZjDuMVj8Tm7FFIlMJnqU= +github.com/yuin/goldmark v1.5.2/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= +github.com/yuin/goldmark-emoji v1.0.1 h1:ctuWEyzGBwiucEqxzwe0SOYDXPAucOrE9NQC18Wa1os= +github.com/yuin/goldmark-emoji v1.0.1/go.mod h1:2w1E6FEWLcDQkoTE+7HU6QF1F6SLlNGjRIBbIZQFqkQ= +golang.org/x/net v0.0.0-20221002022538-bcab6841153b h1:6e93nYa3hNqAvLr0pD4PN1fFS+gKzp2zAXqrnTCstqU= +golang.org/x/net v0.0.0-20221002022538-bcab6841153b/go.mod h1:YDH+HFinaLZZlnHAfSS6ZXJJ9M9t4Dl22yv3iI2vPwk= +golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220728004956-3c1f35247d10/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab h1:2QkjZIsXupsJbJIdSjjUOgWK3aEtzyuh2mPt3l/CkeU= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= +golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/main.go b/main.go new file mode 100644 index 0000000..88e787d --- /dev/null +++ b/main.go @@ -0,0 +1,78 @@ +/* +Copyright © 2023 Kirill Kovalchuk +*/ +package main + +import ( + "context" + "fmt" + "os" + "strings" + + "github.com/nemoden/chat/chat" + "github.com/nemoden/chat/cmd" + "github.com/nemoden/chat/config" + gogpt "github.com/sashabaranov/go-gpt3" +) + +func main() { + args := os.Args[1:] + options := getOptions(args) + positionalArgs := getPotisionalArgs(args) + if len(positionalArgs) > 0 { + subcommands := cmd.GetSubcommandsMap() + // no positional args + _, ok := subcommands[positionalArgs[0]] + // we have no such subcommand, chat implied. + if !ok { + c := config.LoadConfig(options) + input := strings.Join(positionalArgs, " ") + prompt := "" + if c.Format == "markdown" { + prompt += "Return response in markdown format. Prompt on a new line:\n" + } + prompt += input + client, err := chat.NewClient(c) + if err != nil { + fmt.Println(err) + os.Exit(0) + } + ctx := context.Background() + request := gogpt.CompletionRequest{ + Model: gogpt.GPT3TextDavinci003, + Prompt: prompt, + MaxTokens: 1000, + Temperature: 0.5, + } + response, _ := client.GptClient().CreateCompletionStream(ctx, request) + c.Renderer.Render(response) + os.Exit(0) + } + } + cmd.Execute() +} + +func getOptions(args []string) []string { + var opts []string + for _, arg := range args { + if strings.HasPrefix(arg, "--") { + opts = append(opts, arg) + } + } + return opts +} + +func getPotisionalArgs(args []string) []string { + i := 0 + for idx, arg := range args { + if strings.HasPrefix(arg, "--") { + i = idx + 1 + } else { + break + } + } + if len(args) >= i { + return args[i:] + } + return []string{} +} diff --git a/renderer/main.go b/renderer/main.go new file mode 100644 index 0000000..1b78ea0 --- /dev/null +++ b/renderer/main.go @@ -0,0 +1,10 @@ +package renderer + +import ( + gogpt "github.com/sashabaranov/go-gpt3" +) + +type Renderer interface { + // Renders response from ChatGPT API to a file. Returns the output as a string. + Render(stream *gogpt.CompletionStream) string +} diff --git a/renderer/markdown.go b/renderer/markdown.go new file mode 100644 index 0000000..6395f89 --- /dev/null +++ b/renderer/markdown.go @@ -0,0 +1,51 @@ +package renderer + +import ( + "errors" + "fmt" + "io" + "os" + + "github.com/charmbracelet/glamour" + "github.com/gosuri/uilive" + gogpt "github.com/sashabaranov/go-gpt3" +) + +type MarkdownRenderer struct { + out *os.File + prefix string +} + +func NewMarkdownRenderer(out *os.File, prefix string) *MarkdownRenderer { + return &MarkdownRenderer{out, prefix} +} + +func (r *MarkdownRenderer) Render(stream *gogpt.CompletionStream) string { + glamourRenderer, _ := glamour.NewTermRenderer(glamour.WithAutoStyle(), glamour.WithEmoji(), glamour.WithPreservedNewLines()) + writer := uilive.New() + writer.Out = r.out + writer.Start() + defer writer.Stop() + previousResponse := "" + var currentResponse string + wholeResponse := "" + for { + response, err := stream.Recv() + + if errors.Is(err, io.EOF) { + fmt.Fprintf(writer, "EOF: Printing a new line") + fmt.Fprintf(writer, "\n") + previousResponse = "" + break + } + + if len(response.Choices) > 0 { + wholeResponse += response.Choices[0].Text + currentResponse = previousResponse + response.Choices[0].Text + out, _ := glamourRenderer.Render(currentResponse) + fmt.Fprintf(writer, out) + previousResponse = currentResponse + } + } + return wholeResponse +} diff --git a/renderer/passthru.go b/renderer/passthru.go new file mode 100644 index 0000000..973ad9e --- /dev/null +++ b/renderer/passthru.go @@ -0,0 +1,42 @@ +package renderer + +import ( + "bufio" + "errors" + "io" + "os" + + gogpt "github.com/sashabaranov/go-gpt3" +) + +// Passthru renderer simlpy passes thru the response from ChatGPT API and renders it as is. +type PassthruRenderer struct { + out *os.File + prefix string +} + +func NewPassthruRenderer(out *os.File, prefix string) *PassthruRenderer { + return &PassthruRenderer{out, prefix} +} + +func (r *PassthruRenderer) Render(stream *gogpt.CompletionStream) string { + wholeResponse := "" + writer := bufio.NewWriter(r.out) + for { + response, err := stream.Recv() + + if errors.Is(err, io.EOF) { + writer.WriteString("EOF: Printing a new line") + writer.WriteString("\n") + writer.Flush() + break + } + + if len(response.Choices) > 0 { + wholeResponse += response.Choices[0].Text + writer.WriteString(response.Choices[0].Text) + writer.Flush() + } + } + return wholeResponse +}