@ibiz-template/vue3-components 0.7.41-alpha.151 → 0.7.41-alpha.153

package/src/util/ai-chat-util/ai-chat-util.ts

@@ -345,6 +345,14 @@ export class AIChatUtil implements IApiAIChatUtil {
         ibiz.config.common.aiChatMessagePageSize;
     }
 
+    // AI会话是否启用工作空间文件系统相关操作
+    if (editorParams.srfaienableworkspace) {
+      chatOptions.enableWorkSpace =
+        editorParams.srfaienableworkspace === 'true';
+    } else {
+      chatOptions.enableWorkSpace = ibiz.config.common.aiEnableWorkSpace;
+    }
+
     // 智能体选项(由内部获取，优化打开速度)
     chatOptions.fetchAgentList = (args: IParams = {}) => {
       const { query, agentid } = args;
@@ -378,9 +386,17 @@ export class AIChatUtil implements IApiAIChatUtil {
     let id: string = '';
     let abortController: AbortController;
     const { chatInstance, view, ctrl } = args;
+
     // 文件系统操作参数
-    const fileSystemOptions = await this.getAIFileSystemOptions();
+    // AI聊天框文件预览模式
+    let filePreviewMode = ibiz.config.common.aiFilePreviewMode;
+    if (editorParams.srfaifilepreviewmode) {
+      filePreviewMode = editorParams.srfaifilepreviewmode;
+    }
+    const fileSystemOptions =
+      await this.getAIFileSystemOptions(filePreviewMode);
     chatOptions.fileSystemOptions = fileSystemOptions;
+
     // 请求历史
     chatOptions.history = async (
       ctx: IContext,
@@ -508,10 +524,14 @@ export class AIChatUtil implements IApiAIChatUtil {
         if (extParams && Object.keys(extParams).length > 0) {
           Object.assign(questionRequestData, { srfextparams: extParams });
         }
+        // 新增等待回复消息（AI首条内容到达前显示loading气泡）
+        aiChat.addWaitingMessage(id);
         await deService.aiChatSse(
           (msg: IPortalAsyncAction) => {
             // 20: 持续回答中，消息会持续推送。同一个消息 id 会显示在同一个框内
             if (msg.actionstate === 20 && msg.actionresult) {
+              // 移除等待回复消息
+              aiChat.removeWaitingMessage(id);
               // 更新用户消息真实id
               if (msg.asextparam && localMsgID && updateFlag) {
                 aiChat.updateMsgRealMessageID(localMsgID, msg.asextparam);
@@ -588,6 +608,8 @@ export class AIChatUtil implements IApiAIChatUtil {
         }
         abortController?.abort();
       } finally {
+        // 移除等待回复消息
+        aiChat.removeWaitingMessage(id);
         // 标记当前消息已经交互完成
         aiChat.completeMessage(id, true);
         // 清空异步消息任务
@@ -667,10 +689,14 @@ export class AIChatUtil implements IApiAIChatUtil {
             srfproject: other.srfproject,
           });
         try {
+          // 新增等待回复消息（AI首条内容到达前显示loading气泡）
+          aiChat.addWaitingMessage(id);
           await deService.aiResumeSse(
             (msg: IPortalAsyncAction) => {
               // 20: 持续回答中，消息会持续推送。同一个消息 id 会显示在同一个框内
               if (msg.actionstate === 20 && msg.actionresult) {
+                // 移除等待回复消息
+                aiChat.removeWaitingMessage(id);
                 // 设置当前聊天在加载中
                 aiChat.isLoading.value = true;
                 // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
@@ -748,6 +774,8 @@ export class AIChatUtil implements IApiAIChatUtil {
           }
           abortController?.abort();
         } finally {
+          // 移除等待回复消息
+          aiChat.removeWaitingMessage(id);
           // 设置当前聊天加载状态已完成
           aiChat.isLoading.value = false;
           // 标记当前消息已经交互完成
@@ -1342,6 +1370,14 @@ export class AIChatUtil implements IApiAIChatUtil {
         ibiz.config.common.aiChatMessagePageSize;
     }
 
+    // AI会话是否启用工作空间文件系统相关操作
+    if (params.hasOwnProperty('srfaienableworkspace')) {
+      chatOptions.enableWorkSpace = params.srfaienableworkspace === 'true';
+      delete params.srfaienableworkspace;
+    } else {
+      chatOptions.enableWorkSpace = ibiz.config.common.aiEnableWorkSpace;
+    }
+
     // 智能体选项(由内部获取，优化打开速度)
     chatOptions.fetchAgentList = (args: IParams = {}) => {
       const { query, agentid } = args;
@@ -1373,8 +1409,17 @@ export class AIChatUtil implements IApiAIChatUtil {
     let abortController: AbortController;
     const { chatInstance, view, ctrl } = args;
     // 文件系统操作参数
-    const fileSystemOptions = await this.getAIFileSystemOptions();
+
+    // AI聊天框文件预览模式
+    let filePreviewMode = ibiz.config.common.aiFilePreviewMode;
+    if (params.hasOwnProperty('srfaifilepreviewmode')) {
+      filePreviewMode = params.srfaifilepreviewmode;
+      delete params.srfaifilepreviewmode;
+    }
+    const fileSystemOptions =
+      await this.getAIFileSystemOptions(filePreviewMode);
     chatOptions.fileSystemOptions = fileSystemOptions;
+
     // 查询历史
     chatOptions.history = async (
       ctx: IContext,
@@ -1500,10 +1545,14 @@ export class AIChatUtil implements IApiAIChatUtil {
         if (extParams && Object.keys(extParams).length > 0) {
           Object.assign(questionRequestData, { srfextparams: extParams });
         }
+        // 新增等待回复消息（AI首条内容到达前显示loading气泡）
+        aiChat.addWaitingMessage(id);
         await deService.aiChatSse(
           (msg: IPortalAsyncAction) => {
             // 20: 持续回答中，消息会持续推送。同一个消息 id 会显示在同一个框内
             if (msg.actionstate === 20 && msg.actionresult) {
+              // 移除等待回复消息
+              aiChat.removeWaitingMessage(id);
               // 更新用户消息真实id
               if (msg.asextparam && localMsgID && updateFlag) {
                 aiChat.updateMsgRealMessageID(localMsgID, msg.asextparam);
@@ -1580,6 +1629,8 @@ export class AIChatUtil implements IApiAIChatUtil {
         }
         abortController?.abort();
       } finally {
+        // 移除等待回复消息
+        aiChat.removeWaitingMessage(id);
         // 清空异步消息任务
         aiChat.setActiveAsyncacitonId('');
         // 标记当前消息已经交互完成
@@ -1659,10 +1710,14 @@ export class AIChatUtil implements IApiAIChatUtil {
             srfproject: other.srfproject,
           });
         try {
+          // 新增等待回复消息（AI首条内容到达前显示loading气泡）
+          aiChat.addWaitingMessage(id);
           await deService.aiResumeSse(
             (msg: IPortalAsyncAction) => {
               // 20: 持续回答中，消息会持续推送。同一个消息 id 会显示在同一个框内
               if (msg.actionstate === 20 && msg.actionresult) {
+                // 移除等待回复消息
+                aiChat.removeWaitingMessage(id);
                 // 设置当前聊天在加载中
                 aiChat.isLoading.value = true;
                 // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
@@ -1740,6 +1795,8 @@ export class AIChatUtil implements IApiAIChatUtil {
           }
           abortController?.abort();
         } finally {
+          // 移除等待回复消息
+          aiChat.removeWaitingMessage(id);
           // 设置当前聊天加载状态已完成
           aiChat.isLoading.value = false;
           // 标记当前消息已经交互完成
@@ -2399,8 +2456,42 @@ export class AIChatUtil implements IApiAIChatUtil {
    * @returns {*}  {Promise<IData>}
    * @memberof AIChatUtil
    */
-  async getAIFileSystemOptions(): Promise<IData> {
+  async getAIFileSystemOptions(
+    filePreviewMode: 'DEFAULT' | 'CUSTOM',
+  ): Promise<IData> {
     const fileSystemOptions: IData = {};
+    // 文件预览模式
+    fileSystemOptions.filePreviewMode = filePreviewMode;
+    // 获取自定义文件预览路径
+    fileSystemOptions.getCustomFilePreviewUrl = async (
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+    ): Promise<string> => {
+      const filePreviewAddress = ibiz.env.customParams?.file_preview_address;
+      if (!filePreviewAddress) {
+        ibiz.message.error(
+          ibiz.i18n.t('util.aiChartUtil.noFilePreviewAddress'),
+        );
+        return '';
+      }
+      const { id, name, folder } = other;
+      let { downloadUrl } = ibiz.util.file.calcFileUpDownUrl(
+        ctx,
+        param,
+        {},
+        { osscat: folder },
+      );
+      downloadUrl = downloadUrl.replace('%fileId%', id);
+      const { protocol, hostname, port } = window.location;
+      // 没有端口使用默认端口
+      const actualPort = port || (protocol === 'https:' ? '443' : '80');
+      const windowInfo = `${protocol}//${hostname}:${actualPort}`;
+      const filedownloadurl = `${windowInfo}${downloadUrl}?fullfilename=${name}`;
+      const b64Encoded = ibiz.util.base64.encode(filedownloadurl);
+      const previewUrl = `${filePreviewAddress}/onlinePreview?url=${encodeURIComponent(b64Encoded)}`;
+      return previewUrl;
+    };
     // 查询文件列表
     fileSystemOptions.getFileList = async (
       ctx: IContext,
@@ -2430,11 +2521,91 @@ export class AIChatUtil implements IApiAIChatUtil {
       const response = await deService.aiChatProject(
         ctx,
         { srfactag: param.srfactag },
-        { ...other, action: 'UPLOAD' },
+        { action: 'UPLOAD', ...other },
       );
       if (response.ok) return response.data;
       return undefined;
     };
+    // 新建目录
+    fileSystemOptions.createFolder = async (
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+    ): Promise<IData> => {
+      const deService = await ibiz.hub
+        .getApp(ctx.srfappid)
+        .deService.getService(ctx, param.appDataEntityId);
+      const response = await deService.aiChatProject(
+        ctx,
+        { srfactag: param.srfactag },
+        { action: 'MKDIR', ...other },
+      );
+      return response.data;
+    };
+    // 新建文件
+    fileSystemOptions.createFile = async (
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+    ): Promise<IData> => {
+      const deService = await ibiz.hub
+        .getApp(ctx.srfappid)
+        .deService.getService(ctx, param.appDataEntityId);
+      const response = await deService.aiChatProject(
+        ctx,
+        { srfactag: param.srfactag },
+        { action: 'WRITE', ...other },
+      );
+      return response.data;
+    };
+    // 移动文件夹/文件
+    fileSystemOptions.move = async (
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+    ): Promise<IData> => {
+      const deService = await ibiz.hub
+        .getApp(ctx.srfappid)
+        .deService.getService(ctx, param.appDataEntityId);
+      const response = await deService.aiChatProject(
+        ctx,
+        { srfactag: param.srfactag },
+        { action: 'MOVE', ...other },
+      );
+      return response.data;
+    };
+    // 删除文件夹/文件
+    fileSystemOptions.delete = async (
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+    ): Promise<IData> => {
+      const deService = await ibiz.hub
+        .getApp(ctx.srfappid)
+        .deService.getService(ctx, param.appDataEntityId);
+      const response = await deService.aiChatProject(
+        ctx,
+        { srfactag: param.srfactag },
+        { recursive: true, action: 'DELETE', ...other },
+      );
+      return response.data;
+    };
+    // 读取文件内容
+    fileSystemOptions.readFile = async (
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+    ): Promise<IData> => {
+      const deService = await ibiz.hub
+        .getApp(ctx.srfappid)
+        .deService.getService(ctx, param.appDataEntityId);
+      const response = await deService.aiChatProject(
+        ctx,
+        { srfactag: param.srfactag },
+        { action: 'READ', ...other },
+      );
+      return response.data;
+    };
     // 获取文件内容
     fileSystemOptions.getContent = async (
       ctx: IContext,