chatgpt-cli/cmd/chatgpt/main.go

package main

import (
	"errors"
	"fmt"
	"github.com/kardolus/chatgpt-poc/client"
	"github.com/kardolus/chatgpt-poc/http"
	"log"
	"os"
	"strings"
)

const secretEnv = "CHAT_GPT_SECRET_KEY"

func main() {
	exit(run())
}

func exit(err error) {
	if err == nil {
		os.Exit(0)
	}
	log.Printf("Error: %s\n", err)
	os.Exit(1)
}

func run() error {
	if len(os.Args) <= 1 {
		return errors.New("you must specify your query")
	}

	secret := os.Getenv(secretEnv)
	if secret == "" {
		return errors.New("missing environment variable: " + secretEnv)
	}
	client := client.New(http.New().WithSecret(secret))

	result, err := client.Query(strings.Join(os.Args[1:], " "))
	if err != nil {
		return err
	}
	fmt.Println(result)

	return nil
}