web/src/pages/user-setting/setting-model/api-key-modal/index.tsx

import { IModalManagerChildrenProps } from '@/components/modal-manager';
import { Form, Input, Modal } from 'antd';
import { useEffect } from 'react';

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

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

const ApiKeyModal = ({
  visible,
  hideModal,
  llmFactory,
  loading,
  initialValue,
  onOk,
}: IProps) => {
  const [form] = Form.useForm();

  const handleOk = async () => {
    const ret = await form.validateFields();

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

  const handleCancel = () => {
    hideModal();
  };

  const onFinish = (values: any) => {
    console.log('Success:', values);
  };

  const onFinishFailed = (errorInfo: any) => {
    console.log('Failed:', errorInfo);
  };

  useEffect(() => {
    form.setFieldValue('api_key', initialValue);
  }, [initialValue, form]);

  return (
    <Modal
      title="Modify"
      open={visible}
      onOk={handleOk}
      onCancel={handleCancel}
      okButtonProps={{ loading }}
      confirmLoading={loading}
    >
      <Form
        name="basic"
        labelCol={{ span: 6 }}
        wrapperCol={{ span: 18 }}
        style={{ maxWidth: 600 }}
        onFinish={onFinish}
        onFinishFailed={onFinishFailed}
        autoComplete="off"
        form={form}
      >
        <Form.Item<FieldType>
          label="Api-Key"
          name="api_key"
          tooltip="The API key can be obtained by registering the corresponding LLM supplier."
          rules={[{ required: true, message: 'Please input api key!' }]}
        >
          <Input />
        </Form.Item>
        {llmFactory === 'OpenAI' && (
          <Form.Item<FieldType>
            label="Base-Url"
            name="base_url"
            tooltip="If your API key is from OpenAI, just ignore it. Any other intermediate providers will give this base url with the API key."
          >
            <Input placeholder="https://api.openai.com/v1" />
          </Form.Item>
        )}
      </Form>
    </Modal>
  );
};

export default ApiKeyModal;