feat(web): add Ontology menu

web/src/views/Ontology/components/OntologyClassExtractModal.tsx

@@ -0,0 +1,173 @@
+import { forwardRef, useImperativeHandle, useState } from 'react';
+import { Form, Input, App, Transfer, type TransferProps, Flex } from 'antd';
+import { useTranslation } from 'react-i18next';
+
+import type { OntologyClassData, ExtractData, OntologyClassExtractModalData, OntologyClassExtractModalRef } from '../types'
+import RbModal from '@/components/RbModal'
+import { extractOntologyTypes, createOntologyClass } from '@/api/ontology'
+import CustomSelect from '@/components/CustomSelect';
+import { getModelListUrl } from '@/api/models'
+import RbCard from '@/components/RbCard/Card';
+import Tag from '@/components/Tag';
+
+const FormItem = Form.Item;
+
+interface OntologyClassExtractModalProps {
+  refresh: () => void;
+}
+
+const OntologyClassExtractModal = forwardRef<OntologyClassExtractModalRef, OntologyClassExtractModalProps>(({
+  refresh
+}, ref) => {
+  const { t } = useTranslation();
+  const { message } = App.useApp();
+  const [visible, setVisible] = useState(false);
+  const [form] = Form.useForm<OntologyClassExtractModalData>();
+  const [loading, setLoading] = useState(false)
+  const [data, setData] = useState<OntologyClassData | null>(null)
+  const [extractData, setExtractData] = useState<ExtractData | null>(null)
+  const [targetKeys, setTargetKeys] = useState<TransferProps['targetKeys']>([]);
+  const [selectedKeys, setSelectedKeys] = useState<TransferProps['selectedKeys']>([]);
+
+  // 封装取消方法，添加关闭弹窗逻辑
+  const handleClose = () => {
+    setVisible(false);
+    form.resetFields();
+    setLoading(false)
+    setData(null)
+    setExtractData(null)
+  };
+
+  const handleOpen = (vo: OntologyClassData) => {
+    form.resetFields();
+    setVisible(true);
+    setData(vo)
+  };
+  // 封装保存方法，添加提交逻辑
+  const handleSave = () => {
+    if (!data?.scene_id) return;
+    form
+      .validateFields()
+      .then((values) => {
+        setLoading(true)
+        extractOntologyTypes({
+          ...values,
+          scene_id: data.scene_id,
+          domain: data.scene_name,
+        }).then((res) => {
+          const response = res as ExtractData
+          setExtractData(response)
+          setSelectedKeys([])
+          setTargetKeys(response.classes.map(vo => vo.id))
+        })
+        .finally(() => {
+          setLoading(false)
+        })
+      })
+      .catch((err) => {
+        console.log('err', err)
+      });
+  }
+
+  const handleConfirm = () => {
+    if (!extractData) {
+      handleSave()
+    } else {
+      if (!data?.scene_id) return;
+      if (!targetKeys || targetKeys.length === 0) {
+        message.warning(t('common.selectPlaceholder', { title: t('ontology.classType') }))
+        return
+      }
+      console.log('targetKeys', targetKeys)
+      createOntologyClass({
+        scene_id: data?.scene_id,
+        classes: extractData.classes.filter(vo => targetKeys?.includes(vo.id)).map(vo => ({ class_name: vo.name, class_description: vo.description }))
+      }).then(() => {
+        message.success(t('common.createSuccess'))
+        refresh()
+        handleClose()
+      }).finally(() => {
+        setLoading(false)
+      })
+    }
+  }
+
+
+  const onChange: TransferProps['onChange'] = (nextTargetKeys) => {
+    setTargetKeys(nextTargetKeys.filter(Boolean));
+  };
+
+  const onSelectChange: TransferProps['onSelectChange'] = (
+    sourceSelectedKeys,
+    targetSelectedKeys,
+  ) => {
+    setSelectedKeys([...sourceSelectedKeys, ...targetSelectedKeys].filter(Boolean));
+  };
+
+  // 暴露给父组件的方法
+  useImperativeHandle(ref, () => ({
+    handleOpen,
+  }));
+
+  return (
+    <RbModal
+      title={t('ontology.extract')}
+      open={visible}
+      onCancel={handleClose}
+      okText={extractData ? `${t('ontology.extractConfirm')}(${targetKeys?.length})` : loading ? t('ontology.loadingConfirm') : t('ontology.run')}
+      onOk={handleConfirm}
+      confirmLoading={loading}
+      okButtonProps={{ disabled: extractData !== null && targetKeys?.length === 0 }}
+      width={1000}
+    >
+      <Form
+        form={form}
+        layout="vertical"
+      >
+        <FormItem
+          name="llm_id"
+          label={t('ontology.llm_id')}
+          rules={[{ required: true, message: t('common.pleaseSelect') }]}
+        >
+          <CustomSelect
+            url={getModelListUrl}
+            valueKey="id"
+            labelKey="name"
+            hasAll={false}
+            placeholder={t('common.pleaseSelect')}
+            params={{ type: 'llm,chat', pagesize: 100, is_active: true }}
+          />
+        </FormItem>
+
+        <FormItem
+          name="scenario"
+          label={t('ontology.scenario')}
+          rules={[{ required: true, message: t('common.pleaseEnter') }]}
+        >
+          <Input.TextArea placeholder={t('ontology.scenarioPlaceholder')} />
+        </FormItem>
+      </Form>
+
+      {extractData && <RbCard
+        title={t('ontology.classType')}
+        bodyClassName='rb:flex rb:justify-center rb:h-[450px]!'
+      >
+        <Transfer
+          titles={[t('ontology.source'), t('ontology.target')]}
+          dataSource={extractData?.classes?.map(vo => ({ ...vo, key: vo.id }))}
+          targetKeys={targetKeys}
+          selectedKeys={selectedKeys}
+          onChange={onChange}
+          onSelectChange={onSelectChange}
+          render={(item) => (<div>
+            {item.name}
+            <Flex wrap gap={8}>{item.examples.map((vo, index) => <Tag color="default" key={index}>{vo}</Tag>)}</Flex>
+          </div>)}
+          listStyle={{ width: '400px', height: '100%' }}
+        />
+      </RbCard>}
+    </RbModal>
+  );
+});
+
+export default OntologyClassExtractModal;

