mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-18 01:26:37 +08:00
Merge remote-tracking branch 'origin/upstream/main'
This commit is contained in:
commit
c1418df1fd
@ -63,20 +63,36 @@ func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
|
|||||||
suffix := ""
|
suffix := ""
|
||||||
if strings.HasPrefix(meta.ActualModelName, "gemini") {
|
if strings.HasPrefix(meta.ActualModelName, "gemini") {
|
||||||
if meta.IsStream {
|
if meta.IsStream {
|
||||||
suffix = "streamGenerateContent"
|
suffix = "streamGenerateContent?alt=sse"
|
||||||
} else {
|
} else {
|
||||||
suffix = "generateContent"
|
suffix = "generateContent"
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if meta.IsStream {
|
if meta.IsStream {
|
||||||
suffix = "streamRawPredict"
|
suffix = "streamRawPredict?alt=sse"
|
||||||
} else {
|
} else {
|
||||||
suffix = "rawPredict"
|
suffix = "rawPredict"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
baseUrl := fmt.Sprintf("https://%s-aiplatform.googleapis.com/v1/projects/%s/locations/%s/publishers/google/models/%s:%s", meta.Config.Region, meta.Config.VertexAIProjectID, meta.Config.Region, meta.ActualModelName, suffix)
|
if meta.BaseURL != "" {
|
||||||
return baseUrl, nil
|
return fmt.Sprintf(
|
||||||
|
"%s/v1/projects/%s/locations/%s/publishers/google/models/%s:%s",
|
||||||
|
meta.BaseURL,
|
||||||
|
meta.Config.VertexAIProjectID,
|
||||||
|
meta.Config.Region,
|
||||||
|
meta.ActualModelName,
|
||||||
|
suffix,
|
||||||
|
), nil
|
||||||
|
}
|
||||||
|
return fmt.Sprintf(
|
||||||
|
"https://%s-aiplatform.googleapis.com/v1/projects/%s/locations/%s/publishers/google/models/%s:%s",
|
||||||
|
meta.Config.Region,
|
||||||
|
meta.Config.VertexAIProjectID,
|
||||||
|
meta.Config.Region,
|
||||||
|
meta.ActualModelName,
|
||||||
|
suffix,
|
||||||
|
), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {
|
func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {
|
||||||
|
@ -31,6 +31,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
|
|||||||
AnthropicVersion: anthropicVersion,
|
AnthropicVersion: anthropicVersion,
|
||||||
// Model: claudeReq.Model,
|
// Model: claudeReq.Model,
|
||||||
Messages: claudeReq.Messages,
|
Messages: claudeReq.Messages,
|
||||||
|
System: claudeReq.System,
|
||||||
MaxTokens: claudeReq.MaxTokens,
|
MaxTokens: claudeReq.MaxTokens,
|
||||||
Temperature: claudeReq.Temperature,
|
Temperature: claudeReq.Temperature,
|
||||||
TopP: claudeReq.TopP,
|
TopP: claudeReq.TopP,
|
||||||
|
Loading…
Reference in New Issue
Block a user