@sogni-ai/expo-client 1.0.0-alpha.10

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 (262) hide show
  1. package/CHANGELOG.md +69 -0
  2. package/README.md +411 -0
  3. package/dist/cjs/Account/index.d.ts +60 -0
  4. package/dist/cjs/Account/index.d.ts.map +1 -0
  5. package/dist/cjs/Account/index.js +188 -0
  6. package/dist/cjs/Account/index.js.map +1 -0
  7. package/dist/cjs/Account/types.d.ts +43 -0
  8. package/dist/cjs/Account/types.d.ts.map +1 -0
  9. package/dist/cjs/Account/types.js +18 -0
  10. package/dist/cjs/Account/types.js.map +1 -0
  11. package/dist/cjs/ApiClient/WebSocketClient/events.d.ts +139 -0
  12. package/dist/cjs/ApiClient/WebSocketClient/events.d.ts.map +1 -0
  13. package/dist/cjs/ApiClient/WebSocketClient/events.js +3 -0
  14. package/dist/cjs/ApiClient/WebSocketClient/events.js.map +1 -0
  15. package/dist/cjs/ApiClient/WebSocketClient/index.d.ts +45 -0
  16. package/dist/cjs/ApiClient/WebSocketClient/index.d.ts.map +1 -0
  17. package/dist/cjs/ApiClient/WebSocketClient/index.js +283 -0
  18. package/dist/cjs/ApiClient/WebSocketClient/index.js.map +1 -0
  19. package/dist/cjs/ApiClient/WebSocketClient/normalize.d.ts +45 -0
  20. package/dist/cjs/ApiClient/WebSocketClient/normalize.d.ts.map +1 -0
  21. package/dist/cjs/ApiClient/WebSocketClient/normalize.js +245 -0
  22. package/dist/cjs/ApiClient/WebSocketClient/normalize.js.map +1 -0
  23. package/dist/cjs/ApiClient/WebSocketClient/types.d.ts +9 -0
  24. package/dist/cjs/ApiClient/WebSocketClient/types.d.ts.map +1 -0
  25. package/dist/cjs/ApiClient/WebSocketClient/types.js +3 -0
  26. package/dist/cjs/ApiClient/WebSocketClient/types.js.map +1 -0
  27. package/dist/cjs/ApiClient/WebSocketClient/wireTypes.d.ts +161 -0
  28. package/dist/cjs/ApiClient/WebSocketClient/wireTypes.d.ts.map +1 -0
  29. package/dist/cjs/ApiClient/WebSocketClient/wireTypes.js +11 -0
  30. package/dist/cjs/ApiClient/WebSocketClient/wireTypes.js.map +1 -0
  31. package/dist/cjs/ApiClient/index.d.ts +60 -0
  32. package/dist/cjs/ApiClient/index.d.ts.map +1 -0
  33. package/dist/cjs/ApiClient/index.js +87 -0
  34. package/dist/cjs/ApiClient/index.js.map +1 -0
  35. package/dist/cjs/Chat/ChatStream.d.ts +42 -0
  36. package/dist/cjs/Chat/ChatStream.d.ts.map +1 -0
  37. package/dist/cjs/Chat/ChatStream.js +146 -0
  38. package/dist/cjs/Chat/ChatStream.js.map +1 -0
  39. package/dist/cjs/Chat/index.d.ts +61 -0
  40. package/dist/cjs/Chat/index.d.ts.map +1 -0
  41. package/dist/cjs/Chat/index.js +179 -0
  42. package/dist/cjs/Chat/index.js.map +1 -0
  43. package/dist/cjs/Chat/tools.d.ts +13 -0
  44. package/dist/cjs/Chat/tools.d.ts.map +1 -0
  45. package/dist/cjs/Chat/tools.js +22 -0
  46. package/dist/cjs/Chat/tools.js.map +1 -0
  47. package/dist/cjs/Chat/types.d.ts +153 -0
  48. package/dist/cjs/Chat/types.d.ts.map +1 -0
  49. package/dist/cjs/Chat/types.js +3 -0
  50. package/dist/cjs/Chat/types.js.map +1 -0
  51. package/dist/cjs/Projects/ControlNet.d.ts +30 -0
  52. package/dist/cjs/Projects/ControlNet.d.ts.map +1 -0
  53. package/dist/cjs/Projects/ControlNet.js +45 -0
  54. package/dist/cjs/Projects/ControlNet.js.map +1 -0
  55. package/dist/cjs/Projects/createJobRequestMessage.d.ts +11 -0
  56. package/dist/cjs/Projects/createJobRequestMessage.d.ts.map +1 -0
  57. package/dist/cjs/Projects/createJobRequestMessage.js +239 -0
  58. package/dist/cjs/Projects/createJobRequestMessage.js.map +1 -0
  59. package/dist/cjs/Projects/index.d.ts +96 -0
  60. package/dist/cjs/Projects/index.d.ts.map +1 -0
  61. package/dist/cjs/Projects/index.js +536 -0
  62. package/dist/cjs/Projects/index.js.map +1 -0
  63. package/dist/cjs/Projects/restNormalize.d.ts +60 -0
  64. package/dist/cjs/Projects/restNormalize.d.ts.map +1 -0
  65. package/dist/cjs/Projects/restNormalize.js +89 -0
  66. package/dist/cjs/Projects/restNormalize.js.map +1 -0
  67. package/dist/cjs/Projects/types.d.ts +233 -0
  68. package/dist/cjs/Projects/types.d.ts.map +1 -0
  69. package/dist/cjs/Projects/types.js +3 -0
  70. package/dist/cjs/Projects/types.js.map +1 -0
  71. package/dist/cjs/Projects/utils.d.ts +23 -0
  72. package/dist/cjs/Projects/utils.d.ts.map +1 -0
  73. package/dist/cjs/Projects/utils.js +65 -0
  74. package/dist/cjs/Projects/utils.js.map +1 -0
  75. package/dist/cjs/index.d.ts +63 -0
  76. package/dist/cjs/index.d.ts.map +1 -0
  77. package/dist/cjs/index.js +116 -0
  78. package/dist/cjs/index.js.map +1 -0
  79. package/dist/cjs/lib/AuthManager/ApiKeyAuthManager.d.ts +15 -0
  80. package/dist/cjs/lib/AuthManager/ApiKeyAuthManager.d.ts.map +1 -0
  81. package/dist/cjs/lib/AuthManager/ApiKeyAuthManager.js +47 -0
  82. package/dist/cjs/lib/AuthManager/ApiKeyAuthManager.js.map +1 -0
  83. package/dist/cjs/lib/AuthManager/AuthManagerBase.d.ts +23 -0
  84. package/dist/cjs/lib/AuthManager/AuthManagerBase.d.ts.map +1 -0
  85. package/dist/cjs/lib/AuthManager/AuthManagerBase.js +14 -0
  86. package/dist/cjs/lib/AuthManager/AuthManagerBase.js.map +1 -0
  87. package/dist/cjs/lib/AuthManager/AuthStorage.d.ts +11 -0
  88. package/dist/cjs/lib/AuthManager/AuthStorage.d.ts.map +1 -0
  89. package/dist/cjs/lib/AuthManager/AuthStorage.js +50 -0
  90. package/dist/cjs/lib/AuthManager/AuthStorage.js.map +1 -0
  91. package/dist/cjs/lib/AuthManager/TokenAuthManager.d.ts +35 -0
  92. package/dist/cjs/lib/AuthManager/TokenAuthManager.d.ts.map +1 -0
  93. package/dist/cjs/lib/AuthManager/TokenAuthManager.js +164 -0
  94. package/dist/cjs/lib/AuthManager/TokenAuthManager.js.map +1 -0
  95. package/dist/cjs/lib/AuthManager/index.d.ts +7 -0
  96. package/dist/cjs/lib/AuthManager/index.d.ts.map +1 -0
  97. package/dist/cjs/lib/AuthManager/index.js +13 -0
  98. package/dist/cjs/lib/AuthManager/index.js.map +1 -0
  99. package/dist/cjs/lib/Cache.d.ts +12 -0
  100. package/dist/cjs/lib/Cache.d.ts.map +1 -0
  101. package/dist/cjs/lib/Cache.js +36 -0
  102. package/dist/cjs/lib/Cache.js.map +1 -0
  103. package/dist/cjs/lib/DefaultLogger.d.ts +17 -0
  104. package/dist/cjs/lib/DefaultLogger.d.ts.map +1 -0
  105. package/dist/cjs/lib/DefaultLogger.js +36 -0
  106. package/dist/cjs/lib/DefaultLogger.js.map +1 -0
  107. package/dist/cjs/lib/RestClient.d.ts +18 -0
  108. package/dist/cjs/lib/RestClient.d.ts.map +1 -0
  109. package/dist/cjs/lib/RestClient.js +67 -0
  110. package/dist/cjs/lib/RestClient.js.map +1 -0
  111. package/dist/cjs/lib/TypedEventEmitter.d.ts +16 -0
  112. package/dist/cjs/lib/TypedEventEmitter.d.ts.map +1 -0
  113. package/dist/cjs/lib/TypedEventEmitter.js +46 -0
  114. package/dist/cjs/lib/TypedEventEmitter.js.map +1 -0
  115. package/dist/cjs/lib/base64.d.ts +3 -0
  116. package/dist/cjs/lib/base64.d.ts.map +1 -0
  117. package/dist/cjs/lib/base64.js +21 -0
  118. package/dist/cjs/lib/base64.js.map +1 -0
  119. package/dist/cjs/lib/getUUID.d.ts +12 -0
  120. package/dist/cjs/lib/getUUID.d.ts.map +1 -0
  121. package/dist/cjs/lib/getUUID.js +29 -0
  122. package/dist/cjs/lib/getUUID.js.map +1 -0
  123. package/dist/cjs/lib/mediaInput.d.ts +33 -0
  124. package/dist/cjs/lib/mediaInput.d.ts.map +1 -0
  125. package/dist/cjs/lib/mediaInput.js +32 -0
  126. package/dist/cjs/lib/mediaInput.js.map +1 -0
  127. package/dist/cjs/lib/utils.d.ts +10 -0
  128. package/dist/cjs/lib/utils.d.ts.map +1 -0
  129. package/dist/cjs/lib/utils.js +24 -0
  130. package/dist/cjs/lib/utils.js.map +1 -0
  131. package/dist/cjs/types/ErrorData.d.ts +7 -0
  132. package/dist/cjs/types/ErrorData.d.ts.map +1 -0
  133. package/dist/cjs/types/ErrorData.js +3 -0
  134. package/dist/cjs/types/ErrorData.js.map +1 -0
  135. package/dist/cjs/types/json.d.ts +6 -0
  136. package/dist/cjs/types/json.d.ts.map +1 -0
  137. package/dist/cjs/types/json.js +3 -0
  138. package/dist/cjs/types/json.js.map +1 -0
  139. package/dist/cjs/types/token.d.ts +2 -0
  140. package/dist/cjs/types/token.d.ts.map +1 -0
  141. package/dist/cjs/types/token.js +3 -0
  142. package/dist/cjs/types/token.js.map +1 -0
  143. package/dist/cjs/version.d.ts +2 -0
  144. package/dist/cjs/version.d.ts.map +1 -0
  145. package/dist/cjs/version.js +5 -0
  146. package/dist/cjs/version.js.map +1 -0
  147. package/dist/cjs/wallet.d.ts +59 -0
  148. package/dist/cjs/wallet.d.ts.map +1 -0
  149. package/dist/cjs/wallet.js +125 -0
  150. package/dist/cjs/wallet.js.map +1 -0
  151. package/dist/esm/Account/index.js +183 -0
  152. package/dist/esm/Account/index.js.map +1 -0
  153. package/dist/esm/Account/types.js +15 -0
  154. package/dist/esm/Account/types.js.map +1 -0
  155. package/dist/esm/ApiClient/WebSocketClient/events.js +2 -0
  156. package/dist/esm/ApiClient/WebSocketClient/events.js.map +1 -0
  157. package/dist/esm/ApiClient/WebSocketClient/index.js +278 -0
  158. package/dist/esm/ApiClient/WebSocketClient/index.js.map +1 -0
  159. package/dist/esm/ApiClient/WebSocketClient/normalize.js +229 -0
  160. package/dist/esm/ApiClient/WebSocketClient/normalize.js.map +1 -0
  161. package/dist/esm/ApiClient/WebSocketClient/types.js +2 -0
  162. package/dist/esm/ApiClient/WebSocketClient/types.js.map +1 -0
  163. package/dist/esm/ApiClient/WebSocketClient/wireTypes.js +10 -0
  164. package/dist/esm/ApiClient/WebSocketClient/wireTypes.js.map +1 -0
  165. package/dist/esm/ApiClient/index.js +80 -0
  166. package/dist/esm/ApiClient/index.js.map +1 -0
  167. package/dist/esm/Chat/ChatStream.js +144 -0
  168. package/dist/esm/Chat/ChatStream.js.map +1 -0
  169. package/dist/esm/Chat/index.js +174 -0
  170. package/dist/esm/Chat/index.js.map +1 -0
  171. package/dist/esm/Chat/tools.js +19 -0
  172. package/dist/esm/Chat/tools.js.map +1 -0
  173. package/dist/esm/Chat/types.js +2 -0
  174. package/dist/esm/Chat/types.js.map +1 -0
  175. package/dist/esm/Projects/ControlNet.js +41 -0
  176. package/dist/esm/Projects/ControlNet.js.map +1 -0
  177. package/dist/esm/Projects/createJobRequestMessage.js +236 -0
  178. package/dist/esm/Projects/createJobRequestMessage.js.map +1 -0
  179. package/dist/esm/Projects/index.js +531 -0
  180. package/dist/esm/Projects/index.js.map +1 -0
  181. package/dist/esm/Projects/restNormalize.js +86 -0
  182. package/dist/esm/Projects/restNormalize.js.map +1 -0
  183. package/dist/esm/Projects/types.js +2 -0
  184. package/dist/esm/Projects/types.js.map +1 -0
  185. package/dist/esm/Projects/utils.js +55 -0
  186. package/dist/esm/Projects/utils.js.map +1 -0
  187. package/dist/esm/index.js +72 -0
  188. package/dist/esm/index.js.map +1 -0
  189. package/dist/esm/lib/AuthManager/ApiKeyAuthManager.js +42 -0
  190. package/dist/esm/lib/AuthManager/ApiKeyAuthManager.js.map +1 -0
  191. package/dist/esm/lib/AuthManager/AuthManagerBase.js +9 -0
  192. package/dist/esm/lib/AuthManager/AuthManagerBase.js.map +1 -0
  193. package/dist/esm/lib/AuthManager/AuthStorage.js +13 -0
  194. package/dist/esm/lib/AuthManager/AuthStorage.js.map +1 -0
  195. package/dist/esm/lib/AuthManager/TokenAuthManager.js +159 -0
  196. package/dist/esm/lib/AuthManager/TokenAuthManager.js.map +1 -0
  197. package/dist/esm/lib/AuthManager/index.js +5 -0
  198. package/dist/esm/lib/AuthManager/index.js.map +1 -0
  199. package/dist/esm/lib/Cache.js +33 -0
  200. package/dist/esm/lib/Cache.js.map +1 -0
  201. package/dist/esm/lib/DefaultLogger.js +32 -0
  202. package/dist/esm/lib/DefaultLogger.js.map +1 -0
  203. package/dist/esm/lib/RestClient.js +62 -0
  204. package/dist/esm/lib/RestClient.js.map +1 -0
  205. package/dist/esm/lib/TypedEventEmitter.js +44 -0
  206. package/dist/esm/lib/TypedEventEmitter.js.map +1 -0
  207. package/dist/esm/lib/base64.js +17 -0
  208. package/dist/esm/lib/base64.js.map +1 -0
  209. package/dist/esm/lib/getUUID.js +26 -0
  210. package/dist/esm/lib/getUUID.js.map +1 -0
  211. package/dist/esm/lib/mediaInput.js +28 -0
  212. package/dist/esm/lib/mediaInput.js.map +1 -0
  213. package/dist/esm/lib/utils.js +19 -0
  214. package/dist/esm/lib/utils.js.map +1 -0
  215. package/dist/esm/types/ErrorData.js +2 -0
  216. package/dist/esm/types/ErrorData.js.map +1 -0
  217. package/dist/esm/types/json.js +2 -0
  218. package/dist/esm/types/json.js.map +1 -0
  219. package/dist/esm/types/token.js +2 -0
  220. package/dist/esm/types/token.js.map +1 -0
  221. package/dist/esm/version.js +2 -0
  222. package/dist/esm/version.js.map +1 -0
  223. package/dist/esm/wallet.js +117 -0
  224. package/dist/esm/wallet.js.map +1 -0
  225. package/package.json +92 -0
  226. package/src/Account/index.ts +214 -0
  227. package/src/Account/types.ts +61 -0
  228. package/src/ApiClient/WebSocketClient/events.ts +168 -0
  229. package/src/ApiClient/WebSocketClient/index.ts +345 -0
  230. package/src/ApiClient/WebSocketClient/normalize.ts +287 -0
  231. package/src/ApiClient/WebSocketClient/types.ts +10 -0
  232. package/src/ApiClient/WebSocketClient/wireTypes.ts +183 -0
  233. package/src/ApiClient/index.ts +147 -0
  234. package/src/Chat/ChatStream.ts +164 -0
  235. package/src/Chat/index.ts +224 -0
  236. package/src/Chat/tools.ts +23 -0
  237. package/src/Chat/types.ts +165 -0
  238. package/src/Projects/ControlNet.ts +84 -0
  239. package/src/Projects/createJobRequestMessage.ts +241 -0
  240. package/src/Projects/index.ts +608 -0
  241. package/src/Projects/restNormalize.ts +154 -0
  242. package/src/Projects/types.ts +266 -0
  243. package/src/Projects/utils.ts +60 -0
  244. package/src/index.ts +217 -0
  245. package/src/lib/AuthManager/ApiKeyAuthManager.ts +52 -0
  246. package/src/lib/AuthManager/AuthManagerBase.ts +35 -0
  247. package/src/lib/AuthManager/AuthStorage.ts +21 -0
  248. package/src/lib/AuthManager/TokenAuthManager.ts +179 -0
  249. package/src/lib/AuthManager/index.ts +8 -0
  250. package/src/lib/Cache.ts +45 -0
  251. package/src/lib/DefaultLogger.ts +47 -0
  252. package/src/lib/RestClient.ts +77 -0
  253. package/src/lib/TypedEventEmitter.ts +52 -0
  254. package/src/lib/base64.ts +17 -0
  255. package/src/lib/getUUID.ts +27 -0
  256. package/src/lib/mediaInput.ts +54 -0
  257. package/src/lib/utils.ts +21 -0
  258. package/src/types/ErrorData.ts +7 -0
  259. package/src/types/json.ts +5 -0
  260. package/src/types/token.ts +1 -0
  261. package/src/version.ts +1 -0
  262. package/src/wallet.ts +158 -0
