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 (