@ai-group/chat-sdk 3.6.6 → 3.6.8

package/dist/cjs/components/XAdkChatbot/components/FunctionCallRender/index.js.map

@@ -1,7 +1,7 @@
 {
   "version": 3,
   "sources": ["../../../../../../src/components/XAdkChatbot/components/FunctionCallRender/index.tsx"],
-  "sourcesContent": ["import React from \"react\";\nimport { Alert, Button, Flex, Form, Popover, Tooltip, Typography } from \"antd\";\nimport {\n  CopyOutlined,\n  ToolOutlined,\n  CheckCircleTwoTone,\n  LoadingOutlined,\n  UserSwitchOutlined,\n  InfoCircleOutlined,\n} from \"@ant-design/icons\";\nimport type { FunctionCallRenderProps, FunctionCall } from \"@/types\";\nimport { useStyles } from \"./styles\";\nimport { defaultToolKindResolver } from \"@/types/FunctionCallRender\";\nimport JsonView from \"@/components/XAiJsonView\";\n\nconst FunctionCallRender: React.FC<FunctionCallRenderProps> = ({\n  msg,\n  showDetail,\n  onConfirm,\n  kind: kindProp,\n  toolKindResolver = defaultToolKindResolver,\n  renderApproval,\n  renderHandoff,\n}) => {\n  const styles = useStyles();\n  const fnCall = msg.functionCall as FunctionCall;\n  const fnRes = msg.functionResponse;\n  const name = fnCall?.name || fnRes?.name;\n  const kind = kindProp ?? toolKindResolver(name);\n\n  if (kind === \"approval\") {\n    if (renderApproval) {\n      const custom = renderApproval(msg, onConfirm);\n      if (custom !== null && custom !== undefined) return <>{custom}</>;\n    }\n    if (fnRes) {\n      if (msg.role === \"bot\") return null;\n      const confirmed = fnRes.response?.confirmed;\n      return (\n        <div key={msg.id} className={styles.wrapper}>\n          {confirmed ? (\n            <Alert\n              message=\"已确认\"\n              type=\"success\"\n              showIcon\n              className={styles.alert}\n            />\n          ) : (\n            <Alert\n              message=\"已取消\"\n              type=\"error\"\n              showIcon\n              className={styles.alert}\n            />\n          )}\n        </div>\n      );\n    }\n    const originalFn = msg?.functionCall?.args?.originalFunctionCall;\n    return (\n      <div\n        key={msg.id}\n        className={styles.wrapper}\n        style={{ justifyContent: \"flex-end\" }}\n      >\n        <div className={styles.confirm}>\n          <div className={styles.confirmText}>\n            请确认是否执行工具：{originalFn?.name}\n          </div>\n          <Flex justify=\"end\" gap={8}>\n            <Button\n              type=\"default\"\n              size=\"small\"\n              onClick={() => onConfirm?.(fnCall, false)}\n            >\n              取消\n            </Button>\n            <Button\n              type=\"primary\"\n              size=\"small\"\n              onClick={() => onConfirm?.(fnCall, true)}\n            >\n              确认\n            </Button>\n          </Flex>\n        </div>\n      </div>\n    );\n  }\n  if (kind === \"handoff\") {\n    if (renderHandoff) {\n      const custom = renderHandoff(msg);\n      if (custom !== null && custom !== undefined) return <>{custom}</>;\n    }\n    const agentName = fnCall?.args?.agentName;\n    return (\n      <div key={msg.id} className={styles.wrapper}>\n        <div className={styles.fnCall}>\n          <UserSwitchOutlined style={{ color: \"#888\" }} />\n          <Typography.Text\n            ellipsis={{ tooltip: agentName }}\n            style={{ maxWidth: 300, lineHeight: \"16px\" }}\n          >\n            任务转交：{agentName}\n          </Typography.Text>\n          {msg.functionResponse ? (\n            <CheckCircleTwoTone twoToneColor=\"#52c41a\" />\n          ) : (\n            <LoadingOutlined />\n          )}\n        </div>\n      </div>\n    );\n  }\n  const renderItem = (obj: any, text: string) => {\n    const str = JSON.stringify(obj, null, 2);\n    return (\n      <Form.Item\n        label={\n          <Flex align=\"center\">\n            <span className=\"mr-2\">{text}</span>\n            <Typography.Text\n              copyable={{\n                text: str,\n                icon: <CopyOutlined />,\n              }}\n            />\n          </Flex>\n        }\n        style={{ marginBottom: 12 }}\n      >\n        <JsonView value={str} />\n      </Form.Item>\n    );\n  };\n  const renderResponse = () => {\n    const res = msg.functionResponse;\n    if (res) {\n      if (res.response?.error) {\n        return (\n          <Tooltip title={res.response.error}>\n            <InfoCircleOutlined style={{ color: \"#faad14\" }} />\n          </Tooltip>\n        );\n      }\n      return <CheckCircleTwoTone twoToneColor=\"#52c41a\" />;\n    }\n    return <LoadingOutlined />;\n  };\n  return (\n    <div key={msg.id} className={styles.wrapper}>\n      {showDetail ? (\n        <Popover\n          placement=\"left\"\n          content={\n            <Form style={{ width: 400 }} layout=\"vertical\">\n              {renderItem(fnCall?.args, \"工具参数\")}\n              {renderItem(msg.functionResponse?.response, \"工具响应\")}\n            </Form>\n          }\n        >\n          <div className={styles.fnCall}>\n            <ToolOutlined style={{ color: \"#888\" }} />\n            <Typography.Text\n              ellipsis={{ tooltip: name }}\n              style={{ maxWidth: 300, lineHeight: \"16px\" }}\n            >\n              调用工具：{name}\n            </Typography.Text>\n            {renderResponse()}\n          </div>\n        </Popover>\n      ) : (\n        <div className={styles.fnCall}>\n          <ToolOutlined style={{ color: \"#888\" }} />\n          <Typography.Text\n            ellipsis={{ tooltip: name }}\n            style={{ maxWidth: 300, lineHeight: \"16px\" }}\n          >\n            调用工具：{name}\n          </Typography.Text>\n          {renderResponse()}\n        </div>\n      )}\n    </div>\n  );\n};\n\nexport default FunctionCallRender;\n"],
-  "mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AACA,kBAAwE;AACxE,mBAOO;AAEP,oBAA0B;AAC1B,gCAAwC;AACxC,yBAAqB;AAoBqC;AAlB1D,IAAM,qBAAwD,CAAC;AAAA,EAC7D;AAAA,EACA;AAAA,EACA;AAAA,EACA,MAAM;AAAA,EACN,mBAAmB;AAAA,EACnB;AAAA,EACA;AACF,MAAM;AAvBN;AAwBE,QAAM,aAAS,yBAAU;AACzB,QAAM,SAAS,IAAI;AACnB,QAAM,QAAQ,IAAI;AAClB,QAAM,QAAO,iCAAQ,UAAQ,+BAAO;AACpC,QAAM,OAAO,YAAY,iBAAiB,IAAI;AAE9C,MAAI,SAAS,YAAY;AACvB,QAAI,gBAAgB;AAClB,YAAM,SAAS,eAAe,KAAK,SAAS;AAC5C,UAAI,WAAW,QAAQ,WAAW;AAAW,eAAO,2EAAG,kBAAO;AAAA,IAChE;AACA,QAAI,OAAO;AACT,UAAI,IAAI,SAAS;AAAO,eAAO;AAC/B,YAAM,aAAY,WAAM,aAAN,mBAAgB;AAClC,aACE,4CAAC,SAAiB,WAAW,OAAO,SACjC,sBACC;AAAA,QAAC;AAAA;AAAA,UACC,SAAQ;AAAA,UACR,MAAK;AAAA,UACL,UAAQ;AAAA,UACR,WAAW,OAAO;AAAA;AAAA,MACpB,IAEA;AAAA,QAAC;AAAA;AAAA,UACC,SAAQ;AAAA,UACR,MAAK;AAAA,UACL,UAAQ;AAAA,UACR,WAAW,OAAO;AAAA;AAAA,MACpB,KAdM,IAAI,EAgBd;AAAA,IAEJ;AACA,UAAM,cAAa,sCAAK,iBAAL,mBAAmB,SAAnB,mBAAyB;AAC5C,WACE;AAAA,MAAC;AAAA;AAAA,QAEC,WAAW,OAAO;AAAA,QAClB,OAAO,EAAE,gBAAgB,WAAW;AAAA,QAEpC,uDAAC,SAAI,WAAW,OAAO,SACrB;AAAA,uDAAC,SAAI,WAAW,OAAO,aAAa;AAAA;AAAA,YACvB,yCAAY;AAAA,aACzB;AAAA,UACA,6CAAC,oBAAK,SAAQ,OAAM,KAAK,GACvB;AAAA;AAAA,cAAC;AAAA;AAAA,gBACC,MAAK;AAAA,gBACL,MAAK;AAAA,gBACL,SAAS,MAAM,uCAAY,QAAQ;AAAA,gBACpC;AAAA;AAAA,YAED;AAAA,YACA;AAAA,cAAC;AAAA;AAAA,gBACC,MAAK;AAAA,gBACL,MAAK;AAAA,gBACL,SAAS,MAAM,uCAAY,QAAQ;AAAA,gBACpC;AAAA;AAAA,YAED;AAAA,aACF;AAAA,WACF;AAAA;AAAA,MAxBK,IAAI;AAAA,IAyBX;AAAA,EAEJ;AACA,MAAI,SAAS,WAAW;AACtB,QAAI,eAAe;AACjB,YAAM,SAAS,cAAc,GAAG;AAChC,UAAI,WAAW,QAAQ,WAAW;AAAW,eAAO,2EAAG,kBAAO;AAAA,IAChE;AACA,UAAM,aAAY,sCAAQ,SAAR,mBAAc;AAChC,WACE,4CAAC,SAAiB,WAAW,OAAO,SAClC,uDAAC,SAAI,WAAW,OAAO,QACrB;AAAA,kDAAC,mCAAmB,OAAO,EAAE,OAAO,OAAO,GAAG;AAAA,MAC9C;AAAA,QAAC,uBAAW;AAAA,QAAX;AAAA,UACC,UAAU,EAAE,SAAS,UAAU;AAAA,UAC/B,OAAO,EAAE,UAAU,KAAK,YAAY,OAAO;AAAA,UAC5C;AAAA;AAAA,YACO;AAAA;AAAA;AAAA,MACR;AAAA,MACC,IAAI,mBACH,4CAAC,mCAAmB,cAAa,WAAU,IAE3C,4CAAC,gCAAgB;AAAA,OAErB,KAdQ,IAAI,EAed;AAAA,EAEJ;AACA,QAAM,aAAa,CAAC,KAAU,SAAiB;AAC7C,UAAM,MAAM,KAAK,UAAU,KAAK,MAAM,CAAC;AACvC,WACE;AAAA,MAAC,iBAAK;AAAA,MAAL;AAAA,QACC,OACE,6CAAC,oBAAK,OAAM,UACV;AAAA,sDAAC,UAAK,WAAU,QAAQ,gBAAK;AAAA,UAC7B;AAAA,YAAC,uBAAW;AAAA,YAAX;AAAA,cACC,UAAU;AAAA,gBACR,MAAM;AAAA,gBACN,MAAM,4CAAC,6BAAa;AAAA,cACtB;AAAA;AAAA,UACF;AAAA,WACF;AAAA,QAEF,OAAO,EAAE,cAAc,GAAG;AAAA,QAE1B,sDAAC,mBAAAA,SAAA,EAAS,OAAO,KAAK;AAAA;AAAA,IACxB;AAAA,EAEJ;AACA,QAAM,iBAAiB,MAAM;AAvI/B,QAAAC;AAwII,UAAM,MAAM,IAAI;AAChB,QAAI,KAAK;AACP,WAAIA,MAAA,IAAI,aAAJ,gBAAAA,IAAc,OAAO;AACvB,eACE,4CAAC,uBAAQ,OAAO,IAAI,SAAS,OAC3B,sDAAC,mCAAmB,OAAO,EAAE,OAAO,UAAU,GAAG,GACnD;AAAA,MAEJ;AACA,aAAO,4CAAC,mCAAmB,cAAa,WAAU;AAAA,IACpD;AACA,WAAO,4CAAC,gCAAgB;AAAA,EAC1B;AACA,SACE,4CAAC,SAAiB,WAAW,OAAO,SACjC,uBACC;AAAA,IAAC;AAAA;AAAA,MACC,WAAU;AAAA,MACV,SACE,6CAAC,oBAAK,OAAO,EAAE,OAAO,IAAI,GAAG,QAAO,YACjC;AAAA,mBAAW,iCAAQ,MAAM,MAAM;AAAA,QAC/B,YAAW,SAAI,qBAAJ,mBAAsB,UAAU,MAAM;AAAA,SACpD;AAAA,MAGF,uDAAC,SAAI,WAAW,OAAO,QACrB;AAAA,oDAAC,6BAAa,OAAO,EAAE,OAAO,OAAO,GAAG;AAAA,QACxC;AAAA,UAAC,uBAAW;AAAA,UAAX;AAAA,YACC,UAAU,EAAE,SAAS,KAAK;AAAA,YAC1B,OAAO,EAAE,UAAU,KAAK,YAAY,OAAO;AAAA,YAC5C;AAAA;AAAA,cACO;AAAA;AAAA;AAAA,QACR;AAAA,QACC,eAAe;AAAA,SAClB;AAAA;AAAA,EACF,IAEA,6CAAC,SAAI,WAAW,OAAO,QACrB;AAAA,gDAAC,6BAAa,OAAO,EAAE,OAAO,OAAO,GAAG;AAAA,IACxC;AAAA,MAAC,uBAAW;AAAA,MAAX;AAAA,QACC,UAAU,EAAE,SAAS,KAAK;AAAA,QAC1B,OAAO,EAAE,UAAU,KAAK,YAAY,OAAO;AAAA,QAC5C;AAAA;AAAA,UACO;AAAA;AAAA;AAAA,IACR;AAAA,IACC,eAAe;AAAA,KAClB,KAhCM,IAAI,EAkCd;AAEJ;AAEA,IAAO,6BAAQ;",
+  "sourcesContent": ["import React from \"react\";\nimport { Alert, Button, Flex, Form, Popover, Tooltip, Typography } from \"antd\";\nimport {\n  CopyOutlined,\n  ToolOutlined,\n  CheckCircleTwoTone,\n  LoadingOutlined,\n  UserSwitchOutlined,\n  InfoCircleOutlined,\n} from \"@ant-design/icons\";\nimport type { FunctionCallRenderProps, FunctionCall } from \"@/types\";\nimport { useStyles } from \"./styles\";\nimport { defaultToolKindResolver } from \"@/types/FunctionCallRender\";\nimport JsonView from \"@/components/XAiJsonView\";\n\n// 工具调用组件\nconst FunctionCallRender: React.FC<FunctionCallRenderProps> = ({\n  msg,\n  showDetail,\n  onConfirm,\n  kind: kindProp,\n  toolKindResolver = defaultToolKindResolver,\n  renderApproval,\n  renderHandoff,\n}) => {\n  const styles = useStyles();\n  const fnCall = msg.functionCall as FunctionCall;\n  const fnRes = msg.functionResponse;\n  const name = fnCall?.name || fnRes?.name;\n  const kind = kindProp ?? toolKindResolver(name);\n\n  if (kind === \"approval\") {\n    if (renderApproval) {\n      const custom = renderApproval(msg, onConfirm);\n      if (custom !== null && custom !== undefined) return <>{custom}</>;\n    }\n    if (fnRes) {\n      if (msg.role === \"bot\") return null;\n      const confirmed = fnRes.response?.confirmed;\n      return (\n        <div key={msg.id} className={styles.wrapper}>\n          {confirmed ? (\n            <Alert\n              message=\"已确认\"\n              type=\"success\"\n              showIcon\n              className={styles.alert}\n            />\n          ) : (\n            <Alert\n              message=\"已取消\"\n              type=\"error\"\n              showIcon\n              className={styles.alert}\n            />\n          )}\n        </div>\n      );\n    }\n    const originalFn = msg?.functionCall?.args?.originalFunctionCall;\n    return (\n      <div\n        key={msg.id}\n        className={styles.wrapper}\n        style={{ justifyContent: \"flex-end\" }}\n      >\n        <div className={styles.confirm}>\n          <div className={styles.confirmText}>\n            请确认是否执行工具：{originalFn?.name}\n          </div>\n          <Flex justify=\"end\" gap={8}>\n            <Button\n              type=\"default\"\n              size=\"small\"\n              onClick={() => onConfirm?.(fnCall, false)}\n            >\n              取消\n            </Button>\n            <Button\n              type=\"primary\"\n              size=\"small\"\n              onClick={() => onConfirm?.(fnCall, true)}\n            >\n              确认\n            </Button>\n          </Flex>\n        </div>\n      </div>\n    );\n  }\n  if (kind === \"handoff\") {\n    if (renderHandoff) {\n      const custom = renderHandoff(msg);\n      if (custom !== null && custom !== undefined) return <>{custom}</>;\n    }\n    const agentName = fnCall?.args?.agentName;\n    return (\n      <div key={msg.id} className={styles.wrapper}>\n        <div className={styles.fnCall}>\n          <UserSwitchOutlined style={{ color: \"#888\" }} />\n          <Typography.Text\n            ellipsis={{ tooltip: agentName }}\n            style={{ maxWidth: 300, lineHeight: \"16px\" }}\n          >\n            任务转交：{agentName}\n          </Typography.Text>\n          {msg.functionResponse ? (\n            <CheckCircleTwoTone twoToneColor=\"#52c41a\" />\n          ) : (\n            <LoadingOutlined />\n          )}\n        </div>\n      </div>\n    );\n  }\n  const renderItem = (obj: any, text: string) => {\n    const str = JSON.stringify(obj, null, 2);\n    return (\n      <Form.Item\n        label={\n          <Flex align=\"center\">\n            <span className=\"mr-2\">{text}</span>\n            <Typography.Text\n              copyable={{\n                text: str,\n                icon: <CopyOutlined />,\n              }}\n            />\n          </Flex>\n        }\n        style={{ marginBottom: 12 }}\n      >\n        <JsonView value={str} />\n      </Form.Item>\n    );\n  };\n  const renderResponse = () => {\n    const res = msg.functionResponse;\n    if (res) {\n      if (res.response?.error) {\n        return (\n          <Tooltip title={res.response.error}>\n            <InfoCircleOutlined style={{ color: \"#faad14\" }} />\n          </Tooltip>\n        );\n      }\n      return <CheckCircleTwoTone twoToneColor=\"#52c41a\" />;\n    }\n    return <LoadingOutlined />;\n  };\n  return (\n    <div key={msg.id} className={styles.wrapper}>\n      {showDetail ? (\n        <Popover\n          placement=\"left\"\n          content={\n            <Form style={{ width: 400 }} layout=\"vertical\">\n              {renderItem(fnCall?.args, \"工具参数\")}\n              {renderItem(msg.functionResponse?.response, \"工具响应\")}\n            </Form>\n          }\n        >\n          <div className={styles.fnCall}>\n            <ToolOutlined style={{ color: \"#888\" }} />\n            <Typography.Text\n              ellipsis={{ tooltip: name }}\n              style={{ maxWidth: 300, lineHeight: \"16px\" }}\n            >\n              调用工具：{name}\n            </Typography.Text>\n            {renderResponse()}\n          </div>\n        </Popover>\n      ) : (\n        <div className={styles.fnCall}>\n          <ToolOutlined style={{ color: \"#888\" }} />\n          <Typography.Text\n            ellipsis={{ tooltip: name }}\n            style={{ maxWidth: 300, lineHeight: \"16px\" }}\n          >\n            调用工具：{name}\n          </Typography.Text>\n          {renderResponse()}\n        </div>\n      )}\n    </div>\n  );\n};\n\nexport default FunctionCallRender;\n"],
+  "mappings": ";;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AACA,kBAAwE;AACxE,mBAOO;AAEP,oBAA0B;AAC1B,gCAAwC;AACxC,yBAAqB;AAqBqC;AAlB1D,IAAM,qBAAwD,CAAC;AAAA,EAC7D;AAAA,EACA;AAAA,EACA;AAAA,EACA,MAAM;AAAA,EACN,mBAAmB;AAAA,EACnB;AAAA,EACA;AACF,MAAM;AAxBN;AAyBE,QAAM,aAAS,yBAAU;AACzB,QAAM,SAAS,IAAI;AACnB,QAAM,QAAQ,IAAI;AAClB,QAAM,QAAO,iCAAQ,UAAQ,+BAAO;AACpC,QAAM,OAAO,YAAY,iBAAiB,IAAI;AAE9C,MAAI,SAAS,YAAY;AACvB,QAAI,gBAAgB;AAClB,YAAM,SAAS,eAAe,KAAK,SAAS;AAC5C,UAAI,WAAW,QAAQ,WAAW;AAAW,eAAO,2EAAG,kBAAO;AAAA,IAChE;AACA,QAAI,OAAO;AACT,UAAI,IAAI,SAAS;AAAO,eAAO;AAC/B,YAAM,aAAY,WAAM,aAAN,mBAAgB;AAClC,aACE,4CAAC,SAAiB,WAAW,OAAO,SACjC,sBACC;AAAA,QAAC;AAAA;AAAA,UACC,SAAQ;AAAA,UACR,MAAK;AAAA,UACL,UAAQ;AAAA,UACR,WAAW,OAAO;AAAA;AAAA,MACpB,IAEA;AAAA,QAAC;AAAA;AAAA,UACC,SAAQ;AAAA,UACR,MAAK;AAAA,UACL,UAAQ;AAAA,UACR,WAAW,OAAO;AAAA;AAAA,MACpB,KAdM,IAAI,EAgBd;AAAA,IAEJ;AACA,UAAM,cAAa,sCAAK,iBAAL,mBAAmB,SAAnB,mBAAyB;AAC5C,WACE;AAAA,MAAC;AAAA;AAAA,QAEC,WAAW,OAAO;AAAA,QAClB,OAAO,EAAE,gBAAgB,WAAW;AAAA,QAEpC,uDAAC,SAAI,WAAW,OAAO,SACrB;AAAA,uDAAC,SAAI,WAAW,OAAO,aAAa;AAAA;AAAA,YACvB,yCAAY;AAAA,aACzB;AAAA,UACA,6CAAC,oBAAK,SAAQ,OAAM,KAAK,GACvB;AAAA;AAAA,cAAC;AAAA;AAAA,gBACC,MAAK;AAAA,gBACL,MAAK;AAAA,gBACL,SAAS,MAAM,uCAAY,QAAQ;AAAA,gBACpC;AAAA;AAAA,YAED;AAAA,YACA;AAAA,cAAC;AAAA;AAAA,gBACC,MAAK;AAAA,gBACL,MAAK;AAAA,gBACL,SAAS,MAAM,uCAAY,QAAQ;AAAA,gBACpC;AAAA;AAAA,YAED;AAAA,aACF;AAAA,WACF;AAAA;AAAA,MAxBK,IAAI;AAAA,IAyBX;AAAA,EAEJ;AACA,MAAI,SAAS,WAAW;AACtB,QAAI,eAAe;AACjB,YAAM,SAAS,cAAc,GAAG;AAChC,UAAI,WAAW,QAAQ,WAAW;AAAW,eAAO,2EAAG,kBAAO;AAAA,IAChE;AACA,UAAM,aAAY,sCAAQ,SAAR,mBAAc;AAChC,WACE,4CAAC,SAAiB,WAAW,OAAO,SAClC,uDAAC,SAAI,WAAW,OAAO,QACrB;AAAA,kDAAC,mCAAmB,OAAO,EAAE,OAAO,OAAO,GAAG;AAAA,MAC9C;AAAA,QAAC,uBAAW;AAAA,QAAX;AAAA,UACC,UAAU,EAAE,SAAS,UAAU;AAAA,UAC/B,OAAO,EAAE,UAAU,KAAK,YAAY,OAAO;AAAA,UAC5C;AAAA;AAAA,YACO;AAAA;AAAA;AAAA,MACR;AAAA,MACC,IAAI,mBACH,4CAAC,mCAAmB,cAAa,WAAU,IAE3C,4CAAC,gCAAgB;AAAA,OAErB,KAdQ,IAAI,EAed;AAAA,EAEJ;AACA,QAAM,aAAa,CAAC,KAAU,SAAiB;AAC7C,UAAM,MAAM,KAAK,UAAU,KAAK,MAAM,CAAC;AACvC,WACE;AAAA,MAAC,iBAAK;AAAA,MAAL;AAAA,QACC,OACE,6CAAC,oBAAK,OAAM,UACV;AAAA,sDAAC,UAAK,WAAU,QAAQ,gBAAK;AAAA,UAC7B;AAAA,YAAC,uBAAW;AAAA,YAAX;AAAA,cACC,UAAU;AAAA,gBACR,MAAM;AAAA,gBACN,MAAM,4CAAC,6BAAa;AAAA,cACtB;AAAA;AAAA,UACF;AAAA,WACF;AAAA,QAEF,OAAO,EAAE,cAAc,GAAG;AAAA,QAE1B,sDAAC,mBAAAA,SAAA,EAAS,OAAO,KAAK;AAAA;AAAA,IACxB;AAAA,EAEJ;AACA,QAAM,iBAAiB,MAAM;AAxI/B,QAAAC;AAyII,UAAM,MAAM,IAAI;AAChB,QAAI,KAAK;AACP,WAAIA,MAAA,IAAI,aAAJ,gBAAAA,IAAc,OAAO;AACvB,eACE,4CAAC,uBAAQ,OAAO,IAAI,SAAS,OAC3B,sDAAC,mCAAmB,OAAO,EAAE,OAAO,UAAU,GAAG,GACnD;AAAA,MAEJ;AACA,aAAO,4CAAC,mCAAmB,cAAa,WAAU;AAAA,IACpD;AACA,WAAO,4CAAC,gCAAgB;AAAA,EAC1B;AACA,SACE,4CAAC,SAAiB,WAAW,OAAO,SACjC,uBACC;AAAA,IAAC;AAAA;AAAA,MACC,WAAU;AAAA,MACV,SACE,6CAAC,oBAAK,OAAO,EAAE,OAAO,IAAI,GAAG,QAAO,YACjC;AAAA,mBAAW,iCAAQ,MAAM,MAAM;AAAA,QAC/B,YAAW,SAAI,qBAAJ,mBAAsB,UAAU,MAAM;AAAA,SACpD;AAAA,MAGF,uDAAC,SAAI,WAAW,OAAO,QACrB;AAAA,oDAAC,6BAAa,OAAO,EAAE,OAAO,OAAO,GAAG;AAAA,QACxC;AAAA,UAAC,uBAAW;AAAA,UAAX;AAAA,YACC,UAAU,EAAE,SAAS,KAAK;AAAA,YAC1B,OAAO,EAAE,UAAU,KAAK,YAAY,OAAO;AAAA,YAC5C;AAAA;AAAA,cACO;AAAA;AAAA;AAAA,QACR;AAAA,QACC,eAAe;AAAA,SAClB;AAAA;AAAA,EACF,IAEA,6CAAC,SAAI,WAAW,OAAO,QACrB;AAAA,gDAAC,6BAAa,OAAO,EAAE,OAAO,OAAO,GAAG;AAAA,IACxC;AAAA,MAAC,uBAAW;AAAA,MAAX;AAAA,QACC,UAAU,EAAE,SAAS,KAAK;AAAA,QAC1B,OAAO,EAAE,UAAU,KAAK,YAAY,OAAO;AAAA,QAC5C;AAAA;AAAA,UACO;AAAA;AAAA;AAAA,IACR;AAAA,IACC,eAAe;AAAA,KAClB,KAhCM,IAAI,EAkCd;AAEJ;AAEA,IAAO,6BAAQ;",
   "names": ["JsonView", "_a"]
 }

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

