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

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

@@ -50,6 +50,10 @@ export class AIChatUtil implements IApiAIChatUtil {
   private PRE_ERROR_MESSAGE: string =
     'Expected content-type to be text/event-stream, Actual: application/json;charset=UTF-8';
 
+  // 无流内容回来
+  private NONE_ERROR_MESSAGE: string =
+    'Expected content-type to be text/event-stream, Actual: null';
+
   /**
    * 获取AI聊天对象
    */
@@ -524,7 +528,10 @@ export class AIChatUtil implements IApiAIChatUtil {
           (msg: IPortalAsyncAction) => {
             // 20: 持续回答中，消息会持续推送。同一个消息 id 会显示在同一个框内
             if (msg.actionstate === 20 && msg.actionresult) {
-              aiChat.setActiveAsyncacitonId(msg.asyncacitonid);
+              // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
+              if (!aiChat.getActiveAsyncacitonId()) {
+                aiChat.setActiveAsyncacitonId(msg.asyncacitonid);
+              }
               aiChat.addMessage({
                 messageid: id,
                 state: msg.actionstate,
@@ -594,6 +601,8 @@ export class AIChatUtil implements IApiAIChatUtil {
       } finally {
         // 标记当前消息已经交互完成
         aiChat.completeMessage(id, true);
+        // 清空异步消息任务
+        aiChat.setActiveAsyncacitonId('');
         return true;
       }
     };
@@ -639,6 +648,127 @@ export class AIChatUtil implements IApiAIChatUtil {
       // 标记当前消息已经交互完成
       await aiChat.completeMessage(id, true);
     };
+    // 恢复会话
+    chatOptions.resumeQuestion = async (
+      aiChat: any,
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+      arr: IChatMessage[],
+    ) => {
+      if (other.asyncacitonid) {
+        id = createUUID();
+        abortController = new AbortController();
+        const deService = await ibiz.hub
+          .getApp(ctx.srfappid)
+          .deService.getService(ctx, other.appDataEntityId);
+        const resumeRequestData: IData = { asyncacitonid: other.asyncacitonid };
+        if (other.sessionid) {
+          Object.assign(resumeRequestData, {
+            sessionid: other.sessionid,
+          });
+        }
+        if (other.srfaiagent) {
+          Object.assign(resumeRequestData, {
+            srfaiagent: other.srfaiagent,
+          });
+        }
+        if (other.srfproject)
+          Object.assign(resumeRequestData, {
+            srfproject: other.srfproject,
+          });
+        try {
+          await deService.aiResumeSse(
+            (msg: IPortalAsyncAction) => {
+              // 20: 持续回答中，消息会持续推送。同一个消息 id 会显示在同一个框内
+              if (msg.actionstate === 20 && msg.actionresult) {
+                // 设置当前聊天在加载中
+                aiChat.isLoading.value = true;
+                // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
+                if (!aiChat.getActiveAsyncacitonId()) {
+                  aiChat.setActiveAsyncacitonId(msg.asyncacitonid);
+                }
+                aiChat.addMessage({
+                  messageid: id,
+                  state: msg.actionstate,
+                  type: 'DEFAULT',
+                  role: 'ASSISTANT',
+                  content: msg.actionresult as string,
+                  status: 'pending',
+                });
+              }
+              // 30: 回答完成，包含具体所有消息内容。直接覆盖之前的临时拼接消息
+              else if (msg.actionstate === 30 && msg.actionresult) {
+                const result = JSON.parse(msg.actionresult as string);
+                if (result.usage) {
+                  aiChat.updateMsgUsage(id, result.usage);
+                }
+                const choices = result.choices;
+                if (choices && choices.length > 0) {
+                  aiChat.replaceMessage({
+                    messageid: id,
+                    state: msg.actionstate,
+                    type: 'DEFAULT',
+                    role: 'ASSISTANT',
+                    content: choices[0].content || '',
+                    realmessageid: choices[0].messageid,
+                    status: 'sent',
+                  });
+                }
+              }
+              // 40: 回答报错，展示错误信息
+              else if (msg.actionstate === 40) {
+                aiChat.replaceMessage({
+                  messageid: id,
+                  state: msg.actionstate,
+                  type: 'ERROR',
+                  role: 'ASSISTANT',
+                  errorText: msg.actionresult as string,
+                  status: 'failed',
+                });
+              }
+            },
+            abortController,
+            ctx,
+            { srfactag: param.srfactag },
+            { ...resumeRequestData },
+          );
+        } catch (error) {
+          // 等于NONE_ERROR_MESSAGE，代码该异步行为已经完成，不需要做处理
+          if (error && (error as Error).message !== this.NONE_ERROR_MESSAGE) {
+            if (error && (error as Error).message !== this.PRE_ERROR_MESSAGE) {
+              aiChat.replaceMessage({
+                messageid: id,
+                state: 40,
+                type: 'ERROR',
+                role: 'ASSISTANT',
+                errorText:
+                  (error as IData).message || ibiz.i18n.t('app.aiError'),
+                status: 'failed',
+              });
+            } else {
+              const lastMessage = arr[arr.length - 1];
+              aiChat.replaceMessage(
+                {
+                  ...lastMessage,
+                  state: 40,
+                },
+                false,
+              );
+            }
+          }
+          abortController?.abort();
+        } finally {
+          // 设置当前聊天加载状态已完成
+          aiChat.isLoading.value = false;
+          // 标记当前消息已经交互完成
+          aiChat.completeMessage(id, true);
+          // 清空异步消息任务
+          aiChat.setActiveAsyncacitonId('');
+          return true;
+        }
+      }
+    };
     // 调整方向
     chatOptions.supplementQuestion = async (
       aiChat: any,
@@ -1401,7 +1531,10 @@ export class AIChatUtil implements IApiAIChatUtil {
           (msg: IPortalAsyncAction) => {
             // 20: 持续回答中，消息会持续推送。同一个消息 id 会显示在同一个框内
             if (msg.actionstate === 20 && msg.actionresult) {
-              aiChat.setActiveAsyncacitonId(msg.asyncacitonid);
+              // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
+              if (!aiChat.getActiveAsyncacitonId()) {
+                aiChat.setActiveAsyncacitonId(msg.asyncacitonid);
+              }
               aiChat.addMessage({
                 messageid: id,
                 state: msg.actionstate,
@@ -1469,6 +1602,8 @@ export class AIChatUtil implements IApiAIChatUtil {
         }
         abortController?.abort();
       } finally {
+        // 清空异步消息任务
+        aiChat.setActiveAsyncacitonId('');
         // 标记当前消息已经交互完成
         aiChat.completeMessage(id, true);
         return true;
@@ -1516,6 +1651,127 @@ export class AIChatUtil implements IApiAIChatUtil {
       // 标记当前消息已经交互完成
       await aiChat.completeMessage(id, true);
     };
+    // 恢复会话
+    chatOptions.resumeQuestion = async (
+      aiChat: any,
+      ctx: IContext,
+      param: IParams,
+      other: IParams,
+      arr: IChatMessage[],
+    ) => {
+      if (other.asyncacitonid) {
+        id = createUUID();
+        abortController = new AbortController();
+        const deService = await ibiz.hub
+          .getApp(ctx.srfappid)
+          .deService.getService(ctx, other.appDataEntityId);
+        const resumeRequestData: IData = { asyncacitonid: other.asyncacitonid };
+        if (other.sessionid) {
+          Object.assign(resumeRequestData, {
+            sessionid: other.sessionid,
+          });
+        }
+        if (other.srfaiagent) {
+          Object.assign(resumeRequestData, {
+            srfaiagent: other.srfaiagent,
+          });
+        }
+        if (other.srfproject)
+          Object.assign(resumeRequestData, {
+            srfproject: other.srfproject,
+          });
+        try {
+          await deService.aiResumeSse(
+            (msg: IPortalAsyncAction) => {
+              // 20: 持续回答中，消息会持续推送。同一个消息 id 会显示在同一个框内
+              if (msg.actionstate === 20 && msg.actionresult) {
+                // 设置当前聊天在加载中
+                aiChat.isLoading.value = true;
+                // 不存在才设置，存在则不设置（下方设置的时候需要去存储话题数据中）
+                if (!aiChat.getActiveAsyncacitonId()) {
+                  aiChat.setActiveAsyncacitonId(msg.asyncacitonid);
+                }
+                aiChat.addMessage({
+                  messageid: id,
+                  state: msg.actionstate,
+                  type: 'DEFAULT',
+                  role: 'ASSISTANT',
+                  content: msg.actionresult as string,
+                  status: 'pending',
+                });
+              }
+              // 30: 回答完成，包含具体所有消息内容。直接覆盖之前的临时拼接消息
+              else if (msg.actionstate === 30 && msg.actionresult) {
+                const result = JSON.parse(msg.actionresult as string);
+                if (result.usage) {
+                  aiChat.updateMsgUsage(id, result.usage);
+                }
+                const choices = result.choices;
+                if (choices && choices.length > 0) {
+                  aiChat.replaceMessage({
+                    messageid: id,
+                    state: msg.actionstate,
+                    type: 'DEFAULT',
+                    role: 'ASSISTANT',
+                    content: choices[0].content || '',
+                    realmessageid: choices[0].messageid,
+                    status: 'sent',
+                  });
+                }
+              }
+              // 40: 回答报错，展示错误信息
+              else if (msg.actionstate === 40) {
+                aiChat.replaceMessage({
+                  messageid: id,
+                  state: msg.actionstate,
+                  type: 'ERROR',
+                  role: 'ASSISTANT',
+                  errorText: msg.actionresult as string,
+                  status: 'failed',
+                });
+              }
+            },
+            abortController,
+            ctx,
+            { srfactag: param.srfactag },
+            { ...resumeRequestData },
+          );
+        } catch (error) {
+          // 等于NONE_ERROR_MESSAGE，代码该异步行为已经完成，不需要做处理
+          if (error && (error as Error).message !== this.NONE_ERROR_MESSAGE) {
+            if (error && (error as Error).message !== this.PRE_ERROR_MESSAGE) {
+              aiChat.replaceMessage({
+                messageid: id,
+                state: 40,
+                type: 'ERROR',
+                role: 'ASSISTANT',
+                errorText:
+                  (error as IData).message || ibiz.i18n.t('app.aiError'),
+                status: 'failed',
+              });
+            } else {
+              const lastMessage = arr[arr.length - 1];
+              aiChat.replaceMessage(
+                {
+                  ...lastMessage,
+                  state: 40,
+                },
+                false,
+              );
+            }
+          }
+          abortController?.abort();
+        } finally {
+          // 设置当前聊天加载状态已完成
+          aiChat.isLoading.value = false;
+          // 标记当前消息已经交互完成
+          aiChat.completeMessage(id, true);
+          // 清空异步消息任务
+          aiChat.setActiveAsyncacitonId('');
+          return true;
+        }
+      }
+    };
     // 调整方向
     chatOptions.supplementQuestion = async (
       aiChat: any,