@ibiz-template/vue3-components 0.7.41-alpha.150 → 0.7.41-alpha.152

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

@@ -378,28 +378,9 @@ export class AIChatUtil implements IApiAIChatUtil {
     let id: string = '';
     let abortController: AbortController;
     const { chatInstance, view, ctrl } = args;
-    // 查询项目
-    chatOptions.fetchProject = async (
-      ctx: IContext,
-      param: IParams,
-      other: IParams,
-    ) => {
-      const deService = await ibiz.hub
-        .getApp(ctx.srfappid)
-        .deService.getService(ctx, param.appDataEntityId);
-      const response = await deService.aiChatProject(
-        ctx,
-        { srfactag: param.srfactag },
-        other,
-      );
-      let result: IData[] = [];
-      if (response.data) {
-        result = Array.isArray(response.data)
-          ? response.data
-          : response.data.children;
-      }
-      return result;
-    };
+    // 文件系统操作参数
+    const fileSystemOptions = await this.getAIFileSystemOptions();
+    chatOptions.fileSystemOptions = fileSystemOptions;
     // 请求历史
     chatOptions.history = async (
       ctx: IContext,
@@ -483,7 +464,10 @@ export class AIChatUtil implements IApiAIChatUtil {
       mcpservers: string | undefined,
       sessionmeta: IData | undefined,
       srfproject: string | undefined,
+      localMsgID: string,
     ) => {
+      // 更新标记，主要用于更新用户消息真实消息ID（每条消息都会回asextparam，只需要设置一次即可）
+      let updateFlag = true;
       id = createUUID();
       abortController = new AbortController();
       const deService = await ibiz.hub
@@ -524,10 +508,19 @@ 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);
+                updateFlag = false;
+              }
               // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
               if (!aiChat.getActiveAsyncacitonId()) {
                 aiChat.setActiveAsyncacitonId(msg.asyncacitonid);
@@ -599,6 +592,8 @@ export class AIChatUtil implements IApiAIChatUtil {
         }
         abortController?.abort();
       } finally {
+        // 移除等待回复消息
+        aiChat.removeWaitingMessage(id);
         // 标记当前消息已经交互完成
         aiChat.completeMessage(id, true);
         // 清空异步消息任务
@@ -678,10 +673,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;
                 // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
@@ -759,6 +758,8 @@ export class AIChatUtil implements IApiAIChatUtil {
           }
           abortController?.abort();
         } finally {
+          // 移除等待回复消息
+          aiChat.removeWaitingMessage(id);
           // 设置当前聊天加载状态已完成
           aiChat.isLoading.value = false;
           // 标记当前消息已经交互完成
@@ -1383,28 +1384,9 @@ export class AIChatUtil implements IApiAIChatUtil {
     let id: string = '';
     let abortController: AbortController;
     const { chatInstance, view, ctrl } = args;
-    // 查询项目
-    chatOptions.fetchProject = async (
-      ctx: IContext,
-      param: IParams,
-      other: IParams,
-    ) => {
-      const deService = await ibiz.hub
-        .getApp(ctx.srfappid)
-        .deService.getService(ctx, param.appDataEntityId);
-      const response = await deService.aiChatProject(
-        ctx,
-        { srfactag: param.srfactag },
-        other,
-      );
-      let result: IData[] = [];
-      if (response.data) {
-        result = Array.isArray(response.data)
-          ? response.data
-          : response.data.children;
-      }
-      return result;
-    };
+    // 文件系统操作参数
+    const fileSystemOptions = await this.getAIFileSystemOptions();
+    chatOptions.fileSystemOptions = fileSystemOptions;
     // 查询历史
     chatOptions.history = async (
       ctx: IContext,
@@ -1488,7 +1470,10 @@ export class AIChatUtil implements IApiAIChatUtil {
       mcpservers: string | undefined,
       sessionmeta: IData | undefined,
       srfproject: string | undefined,
+      localMsgID: string,
     ) => {
+      // 更新标记，主要用于更新用户消息真实消息ID（每条消息都会回asextparam，只需要设置一次即可）
+      let updateFlag = true;
       id = createUUID();
       abortController = new AbortController();
       const deService = await ibiz.hub
@@ -1527,10 +1512,19 @@ 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);
+                updateFlag = false;
+              }
               // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
               if (!aiChat.getActiveAsyncacitonId()) {
                 aiChat.setActiveAsyncacitonId(msg.asyncacitonid);
@@ -1602,6 +1596,8 @@ export class AIChatUtil implements IApiAIChatUtil {
         }
         abortController?.abort();
       } finally {
+        // 移除等待回复消息
+        aiChat.removeWaitingMessage(id);
         // 清空异步消息任务
         aiChat.setActiveAsyncacitonId('');
         // 标记当前消息已经交互完成
@@ -1681,10 +1677,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;
                 // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
@@ -1762,6 +1762,8 @@ export class AIChatUtil implements IApiAIChatUtil {
           }
           abortController?.abort();
         } finally {
+          // 移除等待回复消息
+          aiChat.removeWaitingMessage(id);
           // 设置当前聊天加载状态已完成
           aiChat.isLoading.value = false;
           // 标记当前消息已经交互完成
@@ -2415,4 +2417,88 @@ export class AIChatUtil implements IApiAIChatUtil {
     };
     return resourceOptions;
   }
+
+  /**
+   * @description 获取AI文件系统相关参数
+   * @returns {*}  {Promise<IData>}
+   * @memberof AIChatUtil
+   */
+  async getAIFileSystemOptions(): Promise<IData> {
+    const fileSystemOptions: IData = {};
+    // 查询文件列表
+    fileSystemOptions.getFileList = async (
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+    ): Promise<IData | undefined> => {
+      const deService = await ibiz.hub
+        .getApp(ctx.srfappid)
+        .deService.getService(ctx, param.appDataEntityId);
+      const response = await deService.aiChatProject(
+        ctx,
+        { srfactag: param.srfactag },
+        { ...other },
+      );
+      if (response.ok) return response.data;
+      return undefined;
+    };
+    // 上传文件
+    fileSystemOptions.uploadFile = async (
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+    ): Promise<IData | undefined> => {
+      const deService = await ibiz.hub
+        .getApp(ctx.srfappid)
+        .deService.getService(ctx, param.appDataEntityId);
+      const response = await deService.aiChatProject(
+        ctx,
+        { srfactag: param.srfactag },
+        { ...other, action: 'UPLOAD' },
+      );
+      if (response.ok) return response.data;
+      return undefined;
+    };
+    // 获取文件内容
+    fileSystemOptions.getContent = async (
+      ctx: IContext,
+      param: IParams,
+      id: string,
+      type: string,
+      cat: string,
+    ): Promise<string | IData | undefined> => {
+      const app = await ibiz.hub.getAppAsync(ctx.appId);
+      const requestUrl = `/ibizutil/downloadtxt/${cat ? `${cat}/` : ''}${id}?type=${type}`;
+      const res = await app.net.get(requestUrl);
+      if (res && res.ok && res.data) {
+        return res.data;
+      }
+    };
+    // 获取图片内容
+    fileSystemOptions.getImgContent = async (
+      ctx: IContext,
+      param: IParams,
+      id: string,
+      type: string,
+      cat: string,
+    ): Promise<string> => {
+      const { downloadUrl } = ibiz.util.file.calcFileUpDownUrl(
+        ctx,
+        param,
+        {},
+        { osscat: cat },
+      );
+      let content = downloadUrl.replace('%fileId%', id);
+      if (type === 'svg') {
+        const response = await ibiz.net.request(content, {
+          method: 'get',
+          responseType: 'blob',
+          baseURL: '',
+        });
+        content = await response.data.text();
+      }
+      return content;
+    };
+    return fileSystemOptions;
+  }
 }