Просмотр исходного кода

feat: add base_url to ApiKeyModal (#167)

tags/v0.1.0
balibabu 1 год назад
Родитель
Сommit
7bae41c71f
Аккаунт пользователя с таким Email не найден

+ 1
- 1
web/src/hooks/llmHooks.ts Просмотреть файл

api_key: string; api_key: string;
llm_name?: string; llm_name?: string;
model_type?: string; model_type?: string;
api_base?: string;
base_url?: string;
} }


export const useSaveApiKey = () => { export const useSaveApiKey = () => {

+ 17
- 4
web/src/pages/user-setting/setting-model/api-key-modal/index.tsx Просмотреть файл

interface IProps extends Omit<IModalManagerChildrenProps, 'showModal'> { interface IProps extends Omit<IModalManagerChildrenProps, 'showModal'> {
loading: boolean; loading: boolean;
initialValue: string; initialValue: string;
onOk: (name: string) => void;
llmFactory: string;
onOk: (name: string, baseUrl: string) => void;
showModal?(): void; showModal?(): void;
} }


type FieldType = { type FieldType = {
api_key?: string; api_key?: string;
base_url?: string;
}; };


const ApiKeyModal = ({ const ApiKeyModal = ({
visible, visible,
hideModal, hideModal,
llmFactory,
loading, loading,
initialValue, initialValue,
onOk, onOk,
const handleOk = async () => { const handleOk = async () => {
const ret = await form.validateFields(); const ret = await form.validateFields();


return onOk(ret.api_key);
return onOk(ret.api_key, ret.base_url);
}; };


const handleCancel = () => { const handleCancel = () => {
> >
<Form <Form
name="basic" name="basic"
labelCol={{ span: 4 }}
wrapperCol={{ span: 20 }}
labelCol={{ span: 6 }}
wrapperCol={{ span: 18 }}
style={{ maxWidth: 600 }} style={{ maxWidth: 600 }}
onFinish={onFinish} onFinish={onFinish}
onFinishFailed={onFinishFailed} onFinishFailed={onFinishFailed}
> >
<Input /> <Input />
</Form.Item> </Form.Item>
{llmFactory === 'OpenAI' && (
<Form.Item<FieldType>
label="Base-Url"
name="base_url"
tooltip="The API key can be obtained by registering the corresponding LLM supplier."
rules={[{ required: true, message: 'Please input base url!' }]}
>
<Input />
</Form.Item>
)}
</Form> </Form>
</Modal> </Modal>
); );

+ 7
- 2
web/src/pages/user-setting/setting-model/hooks.ts Просмотреть файл

} = useSetModalState(); } = useSetModalState();


const onApiKeySavingOk = useCallback( const onApiKeySavingOk = useCallback(
async (apiKey: string) => {
const ret = await saveApiKey({ ...savingParams, api_key: apiKey });
async (apiKey: string, baseUrl: string) => {
const ret = await saveApiKey({
...savingParams,
api_key: apiKey,
base_url: baseUrl,
});


if (ret === 0) { if (ret === 0) {
hideApiKeyModal(); hideApiKeyModal();
return { return {
saveApiKeyLoading: loading, saveApiKeyLoading: loading,
initialApiKey: '', initialApiKey: '',
llmFactory: savingParams.llm_factory,
onApiKeySavingOk, onApiKeySavingOk,
apiKeyVisible, apiKeyVisible,
hideApiKeyModal, hideApiKeyModal,

+ 2
- 0
web/src/pages/user-setting/setting-model/index.tsx Просмотреть файл

const { const {
saveApiKeyLoading, saveApiKeyLoading,
initialApiKey, initialApiKey,
llmFactory,
onApiKeySavingOk, onApiKeySavingOk,
apiKeyVisible, apiKeyVisible,
hideApiKeyModal, hideApiKeyModal,
loading={saveApiKeyLoading} loading={saveApiKeyLoading}
initialValue={initialApiKey} initialValue={initialApiKey}
onOk={onApiKeySavingOk} onOk={onApiKeySavingOk}
llmFactory={llmFactory}
></ApiKeyModal> ></ApiKeyModal>
<SystemModelSettingModal <SystemModelSettingModal
visible={systemSettingVisible} visible={systemSettingVisible}

Загрузка…
Отмена
Сохранить