fumiama/deepinfra
1
0
Fork 0
mirror of https://github.com/fumiama/deepinfra.git synced 2026-06-17 01:40:34 +08:00
Code Activity

feat: add Google GenAI API

Browse Source
This commit is contained in:
源文雨
2025-03-30 01:48:52 +09:00
parent bf0f65bcaa
commit db7712b6d2
7 changed files with 153 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api.go
View File

@@ -8,7 +8,8 @@ import (
)
const (
APIDeepInfra = "https://api.deepinfra.com/v1/openai/chat/completions"
OpenAIDeepInfra = "https://api.deepinfra.com/v1/openai/chat/completions"
GenAIGoogle = "https://generativelanguage.googleapis.com/v1beta"
)
type API struct {
@@ -21,12 +22,11 @@ func NewAPI(api, key string) API {
}
func (api *API) Request(model Model) (string, error) {
req, err := http.NewRequest("POST", api.api, model.Body())
req, err := http.NewRequest("POST", model.API(api.api, api.key), model.Body())
if err != nil {
return "", err
}
req.Header.Add("Content-Type", "application/json")
req.Header.Add("Authorization", "Bearer "+api.key)
model.Header(api.key, req.Header)
resp, err := http.DefaultClient.Do(req)
if err != nil {
return "", err