@gientech/modual 2.2.3-fix.6 → 2.2.3-fix.7

package/src/modules/assistantConfig/index.tsx

@@ -1,785 +0,0 @@
-import React, { useState, useMemo, useEffect, useRef } from 'react';
-import { Form, Input, Select, Checkbox, Switch, Row, Col, message, Button } from 'antd';
-import { ArrowLeft } from 'lucide-react';
-import Paragraph from './components/Paragraph';
-import NotFoundContent from './components/NotFoundContent';
-import { Styles } from '@mxmweb/zui';
-import { CustomComponents } from '@mxmweb/xviewer';
-import {
-  EXPLANATION_PROMPT,
-  EXPLANATION_DEPENDON_KB,
-  DEFAULT_PROMPT,
-  SearchPolicyOptions,
-  ASSISTANT_MIN_MAX,
-  ASSISTANT_DEFAULT,
-  TraceabilityStrategy,
-  DEFAULT_ANSWER,
-} from './constants';
-import AxiosInstance from './server';
-import SensitiveSection from './components/Sensitive';
-import KnowledgeSection from './components/Knowledge';
-import DatabaseSection from './components/Database';
-import GraphSection from './components/Graph';
-import ParamsItem from './components/ParamsItem';
-
-import '@mxmweb/zui/style.css';
-
-const { Item } = Form;
-
-interface ConversationAssistantProps {
-  url?: string;
-  token?: string;
-  role?: string;
-  id?: string;
-  showDatabase?: boolean;
-  showGraph?: boolean;
-  showGraphEnhanced?: boolean;
-  isShare?: boolean;
-  styles?: Styles;
-  disabled?: boolean;
-  CustomComponents?: CustomComponents;
-  eventsEmit?: (name: string, data?: any) => void;
-}
-
-const ConversationAssistant: React.FC<ConversationAssistantProps> = ({
-  url,
-  token,
-  role,
-  id,
-  showDatabase,
-  showGraph,
-  isShare,
-  disabled, //查看时，不可修改
-  eventsEmit,
-  showGraphEnhanced,
-  CustomComponents,
-}) => {
-  const { ContentLoading } = CustomComponents;
-  const [form] = Form.useForm();
-  const [configData, setConfigData]: any = useState();
-  const [modelList, setModelList]: any = useState();
-  const [rerankOptions, setRerankOptions]: any = useState();
-  const [multiModelOptions, setMultiModalOptions]: any = useState();
-  const [embeddingModelOptions, setEmbeddingModalOptions]: any = useState();
-  const [loading, setLoading] = useState(true);
-  const [btnLoading, seBtntLoading] = useState(false);
-  const [configScale, setConfigScale] = useState(ASSISTANT_MIN_MAX);
-
-  const resourceBindData = useRef<{
-    knowledge_base_name?: string;
-    knowledge_base_name_source?: string;
-    selectedKnowledgeBase?: any[];
-    sensitiveWordIds?: string;
-    database_name?: any[];
-    graph_nb_name?: any[];
-  }>({}); //存储资源绑定里选择的数据
-
-  const {
-    getConfigById,
-    fetchModels,
-    fetchConfigItemLimit,
-    checkName,
-    addConversationAssistant,
-    updateConversationAssistant,
-  } = AxiosInstance({
-    url,
-    token,
-    role,
-    eventsEmit,
-  });
-
-  useEffect(() => {
-    init();
-  }, [id]);
-
-  useEffect(() => {
-    if (id) {
-      //编辑的话，只有模型列表返回为空才需要校验
-      if (Array.isArray(modelList) && Array.isArray(rerankOptions)) {
-        form.validateFields([
-          'model_name',
-          'plan_model_name',
-          'recommend_model_name',
-          'rerank_model_name',
-          'multi_model_name',
-          'embedding_model_name',
-        ]);
-      }
-      if (Array.isArray(multiModelOptions)) {
-        form.validateFields(['multi_model_name']);
-      }
-    } else {
-      //新增的话，只有模型列表返回为空才需要校验
-      if (Array.isArray(modelList) && modelList?.length == 0) {
-        form.validateFields([
-          'model_name',
-          'plan_model_name',
-          'recommend_model_name',
-          'embedding_model_name',
-        ]);
-      }
-
-      if (Array.isArray(rerankOptions) && rerankOptions?.length == 0) {
-        form.validateFields(['rerank_model_name']);
-      }
-      if (Array.isArray(multiModelOptions) && multiModelOptions?.length == 0) {
-        form.validateFields(['multi_model_name']);
-      }
-      if (Array.isArray(embeddingModelOptions) && embeddingModelOptions?.length == 0) {
-        form.validateFields(['embedding_model_name']);
-      }
-    }
-  }, [rerankOptions, modelList, multiModelOptions, embeddingModelOptions, id]);
-
-  // 初始化数据
-  const init = async () => {
-    let _configData;
-    if (id) {
-      if (!loading) setLoading(true);
-      try {
-        const res = await getConfigById(id);
-        if (res) {
-          const { configJson, ...rest } = res;
-          const config = configJson && JSON.parse(configJson);
-          const {
-            knowledge_base_name,
-            knowledge_base_name_source,
-            selectedKnowledgeBase,
-            sensitiveWordIds,
-            database_name,
-            graph_nb_name,
-          } = config;
-          _configData = {
-            ...rest,
-            ...config,
-          };
-
-          resourceBindData.current = {
-            knowledge_base_name,
-            knowledge_base_name_source,
-            selectedKnowledgeBase,
-            sensitiveWordIds,
-            database_name,
-            graph_nb_name,
-          };
-
-          setConfigData(_configData);
-
-          form.setFieldsValue(_configData);
-          setLoading(false);
-        }
-      } catch (e) { }
-    } else {
-      setLoading(false);
-    }
-
-    await fetchList(_configData);
-  };
-
-  const fetchList = async (configData?: any) => {
-
-    try {
-      const [modelRes, recallModelRes, multiModelRes, configItemLimitRes, embeddingModelRes] =
-        await Promise.allSettled([
-          fetchModels({
-            pageNum: 1,
-            pageSize: 9999,
-            status: 1,
-            type: 1,
-          }),
-          fetchModels({
-            pageNum: 1,
-            pageSize: 9999,
-            status: 1,
-            type: 3,
-          }),
-          fetchModels({
-            pageNum: 1,
-            pageSize: 9999,
-            status: 1,
-            type: 4,
-          }),
-          fetchConfigItemLimit(),
-          fetchModels({
-            pageNum: 1,
-            pageSize: 9999,
-            status: 1,
-            type: 2,
-          }),
-        ]);
-      if (modelRes.status === 'fulfilled' && modelRes.value) {
-        const list = modelRes.value?.records || [];
-        const options = list.map((item: any) => ({
-          ...item,
-          label: item.modelName,
-          value: item.modelName,
-        }));
-        if (options.length > 0) {
-          const cur_max_tokens = options.find((m) => m.value == configData?.model_name)?.maxTokens
-          if (configData?.max_tokens > cur_max_tokens) {
-            form.setFieldsValue({ max_tokens: cur_max_tokens });
-          }
-
-        }
-        setModelList(options);
-      }
-
-      if (recallModelRes.status === 'fulfilled' && recallModelRes.value) {
-        const options = (recallModelRes.value?.records || []).map((item: any) => ({
-          label: item.modelName || item.name,
-          value: item.modelName || item.name,
-        }));
-        setRerankOptions(options);
-        if (options.length > 0) {
-          !configData?.rerank_model_name &&
-            form.setFieldsValue({ rerank_model_name: options[0].value });
-        }
-      }
-      if (multiModelRes.status === 'fulfilled' && multiModelRes.value) {
-        const options = (multiModelRes.value?.records || []).map((item: any) => ({
-          label: item.modelName || item.name,
-          value: item.modelName || item.name,
-        }));
-        setMultiModalOptions(options);
-        if (options.length > 0) {
-          !configData?.multi_model_name &&
-            form.setFieldsValue({ multi_model_name: options[0].value });
-        }
-      }
-      if (embeddingModelRes.status === 'fulfilled' && embeddingModelRes.value) {
-        const options = (embeddingModelRes.value?.records || []).map((item: any) => ({
-          label: item.modelName || item.name,
-          value: item.modelName || item.name,
-        }));
-        setEmbeddingModalOptions(options);
-        if (options.length > 0) {
-          !configData?.embedding_model_name &&
-            form.setFieldsValue({ embedding_model_name: options[0].value });
-        }
-      }
-      if (configItemLimitRes.status === 'fulfilled' && configItemLimitRes.value) {
-        if (configItemLimitRes.value) {
-          const config = configItemLimitRes.value;
-
-          Object.keys(ASSISTANT_MIN_MAX).forEach(key => {
-            if (config.hasOwnProperty(key)) {
-              const { max, min } = config[key];
-              ASSISTANT_MIN_MAX[key] = {
-                ...ASSISTANT_MIN_MAX[key],
-                min,
-                max,
-              };
-            }
-          });
-          setConfigScale(ASSISTANT_MIN_MAX);
-        }
-      }
-    } catch (e) {
-      console.error('获取列表失败:', e);
-    }
-  };
-
-  const checkConfigName = async (_, value: any) => {
-    const res = await checkName(value, id);
-    if (res) {
-      message.warning('名称重复！', 2);
-      return Promise.reject('名称重复');
-    }
-    return Promise.resolve();
-  };
-
-  const validateModel = (options: any[], msg: string) => ({
-    validator(_: any, value: any) {
-      if (!value) {
-        if (Array.isArray(options) && options.length === 0) {
-          return Promise.reject('暂无可用模型，请联系管理员');
-        } else {
-          return Promise.reject(msg);
-        }
-      }
-      if (
-        (Array.isArray(options) && options.length === 0) ||
-        !options.find((m: any) => m.value == value)
-      ) {
-        return Promise.reject('模型无法使用，请联系管理员');
-      }
-      return Promise.resolve();
-    },
-  });
-
-  const handleEventsEmit = (name: string, data?: any) => {
-    switch (name) {
-      case 'knowledge_selected':
-        resourceBindData.current = {
-          ...resourceBindData.current,
-          ...data,
-        };
-        break;
-      case 'sensitive_selected':
-        resourceBindData.current = {
-          ...resourceBindData.current,
-          sensitiveWordIds: data,
-        };
-        break;
-      case 'database_selected':
-        resourceBindData.current = {
-          ...resourceBindData.current,
-          database_name: data,
-        };
-        break;
-      case 'graph_selected':
-        resourceBindData.current = {
-          ...resourceBindData.current,
-          graph_nb_name: data,
-        };
-        break;
-    }
-  };
-
-  const handleSubmit = async () => {
-    let params = {};
-    try {
-      seBtntLoading(true);
-      const data = await form.validateFields();
-      const {
-        knowledge_base_name,
-        knowledge_base_name_source,
-        selectedKnowledgeBase,
-        sensitiveWordIds,
-        database_name,
-        graph_nb_name,
-      } = resourceBindData.current;
-
-      data.thinking =
-        (modelList.find((item: any) => item.modelName == data.model_name) as any)?.thinking ||
-        false;
-      if (showGraph) {
-        // 如果展示图谱选择，则打开图谱开关，这个参数算法要用
-        data.enabled_nb_graph = true;
-      }
-
-      const configJson = JSON.stringify({
-        ...data,
-        knowledge_base_name,
-        knowledge_base_name_source,
-        selectedKnowledgeBase,
-        sensitiveWordIds,
-        ...(showDatabase ? { database_name } : {}),
-        ...(showGraph ? { graph_nb_name } : {}),
-      });
-
-      params = {
-        configJson,
-        name: data.config_name,
-        isDefault: data.isDefault,
-        defaultAnswer: data.defaultAnswer,
-        dependOnKb: data.dependOnKb,
-        sensitiveWordIds: sensitiveWordIds,
-      };
-      if (id) {
-        const res = await updateConversationAssistant(id, params);
-        if (res) {
-          message.success('更新助手配置成功');
-          eventsEmit && eventsEmit('go_back');
-        }
-      } else {
-        const res = await addConversationAssistant(params);
-        if (res) {
-          message.success('新增助手配置成功');
-          eventsEmit && eventsEmit('go_back');
-        }
-      }
-    } catch (e) {
-      console.error('submit error', e);
-    } finally {
-      seBtntLoading(false);
-    }
-  };
-
-  const formInitialValue = useMemo(() => {
-    // 获取 检索与生成参数(ASSISTANT_MIN_MAX) 和 功能开关的初始值(ASSISTANT_DEFAULT)
-    const _default: any = {};
-    Object.keys(ASSISTANT_MIN_MAX).forEach(key => {
-      _default[key] = ASSISTANT_MIN_MAX[key]?.default;
-    });
-    return {
-      ...ASSISTANT_DEFAULT,
-      ..._default,
-    };
-  }, []);
-
-  return (
-    <div
-      style={{
-        height: '100%',
-        width: '100%',
-      }}
-    >
-      <div
-        className="flex items-center justify-between"
-        style={{
-          width: '100%',
-          padding: '4px 10px 16px 10px',
-        }}
-      >
-        <div className="flex items-center">
-          <ArrowLeft color="#2468F2" onClick={() => eventsEmit && eventsEmit('go_back')} />
-          <div
-            className="text-[16px] text-[#1D2129] bold"
-            style={{
-              marginLeft: 20,
-            }}
-          >
-            {id ? '编辑会话助手' : '新建会话助手'}
-          </div>
-        </div>
-        {!loading && (
-          <Button
-            type="primary"
-            onClick={() => handleSubmit()}
-            loading={btnLoading}
-            disabled={disabled}
-          // style={{
-          //   backgroundColor: '#055aff',
-          // }}
-          >
-            确定
-          </Button>
-        )}
-      </div>
-
-      {loading || false ? (
-        ContentLoading && <ContentLoading />
-      ) : (
-        <Form
-          style={{
-            padding: 20,
-            height: 'calc(100vh - 150px)',
-            backgroundColor: '#fff',
-          }}
-          form={form}
-          layout="vertical"
-          initialValues={formInitialValue}
-          disabled={disabled}
-          colon={false}
-        >
-          <Row
-            gutter={24}
-            style={{
-              height: '100%',
-            }}
-          >
-            <Col
-              span={12}
-              style={{
-                borderRight: '1px solid #E5E6EB',
-                padding: '10px 32px',
-                height: '100%',
-                overflowY: 'auto',
-              }}
-            >
-              <Item
-                label="助手名称"
-                name="config_name"
-                validateTrigger="onBlur"
-                rules={[
-                  { required: true, message: '请输入助手名称' },
-                  {
-                    validator: checkConfigName,
-                  },
-                ]}
-              >
-                <Input placeholder="请输入" />
-              </Item>
-
-              <Row gutter={16}>
-                <Col span={12}>
-                  <Item
-                    label="问答模型"
-                    name="model_name"
-                    rules={[
-                      { required: true, message: '' },
-                      validateModel(modelList, '请选择问答模型'),
-                    ]}
-                  >
-                    <Select
-                      placeholder="请选择"
-                      options={modelList}
-                      notFoundContent={<NotFoundContent />}
-                    />
-                  </Item>
-                </Col>
-                <Col span={12}>
-                  <Item
-                    label="规划模型"
-                    name="plan_model_name"
-                    rules={[
-                      { required: true, message: '' },
-                      validateModel(modelList, '请选择规划模型'),
-                    ]}
-                  >
-                    <Select
-                      placeholder="请选择"
-                      options={modelList}
-                      notFoundContent={<NotFoundContent />}
-                    />
-                  </Item>
-                </Col>
-              </Row>
-              <Row gutter={16}>
-                <Col span={12}>
-                  <Item
-                    label="上传文档问答多模态模型"
-                    name="multi_model_name"
-                    rules={[
-                      { required: true, message: '' },
-                      validateModel(multiModelOptions, '请选择上传文档问答多模态模型'),
-                    ]}
-                  >
-                    <Select
-                      placeholder="请选择"
-                      options={multiModelOptions}
-                      notFoundContent={<NotFoundContent />}
-                    />
-                  </Item>
-                </Col>
-                <Col span={12}>
-                  <Item
-                    label="Rerank模型"
-                    name="rerank_model_name"
-                    rules={[
-                      { required: true, message: '' },
-                      validateModel(rerankOptions, '请选择Rerank模型'),
-                    ]}
-                  >
-                    <Select
-                      placeholder="请选择"
-                      options={rerankOptions}
-                      notFoundContent={<NotFoundContent />}
-                    />
-                  </Item>
-                </Col>
-              </Row>
-              <Row gutter={16}>
-                <Col span={24}>
-                  <Item
-                    label="向量解析模型"
-                    name="embedding_model_name"
-                    rules={[
-                      { required: true, message: '' },
-                      validateModel(embeddingModelOptions, '请选择向量解析模型'),
-                    ]}
-                  >
-                    <Select
-                      placeholder="请选择"
-                      options={embeddingModelOptions}
-                      notFoundContent={<NotFoundContent />}
-                    />
-                  </Item>
-                </Col>
-              </Row>
-              <Item label="开场白" name="prologue">
-                <Input.TextArea
-                  maxLength={50}
-                  showCount
-                  placeholder="请输入开场白"
-                  autoSize={{ minRows: 2 }}
-                />
-              </Item>
-
-              <Item label="提示词" name="system_prompt" tooltip={EXPLANATION_PROMPT}>
-                <Input.TextArea
-                  showCount
-                  rows={8}
-                  maxLength={3000}
-                  autoSize={{ minRows: 10 }}
-                  placeholder={`例如：${DEFAULT_PROMPT}`}
-                />
-              </Item>
-
-              <Item
-                name="isDefault"
-                valuePropName="checked"
-                normalize={value => (value === true ? 1 : 0)}
-              >
-                <Checkbox>选择为默认助手</Checkbox>
-              </Item>
-            </Col>
-            <Col
-              span={12}
-              style={{
-                height: '100%',
-                padding: '10px 32px',
-                overflowY: 'auto',
-              }}
-            >
-              <Paragraph title="检索与生成参数">
-                <Item
-                  label="问题推荐模型"
-                  name="recommend_model_name"
-                  rules={[
-                    { required: true, message: '' },
-                    validateModel(modelList, '请选择问题推荐模型'),
-                  ]}
-                >
-                  <Select
-                    placeholder="请选择"
-                    options={modelList}
-                    notFoundContent={<NotFoundContent />}
-                  />
-                </Item>
-
-                <Item label="检索策略" name="search_policy">
-                  <Select options={SearchPolicyOptions} placeholder="请选择" />
-                </Item>
-
-                <Row gutter={16}>
-                  <ParamsItem label="温度" name="temperature" configScale={configScale} />
-                  <ParamsItem label="多样性" name="top_p" configScale={configScale} />
-                </Row>
-
-                <Row gutter={16}>
-                  <ParamsItem label="召回数" name="top_k" configScale={configScale} />
-                  <ParamsItem label="相似度阈值" name="score_threshold" configScale={configScale} />
-                </Row>
-                <Row gutter={16}>
-                  <Item
-                    noStyle
-                    shouldUpdate={(prev, cur) => prev.model_name !== cur.model_name}
-                  >
-                    {({ getFieldValue }) => {
-                      const model = getFieldValue("model_name")
-                      const model_maxToken = (modelList || []).find((m) => m.value == model)?.maxTokens
-                      const cur_max_tokens = getFieldValue("max_tokens")
-                      if (cur_max_tokens > model_maxToken) {
-                        form.setFieldValue("max_tokens", model_maxToken)
-                      }
-                      return <ParamsItem label="最大令牌数" name="max_tokens" configScale={{
-                        ...configScale,
-                        max_tokens: {
-                          ...configScale.max_tokens,
-                          max: model_maxToken ? model_maxToken : 10000
-                        }
-                      }} />
-                    }
-                    }
-                  </Item>
-
-                  <ParamsItem label="上下文最大轮数" name="max_round" configScale={configScale} />
-                </Row>
-              </Paragraph>
-              <Paragraph title="功能开关">
-                <Item
-                  label="开启FAQ优先"
-                  name="enabled_faq"
-                  valuePropName="checked"
-                  layout="horizontal"
-                >
-                  <Switch />
-                </Item>
-                {showGraphEnhanced && (
-                  <Item
-                    label="启用图谱增强"
-                    name="enabled_graph"
-                    valuePropName="checked"
-                    layout="horizontal"
-                  >
-                    <Switch />
-                  </Item>
-                )}
-                <Item
-                  label="开启强企业知识依赖"
-                  name="dependOnKb"
-                  valuePropName="checked"
-                  layout="horizontal"
-                  tooltip={EXPLANATION_DEPENDON_KB}
-                  normalize={value => (value === true ? 1 : 0)}
-                >
-                  <Switch />
-                </Item>
-                <Item
-                  label="开启问题改写"
-                  name="enabled_question_rewrite"
-                  valuePropName="checked"
-                  layout="horizontal"
-                >
-                  <Switch />
-                </Item>
-                <Item
-                  label="开启溯源"
-                  name="need_trace_source"
-                  valuePropName="checked"
-                  layout="horizontal"
-                >
-                  <Switch />
-                </Item>
-                <Item
-                  noStyle
-                  shouldUpdate={(prev, cur) => prev.need_trace_source !== cur.need_trace_source}
-                >
-                  {({ getFieldValue }) =>
-                    getFieldValue('need_trace_source') ? (
-                      <>
-                        <Item label="溯源策略" name="trace_by_llm" layout="horizontal">
-                          <Select placeholder="请选择" options={TraceabilityStrategy} />
-                        </Item>
-                        <Item name="is_download" valuePropName="checked" layout="horizontal">
-                          <Checkbox>允许下载溯源文件</Checkbox>
-                        </Item>
-                      </>
-                    ) : null
-                  }
-                </Item>
-              </Paragraph>
-              <Paragraph title="资源绑定">
-                <SensitiveSection
-                  url={url}
-                  token={token}
-                  role={role}
-                  sectionData={configData?.sensitiveWordIds}
-                  eventsEmit={handleEventsEmit}
-                  defaultAnswer={configData?.defaultAnswer}
-                  form={form}
-                />
-                <KnowledgeSection
-                  url={url}
-                  token={token}
-                  role={role}
-                  sectionData={{
-                    knowledge_base_name: configData?.knowledge_base_name,
-                    knowledge_base_name_source: configData?.knowledge_base_name_source,
-                    selectedKnowledgeBase: configData?.selectedKnowledgeBase,
-                  }}
-                  createBy={isShare ? configData?.createBy : undefined}
-                  eventsEmit={handleEventsEmit}
-                />
-
-                {showDatabase && (
-                  <DatabaseSection
-                    url={url}
-                    token={token}
-                    role={role}
-                    disabled={disabled}
-                    sectionData={configData?.database_name}
-                    eventsEmit={handleEventsEmit}
-                  />
-                )}
-                {showGraph && (
-                  <GraphSection
-                    url={url}
-                    token={token}
-                    role={role}
-                    sectionData={configData?.graph_nb_name}
-                    eventsEmit={handleEventsEmit}
-                  />
-                )}
-              </Paragraph>
-            </Col>
-          </Row>
-        </Form>
-      )}
-    </div>
-  );
-};
-
-export default ConversationAssistant;