feat: chat status

client/src/pages/api/openapi/v1/chat/completions.ts

@@ -116,10 +116,12 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
     }
 
     // 创建响应流
-    res.setHeader('Content-Type', 'text/event-stream;charset=utf-8');
-    res.setHeader('Access-Control-Allow-Origin', '*');
-    res.setHeader('X-Accel-Buffering', 'no');
-    res.setHeader('Cache-Control', 'no-cache, no-transform');
+    if (stream) {
+      res.setHeader('Content-Type', 'text/event-stream;charset=utf-8');
+      res.setHeader('Access-Control-Allow-Origin', '*');
+      res.setHeader('X-Accel-Buffering', 'no');
+      res.setHeader('Cache-Control', 'no-cache, no-transform');
+    }
 
     /* start process */
     const { responseData, answerText } = await dispatchModules({
@@ -320,6 +322,14 @@ export async function dispatchModules({
     if (res.closed) return Promise.resolve();
     console.log('run=========', module.flowType);
 
+    if (stream && module.showStatus) {
+      responseStatus({
+        res,
+        name: module.name,
+        status: 'running'
+      });
+    }
+
     // get fetch params
     const params: Record<string, any> = {};
     module.inputs.forEach((item: any) => {
@@ -370,7 +380,9 @@ function loadModules(
   return modules.map((module) => {
     return {
       moduleId: module.moduleId,
+      name: module.name,
       flowType: module.flowType,
+      showStatus: module.showStatus,
       inputs: module.inputs
         .filter((item) => item.connected) // filter unconnected target input
         .map((item) => {
@@ -401,3 +413,23 @@ function loadModules(
     };
   });
 }
+
+function responseStatus({
+  res,
+  status,
+  name
+}: {
+  res: NextApiResponse;
+  status: 'running' | 'finish';
+  name?: string;
+}) {
+  if (!name) return;
+  sseResponse({
+    res,
+    event: sseResponseEventEnum.moduleStatus,
+    data: JSON.stringify({
+      status,
+      name
+    })
+  });
+}

client/src/pages/app/detail/components/AdEdit/components/TemplateList.tsx

@@ -1,10 +1,9 @@
 import React, { useMemo } from 'react';
 import { Box, Flex } from '@chakra-ui/react';
 import { ModuleTemplates } from '@/constants/flow/ModuleTemplate';
-import { FlowModuleTemplateType } from '@/types/flow';
+import { FlowModuleItemType, FlowModuleTemplateType } from '@/types/flow';
 import type { Node, XYPosition } from 'reactflow';
 import { useGlobalStore } from '@/store/global';
-import type { AppModuleItemType } from '@/types/app';
 import Avatar from '@/components/Avatar';
 import { FlowModuleTypeEnum } from '@/constants/flow';
 
@@ -14,7 +13,7 @@ const ModuleTemplateList = ({
   onAddNode,
   onClose
 }: {
-  nodes?: Node<AppModuleItemType>[];
+  nodes?: Node<FlowModuleItemType>[];
   isOpen: boolean;
   onAddNode: (e: { template: FlowModuleTemplateType; position: XYPosition }) => void;
   onClose: () => void;

client/src/pages/app/detail/components/AdEdit/index.tsx

@@ -158,7 +158,9 @@ const AppEdit = ({ app, fullScreen, onFullScreen }: Props) => {
   const flow2AppModules = useCallback(() => {
     const modules: AppModuleItemType[] = nodes.map((item) => ({
       moduleId: item.data.moduleId,
+      name: item.data.name,
       flowType: item.data.flowType,
+      showStatus: item.data.showStatus,
       position: item.position,
       inputs: item.data.inputs.map((item) => ({
         ...item,