web/src/views/Ontology/components/OntologyClassModal.tsx

@@ -0,0 +1,96 @@
+import { forwardRef, useImperativeHandle, useState } from 'react';
+import { Form, Input, App } from 'antd';
+import { useTranslation } from 'react-i18next';
+
+import type { AddClassItem, OntologyClassModalRef } from '../types'
+import RbModal from '@/components/RbModal'
+import { createOntologyClass } from '@/api/ontology'
+
+const FormItem = Form.Item;
+
+interface OntologyClassModalProps {
+  refresh: () => void;
+}
+
+const OntologyClassModal = forwardRef<OntologyClassModalRef, OntologyClassModalProps>(({
+  refresh
+}, ref) => {
+  const { t } = useTranslation();
+  const { message } = App.useApp();
+  const [visible, setVisible] = useState(false);
+  const [form] = Form.useForm<AddClassItem>();
+  const [loading, setLoading] = useState(false)
+  const [scene_id, setSceneId] = useState<string | null>(null)
+
+  // 封装取消方法，添加关闭弹窗逻辑
+  const handleClose = () => {
+    setVisible(false);
+    form.resetFields();
+    setLoading(false)
+  };
+
+  const handleOpen = (scene_id: string) => {
+    form.resetFields();
+    setVisible(true);
+    setSceneId(scene_id)
+  };
+  // 封装保存方法，添加提交逻辑
+  const handleSave = () => {
+    if (!scene_id) return;
+    form
+      .validateFields()
+      .then((values) => {
+        setLoading(true)
+        createOntologyClass({
+          scene_id: scene_id,
+          classes: [{ ...values }]
+        }).then(() => {
+            message.success(t('common.saveSuccess'));
+            handleClose();
+            refresh();
+          })
+          .finally(() => setLoading(false))
+      })
+      .catch((err) => {
+        console.log('err', err)
+      });
+  }
+
+  // 暴露给父组件的方法
+  useImperativeHandle(ref, () => ({
+    handleOpen,
+  }));
+
+  return (
+    <RbModal
+      title={t('ontology.addClass')}
+      open={visible}
+      onCancel={handleClose}
+      okText={t('common.create')}
+      onOk={handleSave}
+      confirmLoading={loading}
+    >
+      <Form
+        form={form}
+        layout="vertical"
+      >
+        <FormItem
+          name="class_name"
+          label={t('ontology.class_name')}
+          rules={[{ required: true, message: t('common.pleaseEnter') }]}
+        >
+          <Input placeholder={t('common.enter')} />
+        </FormItem>
+
+        <FormItem
+          name="class_description"
+          label={t('ontology.class_description')}
+        >
+          <Input.TextArea placeholder={t('ontology.classDescriptionPlaceholder')} />
+        </FormItem>
+      </Form>
+    </RbModal>
+  );
+});
+
+export default OntologyClassModal;

web/src/views/Ontology/components/OntologyModal.tsx

