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)
This commit is contained in:
balibabu 2024-07-17 09:45:10 +08:00 committed by GitHub
parent 00943dc04a
commit baeedc699d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 17 additions and 5 deletions

View File

@ -0,0 +1,5 @@
export interface ApiKeyPostBody {
api_key: string;
base_url: string;
group_id?: string;
}

View File

@ -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>
);

View File

@ -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) {