Merge pull request #14 from Kerwin1202/main

fix: the model was not saved successfully(Close Chanzhaoyu#152)
EvanDbg · May 22, 2023 · 19b5d25 · 19b5d25
2 parents b80db6b + 1b120f8
commit 19b5d25
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/service/src/index.ts b/service/src/index.ts
@@ -649,7 +649,7 @@ router.post('/user-info', auth, async (req, res) => {
 
 router.post('/user-chat-model', auth, async (req, res) => {
   try {
-    const { chatModel } = req.query as { chatModel: CHATMODEL }
+    const { chatModel } = req.body as { chatModel: CHATMODEL }
     const userId = req.headers.userId.toString()
 
     const user = await getUserById(userId)

diff --git a/service/src/storage/mongo.ts b/service/src/storage/mongo.ts
@@ -223,10 +223,11 @@ export async function getUsers(page: number, size: number): Promise<{ users: Use
 
 export async function getUserById(userId: string): Promise<UserInfo> {
   const userInfo = await userCol.findOne({ _id: new ObjectId(userId) }) as UserInfo
-  if (userInfo.config == null) {
+  if (userInfo.config == null)
     userInfo.config = new UserConfig()
+  if (userInfo.config.chatModel == null)
     userInfo.config.chatModel = 'gpt-3.5-turbo'
-  }
+
   return userInfo
 }
 

diff --git a/src/components/common/Setting/index.vue b/src/components/common/Setting/index.vue
@@ -100,7 +100,7 @@ const show = computed({
           </template>
           <Audit />
         </NTabPane>
-        <NTabPane name="UserConfig" tab="UserConfig">
+        <NTabPane v-if="userStore.userInfo.root" name="UserConfig" tab="UserConfig">
           <template #tab>
             <SvgIcon class="text-lg" icon="ri-user-5-line" />
             <span class="ml-2">{{ $t('setting.userConfig') }}</span>