xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-12-26 01:35:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge commit 'af543ab8ecb6827cbbc151c2cff181cdc3286274'

Browse Source
This commit is contained in:
Laisky.Cai
2024-04-08 01:13:00 +00:00
parent bf63b1e025 af543ab8ec
commit 9cb3a06dd5
49 changed files with 1024 additions and 335 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
relay/adaptor/gemini/adaptor.go
View File

@@ -7,6 +7,7 @@ import (
"github.com/Laisky/errors/v2"
"github.com/gin-gonic/gin"
"github.com/songquanpeng/one-api/common/config"
"github.com/songquanpeng/one-api/common/helper"
channelhelper "github.com/songquanpeng/one-api/relay/adaptor"
"github.com/songquanpeng/one-api/relay/adaptor/openai"
@@ -22,7 +23,7 @@ func (a *Adaptor) Init(meta *meta.Meta) {
}
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
version := helper.AssignOrDefault(meta.APIVersion, "v1")
version := helper.AssignOrDefault(meta.APIVersion, config.GeminiVersion)
action := "generateContent"
if meta.IsStream {
action = "streamGenerateContent"

4
relay/billing/ratio/model.go
View File

@@ -155,6 +155,10 @@ var ModelRatio = map[string]float64{
"yi-34b-chat-0205": 2.5 / 1000 * RMB,
"yi-34b-chat-200k": 12.0 / 1000 * RMB,
"yi-vl-plus": 6.0 / 1000 * RMB,
// stepfun todo
"step-1v-32k": 0.024 * RMB,
"step-1-32k": 0.024 * RMB,
"step-1-200k": 0.15 * RMB,
}
var CompletionRatio = map[string]float64{}