@llumiverse/drivers 1.3.0 → 1.4.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 (409) hide show
  1. package/lib/{types/adobe → adobe}/firefly.d.ts +2 -1
  2. package/lib/adobe/firefly.d.ts.map +1 -0
  3. package/lib/adobe/firefly.js +120 -0
  4. package/lib/adobe/firefly.js.map +1 -0
  5. package/lib/{types/anthropic → anthropic}/index.d.ts +4 -2
  6. package/lib/anthropic/index.d.ts.map +1 -0
  7. package/lib/{esm/anthropic → anthropic}/index.js +19 -8
  8. package/lib/anthropic/index.js.map +1 -0
  9. package/lib/{types/azure → azure}/azure_foundry.d.ts +9 -7
  10. package/lib/azure/azure_foundry.d.ts.map +1 -0
  11. package/lib/{esm/azure → azure}/azure_foundry.js +138 -124
  12. package/lib/azure/azure_foundry.js.map +1 -0
  13. package/lib/{types/bedrock → bedrock}/converse.d.ts +2 -2
  14. package/lib/bedrock/converse.d.ts.map +1 -0
  15. package/lib/{esm/bedrock → bedrock}/converse.js +86 -79
  16. package/lib/bedrock/converse.js.map +1 -0
  17. package/lib/bedrock/embeddings.d.ts +4 -0
  18. package/lib/bedrock/embeddings.d.ts.map +1 -0
  19. package/lib/bedrock/embeddings.js +359 -0
  20. package/lib/bedrock/embeddings.js.map +1 -0
  21. package/lib/{types/bedrock → bedrock}/index.d.ts +33 -9
  22. package/lib/bedrock/index.d.ts.map +1 -0
  23. package/lib/{esm/bedrock → bedrock}/index.js +599 -324
  24. package/lib/bedrock/index.js.map +1 -0
  25. package/lib/{types/bedrock → bedrock}/nova-image-payload.d.ts +5 -5
  26. package/lib/bedrock/nova-image-payload.d.ts.map +1 -0
  27. package/lib/{esm/bedrock → bedrock}/nova-image-payload.js +31 -36
  28. package/lib/bedrock/nova-image-payload.js.map +1 -0
  29. package/lib/{types/bedrock → bedrock}/payloads.d.ts +1 -1
  30. package/lib/bedrock/payloads.d.ts.map +1 -0
  31. package/lib/{cjs/bedrock → bedrock}/payloads.js.map +1 -1
  32. package/lib/{types/bedrock → bedrock}/s3.d.ts +3 -2
  33. package/lib/bedrock/s3.d.ts.map +1 -0
  34. package/lib/{esm/bedrock → bedrock}/s3.js +27 -23
  35. package/lib/bedrock/s3.js.map +1 -0
  36. package/lib/{types/bedrock → bedrock}/twelvelabs.d.ts +8 -9
  37. package/lib/bedrock/twelvelabs.d.ts.map +1 -0
  38. package/lib/{esm/bedrock → bedrock}/twelvelabs.js +17 -21
  39. package/lib/bedrock/twelvelabs.js.map +1 -0
  40. package/lib/{types/groq → groq}/index.d.ts +6 -4
  41. package/lib/groq/index.d.ts.map +1 -0
  42. package/lib/{esm/groq → groq}/index.js +72 -39
  43. package/lib/groq/index.js.map +1 -0
  44. package/lib/{types/huggingface_ie.d.ts → huggingface_ie.d.ts} +4 -3
  45. package/lib/huggingface_ie.d.ts.map +1 -0
  46. package/lib/{esm/huggingface_ie.js → huggingface_ie.js} +32 -31
  47. package/lib/huggingface_ie.js.map +1 -0
  48. package/lib/index.d.ts +16 -0
  49. package/lib/index.d.ts.map +1 -0
  50. package/lib/index.js +16 -0
  51. package/lib/index.js.map +1 -0
  52. package/lib/{types/mistral → mistral}/index.d.ts +6 -5
  53. package/lib/mistral/index.d.ts.map +1 -0
  54. package/lib/{esm/mistral → mistral}/index.js +66 -38
  55. package/lib/mistral/index.js.map +1 -0
  56. package/lib/{types/mistral → mistral}/types.d.ts +1 -0
  57. package/lib/mistral/types.d.ts.map +1 -0
  58. package/lib/mistral/types.js.map +1 -0
  59. package/lib/{types/openai → openai}/azure_openai.d.ts +5 -4
  60. package/lib/openai/azure_openai.d.ts.map +1 -0
  61. package/lib/{esm/openai → openai}/azure_openai.js +17 -14
  62. package/lib/openai/azure_openai.js.map +1 -0
  63. package/lib/openai/drivers.d.ts +4 -0
  64. package/lib/openai/drivers.d.ts.map +1 -0
  65. package/lib/openai/drivers.js +9 -0
  66. package/lib/openai/drivers.js.map +1 -0
  67. package/lib/{types/openai → openai}/index.d.ts +16 -5
  68. package/lib/openai/index.d.ts.map +1 -0
  69. package/lib/{esm/openai → openai}/index.js +242 -141
  70. package/lib/openai/index.js.map +1 -0
  71. package/lib/{types/openai → openai}/openai.d.ts +3 -3
  72. package/lib/openai/openai.d.ts.map +1 -0
  73. package/lib/openai/openai.js +15 -0
  74. package/lib/openai/openai.js.map +1 -0
  75. package/lib/{types/openai → openai}/openai_compatible.d.ts +3 -3
  76. package/lib/openai/openai_compatible.d.ts.map +1 -0
  77. package/lib/{esm/openai → openai}/openai_compatible.js +14 -11
  78. package/lib/openai/openai_compatible.js.map +1 -0
  79. package/lib/{types/openai → openai}/openai_format.d.ts +3 -2
  80. package/lib/openai/openai_format.d.ts.map +1 -0
  81. package/lib/{esm/openai → openai}/openai_format.js +66 -24
  82. package/lib/openai/openai_format.js.map +1 -0
  83. package/lib/{types/replicate.d.ts → replicate.d.ts} +3 -2
  84. package/lib/replicate.d.ts.map +1 -0
  85. package/lib/{esm/replicate.js → replicate.js} +59 -49
  86. package/lib/replicate.js.map +1 -0
  87. package/lib/{types/shared → shared}/claude-messages.d.ts +25 -1
  88. package/lib/shared/claude-messages.d.ts.map +1 -0
  89. package/lib/{esm/shared → shared}/claude-messages.js +212 -59
  90. package/lib/shared/claude-messages.js.map +1 -0
  91. package/lib/{types/shared → shared}/claude-thinking.d.ts +1 -1
  92. package/lib/shared/claude-thinking.d.ts.map +1 -0
  93. package/lib/{esm/shared → shared}/claude-thinking.js +5 -7
  94. package/lib/shared/claude-thinking.js.map +1 -0
  95. package/lib/shared/debug-prompt.d.ts +5 -0
  96. package/lib/shared/debug-prompt.d.ts.map +1 -0
  97. package/lib/shared/debug-prompt.js +25 -0
  98. package/lib/shared/debug-prompt.js.map +1 -0
  99. package/lib/{types/test-driver → test-driver}/TestErrorCompletionStream.d.ts +1 -1
  100. package/lib/test-driver/TestErrorCompletionStream.d.ts.map +1 -0
  101. package/lib/test-driver/TestErrorCompletionStream.js +16 -0
  102. package/lib/test-driver/TestErrorCompletionStream.js.map +1 -0
  103. package/lib/{types/test-driver → test-driver}/TestValidationErrorCompletionStream.d.ts +1 -1
  104. package/lib/test-driver/TestValidationErrorCompletionStream.d.ts.map +1 -0
  105. package/lib/{esm/test-driver → test-driver}/TestValidationErrorCompletionStream.js +4 -4
  106. package/lib/test-driver/TestValidationErrorCompletionStream.js.map +1 -0
  107. package/lib/{types/test-driver → test-driver}/index.d.ts +3 -3
  108. package/lib/test-driver/index.d.ts.map +1 -0
  109. package/lib/{esm/test-driver → test-driver}/index.js +20 -20
  110. package/lib/test-driver/index.js.map +1 -0
  111. package/lib/{types/test-driver → test-driver}/utils.d.ts +1 -1
  112. package/lib/test-driver/utils.d.ts.map +1 -0
  113. package/lib/{esm/test-driver → test-driver}/utils.js +4 -4
  114. package/lib/test-driver/utils.js.map +1 -0
  115. package/lib/togetherai/index.d.ts +15 -0
  116. package/lib/togetherai/index.d.ts.map +1 -0
  117. package/lib/togetherai/index.js +54 -0
  118. package/lib/togetherai/index.js.map +1 -0
  119. package/lib/{types/togetherai → togetherai}/interfaces.d.ts +1 -1
  120. package/lib/togetherai/interfaces.d.ts.map +1 -0
  121. package/lib/{cjs/watsonx → togetherai}/interfaces.js.map +1 -1
  122. package/lib/vertexai/debug.d.ts +2 -0
  123. package/lib/vertexai/debug.d.ts.map +1 -0
  124. package/lib/{esm/vertexai → vertexai}/debug.js +1 -1
  125. package/lib/vertexai/debug.js.map +1 -0
  126. package/lib/vertexai/embeddings/embed-legacy-multimodal.d.ts +9 -0
  127. package/lib/vertexai/embeddings/embed-legacy-multimodal.d.ts.map +1 -0
  128. package/lib/vertexai/embeddings/embed-legacy-multimodal.js +141 -0
  129. package/lib/vertexai/embeddings/embed-legacy-multimodal.js.map +1 -0
  130. package/lib/vertexai/embeddings/embed.d.ts +13 -0
  131. package/lib/vertexai/embeddings/embed.d.ts.map +1 -0
  132. package/lib/vertexai/embeddings/embed.js +162 -0
  133. package/lib/vertexai/embeddings/embed.js.map +1 -0
  134. package/lib/vertexai/embeddings/source-utils.d.ts +9 -0
  135. package/lib/vertexai/embeddings/source-utils.d.ts.map +1 -0
  136. package/lib/vertexai/embeddings/source-utils.js +18 -0
  137. package/lib/vertexai/embeddings/source-utils.js.map +1 -0
  138. package/lib/{types/vertexai → vertexai}/index.d.ts +31 -19
  139. package/lib/vertexai/index.d.ts.map +1 -0
  140. package/lib/{esm/vertexai → vertexai}/index.js +221 -159
  141. package/lib/vertexai/index.js.map +1 -0
  142. package/lib/{types/vertexai → vertexai}/models/claude.d.ts +4 -4
  143. package/lib/vertexai/models/claude.d.ts.map +1 -0
  144. package/lib/{esm/vertexai → vertexai}/models/claude.js +17 -20
  145. package/lib/vertexai/models/claude.js.map +1 -0
  146. package/lib/{types/vertexai → vertexai}/models/gemini.d.ts +28 -4
  147. package/lib/vertexai/models/gemini.d.ts.map +1 -0
  148. package/lib/{esm/vertexai → vertexai}/models/gemini.js +238 -116
  149. package/lib/vertexai/models/gemini.js.map +1 -0
  150. package/lib/{types/vertexai → vertexai}/models/imagen.d.ts +12 -11
  151. package/lib/vertexai/models/imagen.d.ts.map +1 -0
  152. package/lib/{esm/vertexai → vertexai}/models/imagen.js +74 -47
  153. package/lib/vertexai/models/imagen.js.map +1 -0
  154. package/lib/{types/vertexai → vertexai}/models/llama.d.ts +4 -4
  155. package/lib/vertexai/models/llama.d.ts.map +1 -0
  156. package/lib/{esm/vertexai → vertexai}/models/llama.js +37 -33
  157. package/lib/vertexai/models/llama.js.map +1 -0
  158. package/lib/vertexai/models.d.ts +21 -0
  159. package/lib/vertexai/models.d.ts.map +1 -0
  160. package/lib/{esm/vertexai → vertexai}/models.js +12 -9
  161. package/lib/vertexai/models.js.map +1 -0
  162. package/lib/{types/watsonx → watsonx}/index.d.ts +4 -3
  163. package/lib/watsonx/index.d.ts.map +1 -0
  164. package/lib/{esm/watsonx → watsonx}/index.js +69 -41
  165. package/lib/watsonx/index.js.map +1 -0
  166. package/lib/watsonx/interfaces.d.ts.map +1 -0
  167. package/lib/{esm/watsonx → watsonx}/interfaces.js.map +1 -1
  168. package/lib/{types/xai → xai}/index.d.ts +4 -4
  169. package/lib/xai/index.d.ts.map +1 -0
  170. package/lib/{esm/xai → xai}/index.js +22 -18
  171. package/lib/xai/index.js.map +1 -0
  172. package/package.json +87 -36
  173. package/src/adobe/firefly.ts +60 -43
  174. package/src/anthropic/index.ts +37 -20
  175. package/src/azure/azure_foundry.ts +210 -165
  176. package/src/bedrock/converse.ts +124 -101
  177. package/src/bedrock/embeddings.test.ts +508 -0
  178. package/src/bedrock/embeddings.ts +517 -0
  179. package/src/bedrock/error-handling.test.ts +62 -14
  180. package/src/bedrock/index.ts +832 -466
  181. package/src/bedrock/nova-image-payload.ts +98 -91
  182. package/src/bedrock/payloads.ts +9 -9
  183. package/src/bedrock/s3.test.ts +39 -0
  184. package/src/bedrock/s3.ts +57 -38
  185. package/src/bedrock/streaming-tool-use.test.ts +86 -69
  186. package/src/bedrock/twelvelabs.ts +35 -32
  187. package/src/groq/index.ts +143 -60
  188. package/src/http-timeout-wiring.test.ts +263 -0
  189. package/src/huggingface_ie.ts +45 -61
  190. package/src/index.ts +15 -16
  191. package/src/mistral/index.ts +97 -48
  192. package/src/mistral/types.ts +13 -12
  193. package/src/openai/azure_openai.ts +35 -27
  194. package/src/openai/drivers.ts +9 -0
  195. package/src/openai/error-handling.test.ts +308 -113
  196. package/src/openai/index.ts +378 -248
  197. package/src/openai/openai.ts +6 -13
  198. package/src/openai/openai_compatible.ts +35 -25
  199. package/src/openai/openai_format.ts +90 -40
  200. package/src/replicate.ts +108 -86
  201. package/src/shared/claude-messages.test.ts +37 -0
  202. package/src/shared/claude-messages.ts +303 -97
  203. package/src/shared/claude-thinking.ts +6 -12
  204. package/src/shared/debug-prompt.ts +28 -0
  205. package/src/test-driver/TestErrorCompletionStream.ts +9 -10
  206. package/src/test-driver/TestValidationErrorCompletionStream.ts +9 -10
  207. package/src/test-driver/index.ts +43 -27
  208. package/src/test-driver/utils.ts +8 -8
  209. package/src/togetherai/index.test.ts +32 -0
  210. package/src/togetherai/index.ts +64 -128
  211. package/src/togetherai/interfaces.ts +5 -5
  212. package/src/vertexai/debug.ts +3 -3
  213. package/src/vertexai/embeddings/embed-legacy-multimodal.ts +211 -0
  214. package/src/vertexai/embeddings/embed.ts +206 -0
  215. package/src/vertexai/embeddings/embeddings.test.ts +497 -0
  216. package/src/vertexai/embeddings/source-utils.test.ts +54 -0
  217. package/src/vertexai/embeddings/source-utils.ts +29 -0
  218. package/src/vertexai/index.ts +331 -244
  219. package/src/vertexai/models/claude-error-handling.test.ts +83 -36
  220. package/src/vertexai/models/claude-streaming-spacing.test.ts +82 -75
  221. package/src/vertexai/models/claude.ts +55 -40
  222. package/src/vertexai/models/gemini-conversation-mutation.test.ts +65 -23
  223. package/src/vertexai/models/gemini-error-handling.test.ts +126 -10
  224. package/src/vertexai/models/gemini-orphaned-tool-result.test.ts +118 -0
  225. package/src/vertexai/models/gemini.ts +349 -172
  226. package/src/vertexai/models/imagen.ts +164 -100
  227. package/src/vertexai/models/llama.ts +70 -48
  228. package/src/vertexai/models.ts +36 -16
  229. package/src/watsonx/index.ts +101 -65
  230. package/src/watsonx/interfaces.ts +10 -15
  231. package/src/xai/index.ts +30 -30
  232. package/lib/cjs/adobe/firefly.js +0 -120
  233. package/lib/cjs/adobe/firefly.js.map +0 -1
  234. package/lib/cjs/anthropic/index.js +0 -64
  235. package/lib/cjs/anthropic/index.js.map +0 -1
  236. package/lib/cjs/azure/azure_foundry.js +0 -432
  237. package/lib/cjs/azure/azure_foundry.js.map +0 -1
  238. package/lib/cjs/bedrock/converse.js +0 -359
  239. package/lib/cjs/bedrock/converse.js.map +0 -1
  240. package/lib/cjs/bedrock/index.js +0 -1508
  241. package/lib/cjs/bedrock/index.js.map +0 -1
  242. package/lib/cjs/bedrock/nova-image-payload.js +0 -207
  243. package/lib/cjs/bedrock/nova-image-payload.js.map +0 -1
  244. package/lib/cjs/bedrock/payloads.js +0 -3
  245. package/lib/cjs/bedrock/s3.js +0 -107
  246. package/lib/cjs/bedrock/s3.js.map +0 -1
  247. package/lib/cjs/bedrock/twelvelabs.js +0 -87
  248. package/lib/cjs/bedrock/twelvelabs.js.map +0 -1
  249. package/lib/cjs/groq/index.js +0 -328
  250. package/lib/cjs/groq/index.js.map +0 -1
  251. package/lib/cjs/huggingface_ie.js +0 -201
  252. package/lib/cjs/huggingface_ie.js.map +0 -1
  253. package/lib/cjs/index.js +0 -32
  254. package/lib/cjs/index.js.map +0 -1
  255. package/lib/cjs/mistral/index.js +0 -176
  256. package/lib/cjs/mistral/index.js.map +0 -1
  257. package/lib/cjs/mistral/types.js +0 -83
  258. package/lib/cjs/mistral/types.js.map +0 -1
  259. package/lib/cjs/openai/azure_openai.js +0 -72
  260. package/lib/cjs/openai/azure_openai.js.map +0 -1
  261. package/lib/cjs/openai/index.js +0 -1135
  262. package/lib/cjs/openai/index.js.map +0 -1
  263. package/lib/cjs/openai/openai.js +0 -21
  264. package/lib/cjs/openai/openai.js.map +0 -1
  265. package/lib/cjs/openai/openai_compatible.js +0 -63
  266. package/lib/cjs/openai/openai_compatible.js.map +0 -1
  267. package/lib/cjs/openai/openai_format.js +0 -131
  268. package/lib/cjs/openai/openai_format.js.map +0 -1
  269. package/lib/cjs/package.json +0 -3
  270. package/lib/cjs/replicate.js +0 -275
  271. package/lib/cjs/replicate.js.map +0 -1
  272. package/lib/cjs/shared/claude-messages.js +0 -737
  273. package/lib/cjs/shared/claude-messages.js.map +0 -1
  274. package/lib/cjs/shared/claude-thinking.js +0 -60
  275. package/lib/cjs/shared/claude-thinking.js.map +0 -1
  276. package/lib/cjs/test-driver/TestErrorCompletionStream.js +0 -20
  277. package/lib/cjs/test-driver/TestErrorCompletionStream.js.map +0 -1
  278. package/lib/cjs/test-driver/TestValidationErrorCompletionStream.js +0 -24
  279. package/lib/cjs/test-driver/TestValidationErrorCompletionStream.js.map +0 -1
  280. package/lib/cjs/test-driver/index.js +0 -109
  281. package/lib/cjs/test-driver/index.js.map +0 -1
  282. package/lib/cjs/test-driver/utils.js +0 -30
  283. package/lib/cjs/test-driver/utils.js.map +0 -1
  284. package/lib/cjs/togetherai/index.js +0 -126
  285. package/lib/cjs/togetherai/index.js.map +0 -1
  286. package/lib/cjs/togetherai/interfaces.js +0 -3
  287. package/lib/cjs/togetherai/interfaces.js.map +0 -1
  288. package/lib/cjs/vertexai/debug.js +0 -12
  289. package/lib/cjs/vertexai/debug.js.map +0 -1
  290. package/lib/cjs/vertexai/embeddings/embeddings-image.js +0 -27
  291. package/lib/cjs/vertexai/embeddings/embeddings-image.js.map +0 -1
  292. package/lib/cjs/vertexai/embeddings/embeddings-text.js +0 -23
  293. package/lib/cjs/vertexai/embeddings/embeddings-text.js.map +0 -1
  294. package/lib/cjs/vertexai/index.js +0 -644
  295. package/lib/cjs/vertexai/index.js.map +0 -1
  296. package/lib/cjs/vertexai/models/claude.js +0 -72
  297. package/lib/cjs/vertexai/models/claude.js.map +0 -1
  298. package/lib/cjs/vertexai/models/gemini.js +0 -864
  299. package/lib/cjs/vertexai/models/gemini.js.map +0 -1
  300. package/lib/cjs/vertexai/models/imagen.js +0 -303
  301. package/lib/cjs/vertexai/models/imagen.js.map +0 -1
  302. package/lib/cjs/vertexai/models/llama.js +0 -183
  303. package/lib/cjs/vertexai/models/llama.js.map +0 -1
  304. package/lib/cjs/vertexai/models.js +0 -35
  305. package/lib/cjs/vertexai/models.js.map +0 -1
  306. package/lib/cjs/watsonx/index.js +0 -161
  307. package/lib/cjs/watsonx/index.js.map +0 -1
  308. package/lib/cjs/watsonx/interfaces.js +0 -3
  309. package/lib/cjs/xai/index.js +0 -65
  310. package/lib/cjs/xai/index.js.map +0 -1
  311. package/lib/esm/adobe/firefly.js +0 -116
  312. package/lib/esm/adobe/firefly.js.map +0 -1
  313. package/lib/esm/anthropic/index.js.map +0 -1
  314. package/lib/esm/azure/azure_foundry.js.map +0 -1
  315. package/lib/esm/bedrock/converse.js.map +0 -1
  316. package/lib/esm/bedrock/index.js.map +0 -1
  317. package/lib/esm/bedrock/nova-image-payload.js.map +0 -1
  318. package/lib/esm/bedrock/payloads.js.map +0 -1
  319. package/lib/esm/bedrock/s3.js.map +0 -1
  320. package/lib/esm/bedrock/twelvelabs.js.map +0 -1
  321. package/lib/esm/groq/index.js.map +0 -1
  322. package/lib/esm/huggingface_ie.js.map +0 -1
  323. package/lib/esm/index.js +0 -16
  324. package/lib/esm/index.js.map +0 -1
  325. package/lib/esm/mistral/index.js.map +0 -1
  326. package/lib/esm/mistral/types.js.map +0 -1
  327. package/lib/esm/openai/azure_openai.js.map +0 -1
  328. package/lib/esm/openai/index.js.map +0 -1
  329. package/lib/esm/openai/openai.js +0 -14
  330. package/lib/esm/openai/openai.js.map +0 -1
  331. package/lib/esm/openai/openai_compatible.js.map +0 -1
  332. package/lib/esm/openai/openai_format.js.map +0 -1
  333. package/lib/esm/replicate.js.map +0 -1
  334. package/lib/esm/shared/claude-messages.js.map +0 -1
  335. package/lib/esm/shared/claude-thinking.js.map +0 -1
  336. package/lib/esm/test-driver/TestErrorCompletionStream.js +0 -16
  337. package/lib/esm/test-driver/TestErrorCompletionStream.js.map +0 -1
  338. package/lib/esm/test-driver/TestValidationErrorCompletionStream.js.map +0 -1
  339. package/lib/esm/test-driver/index.js.map +0 -1
  340. package/lib/esm/test-driver/utils.js.map +0 -1
  341. package/lib/esm/togetherai/index.js +0 -122
  342. package/lib/esm/togetherai/index.js.map +0 -1
  343. package/lib/esm/togetherai/interfaces.js.map +0 -1
  344. package/lib/esm/vertexai/debug.js.map +0 -1
  345. package/lib/esm/vertexai/embeddings/embeddings-image.js +0 -24
  346. package/lib/esm/vertexai/embeddings/embeddings-image.js.map +0 -1
  347. package/lib/esm/vertexai/embeddings/embeddings-text.js +0 -20
  348. package/lib/esm/vertexai/embeddings/embeddings-text.js.map +0 -1
  349. package/lib/esm/vertexai/index.js.map +0 -1
  350. package/lib/esm/vertexai/models/claude.js.map +0 -1
  351. package/lib/esm/vertexai/models/gemini.js.map +0 -1
  352. package/lib/esm/vertexai/models/imagen.js.map +0 -1
  353. package/lib/esm/vertexai/models/llama.js.map +0 -1
  354. package/lib/esm/vertexai/models.js.map +0 -1
  355. package/lib/esm/watsonx/index.js.map +0 -1
  356. package/lib/esm/xai/index.js.map +0 -1
  357. package/lib/types/adobe/firefly.d.ts.map +0 -1
  358. package/lib/types/anthropic/index.d.ts.map +0 -1
  359. package/lib/types/azure/azure_foundry.d.ts.map +0 -1
  360. package/lib/types/bedrock/converse.d.ts.map +0 -1
  361. package/lib/types/bedrock/index.d.ts.map +0 -1
  362. package/lib/types/bedrock/nova-image-payload.d.ts.map +0 -1
  363. package/lib/types/bedrock/payloads.d.ts.map +0 -1
  364. package/lib/types/bedrock/s3.d.ts.map +0 -1
  365. package/lib/types/bedrock/twelvelabs.d.ts.map +0 -1
  366. package/lib/types/groq/index.d.ts.map +0 -1
  367. package/lib/types/huggingface_ie.d.ts.map +0 -1
  368. package/lib/types/index.d.ts +0 -16
  369. package/lib/types/index.d.ts.map +0 -1
  370. package/lib/types/mistral/index.d.ts.map +0 -1
  371. package/lib/types/mistral/types.d.ts.map +0 -1
  372. package/lib/types/openai/azure_openai.d.ts.map +0 -1
  373. package/lib/types/openai/index.d.ts.map +0 -1
  374. package/lib/types/openai/openai.d.ts.map +0 -1
  375. package/lib/types/openai/openai_compatible.d.ts.map +0 -1
  376. package/lib/types/openai/openai_format.d.ts.map +0 -1
  377. package/lib/types/replicate.d.ts.map +0 -1
  378. package/lib/types/shared/claude-messages.d.ts.map +0 -1
  379. package/lib/types/shared/claude-thinking.d.ts.map +0 -1
  380. package/lib/types/test-driver/TestErrorCompletionStream.d.ts.map +0 -1
  381. package/lib/types/test-driver/TestValidationErrorCompletionStream.d.ts.map +0 -1
  382. package/lib/types/test-driver/index.d.ts.map +0 -1
  383. package/lib/types/test-driver/utils.d.ts.map +0 -1
  384. package/lib/types/togetherai/index.d.ts +0 -23
  385. package/lib/types/togetherai/index.d.ts.map +0 -1
  386. package/lib/types/togetherai/interfaces.d.ts.map +0 -1
  387. package/lib/types/vertexai/debug.d.ts +0 -2
  388. package/lib/types/vertexai/debug.d.ts.map +0 -1
  389. package/lib/types/vertexai/embeddings/embeddings-image.d.ts +0 -11
  390. package/lib/types/vertexai/embeddings/embeddings-image.d.ts.map +0 -1
  391. package/lib/types/vertexai/embeddings/embeddings-text.d.ts +0 -10
  392. package/lib/types/vertexai/embeddings/embeddings-text.d.ts.map +0 -1
  393. package/lib/types/vertexai/index.d.ts.map +0 -1
  394. package/lib/types/vertexai/models/claude.d.ts.map +0 -1
  395. package/lib/types/vertexai/models/gemini.d.ts.map +0 -1
  396. package/lib/types/vertexai/models/imagen.d.ts.map +0 -1
  397. package/lib/types/vertexai/models/llama.d.ts.map +0 -1
  398. package/lib/types/vertexai/models.d.ts +0 -20
  399. package/lib/types/vertexai/models.d.ts.map +0 -1
  400. package/lib/types/watsonx/index.d.ts.map +0 -1
  401. package/lib/types/watsonx/interfaces.d.ts.map +0 -1
  402. package/lib/types/xai/index.d.ts.map +0 -1
  403. package/src/vertexai/embeddings/embeddings-image.ts +0 -52
  404. package/src/vertexai/embeddings/embeddings-text.ts +0 -54
  405. /package/lib/{esm/bedrock → bedrock}/payloads.js +0 -0
  406. /package/lib/{esm/mistral → mistral}/types.js +0 -0
  407. /package/lib/{esm/togetherai → togetherai}/interfaces.js +0 -0
  408. /package/lib/{types/watsonx → watsonx}/interfaces.d.ts +0 -0
  409. /package/lib/{esm/watsonx → watsonx}/interfaces.js +0 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../../src/vertexai/models/gemini.ts"],"names":[],"mappings":"AACA,OAAO,EAEH,YAAY,EACZ,yBAAyB,EAKzB,kBAAkB,EAClB,YAAY,EACZ,QAAQ,EAER,eAAe,EAGf,aAAa,GAEhB,MAAM,eAAe,CAAC;AACvB,OAAO,EAOH,mBAAmB,EACnB,qBAAqB,EACrB,yBAAyB,EACzB,uBAAuB,EAEvB,eAAe,EAEf,SAAS,EAET,UAAU,EAEV,kBAAkB,EAElB,iCAAiC,EACjC,+BAA+B,EAG/B,+BAA+B,EAC/B,uBAAuB,GAE1B,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAMtE,SAAS,wBAAwB,CAAC,OAAsB;IACpD,+DAA+D;IAC/D,OAAO,CAAC,CAAC,OAAO,CAAC,aAAa,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;AACvE,CAAC;AAED,MAAM,oBAAoB,GAAoB;IAC1C;QACI,QAAQ,EAAE,YAAY,CAAC,+BAA+B;QACtD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,wBAAwB;QAC/C,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,+BAA+B;QACtD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,yBAAyB;QAChD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,yBAAyB;QAChD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;IACD;QACI,QAAQ,EAAE,YAAY,CAAC,6BAA6B;QACpD,SAAS,EAAE,kBAAkB,CAAC,eAAe;KAChD;CACJ,CAAC;AAEF,SAAS,2BAA2B,CAAC,OAAgB;IACjD,OAAO;QACH,GAAG,OAAO;QACV,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAC/B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC;gBACzB,OAAO,IAAI,CAAC;YAChB,CAAC;YACD,OAAO;gBACH,GAAG,IAAI;gBACP,UAAU,EAAE;oBACR,GAAG,IAAI,CAAC,UAAU;oBAClB,IAAI,EAAE,sBAAsB,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;iBACrD;aACW,CAAC;QACrB,CAAC,CAAC;KACL,CAAC;AACN,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,MAA6B;IACjE,OAAO;QACH,GAAG,MAAM;QACT,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,2BAA2B,CAAC;QAC1D,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,2BAA2B,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;KACjF,CAAC;AACN,CAAC;AAED,kGAAkG;AAClG,SAAS,wBAAwB,CAC7B,eAAsG;IAEtG,QAAQ,eAAe,EAAE,CAAC;QACtB,KAAK,wBAAwB;YACzB,OAAO,eAAe,CAAC,sBAAsB,CAAC;QAClD,KAAK,wBAAwB;YACzB,OAAO,eAAe,CAAC,sBAAsB,CAAC;QAClD,KAAK,8BAA8B;YAC/B,OAAO,eAAe,CAAC,4BAA4B,CAAC;QACxD;YACI,OAAO,SAAS,CAAC;IACzB,CAAC;AACL,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,OAAyB,EAAE,MAA6B;IACrF,MAAM,aAAa,GAAG,OAAO,CAAC,aAAkD,CAAC;IACjF,MAAM,KAAK,GAAG,kBAAkB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAEhD,2FAA2F;IAC3F,6EAA6E;IAC7E,2EAA2E;IAC3E,IAAI,eAAe,GAAG,oBAAoB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC5D,IAAI,CAAC,KAAK,IAAI,eAAe,EAAE,CAAC;QAC5B,MAAM,YAAY,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC;QAC7G,IAAI,YAAY,EAAE,CAAC;YACf,eAAe,GAAG,gCAAgC,CAAC,eAAe,CAAC,CAAC;QACxE,CAAC;IACL,CAAC;IACD,6EAA6E;IAC7E,6EAA6E;IAC7E,IAAI,eAAe,EAAE,CAAC;QAClB,eAAe,GAAG,sBAAsB,CAAC,eAAe,CAAC,CAAC;IAC9D,CAAC;IAED,MAAM,mBAAmB,GAAG,wBAAwB,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;IAExE,MAAM,gBAAgB,GAA0B;QAC5C,iBAAiB,EAAE,MAAM,CAAC,MAAM;QAChC,cAAc,EAAE,oBAAoB;QACpC,kBAAkB,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,2EAA2E;QAChI,cAAc,EAAE,CAAC;QACjB,eAAe;QACf,WAAW,EAAE,aAAa,EAAE,WAAW;QACvC,IAAI,EAAE,aAAa,EAAE,KAAK;QAC1B,eAAe,EAAE,aAAa,EAAE,UAAU;QAC1C,aAAa,EAAE,aAAa,EAAE,aAAa;QAC3C,cAAc,EAAE,oBAAoB,CAAC,OAAO,CAAC;QAC7C,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,WAAW,EAAE;YACT,SAAS,EAAE,aAAa,EAAE,UAAU;YACpC,WAAW,EAAE,aAAa,EAAE,kBAAkB;YAC9C,gBAAgB,EAAE,aAAa,EAAE,iBAAiB;YAClD,eAAe,EAAE,wBAAwB,CAAC,aAAa,EAAE,gBAAgB,CAAC;YAC1E,cAAc,EAAE,aAAa,EAAE,gBAAgB;YAC/C,wBAAwB,EAAE,aAAa,EAAE,0BAA0B;SACtE;KACJ,CAAC;IAEF,MAAM,MAAM,GAA0B;QAClC,iBAAiB,EAAE,MAAM,CAAC,MAAM;QAChC,cAAc,EAAE,oBAAoB;QACpC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS;QAClC,UAAU,EAAE,KAAK;YACb,CAAC,CAAC;gBACI,qBAAqB,EAAE;oBACnB,IAAI,EAAE,yBAAyB,CAAC,IAAI;iBACvC;aACJ;YACH,CAAC,CAAC,SAAS;QACf,cAAc,EAAE,CAAC;QACjB,wBAAwB;QACxB,gBAAgB,EAAE,mBAAmB,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;QACtE,kBAAkB,EAAE,mBAAmB,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS;QAC3E,eAAe;QACf,WAAW,EAAE,aAAa,EAAE,WAAW;QACvC,IAAI,EAAE,aAAa,EAAE,KAAK;QAC1B,IAAI,EAAE,aAAa,EAAE,KAAK;QAC1B,eAAe,EAAE,aAAa,EAAE,UAAU;QAC1C,aAAa,EAAE,aAAa,EAAE,aAAa;QAC3C,eAAe,EAAE,aAAa,EAAE,gBAAgB;QAChD,gBAAgB,EAAE,aAAa,EAAE,iBAAiB;QAClD,IAAI,EAAE,aAAa,EAAE,IAAI;QACzB,cAAc,EAAE,oBAAoB,CAAC,OAAO,CAAC;QAC7C,MAAM,EAAE,OAAO,CAAC,MAAM;KACzB,CAAC;IAEF,OAAO;QACH,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,QAAQ,EAAE,eAAe;QACzB,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,MAAM;KACpF,CAAC;AACN,CAAC;AAED;;;GAGG;AACH,SAAS,wBAAwB,CAAC,OAAgB;IAC9C,MAAM,OAAO,GAAuB,EAAE,CAAC;IACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,IAAI,KAAK,EAAE,CAAC;QACR,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACvB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,IAAI,CAAC,IAAI;iBACnB,CAAC,CAAC;YACP,CAAC;iBAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACzB,MAAM,gBAAgB,GAAW,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,EAAE,CAAC;gBAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,IAAI,WAAW,CAAC;gBACzD,MAAM,QAAQ,GAAG,QAAQ,QAAQ,WAAW,gBAAgB,EAAE,CAAC;gBAC/D,OAAO,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,QAAQ;iBAClB,CAAC,CAAC;YACP,CAAC;QACL,CAAC;IACL,CAAC;IACD,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAgB;IACzC,MAAM,GAAG,GAAc,EAAE,CAAC;IAC1B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,MAAM,OAAO,GAAY;gBACrB,EAAE,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE;gBAChC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE;gBACvC,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,IAAkB;aACnD,CAAC;YACF,mEAAmE;YACnE,sDAAsD;YACtD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACxB,OAAO,CAAC,iBAAiB,GAAG,IAAI,CAAC,gBAAgB,CAAC;YACtD,CAAC;YACD,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC;IACL,CAAC;IACD,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,QAA+B;IAChE,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAElD,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAC9B,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CACnF,CAAC;IACF,8CAA8C;IAC9C,IAAI,CAAC,YAAY,EAAE,CAAC;QAChB,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,IAAI,cAAc,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;IAE/E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,IAAI,cAAc,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC3C,mEAAmE;YACnE,cAAc,CAAC,KAAK,GAAG,CAAC,cAAc,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7F,CAAC;aAAM,CAAC;YACJ,8CAA8C;YAC9C,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC5B,cAAc,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC;QAC/E,CAAC;IACL,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC5B,OAAO,MAAM,CAAC;AAClB,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,sBAAsB,CAAC,QAAmB;IACtD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAC;IAC3C,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC5B,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrB,SAAS;QACb,CAAC;QACD,MAAM,mBAAmB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAChF,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrB,SAAS;QACb,CAAC;QACD,MAAM,IAAI,GAAG,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7B,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;QACvC,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBAC5B,IAAI,IAAI,CAAC,YAAY,EAAE,IAAI;oBAAE,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC1E,CAAC;QACL,CAAC;QACD,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAC3C,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CACpF,CAAC;QACF,mEAAmE;QACnE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QACpC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;IACtG,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,sBAAsB,GAAmB;IAC3C,YAAY,CAAC,UAAU;IACvB,YAAY,CAAC,IAAI;IACjB,YAAY,CAAC,yBAAyB;CACzC,CAAC;AAEF,mFAAmF;AACnF,6EAA6E;AAC7E,4DAA4D;AAC5D,MAAM,0BAA0B,GAAG;IAC/B,sBAAsB,EAAE,kCAAkC;CAC7D,CAAC;AAEF,SAAS,oBAAoB,CAAC,MAAiC;IAC3D,MAAM,aAAa,GAAG,MAAM,CAAC,aAAkD,CAAC;IAChF,gFAAgF;IAChF,IAAI,aAAa,EAAE,sBAAsB,KAAK,SAAS,EAAE,CAAC;QACtD,OAAO,aAAa,CAAC,sBAAsB,CAAC;IAChD,CAAC;IACD,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;QACxB,OAAO,qBAAqB,CAAC,MAAM,CAAC,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;IACrE,CAAC;IACD,yCAAyC;IACzC,kEAAkE;IAClE,IAAI,qBAAqB,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,KAAK,EAAE,CAAC;QAChD,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,GAAG,CAAC;QACf,CAAC;QACD,OAAO,CAAC,CAAC;IACb,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,4BAA4B,CACjC,KAAa,EACb,MAAuC;IAEvC,IAAI,KAAK,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;QACvC,OAAO,aAAa,CAAC,IAAI,CAAC;IAC9B,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;QAC3C,OAAO,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC;IACzE,CAAC;IACD,QAAQ,MAAM,EAAE,CAAC;QACb,KAAK,KAAK;YACN,OAAO,aAAa,CAAC,GAAG,CAAC;QAC7B,KAAK,QAAQ;YACT,OAAO,aAAa,CAAC,MAAM,CAAC;QAChC,KAAK,MAAM;YACP,OAAO,aAAa,CAAC,IAAI,CAAC;QAC9B;YACI,OAAO,SAAS,CAAC;IACzB,CAAC;AACL,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAa,EAAE,MAAoD;IAC9F,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC;IACxD,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEpC,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;QACnB,IAAI,KAAK;YAAE,OAAO,GAAG,CAAC;QACtB,IAAI,WAAW;YAAE,OAAO,GAAG,CAAC;QAC5B,IAAI,OAAO;YAAE,OAAO,CAAC,CAAC;QACtB,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,IAAI,KAAK;QAAE,OAAO,KAAK,CAAC;IACxB,IAAI,OAAO,IAAI,WAAW;QAAE,OAAO,KAAK,CAAC;IACzC,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAiC;IAC3D,MAAM,aAAa,GAAG,MAAM,CAAC,aAAkD,CAAC;IAEhF,6EAA6E;IAC7E,MAAM,gBAAgB,GAAG,aAAa,EAAE,gBAAgB,IAAI,KAAK,CAAC;IAClE,IAAI,aAAa,EAAE,sBAAsB,KAAK,SAAS,IAAI,aAAa,EAAE,cAAc,EAAE,CAAC;QACvF,OAAO;YACH,eAAe,EAAE,gBAAgB;YACjC,cAAc,EAAE,aAAa,CAAC,sBAAsB;YACpD,aAAa,EAAE,aAAa,CAAC,cAAc;SAC9C,CAAC;IACN,CAAC;IACD,IAAI,aAAa,EAAE,MAAM,EAAE,CAAC;QACxB,IAAI,uBAAuB,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;YAC/C,OAAO;gBACH,eAAe,EAAE,gBAAgB;gBACjC,aAAa,EAAE,4BAA4B,CAAC,MAAM,CAAC,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC;aAClF,CAAC;QACN,CAAC;QACD,OAAO;YACH,eAAe,EAAE,gBAAgB;YACjC,cAAc,EAAE,qBAAqB,CAAC,MAAM,CAAC,KAAK,EAAE,aAAa,CAAC,MAAM,CAAC;SAC5E,CAAC;IACN,CAAC;IAED,uCAAuC;IACvC,kEAAkE;IAClE,sEAAsE;IACtE,IAAI,uBAAuB,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;QAC/C,OAAO;YACH,eAAe,EAAE,gBAAgB;YACjC,aAAa,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,CAAC,GAAG;SACtG,CAAC;IACN,CAAC;IACD,IAAI,uBAAuB,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC;QAC/C,MAAM,sBAAsB,GAAG,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACjE,OAAO;YACH,eAAe,EAAE,gBAAgB;YACjC,cAAc,EAAE,sBAAsB;SACzC,CAAC;IACN,CAAC;AACL,CAAC;AAED,MAAM,OAAO,qBAAqB;IAC9B,KAAK,CAAU;IAEf,YAAY,OAAe;QACvB,IAAI,CAAC,KAAK,GAAG;YACT,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,UAAU;YACpB,IAAI,EAAE,SAAS,CAAC,IAAI;YACpB,UAAU,EAAE,IAAI;SACD,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,YAAY,CACd,OAAuB,EACvB,QAAyB,EACzB,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAE3C,MAAM,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;QACrC,IAAI,QAAQ,GAAc,EAAE,CAAC;QAC7B,IAAI,MAAM,GAAwB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,2CAA2C;QAE1G,MAAM,MAAM,GAAc,EAAE,CAAC;QAE7B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YACzB,yBAAyB;YACzB,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;gBACjC,gCAAgC;gBAChC,IAAI,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpC,MAAM,IAAI,KAAK,CACX,6FAA6F,CAChG,CAAC;gBACN,CAAC;gBAED,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBACd,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC;wBACf,IAAI,EAAE,GAAG,CAAC,OAAO;qBACpB,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;gBACtC,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;oBACnB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;gBACzD,CAAC;gBACD,yFAAyF;gBACzF,MAAM,oBAAoB,GAAS;oBAC/B,gBAAgB,EAAE;wBACd,IAAI,EAAE,GAAG,CAAC,WAAW;wBACrB,QAAQ,EAAE,sBAAsB,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;qBACtD;oBACD,wFAAwF;oBACxF,gBAAgB,EAAE,GAAG,CAAC,iBAAiB;iBAC1C,CAAC;gBACF,QAAQ,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,oBAAoB,CAAC;iBAChC,CAAC,CAAC;YACP,CAAC;iBAAM,CAAC;gBACJ,2DAA2D;gBAC3D,MAAM,KAAK,GAAW,EAAE,CAAC;gBACzB,wBAAwB;gBACxB,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBACd,KAAK,CAAC,IAAI,CAAC;wBACP,IAAI,EAAE,GAAG,CAAC,OAAO;qBACpB,CAAC,CAAC;gBACP,CAAC;gBAED,wBAAwB;gBACxB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBACZ,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;wBACxB,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,MAAM,EAAE,CAAC;wBACjC,MAAM,OAAO,GACT,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,iCAAiC,CAAC,CAAC;wBAEzF,IAAI,OAAO,EAAE,CAAC;4BACV,KAAK,CAAC,IAAI,CAAC;gCACP,QAAQ,EAAE;oCACN,OAAO;oCACP,QAAQ,EAAE,CAAC,CAAC,SAAS;iCACxB;6BACJ,CAAC,CAAC;wBACP,CAAC;6BAAM,CAAC;4BACJ,uBAAuB;4BACvB,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,SAAS,EAAE,CAAC;4BACnC,MAAM,IAAI,GAAG,MAAM,kBAAkB,CAAC,MAAM,CAAC,CAAC;4BAC9C,KAAK,CAAC,IAAI,CAAC;gCACP,UAAU,EAAE;oCACR,IAAI;oCACJ,QAAQ,EAAE,CAAC,CAAC,SAAS;iCACxB;6BACJ,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;gBACL,CAAC;gBAED,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACnB,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;wBACjC,MAAM,CAAC,IAAI,CAAC;4BACR,IAAI,EAAE,MAAM;4BACZ,KAAK;yBACR,CAAC,CAAC;oBACP,CAAC;yBAAM,CAAC;wBACJ,QAAQ,CAAC,IAAI,CAAC;4BACV,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;4BAC1D,KAAK;yBACR,CAAC,CAAC;oBACP,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAED,uCAAuC;QACvC,IAAI,MAAM,EAAE,CAAC;YACT,IAAI,wBAAwB,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBACtD,6FAA6F;gBAC7F,qFAAqF;gBACrF,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE,iDAAiD,EAAE,CAAC,CAAC;YACpF,CAAC;iBAAM,CAAC;gBACJ,6FAA6F;gBAC7F,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;oBAChB,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC;wBACf,IAAI,EAAE,8FAA8F,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;qBAC/H,CAAC,CAAC;gBACP,CAAC;qBAAM,CAAC;oBACJ,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC;wBACf,IAAI,EAAE,sEAAsE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;qBACvG,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;QACL,CAAC;QAED,kDAAkD;QAClD,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7C,MAAM,GAAG,SAAS,CAAC;QACvB,CAAC;QAED,qGAAqG;QACrG,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,wIAAwI;QACxI,QAAQ,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAE1C,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;IAChC,CAAC;IAED,yBAAyB,CACrB,MAAsB,EACtB,aAA+D;QAE/D,IAAI,CAAC,aAAa,EAAE,eAAe,EAAE,CAAC;YAClC,OAAO,EAAE,CAAC;QACd,CAAC;QACD,MAAM,UAAU,GAAwB;YACpC,KAAK,EAAE,aAAa,CAAC,eAAe;YACpC,MAAM,EAAE,aAAa,CAAC,gBAAgB;YACtC,aAAa,EAAE,aAAa,CAAC,uBAAuB,IAAI,SAAS;YACjE,UAAU,EAAE,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,uBAAuB,IAAI,CAAC,CAAC;SACnG,CAAC;QAEF,sBAAsB;QACtB,UAAU,CAAC,MAAM;YACb,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC,CAAC;gBACzC,CAAC,aAAa,CAAC,kBAAkB,IAAI,CAAC,CAAC;gBACvC,CAAC,aAAa,CAAC,uBAAuB,IAAI,CAAC,CAAC,CAAC;QAEjD,IAAI,CAAC,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC;YAC3E,mEAAmE;YACnE,mEAAmE;YACnE,kEAAkE;YAClE,+DAA+D;YAC/D,MAAM,CAAC,MAAM,CAAC,IAAI,CACd,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,EACjF,8EAA8E,CACjF,CAAC;QACN,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YACrB,UAAU,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,kCAAkC;QACrE,CAAC;QAED,OAAO,UAAU,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,qBAAqB,CACvB,MAAsB,EACtB,MAA6B,EAC7B,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,MAA0B,CAAC;QAC/B,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAE3C,iEAAiE;QACjE,2FAA2F;QAC3F,yFAAyF;QACzF,0FAA0F;QAC1F,MAAM,cAAc,GAAG,6BAA6B,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC3E,IAAI,cAAc,EAAE,CAAC;YACjB,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;QACnC,CAAC;QAED,IAAI,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7E,MAAM,CAAC,QAAQ,GAAG,YAAY,CAAC;QAE/B,6DAA6D;QAC7D,IAAI,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;YACnD,MAAM,GAAG,QAAQ,CAAC,CAAC,oFAAoF;QAC3G,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,aAAkD,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,CAAC,oBAAoB,CAAC,MAAM,EAAE,aAAa,EAAE,IAAI,IAAI,KAAK,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QAEtG,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAE9D,MAAM,WAAW,GAAwB,IAAI,CAAC,yBAAyB,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;QAExG,IAAI,QAA+B,CAAC;QACpC,IAAI,aAAiC,EAAE,MAAsC,CAAC;QAC9E,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QAC3C,IAAI,SAAS,EAAE,CAAC;YACZ,QAAQ,SAAS,CAAC,YAAY,EAAE,CAAC;gBAC7B,KAAK,YAAY,CAAC,UAAU;oBACxB,aAAa,GAAG,QAAQ,CAAC;oBACzB,MAAM;gBACV,KAAK,YAAY,CAAC,IAAI;oBAClB,aAAa,GAAG,MAAM,CAAC;oBACvB,MAAM;gBACV;oBACI,aAAa,GAAG,SAAS,CAAC,YAAY,CAAC;YAC/C,CAAC;YACD,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC;YAElC,+EAA+E;YAC/E,MAAM,qBAAqB,GAAG,0BAA0B,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAsB,CAAC,CAAC;YACpG,IACI,SAAS,CAAC,YAAY;gBACtB,CAAC,sBAAsB,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAY,CAAC;gBACxD,CAAC,qBAAqB,EACxB,CAAC;gBACC,MAAM,IAAI,KAAK,CACX,8BAA8B,SAAS,CAAC,YAAY,IAAI;oBACpD,mBAAmB,SAAS,CAAC,aAAa,IAAI;oBAC9C,YAAY,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAClH,CAAC;YACN,CAAC;YAED,IAAI,OAAO,EAAE,CAAC;gBACV,QAAQ,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;gBAExC,wEAAwE;gBACxE,6DAA6D;gBAC7D,yEAAyE;gBACzE,sEAAsE;gBACtE,qEAAqE;gBACrE,IAAI,qBAAqB,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC3D,MAAM,CAAC,MAAM,CAAC,IAAI,CACd,yCAAyC,SAAS,CAAC,YAAY,KAAK;wBAChE,2CAA2C,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/F,CAAC;gBACN,CAAC;gBAED,MAAM,GAAG,wBAAwB,CAAC,OAAO,CAAC,CAAC;gBAC3C,YAAY,GAAG,kBAAkB,CAAC,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YAC/D,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACX,aAAa,GAAG,UAAU,CAAC;QAC/B,CAAC;QAED,gDAAgD;QAChD,YAAY,GAAG,yBAAyB,CAAC,YAAY,CAAc,CAAC;QAEpE,uEAAuE;QACvE,MAAM,WAAW,GAAG,mBAAmB,CAAC,YAAY,CAAC,CAAC,UAAU,CAAC;QACjE,MAAM,YAAY,GAAG;YACjB,YAAY,EAAE,OAAO,CAAC,qBAAqB,IAAI,QAAQ;YACvD,WAAW;YACX,aAAa,EAAE,OAAO,CAAC,kBAAkB;SAC5C,CAAC;QACF,IAAI,qBAAqB,GAAG,iCAAiC,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAE1F,4CAA4C;QAC5C,qBAAqB,GAAG,+BAA+B,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;QAE7F,sCAAsC;QACtC,qBAAqB,GAAG,+BAA+B,CAAC,qBAAqB,EAAE;YAC3E,YAAY,EAAE,OAAO,CAAC,yBAAyB,IAAI,CAAC;YACpD,WAAW;SACd,CAAC,CAAC;QAEH,qEAAqE;QACrE,MAAM,iBAAiB,GAAG,yBAAyB,CAAC,qBAAqB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAE1F,OAAO;YACH,MAAM,EAAE,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YACrF,WAAW,EAAE,WAAW;YACxB,aAAa,EAAE,aAAa;YAC5B,iBAAiB,EAAE,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;YAC3E,YAAY,EAAE,iBAAiB;YAC/B,QAAQ;SACU,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,2BAA2B,CAC7B,MAAsB,EACtB,MAA6B,EAC7B,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,IAAI,MAA0B,CAAC;QAC/B,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAClD,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAE3C,iEAAiE;QACjE,2FAA2F;QAC3F,yFAAyF;QACzF,0FAA0F;QAC1F,MAAM,cAAc,GAAG,6BAA6B,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC3E,IAAI,cAAc,EAAE,CAAC;YACjB,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;QACnC,CAAC;QAED,0EAA0E;QAC1E,MAAM,YAAY,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC/E,MAAM,CAAC,QAAQ,GAAG,YAAY,CAAC;QAE/B,IAAI,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;YACnD,MAAM,GAAG,QAAQ,CAAC,CAAC,oFAAoF;QAC3G,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,aAAkD,CAAC;QACjF,MAAM,MAAM,GAAG,MAAM,CAAC,oBAAoB,CAAC,MAAM,EAAE,aAAa,EAAE,IAAI,IAAI,KAAK,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QAEtG,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAClD,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC7C,MAAM,WAAW,GAAwB,IAAI,CAAC,yBAAyB,CAAC,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YACpG,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACtC,IAAI,QAA+B,CAAC;oBACpC,IAAI,aAAiC,CAAC;oBACtC,QAAQ,SAAS,CAAC,YAAY,EAAE,CAAC;wBAC7B,KAAK,YAAY,CAAC,UAAU;4BACxB,aAAa,GAAG,QAAQ,CAAC;4BACzB,MAAM;wBACV,KAAK,YAAY,CAAC,IAAI;4BAClB,aAAa,GAAG,MAAM,CAAC;4BACvB,MAAM;wBACV;4BACI,aAAa,GAAG,SAAS,CAAC,YAAY,CAAC;oBAC/C,CAAC;oBACD,+EAA+E;oBAC/E,MAAM,qBAAqB,GAAG,0BAA0B,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAsB,CAAC,CAAC;oBACpG,IACI,SAAS,CAAC,YAAY;wBACtB,CAAC,sBAAsB,CAAC,QAAQ,CAAC,SAAS,CAAC,YAAY,CAAC;wBACxD,CAAC,qBAAqB,EACxB,CAAC;wBACC,MAAM,IAAI,KAAK,CACX,8BAA8B,SAAS,CAAC,YAAY,IAAI;4BACpD,mBAAmB,SAAS,CAAC,aAAa,IAAI;4BAC9C,YAAY,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,aAAa,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAC5H,CAAC;oBACN,CAAC;oBACD,IAAI,SAAS,CAAC,OAAO,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;wBACtC,+CAA+C;wBAC/C,MAAM,eAAe,GAAG,wBAAwB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;wBACpE,QAAQ,GAAG,mBAAmB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;wBAClD,IAAI,QAAQ,EAAE,CAAC;4BACX,aAAa,GAAG,UAAU,CAAC;4BAC3B,yDAAyD;4BACzD,yDAAyD;4BACzD,kDAAkD;4BAClD,kBAAkB;4BAClB,IAAI,qBAAqB,EAAE,CAAC;gCACxB,MAAM,CAAC,MAAM,CAAC,IAAI,CACd,yCAAyC,SAAS,CAAC,YAAY,KAAK;oCAChE,2CAA2C,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/F,CAAC;4BACN,CAAC;wBACL,CAAC;wBACD,OAAO;4BACH,MAAM,EAAE,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE;4BACzD,WAAW,EAAE,WAAW;4BACxB,aAAa,EAAE,aAAa;4BAC5B,QAAQ;yBACX,CAAC;oBACN,CAAC;gBACL,CAAC;YACL,CAAC;YACD,wDAAwD;YACxD,OAAO;gBACH,MAAM,EAAE,IAAI,CAAC,cAAc,EAAE,kBAAkB;oBAC3C,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,KAAK,EAAE,IAAI,CAAC,cAAc,CAAC,kBAAkB,EAAE,CAAC;oBAC5E,CAAC,CAAC,EAAE;gBACR,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,WAAW,IAAI,EAAE;gBACrD,WAAW,EAAE,WAAW;aAC3B,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAClB,CAAC;IAED;;;;;;;;;;;;;;;;;;;OAmBG;IACH,qBAAqB,CAAC,OAAuB,EAAE,KAAc,EAAE,OAA+B;QAC1F,oDAAoD;QACpD,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAEhD,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,+CAA+C;YAC/C,oEAAoE;YACpE,MAAM,KAAK,CAAC;QAChB,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC;QACvB,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC;QAEvC,wBAAwB;QACxB,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;QAEjC,6CAA6C;QAC7C,IAAI,WAAW,GAAG,OAAO,CAAC;QAE1B,2DAA2D;QAC3D,IAAI,cAAc,EAAE,CAAC;YACjB,WAAW,GAAG,IAAI,cAAc,KAAK,WAAW,EAAE,CAAC;QACvD,CAAC;QAED,qDAAqD;QACrD,MAAM,SAAS,GAAG,IAAI,CAAC,sBAAsB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAEvE,kDAAkD;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAEjD,OAAO,IAAI,eAAe,CACtB,IAAI,OAAO,CAAC,QAAQ,KAAK,WAAW,EAAE,EACtC,SAAS,EACT,OAAO,EACP,KAAK,EACL,cAAc,EACd,SAAS,CACZ,CAAC;IACN,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,KAAc;QACnC,OAAO,CACH,KAAK,KAAK,IAAI;YACd,OAAO,KAAK,KAAK,QAAQ;YACzB,QAAQ,IAAI,KAAK;YACjB,OAAQ,KAA8B,CAAC,MAAM,KAAK,QAAQ;YAC1D,SAAS,IAAI,KAAK;YAClB,OAAQ,KAA+B,CAAC,OAAO,KAAK,QAAQ,CAC/D,CAAC;IACN,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BG;IACK,sBAAsB,CAAC,cAAsB,EAAE,OAAgB;QACnE,IAAI,OAAO,IAAI,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC;YAAE,OAAO,KAAK,CAAC;QAEnE,yBAAyB;QACzB,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,kBAAkB;QAC3D,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,mBAAmB;QAC5D,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,sCAAsC;QAC/E,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,cAAc;QACvD,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,sBAAsB;QAC/D,IAAI,cAAc,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,kBAAkB;QAC3D,IAAI,cAAc,IAAI,GAAG,IAAI,cAAc,GAAG,GAAG;YAAE,OAAO,IAAI,CAAC,CAAC,0BAA0B;QAE1F,+EAA+E;QAC/E,qEAAqE;QACrE,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,EAAE,CAAC;YACpC,IACI,OAAO,CAAC,QAAQ,CAAC,wCAAwC,CAAC;gBAC1D,OAAO,CAAC,QAAQ,CAAC,oCAAoC,CAAC,EACxD,CAAC;gBACC,OAAO,IAAI,CAAC;YAChB,CAAC;QACL,CAAC;QAED,2FAA2F;QAC3F,mFAAmF;QACnF,sFAAsF;QACtF,IAAI,OAAO,EAAE,CAAC;YACV,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YACpC,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAAE,OAAO,IAAI,CAAC;QAC5G,CAAC;QAED,kCAAkC;QAClC,IAAI,cAAc,IAAI,GAAG,IAAI,cAAc,GAAG,GAAG;YAAE,OAAO,KAAK,CAAC;QAEhE,4DAA4D;QAC5D,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,uBAAuB,CAAC,OAAe;QAC3C,MAAM,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAC3C,OAAO,YAAY,CAAC,QAAQ,CAAC,eAAe,CAAC,IAAI,YAAY,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;IAClG,CAAC;IAED;;;;OAIG;IACK,gBAAgB,CAAC,OAAe;QACpC,+BAA+B;QAC/B,MAAM,QAAQ,GAAG;YACb,+BAA+B,EAAE,sCAAsC;YACvE,aAAa,EAAE,wBAAwB;YACvC,eAAe,EAAE,yBAAyB;YAC1C,cAAc,EAAE,4BAA4B;SAC/C,CAAC;QAEF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACrC,IAAI,KAAK,EAAE,CAAC;gBACR,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC;YACpB,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ;AAED;;;;GAIG;AACH,MAAM,UAAU,gCAAgC,CAAC,QAAmB;IAChE,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;QAC5B,IAAI,CAAC,OAAO,CAAC,KAAK;YAAE,OAAO,OAAO,CAAC;QACnC,MAAM,gBAAgB,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,gBAAgB,CAAC,CAAC;QACzF,IAAI,CAAC,gBAAgB;YAAE,OAAO,OAAO,CAAC;QAEtC,MAAM,QAAQ,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACxC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpB,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACrF,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;gBACrF,OAAO,EAAE,IAAI,EAAE,eAAe,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,SAAS,IAAI,EAAE,CAAC;YAC5E,CAAC;YACD,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ;oBAC1C,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC;oBAChD,CAAC,CAAC,aAAa,CAAC;gBACpB,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC;gBACrF,OAAO,EAAE,IAAI,EAAE,oBAAoB,IAAI,CAAC,gBAAgB,CAAC,IAAI,KAAK,SAAS,GAAG,EAAE,CAAC;YACrF,CAAC;YACD,OAAO,IAAI,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,kBAAkB,CAAC,KAA0C;IAClE,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,SAAS,CAAC;IACrB,CAAC;IACD,oDAAoD;IACpD,8DAA8D;IAC9D,OAAO;QACH,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,eAAe,CAAC;KACnD,CAAC;AACN,CAAC;AAED,SAAS,eAAe,CAAC,IAAoB;IACzC,OAAO;QACH,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,0DAA0D;QAC1D,8EAA8E;QAC9E,sFAAsF;QACtF,oBAAoB,EAAE,IAAI,CAAC,YAAY;KAC1C,CAAC;AACN,CAAC;AAED;;;;;GAKG;AACH,SAAS,kBAAkB,CAAC,YAAqB,EAAE,MAAiB;IAChE,oDAAoD;IACpD,MAAM,SAAS,GAAG,uBAAuB,CAAU,YAAY,CAAC,CAAC;IACjE,MAAM,SAAS,GAAG,SAAS,IAAI,CAAE,YAA0B,IAAI,EAAE,CAAC,CAAC;IACnE,OAAO,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,GAAG,oBAAoB,CAAC;AAExC;;;GAGG;AACH,SAAS,6BAA6B,CAAC,YAAqB;IACxD,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;QAC5D,MAAM,CAAC,GAAG,YAAuC,CAAC;QAClD,IAAI,CAAC,CAAC,UAAU,CAAC,IAAI,OAAO,CAAC,CAAC,UAAU,CAAC,KAAK,QAAQ,EAAE,CAAC;YACrD,OAAO,CAAC,CAAC,UAAU,CAAY,CAAC;QACpC,CAAC;IACL,CAAC;IACD,OAAO,SAAS,CAAC;AACrB,CAAC;AAED;;;;;GAKG;AACH,SAAS,yBAAyB,CAAC,YAAqB,EAAE,MAA2B;IACjF,IAAI,CAAC,MAAM;QAAE,OAAO,YAAY,CAAC;IACjC,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;QAC5D,OAAO,EAAE,GAAI,YAAuB,EAAE,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACjE,CAAC;IACD,OAAO,YAAY,CAAC;AACxB,CAAC;AAED;;;;;;;;;;GAUG;AACH,SAAS,sBAAsB,CAAC,QAAgB;IAC5C,QAAQ,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC3B,IAAI,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACrD,IAAI,CAAC;YACD,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;QAAC,MAAM,CAAC;YACL,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;QAChC,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IAChC,CAAC;AACL,CAAC"}
