117 lines
2.0 KiB
Go
117 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
util "github.com/olemorud/chatgpt-cli/v2"
|
|
openai "github.com/sashabaranov/go-openai"
|
|
)
|
|
|
|
func main() {
|
|
env, err := util.ReadEnvFile(".env")
|
|
|
|
if err != nil {
|
|
fmt.Println("failed to read .env", err)
|
|
}
|
|
|
|
// parse command line arguments
|
|
token := env["OPENAI_API_KEY"]
|
|
|
|
model := *flag.String("model", openai.GPT3Dot5Turbo,
|
|
"OpenAI Model to use.\n"+
|
|
"List of models:\n"+
|
|
"https://platform.openai.com/docs/models/overview\n")
|
|
|
|
flag.Parse()
|
|
|
|
args := flag.Args()
|
|
|
|
if len(args) == 0 {
|
|
err = runInteractive(token, model)
|
|
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
} else {
|
|
query := strings.Join(args, " ")
|
|
err = askGpt(token, model, query)
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func askGpt(token string, model string, query string) error {
|
|
client := openai.NewClient("your token")
|
|
|
|
resp, err := client.CreateChatCompletion(
|
|
context.Background(),
|
|
openai.ChatCompletionRequest{
|
|
Model: model,
|
|
Messages: []openai.ChatCompletionMessage{
|
|
{
|
|
Role: openai.ChatMessageRoleUser,
|
|
Content: query,
|
|
},
|
|
},
|
|
},
|
|
)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(resp.Choices[0].Message.Content)
|
|
|
|
return nil
|
|
}
|
|
|
|
func runInteractive(token string, model string) error {
|
|
client := openai.NewClient("your token")
|
|
messages := make([]openai.ChatCompletionMessage, 0)
|
|
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
|
|
fmt.Println("ChatGPT", model, "interactive mode")
|
|
|
|
fmt.Println("->")
|
|
|
|
for scanner.Scan() {
|
|
|
|
text := scanner.Text()
|
|
|
|
messages = append(messages, openai.ChatCompletionMessage{
|
|
Role: openai.ChatMessageRoleUser,
|
|
Content: text,
|
|
})
|
|
|
|
resp, err := client.CreateChatCompletion(
|
|
context.Background(),
|
|
openai.ChatCompletionRequest{
|
|
Model: model,
|
|
Messages: messages,
|
|
},
|
|
)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(resp.Choices[0].Message.Content)
|
|
|
|
fmt.Println("->")
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
fmt.Fprintln(os.Stderr, "reading standard input:", err)
|
|
}
|
|
|
|
return nil
|
|
}
|