CaIon 2 лет назад
Родитель
Сommit
07d653aedb
4 измененных файлов с 12 добавлено и 1 удалено
  1. 1 0
      common/model-ratio.go
  2. 9 0
      controller/model.go
  3. 1 1
      docker-compose.yml
  4. 1 0
      web/src/constants/channel.constants.js

+ 1 - 0
common/model-ratio.go

@@ -14,6 +14,7 @@ import (
 // 1 === $0.002 / 1K tokens
 // 1 === ¥0.014 / 1k tokens
 var ModelRatio = map[string]float64{
+	"midjourney":                50,
 	"gpt-4":                     15,
 	"gpt-4-0314":                15,
 	"gpt-4-0613":                15,

+ 9 - 0
controller/model.go

@@ -54,6 +54,15 @@ func init() {
 	})
 	// https://platform.openai.com/docs/models/model-endpoint-compatibility
 	openAIModels = []OpenAIModels{
+		{
+			Id:         "midjourney",
+			Object:     "model",
+			Created:    1677649963,
+			OwnedBy:    "Midjourney",
+			Permission: permission,
+			Root:       "midjourney",
+			Parent:     nil,
+		},
 		{
 			Id:         "dall-e-2",
 			Object:     "model",

+ 1 - 1
docker-compose.yml

@@ -2,7 +2,7 @@ version: '3.4'
 
 services:
   one-api:
-    image: justsong/one-api:latest
+    image: calciumion/neko-api:main
     container_name: one-api
     restart: always
     command: --log-dir /app/logs

+ 1 - 0
web/src/constants/channel.constants.js

@@ -1,5 +1,6 @@
 export const CHANNEL_OPTIONS = [
   { key: 1, text: 'OpenAI', value: 1, color: 'green' },
+  { key: 99, text: 'Midjourney-Proxy', value: 99, color: 'green' },
   { key: 14, text: 'Anthropic Claude', value: 14, color: 'black' },
   { key: 3, text: 'Azure OpenAI', value: 3, color: 'olive' },
   { key: 11, text: 'Google PaLM2', value: 11, color: 'orange' },