mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-18 09:36:37 +08:00
f7a1f72d68
The GetAdaptor function in the Adaptor struct has been updated to use the actual model name instead of the origin model name. This change ensures that the correct adaptor is retrieved for processing the response.
62 lines
1.8 KiB
Go
62 lines
1.8 KiB
Go
package meta
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/songquanpeng/one-api/common/ctxkey"
|
|
"github.com/songquanpeng/one-api/model"
|
|
"github.com/songquanpeng/one-api/relay/channeltype"
|
|
"github.com/songquanpeng/one-api/relay/relaymode"
|
|
)
|
|
|
|
type Meta struct {
|
|
Mode int
|
|
ChannelType int
|
|
ChannelId int
|
|
TokenId int
|
|
TokenName string
|
|
UserId int
|
|
Group string
|
|
ModelMapping map[string]string
|
|
BaseURL string
|
|
APIKey string
|
|
APIType int
|
|
Config model.ChannelConfig
|
|
IsStream bool
|
|
// OriginModelName is the model name from the raw user request
|
|
OriginModelName string
|
|
// ActualModelName is the model name after mapping
|
|
ActualModelName string
|
|
RequestURLPath string
|
|
PromptTokens int // only for DoResponse
|
|
ChannelRatio float64
|
|
}
|
|
|
|
func GetByContext(c *gin.Context) *Meta {
|
|
meta := Meta{
|
|
Mode: relaymode.GetByPath(c.Request.URL.Path),
|
|
ChannelType: c.GetInt(ctxkey.Channel),
|
|
ChannelId: c.GetInt(ctxkey.ChannelId),
|
|
TokenId: c.GetInt(ctxkey.TokenId),
|
|
TokenName: c.GetString(ctxkey.TokenName),
|
|
UserId: c.GetInt(ctxkey.Id),
|
|
Group: c.GetString(ctxkey.Group),
|
|
ModelMapping: c.GetStringMapString(ctxkey.ModelMapping),
|
|
OriginModelName: c.GetString(ctxkey.RequestModel),
|
|
BaseURL: c.GetString(ctxkey.BaseURL),
|
|
APIKey: strings.TrimPrefix(c.Request.Header.Get("Authorization"), "Bearer "),
|
|
RequestURLPath: c.Request.URL.String(),
|
|
ChannelRatio: c.GetFloat64(ctxkey.ChannelRatio), // add by Laisky
|
|
}
|
|
cfg, ok := c.Get(ctxkey.Config)
|
|
if ok {
|
|
meta.Config = cfg.(model.ChannelConfig)
|
|
}
|
|
if meta.BaseURL == "" {
|
|
meta.BaseURL = channeltype.ChannelBaseURLs[meta.ChannelType]
|
|
}
|
|
meta.APIType = channeltype.ToAPIType(meta.ChannelType)
|
|
return &meta
|
|
}
|