diff --git a/conversion/requests/chatgpt/convert.go b/conversion/requests/chatgpt/convert.go index 49e25bf..0105e25 100644 --- a/conversion/requests/chatgpt/convert.go +++ b/conversion/requests/chatgpt/convert.go @@ -18,6 +18,10 @@ func ConvertAPIRequest(api_request official_types.APIRequest, account string, se chatgpt_request.Model = "gpt-4o" } else if strings.HasPrefix(api_request.Model, "gpt-4") { chatgpt_request.Model = "gpt-4" + } else if strings.HasPrefix(api_request.Model, "o1-preview") { + chatgpt_request.Model = "o1-preview" + } else if strings.HasPrefix(api_request.Model, "o1-mini") { + chatgpt_request.Model = "o1-mini" } matches := gptsRegexp.FindStringSubmatch(api_request.Model) if len(matches) == 2 { diff --git a/handlers.go b/handlers.go index c225081..a36c938 100644 --- a/handlers.go +++ b/handlers.go @@ -84,6 +84,18 @@ func simulateModel(c *gin.Context) { "created": 1688888888, "owned_by": "chatgpt-to-api", }, + { + "id": "o1-preview", + "object": "model", + "created": 1688888888, + "owned_by": "chatgpt-to-api", + }, + { + "id": "o1-mini", + "object": "model", + "created": 1688888888, + "owned_by": "chatgpt-to-api", + }, }, }) }