@unif/react-native-chat-sdk 0.1.1 → 0.2.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (162) hide show
  1. package/lib/commonjs/hooks/ChatProvider.js +76 -0
  2. package/lib/commonjs/hooks/ChatProvider.js.map +1 -0
  3. package/lib/commonjs/hooks/useXChat.js +173 -0
  4. package/lib/commonjs/hooks/useXChat.js.map +1 -0
  5. package/lib/commonjs/hooks/useXConversations.js +69 -0
  6. package/lib/commonjs/hooks/useXConversations.js.map +1 -0
  7. package/lib/commonjs/hooks/useXModel.js +27 -0
  8. package/lib/commonjs/hooks/useXModel.js.map +1 -0
  9. package/lib/commonjs/index.js +82 -0
  10. package/lib/commonjs/index.js.map +1 -0
  11. package/lib/commonjs/index.md +188 -0
  12. package/lib/commonjs/package.json +1 -0
  13. package/lib/commonjs/providers/AbstractChatProvider.js +64 -0
  14. package/lib/commonjs/providers/AbstractChatProvider.js.map +1 -0
  15. package/lib/commonjs/providers/DeepSeekChatProvider.js +88 -0
  16. package/lib/commonjs/providers/DeepSeekChatProvider.js.map +1 -0
  17. package/lib/commonjs/providers/OpenAIChatProvider.js +114 -0
  18. package/lib/commonjs/providers/OpenAIChatProvider.js.map +1 -0
  19. package/lib/commonjs/stores/chatStore.js +81 -0
  20. package/lib/commonjs/stores/chatStore.js.map +1 -0
  21. package/lib/commonjs/stores/historyStore.js +79 -0
  22. package/lib/commonjs/stores/historyStore.js.map +1 -0
  23. package/lib/commonjs/stores/modelStore.js +20 -0
  24. package/lib/commonjs/stores/modelStore.js.map +1 -0
  25. package/lib/commonjs/stores/sessionStore.js +24 -0
  26. package/lib/commonjs/stores/sessionStore.js.map +1 -0
  27. package/lib/commonjs/stores/storage.js +2 -0
  28. package/lib/commonjs/stores/storage.js.map +1 -0
  29. package/lib/commonjs/tools/SSEParser.js +56 -0
  30. package/lib/commonjs/tools/SSEParser.js.map +1 -0
  31. package/lib/commonjs/tools/XRequest.js +87 -0
  32. package/lib/commonjs/tools/XRequest.js.map +1 -0
  33. package/lib/commonjs/tools/XStream.js +84 -0
  34. package/lib/commonjs/tools/XStream.js.map +1 -0
  35. package/lib/commonjs/types/message.js +2 -0
  36. package/lib/commonjs/types/message.js.map +1 -0
  37. package/lib/commonjs/types/provider.js +6 -0
  38. package/lib/commonjs/types/provider.js.map +1 -0
  39. package/lib/commonjs/types/sse.js +2 -0
  40. package/lib/commonjs/types/sse.js.map +1 -0
  41. package/lib/module/hooks/ChatProvider.js +70 -0
  42. package/lib/module/hooks/ChatProvider.js.map +1 -0
  43. package/lib/module/hooks/useXChat.js +169 -0
  44. package/lib/module/hooks/useXChat.js.map +1 -0
  45. package/lib/module/hooks/useXConversations.js +65 -0
  46. package/lib/module/hooks/useXConversations.js.map +1 -0
  47. package/lib/module/hooks/useXModel.js +23 -0
  48. package/lib/module/hooks/useXModel.js.map +1 -0
  49. package/lib/module/index.js +27 -0
  50. package/lib/module/index.js.map +1 -0
  51. package/lib/module/index.md +188 -0
  52. package/lib/module/package.json +1 -0
  53. package/lib/module/providers/AbstractChatProvider.js +59 -0
  54. package/lib/module/providers/AbstractChatProvider.js.map +1 -0
  55. package/lib/module/providers/DeepSeekChatProvider.js +83 -0
  56. package/lib/module/providers/DeepSeekChatProvider.js.map +1 -0
  57. package/lib/module/providers/OpenAIChatProvider.js +109 -0
  58. package/lib/module/providers/OpenAIChatProvider.js.map +1 -0
  59. package/lib/module/stores/chatStore.js +76 -0
  60. package/lib/module/stores/chatStore.js.map +1 -0
  61. package/lib/module/stores/historyStore.js +75 -0
  62. package/lib/module/stores/historyStore.js.map +1 -0
  63. package/lib/module/stores/modelStore.js +15 -0
  64. package/lib/module/stores/modelStore.js.map +1 -0
  65. package/lib/module/stores/sessionStore.js +19 -0
  66. package/lib/module/stores/sessionStore.js.map +1 -0
  67. package/lib/module/stores/storage.js +2 -0
  68. package/lib/module/stores/storage.js.map +1 -0
  69. package/lib/module/tools/SSEParser.js +51 -0
  70. package/lib/module/tools/SSEParser.js.map +1 -0
  71. package/lib/module/tools/XRequest.js +82 -0
  72. package/lib/module/tools/XRequest.js.map +1 -0
  73. package/lib/module/tools/XStream.js +79 -0
  74. package/lib/module/tools/XStream.js.map +1 -0
  75. package/lib/module/types/message.js +2 -0
  76. package/lib/module/types/message.js.map +1 -0
  77. package/lib/module/types/provider.js +4 -0
  78. package/lib/module/types/provider.js.map +1 -0
  79. package/lib/module/types/sse.js +2 -0
  80. package/lib/module/types/sse.js.map +1 -0
  81. package/lib/typescript/commonjs/hooks/ChatProvider.d.ts +54 -0
  82. package/lib/typescript/commonjs/hooks/ChatProvider.d.ts.map +1 -0
  83. package/lib/typescript/commonjs/hooks/useXChat.d.ts +40 -0
  84. package/lib/typescript/commonjs/hooks/useXChat.d.ts.map +1 -0
  85. package/lib/typescript/commonjs/hooks/useXConversations.d.ts +24 -0
  86. package/lib/typescript/commonjs/hooks/useXConversations.d.ts.map +1 -0
  87. package/lib/typescript/commonjs/hooks/useXModel.d.ts +18 -0
  88. package/lib/typescript/commonjs/hooks/useXModel.d.ts.map +1 -0
  89. package/lib/typescript/commonjs/index.d.ts +27 -0
  90. package/lib/typescript/commonjs/index.d.ts.map +1 -0
  91. package/lib/typescript/commonjs/package.json +1 -0
  92. package/lib/typescript/commonjs/providers/AbstractChatProvider.d.ts +38 -0
  93. package/lib/typescript/commonjs/providers/AbstractChatProvider.d.ts.map +1 -0
  94. package/lib/typescript/commonjs/providers/DeepSeekChatProvider.d.ts +23 -0
  95. package/lib/typescript/commonjs/providers/DeepSeekChatProvider.d.ts.map +1 -0
  96. package/lib/typescript/commonjs/providers/OpenAIChatProvider.d.ts +33 -0
  97. package/lib/typescript/commonjs/providers/OpenAIChatProvider.d.ts.map +1 -0
  98. package/lib/typescript/commonjs/stores/chatStore.d.ts +27 -0
  99. package/lib/typescript/commonjs/stores/chatStore.d.ts.map +1 -0
  100. package/lib/typescript/commonjs/stores/historyStore.d.ts +22 -0
  101. package/lib/typescript/commonjs/stores/historyStore.d.ts.map +1 -0
  102. package/lib/typescript/commonjs/stores/modelStore.d.ts +11 -0
  103. package/lib/typescript/commonjs/stores/modelStore.d.ts.map +1 -0
  104. package/lib/typescript/commonjs/stores/sessionStore.d.ts +13 -0
  105. package/lib/typescript/commonjs/stores/sessionStore.d.ts.map +1 -0
  106. package/lib/typescript/commonjs/stores/storage.d.ts +10 -0
  107. package/lib/typescript/commonjs/stores/storage.d.ts.map +1 -0
  108. package/lib/typescript/commonjs/tools/SSEParser.d.ts +19 -0
  109. package/lib/typescript/commonjs/tools/SSEParser.d.ts.map +1 -0
  110. package/lib/typescript/commonjs/tools/XRequest.d.ts +33 -0
  111. package/lib/typescript/commonjs/tools/XRequest.d.ts.map +1 -0
  112. package/lib/typescript/commonjs/tools/XStream.d.ts +34 -0
  113. package/lib/typescript/commonjs/tools/XStream.d.ts.map +1 -0
  114. package/lib/typescript/commonjs/types/message.d.ts +22 -0
  115. package/lib/typescript/commonjs/types/message.d.ts.map +1 -0
  116. package/lib/typescript/commonjs/types/provider.d.ts +23 -0
  117. package/lib/typescript/commonjs/types/provider.d.ts.map +1 -0
  118. package/lib/typescript/commonjs/types/sse.d.ts +63 -0
  119. package/lib/typescript/commonjs/types/sse.d.ts.map +1 -0
  120. package/lib/typescript/module/hooks/ChatProvider.d.ts +54 -0
  121. package/lib/typescript/module/hooks/ChatProvider.d.ts.map +1 -0
  122. package/lib/typescript/module/hooks/useXChat.d.ts +40 -0
  123. package/lib/typescript/module/hooks/useXChat.d.ts.map +1 -0
  124. package/lib/typescript/module/hooks/useXConversations.d.ts +24 -0
  125. package/lib/typescript/module/hooks/useXConversations.d.ts.map +1 -0
  126. package/lib/typescript/module/hooks/useXModel.d.ts +18 -0
  127. package/lib/typescript/module/hooks/useXModel.d.ts.map +1 -0
  128. package/lib/typescript/module/index.d.ts +27 -0
  129. package/lib/typescript/module/index.d.ts.map +1 -0
  130. package/lib/typescript/module/package.json +1 -0
  131. package/lib/typescript/module/providers/AbstractChatProvider.d.ts +38 -0
  132. package/lib/typescript/module/providers/AbstractChatProvider.d.ts.map +1 -0
  133. package/lib/typescript/module/providers/DeepSeekChatProvider.d.ts +23 -0
  134. package/lib/typescript/module/providers/DeepSeekChatProvider.d.ts.map +1 -0
  135. package/lib/typescript/module/providers/OpenAIChatProvider.d.ts +33 -0
  136. package/lib/typescript/module/providers/OpenAIChatProvider.d.ts.map +1 -0
  137. package/lib/typescript/module/stores/chatStore.d.ts +27 -0
  138. package/lib/typescript/module/stores/chatStore.d.ts.map +1 -0
  139. package/lib/typescript/module/stores/historyStore.d.ts +22 -0
  140. package/lib/typescript/module/stores/historyStore.d.ts.map +1 -0
  141. package/lib/typescript/module/stores/modelStore.d.ts +11 -0
  142. package/lib/typescript/module/stores/modelStore.d.ts.map +1 -0
  143. package/lib/typescript/module/stores/sessionStore.d.ts +13 -0
  144. package/lib/typescript/module/stores/sessionStore.d.ts.map +1 -0
  145. package/lib/typescript/module/stores/storage.d.ts +10 -0
  146. package/lib/typescript/module/stores/storage.d.ts.map +1 -0
  147. package/lib/typescript/module/tools/SSEParser.d.ts +19 -0
  148. package/lib/typescript/module/tools/SSEParser.d.ts.map +1 -0
  149. package/lib/typescript/module/tools/XRequest.d.ts +33 -0
  150. package/lib/typescript/module/tools/XRequest.d.ts.map +1 -0
  151. package/lib/typescript/module/tools/XStream.d.ts +34 -0
  152. package/lib/typescript/module/tools/XStream.d.ts.map +1 -0
  153. package/lib/typescript/module/types/message.d.ts +22 -0
  154. package/lib/typescript/module/types/message.d.ts.map +1 -0
  155. package/lib/typescript/module/types/provider.d.ts +23 -0
  156. package/lib/typescript/module/types/provider.d.ts.map +1 -0
  157. package/lib/typescript/module/types/sse.d.ts +63 -0
  158. package/lib/typescript/module/types/sse.d.ts.map +1 -0
  159. package/package.json +40 -8
  160. package/src/hooks/ChatProvider.tsx +14 -1
  161. package/src/hooks/useXChat.ts +129 -4
  162. package/src/stores/chatStore.ts +35 -0
