@ai-group/chat-sdk 2.1.13 → 2.1.14

package/dist/cjs/hooks/useADKChat.d.ts

@@ -3,7 +3,7 @@ import { DebugOptions, FunctionCall, SendContent, SessionData, Event, IMessage,
 import { type ChatConfig } from "../services/api";
 declare function useADKChat({ url, token, config, type, onError, onMessage, onSuccess, onStream, }: DebugOptions): {
     appInfo: ChatConfig | null;
-    startChat: ({ text, files, functionResponse }: SendContent) => Promise<void>;
+    startChat: ({ text, files, functionResponse, }: SendContent) => Promise<void>;
     reChat: () => void;
     stopChat: () => void;
     clearChat: () => void;
@@ -24,6 +24,6 @@ declare function useADKChat({ url, token, config, type, onError, onMessage, onSu
     setCurrentSessionDetail: (sessionId: string) => Promise<void>;
     setMessages: import("react").Dispatch<import("react").SetStateAction<IMessage[]>>;
     storeEvents: (part: Part | null, event: any) => void;
-    storeMessage: (part: Part | null, event: Event, role: 'user' | 'bot') => void;
+    storeMessage: (part: Part | null, event: Event, role: "user" | "bot") => void;
 };
 export default useADKChat;

package/dist/cjs/hooks/useADKChat.js

@@ -64,8 +64,12 @@ function useADKChat({
   const [appInfo, setAppInfo] = (0, import_react.useState)(null);
   const [initialized, setInitialized] = (0, import_react.useState)(false);
   const [sessionList, setSessionList] = (0, import_react.useState)([]);
-  const [currentSessionId, setCurrentSessionId] = (0, import_react.useState)("");
-  const [sessionPagination, setSessionPagination] = (0, import_react.useState)({ total: 0, pageNum: 1, pageSize: import_constants.SESSION_PAGE_SIZE });
+  const [currentSessionId, setCurrentSessionId] = (0, import_react.useState)(() => (0, import_uuid.v4)());
+  const [sessionPagination, setSessionPagination] = (0, import_react.useState)({
+    total: 0,
+    pageNum: 1,
+    pageSize: import_constants.SESSION_PAGE_SIZE
+  });
   const [prologue, setPrologue] = (0, import_react.useState)("");
   const [suggestedQuestions, setSuggestedQuestions] = (0, import_react.useState)([]);
   const [messages, setMessages] = (0, import_react.useState)([]);
@@ -105,7 +109,6 @@ function useADKChat({
     if (messages.length === 0 && suggested_questions.length) {
       setSuggestedQuestions(suggested_questions);
     }
-    ;
   };
   const insertPrologue = (prologue2) => {
     if (prologue2) {
@@ -147,185 +150,210 @@ function useADKChat({
     newEventData.set(event.id, updatedEvent);
     eventDataRef.current = newEventData;
   }, []);
-  const storeMessage = (0, import_react.useCallback)((part, event, role) => {
-    var _a, _b;
-    const msg = {
-      id: (0, import_uuid.v4)(),
-      author: event.author,
-      invocationId: event.invocationId,
-      eventId: event.id,
-      timestamp: event.timestamp,
-      role
-    };
-    if (part) {
-      if (part.inlineData) {
-        msg.inlineData = {
-          displayName: part.inlineData.displayName,
-          data: part.inlineData.data,
-          mimeType: part.inlineData.mimeType
-        };
-      } else if (part.text) {
-        msg.text = part.text;
-        msg.thought = part.thought;
-        if ((_b = (_a = event == null ? void 0 : event.groundingMetadata) == null ? void 0 : _a.searchEntryPoint) == null ? void 0 : _b.renderedContent) {
-          msg.renderedContent = event.groundingMetadata.searchEntryPoint.renderedContent;
+  const storeMessage = (0, import_react.useCallback)(
+    (part, event, role) => {
+      var _a, _b;
+      const msg = {
+        id: (0, import_uuid.v4)(),
+        author: event.author,
+        invocationId: event.invocationId,
+        eventId: event.id,
+        timestamp: event.timestamp,
+        role
+      };
+      if (part) {
+        if (part.inlineData) {
+          msg.inlineData = {
+            displayName: part.inlineData.displayName,
+            data: part.inlineData.data,
+            mimeType: part.inlineData.mimeType
+          };
+        } else if (part.text) {
+          msg.text = part.text;
+          msg.thought = part.thought;
+          if ((_b = (_a = event == null ? void 0 : event.groundingMetadata) == null ? void 0 : _a.searchEntryPoint) == null ? void 0 : _b.renderedContent) {
+            msg.renderedContent = event.groundingMetadata.searchEntryPoint.renderedContent;
+          }
+        } else if (part.fileData) {
+          msg.fileData = [part.fileData];
+        } else if (part.functionCall) {
+          msg.functionCall = part.functionCall;
+        } else if (part.functionResponse) {
+          msg.functionResponse = part.functionResponse;
+        } else if (part.executableCode) {
+          msg.executableCode = part.executableCode;
+        } else if (part.codeExecutionResult) {
+          msg.codeExecutionResult = part.codeExecutionResult;
         }
-      } else if (part.fileData) {
-        msg.fileData = [part.fileData];
-      } else if (part.functionCall) {
-        msg.functionCall = part.functionCall;
-      } else if (part.functionResponse) {
-        msg.functionResponse = part.functionResponse;
-      } else if (part.executableCode) {
-        msg.executableCode = part.executableCode;
-      } else if (part.codeExecutionResult) {
-        msg.codeExecutionResult = part.codeExecutionResult;
       }
-    }
-    if (part && Object.keys(part).length > 0) {
-      insertMessage(msg);
-    }
-  }, [insertMessage]);
-  const processPart = (0, import_react.useCallback)((event, part) => {
-    var _a, _b;
-    const renderedContent = (_b = (_a = event.groundingMetadata) == null ? void 0 : _a.searchEntryPoint) == null ? void 0 : _b.renderedContent;
-    if (part.text) {
-      const newChunk = part.text;
-      if (!textMsgRef.current || textMsgRef.current.thought !== part.thought) {
-        const newMessage = {
-          id: (0, import_uuid.v4)(),
-          author: event.author,
-          invocationId: event.invocationId,
-          eventId: event.id,
-          timestamp: event.timestamp,
-          role: "bot",
-          text: newChunk,
-          thought: part.thought
-        };
-        if (renderedContent) {
-          newMessage.renderedContent = renderedContent;
-        }
-        textMsgRef.current = newMessage;
-        storeEvents(part, event);
-        insertMessage(newMessage);
-      } else {
-        if (event.partial === false) {
+      if (part && Object.keys(part).length > 0) {
+        insertMessage(msg);
+      }
+    },
+    [insertMessage]
+  );
+  const processPart = (0, import_react.useCallback)(
+    (event, part) => {
+      var _a, _b;
+      const renderedContent = (_b = (_a = event.groundingMetadata) == null ? void 0 : _a.searchEntryPoint) == null ? void 0 : _b.renderedContent;
+      if (part.text) {
+        const newChunk = part.text;
+        if (!textMsgRef.current || textMsgRef.current.thought !== part.thought) {
+          const newMessage = {
+            id: (0, import_uuid.v4)(),
+            author: event.author,
+            invocationId: event.invocationId,
+            eventId: event.id,
+            timestamp: event.timestamp,
+            role: "bot",
+            text: newChunk,
+            thought: part.thought
+          };
+          if (renderedContent) {
+            newMessage.renderedContent = renderedContent;
+          }
+          textMsgRef.current = newMessage;
           storeEvents(part, event);
-          textMsgRef.current = null;
-          return;
-        }
-        if (renderedContent) {
+          insertMessage(newMessage);
+        } else {
+          if (event.partial === false) {
+            storeEvents(part, event);
+            textMsgRef.current = null;
+            return;
+          }
+          if (renderedContent) {
+            textMsgRef.current = {
+              ...textMsgRef.current,
+              renderedContent
+            };
+          }
           textMsgRef.current = {
             ...textMsgRef.current,
-            renderedContent
+            text: (textMsgRef.current.text || "") + newChunk
           };
+          updateMessage(textMsgRef.current);
         }
-        textMsgRef.current = {
-          ...textMsgRef.current,
-          text: (textMsgRef.current.text || "") + newChunk
-        };
-        updateMessage(textMsgRef.current);
+      } else {
+        textMsgRef.current = null;
+        storeEvents(part, event);
+        storeMessage(part, event, "bot");
       }
-    } else {
-      textMsgRef.current = null;
+    },
+    [storeEvents, storeMessage, insertMessage, updateMessage]
+  );
+  const processFollowupPart = (0, import_react.useCallback)(
+    (event) => {
+      var _a, _b;
+      const part = ((_b = (_a = event == null ? void 0 : event.content) == null ? void 0 : _a.parts) == null ? void 0 : _b[0]) || null;
+      const text = (part == null ? void 0 : part.text) || "";
       storeEvents(part, event);
-      storeMessage(
-        part,
-        event,
-        "bot"
-      );
-    }
-  }, [storeEvents, storeMessage, insertMessage, updateMessage]);
-  const processFollowupPart = (0, import_react.useCallback)((event) => {
-    var _a, _b;
-    const part = ((_b = (_a = event == null ? void 0 : event.content) == null ? void 0 : _a.parts) == null ? void 0 : _b[0]) || null;
-    const text = (part == null ? void 0 : part.text) || "";
-    storeEvents(part, event);
-    if (text) {
-      setSuggestedQuestions((prev) => [...prev, text]);
-    }
-  }, [storeEvents, insertMessage]);
-  const processErrorMessage = (0, import_react.useCallback)((chunkJson) => {
-    storeEvents(null, chunkJson);
-    insertMessage({
-      id: (0, import_uuid.v4)(),
-      text: chunkJson.errorMessage,
-      role: "bot"
-    });
-  }, [storeEvents, insertMessage]);
-  const processActionArtifact = (0, import_react.useCallback)((event) => {
-    if (event.actions && event.actions.artifactDelta) {
-      storeEvents(null, event);
-      storeMessage(null, event, "bot");
-    }
-  }, [storeEvents, storeMessage]);
-  const sendMessageRequest = (0, import_react.useCallback)((values) => {
-    return new Promise((resolve) => {
-      if (ctrl.current) {
-        ctrl.current.abort();
+      if (text) {
+        setSuggestedQuestions((prev) => [...prev, text]);
       }
-      setLoading(true);
-      ctrl.current = new AbortController();
-      const requestUrl = `${url}/api/gateway-web/openApi/v1/aizt/app/${appNo}/sendMsgStreaming`;
-      (0, import_fetch_event_source.fetchEventSource)(requestUrl, {
-        method: "POST",
-        signal: ctrl.current.signal,
-        body: JSON.stringify(values),
-        openWhenHidden: true,
-        headers: {
-          "Content-Type": "application/json",
-          "Cache-Control": "no-cache",
-          stream: "true",
-          Authorization: `Bearer ${token}`
-        },
-        async onopen(response) {
-          if (response.ok && response.headers.get("content-type") === import_fetch_event_source.EventStreamContentType) {
-          } else if (response.status >= 400 && response.status < 500 && response.status !== 429) {
-            resolve();
-          } else {
-            resolve();
-          }
-        },
-        onmessage: (event) => {
-          const chunkJson = (0, import_utils.safeJsonParse)(event.data, {});
-          if (chunkJson.error) {
-            import_antd.message.warning(chunkJson.error);
-            return;
-          }
-          onStream == null ? void 0 : onStream(chunkJson);
-          if (chunkJson.content) {
-            if (chunkJson.content.role === "followup") {
-              processFollowupPart(chunkJson);
+    },
+    [storeEvents, insertMessage]
+  );
+  const processErrorMessage = (0, import_react.useCallback)(
+    (chunkJson) => {
+      storeEvents(null, chunkJson);
+      insertMessage({
+        id: (0, import_uuid.v4)(),
+        text: chunkJson.errorMessage,
+        role: "bot"
+      });
+    },
+    [storeEvents, insertMessage]
+  );
+  const processActionArtifact = (0, import_react.useCallback)(
+    (event) => {
+      if (event.actions && event.actions.artifactDelta) {
+        storeEvents(null, event);
+        storeMessage(null, event, "bot");
+      }
+    },
+    [storeEvents, storeMessage]
+  );
+  const sendMessageRequest = (0, import_react.useCallback)(
+    (values) => {
+      return new Promise((resolve) => {
+        if (ctrl.current) {
+          ctrl.current.abort();
+        }
+        setLoading(true);
+        ctrl.current = new AbortController();
+        const requestUrl = `${url}/api/gateway-web/openApi/v1/aizt/app/${appNo}/sendMsgStreaming`;
+        (0, import_fetch_event_source.fetchEventSource)(requestUrl, {
+          method: "POST",
+          signal: ctrl.current.signal,
+          body: JSON.stringify(values),
+          openWhenHidden: true,
+          headers: {
+            "Content-Type": "application/json",
+            "Cache-Control": "no-cache",
+            stream: "true",
+            Authorization: `Bearer ${token}`
+          },
+          async onopen(response) {
+            if (response.ok && response.headers.get("content-type") === import_fetch_event_source.EventStreamContentType) {
+            } else if (response.status >= 400 && response.status < 500 && response.status !== 429) {
+              resolve();
+            } else {
+              resolve();
+            }
+          },
+          onmessage: (event) => {
+            const chunkJson = (0, import_utils.safeJsonParse)(event.data, {});
+            if (chunkJson.error) {
+              import_antd.message.warning(chunkJson.error);
               return;
             }
-            for (const part of combineTextParts(chunkJson.content.parts)) {
-              processPart(chunkJson, part);
+            onStream == null ? void 0 : onStream(chunkJson);
+            if (chunkJson.content) {
+              if (chunkJson.content.role === "followup") {
+                processFollowupPart(chunkJson);
+                return;
+              }
+              for (const part of combineTextParts(chunkJson.content.parts)) {
+                processPart(chunkJson, part);
+              }
+            } else if (chunkJson.errorMessage) {
+              processErrorMessage(chunkJson);
+            } else if (chunkJson.actions) {
+              processActionArtifact(chunkJson);
             }
-          } else if (chunkJson.errorMessage) {
-            processErrorMessage(chunkJson);
-          } else if (chunkJson.actions) {
-            processActionArtifact(chunkJson);
-          }
-        },
-        onclose: () => {
-          var _a;
-          setLoading(false);
-          if (textMsgRef.current) {
-            onMessage == null ? void 0 : onMessage(((_a = textMsgRef.current) == null ? void 0 : _a.text) || "", textMsgRef.current);
+          },
+          onclose: () => {
+            var _a;
+            setLoading(false);
+            if (textMsgRef.current) {
+              onMessage == null ? void 0 : onMessage(((_a = textMsgRef.current) == null ? void 0 : _a.text) || "", textMsgRef.current);
+            }
+            textMsgRef.current = null;
+            resolve();
+          },
+          onerror: (error) => {
+            setLoading(false);
+            resolve();
+            console.error("EventSource failed:", error);
+            throw error;
           }
-          textMsgRef.current = null;
-          resolve();
-        },
-        onerror: (error) => {
-          setLoading(false);
-          resolve();
-          console.error("EventSource failed:", error);
-          throw error;
-        }
+        });
       });
-    });
-  }, [type, currentSessionId, processActionArtifact, processErrorMessage, processPart]);
-  const startChat = async ({ text = "", files = [], functionResponse }) => {
+    },
+    [
+      type,
+      currentSessionId,
+      processActionArtifact,
+      processErrorMessage,
+      processPart
+    ]
+  );
+  const startChat = async ({
+    text = "",
+    files = [],
+    functionResponse
+  }) => {
+    console.log(text, loading);
     if (loading)
       return;
     if (!text.trim() && !functionResponse)
@@ -390,12 +418,16 @@ function useADKChat({
       return;
     setMessages((prev) => {
       const newMessages = [...prev];
-      const lastAIIndex = newMessages.findLastIndex((msg) => msg.role === "bot");
+      const lastAIIndex = newMessages.findLastIndex(
+        (msg) => msg.role === "bot"
+      );
       if (lastAIIndex !== -1) {
         newMessages.splice(lastAIIndex, 1);
       }
       setSuggestedQuestions([]);
-      const lastUserMessage = newMessages.findLast((msg) => msg.role === "user");
+      const lastUserMessage = newMessages.findLast(
+        (msg) => msg.role === "user"
+      );
       if (lastUserMessage) {
         const { text = "", fileData = [] } = lastUserMessage;
         sendMessageRequest({
@@ -403,9 +435,11 @@ function useADKChat({
           sessionId: currentSessionId,
           content: {
             role: "user",
-            parts: [{
-              text
-            }]
+            parts: [
+              {
+                text
+              }
+            ]
           },
           files: fileData == null ? void 0 : fileData.map((file) => ({
             fileName: file.displayName,
@@ -501,10 +535,12 @@ function useADKChat({
         if (part.errorMessage)
           msg.text = part.errorMessage;
         if (part.functionResponse) {
-          const functionCallMsg = mapped.find((msg2) => {
-            var _a, _b;
-            return ((_a = msg2.functionCall) == null ? void 0 : _a.id) === ((_b = part.functionResponse) == null ? void 0 : _b.id);
-          });
+          const functionCallMsg = mapped.find(
+            (msg2) => {
+              var _a, _b;
+              return ((_a = msg2.functionCall) == null ? void 0 : _a.id) === ((_b = part.functionResponse) == null ? void 0 : _b.id);
+            }
+          );
           if (functionCallMsg) {
             functionCallMsg.functionResponse = part.functionResponse;
           }
@@ -533,7 +569,9 @@ function useADKChat({
         if (Array.isArray(data) && data.length > 0) {
           formatMessages(data, true);
         } else {
-          setSuggestedQuestions(((_b = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b.suggested_questions) || []);
+          setSuggestedQuestions(
+            ((_b = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b.suggested_questions) || []
+          );
         }
       }
     } else {
@@ -544,7 +582,11 @@ function useADKChat({
     var _a, _b;
     try {
       setLoading(true);
-      const { data = [], pagination, result } = await (0, import_api.fetchSessionList)({
+      const {
+        data = [],
+        pagination,
+        result
+      } = await (0, import_api.fetchSessionList)({
         url,
         appNo,
         token,
@@ -564,13 +606,19 @@ function useADKChat({
             };
           });
           setSessionList(formatSessions);
-          setSessionPagination({ pageNum: sessionPagination.pageNum, pageSize: sessionPagination.pageSize, total: (pagination == null ? void 0 : pagination.total) || 0 });
+          setSessionPagination({
+            pageNum: sessionPagination.pageNum,
+            pageSize: sessionPagination.pageSize,
+            total: (pagination == null ? void 0 : pagination.total) || 0
+          });
           const { sessionId } = data == null ? void 0 : data[0];
           if (showFirstSession) {
             fetchDetail && setCurrentSessionDetail(sessionId);
           } else {
             setPrologue(((_a = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _a.prologue) || "");
-            setSuggestedQuestions(((_b = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b.suggested_questions) || []);
+            setSuggestedQuestions(
+              ((_b = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _b.suggested_questions) || []
+            );
           }
           if (!initialized) {
             onSuccess == null ? void 0 : onSuccess({