From baeedc699d0845e39213e692b3f190a79aa9e8c9 Mon Sep 17 00:00:00 2001
From: balibabu <cike8899@users.noreply.github.com>
Date: Wed, 17 Jul 2024 09:45:10 +0800
Subject: [PATCH] fix: add group id field to ApiKeyModal #1353 (#1540)

### What problem does this PR solve?

fix: add group id field to ApiKeyModal #1353
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 web/src/pages/user-setting/interface.ts               |  5 +++++
 .../setting-model/api-key-modal/index.tsx             | 11 +++++++++--
 web/src/pages/user-setting/setting-model/hooks.ts     |  6 +++---
 3 files changed, 17 insertions(+), 5 deletions(-)
 create mode 100644 web/src/pages/user-setting/interface.ts

diff --git a/web/src/pages/user-setting/interface.ts b/web/src/pages/user-setting/interface.ts
new file mode 100644
index 00000000..1c9114ee
--- /dev/null
+++ b/web/src/pages/user-setting/interface.ts
@@ -0,0 +1,5 @@
+export interface ApiKeyPostBody {
+  api_key: string;
+  base_url: string;
+  group_id?: string;
+}
diff --git a/web/src/pages/user-setting/setting-model/api-key-modal/index.tsx b/web/src/pages/user-setting/setting-model/api-key-modal/index.tsx
index 14d3fa94..ca5a843e 100644
--- a/web/src/pages/user-setting/setting-model/api-key-modal/index.tsx
+++ b/web/src/pages/user-setting/setting-model/api-key-modal/index.tsx
@@ -2,18 +2,20 @@ import { IModalManagerChildrenProps } from '@/components/modal-manager';
 import { useTranslate } from '@/hooks/commonHooks';
 import { Form, Input, Modal } from 'antd';
 import { useEffect } from 'react';
+import { ApiKeyPostBody } from '../../interface';
 
 interface IProps extends Omit<IModalManagerChildrenProps, 'showModal'> {
   loading: boolean;
   initialValue: string;
   llmFactory: string;
-  onOk: (name: string, baseUrl: string) => void;
+  onOk: (postBody: ApiKeyPostBody) => void;
   showModal?(): void;
 }
 
 type FieldType = {
   api_key?: string;
   base_url?: string;
+  group_id?: string;
 };
 
 const modelsWithBaseUrl = ['OpenAI', 'Azure-OpenAI'];
@@ -32,7 +34,7 @@ const ApiKeyModal = ({
   const handleOk = async () => {
     const ret = await form.validateFields();
 
-    return onOk(ret.api_key, ret.base_url);
+    return onOk(ret);
   };
 
   useEffect(() => {
@@ -75,6 +77,11 @@ const ApiKeyModal = ({
             <Input placeholder="https://api.openai.com/v1" />
           </Form.Item>
         )}
+        {llmFactory === 'Minimax' && (
+          <Form.Item<FieldType> label={'Group ID'} name="group_id">
+            <Input />
+          </Form.Item>
+        )}
       </Form>
     </Modal>
   );
diff --git a/web/src/pages/user-setting/setting-model/hooks.ts b/web/src/pages/user-setting/setting-model/hooks.ts
index dfdce7e2..460ca8e5 100644
--- a/web/src/pages/user-setting/setting-model/hooks.ts
+++ b/web/src/pages/user-setting/setting-model/hooks.ts
@@ -16,6 +16,7 @@ import {
 } from '@/hooks/userSettingHook';
 import { IAddLlmRequestBody } from '@/interfaces/request/llm';
 import { useCallback, useEffect, useState } from 'react';
+import { ApiKeyPostBody } from '../interface';
 
 type SavingParamsState = Omit<IApiKeySavingParams, 'api_key'>;
 
@@ -31,11 +32,10 @@ export const useSubmitApiKey = () => {
   } = useSetModalState();
 
   const onApiKeySavingOk = useCallback(
-    async (apiKey: string, baseUrl: string) => {
+    async (postBody: ApiKeyPostBody) => {
       const ret = await saveApiKey({
         ...savingParams,
-        api_key: apiKey,
-        base_url: baseUrl,
+        ...postBody,
       });
 
       if (ret === 0) {