@@ -0,0 +1,22 @@
1
+ /**
2
+ * 历史会话持久化
3
+ * 使用 ChatStorage 抽象,不直接依赖 AsyncStorage
4
+ */
5
+ import type { ChatMessage } from '../types/message';
6
+ import type { ChatStorage } from './storage';
7
+ export interface SessionSummary {
8
+ id: string;
9
+ title: string;
10
+ lastMessage: string;
11
+ timestamp: number;
12
+ messageCount: number;
13
+ }
14
+ export interface HistoryStore {
15
+ sessions: SessionSummary[];
16
+ archiveSession: (id: string, title: string, messages: ChatMessage[]) => Promise<void>;
17
+ loadSessionMessages: (id: string) => Promise<ChatMessage[]>;
18
+ deleteSession: (id: string) => Promise<void>;
19
+ clearAll: () => Promise<void>;
20
+ }
21
+ export declare function createHistoryStore(storage: ChatStorage): HistoryStore;
22
+ //# sourceMappingURL=historyStore.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"historyStore.d.ts","sourceRoot":"","sources":["../../../../src/stores/historyStore.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,kBAAkB,CAAC;AAClD,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,WAAW,CAAC;AAK3C,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,cAAc,EAAE,CAAC;IAC3B,cAAc,EAAE,CACd,EAAE,EAAE,MAAM,EACV,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,WAAW,EAAE,KACpB,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,mBAAmB,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;IAC5D,aAAa,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,QAAQ,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC/B;AAED,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,WAAW,GAAG,YAAY,CAqFrE"}
@@ -0,0 +1,11 @@
1
+ /**
2
+ * 模型选择状态(Zustand)
3
+ * 内部 store,通过 useXModel hook 封装后对外暴露
4
+ */
5
+ interface ModelState {
6
+ selectedModel: string;
7
+ setSelectedModel: (id: string) => void;
8
+ }
9
+ export declare const createModelStore: (defaultModel: string) => import("zustand").UseBoundStore<import("zustand").StoreApi<ModelState>>;
10
+ export {};
11
+ //# sourceMappingURL=modelStore.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"modelStore.d.ts","sourceRoot":"","sources":["../../../../src/stores/modelStore.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,UAAU,UAAU;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,gBAAgB,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,IAAI,CAAC;CACxC;AAED,eAAO,MAAM,gBAAgB,GAAI,cAAc,MAAM,4EAIhD,CAAC"}
@@ -0,0 +1,13 @@
1
+ /**
2
+ * 会话状态管理(Zustand)
3
+ * 内部 store,通过 useXConversations hook 封装后对外暴露
4
+ */
5
+ interface SessionState {
6
+ sessionId: string;
7
+ sessionTitle: string;
8
+ setSessionId: (id: string) => void;
9
+ setSessionTitle: (title: string) => void;
10
+ }
11
+ export declare const createSessionStore: (generateId: () => string) => import("zustand").UseBoundStore<import("zustand").StoreApi<SessionState>>;
12
+ export {};
13
+ //# sourceMappingURL=sessionStore.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"sessionStore.d.ts","sourceRoot":"","sources":["../../../../src/stores/sessionStore.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,UAAU,YAAY;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,IAAI,CAAC;IACnC,eAAe,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CAC1C;AAED,eAAO,MAAM,kBAAkB,GAAI,YAAY,MAAM,MAAM,8EAMtD,CAAC"}
@@ -0,0 +1,10 @@
1
+ /**
2
+ * ChatStorage — 存储抽象接口
3
+ * 不绑定 AsyncStorage,支持注入任意存储实现
4
+ */
5
+ export interface ChatStorage {
6
+ getItem: (key: string) => Promise<string | null>;
7
+ setItem: (key: string, value: string) => Promise<void>;
8
+ removeItem: (key: string) => Promise<void>;
9
+ }
10
+ //# sourceMappingURL=storage.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../../../src/stores/storage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IACjD,OAAO,EAAE,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACvD,UAAU,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC5C"}
@@ -0,0 +1,19 @@
1
+ /**
2
+ * SSE 解析器
3
+ * - event_id 幂等去重
4
+ * - seq 严格排序(乱序缓冲,顺序到达后 flush)
5
+ * - 每轮对话调用 reset() 重置
6
+ */
7
+ import type { SSEEnvelope } from '../types/sse';
8
+ export declare class SSEParser {
9
+ private processedIds;
10
+ private buffer;
11
+ private nextSeq;
12
+ reset(): void;
13
+ /**
14
+ * 解析原始 SSE data 字符串,返回按 seq 排序的事件数组
15
+ * 已处理的 event_id 会被自动跳过(幂等)
16
+ */
17
+ process(raw: string): SSEEnvelope[];
18
+ }
19
+ //# sourceMappingURL=SSEParser.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"SSEParser.d.ts","sourceRoot":"","sources":["../../../../src/tools/SSEParser.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,cAAc,CAAC;AAE9C,qBAAa,SAAS;IACpB,OAAO,CAAC,YAAY,CAAqB;IACzC,OAAO,CAAC,MAAM,CAAkC;IAChD,OAAO,CAAC,OAAO,CAAK;IAEpB,KAAK,IAAI,IAAI;IAMb;;;OAGG;IACH,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,WAAW,EAAE;CA2BpC"}
@@ -0,0 +1,33 @@
1
+ /**
2
+ * XRequest — 面向 LLM 的 HTTP 请求工具
3
+ *
4
+ * 组合 XStream + SSEParser 实现完整 SSE 请求链路
5
+ */
6
+ import type { SSEOutput } from '../types/provider';
7
+ export interface XRequestConfig {
8
+ baseURL: string;
9
+ endpoint?: string;
10
+ model?: string;
11
+ headers?: Record<string, string>;
12
+ timeout?: number;
13
+ getToken?: () => Promise<string | null>;
14
+ }
15
+ export interface XRequestCallbacks {
16
+ onStream?: (chunk: SSEOutput) => void;
17
+ onSuccess?: (message: string) => void;
18
+ onError?: (error: {
19
+ type: string;
20
+ message: string;
21
+ }) => void;
22
+ onAbort?: () => void;
23
+ }
24
+ export declare class XRequest {
25
+ private config;
26
+ private stream;
27
+ private parser;
28
+ constructor(config: XRequestConfig);
29
+ create(params: Record<string, unknown>, callbacks: XRequestCallbacks): Promise<void>;
30
+ abort(): void;
31
+ dispose(): void;
32
+ }
33
+ //# sourceMappingURL=XRequest.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"XRequest.d.ts","sourceRoot":"","sources":["../../../../src/tools/XRequest.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,mBAAmB,CAAC;AAGjD,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;CACzC;AAED,MAAM,WAAW,iBAAiB;IAChC,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,IAAI,CAAC;IACtC,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IACtC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,CAAC;IAC7D,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;CACtB;AAED,qBAAa,QAAQ;IACnB,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,MAAM,CAAwB;IACtC,OAAO,CAAC,MAAM,CAAmB;gBAErB,MAAM,EAAE,cAAc;IAI5B,MAAM,CACV,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC/B,SAAS,EAAE,iBAAiB,GAC3B,OAAO,CAAC,IAAI,CAAC;IA0DhB,KAAK,IAAI,IAAI;IAOb,OAAO,IAAI,IAAI;CAGhB"}
@@ -0,0 +1,34 @@
1
+ /**
2
+ * XStream — SSE 流适配器(React Native 版本)
3
+ *
4
+ * 基于 react-native-sse 封装,支持 POST 方法 + 自定义请求体
5
+ */
6
+ import type { SSEOutput } from '../types/provider';
7
+ export interface XStreamConfig {
8
+ url: string;
9
+ method?: 'GET' | 'POST';
10
+ headers?: Record<string, string>;
11
+ body?: Record<string, unknown>;
12
+ timeout?: number;
13
+ transformStream?: (event: {
14
+ data: string;
15
+ event?: string;
16
+ }) => SSEOutput;
17
+ }
18
+ export interface XStreamCallbacks {
19
+ onMessage: (data: SSEOutput) => void;
20
+ onError?: (error: Error) => void;
21
+ onComplete?: () => void;
22
+ onOpen?: () => void;
23
+ }
24
+ export declare class XStream {
25
+ private config;
26
+ private es;
27
+ private timeoutTimer;
28
+ constructor(config: XStreamConfig);
29
+ connect(callbacks: XStreamCallbacks): void;
30
+ abort(): void;
31
+ dispose(): void;
32
+ private clearTimer;
33
+ }
34
+ //# sourceMappingURL=XStream.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"XStream.d.ts","sourceRoot":"","sources":["../../../../src/tools/XStream.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,mBAAmB,CAAC;AAEjD,MAAM,WAAW,aAAa;IAC5B,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,KAAK,SAAS,CAAC;CAC1E;AAED,MAAM,WAAW,gBAAgB;IAC/B,SAAS,EAAE,CAAC,IAAI,EAAE,SAAS,KAAK,IAAI,CAAC;IACrC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IACjC,UAAU,CAAC,EAAE,MAAM,IAAI,CAAC;IACxB,MAAM,CAAC,EAAE,MAAM,IAAI,CAAC;CACrB;AAUD,qBAAa,OAAO;IAClB,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,EAAE,CAA4B;IACtC,OAAO,CAAC,YAAY,CAA8C;gBAEtD,MAAM,EAAE,aAAa;IAIjC,OAAO,CAAC,SAAS,EAAE,gBAAgB,GAAG,IAAI;IAmD1C,KAAK,IAAI,IAAI;IAQb,OAAO,IAAI,IAAI;IAIf,OAAO,CAAC,UAAU;CAMnB"}
@@ -0,0 +1,22 @@
1
+ /**
2
+ * Chat 消息类型定义
3
+ */
4
+ export type MessageRole = 'user' | 'assistant' | 'system';
5
+ export type MessageType = 'text' | 'card' | 'system';
6
+ export type MessageStatus = 'local' | 'loading' | 'updating' | 'success' | 'error' | 'abort';
7
+ export interface ChatMessage {
8
+ id: string;
9
+ text: string;
10
+ role: MessageRole;
11
+ createdAt: Date;
12
+ messageType: MessageType;
13
+ status: MessageStatus;
14
+ cardType?: string;
15
+ cardData?: {
16
+ data: Record<string, unknown>;
17
+ actions: string[];
18
+ };
19
+ turnId: string;
20
+ extra?: Record<string, unknown>;
21
+ }
22
+ //# sourceMappingURL=message.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"message.d.ts","sourceRoot":"","sources":["../../../../src/types/message.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;AAC1D,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,MAAM,GAAG,QAAQ,CAAC;AACrD,MAAM,MAAM,aAAa,GACrB,OAAO,GACP,SAAS,GACT,UAAU,GACV,SAAS,GACT,OAAO,GACP,OAAO,CAAC;AAEZ,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,WAAW,CAAC;IAClB,SAAS,EAAE,IAAI,CAAC;IAChB,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,EAAE,aAAa,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAAC,OAAO,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IAChE,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACjC"}
@@ -0,0 +1,23 @@
1
+ /**
2
+ * Provider 协议类型
3
+ * 定义 ChatProvider ↔ useXChat 之间的数据契约
4
+ */
5
+ import type { ChatMessage } from './message';
6
+ import type { SuggestionItem } from './sse';
7
+ export interface RequestParams<T = ChatMessage> {
8
+ message: T;
9
+ messages?: T[];
10
+ model?: string;
11
+ extra?: Record<string, unknown>;
12
+ }
13
+ export interface SSEOutput {
14
+ data: string;
15
+ event?: string;
16
+ }
17
+ export interface ProviderCallbacks<Message = ChatMessage> {
18
+ onUpdate: (message: Message) => void;
19
+ onSuccess: (message: Message) => void;
20
+ onError: (error: Error) => void;
21
+ onSuggestion?: (items: SuggestionItem[]) => void;
22
+ }
23
+ //# sourceMappingURL=provider.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"provider.d.ts","sourceRoot":"","sources":["../../../../src/types/provider.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,WAAW,CAAC;AAC3C,OAAO,KAAK,EAAC,cAAc,EAAC,MAAM,OAAO,CAAC;AAE1C,MAAM,WAAW,aAAa,CAAC,CAAC,GAAG,WAAW;IAC5C,OAAO,EAAE,CAAC,CAAC;IACX,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACjC;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,iBAAiB,CAAC,OAAO,GAAG,WAAW;IACtD,QAAQ,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IACrC,SAAS,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IACtC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IAChC,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,cAAc,EAAE,KAAK,IAAI,CAAC;CAClD"}
@@ -0,0 +1,63 @@
1
+ /**
2
+ * SSEEnvelope 协议类型定义
3
+ */
4
+ export type SSEEventType = 'text' | 'card_data' | 'suggestion' | 'error' | 'done';
5
+ export interface SSEEnvelopeBase {
6
+ event_id: string;
7
+ turn_id: string;
8
+ seq: number;
9
+ ts: string;
10
+ session_id: string;
11
+ }
12
+ export interface SSETextEvent extends SSEEnvelopeBase {
13
+ type: 'text';
14
+ payload: {
15
+ content: string;
16
+ };
17
+ }
18
+ export interface SSECardDataEvent extends SSEEnvelopeBase {
19
+ type: 'card_data';
20
+ payload: {
21
+ cardType: string;
22
+ payload: CardPayload;
23
+ };
24
+ }
25
+ export interface SSESuggestionEvent extends SSEEnvelopeBase {
26
+ type: 'suggestion';
27
+ payload: {
28
+ items: SuggestionItem[];
29
+ };
30
+ }
31
+ export interface SSEErrorEvent extends SSEEnvelopeBase {
32
+ type: 'error';
33
+ payload: {
34
+ code: string;
35
+ message: string;
36
+ retryable: boolean;
37
+ recover_actions?: ActionPayload[];
38
+ };
39
+ }
40
+ export interface SSEDoneEvent extends SSEEnvelopeBase {
41
+ type: 'done';
42
+ payload: {
43
+ messageId: string;
44
+ reason: 'ok' | 'abort' | 'error';
45
+ };
46
+ }
47
+ export type SSEEnvelope = SSETextEvent | SSECardDataEvent | SSESuggestionEvent | SSEErrorEvent | SSEDoneEvent;
48
+ export interface SuggestionItem {
49
+ id: string;
50
+ label: string;
51
+ action?: ActionPayload;
52
+ mode?: 'action' | 'input_focus';
53
+ }
54
+ export type ActionType = 'confirm_order' | 'select_product' | 'reorder' | 'add_product' | 'clear_draft' | 'push_dealer' | 'new_order' | 'view_detail' | 'add_to_order' | 'browse_products' | 'retry_order';
55
+ export interface ActionPayload {
56
+ type: ActionType;
57
+ payload?: Record<string, unknown>;
58
+ }
59
+ export interface CardPayload {
60
+ data: Record<string, unknown>;
61
+ actions: string[];
62
+ }
63
+ //# sourceMappingURL=sse.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"sse.d.ts","sourceRoot":"","sources":["../../../../src/types/sse.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,MAAM,MAAM,YAAY,GACpB,MAAM,GACN,WAAW,GACX,YAAY,GACZ,OAAO,GACP,MAAM,CAAC;AAEX,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,GAAG,EAAE,MAAM,CAAC;IACZ,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,YAAa,SAAQ,eAAe;IACnD,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;CAC9B;AAED,MAAM,WAAW,gBAAiB,SAAQ,eAAe;IACvD,IAAI,EAAE,WAAW,CAAC;IAClB,OAAO,EAAE;QACP,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,WAAW,CAAC;KACtB,CAAC;CACH;AAED,MAAM,WAAW,kBAAmB,SAAQ,eAAe;IACzD,IAAI,EAAE,YAAY,CAAC;IACnB,OAAO,EAAE;QACP,KAAK,EAAE,cAAc,EAAE,CAAC;KACzB,CAAC;CACH;AAED,MAAM,WAAW,aAAc,SAAQ,eAAe;IACpD,IAAI,EAAE,OAAO,CAAC;IACd,OAAO,EAAE;QACP,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,OAAO,CAAC;QACnB,eAAe,CAAC,EAAE,aAAa,EAAE,CAAC;KACnC,CAAC;CACH;AAED,MAAM,WAAW,YAAa,SAAQ,eAAe;IACnD,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE;QACP,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE,IAAI,GAAG,OAAO,GAAG,OAAO,CAAC;KAClC,CAAC;CACH;AAED,MAAM,MAAM,WAAW,GACnB,YAAY,GACZ,gBAAgB,GAChB,kBAAkB,GAClB,aAAa,GACb,YAAY,CAAC;AAEjB,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,aAAa,CAAC;IACvB,IAAI,CAAC,EAAE,QAAQ,GAAG,aAAa,CAAC;CACjC;AAED,MAAM,MAAM,UAAU,GAClB,eAAe,GACf,gBAAgB,GAChB,SAAS,GACT,aAAa,GACb,aAAa,GACb,aAAa,GACb,WAAW,GACX,aAAa,GACb,cAAc,GACd,iBAAiB,GACjB,aAAa,CAAC;AAElB,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,UAAU,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC9B,OAAO,EAAE,MAAM,EAAE,CAAC;CACnB"}
package/package.json CHANGED
@@ -1,14 +1,24 @@
1
1
  {
2
2
  "name": "@unif/react-native-chat-sdk",
3
- "version": "0.1.1",
3
+ "version": "0.2.0",
4
4
  "description": "UNIF React Native Chat SDK — 流式通信、Provider、状态管理",
5
- "main": "./src/index.ts",
6
- "types": "./src/index.ts",
5
+ "main": "./lib/commonjs/index.js",
6
+ "module": "./lib/module/index.js",
7
+ "types": "./lib/typescript/commonjs/index.d.ts",
8
+ "react-native": "./src/index.ts",
9
+ "source": "./src/index.ts",
7
10
  "exports": {
8
11
  ".": {
9
12
  "source": "./src/index.ts",
10
- "types": "./src/index.ts",
11
- "default": "./src/index.ts"
13
+ "react-native": "./src/index.ts",
14
+ "import": {
15
+ "types": "./lib/typescript/module/index.d.ts",
16
+ "default": "./lib/module/index.js"
17
+ },
18
+ "require": {
19
+ "types": "./lib/typescript/commonjs/index.d.ts",
20
+ "default": "./lib/commonjs/index.js"
21
+ }
12
22
  },
13
23
  "./package.json": "./package.json"
14
24
  },
@@ -22,14 +32,25 @@
22
32
  "scripts": {
23
33
  "build": "bob build",
24
34
  "clean": "del-cli lib",
35
+ "prepublishOnly": "bob build",
25
36
  "typecheck": "tsc"
26
37
  },
27
- "keywords": ["react-native", "chat", "sdk", "sse", "streaming"],
38
+ "keywords": [
39
+ "react-native",
40
+ "chat",
41
+ "sdk",
42
+ "sse",
43
+ "streaming"
44
+ ],
28
45
  "author": "qq382724935 <liulijun@pec.com.cn>",
29
46
  "license": "MIT",
30
47
  "publishConfig": {
31
48
  "registry": "https://registry.npmjs.org/"
32
49
  },
50
+ "devDependencies": {
51
+ "react-native-sse": "^1.2.1",
52
+ "typescript": "^5.9.2"
53
+ },
33
54
  "dependencies": {
34
55
  "zustand": "^4.5.0"
35
56
  },
@@ -42,8 +63,19 @@
42
63
  "source": "src",
43
64
  "output": "lib",
44
65
  "targets": [
45
- ["module", { "esm": true }],
46
- ["typescript", { "project": "tsconfig.build.json" }]
66
+ "commonjs",
67
+ [
68
+ "module",
69
+ {
70
+ "esm": true
71
+ }
72
+ ],
73
+ [
74
+ "typescript",
75
+ {
76
+ "project": "tsconfig.build.json"
77
+ }
78
+ ]
47
79
  ]
48
80
  }
