yt-chat-components 0.9.9 → 1.0.1

package/src/YtChatView/chatWidget/chatWindow/controllers/index.ts

@@ -1,247 +1,250 @@
-// @ts-nocheck
-import axios from "axios";
-
-
-/**
- * 发送消息
- * @param embed_app_extend 扩展参数
- * @param isStream 是否启用流式传输
- * @param handleMessageContent 处理消息内容的回调函数
- * @param signal 用于主动关闭请求流
- * @param baseUrl 接口地址
- * @param flowId AI应用ID
- * @param message 消息内容
- * @param input_type 输入类型
- * @param output_type 输出类型
- * @param sessionId 对话ID
- * @param output_component
- * @param tweaks
- * @param api_key 接口密钥
- * @param additional_headers
- */
-export async function sendMessage(
-  embed_app_extend: object,
-  isStream: boolean,
-  handleMessageContent: Function,
-  signal,
-  baseUrl: string,
-  flowId: string,
-  message: string,
-  input_type: string,
-  output_type: string,
-  sessionId: React.MutableRefObject<string>,
-  output_component?: string,
-  tweaks?: Object,
-  api_key?: string,
-  additional_headers?: {
-    [key: string]: string;
-  },
-) {
-  let data: any;
-  data = {input_type, input_value: message, output_type};
-  if (tweaks) {
-    data['tweaks'] = tweaks;
-  }
-  if (output_component) {
-    data['output_component'] = output_component;
-  }
-  if (embed_app_extend) {
-    data['embed_app_extend'] = embed_app_extend;
-  }
-  let headers: { [key: string]: string } = {'Content-Type': 'application/json', 'timeout': 600000};
-  if (api_key) {
-    headers['x-api-key'] = api_key;
-  }
-  if (additional_headers) {
-    headers = Object.assign(headers, additional_headers);
-    // headers = {...headers, ...additional_headers};
-  }
-  // @ts-ignore
-  if (sessionId && sessionId != '') {
-    data.session_id = sessionId;
-  }
-
-  if (isStream) {
-    return await fetchCustomStream(
-      `${baseUrl}/api/v1/run/${flowId}?stream=true`,
-      data,
-      headers,
-      handleMessageContent,
-      signal,
-    );
-  } else {
-    let response = axios.post(`${baseUrl}/api/v1/run/${flowId}`, data, { headers });
-    return response;
-  }
-}
-
-// 缓冲区用于累积未完成的 JSON 数据
-// let buffer = '';
-
-async function fetchCustomStream(url, body, headers, handleMessageContent, signal) {
-  const response = await fetch(url, {
-    method: 'POST',
-    headers: {...headers, timeout: 600000},
-    body: JSON.stringify(body),
-    signal: signal,
-  });
-
-  if (!response.ok) {
-    throw new Error('请求失败');
-  }
-
-  // 手动读取响应体作为可读流
-  const reader = response.body.getReader();
-  const decoder = new TextDecoder();
-  let buffer = '';
-
-  while (true) {
-    const {done, value} = await reader.read();
-    if (done) break;
-
-    // 将二进制数据解码为字符串，并累积到缓冲区
-    const chunk = decoder.decode(value, {stream: true});
-    buffer += chunk;
-
-    // 提取并解析 JSON
-    const {parsedData, remainingBuffer} = extractAndParseJSON(buffer, handleMessageContent);
-
-    // 更新缓冲区为剩余未处理的数据
-    buffer = remainingBuffer;
-  }
-
-  // 如果缓冲区中还有剩余数据，可能是不完整的 JSON
-  if (buffer.trim() !== '') {
-    console.warn('缓冲区中存在未完成的 JSON 数据:', buffer);
-  }
-}
-
-// 辅助函数：从缓冲区中提取并解析所有完整的 JSON 对象
-function extractAndParseJSON(buffer, handleMessageContent) {
-  let remainingBuffer = buffer; // 剩余未处理的数据
-  const parsedData = []; // 存储解析成功的 JSON 对象
-  let bracketCount = 0; // 记录大括号的嵌套层次
-  let jsonStartIndex = -1; // 当前 JSON 对象的起始索引
-  let inString = false; // 是否处于字符串内部
-  let escapeChar = false; // 是否遇到转义字符
-
-  for (let i = 0; i < remainingBuffer.length; i++) {
-    const char = remainingBuffer[i];
-
-    if (escapeChar) {
-      escapeChar = false; // 跳过转义字符后的下一个字符
-      continue;
-    }
-
-    if (char === '\\') {
-      escapeChar = true; // 标记转义字符
-      continue;
-    }
-
-    if (char === '"' && !escapeChar) {
-      inString = !inString; // 切换字符串状态
-      continue;
-    }
-
-    if (!inString) {
-      if (char === '{') {
-        if (bracketCount === 0) {
-          jsonStartIndex = i; // 标记 JSON 开始位置
-        }
-        bracketCount++;
-      } else if (char === '}') {
-        bracketCount--;
-        if (bracketCount === 0 && jsonStartIndex !== -1) {
-          // 找到一个完整的 JSON 对象
-          const jsonString = remainingBuffer.slice(jsonStartIndex, i + 1);
-
-          try {
-            const parsedJson = JSON.parse(jsonString); // 解析 JSON
-            handleMessageContent(parsedJson['event'], parsedJson['data']);
-            parsedData.push(parsedJson); // 添加到解析结果中
-          } catch (error) {
-            console.warn('无法解析 JSON 数据:', error);
-          }
-
-          // 更新剩余缓冲区的起始位置
-          remainingBuffer = remainingBuffer.slice(i + 1);
-          i = -1; // 重置索引以重新开始扫描
-          jsonStartIndex = -1; // 重置 JSON 起始索引
-        }
-      }
-    }
-  }
-
-  return {parsedData, remainingBuffer}; // 返回解析结果和剩余缓冲区
-}
-
-/**
- * 获取对话历史列表
- * @param baseUrl
- * @param flowId
- * @param operatorId
- */
-export async function getHistoryList(baseUrl: string, flowId: string, operatorId: string) {
-  return axios.get(`${baseUrl}/api/v1/embed/chat_history_list/${flowId}/${operatorId}`, {timeout: 600000});
-}
-
-/**
- * 获取对话历史记录
- * @param baseUrl
- * @param flowI
- * @param sessionId
- * @param operatorId
- */
-export async function getChatHistory(
-  baseUrl: string,
-  flowId: string,
-  sessionId: string,
-  operatorId: string,
-) {
-  return axios.get(`${baseUrl}/api/v1/embed/chat_history/${flowId}/${sessionId}/${operatorId}`);
-}
-
-
-/**
- * 上传文件
- * @param baseUrl
- * @param file
- * @param flowId
- * @param api_key
- */
-export async function fetchUploadFile(baseUrl: string, file: File, flowId: string, api_key: string) {
-  const headers = {
-    'Content-Type': 'multipart/form-data',
-    'x-api-key': api_key,
-    timeout:600000
-  };
-  const formData = new FormData();
-  formData.append('file', file);
-  formData.append('folder', flowId);
-
-  return axios.post(`${baseUrl}/api/t1/file/upload`, formData, {headers});
-}
-
-/**
- * 获取场景信息
- * @param baseUrl
- * @param sceneId
- * @param api_key
- */
-export async function getSceneInfo(baseUrl: string, sceneId: string, api_key: string) {
-  const headers = {
-    'Content-Type': 'multipart/form-data',
-    'x-api-key': api_key,
-    timeout:600000
-  };
-  return axios.get(`${baseUrl}/api/t1/scene/detail/${sceneId}`,{headers});
-}
-
-export async function getFlowInfo(baseUrl: string, flowId: string, api_key: string) {
-  const headers = {
-    'Content-Type': 'multipart/form-data',
-    'x-api-key': api_key,
-    timeout:600000
-  };
-  return axios.get(`${baseUrl}/api/t1/flow/detail/${flowId}`,{headers});
+// @ts-nocheck
+import axios from "axios";
+import {InputValueType} from "../types/chatWidget";
+
+
+/**
+ * 发送消息
+ * @param input_value_type
+ * @param isStream
+ * @param embed_app_extend 扩展参数 * @param isStream 是否启用流式传输
+ * @param handleMessageContent 处理消息内容的回调函数
+ * @param signal 用于主动关闭请求流
+ * @param baseUrl 接口地址
+ * @param flowId AI应用ID
+ * @param message 消息内容
+ * @param input_type 输入类型
+ * @param output_type 输出类型
+ * @param sessionId 对话ID
+ * @param output_component
+ * @param tweaks
+ * @param api_key 接口密钥
+ * @param additional_headers
+ */
+export async function sendMessage(
+  input_value_type: string = InputValueType.text,
+  embed_app_extend: object,
+  isStream: boolean,
+  handleMessageContent: Function,
+  signal,
+  baseUrl: string,
+  flowId: string,
+  message: string,
+  input_type: string,
+  output_type: string,
+  sessionId: React.MutableRefObject<string>,
+  output_component?: string,
+  tweaks?: Object,
+  api_key?: string,
+  additional_headers?: {
+    [key: string]: string;
+  },
+) {
+  let data: any;
+  data = {input_value_type, input_type, input_value: message, output_type};
+  if (tweaks) {
+    data['tweaks'] = tweaks;
+  }
+  if (output_component) {
+    data['output_component'] = output_component;
+  }
+  if (embed_app_extend) {
+    data['embed_app_extend'] = embed_app_extend;
+  }
+  let headers: { [key: string]: string } = {'Content-Type': 'application/json', 'timeout': 600000};
+  if (api_key) {
+    headers['x-api-key'] = api_key;
+  }
+  if (additional_headers) {
+    headers = Object.assign(headers, additional_headers);
+    // headers = {...headers, ...additional_headers};
+  }
+  // @ts-ignore
+  if (sessionId && sessionId != '') {
+    data.session_id = sessionId;
+  }
+
+  if (isStream) {
+    return await fetchCustomStream(
+      `${baseUrl}/api/v1/run/${flowId}?stream=true`,
+      data,
+      headers,
+      handleMessageContent,
+      signal,
+    );
+  } else {
+    let response = axios.post(`${baseUrl}/api/v1/run/${flowId}`, data, { headers });
+    return response;
+  }
+}
+
+// 缓冲区用于累积未完成的 JSON 数据
+// let buffer = '';
+
+async function fetchCustomStream(url, body, headers, handleMessageContent, signal) {
+  const response = await fetch(url, {
+    method: 'POST',
+    headers: {...headers, timeout: 600000},
+    body: JSON.stringify(body),
+    signal: signal,
+  });
+
+  if (!response.ok) {
+    throw new Error('请求失败');
+  }
+
+  // 手动读取响应体作为可读流
+  const reader = response.body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = '';
+
+  while (true) {
+    const {done, value} = await reader.read();
+    if (done) break;
+
+    // 将二进制数据解码为字符串，并累积到缓冲区
+    const chunk = decoder.decode(value, {stream: true});
+    buffer += chunk;
+
+    // 提取并解析 JSON
+    const {parsedData, remainingBuffer} = extractAndParseJSON(buffer, handleMessageContent);
+
+    // 更新缓冲区为剩余未处理的数据
+    buffer = remainingBuffer;
+  }
+
+  // 如果缓冲区中还有剩余数据，可能是不完整的 JSON
+  if (buffer.trim() !== '') {
+    console.warn('缓冲区中存在未完成的 JSON 数据:', buffer);
+  }
+}
+
+// 辅助函数：从缓冲区中提取并解析所有完整的 JSON 对象
+function extractAndParseJSON(buffer, handleMessageContent) {
+  let remainingBuffer = buffer; // 剩余未处理的数据
+  const parsedData = []; // 存储解析成功的 JSON 对象
+  let bracketCount = 0; // 记录大括号的嵌套层次
+  let jsonStartIndex = -1; // 当前 JSON 对象的起始索引
+  let inString = false; // 是否处于字符串内部
+  let escapeChar = false; // 是否遇到转义字符
+
+  for (let i = 0; i < remainingBuffer.length; i++) {
+    const char = remainingBuffer[i];
+
+    if (escapeChar) {
+      escapeChar = false; // 跳过转义字符后的下一个字符
+      continue;
+    }
+
+    if (char === '\\') {
+      escapeChar = true; // 标记转义字符
+      continue;
+    }
+
+    if (char === '"' && !escapeChar) {
+      inString = !inString; // 切换字符串状态
+      continue;
+    }
+
+    if (!inString) {
+      if (char === '{') {
+        if (bracketCount === 0) {
+          jsonStartIndex = i; // 标记 JSON 开始位置
+        }
+        bracketCount++;
+      } else if (char === '}') {
+        bracketCount--;
+        if (bracketCount === 0 && jsonStartIndex !== -1) {
+          // 找到一个完整的 JSON 对象
+          const jsonString = remainingBuffer.slice(jsonStartIndex, i + 1);
+
+          try {
+            const parsedJson = JSON.parse(jsonString); // 解析 JSON
+            handleMessageContent(parsedJson['event'], parsedJson['data']);
+            parsedData.push(parsedJson); // 添加到解析结果中
+          } catch (error) {
+            console.warn('无法解析 JSON 数据:', error);
+          }
+
+          // 更新剩余缓冲区的起始位置
+          remainingBuffer = remainingBuffer.slice(i + 1);
+          i = -1; // 重置索引以重新开始扫描
+          jsonStartIndex = -1; // 重置 JSON 起始索引
+        }
+      }
+    }
+  }
+
+  return {parsedData, remainingBuffer}; // 返回解析结果和剩余缓冲区
+}
+
+/**
+ * 获取对话历史列表
+ * @param baseUrl
+ * @param flowId
+ * @param operatorId
+ */
+export async function getHistoryList(baseUrl: string, flowId: string, operatorId: string) {
+  return axios.get(`${baseUrl}/api/v1/embed/chat_history_list/${flowId}/${operatorId}`, {timeout: 600000});
+}
+
+/**
+ * 获取对话历史记录
+ * @param baseUrl
+ * @param flowI
+ * @param sessionId
+ * @param operatorId
+ */
+export async function getChatHistory(
+  baseUrl: string,
+  flowId: string,
+  sessionId: string,
+  operatorId: string,
+) {
+  return axios.get(`${baseUrl}/api/v1/embed/chat_history/${flowId}/${sessionId}/${operatorId}`);
+}
+
+
+/**
+ * 上传文件
+ * @param baseUrl
+ * @param file
+ * @param flowId
+ * @param api_key
+ */
+export async function fetchUploadFile(baseUrl: string, file: File, flowId: string, api_key: string) {
+  const headers = {
+    'Content-Type': 'multipart/form-data',
+    'x-api-key': api_key,
+    timeout:600000
+  };
+  const formData = new FormData();
+  formData.append('file', file);
+  formData.append('folder', flowId);
+
+  return axios.post(`${baseUrl}/api/t1/file/upload`, formData, {headers});
+}
+
+/**
+ * 获取场景信息
+ * @param baseUrl
+ * @param sceneId
+ * @param api_key
+ */
+export async function getSceneInfo(baseUrl: string, sceneId: string, api_key: string) {
+  const headers = {
+    'Content-Type': 'multipart/form-data',
+    'x-api-key': api_key,
+    timeout:600000
+  };
+  return axios.get(`${baseUrl}/api/t1/scene/detail/${sceneId}`,{headers});
+}
+
+export async function getFlowInfo(baseUrl: string, flowId: string, api_key: string) {
+  const headers = {
+    'Content-Type': 'multipart/form-data',
+    'x-api-key': api_key,
+    timeout:600000
+  };
+  return axios.get(`${baseUrl}/api/t1/flow/detail/${flowId}`,{headers});
 }