@@ -0,0 +1,229 @@
1
+ /**
2
+ * The ONLY place where wire-level message shapes are translated into clean
3
+ * public types. Every quirk in the sogni-socket protocol gets masked here so
4
+ * the rest of the SDK and downstream apps never see them.
5
+ *
6
+ * Authoritative wire source:
7
+ * ../sogni-socket/controllers/jobsController.js
8
+ * ../sogni-socket/controllers/websocketsController.js
9
+ *
10
+ * Known quirks masked (catalog grows as later phases add events):
11
+ * - `jobID` on the wire is actually the PROJECT id (Phase 3)
12
+ * - `imgID` on the wire is actually the JOB id (Phase 3)
13
+ * - `error_message` is snake_case in an otherwise camelCase API (Phase 3)
14
+ * - `balanceUpdate` arrives in two shapes (v1 flat, v2 nested) — always
15
+ * normalized to v2 here.
16
+ * - `authenticated.tokens` similarly v1/v2 — normalized to v2.
17
+ */
18
+ function isV2Tokens(t) {
19
+ return (typeof t === 'object' && t !== null && 'sogni' in t && 'spark' in t);
20
+ }
21
+ function normalizeBalanceTokens(raw) {
22
+ if (isV2Tokens(raw)) {
23
+ return {
24
+ sogni: {
25
+ settled: String(raw.sogni.settled ?? '0'),
26
+ credit: String(raw.sogni.credit ?? '0'),
27
+ debit: String(raw.sogni.debit ?? '0'),
28
+ net: String(raw.sogni.net ?? '0')
29
+ },
30
+ spark: {
31
+ settled: String(raw.spark.settled ?? '0'),
32
+ credit: String(raw.spark.credit ?? '0'),
33
+ debit: String(raw.spark.debit ?? '0'),
34
+ net: String(raw.spark.net ?? '0'),
35
+ premiumCredit: String(raw.spark.premiumCredit ?? '0')
36
+ }
37
+ };
38
+ }
39
+ // v1 → synthesize v2 wrapper with empty spark
40
+ const v1 = raw;
41
+ return {
42
+ sogni: {
43
+ settled: String(v1.settled ?? '0'),
44
+ credit: String(v1.credit ?? '0'),
45
+ debit: String(v1.debit ?? '0'),
46
+ net: String(v1.net ?? '0')
47
+ },
48
+ spark: {
49
+ settled: '0',
50
+ credit: '0',
51
+ debit: '0',
52
+ net: '0',
53
+ premiumCredit: '0'
54
+ }
55
+ };
56
+ }
57
+ export function normalizeAuthenticated(raw) {
58
+ return {
59
+ username: raw.username,
60
+ walletAddress: raw.address,
61
+ balance: normalizeBalanceTokens(raw.tokens),
62
+ isMainnet: !!raw.isMainnet,
63
+ accountWasMigrated: !!raw.accountWasMigrated,
64
+ hasUnclaimedAirdrop: !!raw.hasUnclaimedAirdrop,
65
+ firstLoginAfterMigration: !!raw.firstLoginAfterMigration
66
+ };
67
+ }
68
+ export function normalizeBalanceUpdate(raw) {
69
+ return { balance: normalizeBalanceTokens(raw) };
70
+ }
71
+ export function normalizeChangeNetwork(raw) {
72
+ return { network: raw.network };
73
+ }
74
+ export function normalizeToastMessage(raw) {
75
+ return {
76
+ level: raw.type,
77
+ message: raw.message,
78
+ ...(raw.stickyID ? { stickyID: raw.stickyID } : {})
79
+ };
80
+ }
81
+ /* ============================================================================
82
+ * Project / job lifecycle normalizers.
83
+ * - wire `jobID` -> `projectId` (uppercased on both sides)
84
+ * - wire `imgID` -> `jobId`
85
+ * - wire `error_message` -> `errorMessage`
86
+ * - mixed casing on error reasons normalized to one spelling, with the raw
87
+ * code preserved in `rawCode` for forward compat
88
+ * ========================================================================== */
89
+ const upper = (s) => (s ? s.toUpperCase() : null);
90
+ /** Split the composite jobState message into distinct typed events. */
91
+ export function normalizeJobState(raw) {
92
+ const projectId = upper(raw.jobID);
93
+ if (!projectId)
94
+ return null;
95
+ const jobId = upper(raw.imgID);
96
+ switch (raw.type) {
97
+ case 'jobStarted':
98
+ return {
99
+ kind: 'jobStarted',
100
+ event: { projectId, jobId, workerName: raw.workerName ?? null }
101
+ };
102
+ case 'initiatingModel':
103
+ return { kind: 'jobInitiatingModel', event: { projectId, jobId } };
104
+ case 'jobCompleted':
105
+ return { kind: 'jobLifecycleCompleted', event: { projectId, jobId } };
106
+ default:
107
+ return null;
108
+ }
109
+ }
110
+ export function normalizeJobProgress(raw) {
111
+ const projectId = upper(raw.jobID);
112
+ const jobId = upper(raw.imgID);
113
+ if (!projectId || !jobId)
114
+ return null;
115
+ const stepCount = Math.max(0, raw.stepCount ?? 0);
116
+ const step = Math.max(0, raw.step ?? 0);
117
+ const progress = stepCount > 0 ? Math.min(1, step / stepCount) : 0;
118
+ return { projectId, jobId, progress, step, stepCount };
119
+ }
120
+ export function normalizeJobResult(raw) {
121
+ const projectId = upper(raw.jobID);
122
+ const jobId = upper(raw.imgID);
123
+ if (!projectId || !jobId)
124
+ return null;
125
+ return {
126
+ projectId,
127
+ jobId,
128
+ seed: raw.lastSeed,
129
+ performedStepCount: raw.performedStepCount,
130
+ userCanceled: !!raw.userCanceled,
131
+ triggeredNSFWFilter: !!raw.triggeredNSFWFilter
132
+ };
133
+ }
134
+ const ERROR_REASON_MAP = {
135
+ serverRestarting: 'serverRestarting',
136
+ workerDisconnected: 'workerDisconnected',
137
+ jobTimedOut: 'jobTimedOut',
138
+ artistCanceled: 'artistCanceled',
139
+ workerCancelled: 'workerCanceled',
140
+ workerCanceled: 'workerCanceled',
141
+ sensitiveContent: 'sensitiveContent',
142
+ imgUploadFailure: 'imgUploadFailure',
143
+ imgDownloadFailure: 'imgDownloadFailure'
144
+ };
145
+ export function normalizeJobError(raw) {
146
+ const projectId = upper(raw.jobID);
147
+ if (!projectId)
148
+ return null;
149
+ return {
150
+ projectId,
151
+ jobId: upper(raw.imgID),
152
+ reason: ERROR_REASON_MAP[raw.error] ?? 'unknown',
153
+ rawCode: raw.error,
154
+ message: raw.error_message,
155
+ source: raw.isFromWorker ? 'worker' : 'server'
156
+ };
157
+ }
158
+ export function normalizeJobETA(raw) {
159
+ const projectId = upper(raw.jobID);
160
+ const jobId = upper(raw.imgID);
161
+ if (!projectId || !jobId)
162
+ return null;
163
+ return { projectId, jobId, etaSeconds: Number(raw.etaSeconds ?? 0) };
164
+ }
165
+ export function normalizeArtistCancelConfirmation(raw) {
166
+ const projectId = upper(raw.jobID);
167
+ if (!projectId)
168
+ return null;
169
+ return {
170
+ projectId,
171
+ didCancel: !!raw.didCancel,
172
+ ...(raw.error_message ? { errorMessage: raw.error_message } : {})
173
+ };
174
+ }
175
+ /* === LLM (chat) ===
176
+ * Chat jobs have only one ID (the completion's jobID — no project/job hierarchy).
177
+ * We just rename to camelCase. */
178
+ export function normalizeChatTokens(raw) {
179
+ const jobId = upper(raw.jobID);
180
+ if (!jobId)
181
+ return null;
182
+ return {
183
+ jobId,
184
+ content: raw.content ?? '',
185
+ role: raw.role,
186
+ finishReason: raw.finishReason ?? null,
187
+ usage: raw.usage,
188
+ tool_calls: raw.tool_calls
189
+ };
190
+ }
191
+ export function normalizeChatResult(raw) {
192
+ const jobId = upper(raw.jobID);
193
+ if (!jobId)
194
+ return null;
195
+ return {
196
+ jobId,
197
+ usage: raw.usage,
198
+ timeTaken: raw.timeTaken,
199
+ ...(raw.workerName ? { workerName: raw.workerName } : {})
200
+ };
201
+ }
202
+ export function normalizeChatError(raw) {
203
+ const jobId = upper(raw.jobID);
204
+ if (!jobId)
205
+ return null;
206
+ return {
207
+ jobId,
208
+ rawCode: raw.error,
209
+ message: raw.error_message,
210
+ ...(raw.workerName ? { workerName: raw.workerName } : {})
211
+ };
212
+ }
213
+ export function normalizeSwarmLLMModels(raw) {
214
+ const models = {};
215
+ for (const [modelId, value] of Object.entries(raw || {})) {
216
+ if (typeof value === 'number') {
217
+ models[modelId] = { workers: value };
218
+ }
219
+ else if (value && typeof value === 'object') {
220
+ models[modelId] = {
221
+ workers: value.workers ?? 0,
222
+ ...(value.displayName ? { displayName: value.displayName } : {}),
223
+ ...(value.contextLength ? { contextLength: value.contextLength } : {})
224
+ };
225
+ }
226
+ }
227
+ return { models };
228
+ }
229
+ //# sourceMappingURL=normalize.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"normalize.js","sourceRoot":"","sources":["../../../../src/ApiClient/WebSocketClient/normalize.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AA0CH,SAAS,UAAU,CAAC,CAAmB;IACrC,OAAO,CACL,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,IAAI,OAAO,IAAI,CAAC,IAAI,OAAO,IAAK,CAAwB,CAC5F,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,GAAqB;IACnD,IAAI,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACpB,OAAO;YACL,KAAK,EAAE;gBACL,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,GAAG,CAAC;gBACzC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,IAAI,GAAG,CAAC;gBACvC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC;gBACrC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,CAAC;aAClC;YACD,KAAK,EAAE;gBACL,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,GAAG,CAAC;gBACzC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,IAAI,GAAG,CAAC;gBACvC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC;gBACrC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,IAAI,GAAG,CAAC;gBACjC,aAAa,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,IAAI,GAAG,CAAC;aACtD;SACF,CAAC;IACJ,CAAC;IACD,8CAA8C;IAC9C,MAAM,EAAE,GAAG,GAAyB,CAAC;IACrC,OAAO;QACL,KAAK,EAAE;YACL,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC,OAAO,IAAI,GAAG,CAAC;YAClC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,MAAM,IAAI,GAAG,CAAC;YAChC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC,KAAK,IAAI,GAAG,CAAC;YAC9B,GAAG,EAAE,MAAM,CAAC,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC;SAC3B;QACD,KAAK,EAAE;YACL,OAAO,EAAE,GAAG;YACZ,MAAM,EAAE,GAAG;YACX,KAAK,EAAE,GAAG;YACV,GAAG,EAAE,GAAG;YACR,aAAa,EAAE,GAAG;SACnB;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,GAAqB;IAC1D,OAAO;QACL,QAAQ,EAAE,GAAG,CAAC,QAAQ;QACtB,aAAa,EAAE,GAAG,CAAC,OAAO;QAC1B,OAAO,EAAE,sBAAsB,CAAC,GAAG,CAAC,MAAM,CAAC;QAC3C,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS;QAC1B,kBAAkB,EAAE,CAAC,CAAC,GAAG,CAAC,kBAAkB;QAC5C,mBAAmB,EAAE,CAAC,CAAC,GAAG,CAAC,mBAAmB;QAC9C,wBAAwB,EAAE,CAAC,CAAC,GAAG,CAAC,wBAAwB;KACzD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,GAAqB;IAC1D,OAAO,EAAE,OAAO,EAAE,sBAAsB,CAAC,GAAG,CAAC,EAAE,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,GAAqB;IAC1D,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,GAAoB;IACxD,OAAO;QACL,KAAK,EAAE,GAAG,CAAC,IAAI;QACf,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACpD,CAAC;AACJ,CAAC;AAED;;;;;;;gFAOgF;AAEhF,MAAM,KAAK,GAAG,CAAC,CAA4B,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AAO7E,uEAAuE;AACvE,MAAM,UAAU,iBAAiB,CAAC,GAAgB;IAChD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;QACjB,KAAK,YAAY;YACf,OAAO;gBACL,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,IAAI,IAAI,EAAE;aAChE,CAAC;QACJ,KAAK,iBAAiB;YACpB,OAAO,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;QACrE,KAAK,cAAc;YACjB,OAAO,EAAE,IAAI,EAAE,uBAAuB,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;QACxE;YACE,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,GAAmB;IACtD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,SAAS,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACtC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC;IAClD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;IACxC,MAAM,QAAQ,GAAG,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,GAAiB;IAClD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,SAAS,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACtC,OAAO;QACL,SAAS;QACT,KAAK;QACL,IAAI,EAAE,GAAG,CAAC,QAAQ;QAClB,kBAAkB,EAAE,GAAG,CAAC,kBAAkB;QAC1C,YAAY,EAAE,CAAC,CAAC,GAAG,CAAC,YAAY;QAChC,mBAAmB,EAAE,CAAC,CAAC,GAAG,CAAC,mBAAmB;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,gBAAgB,GAAmC;IACvD,gBAAgB,EAAE,kBAAkB;IACpC,kBAAkB,EAAE,oBAAoB;IACxC,WAAW,EAAE,aAAa;IAC1B,cAAc,EAAE,gBAAgB;IAChC,eAAe,EAAE,gBAAgB;IACjC,cAAc,EAAE,gBAAgB;IAChC,gBAAgB,EAAE,kBAAkB;IACpC,gBAAgB,EAAE,kBAAkB;IACpC,kBAAkB,EAAE,oBAAoB;CACzC,CAAC;AAEF,MAAM,UAAU,iBAAiB,CAAC,GAAgB;IAChD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5B,OAAO;QACL,SAAS;QACT,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC;QACvB,MAAM,EAAE,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,SAAS;QAChD,OAAO,EAAE,GAAG,CAAC,KAAK;QAClB,OAAO,EAAE,GAAG,CAAC,aAAa;QAC1B,MAAM,EAAE,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;KAC/C,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,GAAc;IAC5C,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,SAAS,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACtC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC,EAAE,CAAC;AACvE,CAAC;AAED,MAAM,UAAU,iCAAiC,CAC/C,GAAgC;IAEhC,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACnC,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAC5B,OAAO;QACL,SAAS;QACT,SAAS,EAAE,CAAC,CAAC,GAAG,CAAC,SAAS;QAC1B,GAAG,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClE,CAAC;AACJ,CAAC;AAED;;kCAEkC;AAElC,MAAM,UAAU,mBAAmB,CAAC,GAAiB;IACnD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO;QACL,KAAK;QACL,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;QAC1B,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,IAAI;QACtC,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,UAAU,EAAE,GAAG,CAAC,UAAU;KAC3B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,GAAoB;IACtD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO;QACL,KAAK;QACL,KAAK,EAAE,GAAG,CAAC,KAAK;QAChB,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1D,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,GAAmB;IACpD,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO;QACL,KAAK;QACL,OAAO,EAAE,GAAG,CAAC,KAAK;QAClB,OAAO,EAAE,GAAG,CAAC,aAAa;QAC1B,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1D,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,GAAsB;IAC5D,MAAM,MAAM,GAAiC,EAAE,CAAC;IAChD,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,CAAC;QACzD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QACvC,CAAC;aAAM,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,GAAG;gBAChB,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,CAAC;gBAC3B,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,KAAK,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACvE,CAAC;QACJ,CAAC;IACH,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,CAAC;AACpB,CAAC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/ApiClient/WebSocketClient/types.ts"],"names":[],"mappings":""}
@@ -0,0 +1,10 @@
1
+ /**
2
+ * Wire-level message shapes as the sogni-socket server sends them.
3
+ * These are NOT exposed to consumers. They are translated into clean public
4
+ * types by `normalize.ts`.
5
+ *
6
+ * Authoritative source: ../sogni-socket/controllers/jobsController.js and
7
+ * ../sogni-socket/controllers/websocketsController.js.
8
+ */
9
+ export {};
10
+ //# sourceMappingURL=wireTypes.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"wireTypes.js","sourceRoot":"","sources":["../../../../src/ApiClient/WebSocketClient/wireTypes.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG"}
@@ -0,0 +1,80 @@
1
+ import RestClient from '../lib/RestClient';
2
+ import TypedEventEmitter from '../lib/TypedEventEmitter';
3
+ import { ApiKeyAuthManager, TokenAuthManager } from '../lib/AuthManager';
4
+ import WebSocketClient from './WebSocketClient';
5
+ export class ApiError extends Error {
6
+ constructor(status, payload) {
7
+ super(payload.message);
8
+ this.status = status;
9
+ this.payload = payload;
10
+ }
11
+ }
12
+ class ApiClient extends TypedEventEmitter {
13
+ constructor({ baseUrl, socketUrl, appId, appSource, networkType, authType, logger, authStorage, disableSocket = false }) {
14
+ super();
15
+ this.appId = appId;
16
+ this.appSource = appSource?.trim() || undefined;
17
+ this.logger = logger;
18
+ this._disableSocket = disableSocket;
19
+ if (authType === 'apiKey') {
20
+ this._auth = new ApiKeyAuthManager(logger);
21
+ }
22
+ else {
23
+ this._auth = new TokenAuthManager(baseUrl, logger, authStorage);
24
+ }
25
+ this._rest = new RestClient(baseUrl, this._auth, logger);
26
+ this._socketRest = new RestClient(normalizeHttpUrl(socketUrl), this._auth, logger);
27
+ this._socket = new WebSocketClient(socketUrl, this._auth, appId, networkType, logger, this.appSource);
28
+ this._auth.on('updated', (isAuthenticated) => {
29
+ this.emit('authUpdated', isAuthenticated);
30
+ this._handleAuthUpdated(isAuthenticated);
31
+ });
32
+ this._socket.on('connected', (e) => this.emit('connected', e));
33
+ this._socket.on('disconnected', (e) => this.emit('disconnected', e));
34
+ }
35
+ get isAuthenticated() {
36
+ return this._auth.isAuthenticated;
37
+ }
38
+ get auth() {
39
+ return this._auth;
40
+ }
41
+ get rest() {
42
+ return this._rest;
43
+ }
44
+ /** REST client bound to the socket server URL (used for model lists, image upload/download URLs, etc.). */
45
+ get socketRest() {
46
+ return this._socketRest;
47
+ }
48
+ get socket() {
49
+ return this._socket;
50
+ }
51
+ get socketEnabled() {
52
+ return !this._disableSocket;
53
+ }
54
+ _handleAuthUpdated(isAuthenticated) {
55
+ if (this._disableSocket)
56
+ return;
57
+ if (isAuthenticated && !this._socket.isConnected) {
58
+ void this._socket.connect();
59
+ }
60
+ else if (!isAuthenticated && this._socket.isConnected) {
61
+ this._socket.disconnect();
62
+ }
63
+ }
64
+ dispose() {
65
+ this._socket.dispose();
66
+ this._auth.removeAllListeners();
67
+ this.removeAllListeners();
68
+ this._auth.clear();
69
+ }
70
+ }
71
+ function normalizeHttpUrl(url) {
72
+ const u = new URL(url);
73
+ if (u.protocol === 'http:' || u.protocol === 'ws:')
74
+ u.protocol = 'http:';
75
+ else
76
+ u.protocol = 'https:';
77
+ return u.toString();
78
+ }
79
+ export default ApiClient;
80
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/ApiClient/index.ts"],"names":[],"mappings":"AAAA,OAAO,UAAU,MAAM,mBAAmB,CAAC;AAC3C,OAAO,iBAAiB,MAAM,0BAA0B,CAAC;AAGzD,OAAO,EAAE,iBAAiB,EAA4B,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACnG,OAAO,eAAe,MAAM,mBAAmB,CAAC;AAehD,MAAM,OAAO,QAAS,SAAQ,KAAK;IAGjC,YAAY,MAAc,EAAE,OAAyB;QACnD,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;CACF;AAoBD,MAAM,SAAU,SAAQ,iBAAkC;IAUxD,YAAY,EACV,OAAO,EACP,SAAS,EACT,KAAK,EACL,SAAS,EACT,WAAW,EACX,QAAQ,EACR,MAAM,EACN,WAAW,EACX,aAAa,GAAG,KAAK,EACJ;QACjB,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,SAAS,EAAE,IAAI,EAAE,IAAI,SAAS,CAAC;QAChD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,GAAG,IAAI,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,GAAG,IAAI,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACzD,IAAI,CAAC,WAAW,GAAG,IAAI,UAAU,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACnF,IAAI,CAAC,OAAO,GAAG,IAAI,eAAe,CAChC,SAAS,EACT,IAAI,CAAC,KAAK,EACV,KAAK,EACL,WAAW,EACX,MAAM,EACN,IAAI,CAAC,SAAS,CACf,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,eAAe,EAAE,EAAE;YAC3C,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;YAC1C,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/D,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC;IACvE,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC;IACpC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,2GAA2G;IAC3G,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC;IAC9B,CAAC;IAEO,kBAAkB,CAAC,eAAwB;QACjD,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAChC,IAAI,eAAe,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YACjD,KAAK,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAC9B,CAAC;aAAM,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YACxD,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO;QACL,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QACvB,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;QAChC,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC1B,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,GAAW;IACnC,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACvB,IAAI,CAAC,CAAC,QAAQ,KAAK,OAAO,IAAI,CAAC,CAAC,QAAQ,KAAK,KAAK;QAAE,CAAC,CAAC,QAAQ,GAAG,OAAO,CAAC;;QACpE,CAAC,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;AACtB,CAAC;AAED,eAAe,SAAS,CAAC"}
@@ -0,0 +1,144 @@
1
+ /**
2
+ * Async iterable representing a streaming chat completion.
3
+ *
4
+ * Iterate with `for await (const chunk of stream)`. Each chunk has `content`
5
+ * (token text) and may carry a `finishReason` on the final chunk. After the
6
+ * stream completes, `stream.result` resolves to the full ChatCompletionResult.
7
+ *
8
+ * Cancel mid-stream with `stream.cancel()` (sends a server-side abort).
9
+ */
10
+ class ChatStream {
11
+ constructor(jobId, onCancel) {
12
+ this._queue = [];
13
+ this._pendingResolver = null;
14
+ this._pendingRejecter = null;
15
+ this._content = '';
16
+ this._role = 'assistant';
17
+ this._finishReason = null;
18
+ this._usage = { prompt_tokens: 0, completion_tokens: 0 };
19
+ this._timeTaken = 0;
20
+ this._toolCallAccumulator = new Map();
21
+ this._done = false;
22
+ this._error = null;
23
+ this.jobId = jobId;
24
+ this._onCancel = onCancel;
25
+ this._resultPromise = new Promise((resolve, reject) => {
26
+ this._resultResolve = resolve;
27
+ this._resultReject = reject;
28
+ });
29
+ }
30
+ /** Resolves to the full result after the stream completes. */
31
+ get result() {
32
+ return this._resultPromise;
33
+ }
34
+ async cancel() {
35
+ await this._onCancel();
36
+ }
37
+ [Symbol.asyncIterator]() {
38
+ return {
39
+ next: () => this._next()
40
+ };
41
+ }
42
+ _next() {
43
+ return new Promise((resolve, reject) => {
44
+ if (this._error) {
45
+ reject(this._error);
46
+ return;
47
+ }
48
+ if (this._queue.length > 0) {
49
+ resolve({ value: this._queue.shift(), done: false });
50
+ return;
51
+ }
52
+ if (this._done) {
53
+ resolve({ value: undefined, done: true });
54
+ return;
55
+ }
56
+ this._pendingResolver = resolve;
57
+ this._pendingRejecter = reject;
58
+ });
59
+ }
60
+ _pushChunk(chunk) {
61
+ if (chunk.content)
62
+ this._content += chunk.content;
63
+ if (chunk.role)
64
+ this._role = chunk.role;
65
+ if (chunk.finishReason !== undefined && chunk.finishReason !== null) {
66
+ this._finishReason = chunk.finishReason;
67
+ }
68
+ if (chunk.usage)
69
+ this._usage = chunk.usage;
70
+ if (chunk.tool_calls)
71
+ this._accumulateToolCalls(chunk.tool_calls);
72
+ if (this._pendingResolver) {
73
+ const resolve = this._pendingResolver;
74
+ this._pendingResolver = null;
75
+ this._pendingRejecter = null;
76
+ resolve({ value: chunk, done: false });
77
+ }
78
+ else {
79
+ this._queue.push(chunk);
80
+ }
81
+ }
82
+ _complete(timeTaken, usage, workerName) {
83
+ if (this._done)
84
+ return;
85
+ if (usage)
86
+ this._usage = usage;
87
+ this._timeTaken = timeTaken;
88
+ if (workerName)
89
+ this._workerName = workerName;
90
+ this._done = true;
91
+ const toolCalls = Array.from(this._toolCallAccumulator.values()).filter((tc) => tc.id && tc.function.name);
92
+ const result = {
93
+ jobId: this.jobId,
94
+ content: this._content,
95
+ role: this._role,
96
+ finishReason: this._finishReason,
97
+ usage: this._usage,
98
+ timeTaken: this._timeTaken,
99
+ ...(this._workerName ? { workerName: this._workerName } : {}),
100
+ ...(toolCalls.length ? { tool_calls: toolCalls } : {})
101
+ };
102
+ this._resultResolve(result);
103
+ if (this._pendingResolver) {
104
+ const resolve = this._pendingResolver;
105
+ this._pendingResolver = null;
106
+ this._pendingRejecter = null;
107
+ resolve({ value: undefined, done: true });
108
+ }
109
+ }
110
+ _fail(error) {
111
+ if (this._done)
112
+ return;
113
+ this._done = true;
114
+ const err = new Error(error.message || 'Chat completion failed');
115
+ err.code = error.code;
116
+ err.rawCode = error.originalCode;
117
+ this._error = err;
118
+ this._resultReject(err);
119
+ if (this._pendingRejecter) {
120
+ const reject = this._pendingRejecter;
121
+ this._pendingResolver = null;
122
+ this._pendingRejecter = null;
123
+ reject(err);
124
+ }
125
+ }
126
+ _accumulateToolCalls(deltas) {
127
+ for (const delta of deltas) {
128
+ const existing = this._toolCallAccumulator.get(delta.index) ?? {
129
+ id: delta.id ?? '',
130
+ type: 'function',
131
+ function: { name: '', arguments: '' }
132
+ };
133
+ if (delta.id)
134
+ existing.id = delta.id;
135
+ if (delta.function?.name)
136
+ existing.function.name += delta.function.name;
137
+ if (delta.function?.arguments)
138
+ existing.function.arguments += delta.function.arguments;
139
+ this._toolCallAccumulator.set(delta.index, existing);
140
+ }
141
+ }
142
+ }
143
+ export default ChatStream;
144
+ //# sourceMappingURL=ChatStream.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ChatStream.js","sourceRoot":"","sources":["../../../src/Chat/ChatStream.ts"],"names":[],"mappings":"AAYA;;;;;;;;GAQG;AACH,MAAM,UAAU;IAmBd,YAAY,KAAa,EAAE,QAA6B;QAjBhD,WAAM,GAA0B,EAAE,CAAC;QACnC,qBAAgB,GAAoB,IAAI,CAAC;QACzC,qBAAgB,GAAoB,IAAI,CAAC;QACzC,aAAQ,GAAG,EAAE,CAAC;QACd,UAAK,GAAW,WAAW,CAAC;QAC5B,kBAAa,GAAkB,IAAI,CAAC;QACpC,WAAM,GAAe,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE,CAAC;QAChE,eAAU,GAAG,CAAC,CAAC;QAEf,yBAAoB,GAAG,IAAI,GAAG,EAAoB,CAAC;QACnD,UAAK,GAAG,KAAK,CAAC;QACd,WAAM,GAAiB,IAAI,CAAC;QAOlC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,cAAc,GAAG,IAAI,OAAO,CAAuB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1E,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;YAC9B,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,8DAA8D;IAC9D,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,MAAM;QACV,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC;IAED,CAAC,MAAM,CAAC,aAAa,CAAC;QACpB,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE;SACzB,CAAC;IACJ,CAAC;IAEO,KAAK;QACX,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACpB,OAAO;YACT,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,OAAO,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAG,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;gBACtD,OAAO;YACT,CAAC;YACD,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,EAAE,KAAK,EAAE,SAA2C,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC5E,OAAO;YACT,CAAC;YACD,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC;YAChC,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,KAA0B;QACnC,IAAI,KAAK,CAAC,OAAO;YAAE,IAAI,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC;QAClD,IAAI,KAAK,CAAC,IAAI;YAAE,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC;QACxC,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS,IAAI,KAAK,CAAC,YAAY,KAAK,IAAI,EAAE,CAAC;YACpE,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;QAC1C,CAAC;QACD,IAAI,KAAK,CAAC,KAAK;YAAE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC;QAC3C,IAAI,KAAK,CAAC,UAAU;YAAE,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAElE,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC;YACtC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC7B,OAAO,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,SAAS,CAAC,SAAiB,EAAE,KAAkB,EAAE,UAAmB;QAClE,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO;QACvB,IAAI,KAAK;YAAE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QAC/B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,UAAU;YAAE,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAElB,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACrE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAClC,CAAC;QACF,MAAM,MAAM,GAAyB;YACnC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO,EAAE,IAAI,CAAC,QAAQ;YACtB,IAAI,EAAE,IAAI,CAAC,KAAK;YAChB,YAAY,EAAE,IAAI,CAAC,aAAa;YAChC,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,SAAS,EAAE,IAAI,CAAC,UAAU;YAC1B,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACvD,CAAC;QACF,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAE5B,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC;YACtC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC7B,OAAO,CAAC,EAAE,KAAK,EAAE,SAA2C,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAgB;QACpB,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO;QACvB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,KAAK,CAAC,OAAO,IAAI,wBAAwB,CAAC,CAAC;QAChE,GAAmD,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACtE,GAAmD,CAAC,OAAO,GAAG,KAAK,CAAC,YAAY,CAAC;QAClF,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC;QAClB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;QACxB,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC;YACrC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC7B,MAAM,CAAC,GAAG,CAAC,CAAC;QACd,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,MAAuB;QAClD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI;gBAC7D,EAAE,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE;gBAClB,IAAI,EAAE,UAAmB;gBACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;aACtC,CAAC;YACF,IAAI,KAAK,CAAC,EAAE;gBAAE,QAAQ,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;YACrC,IAAI,KAAK,CAAC,QAAQ,EAAE,IAAI;gBAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC;YACxE,IAAI,KAAK,CAAC,QAAQ,EAAE,SAAS;gBAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;YACvF,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;CACF;AAED,eAAe,UAAU,CAAC"}