OpenAI GPT-3 API wrapper for Go
Installation:
go get github.com/sashabaranov/go-gpt3
Example usage:
package main
import (
"context"
"fmt"
gogpt "github.com/sashabaranov/go-gpt3"
)
func main() {
c := gogpt.NewClient("your token")
ctx := context.Background()
req := gogpt.CompletionRequest{
MaxTokens: 5,
Prompt: "Lorem ipsum",
}
resp, err := c.CreateCompletion(ctx, "ada", req)
if err != nil {
return
}
fmt.Println(resp.Choices[0].Text)
searchReq := gogpt.SearchRequest{
Documents: []string{"White House", "hospital", "school"},
Query: "the president",
}
searchResp, err := c.Search(ctx, "ada", searchReq)
if err != nil {
return
}
fmt.Println(searchResp.SearchResults)
}