deepinfra/api.go

package deepinfra

import (
	"errors"
	"io"
	"net/http"
	"strings"
)

const (
	APIDeepInfra = "https://api.deepinfra.com/v1/openai/chat/completions"
)

type API struct {
	api string // api to call
	key string // key in Authorization: Bearer
}

func NewAPI(api, key string) API {
	return API{api: api, key: key}
}

func (api *API) Request(model Model) (string, error) {
	req, err := http.NewRequest("POST", api.api, model.Body())
	if err != nil {
		return "", err
	}
	req.Header.Add("Content-Type", "application/json")
	req.Header.Add("Authorization", "Bearer "+api.key)
	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusOK {
		sb := strings.Builder{}
		sb.WriteString(resp.Status)
		sb.WriteByte(' ')
		_, _ = io.Copy(&sb, resp.Body)
		return "", errors.New(sb.String())
	}
	err = model.Parse(resp.Body)
	if err != nil {
		return "", err
	}
	return model.Output(), nil
}