Spaces:
Running
Running
File size: 1,885 Bytes
d669ddb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
package v1
import (
"adams549659584/go-proxy-bingai/common"
"encoding/json"
"net/http"
binglib "github.com/Harry-zklcdc/bing-lib"
)
func ModelsHandler(w http.ResponseWriter, r *http.Request) {
w.Header().Add("Access-Control-Allow-Origin", "*")
w.Header().Add("Access-Control-Allow-Methods", "*")
w.Header().Add("Access-Control-Allow-Headers", "*")
if apikey != "" {
if r.Header.Get("Authorization") != "Bearer "+apikey {
w.WriteHeader(http.StatusUnauthorized)
w.Write([]byte("Unauthorized"))
return
}
}
models := []modelStruct{
{
Id: DALL_E_3,
Object: "model",
Created: 1687579610,
OwnedBy: "Go-Proxy-BingAI",
},
{
Id: GPT_35_TURBO,
Object: "model",
Created: 1687579610,
OwnedBy: "Go-Proxy-BingAI",
},
{
Id: GPT_4_TURBO_PREVIEW,
Object: "model",
Created: 1687579610,
OwnedBy: "Go-Proxy-BingAI",
},
{
Id: GPT_35_TURBO_16K,
Object: "model",
Created: 1687579610,
OwnedBy: "Go-Proxy-BingAI",
},
{
Id: GPT_4_32K,
Object: "model",
Created: 1687579610,
OwnedBy: "Go-Proxy-BingAI",
},
{
Id: GPT_4_VISION,
Object: "model",
Created: 1687579610,
OwnedBy: "Go-Proxy-BingAI",
},
}
for _, model := range binglib.ChatModels {
models = append(models, modelStruct{
Id: model,
Object: "model",
Created: 1687579610,
OwnedBy: "Go-Proxy-BingAI",
}, modelStruct{
Id: model + "-vision",
Object: "model",
Created: 1687579610,
OwnedBy: "Go-Proxy-BingAI",
})
}
resp := modelResponse{
Object: "list",
Data: models,
}
respData, err := json.Marshal(resp)
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
w.Write([]byte(err.Error()))
common.Logger.Error("ModelsHandler Marshal Error: %v", err)
return
}
w.Header().Set("Content-Type", "application/json")
w.Write(respData)
}
|