49
81
  }
@@ -39,10 +39,23 @@ interface ChatContextValue<Message = ChatMessage> {
39
39
  suggestions: SuggestionItem[];
40
40
  error: string | null;
41
41
  onRequest: (input: unknown) => void;
42
+ onReload: (id: string, input: unknown) => void;
42
43
  abort: () => void;
43
44
  resetChat: () => void;
44
45
  loadSession: (messages: ChatMessage[]) => void;
45
46
  clearError: () => void;
47
+ setMessages: (
48
+ messages:
49
+ | MessageInfo<Message>[]
50
+ | ((prev: MessageInfo<Message>[]) => MessageInfo<Message>[]),
51
+ ) => void;
52
+ setMessage: (
53
+ id: string,
54
+ partial:
55
+ | Partial<MessageInfo<Message>>
56
+ | ((msg: MessageInfo<Message>) => Partial<MessageInfo<Message>>),
57
+ ) => void;
58
+ removeMessage: (id: string) => void;
46
59
  // useXConversations
47
60
  sessions: SessionSummary[];
48
61
  activeId: string;
@@ -107,7 +120,7 @@ export function ChatProvider<
107
120
 
108
121
  const model = useXModel({ models, defaultModel });
109
122
 
110
- const value: ChatContextValue<Message> = useMemo(
123
+ const value = useMemo(
111
124
  () => ({
112
125
  ...chat,
113
126
  ...conversations,
@@ -38,15 +38,25 @@ export interface UseXChatReturn<
38
38
  suggestions: SuggestionItem[];
39
39
  error: string | null;
40
40
  onRequest: (input: Input) => void;
41
+ onReload: (id: string, input: Input) => void;
41
42
  abort: () => void;
42
43
  resetChat: () => void;
43
44
  loadSession: (messages: ChatMessage[]) => void;
44
45
  clearError: () => void;
46
+ setMessages: (
47
+ messages:
48
+ | MessageInfo<Message>[]
49
+ | ((prev: MessageInfo<Message>[]) => MessageInfo<Message>[]),
50
+ ) => void;
51
+ setMessage: (
52
+ id: string,
53
+ partial:
54
+ | Partial<MessageInfo<Message>>
55
+ | ((msg: MessageInfo<Message>) => Partial<MessageInfo<Message>>),
56
+ ) => void;
57
+ removeMessage: (id: string) => void;
45
58
  }
46
59
 
47
- let _idCounter = 0;
48
- const defaultGenId = () => `xc-${Date.now()}-${++_idCounter}`;
49
-
50
60
  export function useXChat<
51
61
  Message = ChatMessage,
52
62
  Input = RequestParams<Message>,
@@ -54,7 +64,7 @@ export function useXChat<
54
64
  >(
55
65
  config: UseXChatConfig<Message, Input, Output>
56
66
  ): UseXChatReturn<Message, Input> {
57
- const { provider, generateId = defaultGenId } = config;
67
+ const { provider } = config;
58
68
 
59
69
  const storeRef = useRef(createChatStore());
60
70
  const store = storeRef.current;
@@ -131,6 +141,117 @@ export function useXChat<
131
141
  store.getState().setError(null);
132
142
  }, [store]);
133
143
 
144
+ const onReload = useCallback(
145
+ (id: string, input: Input) => {
146
+ const s = store.getState();
147
+ if (s.isRequesting) return;
148
+
149
+ // 将目标消息标记为 loading,移除其后的 assistant 消息
150
+ const targetIdx = s.messages.findIndex((m) => m.id === id);
151
+ if (targetIdx < 0) return;
152
+
153
+ store.getState().setRequesting(true);
154
+ store.getState().setError(null);
155
+ store.getState().setSuggestions([]);
156
+
157
+ provider.sendMessage(input, {
158
+ onUpdate: (message) => {
159
+ store
160
+ .getState()
161
+ .upsertMessage(message as unknown as ChatMessage);
162
+ },
163
+ onSuccess: (message) => {
164
+ const m = message as unknown as ChatMessage;
165
+ store.getState().upsertMessage({
166
+ ...m,
167
+ status: 'success',
168
+ });
169
+ store.getState().setRequesting(false);
170
+ },
171
+ onError: (error) => {
172
+ store.getState().setError(error.message);
173
+ store.getState().setRequesting(false);
174
+
175
+ if (config.requestFallback) {
176
+ const fallbackMsg = config.requestFallback(input, { error });
177
+ store
178
+ .getState()
179
+ .addMessage(fallbackMsg as unknown as ChatMessage);
180
+ }
181
+ },
182
+ onSuggestion: (items) => {
183
+ store.getState().setSuggestions(items);
184
+ },
185
+ });
186
+ },
187
+ [provider, store, config]
188
+ );
189
+
190
+ const setMessages = useCallback(
191
+ (
192
+ messagesOrFn:
193
+ | MessageInfo<Message>[]
194
+ | ((prev: MessageInfo<Message>[]) => MessageInfo<Message>[]),
195
+ ) => {
196
+ if (typeof messagesOrFn === 'function') {
197
+ const current = store.getState().messages;
198
+ const currentInfos: MessageInfo<Message>[] = current.map((m) => ({
199
+ id: m.id,
200
+ message: m as unknown as Message,
201
+ status: m.status,
202
+ extra: m.extra as Record<string, unknown> | undefined,
203
+ }));
204
+ const next = messagesOrFn(currentInfos);
205
+ store
206
+ .getState()
207
+ .setMessages(next.map((info) => info.message as unknown as ChatMessage));
208
+ } else {
209
+ store
210
+ .getState()
211
+ .setMessages(
212
+ messagesOrFn.map((info) => info.message as unknown as ChatMessage),
213
+ );
214
+ }
215
+ },
216
+ [store]
217
+ );
218
+
219
+ const setMessage = useCallback(
220
+ (
221
+ id: string,
222
+ partialOrFn:
223
+ | Partial<MessageInfo<Message>>
224
+ | ((msg: MessageInfo<Message>) => Partial<MessageInfo<Message>>),
225
+ ) => {
226
+ store.getState().setMessage(id, (m) => {
227
+ const info: MessageInfo<Message> = {
228
+ id: m.id,
229
+ message: m as unknown as Message,
230
+ status: m.status,
231
+ extra: m.extra as Record<string, unknown> | undefined,
232
+ };
233
+ const partial =
234
+ typeof partialOrFn === 'function' ? partialOrFn(info) : partialOrFn;
235
+ const updated: Partial<ChatMessage> = {};
236
+ if (partial.message !== undefined) {
237
+ Object.assign(updated, partial.message);
238
+ }
239
+ if (partial.status !== undefined) {
240
+ updated.status = partial.status;
241
+ }
242
+ return updated;
243
+ });
244
+ },
245
+ [store]
246
+ );
247
+
248
+ const removeMessage = useCallback(
249
+ (id: string) => {
250
+ store.getState().removeMessage(id);
251
+ },
252
+ [store]
253
+ );
254
+
134
255
  useEffect(() => {
135
256
  return () => {
136
257
  provider.abort();
@@ -150,9 +271,13 @@ export function useXChat<
150
271
  suggestions: state.suggestions,
151
272
  error: state.error,
152
273
  onRequest,
274
+ onReload,
153
275
  abort,
154
276
  resetChat,
155
277
  loadSession,
156
278
  clearError,
279
+ setMessages,
280
+ setMessage,
281
+ removeMessage,
157
282
  };
158
283
  }
@@ -17,6 +17,18 @@ interface ChatState {
17
17
  addMessage: (message: ChatMessage) => void;
18
18
  updateMessage: (id: string, updater: (msg: ChatMessage) => ChatMessage) => void;
19
19
  upsertMessage: (message: ChatMessage) => void;
20
+ setMessages: (
21
+ messages:
22
+ | ChatMessage[]
23
+ | ((prev: ChatMessage[]) => ChatMessage[]),
24
+ ) => void;
25
+ setMessage: (
26
+ id: string,
27
+ partial:
28
+ | Partial<ChatMessage>
29
+ | ((msg: ChatMessage) => Partial<ChatMessage>),
30
+ ) => void;
31
+ removeMessage: (id: string) => void;
20
32
  setSuggestions: (items: SuggestionItem[]) => void;
21
33
  setRequesting: (v: boolean) => void;
22
34
  setError: (error: string | null) => void;
@@ -55,6 +67,29 @@ export const createChatStore = () =>
55
67
  }
56
68
  },
57
69
 
70
+ setMessages: (messagesOrFn) =>
71
+ set((state) => ({
72
+ messages:
73
+ typeof messagesOrFn === 'function'
74
+ ? messagesOrFn(state.messages)
75
+ : messagesOrFn,
76
+ })),
77
+
78
+ setMessage: (id, partialOrFn) =>
79
+ set((state) => ({
80
+ messages: state.messages.map((m) => {
81
+ if (m.id !== id) return m;
82
+ const partial =
83
+ typeof partialOrFn === 'function' ? partialOrFn(m) : partialOrFn;
84
+ return { ...m, ...partial };
85
+ }),
86
+ })),
87
+
88
+ removeMessage: (id) =>
89
+ set((state) => ({
90
+ messages: state.messages.filter((m) => m.id !== id),
91
+ })),
92
+
58
93
  setSuggestions: (items) => set({ suggestions: items }),
59
94
  setRequesting: (v) => set({ isRequesting: v }),
60
95
  setError: (error) => set({ error }),