AI

main.go

package main

import (
    "context"
    "errors"
    "fmt"
    "io"
    "os"

    "github.com/sashabaranov/go-openai"
)

func main() {
    inputb, err := io.ReadAll(os.Stdin)
    if err != nil {
        panic(err)
    }
    input := string(inputb)

    client := openai.NewClient("OPENAI_API_KEY")
    resp, err := client.CreateChatCompletionStream(context.Background(), openai.ChatCompletionRequest{
        Model: openai.GPT3Dot5Turbo,
        Messages: []openai.ChatCompletionMessage{
            {
                Role:    openai.ChatMessageRoleSystem,
                Content: fmt.Sprintf("%s:\n%s", input, os.Args[1:]),
            },
        },
    })
    if err != nil {
        panic(err)
    }

    for {
        resp, err := resp.Recv()
        if err != nil {
            if errors.Is(err, io.EOF) {
                return
            }
            panic(err)
        }
        if resp.Choices[0].FinishReason == "stop" {
            return
        }
        fmt.Print(resp.Choices[0].Delta.Content)
    }
}