feat: 修改模型数据可修改问题

e08e8aa0 · archer · becee69d · e08e8aa0 · e08e8aa0 · e08e8aa0
Commit e08e8aa0 authored Apr 04, 2023 by archer
5 changed files
--- a/src/api/model.ts
+++ b/src/api/model.ts
@@ -57,7 +57,7 @@ export const postModelDataJsonData = (
  jsonData: { prompt: string; completion: string; vector?: number[] }[]
 ) => POST(`/model/data/pushModelDataJson`, { modelId, data: jsonData });
-export const putModelDataById = (data: { dataId: string; text: string }) =>
+export const putModelDataById = (data: { dataId: string; text: string; q?: string }) =>
  PUT('/model/data/putModelData', data);
 export const delOneModelData = (dataId: string) =>
  DELETE(`/model/data/delModelDataById?dataId=${dataId}`);
--- a/src/pages/api/chat/lafGpt.ts
+++ b/src/pages/api/chat/lafGpt.ts
@@ -65,27 +65,23 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
        messages: [
          {
            role: 'system',
-            content: `服务端逻辑生成器。根据用户输入的需求，拆解成代码实现的步骤，并按下面格式返回：
+            content: `服务端逻辑生成器。根据用户输入的需求，拆解成代码实现的步骤，并按格式返回： 1.\n2.\n3.\n ......
-          1. 
-          2. 
-          3. 
-          ....
          下面是一些例子：
-          实现一个手机号注册账号的方法
+          实现一个手机号注册账号的方法，包含两个函数
-          发送手机验证码函数： 
+          * 发送手机验证码函数： 
          1. 从 query 中获取 phone 
          2. 校验手机号格式是否正确，不正确返回{error: "手机号格式错误"} 
          3. 给 phone 发送一个短信验证码，验证码长度为6位字符串，内容为：你正在注册laf, 验证码为：code 
          4. 数据库添加数据，表为"codes"，内容为 {phone, code} 
-          注册函数 
+          * 注册函数 
          1. 从 body 中获取 phone 和 code 
          2. 校验手机号格式是否正确，不正确返回{error: "手机号格式错误"} 
          2. 获取数据库数据，表为"codes"，查找是否有符合 phone, code 等于body参数的记录，没有的话返回 {error:"验证码不正确"} 
          4. 添加数据库数据，表为"users" ，内容为{phone, code, createTime}
          5. 删除数据库数据，删除 code 记录
          ---------------
-          更新播客记录。传入blogId，blogText，tags，还需要记录更新的时间
+          更新博客记录。传入blogId，blogText，tags，还需要记录更新的时间
          1. 从 body 中获取 blogId，blogText 和 tags
          2. 校验 blogId 是否为空，为空则返回 {error: "博客ID不能为空"}
          3. 校验 blogText 是否为空，为空则返回 {error: "博客内容不能为空"}

--- a/src/pages/api/model/data/putModelData.ts
+++ b/src/pages/api/model/data/putModelData.ts
@@ -2,13 +2,12 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { authToken } from '@/service/utils/tools';
 import { connectRedis } from '@/service/redis';
+import { ModelDataStatusEnum } from '@/constants/redis';
+import { generateVector } from '@/service/events/generateVector';
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
  try {
-    let { dataId, text } = req.body as {
+    const { dataId, text, q } = req.body as { dataId: string; text: string; q?: string };
-      dataId: string;
-      text: string;
-    };
    const { authorization } = req.headers;
    if (!authorization) {
@@ -31,7 +30,17 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
    }
    // 更新
-    await redis.hSet(dataId, 'text', text);
+    await redis.sendCommand([
+      'HMSET',
+      dataId,
+      ...(q ? ['q', q, 'status', ModelDataStatusEnum.waiting] : []),
+      'text',
+      text
+    ]);
+    if (q) {
+      generateVector();
+    }
    jsonRes(res);
  } catch (err) {

--- a/src/pages/model/detail/components/InputDataModal.tsx
+++ b/src/pages/model/detail/components/InputDataModal.tsx
 import React, { useState, useCallback } from 'react';
 import {
  Box,
-  IconButton,
  Flex,
  Button,
  Modal,
@@ -9,37 +8,40 @@ import {
  ModalContent,
  ModalHeader,
  ModalCloseButton,
-  Input,
  Textarea
 } from '@chakra-ui/react';
-import { useForm, useFieldArray } from 'react-hook-form';
+import { useForm } from 'react-hook-form';
-import { postModelDataInput } from '@/api/model';
+import { postModelDataInput, putModelDataById } from '@/api/model';
 import { useToast } from '@/hooks/useToast';
-import { DeleteIcon } from '@chakra-ui/icons';
 import { customAlphabet } from 'nanoid';
 const nanoid = customAlphabet('abcdefghijklmnopqrstuvwxyz1234567890', 12);
-type FormData = { text: string; q: string };
+export type FormData = { dataId?: string; text: string; q: string };
 const InputDataModal = ({
  onClose,
  onSuccess,
-  modelId
+  modelId,
+  defaultValues = {
+    text: '',
+    q: ''
+  }
 }: {
  onClose: () => void;
  onSuccess: () => void;
  modelId: string;
+  defaultValues?: FormData;
 }) => {
  const [importing, setImporting] = useState(false);
  const { toast } = useToast();
-  const { register, handleSubmit, control } = useForm<FormData>({
+  const { register, handleSubmit } = useForm<FormData>({
-    defaultValues: {
+    defaultValues
-      text: '',
-      q: ''
-    }
  });
+  /**
+   * 确认导入新数据
+   */
  const sureImportData = useCallback(
    async (e: FormData) => {
      setImporting(true);
@@ -72,6 +74,26 @@ const InputDataModal = ({
    [modelId, onClose, onSuccess, toast]
  );
+  const updateData = useCallback(
+    async (e: FormData) => {
+      if (!e.dataId) return;
+      if (e.text === defaultValues.text && e.q === defaultValues.q) return;
+      await putModelDataById({
+        dataId: e.dataId,
+        text: e.text,
+        q: e.q === defaultValues.q ? '' : e.q
+      });
+      toast({
+        title: '修改回答成功',
+        status: 'success'
+      });
+      onClose();
+      onSuccess();
+    },
+    [defaultValues.q, onClose, onSuccess, toast]
+  );
  return (
    <Modal isOpen={true} onClose={onClose} isCentered>
      <ModalOverlay />
@@ -125,7 +147,10 @@ const InputDataModal = ({
          <Button variant={'outline'} mr={3} onClick={onClose}>
            取消
          </Button>
-          <Button isLoading={importing} onClick={handleSubmit(sureImportData)}>
+          <Button
+            isLoading={importing}
+            onClick={handleSubmit(defaultValues.dataId ? updateData : sureImportData)}
+          >
            确认导入
          </Button>
        </Flex>

--- a/src/pages/model/detail/components/ModelDataCard.tsx
+++ b/src/pages/model/detail/components/ModelDataCard.tsx
-import React, { useCallback } from 'react';
+import React, { useCallback, useState } from 'react';
 import {
  Box,
  TableContainer,
@@ -12,7 +12,6 @@ import {
  Flex,
  Button,
  useDisclosure,
-  Textarea,
  Menu,
  MenuButton,
  MenuList,
@@ -25,22 +24,21 @@ import { usePagination } from '@/hooks/usePagination';
 import {
  getModelDataList,
  delOneModelData,
-  putModelDataById,
  getModelSplitDataList,
  getExportDataList
 } from '@/api/model';
-import { DeleteIcon, RepeatIcon } from '@chakra-ui/icons';
+import { DeleteIcon, RepeatIcon, EditIcon } from '@chakra-ui/icons';
 import { useToast } from '@/hooks/useToast';
 import { useLoading } from '@/hooks/useLoading';
 import dynamic from 'next/dynamic';
 import { useMutation, useQuery } from '@tanstack/react-query';
+import type { FormData as InputDataType } from './InputDataModal';
 const InputModel = dynamic(() => import('./InputDataModal'));
 const SelectFileModel = dynamic(() => import('./SelectFileModal'));
 const SelectJsonModel = dynamic(() => import('./SelectJsonModal'));
 const ModelDataCard = ({ model }: { model: ModelSchema }) => {
-  const { toast } = useToast();
  const { Loading } = useLoading();
  const {
@@ -58,26 +56,9 @@ const ModelDataCard = ({ model }: { model: ModelSchema }) => {
    }
  });
-  const updateAnswer = useCallback(
+  const [editInputData, setEditInputData] = useState<InputDataType>();
-    async (dataId: string, text: string) => {
-      await putModelDataById({
-        dataId,
-        text
-      });
-      toast({
-        title: '修改回答成功',
-        status: 'success'
-      });
-    },
-    [toast]
-  );
  const {
-    isOpen: isOpenInputModal,
-    onOpen: onOpenInputModal,
-    onClose: onCloseInputModal
-  } = useDisclosure();
-  const {
    isOpen: isOpenSelectFileModal,
    onOpen: onOpenSelectFileModal,
    onClose: onCloseSelectFileModal
@@ -151,7 +132,16 @@ const ModelDataCard = ({ model }: { model: ModelSchema }) => {
            导入
          </MenuButton>
          <MenuList>
-            <MenuItem onClick={onOpenInputModal}>手动输入</MenuItem>
+            <MenuItem
+              onClick={() =>
+                setEditInputData({
+                  text: '',
+                  q: ''
+                })
+              }
+            >
+              手动输入
+            </MenuItem>
            <MenuItem onClick={onOpenSelectFileModal}>文件导入</MenuItem>
            <MenuItem onClick={onOpenSelectJsonModal}>JSON导入</MenuItem>
          </MenuList>
@@ -170,35 +160,39 @@ const ModelDataCard = ({ model }: { model: ModelSchema }) => {
                <Th>Question</Th>
                <Th>Text</Th>
                <Th>Status</Th>
-                <Th></Th>
+                <Th>操作</Th>
              </Tr>
            </Thead>
            <Tbody>
              {modelDataList.map((item) => (
                <Tr key={item.id}>
-                  <Td w={'350px'}>
+                  <Td>
-                    <Box fontSize={'xs'} w={'100%'} whiteSpace={'pre-wrap'} _notLast={{ mb: 1 }}>
+                    <Box fontSize={'xs'} w={'100%'} whiteSpace={'pre-wrap'}>
                      {item.q}
                    </Box>
                  </Td>
                  <Td minW={'200px'}>
-                    <Textarea
+                    <Box w={'100%'} fontSize={'xs'} whiteSpace={'pre-wrap'}>
-                      w={'100%'}
+                      {item.text}
-                      h={'100%'}
+                    </Box>
-                      defaultValue={item.text}
-                      fontSize={'xs'}
-                      resize={'both'}
-                      onBlur={(e) => {
-                        const oldVal = modelDataList.find((data) => item.id === data.id)?.text;
-                        if (oldVal !== e.target.value) {
-                          updateAnswer(item.id, e.target.value);
-                        }
-                      }}
-                    ></Textarea>
                  </Td>
-                  <Td w={'100px'}>{ModelDataStatusMap[item.status]}</Td>
+                  <Td>{ModelDataStatusMap[item.status]}</Td>
                  <Td>
                    <IconButton
+                      mr={5}
+                      icon={<EditIcon />}
+                      variant={'outline'}
+                      aria-label={'delete'}
+                      size={'sm'}
+                      onClick={() =>
+                        setEditInputData({
+                          dataId: item.id,
+                          q: item.q,
+                          text: item.text
+                        })
+                      }
+                    />
+                    <IconButton
                      icon={<DeleteIcon />}
                      variant={'outline'}
                      colorScheme={'gray'}
@@ -221,8 +215,13 @@ const ModelDataCard = ({ model }: { model: ModelSchema }) => {
      </Box>
      <Loading loading={isLoading} fixed={false} />
-      {isOpenInputModal && (
+      {editInputData !== undefined && (
-        <InputModel modelId={model._id} onClose={onCloseInputModal} onSuccess={refetchData} />
+        <InputModel
+          modelId={model._id}
+          defaultValues={editInputData}
+          onClose={() => setEditInputData(undefined)}
+          onSuccess={refetchData}
+        />
      )}
      {isOpenSelectFileModal && (
        <SelectFileModel