mirror of
https://github.com/stulzq/azure-openai-proxy.git
synced 2025-12-19 07:14:21 +01:00
c40e27c3b4
* add: ability to mimic `/v1/models/` api * fix: missing json dependency * fix: accidentally remove dependency * fix: linter error caught by gh workflow
30 lines
1.0 KiB
Go
30 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/spf13/viper"
|
|
"github.com/stulzq/azure-openai-proxy/azure"
|
|
)
|
|
|
|
// registerRoute registers all routes
|
|
func registerRoute(r *gin.Engine) {
|
|
// https://platform.openai.com/docs/api-reference
|
|
r.HEAD("/", func(c *gin.Context) {
|
|
c.Status(200)
|
|
})
|
|
r.Any("/health", func(c *gin.Context) {
|
|
c.Status(200)
|
|
})
|
|
apiBase := viper.GetString("api_base")
|
|
stripPrefixConverter := azure.NewStripPrefixConverter(apiBase)
|
|
r.GET(stripPrefixConverter.Prefix+"/models", azure.ModelProxy)
|
|
templateConverter := azure.NewTemplateConverter("/openai/deployments/{{.DeploymentName}}/embeddings")
|
|
apiBasedRouter := r.Group(apiBase)
|
|
{
|
|
apiBasedRouter.Any("/engines/:model/embeddings", azure.ProxyWithConverter(templateConverter))
|
|
apiBasedRouter.Any("/completions", azure.ProxyWithConverter(stripPrefixConverter))
|
|
apiBasedRouter.Any("/chat/completions", azure.ProxyWithConverter(stripPrefixConverter))
|
|
apiBasedRouter.Any("/embeddings", azure.ProxyWithConverter(stripPrefixConverter))
|
|
}
|
|
}
|