@@ -1,7 +1,7 @@
1
- import { AIModel, Completion, ExecutionOptions, PromptSegment } from "@llumiverse/core";
2
- import { VertexAIDriver } from "../index.js";
1
+ import { type AIModel, type Completion, type ExecutionOptions, type PromptSegment } from '@llumiverse/core';
2
+ import type { VertexAIDriver } from '../index.js';
3
3
  interface ImagenBaseReference {
4
- referenceType: "REFERENCE_TYPE_RAW" | "REFERENCE_TYPE_MASK" | "REFERENCE_TYPE_SUBJECT" | "REFERENCE_TYPE_CONTROL" | "REFERENCE_TYPE_STYLE";
4
+ referenceType: 'REFERENCE_TYPE_RAW' | 'REFERENCE_TYPE_MASK' | 'REFERENCE_TYPE_SUBJECT' | 'REFERENCE_TYPE_CONTROL' | 'REFERENCE_TYPE_STYLE';
5
5
  referenceId: number;
6
6
  referenceImage: {
7
7
  bytesBase64Encoded: string;
@@ -25,10 +25,10 @@ export declare enum ImagenMaskMode {
25
25
  MASK_MODE_SEMANTIC = "MASK_MODE_SEMANTIC"
26
26
  }
27
27
  interface ImagenReferenceRaw extends ImagenBaseReference {
28
- referenceType: "REFERENCE_TYPE_RAW";
28
+ referenceType: 'REFERENCE_TYPE_RAW';
29
29
  }
30
- interface ImagenReferenceMask extends Omit<ImagenBaseReference, "referenceImage"> {
31
- referenceType: "REFERENCE_TYPE_MASK";
30
+ interface ImagenReferenceMask extends Omit<ImagenBaseReference, 'referenceImage'> {
31
+ referenceType: 'REFERENCE_TYPE_MASK';
32
32
  maskImageConfig: {
33
33
  maskMode?: ImagenMaskMode;
34
34
  maskClasses?: number[];
@@ -39,21 +39,21 @@ interface ImagenReferenceMask extends Omit<ImagenBaseReference, "referenceImage"
39
39
  };
40
40
  }
41
41
  interface ImagenReferenceSubject extends ImagenBaseReference {
42
- referenceType: "REFERENCE_TYPE_SUBJECT";
42
+ referenceType: 'REFERENCE_TYPE_SUBJECT';
43
43
  subjectImageConfig: {
44
44
  subjectDescription: string;
45
- subjectType: "SUBJECT_TYPE_PERSON" | "SUBJECT_TYPE_ANIMAL" | "SUBJECT_TYPE_PRODUCT" | "SUBJECT_TYPE_DEFAULT";
45
+ subjectType: 'SUBJECT_TYPE_PERSON' | 'SUBJECT_TYPE_ANIMAL' | 'SUBJECT_TYPE_PRODUCT' | 'SUBJECT_TYPE_DEFAULT';
46
46
  };
47
47
  }
48
48
  interface ImagenReferenceControl extends ImagenBaseReference {
49
- referenceType: "REFERENCE_TYPE_CONTROL";
49
+ referenceType: 'REFERENCE_TYPE_CONTROL';
50
50
  controlImageConfig: {
51
- controlType: "CONTROL_TYPE_FACE_MESH" | "CONTROL_TYPE_CANNY" | "CONTROL_TYPE_SCRIBBLE";
51
+ controlType: 'CONTROL_TYPE_FACE_MESH' | 'CONTROL_TYPE_CANNY' | 'CONTROL_TYPE_SCRIBBLE';
52
52
  enableControlImageComputation?: boolean;
53
53
  };
54
54
  }
55
55
  interface ImagenReferenceStyle extends ImagenBaseReference {
56
- referenceType: "REFERENCE_TYPE_STYLE";
56
+ referenceType: 'REFERENCE_TYPE_STYLE';
57
57
  styleImageConfig: {
58
58
  styleDescription?: string;
59
59
  };
@@ -65,6 +65,7 @@ export interface ImagenPrompt {
65
65
  subjectDescription?: string;
66
66
  negativePrompt?: string;
67
67
  }
68
+ export declare function formatImagenDebugPrompt(prompt: ImagenPrompt): ImagenPrompt;
68
69
  export declare class ImagenModelDefinition {
69
70
  model: AIModel;
70
71
  constructor(modelId: string);
@@ -0,0 +1 @@
1
+ {"version":3,"file":"imagen.d.ts","sourceRoot":"","sources":["../../../src/vertexai/models/imagen.ts"],"names":[],"mappings":"AAEA,OAAO,EACH,KAAK,OAAO,EACZ,KAAK,UAAU,EACf,KAAK,gBAAgB,EAIrB,KAAK,aAAa,EAErB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAElD,UAAU,mBAAmB;IACzB,aAAa,EACP,oBAAoB,GACpB,qBAAqB,GACrB,wBAAwB,GACxB,wBAAwB,GACxB,sBAAsB,CAAC;IAC7B,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE;QACZ,kBAAkB,EAAE,MAAM,CAAC;KAC9B,CAAC;CACL;AAED,oBAAY,cAAc;IACtB,UAAU,eAAe;IACzB,yBAAyB,8BAA8B;IACvD,2BAA2B,gCAAgC;IAC3D,gBAAgB,qBAAqB;IACrC,kBAAkB,uBAAuB;IACzC,qBAAqB,0BAA0B;IAC/C,mBAAmB,wBAAwB;IAC3C,wBAAwB,6BAA6B;IACrD,sBAAsB,2BAA2B;CACpD;AAED,oBAAY,cAAc;IACtB,uBAAuB,4BAA4B;IACnD,oBAAoB,yBAAyB;IAC7C,oBAAoB,yBAAyB;IAC7C,kBAAkB,uBAAuB;CAC5C;AAED,UAAU,kBAAmB,SAAQ,mBAAmB;IACpD,aAAa,EAAE,oBAAoB,CAAC;CACvC;AAED,UAAU,mBAAoB,SAAQ,IAAI,CAAC,mBAAmB,EAAE,gBAAgB,CAAC;IAC7E,aAAa,EAAE,qBAAqB,CAAC;IACrC,eAAe,EAAE;QACb,QAAQ,CAAC,EAAE,cAAc,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;QACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,cAAc,CAAC,EAAE;QAEb,kBAAkB,EAAE,MAAM,CAAC;KAC9B,CAAC;CACL;AAED,UAAU,sBAAuB,SAAQ,mBAAmB;IACxD,aAAa,EAAE,wBAAwB,CAAC;IACxC,kBAAkB,EAAE;QAChB,kBAAkB,EAAE,MAAM,CAAC;QAC3B,WAAW,EAAE,qBAAqB,GAAG,qBAAqB,GAAG,sBAAsB,GAAG,sBAAsB,CAAC;KAChH,CAAC;CACL;AAED,UAAU,sBAAuB,SAAQ,mBAAmB;IACxD,aAAa,EAAE,wBAAwB,CAAC;IACxC,kBAAkB,EAAE;QAChB,WAAW,EAAE,wBAAwB,GAAG,oBAAoB,GAAG,uBAAuB,CAAC;QACvF,6BAA6B,CAAC,EAAE,OAAO,CAAC;KAC3C,CAAC;CACL;AAED,UAAU,oBAAqB,SAAQ,mBAAmB;IACtD,aAAa,EAAE,sBAAsB,CAAC;IACtC,gBAAgB,EAAE;QACd,gBAAgB,CAAC,EAAE,MAAM,CAAC;KAC7B,CAAC;CACL;AAED,KAAK,aAAa,GACZ,kBAAkB,GAClB,mBAAmB,GACnB,sBAAsB,GACtB,sBAAsB,GACtB,oBAAoB,CAAC;AAE3B,MAAM,WAAW,YAAY;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,aAAa,EAAE,CAAC;IAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,YAAY,GAAG,YAAY,CAgB1E;AAkED,qBAAa,qBAAqB;IAC9B,KAAK,EAAE,OAAO,CAAC;gBAEH,OAAO,EAAE,MAAM;IAUrB,YAAY,CACd,OAAO,EAAE,cAAc,EACvB,QAAQ,EAAE,aAAa,EAAE,EACzB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,YAAY,CAAC;IAkLlB,sBAAsB,CACxB,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,YAAY,EACpB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,UAAU,CAAC;CAoEzB"}
@@ -1,6 +1,7 @@
1
- import { ModelType, PromptRole, readStreamAsBase64 } from "@llumiverse/core";
2
1
  // Import the helper module for converting arbitrary protobuf.Value objects
3
2
  import { helpers } from '@google-cloud/aiplatform';
3
+ import { ModelType, PromptRole, readStreamAsBase64, } from '@llumiverse/core';
4
+ import { truncateBinaryForDebug } from '../../shared/debug-prompt.js';
4
5
  export var ImagenTaskType;
5
6
  (function (ImagenTaskType) {
6
7
  ImagenTaskType["TEXT_IMAGE"] = "TEXT_IMAGE";
@@ -20,13 +21,30 @@ export var ImagenMaskMode;
20
21
  ImagenMaskMode["MASK_MODE_FOREGROUND"] = "MASK_MODE_FOREGROUND";
21
22
  ImagenMaskMode["MASK_MODE_SEMANTIC"] = "MASK_MODE_SEMANTIC";
22
23
  })(ImagenMaskMode || (ImagenMaskMode = {}));
24
+ export function formatImagenDebugPrompt(prompt) {
25
+ return {
26
+ ...prompt,
27
+ referenceImages: prompt.referenceImages?.map((reference) => {
28
+ if (!reference.referenceImage?.bytesBase64Encoded) {
29
+ return reference;
30
+ }
31
+ return {
32
+ ...reference,
33
+ referenceImage: {
34
+ ...reference.referenceImage,
35
+ bytesBase64Encoded: truncateBinaryForDebug(reference.referenceImage.bytesBase64Encoded),
36
+ },
37
+ };
38
+ }),
39
+ };
40
+ }
23
41
  function getImagenParameters(taskType, options) {
24
42
  const commonParameters = {
25
43
  sampleCount: options?.number_of_images,
26
44
  seed: options?.seed,
27
45
  safetySetting: options?.safety_setting,
28
46
  personGeneration: options?.person_generation,
29
- negativePrompt: taskType ? undefined : "", //Filled in later from the prompt
47
+ negativePrompt: taskType ? undefined : '', //Filled in later from the prompt
30
48
  //TODO: Add more safety and prompt rejection information
31
49
  //includeSafetyAttributes: true,
32
50
  //includeRaiReason: true,
@@ -35,7 +53,7 @@ function getImagenParameters(taskType, options) {
35
53
  case ImagenTaskType.EDIT_MODE_INPAINT_REMOVAL:
36
54
  return {
37
55
  ...commonParameters,
38
- editMode: "EDIT_MODE_INPAINT_REMOVAL",
56
+ editMode: 'EDIT_MODE_INPAINT_REMOVAL',
39
57
  editConfig: {
40
58
  baseSteps: options?.edit_steps,
41
59
  },
@@ -43,7 +61,7 @@ function getImagenParameters(taskType, options) {
43
61
  case ImagenTaskType.EDIT_MODE_INPAINT_INSERTION:
44
62
  return {
45
63
  ...commonParameters,
46
- editMode: "EDIT_MODE_INPAINT_INSERTION",
64
+ editMode: 'EDIT_MODE_INPAINT_INSERTION',
47
65
  editConfig: {
48
66
  baseSteps: options?.edit_steps,
49
67
  },
@@ -51,7 +69,7 @@ function getImagenParameters(taskType, options) {
51
69
  case ImagenTaskType.EDIT_MODE_BGSWAP:
52
70
  return {
53
71
  ...commonParameters,
54
- editMode: "EDIT_MODE_BGSWAP",
72
+ editMode: 'EDIT_MODE_BGSWAP',
55
73
  editConfig: {
56
74
  baseSteps: options?.edit_steps,
57
75
  },
@@ -59,7 +77,7 @@ function getImagenParameters(taskType, options) {
59
77
  case ImagenTaskType.EDIT_MODE_OUTPAINT:
60
78
  return {
61
79
  ...commonParameters,
62
- editMode: "EDIT_MODE_OUTPAINT",
80
+ editMode: 'EDIT_MODE_OUTPAINT',
63
81
  editConfig: {
64
82
  baseSteps: options?.edit_steps,
65
83
  },
@@ -80,7 +98,7 @@ function getImagenParameters(taskType, options) {
80
98
  ...commonParameters,
81
99
  };
82
100
  default:
83
- throw new Error("Task type not supported");
101
+ throw new Error('Task type not supported');
84
102
  }
85
103
  }
86
104
  export class ImagenModelDefinition {
@@ -95,13 +113,13 @@ export class ImagenModelDefinition {
95
113
  };
96
114
  }
97
115
  async createPrompt(_driver, segments, options) {
98
- const splits = options.model.split("/");
116
+ const splits = options.model.split('/');
99
117
  const modelName = splits[splits.length - 1];
100
118
  options = { ...options, model: modelName };
101
119
  const prompt = {
102
- prompt: "",
120
+ prompt: '',
103
121
  };
104
- //Collect text prompts, Imagen does not support roles, so everything gets merged together
122
+ //Collect text prompts, Imagen does not support roles, so everything gets merged together
105
123
  // however we still respect our typical pattern. System First, Safety Last.
106
124
  const system = [];
107
125
  const user = [];
@@ -128,96 +146,104 @@ export class ImagenModelDefinition {
128
146
  if (!prompt.referenceImages) {
129
147
  prompt.referenceImages = [];
130
148
  }
131
- //Always required, but only used by customisation.
149
+ //Always required, but only used by customisation.
132
150
  //Each ref ID refers to a single "reference", i.e. object. To provide multiple images of a single ref,
133
151
  //include multiple images in one prompt.
134
152
  const refId = prompt.referenceImages.length + 1;
135
153
  for (const img of msg.files) {
136
- if (img.mime_type?.includes("image")) {
154
+ if (img.mime_type?.includes('image')) {
137
155
  if (msg.role !== PromptRole.mask) {
138
156
  //Editing based mode requires a reference image
139
- if (imagenOptions?.edit_mode?.includes("EDIT_MODE")) {
157
+ if (imagenOptions?.edit_mode?.includes('EDIT_MODE')) {
140
158
  prompt.referenceImages.push({
141
- referenceType: "REFERENCE_TYPE_RAW",
159
+ referenceType: 'REFERENCE_TYPE_RAW',
142
160
  referenceId: refId,
143
161
  referenceImage: {
144
162
  bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
145
- }
163
+ },
146
164
  });
147
165
  //If mask is auto-generated, add a mask reference
148
166
  if (mask_mode !== ImagenMaskMode.MASK_MODE_USER_PROVIDED) {
149
167
  prompt.referenceImages.push({
150
- referenceType: "REFERENCE_TYPE_MASK",
168
+ referenceType: 'REFERENCE_TYPE_MASK',
151
169
  referenceId: refId,
152
170
  maskImageConfig: {
153
171
  maskMode: mask_mode,
154
172
  dilation: imagenOptions?.mask_dilation,
155
- }
173
+ },
156
174
  });
157
175
  }
158
176
  }
159
- else if (options.model_options?.edit_mode === ImagenTaskType.CUSTOMIZATION_SUBJECT) {
177
+ else if (options.model_options?.edit_mode ===
178
+ ImagenTaskType.CUSTOMIZATION_SUBJECT) {
160
179
  //First image is always the control image
161
- if (refId == 1) {
180
+ if (refId === 1) {
162
181
  //Customization subject mode requires a control image
163
182
  prompt.referenceImages.push({
164
- referenceType: "REFERENCE_TYPE_CONTROL",
183
+ referenceType: 'REFERENCE_TYPE_CONTROL',
165
184
  referenceId: refId,
166
185
  referenceImage: {
167
186
  bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
168
187
  },
169
188
  controlImageConfig: {
170
- controlType: imagenOptions?.controlType === "CONTROL_TYPE_FACE_MESH" ? "CONTROL_TYPE_FACE_MESH" : "CONTROL_TYPE_CANNY",
189
+ controlType: imagenOptions?.controlType === 'CONTROL_TYPE_FACE_MESH'
190
+ ? 'CONTROL_TYPE_FACE_MESH'
191
+ : 'CONTROL_TYPE_CANNY',
171
192
  enableControlImageComputation: imagenOptions?.controlImageComputation,
172
- }
193
+ },
173
194
  });
174
195
  }
175
196
  else {
176
197
  // Subject images
177
198
  prompt.referenceImages.push({
178
- referenceType: "REFERENCE_TYPE_SUBJECT",
199
+ referenceType: 'REFERENCE_TYPE_SUBJECT',
179
200
  referenceId: refId,
180
201
  referenceImage: {
181
202
  bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
182
203
  },
183
204
  subjectImageConfig: {
184
205
  subjectDescription: prompt.subjectDescription ?? msg.content,
185
- subjectType: imagenOptions?.subjectType ?? "SUBJECT_TYPE_DEFAULT",
186
- }
206
+ subjectType: imagenOptions?.subjectType ?? 'SUBJECT_TYPE_DEFAULT',
207
+ },
187
208
  });
188
209
  }
189
210
  }
190
- else if (options.model_options?.edit_mode === ImagenTaskType.CUSTOMIZATION_STYLE) {
211
+ else if (options.model_options?.edit_mode ===
212
+ ImagenTaskType.CUSTOMIZATION_STYLE) {
191
213
  // Style images
192
214
  prompt.referenceImages.push({
193
- referenceType: "REFERENCE_TYPE_STYLE",
215
+ referenceType: 'REFERENCE_TYPE_STYLE',
194
216
  referenceId: refId,
195
217
  referenceImage: {
196
218
  bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
197
219
  },
198
220
  styleImageConfig: {
199
221
  styleDescription: prompt.subjectDescription ?? msg.content,
200
- }
222
+ },
201
223
  });
202
224
  }
203
- else if (options.model_options?.edit_mode === ImagenTaskType.CUSTOMIZATION_CONTROLLED) {
225
+ else if (options.model_options?.edit_mode ===
226
+ ImagenTaskType.CUSTOMIZATION_CONTROLLED) {
204
227
  // Control images
205
228
  prompt.referenceImages.push({
206
- referenceType: "REFERENCE_TYPE_CONTROL",
229
+ referenceType: 'REFERENCE_TYPE_CONTROL',
207
230
  referenceId: refId,
208
231
  referenceImage: {
209
232
  bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
210
233
  },
211
234
  controlImageConfig: {
212
- controlType: imagenOptions?.controlType === "CONTROL_TYPE_FACE_MESH" ? "CONTROL_TYPE_FACE_MESH" : "CONTROL_TYPE_CANNY",
235
+ controlType: imagenOptions?.controlType === 'CONTROL_TYPE_FACE_MESH'
236
+ ? 'CONTROL_TYPE_FACE_MESH'
237
+ : 'CONTROL_TYPE_CANNY',
213
238
  enableControlImageComputation: imagenOptions?.controlImageComputation,
214
- }
239
+ },
215
240
  });
216
241
  }
217
- else if (options.model_options?.edit_mode === ImagenTaskType.CUSTOMIZATION_INSTRUCT) {
242
+ else if (options.model_options?.edit_mode ===
243
+ ImagenTaskType.CUSTOMIZATION_INSTRUCT) {
218
244
  // Control images
219
245
  prompt.referenceImages.push({
220
- referenceType: "REFERENCE_TYPE_RAW",
246
+ referenceType: 'REFERENCE_TYPE_RAW',
221
247
  referenceId: refId,
222
248
  referenceImage: {
223
249
  bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
@@ -228,7 +254,7 @@ export class ImagenModelDefinition {
228
254
  //If mask is user-provided, add a mask reference
229
255
  if (msg.role === PromptRole.mask && mask_mode === ImagenMaskMode.MASK_MODE_USER_PROVIDED) {
230
256
  prompt.referenceImages.push({
231
- referenceType: "REFERENCE_TYPE_MASK",
257
+ referenceType: 'REFERENCE_TYPE_MASK',
232
258
  referenceId: refId,
233
259
  referenceImage: {
234
260
  bytesBase64Encoded: await readStreamAsBase64(await img.getStream()),
@@ -236,7 +262,7 @@ export class ImagenModelDefinition {
236
262
  maskImageConfig: {
237
263
  maskMode: mask_mode,
238
264
  dilation: imagenOptions?.mask_dilation,
239
- }
265
+ },
240
266
  });
241
267
  }
242
268
  }
@@ -244,22 +270,23 @@ export class ImagenModelDefinition {
244
270
  }
245
271
  }
246
272
  //Extract the text from the segments
247
- prompt.prompt += [system.join("\n\n"), user.join("\n\n"), safety.join("\n\n")].join("\n\n");
273
+ prompt.prompt += [system.join('\n\n'), user.join('\n\n'), safety.join('\n\n')].join('\n\n');
248
274
  //Negative prompt
249
275
  if (negative.length > 0) {
250
- prompt.negativePrompt = negative.join(", ");
276
+ prompt.negativePrompt = negative.join(', ');
251
277
  }
252
278
  console.log(prompt);
253
279
  return prompt;
254
280
  }
255
281
  async requestImageGeneration(driver, prompt, options) {
256
- if (options.model_options?._option_id !== undefined && options.model_options?._option_id !== "vertexai-imagen") {
257
- driver.logger.debug({ options: options.model_options }, "Unexpected option id");
282
+ if (options.model_options?._option_id !== undefined &&
283
+ options.model_options?._option_id !== 'vertexai-imagen') {
284
+ driver.logger.debug({ options: options.model_options }, 'Unexpected option id');
258
285
  }
259
286
  options.model_options = options.model_options;
260
287
  const taskType = options.model_options?.edit_mode ?? ImagenTaskType.TEXT_IMAGE;
261
- driver.logger.info("Task type: " + taskType);
262
- const modelName = options.model.split("/").pop() ?? '';
288
+ driver.logger.info(`Task type: ${taskType}`);
289
+ const modelName = options.model.split('/').pop() ?? '';
263
290
  // Configure the parent resource
264
291
  // TODO: make location configurable, fixed to us-central1 for now
265
292
  const endpoint = `projects/${driver.options.project}/locations/us-central1/publishers/google/models/${modelName}`;
@@ -268,7 +295,7 @@ export class ImagenModelDefinition {
268
295
  throw new Error('No instance value found');
269
296
  }
270
297
  const instances = [instanceValue];
271
- let parameter = getImagenParameters(taskType, options.model_options ?? { _option_id: "vertexai-imagen" });
298
+ let parameter = getImagenParameters(taskType, options.model_options ?? { _option_id: 'vertexai-imagen' });
272
299
  parameter.negativePrompt = prompt.negativePrompt ?? undefined;
273
300
  const numberOfImages = options.model_options?.number_of_images ?? 1;
274
301
  // Remove all undefined values
@@ -287,11 +314,11 @@ export class ImagenModelDefinition {
287
314
  throw new Error('No predictions found');
288
315
  }
289
316
  // Extract base64 encoded images from predictions
290
- const images = predictions.map(prediction => prediction.structValue?.fields?.bytesBase64Encoded?.stringValue ?? '');
317
+ const images = predictions.map((prediction) => prediction.structValue?.fields?.bytesBase64Encoded?.stringValue ?? '');
291
318
  return {
292
- result: images.map(image => ({
293
- type: "image",
294
- value: image
319
+ result: images.map((image) => ({
320
+ type: 'image',
321
+ value: image,
295
322
  })),
296
323
  };
297
324
  }
@@ -0,0 +1 @@
1
+ {"version":3,"file":"imagen.js","sourceRoot":"","sources":["../../../src/vertexai/models/imagen.ts"],"names":[],"mappings":"AAAA,2EAA2E;AAC3E,OAAO,EAAE,OAAO,EAAe,MAAM,0BAA0B,CAAC;AAChE,OAAO,EAKH,SAAS,EACT,UAAU,EAEV,kBAAkB,GACrB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AAgBtE,MAAM,CAAN,IAAY,cAUX;AAVD,WAAY,cAAc;IACtB,2CAAyB,CAAA;IACzB,yEAAuD,CAAA;IACvD,6EAA2D,CAAA;IAC3D,uDAAqC,CAAA;IACrC,2DAAyC,CAAA;IACzC,iEAA+C,CAAA;IAC/C,6DAA2C,CAAA;IAC3C,uEAAqD,CAAA;IACrD,mEAAiD,CAAA;AACrD,CAAC,EAVW,cAAc,KAAd,cAAc,QAUzB;AAED,MAAM,CAAN,IAAY,cAKX;AALD,WAAY,cAAc;IACtB,qEAAmD,CAAA;IACnD,+DAA6C,CAAA;IAC7C,+DAA6C,CAAA;IAC7C,2DAAyC,CAAA;AAC7C,CAAC,EALW,cAAc,KAAd,cAAc,QAKzB;AAwDD,MAAM,UAAU,uBAAuB,CAAC,MAAoB;IACxD,OAAO;QACH,GAAG,MAAM;QACT,eAAe,EAAE,MAAM,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;YACvD,IAAI,CAAC,SAAS,CAAC,cAAc,EAAE,kBAAkB,EAAE,CAAC;gBAChD,OAAO,SAAS,CAAC;YACrB,CAAC;YACD,OAAO;gBACH,GAAG,SAAS;gBACZ,cAAc,EAAE;oBACZ,GAAG,SAAS,CAAC,cAAc;oBAC3B,kBAAkB,EAAE,sBAAsB,CAAC,SAAS,CAAC,cAAc,CAAC,kBAAkB,CAAC;iBAC1F;aACJ,CAAC;QACN,CAAC,CAAC;KACL,CAAC;AACN,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAgB,EAAE,OAAsB;IACjE,MAAM,gBAAgB,GAAG;QACrB,WAAW,EAAE,OAAO,EAAE,gBAAgB;QACtC,IAAI,EAAE,OAAO,EAAE,IAAI;QACnB,aAAa,EAAE,OAAO,EAAE,cAAc;QACtC,gBAAgB,EAAE,OAAO,EAAE,iBAAiB;QAC5C,cAAc,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,iCAAiC;QAC5E,wDAAwD;QACxD,gCAAgC;QAChC,yBAAyB;KAC5B,CAAC;IACF,QAAQ,QAAQ,EAAE,CAAC;QACf,KAAK,cAAc,CAAC,yBAAyB;YACzC,OAAO;gBACH,GAAG,gBAAgB;gBACnB,QAAQ,EAAE,2BAA2B;gBACrC,UAAU,EAAE;oBACR,SAAS,EAAE,OAAO,EAAE,UAAU;iBACjC;aACJ,CAAC;QACN,KAAK,cAAc,CAAC,2BAA2B;YAC3C,OAAO;gBACH,GAAG,gBAAgB;gBACnB,QAAQ,EAAE,6BAA6B;gBACvC,UAAU,EAAE;oBACR,SAAS,EAAE,OAAO,EAAE,UAAU;iBACjC;aACJ,CAAC;QACN,KAAK,cAAc,CAAC,gBAAgB;YAChC,OAAO;gBACH,GAAG,gBAAgB;gBACnB,QAAQ,EAAE,kBAAkB;gBAC5B,UAAU,EAAE;oBACR,SAAS,EAAE,OAAO,EAAE,UAAU;iBACjC;aACJ,CAAC;QACN,KAAK,cAAc,CAAC,kBAAkB;YAClC,OAAO;gBACH,GAAG,gBAAgB;gBACnB,QAAQ,EAAE,oBAAoB;gBAC9B,UAAU,EAAE;oBACR,SAAS,EAAE,OAAO,EAAE,UAAU;iBACjC;aACJ,CAAC;QACN,KAAK,cAAc,CAAC,UAAU;YAC1B,OAAO;gBACH,GAAG,gBAAgB;gBACnB,6DAA6D;gBAC7D,YAAY,EAAE,OAAO,EAAE,aAAa;gBACpC,WAAW,EAAE,OAAO,EAAE,YAAY;gBAClC,aAAa,EAAE,OAAO,EAAE,cAAc;aACzC,CAAC;QACN,KAAK,cAAc,CAAC,qBAAqB,CAAC;QAC1C,KAAK,cAAc,CAAC,wBAAwB,CAAC;QAC7C,KAAK,cAAc,CAAC,sBAAsB,CAAC;QAC3C,KAAK,cAAc,CAAC,mBAAmB;YACnC,OAAO;gBACH,GAAG,gBAAgB;aACtB,CAAC;QACN;YACI,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IACnD,CAAC;AACL,CAAC;AAED,MAAM,OAAO,qBAAqB;IAC9B,KAAK,CAAU;IAEf,YAAY,OAAe;QACvB,IAAI,CAAC,KAAK,GAAG;YACT,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,UAAU;YACpB,IAAI,EAAE,SAAS,CAAC,KAAK;YACrB,UAAU,EAAE,KAAK;SACpB,CAAC;IACN,CAAC;IAED,KAAK,CAAC,YAAY,CACd,OAAuB,EACvB,QAAyB,EACzB,OAAyB;QAEzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5C,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAE3C,MAAM,MAAM,GAAiB;YACzB,MAAM,EAAE,EAAE;SACb,CAAC;QAEF,yFAAyF;QACzF,2EAA2E;QAC3E,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAa,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,MAAM,SAAS,GAAI,OAAO,CAAC,aAA+B,EAAE,SAAS,CAAC;QACtE,MAAM,aAAa,GAAG,OAAO,CAAC,aAA8B,CAAC;QAE7D,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YACzB,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;gBACjC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7B,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC;gBACxC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7B,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,QAAQ,EAAE,CAAC;gBAC1C,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACJ,yDAAyD;gBACzD,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,CAAC;YACD,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACZ,0BAA0B;gBAC1B,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;oBAC1B,MAAM,CAAC,eAAe,GAAG,EAAE,CAAC;gBAChC,CAAC;gBAED,kDAAkD;gBAClD,sGAAsG;gBACtG,wCAAwC;gBACxC,MAAM,KAAK,GAAG,MAAM,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC;gBAChD,KAAK,MAAM,GAAG,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBAC1B,IAAI,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;wBACnC,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,EAAE,CAAC;4BAC/B,+CAA+C;4BAC/C,IAAI,aAAa,EAAE,SAAS,EAAE,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gCAClD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;oCACxB,aAAa,EAAE,oBAAoB;oCACnC,WAAW,EAAE,KAAK;oCAClB,cAAc,EAAE;wCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;qCACtE;iCACJ,CAAC,CAAC;gCACH,iDAAiD;gCACjD,IAAI,SAAS,KAAK,cAAc,CAAC,uBAAuB,EAAE,CAAC;oCACvD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;wCACxB,aAAa,EAAE,qBAAqB;wCACpC,WAAW,EAAE,KAAK;wCAClB,eAAe,EAAE;4CACb,QAAQ,EAAE,SAAS;4CACnB,QAAQ,EAAE,aAAa,EAAE,aAAa;yCACzC;qCACJ,CAAC,CAAC;gCACP,CAAC;4BACL,CAAC;iCAAM,IACF,OAAO,CAAC,aAA+B,EAAE,SAAS;gCACnD,cAAc,CAAC,qBAAqB,EACtC,CAAC;gCACC,yCAAyC;gCACzC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;oCACd,qDAAqD;oCACrD,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;wCACxB,aAAa,EAAE,wBAAwB;wCACvC,WAAW,EAAE,KAAK;wCAClB,cAAc,EAAE;4CACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;yCACtE;wCACD,kBAAkB,EAAE;4CAChB,WAAW,EACP,aAAa,EAAE,WAAW,KAAK,wBAAwB;gDACnD,CAAC,CAAC,wBAAwB;gDAC1B,CAAC,CAAC,oBAAoB;4CAC9B,6BAA6B,EAAE,aAAa,EAAE,uBAAuB;yCACxE;qCACJ,CAAC,CAAC;gCACP,CAAC;qCAAM,CAAC;oCACJ,iBAAiB;oCACjB,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;wCACxB,aAAa,EAAE,wBAAwB;wCACvC,WAAW,EAAE,KAAK;wCAClB,cAAc,EAAE;4CACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;yCACtE;wCACD,kBAAkB,EAAE;4CAChB,kBAAkB,EAAE,MAAM,CAAC,kBAAkB,IAAI,GAAG,CAAC,OAAO;4CAC5D,WAAW,EAAE,aAAa,EAAE,WAAW,IAAI,sBAAsB;yCACpE;qCACJ,CAAC,CAAC;gCACP,CAAC;4BACL,CAAC;iCAAM,IACF,OAAO,CAAC,aAA+B,EAAE,SAAS;gCACnD,cAAc,CAAC,mBAAmB,EACpC,CAAC;gCACC,eAAe;gCACf,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;oCACxB,aAAa,EAAE,sBAAsB;oCACrC,WAAW,EAAE,KAAK;oCAClB,cAAc,EAAE;wCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;qCACtE;oCACD,gBAAgB,EAAE;wCACd,gBAAgB,EAAE,MAAM,CAAC,kBAAkB,IAAI,GAAG,CAAC,OAAO;qCAC7D;iCACJ,CAAC,CAAC;4BACP,CAAC;iCAAM,IACF,OAAO,CAAC,aAA+B,EAAE,SAAS;gCACnD,cAAc,CAAC,wBAAwB,EACzC,CAAC;gCACC,iBAAiB;gCACjB,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;oCACxB,aAAa,EAAE,wBAAwB;oCACvC,WAAW,EAAE,KAAK;oCAClB,cAAc,EAAE;wCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;qCACtE;oCACD,kBAAkB,EAAE;wCAChB,WAAW,EACP,aAAa,EAAE,WAAW,KAAK,wBAAwB;4CACnD,CAAC,CAAC,wBAAwB;4CAC1B,CAAC,CAAC,oBAAoB;wCAC9B,6BAA6B,EAAE,aAAa,EAAE,uBAAuB;qCACxE;iCACJ,CAAC,CAAC;4BACP,CAAC;iCAAM,IACF,OAAO,CAAC,aAA+B,EAAE,SAAS;gCACnD,cAAc,CAAC,sBAAsB,EACvC,CAAC;gCACC,iBAAiB;gCACjB,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;oCACxB,aAAa,EAAE,oBAAoB;oCACnC,WAAW,EAAE,KAAK;oCAClB,cAAc,EAAE;wCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;qCACtE;iCACJ,CAAC,CAAC;4BACP,CAAC;wBACL,CAAC;wBACD,gDAAgD;wBAChD,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,IAAI,SAAS,KAAK,cAAc,CAAC,uBAAuB,EAAE,CAAC;4BACvF,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC;gCACxB,aAAa,EAAE,qBAAqB;gCACpC,WAAW,EAAE,KAAK;gCAClB,cAAc,EAAE;oCACZ,kBAAkB,EAAE,MAAM,kBAAkB,CAAC,MAAM,GAAG,CAAC,SAAS,EAAE,CAAC;iCACtE;gCACD,eAAe,EAAE;oCACb,QAAQ,EAAE,SAAS;oCACnB,QAAQ,EAAE,aAAa,EAAE,aAAa;iCACzC;6BACJ,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAED,oCAAoC;QACpC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE5F,iBAAiB;QACjB,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,MAAM,CAAC,cAAc,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAEpB,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,sBAAsB,CACxB,MAAsB,EACtB,MAAoB,EACpB,OAAyB;QAEzB,IACI,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,SAAS;YAC/C,OAAO,CAAC,aAAa,EAAE,UAAU,KAAK,iBAAiB,EACzD,CAAC;YACC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,aAAa,EAAE,EAAE,sBAAsB,CAAC,CAAC;QACpF,CAAC;QACD,OAAO,CAAC,aAAa,GAAG,OAAO,CAAC,aAA0C,CAAC;QAE3E,MAAM,QAAQ,GAAW,OAAO,CAAC,aAAa,EAAE,SAAS,IAAI,cAAc,CAAC,UAAU,CAAC;QAEvF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,QAAQ,EAAE,CAAC,CAAC;QAE7C,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;QAEvD,gCAAgC;QAChC,iEAAiE;QACjE,MAAM,QAAQ,GAAG,YAAY,MAAM,CAAC,OAAO,CAAC,OAAO,mDAAmD,SAAS,EAAE,CAAC;QAElH,MAAM,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,aAAa,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC/C,CAAC;QACD,MAAM,SAAS,GAAG,CAAC,aAAa,CAAC,CAAC;QAElC,IAAI,SAAS,GAAG,mBAAmB,CAC/B,QAAQ,EACR,OAAO,CAAC,aAAa,IAAI,EAAE,UAAU,EAAE,iBAAiB,EAAE,CACW,CAAC;QAC1E,SAAS,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,SAAS,CAAC;QAE9D,MAAM,cAAc,GAAG,OAAO,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC,CAAC;QAEpE,8BAA8B;QAC9B,SAAS,GAAG,MAAM,CAAC,WAAW,CAC1B,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAC5C,CAAC;QAEtB,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAE9C,MAAM,OAAO,GAAsD;YAC/D,QAAQ;YACR,SAAS;YACT,UAAU;SACb,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,eAAe,EAAE,CAAC;QAE9C,kBAAkB;QAClB,MAAM,CAAC,QAAQ,CAAC,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,GAAG,cAAc,EAAE,CAAC,CAAC,CAAC,uCAAuC;QAC/H,MAAM,WAAW,GAAG,QAAQ,CAAC,WAAW,CAAC;QAEzC,IAAI,CAAC,WAAW,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC5C,CAAC;QAED,iDAAiD;QACjD,MAAM,MAAM,GAAa,WAAW,CAAC,GAAG,CACpC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,MAAM,EAAE,kBAAkB,EAAE,WAAW,IAAI,EAAE,CACxF,CAAC;QAEF,OAAO;YACH,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC3B,IAAI,EAAE,OAAgB;gBACtB,KAAK,EAAE,KAAK;aACf,CAAC,CAAC;SACN,CAAC;IACN,CAAC;CACJ"}
@@ -1,11 +1,11 @@
1
- import { AIModel, Completion, CompletionChunkObject, ExecutionOptions, PromptOptions, PromptSegment } from "@llumiverse/core";
2
- import { VertexAIDriver } from "../index.js";
3
- import { ModelDefinition } from "../models.js";
1
+ import { type AIModel, type Completion, type CompletionChunkObject, type ExecutionOptions, type PromptOptions, type PromptSegment } from '@llumiverse/core';
2
+ import type { VertexAIDriver } from '../index.js';
3
+ import type { ModelDefinition } from '../models.js';
4
4
  interface LLamaMessage {
5
5
  role: string;
6
6
  content: string;
7
7
  }
8
- interface LLamaPrompt {
8
+ export interface LLamaPrompt {
9
9
  messages: LLamaMessage[];
10
10
  }
11
11
  export declare class LLamaModelDefinition implements ModelDefinition<LLamaPrompt> {
@@ -0,0 +1 @@
1
+ {"version":3,"file":"llama.d.ts","sourceRoot":"","sources":["../../../src/vertexai/models/llama.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,KAAK,OAAO,EACZ,KAAK,UAAU,EACf,KAAK,qBAAqB,EAC1B,KAAK,gBAAgB,EAErB,KAAK,aAAa,EAElB,KAAK,aAAa,EAErB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAEpD,UAAU,YAAY;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,WAAW;IACxB,QAAQ,EAAE,YAAY,EAAE,CAAC;CAC5B;AAqED,qBAAa,oBAAqB,YAAW,eAAe,CAAC,WAAW,CAAC;IACrE,KAAK,EAAE,OAAO,CAAC;gBAEH,OAAO,EAAE,MAAM;IAW3B,mBAAmB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM;IASxC,YAAY,CACd,OAAO,EAAE,cAAc,EACvB,QAAQ,EAAE,aAAa,EAAE,EACzB,OAAO,EAAE,aAAa,GACvB,OAAO,CAAC,WAAW,CAAC;IAwCjB,qBAAqB,CACvB,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,WAAW,EACnB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,UAAU,CAAC;IA6DhB,2BAA2B,CAC7B,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,WAAW,EACnB,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,aAAa,CAAC,qBAAqB,CAAC,CAAC;CAuDnD"}
@@ -1,5 +1,5 @@
1
- import { ModelType, PromptRole } from "@llumiverse/core";
2
- import { transformSSEStream } from "@llumiverse/core/async";
1
+ import { ModelType, PromptRole, } from '@llumiverse/core';
2
+ import { transformSSEStream } from '@llumiverse/core/async';
3
3
  /**
4
4
  * Convert a stream to a string
5
5
  */
@@ -53,7 +53,7 @@ export class LLamaModelDefinition {
53
53
  let messageContent = segment.content || '';
54
54
  if (segment.files && segment.files.length > 0) {
55
55
  for (const file of segment.files) {
56
- if (file.mime_type?.startsWith("text/")) {
56
+ if (file.mime_type?.startsWith('text/')) {
57
57
  const fileStream = await file.getStream();
58
58
  const fileContent = await streamToString(fileStream);
59
59
  messageContent += `\n\nFile content:\n${fileContent}`;
@@ -62,13 +62,13 @@ export class LLamaModelDefinition {
62
62
  }
63
63
  messages.push({
64
64
  role: role,
65
- content: messageContent
65
+ content: messageContent,
66
66
  });
67
67
  }
68
68
  if (options.result_schema) {
69
69
  messages.push({
70
70
  role: 'user',
71
- content: "The answer must be a JSON object using the following JSON Schema:\n" + JSON.stringify(options.result_schema)
71
+ content: `The answer must be a JSON object using the following JSON Schema:\n${JSON.stringify(options.result_schema)}`,
72
72
  });
73
73
  }
74
74
  // Return the prompt in the format expected by Llama MaaS API
@@ -77,7 +77,7 @@ export class LLamaModelDefinition {
77
77
  };
78
78
  }
79
79
  async requestTextCompletion(driver, prompt, options) {
80
- const splits = options.model.split("/");
80
+ const splits = options.model.split('/');
81
81
  const modelName = splits[splits.length - 1];
82
82
  let conversation = updateConversation(options.conversation, prompt);
83
83
  const modelOptions = options.model_options;
@@ -94,41 +94,43 @@ export class LLamaModelDefinition {
94
94
  google: {
95
95
  model_safety_settings: {
96
96
  enabled: false,
97
- llama_guard_settings: {}
98
- }
99
- }
100
- }
97
+ llama_guard_settings: {},
98
+ },
99
+ },
100
+ },
101
101
  };
102
102
  // Make POST request to the Llama MaaS API
103
103
  const region = this.getLlamaModelRegion(modelName);
104
104
  const client = driver.getLLamaClient(region);
105
105
  const openaiEndpoint = `endpoints/openapi/chat/completions`;
106
- const result = await client.post(openaiEndpoint, {
107
- payload
108
- });
106
+ const result = (await client.post(openaiEndpoint, {
107
+ payload,
108
+ }));
109
109
  // Extract response data
110
110
  const assistantMessage = result?.choices[0]?.message;
111
111
  const text = assistantMessage?.content;
112
112
  // Update conversation with the response
113
113
  conversation = updateConversation(conversation, {
114
- messages: [{
114
+ messages: [
115
+ {
115
116
  role: assistantMessage?.role,
116
- content: text
117
- }],
117
+ content: text,
118
+ },
119
+ ],
118
120
  });
119
121
  return {
120
- result: [{ type: "text", value: text }],
122
+ result: [{ type: 'text', value: text }],
121
123
  token_usage: {
122
124
  prompt: result.usage.prompt_tokens,
123
125
  result: result.usage.completion_tokens,
124
- total: result.usage.total_tokens
126
+ total: result.usage.total_tokens,
125
127
  },
126
128
  finish_reason: result.choices[0].finish_reason,
127
- conversation
129
+ conversation,
128
130
  };
129
131
  }
130
132
  async requestTextCompletionStream(driver, prompt, options) {
131
- const splits = options.model.split("/");
133
+ const splits = options.model.split('/');
132
134
  const modelName = splits[splits.length - 1];
133
135
  const conversation = updateConversation(options.conversation, prompt);
134
136
  const modelOptions = options.model_options;
@@ -145,10 +147,10 @@ export class LLamaModelDefinition {
145
147
  google: {
146
148
  model_safety_settings: {
147
149
  enabled: false,
148
- llama_guard_settings: {}
149
- }
150
- }
151
- }
150
+ llama_guard_settings: {},
151
+ },
152
+ },
153
+ },
152
154
  };
153
155
  // Make POST request to the Llama MaaS API
154
156
  //TODO: Fix error handling with the fetch client, errors will return a empty response
@@ -156,22 +158,24 @@ export class LLamaModelDefinition {
156
158
  const region = this.getLlamaModelRegion(modelName);
157
159
  const client = driver.getLLamaClient(region);
158
160
  const openaiEndpoint = `endpoints/openapi/chat/completions`;
159
- const stream = await client.post(openaiEndpoint, {
161
+ const stream = (await client.post(openaiEndpoint, {
160
162
  payload,
161
- reader: 'sse'
162
- });
163
+ reader: 'sse',
164
+ }));
163
165
  return transformSSEStream(stream, (data) => {
164
166
  const json = JSON.parse(data);
165
167
  const choice = json.choices?.[0];
166
168
  const content = choice?.delta?.content ?? '';
167
169
  return {
168
- result: content ? [{ type: "text", value: content }] : [],
170
+ result: content ? [{ type: 'text', value: content }] : [],
169
171
  finish_reason: choice?.finish_reason,
170
- token_usage: json.usage ? {
171
- prompt: json.usage.prompt_tokens,
172
- result: json.usage.completion_tokens,
173
- total: json.usage.total_tokens,
174
- } : undefined
172
+ token_usage: json.usage
173
+ ? {
174
+ prompt: json.usage.prompt_tokens,
175
+ result: json.usage.completion_tokens,
176
+ total: json.usage.total_tokens,
177
+ }
178
+ : undefined,
175
179
  };
176
180
  });
177
181
  }