Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fix wrong implementation for /v1/models (close #128)

Browse Source
This commit is contained in:
JustSong 2023-05-31 14:43:29 +08:00
parent 54215dc303
commit fa71daa8a7
1 changed files with 14 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
controller/model.go
View File

@ -23,20 +23,21 @@ type OpenAIModelPermission struct {
} }
type OpenAIModels struct { type OpenAIModels struct {
Id string `json:"id"` Id string `json:"id"`
Object string `json:"object"` Object string `json:"object"`
Created int `json:"created"` Created int `json:"created"`
OwnedBy string `json:"owned_by"` OwnedBy string `json:"owned_by"`
Permission OpenAIModelPermission `json:"permission"` Permission []OpenAIModelPermission `json:"permission"`
Root string `json:"root"` Root string `json:"root"`
Parent *string `json:"parent"` Parent *string `json:"parent"`
} }
var openAIModels []OpenAIModels var openAIModels []OpenAIModels
var openAIModelsMap map[string]OpenAIModels var openAIModelsMap map[string]OpenAIModels
func init() { func init() {
permission := OpenAIModelPermission{ var permission []OpenAIModelPermission
permission = append(permission, OpenAIModelPermission{
Id: "modelperm-LwHkVFn8AcMItP432fKKDIKJ", Id: "modelperm-LwHkVFn8AcMItP432fKKDIKJ",
Object: "model_permission", Object: "model_permission",
Created: 1626777600, Created: 1626777600,
@ -49,7 +50,7 @@ func init() {
Organization: "*", Organization: "*",
Group: nil, Group: nil,
IsBlocking: false, IsBlocking: false,
} })
// https://platform.openai.com/docs/models/model-endpoint-compatibility // https://platform.openai.com/docs/models/model-endpoint-compatibility
openAIModels = []OpenAIModels{ openAIModels = []OpenAIModels{
{ {
@ -106,15 +107,6 @@ func init() {
Root: "gpt-4-32k-0314", Root: "gpt-4-32k-0314",
Parent: nil, Parent: nil,
}, },
{
Id: "gpt-3.5-turbo",
Object: "model",
Created: 1677649963,
OwnedBy: "openai",
Permission: permission,
Root: "gpt-3.5-turbo",
Parent: nil,
},
{ {
Id: "text-embedding-ada-002", Id: "text-embedding-ada-002",
Object: "model", Object: "model",
@ -132,7 +124,10 @@ func init() {
} }
func ListModels(c *gin.Context) { func ListModels(c *gin.Context) {
c.JSON(200, openAIModels) c.JSON(200, gin.H{
"object": "list",
"data": openAIModels,
})
} }
func RetrieveModel(c *gin.Context) { func RetrieveModel(c *gin.Context) {