models-example.go

package main

import (
	"encoding/json"
	"fmt"
	"os"

	"github.com/Kardbord/gopenai/authentication"
	"github.com/Kardbord/gopenai/models"
	_ "github.com/joho/godotenv/autoload"
)

const OpenAITokenEnv = "OPENAI_API_KEY"

func init() {
	key := os.Getenv(OpenAITokenEnv)
	authentication.SetAPIKey(key)
}

func main() {
	listresp, err := models.MakeListModelsRequest(nil)
	if err != nil {
		fmt.Println(err)
		return
	}

	fmt.Printf("Retreived information for %d models. Calling the \"Retrieve\" endpoint with the first one.\n", len(listresp.Data))
	resp, err := models.MakeRetrieveModelRequest(listresp.Data[0].ID, nil)
	if err != nil {
		fmt.Println(err)
		return
	}

	jsonResp, err := json.MarshalIndent(resp, "", "  ")
	if err != nil {
		fmt.Printf(`Failed to marshal response to JSON for printing, err="%s"\n`, err)
		return
	}
	fmt.Printf("Retrieved Model Info: %s\n", string(jsonResp))
}