fumiama/deepinfra
1
0
Fork 0
mirror of https://github.com/fumiama/deepinfra.git synced 2026-06-09 12:30:31 +08:00
Code Activity

feat: customizable client

Browse Source
This commit is contained in:
源文雨
2025-03-30 01:56:13 +09:00
parent db7712b6d2
commit 9a5ef962fa
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
api.go
View File

@@ -15,19 +15,28 @@ const (
type API struct { type API struct {
api string // api to call api string // api to call
key string // key in Authorization: Bearer key string // key in Authorization: Bearer
cli *http.Client
} }
func NewAPI(api, key string) API { func NewAPI(api, key string) API {
return API{api: api, key: key} return API{api: api, key: key}
} }
func (api *API) SetHTTPClient(c *http.Client) {
api.cli = c
}
func (api *API) Request(model Model) (string, error) { func (api *API) Request(model Model) (string, error) {
req, err := http.NewRequest("POST", model.API(api.api, api.key), model.Body()) req, err := http.NewRequest("POST", model.API(api.api, api.key), model.Body())
if err != nil { if err != nil {
return "", err return "", err
} }
model.Header(api.key, req.Header) model.Header(api.key, req.Header)
resp, err := http.DefaultClient.Do(req) cli := http.DefaultClient
if api.cli != nil {
cli = api.cli
}
resp, err := cli.Do(req)
if err != nil { if err != nil {
return "", err return "", err
} }