feat: support AWS Bedrock #308 (#1617)

### What problem does this PR solve? feat: support AWS Bedrock #308 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
2024-07-19 18:36:49 +08:00
parent 58df013722
commit 657019a5a9
8 changed files with 258 additions and 52 deletions
--- a/web/src/pages/user-setting/setting-model/bedrock-modal/index.tsx
+++ b/web/src/pages/user-setting/setting-model/bedrock-modal/index.tsx
@@ -0,0 +1,119 @@
+import { useTranslate } from '@/hooks/common-hooks';
+import { IModalProps } from '@/interfaces/common';
+import { IAddLlmRequestBody } from '@/interfaces/request/llm';
+import { Flex, Form, Input, Modal, Select, Space } from 'antd';
+import { useMemo } from 'react';
+import { BedrockRegionList } from '../constant';
+
+type FieldType = IAddLlmRequestBody & {
+  bedrock_ak: string;
+  bedrock_sk: string;
+  bedrock_region: string;
+};
+
+const { Option } = Select;
+
+const BedrockModal = ({
+  visible,
+  hideModal,
+  onOk,
+  loading,
+  llmFactory,
+}: IModalProps<IAddLlmRequestBody> & { llmFactory: string }) => {
+  const [form] = Form.useForm<FieldType>();
+
+  const { t } = useTranslate('setting');
+  const options = useMemo(
+    () => BedrockRegionList.map((x) => ({ value: x, label: t(x) })),
+    [t],
+  );
+
+  const handleOk = async () => {
+    const values = await form.validateFields();
+
+    const data = {
+      ...values,
+      llm_factory: llmFactory,
+    };
+
+    onOk?.(data);
+  };
+
+  return (
+    <Modal
+      title={t('addLlmTitle', { name: llmFactory })}
+      open={visible}
+      onOk={handleOk}
+      onCancel={hideModal}
+      okButtonProps={{ loading }}
+      footer={(originNode: React.ReactNode) => {
+        return (
+          <Flex justify={'space-between'}>
+            <a
+              href="https://console.aws.amazon.com/"
+              target="_blank"
+              rel="noreferrer"
+            >
+              {t('ollamaLink', { name: llmFactory })}
+            </a>
+            <Space>{originNode}</Space>
+          </Flex>
+        );
+      }}
+    >
+      <Form
+        name="basic"
+        style={{ maxWidth: 600 }}
+        autoComplete="off"
+        layout={'vertical'}
+        form={form}
+      >
+        <Form.Item<FieldType>
+          label={t('modelType')}
+          name="model_type"
+          initialValue={'chat'}
+          rules={[{ required: true, message: t('modelTypeMessage') }]}
+        >
+          <Select placeholder={t('modelTypeMessage')}>
+            <Option value="chat">chat</Option>
+            <Option value="embedding">embedding</Option>
+          </Select>
+        </Form.Item>
+        <Form.Item<FieldType>
+          label={t('modelName')}
+          name="llm_name"
+          rules={[{ required: true, message: t('bedrockModelNameMessage') }]}
+        >
+          <Input placeholder={t('bedrockModelNameMessage')} />
+        </Form.Item>
+        <Form.Item<FieldType>
+          label={t('addBedrockEngineAK')}
+          name="bedrock_ak"
+          rules={[{ required: true, message: t('bedrockAKMessage') }]}
+        >
+          <Input placeholder={t('bedrockAKMessage')} />
+        </Form.Item>
+        <Form.Item<FieldType>
+          label={t('addBedrockSK')}
+          name="bedrock_sk"
+          rules={[{ required: true, message: t('bedrockSKMessage') }]}
+        >
+          <Input placeholder={t('bedrockSKMessage')} />
+        </Form.Item>
+        <Form.Item<FieldType>
+          label={t('bedrockRegion')}
+          name="bedrock_region"
+          rules={[{ required: true, message: t('bedrockRegionMessage') }]}
+        >
+          <Select
+            placeholder={t('bedrockRegionMessage')}
+            options={options}
+            allowClear
+          ></Select>
+        </Form.Item>
+      </Form>
+    </Modal>
+  );
+};
+
+export default BedrockModal;