@@ -1,7 +1,7 @@
 /// <reference types="react" />
 import { DebugOptions, FunctionCall, SendContent, SessionData, Event, IMessage, Part } from "../types";
 import { type ChatConfig } from "../services/api";
-declare function useADKChat({ url, token, config, type, enabled, // ← 新增: 是否启用 Hook
+declare function useADKChat({ url, token, config, type: _type, enabled, // ← 新增: 是否启用 Hook
 strategies, onError, onMessage, onSuccess, onStream, }: DebugOptions): {
     appInfo: ChatConfig | null;
     startChat: ({ text, files, functionResponse, }: SendContent) => Promise<void>;
@@ -24,7 +24,7 @@ strategies, onError, onMessage, onSuccess, onStream, }: DebugOptions): {
     sessionLoading: boolean;
     messages: IMessage[];
     insertSuggestedQuestions: (suggested_questions?: string[]) => void;
-    insertPrologue: (prologue: string) => void;
+    insertPrologue: (prologueText: string) => void;
     deleteSession: (sessionId: string) => Promise<void>;
     updateSession: (sessionId: string, title: string) => Promise<void>;
     setSuggestedQuestions: import("react").Dispatch<import("react").SetStateAction<string[]>>;

package/dist/cjs/hooks/useADKChat.js

@@ -50,14 +50,46 @@ var combineTextParts = (parts) => {
   return result;
 };
 var DEFAULT_SESSION_EVENTS_POLLING_INTERVAL = 1e4;
+var buildNonTextMessage = (event, part, role = "bot") => {
+  const msg = {
+    id: (0, import_uuid.v4)(),
+    author: event.author,
+    invocationId: event.invocationId,
+    eventId: event.id,
+    timestamp: event.timestamp,
+    isLike: event.isLike || 0,
+    role,
+    modelCode: event.modelCode,
+    usageMetadata: event.usageMetadata,
+    finishReason: event.finishReason,
+    raw: event
+  };
+  if (part.inlineData) {
+    msg.inlineData = {
+      displayName: part.inlineData.displayName,
+      data: part.inlineData.data,
+      mimeType: part.inlineData.mimeType
+    };
+  } 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.text) {
+    msg.text = part.text;
+    msg.thought = part.thought;
+  }
+  return msg;
+};
 var getMessageEventId = (messageItem) => {
   var _a;
   return messageItem.eventId || ((_a = messageItem.raw) == null ? void 0 : _a.id);
 };
-var getMessageInvocationId = (messageItem) => {
-  var _a;
-  return messageItem.invocationId || ((_a = messageItem.raw) == null ? void 0 : _a.invocationId);
-};
 var deduplicateMessages = (messageItems) => {
   const seenMessageIds = /* @__PURE__ */ new Set();
   let hasDuplicate = false;
@@ -71,14 +103,14 @@ var deduplicateMessages = (messageItems) => {
   });
   const botTextGroups = /* @__PURE__ */ new Map();
   uniqueIdMessages.forEach((messageItem) => {
-    const invocationId = getMessageInvocationId(messageItem);
-    if (messageItem.role !== "bot" || !invocationId || typeof messageItem.text !== "string")
+    const eventId = getMessageEventId(messageItem);
+    if (messageItem.role !== "bot" || !eventId || typeof messageItem.text !== "string")
       return;
     const normalizedText = messageItem.text.replace(/\s+/g, " ").trim();
     if (!normalizedText)
       return;
     const fileKey = (messageItem.fileData || []).map((file) => file.fileUri).join("|");
-    const groupKey = `${invocationId}\0${Boolean(
+    const groupKey = `${eventId}\0${Boolean(
       messageItem.thought
     )}\0${fileKey}`;
     const group = botTextGroups.get(groupKey) || [];
@@ -110,6 +142,8 @@ var mapSessionMessages = (sessionEvents) => {
       return;
     if (item.author === "user" && item.content.parts.some((part) => part.task))
       return;
+    if (item.content.role === "user" && item.content.parts.every((part) => !part || part.functionResponse))
+      return;
     const role = (item.content.role || "").toLowerCase() === "user" ? "user" : "bot";
     const parts = item.content.parts.filter((part) => {
       if (!part)
@@ -127,9 +161,10 @@ var mapSessionMessages = (sessionEvents) => {
     const otherParts = parts.filter(
       (part) => !part.text && !part.errorMessage && !part.inlineData && !part.fileData
     );
+    let partCounter = 0;
     if (textParts.length > 0 || fileDataParts.length > 0) {
       const messageItem = {
-        id: (0, import_uuid.v4)(),
+        id: `session-event:${item.id}:${partCounter++}`,
         author: item.author,
         invocationId: item.invocationId,
         eventId: item.id,
@@ -171,7 +206,7 @@ var mapSessionMessages = (sessionEvents) => {
     }
     otherParts.forEach((part) => {
       const messageItem = {
-        id: (0, import_uuid.v4)(),
+        id: `session-event:${item.id}:${partCounter++}`,
         author: item.author,
         invocationId: item.invocationId,
         eventId: item.id,
@@ -206,64 +241,54 @@ var mapSessionMessages = (sessionEvents) => {
   });
   return mapped;
 };
-var getComparableMessageContent = (messageItem) => {
-  var _a;
-  return JSON.stringify({
-    role: messageItem.role,
-    text: ((_a = messageItem.text) == null ? void 0 : _a.replace(/\r\n/g, "\n").trim()) || "",
-    thought: Boolean(messageItem.thought),
-    inlineData: messageItem.inlineData || null,
-    fileData: messageItem.fileData || null,
-    functionCall: messageItem.functionCall || null,
-    functionResponse: messageItem.functionResponse || null,
-    executableCode: messageItem.executableCode || null,
-    codeExecutionResult: messageItem.codeExecutionResult || null
-  });
-};
-var isStreamMessageWithoutServerEventId = (messageItem) => {
-  const eventId = getMessageEventId(messageItem);
-  return Boolean(messageItem.invocationId) && (!eventId || eventId === messageItem.invocationId);
-};
-var insertMissingSessionEvents = (currentMessages, sessionEvents) => {
+var insertMissingSessionEvents = (currentMessages, sessionEvents, removedInvocationIds) => {
   const renderedEventIds = new Set(
     currentMessages.map(getMessageEventId).filter(Boolean)
   );
-  const unmatchedStreamMessageIndexes = new Set(
-    currentMessages.map(
-      (messageItem, index) => isStreamMessageWithoutServerEventId(messageItem) ? index : -1
-    ).filter((index) => index >= 0)
-  );
+  let nextMessages = [...currentMessages];
+  const fnResMap = /* @__PURE__ */ new Map();
+  sessionEvents.forEach((event) => {
+    var _a, _b;
+    if (((_a = event.content) == null ? void 0 : _a.role) !== "user" || !Array.isArray((_b = event.content) == null ? void 0 : _b.parts))
+      return;
+    event.content.parts.forEach((p) => {
+      var _a2;
+      if ((_a2 = p.functionResponse) == null ? void 0 : _a2.id) {
+        fnResMap.set(p.functionResponse.id, p.functionResponse);
+      }
+    });
+  });
+  if (fnResMap.size > 0) {
+    let paired = false;
+    nextMessages = nextMessages.map((msg) => {
+      if (msg.functionCall && !msg.functionResponse) {
+        const callId = msg.functionCall.id;
+        if (callId && fnResMap.has(callId)) {
+          paired = true;
+          return { ...msg, functionResponse: fnResMap.get(callId) };
+        }
+      }
+      return msg;
+    });
+    if (!paired)
+      nextMessages = currentMessages;
+  }
   const missingEvents = sessionEvents.filter((event) => {
     var _a, _b;
     const parts = (_a = event.content) == null ? void 0 : _a.parts;
     const canRender = Boolean(event.id) && ((_b = event.content) == null ? void 0 : _b.role) === "model" && Array.isArray(parts) && parts.length > 0;
     if (!canRender || renderedEventIds.has(event.id))
       return false;
-    const eventMessages = mapSessionMessages([event]);
-    const matchedIndexes = [];
-    const isRenderedByStream = eventMessages.length > 0 && eventMessages.every((eventMessage) => {
-      const comparableContent = getComparableMessageContent(eventMessage);
-      const matchedIndex = currentMessages.findIndex(
-        (currentMessage, index) => unmatchedStreamMessageIndexes.has(index) && !matchedIndexes.includes(index) && currentMessage.invocationId === eventMessage.invocationId && getComparableMessageContent(currentMessage) === comparableContent
-      );
-      if (matchedIndex < 0)
-        return false;
-      matchedIndexes.push(matchedIndex);
-      return true;
-    });
-    if (isRenderedByStream) {
-      matchedIndexes.forEach(
-        (index) => unmatchedStreamMessageIndexes.delete(index)
-      );
+    if (removedInvocationIds && removedInvocationIds.size > 0 && event.invocationId && removedInvocationIds.has(event.invocationId)) {
       return false;
     }
     return true;
   });
   if (missingEvents.length === 0)
-    return currentMessages;
+    return nextMessages;
   const missingMessages = mapSessionMessages(missingEvents);
   if (missingMessages.length === 0)
-    return currentMessages;
+    return nextMessages;
   const eventOrder = new Map(
     sessionEvents.map((event, index) => [event.id, index])
   );
@@ -281,7 +306,6 @@ var insertMissingSessionEvents = (currentMessages, sessionEvents) => {
     }
     return Number.POSITIVE_INFINITY;
   };
-  const nextMessages = [...currentMessages];
   missingMessages.forEach((messageItem) => {
     const targetOrder = getMessageOrder(messageItem);
     let insertIndex = nextMessages.findIndex((currentMessage) => {
@@ -298,7 +322,7 @@ function useADKChat({
   url = window.location.origin,
   token,
   config = {},
-  type = "agentDebug",
+  type: _type = "agentDebug",
   enabled = true,
   // ← 新增: 是否启用 Hook
   strategies,
@@ -338,10 +362,9 @@ function useADKChat({
   const [suggestedQuestions, setSuggestedQuestions] = (0, import_react.useState)([]);
   const [messages, setMessages] = (0, import_react.useState)([]);
   const pendingSessionEventsRef = (0, import_react.useRef)(null);
-  const sessionEventsFlushTimerRef = (0, import_react.useRef)(null);
-  const sessionEventsSyncingRef = (0, import_react.useRef)(false);
   const sessionDetailLoadingRef = (0, import_react.useRef)(false);
   const stateDeltaRef = (0, import_react.useRef)(void 0);
+  const reChatRemovedInvocationIdsRef = (0, import_react.useRef)(/* @__PURE__ */ new Set());
   const backgroundRef = (0, import_react.useRef)(/* @__PURE__ */ new Map());
   const messagesRef = (0, import_react.useRef)(messages);
   messagesRef.current = messages;
@@ -369,42 +392,24 @@ function useADKChat({
   }, [messages]);
   const textMsgRef = (0, import_react.useRef)(null);
   const eventDataRef = (0, import_react.useRef)(/* @__PURE__ */ new Map());
-  const cancelSessionEventsFlush = (0, import_react.useCallback)(() => {
-    if (sessionEventsFlushTimerRef.current === null)
-      return;
-    clearTimeout(sessionEventsFlushTimerRef.current);
-    sessionEventsFlushTimerRef.current = null;
-  }, []);
   const flushPendingSessionEvents = (0, import_react.useCallback)(() => {
     const pending = pendingSessionEventsRef.current;
-    const replyInProgress = loadingRef.current || sendingRef.current || textMsgRef.current !== null;
+    const replyInProgress = sendingRef.current || textMsgRef.current !== null;
     if (!pending || replyInProgress || pending.sessionId !== currentSessionIdRef.current)
       return;
-    const previewMessages = insertMissingSessionEvents(
-      messagesRef.current,
-      pending.events
-    );
     pendingSessionEventsRef.current = null;
-    if (previewMessages === messagesRef.current)
-      return;
-    sessionEventsSyncingRef.current = true;
-    setLoading(true);
-    sessionEventsFlushTimerRef.current = setTimeout(() => {
-      sessionEventsFlushTimerRef.current = null;
-      if (pending.sessionId === currentSessionIdRef.current) {
-        setMessages((prev) => {
-          const nextMessages = insertMissingSessionEvents(
-            prev,
-            pending.events
-          );
-          messagesRef.current = nextMessages;
-          return nextMessages;
-        });
-      }
-      sessionEventsSyncingRef.current = false;
-      setLoading(false);
-    }, 0);
-  }, [setLoading]);
+    setMessages((prev) => {
+      const nextMessages = insertMissingSessionEvents(
+        prev,
+        pending.events,
+        reChatRemovedInvocationIdsRef.current
+      );
+      if (nextMessages === prev)
+        return prev;
+      messagesRef.current = nextMessages;
+      return nextMessages;
+    });
+  }, []);
   const syncSessionEvents = (0, import_react.useCallback)(
     (events, targetSessionId) => {
       if (targetSessionId !== currentSessionIdRef.current)
@@ -419,12 +424,7 @@ function useADKChat({
   );
   (0, import_react.useEffect)(() => {
     pendingSessionEventsRef.current = null;
-    cancelSessionEventsFlush();
-    if (sessionEventsSyncingRef.current) {
-      sessionEventsSyncingRef.current = false;
-      setLoading(false);
-    }
-  }, [cancelSessionEventsFlush, currentSessionId, setLoading]);
+  }, [currentSessionId]);
   (0, import_react.useEffect)(() => {
     if (!loading)
       flushPendingSessionEvents();
@@ -438,18 +438,21 @@ function useADKChat({
       return [...prev, msg];
     });
   }, []);
-  const insertSuggestedQuestions = (suggested_questions = []) => {
-    if (!(suggested_questions == null ? void 0 : suggested_questions.length))
-      return;
-    if (messages.length === 0 && suggested_questions.length) {
-      setSuggestedQuestions(suggested_questions);
-    }
-  };
-  const insertPrologue = (prologue2) => {
-    if (prologue2) {
-      setPrologue(prologue2);
+  const insertSuggestedQuestions = (0, import_react.useCallback)(
+    (suggested_questions = []) => {
+      if (!(suggested_questions == null ? void 0 : suggested_questions.length))
+        return;
+      setSuggestedQuestions(
+        (prev) => messages.length === 0 ? suggested_questions : prev
+      );
+    },
+    [messages.length]
+  );
+  const insertPrologue = (0, import_react.useCallback)((prologueText) => {
+    if (prologueText) {
+      setPrologue(prologueText);
     }
-  };
+  }, []);
   const updateMessage = (0, import_react.useCallback)((msg) => {
     setMessages((prev) => {
       return prev.map((m) => {
@@ -515,47 +518,13 @@ function useADKChat({
   const storeMessage = (0, import_react.useCallback)(
     (part, event, role) => {
       var _a2, _b2;
-      const msg = {
-        id: (0, import_uuid.v4)(),
-        author: event.author,
-        invocationId: event.invocationId,
-        eventId: event.id,
-        timestamp: event.timestamp,
-        isLike: event.isLike || 0,
-        role,
-        modelCode: event.modelCode,
-        usageMetadata: event.usageMetadata,
-        finishReason: event.finishReason,
-        raw: event
-      };
-      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 ((_b2 = (_a2 = event == null ? void 0 : event.groundingMetadata) == null ? void 0 : _a2.searchEntryPoint) == null ? void 0 : _b2.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;
-        }
-      }
-      if (part && Object.keys(part).length > 0) {
-        insertMessage(msg);
+      if (!part || Object.keys(part).length === 0)
+        return;
+      const msg = buildNonTextMessage(event, part, role);
+      if (part.text && ((_b2 = (_a2 = event == null ? void 0 : event.groundingMetadata) == null ? void 0 : _a2.searchEntryPoint) == null ? void 0 : _b2.renderedContent)) {
+        msg.renderedContent = event.groundingMetadata.searchEntryPoint.renderedContent;
       }
+      insertMessage(msg);
     },
     [insertMessage]
   );
@@ -624,7 +593,7 @@ function useADKChat({
         setSuggestedQuestions((prev) => [...prev, text]);
       }
     },
-    [storeEvents, insertMessage]
+    [storeEvents]
   );
   const processErrorMessage = (0, import_react.useCallback)(
     (chunkJson) => {
@@ -750,29 +719,10 @@ function useADKChat({
                     }
                   } else {
                     bgTextMsg = null;
-                    const msg = {
-                      id: (0, import_uuid.v4)(),
-                      author: chunkJson.author,
-                      invocationId: chunkJson.invocationId,
-                      eventId: chunkJson.id,
-                      timestamp: chunkJson.timestamp,
-                      role: "bot"
-                    };
-                    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 {
-                      msg.text = part.text;
-                      msg.thought = part.thought;
-                      if (part.fileData)
-                        msg.fileData = [part.fileData];
-                    }
-                    bg.messages = [...bg.messages, msg];
+                    bg.messages = [
+                      ...bg.messages,
+                      buildNonTextMessage(chunkJson, part)
+                    ];
                   }
                 }
               } else if (chunkJson.errorMessage) {
@@ -809,12 +759,17 @@ function useADKChat({
                 bg.loading = false;
                 bg.textMsgRef = null;
               }
+              if (!backgroundRef.current.has(targetSessionId) && currentSessionIdRef.current === targetSessionId) {
+                setLoading(false);
+                textMsgRef.current = null;
+              }
             } else {
               setLoading(false);
               if (textMsgRef.current) {
                 onMessage == null ? void 0 : onMessage(((_a2 = textMsgRef.current) == null ? void 0 : _a2.text) || "", textMsgRef.current);
               }
               textMsgRef.current = null;
+              ctrl.current = null;
             }
             resolve();
           },
@@ -824,8 +779,13 @@ function useADKChat({
               const bg = backgroundRef.current.get(targetSessionId);
               if (bg)
                 bg.loading = false;
+              if (!backgroundRef.current.has(targetSessionId) && currentSessionIdRef.current === targetSessionId) {
+                setLoading(false);
+                ctrl.current = null;
+              }
             } else {
               setLoading(false);
+              ctrl.current = null;
             }
             resolve();
             console.error("EventSource failed:", error);
@@ -835,8 +795,6 @@ function useADKChat({
       });
     },
     [
-      type,
-      currentSessionId,
       processActionArtifact,
       processErrorMessage,
       processPart,
@@ -854,6 +812,7 @@ function useADKChat({
       return;
     sendingRef.current = true;
     setSuggestedQuestions([]);
+    reChatRemovedInvocationIdsRef.current = /* @__PURE__ */ new Set();
     setMessages((prev) => {
       let temp = [...prev];
       if (text.trim()) {
@@ -929,6 +888,13 @@ function useADKChat({
     }
     const newMessages = messages.slice(0, lastUserIndex + 1);
     setMessages(newMessages);
+    const removedInvocationIds = /* @__PURE__ */ new Set();
+    for (let i = lastUserIndex + 1; i < messages.length; i++) {
+      const msg = messages[i];
+      if (msg.invocationId)
+        removedInvocationIds.add(msg.invocationId);
+    }
+    reChatRemovedInvocationIdsRef.current = removedInvocationIds;
     const lastUserMessage = newMessages[lastUserIndex];
     const { text = "", fileData = [] } = lastUserMessage;
     try {
@@ -976,17 +942,16 @@ function useADKChat({
   const stopChat = (0, import_react.useCallback)(() => {
     var _a2;
     pendingSessionEventsRef.current = null;
-    cancelSessionEventsFlush();
-    sessionEventsSyncingRef.current = false;
     (_a2 = ctrl.current) == null ? void 0 : _a2.abort();
     ctrl.current = null;
     setLoading(false);
     textMsgRef.current = null;
-  }, [cancelSessionEventsFlush, setLoading]);
+  }, [setLoading]);
   const clearChat = () => {
     var _a2, _b2;
     const newSessionId = (0, import_uuid.v4)();
     setCurrentSessionId(newSessionId);
+    reChatRemovedInvocationIdsRef.current = /* @__PURE__ */ new Set();
     stopChat();
     setPrologue(((_a2 = appInfo == null ? void 0 : appInfo.onboardingInfo) == null ? void 0 : _a2.prologue) || "");
     setMessages([]);
@@ -1006,11 +971,7 @@ function useADKChat({
         return;
       }
       pendingSessionEventsRef.current = null;
-      cancelSessionEventsFlush();
-      if (sessionEventsSyncingRef.current) {
-        sessionEventsSyncingRef.current = false;
-        setLoading(false);
-      }
+      reChatRemovedInvocationIdsRef.current = /* @__PURE__ */ new Set();
       const prevSessionId = currentSessionId;
       if (ctrl.current) {
         backgroundRef.current.set(prevSessionId, {
@@ -1033,9 +994,11 @@ function useADKChat({
       const bg = backgroundRef.current.get(sessionId);
       if (bg) {
         setMessages(bg.messages);
-        setLoading(bg.loading);
+        messagesRef.current = bg.messages;
         textMsgRef.current = bg.textMsgRef;
-        ctrl.current = bg.loading ? bg.ctrl : null;
+        const isSseStillActive = bg.loading && !bg.ctrl.signal.aborted;
+        setLoading(isSseStillActive);
+        ctrl.current = isSseStillActive ? bg.ctrl : null;
         setCurrentSessionId(sessionId);
         setInitialized(true);
         backgroundRef.current.delete(sessionId);
@@ -1242,7 +1205,7 @@ function useADKChat({
     let requesting = false;
     const pollingSessionId = currentSessionId;
     const poll = async () => {
-      if (requesting || sessionDetailLoadingRef.current)
+      if (disposed || requesting || sessionDetailLoadingRef.current)
         return;
       requesting = true;
       try {
@@ -1260,7 +1223,6 @@ function useADKChat({
         requesting = false;
       }
     };
-    void poll();
     const timer = window.setInterval(poll, sessionEventsPollingInterval);
     return () => {
       disposed = true;
@@ -1279,7 +1241,6 @@ function useADKChat({
   (0, import_react.useEffect)(() => {
     return () => {
       var _a2;
-      cancelSessionEventsFlush();
       (_a2 = ctrl.current) == null ? void 0 : _a2.abort();
       backgroundRef.current.forEach((bg) => {
         var _a3;
@@ -1287,7 +1248,7 @@ function useADKChat({
       });
       backgroundRef.current.clear();
     };
-  }, [cancelSessionEventsFlush]);
+  }, []);
   return {
     appInfo,
     startChat,