@yoka-ui/ui 1.0.1

package/dist/es/business/AiChat/sse.d.ts

@@ -0,0 +1,16 @@
+/**
+ * SSE 流式响应处理
+ * 支持大文件传输和流式数据处理
+ */
+export interface SSEOptions {
+    onMessage: (data: string) => void;
+    onEnd?: () => void;
+    onError?: (err: Error) => void;
+    onProgress?: (loaded: number, total?: number) => void;
+}
+export declare function createSSE(url: string, body: any, onMessage: (data: string) => void, onEnd?: () => void, onError?: (err: any) => void): Promise<() => void>;
+/**
+ * 创建增强型SSE连接
+ * 支持进度追踪、断点续传等高级功能
+ */
+export declare function createEnhancedSSE(url: string, body: any, options: SSEOptions): Promise<() => void>;

package/dist/es/business/AiChat/sse.js

@@ -0,0 +1,171 @@
+var __async = (__this, __arguments, generator) => {
+  return new Promise((resolve, reject) => {
+    var fulfilled = (value) => {
+      try {
+        step(generator.next(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var rejected = (value) => {
+      try {
+        step(generator.throw(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
+    step((generator = generator.apply(__this, __arguments)).next());
+  });
+};
+
+// src/business/AiChat/sse.ts
+function createSSE(url, body, onMessage, onEnd, onError) {
+  return __async(this, null, function* () {
+    const controller = new AbortController();
+    const signal = controller.signal;
+    try {
+      const res = yield fetch(url, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify(body),
+        signal
+      });
+      if (!res.ok) {
+        const errorText = yield res.text().catch(() => "");
+        throw new Error(`HTTP ${res.status}: ${errorText || res.statusText}`);
+      }
+      if (!res.body)
+        throw new Error("Response body is null, streaming not supported");
+      const reader = res.body.getReader();
+      const decoder = new TextDecoder("utf-8");
+      let buffer = "";
+      const contentLength = res.headers.get("Content-Length");
+      const total = contentLength ? parseInt(contentLength, 10) : void 0;
+      let loaded = 0;
+      while (true) {
+        const { done, value } = yield reader.read();
+        if (done) {
+          if (buffer.trim()) {
+            onMessage(buffer);
+          }
+          onEnd == null ? void 0 : onEnd();
+          break;
+        }
+        loaded += value.byteLength;
+        const chunk = decoder.decode(value, { stream: true });
+        buffer += chunk;
+        const lines = buffer.split("\n");
+        buffer = lines.pop() || "";
+        for (const line of lines) {
+          const trimmed = line.trim();
+          if (!trimmed)
+            continue;
+          if (trimmed.startsWith("data:")) {
+            const data = trimmed.slice(5).trim();
+            if (data === "[DONE]") {
+              onEnd == null ? void 0 : onEnd();
+              return () => controller.abort();
+            }
+            try {
+              const parsed = JSON.parse(data);
+              if (parsed.content) {
+                onMessage(parsed.content);
+              } else {
+                onMessage(data);
+              }
+            } catch (e) {
+              onMessage(data);
+            }
+          } else {
+            onMessage(trimmed);
+          }
+        }
+      }
+      return () => controller.abort();
+    } catch (err) {
+      if (err.name !== "AbortError") {
+        const error = err instanceof Error ? err : new Error(String(err));
+        onError == null ? void 0 : onError(error);
+      }
+      return () => controller.abort();
+    }
+  });
+}
+function createEnhancedSSE(url, body, options) {
+  return __async(this, null, function* () {
+    var _a, _b, _c, _d;
+    const controller = new AbortController();
+    const signal = controller.signal;
+    try {
+      const res = yield fetch(url, {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+          Accept: "text/event-stream"
+        },
+        body: JSON.stringify(body),
+        signal
+      });
+      if (!res.ok) {
+        const errorText = yield res.text().catch(() => "");
+        throw new Error(`HTTP ${res.status}: ${errorText || res.statusText}`);
+      }
+      if (!res.body) {
+        throw new Error("Response body is null");
+      }
+      const reader = res.body.getReader();
+      const decoder = new TextDecoder("utf-8");
+      const contentLength = res.headers.get("Content-Length");
+      const total = contentLength ? parseInt(contentLength, 10) : void 0;
+      let loaded = 0;
+      let buffer = "";
+      while (true) {
+        const { done, value } = yield reader.read();
+        if (done) {
+          if (buffer.trim()) {
+            options.onMessage(buffer);
+          }
+          (_a = options.onEnd) == null ? void 0 : _a.call(options);
+          break;
+        }
+        loaded += value.byteLength;
+        (_b = options.onProgress) == null ? void 0 : _b.call(options, loaded, total);
+        const chunk = decoder.decode(value, { stream: true });
+        buffer += chunk;
+        const lines = buffer.split("\n");
+        buffer = lines.pop() || "";
+        for (const line of lines) {
+          const trimmed = line.trim();
+          if (!trimmed || trimmed.startsWith(":"))
+            continue;
+          if (trimmed.startsWith("data:")) {
+            const data = trimmed.slice(5).trim();
+            if (data === "[DONE]") {
+              (_c = options.onEnd) == null ? void 0 : _c.call(options);
+              return () => controller.abort();
+            }
+            try {
+              const parsed = JSON.parse(data);
+              options.onMessage(parsed.content || data);
+            } catch (e) {
+              options.onMessage(data);
+            }
+          }
+        }
+      }
+      return () => controller.abort();
+    } catch (err) {
+      if (err.name !== "AbortError") {
+        const error = err instanceof Error ? err : new Error(String(err));
+        (_d = options.onError) == null ? void 0 : _d.call(options, error);
+      }
+      return () => controller.abort();
+    }
+  });
+}
+export {
+  createEnhancedSSE,
+  createSSE
+};
+//# sourceMappingURL=sse.js.map

package/dist/es/business/AiChat/sse.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../../src/business/AiChat/sse.ts"],
+  "sourcesContent": ["/**\n * SSE 流式响应处理\n * 支持大文件传输和流式数据处理\n */\n\nexport interface SSEOptions {\n  onMessage: (data: string) => void;\n  onEnd?: () => void;\n  onError?: (err: Error) => void;\n  onProgress?: (loaded: number, total?: number) => void;\n}\n\nexport async function createSSE(\n  url: string,\n  body: any,\n  onMessage: (data: string) => void,\n  onEnd?: () => void,\n  onError?: (err: any) => void,\n) {\n  const controller = new AbortController();\n  const signal = controller.signal;\n\n  try {\n    const res = await fetch(url, {\n      method: 'POST',\n      headers: { 'Content-Type': 'application/json' },\n      body: JSON.stringify(body),\n      signal,\n    });\n\n    if (!res.ok) {\n      const errorText = await res.text().catch(() => '');\n      throw new Error(`HTTP ${res.status}: ${errorText || res.statusText}`);\n    }\n\n    if (!res.body) throw new Error('Response body is null, streaming not supported');\n\n    const reader = res.body.getReader();\n    const decoder = new TextDecoder('utf-8');\n    let buffer = '';\n\n    // 获取内容长度(如果可用)\n    const contentLength = res.headers.get('Content-Length');\n    const total = contentLength ? parseInt(contentLength, 10) : undefined;\n    let loaded = 0;\n\n    while (true) {\n      const { done, value } = await reader.read();\n\n      if (done) {\n        // 处理剩余缓冲区数据\n        if (buffer.trim()) {\n          onMessage(buffer);\n        }\n        onEnd?.();\n        break;\n      }\n\n      loaded += value.byteLength;\n\n      // 解码数据块\n      const chunk = decoder.decode(value, { stream: true });\n      buffer += chunk;\n\n      // 按行处理数据(SSE协议)\n      const lines = buffer.split('\\n');\n      buffer = lines.pop() || ''; // 保留最后不完整的行\n\n      for (const line of lines) {\n        const trimmed = line.trim();\n        if (!trimmed) continue;\n\n        // 处理SSE格式: data: {...}\n        if (trimmed.startsWith('data:')) {\n          const data = trimmed.slice(5).trim();\n          if (data === '[DONE]') {\n            onEnd?.();\n            return () => controller.abort();\n          }\n          try {\n            // 尝试解析JSON\n            const parsed = JSON.parse(data);\n            if (parsed.content) {\n              onMessage(parsed.content);\n            } else {\n              onMessage(data);\n            }\n          } catch {\n            // 非JSON数据,直接传递\n            onMessage(data);\n          }\n        } else {\n          // 非SSE格式,直接传递\n          onMessage(trimmed);\n        }\n      }\n    }\n\n    return () => controller.abort();\n  } catch (err: any) {\n    if (err.name !== 'AbortError') {\n      const error = err instanceof Error ? err : new Error(String(err));\n      onError?.(error);\n    }\n    return () => controller.abort();\n  }\n}\n\n/**\n * 创建增强型SSE连接\n * 支持进度追踪、断点续传等高级功能\n */\nexport async function createEnhancedSSE(url: string, body: any, options: SSEOptions): Promise<() => void> {\n  const controller = new AbortController();\n  const signal = controller.signal;\n\n  try {\n    const res = await fetch(url, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n        Accept: 'text/event-stream',\n      },\n      body: JSON.stringify(body),\n      signal,\n    });\n\n    if (!res.ok) {\n      const errorText = await res.text().catch(() => '');\n      throw new Error(`HTTP ${res.status}: ${errorText || res.statusText}`);\n    }\n\n    if (!res.body) {\n      throw new Error('Response body is null');\n    }\n\n    const reader = res.body.getReader();\n    const decoder = new TextDecoder('utf-8');\n    const contentLength = res.headers.get('Content-Length');\n    const total = contentLength ? parseInt(contentLength, 10) : undefined;\n\n    let loaded = 0;\n    let buffer = '';\n\n    while (true) {\n      const { done, value } = await reader.read();\n\n      if (done) {\n        if (buffer.trim()) {\n          options.onMessage(buffer);\n        }\n        options.onEnd?.();\n        break;\n      }\n\n      loaded += value.byteLength;\n      options.onProgress?.(loaded, total);\n\n      const chunk = decoder.decode(value, { stream: true });\n      buffer += chunk;\n\n      // SSE 协议处理\n      const lines = buffer.split('\\n');\n      buffer = lines.pop() || '';\n\n      for (const line of lines) {\n        const trimmed = line.trim();\n        if (!trimmed || trimmed.startsWith(':')) continue;\n\n        if (trimmed.startsWith('data:')) {\n          const data = trimmed.slice(5).trim();\n          if (data === '[DONE]') {\n            options.onEnd?.();\n            return () => controller.abort();\n          }\n\n          try {\n            const parsed = JSON.parse(data);\n            options.onMessage(parsed.content || data);\n          } catch {\n            options.onMessage(data);\n          }\n        }\n      }\n    }\n\n    return () => controller.abort();\n  } catch (err: any) {\n    if (err.name !== 'AbortError') {\n      const error = err instanceof Error ? err : new Error(String(err));\n      options.onError?.(error);\n    }\n    return () => controller.abort();\n  }\n}\n"],
+  "mappings": ";;;;;;;;;;;;;;;;;;;;;;AAYA,SAAsB,UACpB,KACA,MACA,WACA,OACA,SACA;AAAA;AACA,UAAM,aAAa,IAAI,gBAAgB;AACvC,UAAM,SAAS,WAAW;AAE1B,QAAI;AACF,YAAM,MAAM,MAAM,MAAM,KAAK;AAAA,QAC3B,QAAQ;AAAA,QACR,SAAS,EAAE,gBAAgB,mBAAmB;AAAA,QAC9C,MAAM,KAAK,UAAU,IAAI;AAAA,QACzB;AAAA,MACF,CAAC;AAED,UAAI,CAAC,IAAI,IAAI;AACX,cAAM,YAAY,MAAM,IAAI,KAAK,EAAE,MAAM,MAAM,EAAE;AACjD,cAAM,IAAI,MAAM,QAAQ,IAAI,WAAW,aAAa,IAAI,YAAY;AAAA,MACtE;AAEA,UAAI,CAAC,IAAI;AAAM,cAAM,IAAI,MAAM,gDAAgD;AAE/E,YAAM,SAAS,IAAI,KAAK,UAAU;AAClC,YAAM,UAAU,IAAI,YAAY,OAAO;AACvC,UAAI,SAAS;AAGb,YAAM,gBAAgB,IAAI,QAAQ,IAAI,gBAAgB;AACtD,YAAM,QAAQ,gBAAgB,SAAS,eAAe,EAAE,IAAI;AAC5D,UAAI,SAAS;AAEb,aAAO,MAAM;AACX,cAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAE1C,YAAI,MAAM;AAER,cAAI,OAAO,KAAK,GAAG;AACjB,sBAAU,MAAM;AAAA,UAClB;AACA;AACA;AAAA,QACF;AAEA,kBAAU,MAAM;AAGhB,cAAM,QAAQ,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC;AACpD,kBAAU;AAGV,cAAM,QAAQ,OAAO,MAAM,IAAI;AAC/B,iBAAS,MAAM,IAAI,KAAK;AAExB,mBAAW,QAAQ,OAAO;AACxB,gBAAM,UAAU,KAAK,KAAK;AAC1B,cAAI,CAAC;AAAS;AAGd,cAAI,QAAQ,WAAW,OAAO,GAAG;AAC/B,kBAAM,OAAO,QAAQ,MAAM,CAAC,EAAE,KAAK;AACnC,gBAAI,SAAS,UAAU;AACrB;AACA,qBAAO,MAAM,WAAW,MAAM;AAAA,YAChC;AACA,gBAAI;AAEF,oBAAM,SAAS,KAAK,MAAM,IAAI;AAC9B,kBAAI,OAAO,SAAS;AAClB,0BAAU,OAAO,OAAO;AAAA,cAC1B,OAAO;AACL,0BAAU,IAAI;AAAA,cAChB;AAAA,YACF,SAAQ,GAAN;AAEA,wBAAU,IAAI;AAAA,YAChB;AAAA,UACF,OAAO;AAEL,sBAAU,OAAO;AAAA,UACnB;AAAA,QACF;AAAA,MACF;AAEA,aAAO,MAAM,WAAW,MAAM;AAAA,IAChC,SAAS,KAAP;AACA,UAAI,IAAI,SAAS,cAAc;AAC7B,cAAM,QAAQ,eAAe,QAAQ,MAAM,IAAI,MAAM,OAAO,GAAG,CAAC;AAChE,2CAAU;AAAA,MACZ;AACA,aAAO,MAAM,WAAW,MAAM;AAAA,IAChC;AAAA,EACF;AAAA;AAMA,SAAsB,kBAAkB,KAAa,MAAW,SAA0C;AAAA;AAhH1G;AAiHE,UAAM,aAAa,IAAI,gBAAgB;AACvC,UAAM,SAAS,WAAW;AAE1B,QAAI;AACF,YAAM,MAAM,MAAM,MAAM,KAAK;AAAA,QAC3B,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,QAAQ;AAAA,QACV;AAAA,QACA,MAAM,KAAK,UAAU,IAAI;AAAA,QACzB;AAAA,MACF,CAAC;AAED,UAAI,CAAC,IAAI,IAAI;AACX,cAAM,YAAY,MAAM,IAAI,KAAK,EAAE,MAAM,MAAM,EAAE;AACjD,cAAM,IAAI,MAAM,QAAQ,IAAI,WAAW,aAAa,IAAI,YAAY;AAAA,MACtE;AAEA,UAAI,CAAC,IAAI,MAAM;AACb,cAAM,IAAI,MAAM,uBAAuB;AAAA,MACzC;AAEA,YAAM,SAAS,IAAI,KAAK,UAAU;AAClC,YAAM,UAAU,IAAI,YAAY,OAAO;AACvC,YAAM,gBAAgB,IAAI,QAAQ,IAAI,gBAAgB;AACtD,YAAM,QAAQ,gBAAgB,SAAS,eAAe,EAAE,IAAI;AAE5D,UAAI,SAAS;AACb,UAAI,SAAS;AAEb,aAAO,MAAM;AACX,cAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAE1C,YAAI,MAAM;AACR,cAAI,OAAO,KAAK,GAAG;AACjB,oBAAQ,UAAU,MAAM;AAAA,UAC1B;AACA,wBAAQ,UAAR;AACA;AAAA,QACF;AAEA,kBAAU,MAAM;AAChB,sBAAQ,eAAR,iCAAqB,QAAQ;AAE7B,cAAM,QAAQ,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC;AACpD,kBAAU;AAGV,cAAM,QAAQ,OAAO,MAAM,IAAI;AAC/B,iBAAS,MAAM,IAAI,KAAK;AAExB,mBAAW,QAAQ,OAAO;AACxB,gBAAM,UAAU,KAAK,KAAK;AAC1B,cAAI,CAAC,WAAW,QAAQ,WAAW,GAAG;AAAG;AAEzC,cAAI,QAAQ,WAAW,OAAO,GAAG;AAC/B,kBAAM,OAAO,QAAQ,MAAM,CAAC,EAAE,KAAK;AACnC,gBAAI,SAAS,UAAU;AACrB,4BAAQ,UAAR;AACA,qBAAO,MAAM,WAAW,MAAM;AAAA,YAChC;AAEA,gBAAI;AACF,oBAAM,SAAS,KAAK,MAAM,IAAI;AAC9B,sBAAQ,UAAU,OAAO,WAAW,IAAI;AAAA,YAC1C,SAAQ,GAAN;AACA,sBAAQ,UAAU,IAAI;AAAA,YACxB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAEA,aAAO,MAAM,WAAW,MAAM;AAAA,IAChC,SAAS,KAAP;AACA,UAAI,IAAI,SAAS,cAAc;AAC7B,cAAM,QAAQ,eAAe,QAAQ,MAAM,IAAI,MAAM,OAAO,GAAG,CAAC;AAChE,sBAAQ,YAAR,iCAAkB;AAAA,MACpB;AACA,aAAO,MAAM,WAAW,MAAM;AAAA,IAChC;AAAA,EACF;AAAA;",
+  "names": []
+}

package/dist/es/business/AiChat/type.d.ts

@@ -0,0 +1,156 @@
+/// <reference types="react" />
+export interface Message {
+    id: string;
+    role: 'user' | 'ai' | 'system';
+    content: string;
+    loading?: boolean;
+    category?: MessageCategory;
+    tags?: string[];
+    taskId?: string;
+}
+export type MessageCategory = 'technical' | 'business' | 'general' | 'task_creation' | 'task_query' | 'api_call';
+export interface ChatSession {
+    id: string;
+    title: string;
+    createTime: number;
+    messages: Message[];
+}
+export interface AiChatApiCallbacks {
+    onFetchSessions?: () => Promise<ChatSession[]>;
+    onFetchSessionDetail?: (sessionId: string) => Promise<ChatSession>;
+    onCreateSession?: () => Promise<{
+        sessionId: string;
+    }>;
+    onSendMessage?: (params: {
+        sessionId: string;
+        prompt: string;
+        history: Message[];
+    }) => Promise<void>;
+    onUploadFile?: (params: {
+        sessionId: string;
+        file: File;
+        type: 'image' | 'document';
+    }) => Promise<{
+        url: string;
+        name: string;
+    }>;
+}
+export interface AiChatConfig extends AiChatApiCallbacks {
+    apiPath?: string;
+    storageKey?: string;
+    position?: {
+        bottom?: number;
+        right?: number;
+    };
+    buttonSize?: number;
+    drawerWidth?: number;
+    zIndex?: number;
+    showSessionPanel?: boolean;
+    onBeforeSend?: (content: string) => boolean | Promise<boolean>;
+    onError?: (error: Error) => void;
+    onSuccess?: () => void;
+}
+export interface UseAiChatOptions extends AiChatApiCallbacks {
+    apiPath?: string;
+    storageKey?: string;
+    onBeforeSend?: (content: string) => boolean | Promise<boolean>;
+    onError?: (error: Error) => void;
+    onSuccess?: () => void;
+}
+export interface UseAiChatReturn {
+    sessions: ChatSession[];
+    activeSession: ChatSession | null;
+    loading: boolean;
+    visible: boolean;
+    createNewSession: () => void;
+    switchSession: (id: string) => void;
+    deleteSession: (id: string, e: React.MouseEvent) => void;
+    sendQuestion: (content: string, regenId?: string) => Promise<void>;
+    regenerateAnswer: (msg: Message) => void;
+    deleteSingleMsg: (id: string, e: React.MouseEvent) => void;
+    cancelRequest: () => void;
+    toggleVisible: (v?: boolean) => void;
+}
+export interface Intent {
+    type: IntentType;
+    confidence: number;
+    parameters: Record<string, any>;
+    rawText: string;
+    matchedPattern?: string;
+}
+export type IntentType = 'create_task' | 'query_task' | 'update_task' | 'delete_task' | 'call_api' | 'general_question';
+export interface IntentPattern {
+    type: IntentType;
+    patterns: RegExp[];
+    taskType?: TaskCategory;
+}
+export interface Task {
+    id: string;
+    title: string;
+    description: string;
+    category: TaskCategory;
+    priority: TaskPriority;
+    status: TaskStatus;
+    dueDate?: number;
+    assignee?: string;
+    sourceMessageId?: string;
+    createTime: number;
+    apiResponse?: Record<string, any>;
+}
+export type TaskCategory = 'paimon' | 'data_sync' | 'report' | 'etl' | 'bug' | 'feature' | 'analysis' | 'other';
+export type TaskPriority = 'high' | 'medium' | 'low';
+export type TaskStatus = 'pending' | 'in_progress' | 'completed' | 'cancelled' | 'failed';
+export interface TaskTypeConfig {
+    type: string;
+    name: string;
+    description?: string;
+    apiEndpoint?: string;
+    defaultParams?: Record<string, any>;
+    urlPattern?: string;
+    keywords?: string[];
+}
+export interface TaskCreationResult {
+    success: boolean;
+    taskId?: string;
+    taskData?: any;
+    error?: string;
+    redirectUrl?: string;
+}
+export interface NavigationOptions {
+    replace?: boolean;
+    merge?: boolean;
+    queryParams?: Record<string, string>;
+    fragment?: string;
+}
+export interface EnhancedAiChatConfig extends AiChatConfig {
+    taskWorkflow?: {
+        enabled?: boolean;
+        createTaskApi?: string;
+        supportedTaskTypes?: TaskTypeConfig[];
+        autoNavigate?: boolean;
+        taskIdParamName?: string;
+        taskTypeParamName?: string;
+    };
+    intentRecognition?: {
+        enabled?: boolean;
+        customPatterns?: IntentPattern[];
+        confidenceThreshold?: number;
+    };
+    navigation?: {
+        refreshStrategy?: 'immediate' | 'delayed' | 'manual';
+        refreshDelay?: number;
+        preserveQueryParams?: boolean;
+        urlBuilder?: (taskId: string, taskType: string) => string;
+    };
+}
+export interface EnhancedUseAiChatReturn extends UseAiChatReturn {
+    tasks: Task[];
+    createTask: (task: Omit<Task, 'id' | 'createTime'>) => void;
+    updateTaskStatus: (taskId: string, status: TaskStatus) => void;
+    deleteTask: (taskId: string) => void;
+    recognizeIntent: (text: string) => Intent;
+    createTaskFromNaturalLanguage: (description: string) => Promise<TaskCreationResult>;
+    updateUrlWithTaskId: (taskId: string, options?: NavigationOptions) => void;
+    refreshCurrentPage: (taskId?: string) => void;
+    executeTaskCreationWorkflow: (description: string) => Promise<void>;
+}

package/dist/es/business/AiChat/type.js

@@ -0,0 +1 @@
+//# sourceMappingURL=type.js.map

package/dist/es/business/AiChat/type.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": [],
+  "sourcesContent": [],
+  "mappings": "",
+  "names": []
+}

package/dist/es/business/AiChat/useAiChat.d.ts

@@ -0,0 +1,2 @@
+import type { UseAiChatOptions, UseAiChatReturn } from './type';
+export declare function useAiChat(options?: UseAiChatOptions): UseAiChatReturn;

package/dist/es/business/AiChat/useAiChat.js

@@ -0,0 +1,268 @@
+var __defProp = Object.defineProperty;
+var __defProps = Object.defineProperties;
+var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
+var __async = (__this, __arguments, generator) => {
+  return new Promise((resolve, reject) => {
+    var fulfilled = (value) => {
+      try {
+        step(generator.next(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var rejected = (value) => {
+      try {
+        step(generator.throw(value));
+      } catch (e) {
+        reject(e);
+      }
+    };
+    var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
+    step((generator = generator.apply(__this, __arguments)).next());
+  });
+};
+
+// src/business/AiChat/useAiChat.ts
+import { message } from "antd";
+import { useCallback, useEffect, useState } from "react";
+import { createSSE } from "./sse";
+function useAiChat(options) {
+  const apiPath = (options == null ? void 0 : options.apiPath) || "/api/ai/chat-stream";
+  const storageKey = (options == null ? void 0 : options.storageKey) || "ai_chat_sessions";
+  const [sessions, setSessions] = useState([]);
+  const [activeSession, setActiveSession] = useState(null);
+  const [loading, setLoading] = useState(false);
+  const [cancelFn, setCancelFn] = useState(null);
+  const [visible, setVisible] = useState(false);
+  useEffect(() => {
+    const loadSessions = () => __async(this, null, function* () {
+      if (options == null ? void 0 : options.onFetchSessions) {
+        try {
+          const list = yield options.onFetchSessions();
+          setSessions(list);
+          if (list.length > 0)
+            setActiveSession(list[list.length - 1]);
+        } catch (err) {
+          console.error("加载会话失败:", err);
+        }
+      } else {
+        const saved = localStorage.getItem(storageKey);
+        if (saved) {
+          const list = JSON.parse(saved);
+          setSessions(list);
+          if (list.length > 0)
+            setActiveSession(list[list.length - 1]);
+        }
+      }
+    });
+    loadSessions();
+  }, [storageKey, options == null ? void 0 : options.onFetchSessions]);
+  useEffect(() => {
+    if (!(options == null ? void 0 : options.onFetchSessions) && sessions.length) {
+      localStorage.setItem(storageKey, JSON.stringify(sessions));
+    }
+  }, [sessions, storageKey, options == null ? void 0 : options.onFetchSessions]);
+  const createNewSession = useCallback(() => __async(this, null, function* () {
+    let newSessionId;
+    if (options == null ? void 0 : options.onCreateSession) {
+      try {
+        const result = yield options.onCreateSession();
+        newSessionId = result.sessionId;
+      } catch (err) {
+        console.error("创建会话失败:", err);
+        message.error("创建会话失败");
+        return;
+      }
+    } else {
+      newSessionId = Date.now().toString();
+    }
+    const s = {
+      id: newSessionId,
+      title: "新会话",
+      createTime: Date.now(),
+      messages: []
+    };
+    setSessions((prev) => [...prev, s]);
+    setActiveSession(s);
+    setVisible(true);
+  }), [options == null ? void 0 : options.onCreateSession]);
+  const switchSession = useCallback(
+    (id) => __async(this, null, function* () {
+      if (options == null ? void 0 : options.onFetchSessionDetail) {
+        try {
+          const detail = yield options.onFetchSessionDetail(id);
+          setActiveSession(detail);
+        } catch (err) {
+          console.error("获取会话详情失败:", err);
+          message.error("获取会话详情失败");
+        }
+      } else {
+        const s = sessions.find((x) => x.id === id);
+        if (s)
+          setActiveSession(s);
+      }
+    }),
+    [sessions, options == null ? void 0 : options.onFetchSessionDetail]
+  );
+  const deleteSession = useCallback(
+    (id, e) => {
+      var _a;
+      e.stopPropagation();
+      const ns = sessions.filter((x) => x.id !== id);
+      setSessions(ns);
+      if ((activeSession == null ? void 0 : activeSession.id) === id)
+        setActiveSession((_a = ns.at(-1)) != null ? _a : null);
+    },
+    [sessions, activeSession]
+  );
+  const sendQuestion = useCallback(
+    (content, regenId) => __async(this, null, function* () {
+      var _a, _b;
+      if (!content.trim() || !activeSession)
+        return;
+      if (options == null ? void 0 : options.onBeforeSend) {
+        const canSend = yield options.onBeforeSend(content);
+        if (!canSend)
+          return;
+      }
+      const user = { id: `${Date.now()}_u`, role: "user", content };
+      const ai = {
+        id: `${Date.now()}_ai`,
+        role: "ai",
+        content: "",
+        loading: true
+      };
+      let newMsgs = [];
+      if (regenId) {
+        newMsgs = activeSession.messages.map((m) => m.id === regenId ? ai : m);
+      } else {
+        newMsgs = [...activeSession.messages, user, ai];
+      }
+      const updated = __spreadProps(__spreadValues({}, activeSession), {
+        title: activeSession.messages.length === 0 ? content.slice(0, 20) : activeSession.title,
+        messages: newMsgs
+      });
+      setActiveSession(updated);
+      setSessions((prev) => prev.map((x) => x.id === updated.id ? updated : x));
+      setLoading(true);
+      if (options == null ? void 0 : options.onSendMessage) {
+        try {
+          yield options.onSendMessage({
+            sessionId: activeSession.id,
+            prompt: content,
+            history: updated.messages.filter((m) => !m.loading)
+          });
+          setLoading(false);
+          const ms = updated.messages.map((m) => m.loading ? __spreadProps(__spreadValues({}, m), { loading: false }) : m);
+          const su = __spreadProps(__spreadValues({}, updated), { messages: ms });
+          setActiveSession(su);
+          setSessions((prev) => prev.map((x) => x.id === su.id ? su : x));
+          (_a = options == null ? void 0 : options.onSuccess) == null ? void 0 : _a.call(options);
+        } catch (err) {
+          setLoading(false);
+          const error = err;
+          message.error(`请求异常：${error.message}`);
+          (_b = options == null ? void 0 : options.onError) == null ? void 0 : _b.call(options, error);
+        }
+      } else {
+        const cancel = yield createSSE(
+          apiPath,
+          {
+            prompt: content,
+            history: updated.messages.filter((m) => !m.loading),
+            useMcp: true,
+            sessionId: activeSession.id
+          },
+          (text) => {
+            if (!activeSession)
+              return;
+            const ms = updated.messages.map((m) => m.loading ? __spreadProps(__spreadValues({}, m), { content: m.content + text }) : m);
+            const su = __spreadProps(__spreadValues({}, updated), { messages: ms });
+            setActiveSession(su);
+            setSessions((prev) => prev.map((x) => x.id === su.id ? su : x));
+          },
+          () => {
+            var _a2;
+            setLoading(false);
+            const ms = updated.messages.map((m) => m.loading ? __spreadProps(__spreadValues({}, m), { loading: false }) : m);
+            const su = __spreadProps(__spreadValues({}, updated), { messages: ms });
+            setActiveSession(su);
+            setSessions((prev) => prev.map((x) => x.id === su.id ? su : x));
+            (_a2 = options == null ? void 0 : options.onSuccess) == null ? void 0 : _a2.call(options);
+          },
+          (err) => {
+            var _a2;
+            setLoading(false);
+            const errorMsg = err.message;
+            message.error(`请求异常：${errorMsg}`);
+            (_a2 = options == null ? void 0 : options.onError) == null ? void 0 : _a2.call(options, err);
+          }
+        );
+        setCancelFn(() => cancel);
+      }
+    }),
+    [activeSession, apiPath, options]
+  );
+  const regenerateAnswer = useCallback(
+    (msg) => {
+      if (!activeSession)
+        return;
+      const idx = activeSession.messages.findIndex((x) => x.id === msg.id);
+      const userMsg = activeSession.messages[idx - 1];
+      if ((userMsg == null ? void 0 : userMsg.role) === "user")
+        sendQuestion(userMsg.content, msg.id);
+    },
+    [activeSession, sendQuestion]
+  );
+  const deleteSingleMsg = useCallback(
+    (id, e) => {
+      e.stopPropagation();
+      if (!activeSession)
+        return;
+      const ms = activeSession.messages.filter((x) => x.id !== id);
+      const up = __spreadProps(__spreadValues({}, activeSession), { messages: ms });
+      setActiveSession(up);
+      setSessions((prev) => prev.map((x) => x.id === up.id ? up : x));
+    },
+    [activeSession]
+  );
+  const cancelRequest = useCallback(() => {
+    cancelFn == null ? void 0 : cancelFn();
+    setLoading(false);
+  }, [cancelFn]);
+  const toggleVisible = useCallback((v) => setVisible(v != null ? v : !visible), [visible]);
+  return {
+    sessions,
+    activeSession,
+    loading,
+    visible,
+    createNewSession,
+    switchSession,
+    deleteSession,
+    sendQuestion,
+    regenerateAnswer,
+    deleteSingleMsg,
+    cancelRequest,
+    toggleVisible
+  };
+}
+export {
+  useAiChat
+};
+//# sourceMappingURL=useAiChat.js.map