Просмотр исходного кода

Merge pull request #42 from Tailen/main

feat: 添加 davinci-002 和 babbage-002 模型
Calcium-Ion 2 лет назад
Родитель
Сommit
aa6d623981
2 измененных файлов с 20 добавлено и 0 удалено
  1. 2 0
      common/model-ratio.go
  2. 18 0
      controller/model.go

+ 2 - 0
common/model-ratio.go

@@ -32,6 +32,8 @@ var ModelRatio = map[string]float64{
 	"gpt-3.5-turbo-16k-0613":    1.5,
 	"gpt-3.5-turbo-instruct":    0.75, // $0.0015 / 1K tokens
 	"gpt-3.5-turbo-1106":        0.5,  // $0.001 / 1K tokens
+	"babbage-002":               0.2,  // $0.0004 / 1K tokens
+	"davinci-002":               1,    // $0.002 / 1K tokens
 	"text-ada-001":              0.2,
 	"text-babbage-001":          0.25,
 	"text-curie-001":            1,

+ 18 - 0
controller/model.go

@@ -360,6 +360,24 @@ func init() {
 			Root:       "code-davinci-edit-001",
 			Parent:     nil,
 		},
+		{
+			Id:         "babbage-002",
+			Object:     "model",
+			Created:    1677649963,
+			OwnedBy:    "openai",
+			Permission: permission,
+			Root:       "babbage-002",
+			Parent:     nil,
+		},
+		{
+			Id:         "davinci-002",
+			Object:     "model",
+			Created:    1677649963,
+			OwnedBy:    "openai",
+			Permission: permission,
+			Root:       "davinci-002",
+			Parent:     nil,
+		},
 		{
 			Id:         "claude-instant-1",
 			Object:     "model",