@@ -0,0 +1,99 @@
+import { forwardRef, useImperativeHandle, useState } from 'react';
+import { Form, Input, App } from 'antd';
+import { useTranslation } from 'react-i18next';
+
+import type { OntologyItem, OntologyModalData, OntologyModalRef } from '../types'
+import RbModal from '@/components/RbModal'
+import { createOntologyScene, updateOntologyScene } from '@/api/ontology'
+
+const FormItem = Form.Item;
+
+interface OntologyModalProps {
+  refresh: () => void;
+}
+
+const OntologyModal = forwardRef<OntologyModalRef, OntologyModalProps>(({
+  refresh
+}, ref) => {
+  const { t } = useTranslation();
+  const { message } = App.useApp();
+  const [visible, setVisible] = useState(false);
+  const [editVo, setEditVo] = useState<OntologyItem | null>(null)
+  const [form] = Form.useForm<OntologyModalData>();
+  const [loading, setLoading] = useState(false)
+
+  // 封装取消方法，添加关闭弹窗逻辑
+  const handleClose = () => {
+    setVisible(false);
+    form.resetFields();
+    setLoading(false)
+    setEditVo(null)
+  };
+
+  const handleOpen = (vo?: OntologyItem) => {
+    if (vo) {
+      setEditVo(vo);
+      form.setFieldsValue(vo);
+    } else {
+      form.resetFields();
+    }
+    setVisible(true);
+  };
+  // 封装保存方法，添加提交逻辑
+  const handleSave = () => {
+    form
+      .validateFields()
+      .then((values) => {
+        setLoading(true)
+        const request = editVo?.scene_id ? updateOntologyScene(editVo.scene_id, values) : createOntologyScene(values)
+        request
+          .then(() => {
+            message.success(t('common.saveSuccess'));
+            handleClose();
+            refresh();
+          })
+          .finally(() => setLoading(false))
+      })
+      .catch((err) => {
+        console.log('err', err)
+      });
+  }
+
+  // 暴露给父组件的方法
+  useImperativeHandle(ref, () => ({
+    handleOpen,
+  }));
+
+  return (
+    <RbModal
+      title={editVo?.scene_id ? t('ontology.edit') : t('ontology.create')}
+      open={visible}
+      onCancel={handleClose}
+      okText={editVo?.scene_id ? t('common.save') : t('common.create')}
+      onOk={handleSave}
+      confirmLoading={loading}
+    >
+      <Form
+        form={form}
+        layout="vertical"
+      >
+        <FormItem
+          name="scene_name"
+          label={t('ontology.scene_name')}
+          rules={[{ required: true, message: t('common.pleaseEnter') }]}
+        >
+          <Input placeholder={t('common.enter')} />
+        </FormItem>
+
+        <FormItem
+          name="scene_description"
+          label={t('ontology.scene_description')}
+        >
+          <Input.TextArea placeholder={t('ontology.descriptionPlaceholder')} />
+        </FormItem>
+      </Form>
+    </RbModal>
+  );
+});
+
+export default OntologyModal;

web/src/views/Ontology/components/PageHeader.tsx

@@ -0,0 +1,45 @@
+import { type FC, type ReactNode } from 'react';
+import { useNavigate } from 'react-router-dom';
+import { Layout, Button } from 'antd';
+import { useTranslation } from 'react-i18next';
+import logoutIcon from '@/assets/images/logout_hover.svg'
+
+const { Header } = Layout;
+
+interface ConfigHeaderProps {
+  name?: string;
+  subTitle?: ReactNode | string;
+  extra?: ReactNode;
+}
+const PageHeader: FC<ConfigHeaderProps> = ({ 
+  name,
+  subTitle,
+  extra
+}) => {
+  const { t } = useTranslation();
+  const navigate = useNavigate();
+
+  const goBack = () => {
+    navigate(-1)
+  }
+  return (
+    <Header className="rb:w-full rb:h-16 rb:flex rb:justify-between rb:p-[0_16px_0_24px]! rb:border-b rb:border-[#EAECEE] rb:leading-8">
+      <div className="rb:flex rb:flex-col rb:justify-center rb:gap-1 rb:mr-4">
+        <div className="rb:text-[16px] rb:leading-6 rb:font-medium">
+          {name}
+        </div>
+        <div className="rb:text-[12px] rb:text-[#5B6167] rb:leading-4">{subTitle}</div>
+      </div>
+
+      <div className="rb:flex rb:items-center rb:gap-3">
+        <Button type="primary" ghost className="rb:h-6! rb:px-2! rb:leading-5.5!" onClick={goBack}>
+          <img src={logoutIcon} className="rb:w-4 rb:h-4" />
+          {t('common.return')}
+        </Button>
+        {extra}
+      </div>
+    </Header>
+  );
+};
+
+export default PageHeader;