router.go

package router

import (
	"github.com/gin-gonic/gin"
	"github.com/soulteary/amazing-openai-api/models/azure"
	"github.com/soulteary/amazing-openai-api/models/gemini"
	"github.com/soulteary/amazing-openai-api/models/yi"
)

func RegisterModelRoute(r *gin.Engine, serviceType string) {
	// https://platform.openai.com/docs/api-reference
	apiBase := "/v1"

	switch serviceType {
	case "azure":
		stripPrefixConverter := azure.NewStripPrefixConverter(apiBase)
		r.GET(stripPrefixConverter.Prefix+"/models", azure.ModelProxy)
		apiBasedRouter := r.Group(apiBase)
		{
			apiBasedRouter.Any("/completions", azure.ProxyWithConverter(stripPrefixConverter))
			apiBasedRouter.Any("/chat/completions", azure.ProxyWithConverter(stripPrefixConverter))
		}
	case "yi":
		stripPrefixConverter := yi.NewStripPrefixConverter(apiBase)
		apiBasedRouter := r.Group(apiBase)
		{
			apiBasedRouter.Any("/completions", yi.ProxyWithConverter(stripPrefixConverter))
			apiBasedRouter.Any("/chat/completions", yi.ProxyWithConverter(stripPrefixConverter))
		}
	case "gemini":
		stripPrefixConverter := gemini.NewStripPrefixConverter(apiBase)
		apiBasedRouter := r.Group(apiBase)
		{
			apiBasedRouter.Any("/completions", gemini.ProxyWithConverter(stripPrefixConverter))
			apiBasedRouter.Any("/chat/completions", gemini.ProxyWithConverter(stripPrefixConverter))
		}
	}
}