diff --git a/common/model-ratio.go b/common/model-ratio.go index bb2adc7..f1cc07d 100644 --- a/common/model-ratio.go +++ b/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, diff --git a/controller/model.go b/controller/model.go index f990433..201d643 100644 --- a/controller/model.go +++ b/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", diff --git a/docker-compose.yml b/docker-compose.yml index 9b814a0..6c5350d 100644 --- a/docker-compose.yml +++ b/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 diff --git a/web/src/components/SiderBar.js b/web/src/components/SiderBar.js index fc22b81..95298c9 100644 --- a/web/src/components/SiderBar.js +++ b/web/src/components/SiderBar.js @@ -15,7 +15,7 @@ import { IconLayers, IconSetting, IconCreditCard, - IconSemiLogo, + IconComment, IconHome, IconImage } from '@douyinfe/semi-icons'; @@ -36,7 +36,13 @@ let headerButtons = [ icon: , className: isAdmin()?'semi-navigation-item-normal':'tableHiddle', }, - + { + text: '聊天', + itemKey: 'chat', + to: '/chat', + icon: , + className: localStorage.getItem('chat_link')?'semi-navigation-item-normal':'tableHiddle', + }, { text: '令牌', itemKey: 'token', @@ -89,14 +95,6 @@ let headerButtons = [ // } ]; -if (localStorage.getItem('chat_link')) { - headerButtons.splice(1, 0, { - name: '聊天', - to: '/chat', - icon: 'comments' - }); -} - const HeaderBar = () => { const [userState, userDispatch] = useContext(UserContext); let navigate = useNavigate(); @@ -134,6 +132,7 @@ const HeaderBar = () => { midjourney: "/midjourney", setting: "/setting", about: "/about", + chat: "/chat", }; return (