@angri450/zvec-grep 0.2.2

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 (418) hide show
  1. package/LICENSE +201 -0
  2. package/README.md +319 -0
  3. package/README_CN.md +297 -0
  4. package/dist/authorization/index.d.ts +7 -0
  5. package/dist/authorization/index.js +7 -0
  6. package/dist/authorization/index.js.map +1 -0
  7. package/dist/authorization/manager.d.ts +8 -0
  8. package/dist/authorization/manager.js +19 -0
  9. package/dist/authorization/manager.js.map +1 -0
  10. package/dist/authorization/operation.d.ts +7 -0
  11. package/dist/authorization/operation.js +50 -0
  12. package/dist/authorization/operation.js.map +1 -0
  13. package/dist/authorization/planner.d.ts +21 -0
  14. package/dist/authorization/planner.js +97 -0
  15. package/dist/authorization/planner.js.map +1 -0
  16. package/dist/authorization/prompt.d.ts +12 -0
  17. package/dist/authorization/prompt.js +63 -0
  18. package/dist/authorization/prompt.js.map +1 -0
  19. package/dist/authorization/store.d.ts +20 -0
  20. package/dist/authorization/store.js +251 -0
  21. package/dist/authorization/store.js.map +1 -0
  22. package/dist/authorization/target.d.ts +15 -0
  23. package/dist/authorization/target.js +49 -0
  24. package/dist/authorization/target.js.map +1 -0
  25. package/dist/authorization/types.d.ts +61 -0
  26. package/dist/authorization/types.js +2 -0
  27. package/dist/authorization/types.js.map +1 -0
  28. package/dist/cli/args.d.ts +6 -0
  29. package/dist/cli/args.js +1255 -0
  30. package/dist/cli/args.js.map +1 -0
  31. package/dist/cli/auth.d.ts +21 -0
  32. package/dist/cli/auth.js +209 -0
  33. package/dist/cli/auth.js.map +1 -0
  34. package/dist/cli/commands.d.ts +4 -0
  35. package/dist/cli/commands.js +801 -0
  36. package/dist/cli/commands.js.map +1 -0
  37. package/dist/cli/errors.d.ts +9 -0
  38. package/dist/cli/errors.js +131 -0
  39. package/dist/cli/errors.js.map +1 -0
  40. package/dist/cli/format/context.d.ts +9 -0
  41. package/dist/cli/format/context.js +786 -0
  42. package/dist/cli/format/context.js.map +1 -0
  43. package/dist/cli/format/debug.d.ts +4 -0
  44. package/dist/cli/format/debug.js +60 -0
  45. package/dist/cli/format/debug.js.map +1 -0
  46. package/dist/cli/format/highlight.d.ts +3 -0
  47. package/dist/cli/format/highlight.js +106 -0
  48. package/dist/cli/format/highlight.js.map +1 -0
  49. package/dist/cli/format/progress.d.ts +13 -0
  50. package/dist/cli/format/progress.js +290 -0
  51. package/dist/cli/format/progress.js.map +1 -0
  52. package/dist/cli/format/range.d.ts +3 -0
  53. package/dist/cli/format/range.js +24 -0
  54. package/dist/cli/format/range.js.map +1 -0
  55. package/dist/cli/format/status.d.ts +83 -0
  56. package/dist/cli/format/status.js +530 -0
  57. package/dist/cli/format/status.js.map +1 -0
  58. package/dist/cli/help.d.ts +1 -0
  59. package/dist/cli/help.js +546 -0
  60. package/dist/cli/help.js.map +1 -0
  61. package/dist/cli/index.d.ts +2 -0
  62. package/dist/cli/index.js +40 -0
  63. package/dist/cli/index.js.map +1 -0
  64. package/dist/cli/install.d.ts +4 -0
  65. package/dist/cli/install.js +1735 -0
  66. package/dist/cli/install.js.map +1 -0
  67. package/dist/cli/managed-rg.d.ts +10 -0
  68. package/dist/cli/managed-rg.js +337 -0
  69. package/dist/cli/managed-rg.js.map +1 -0
  70. package/dist/cli/server-search.d.ts +3 -0
  71. package/dist/cli/server-search.js +12 -0
  72. package/dist/cli/server-search.js.map +1 -0
  73. package/dist/cli/types.d.ts +87 -0
  74. package/dist/cli/types.js +10 -0
  75. package/dist/cli/types.js.map +1 -0
  76. package/dist/cli/version.d.ts +1 -0
  77. package/dist/cli/version.js +13 -0
  78. package/dist/cli/version.js.map +1 -0
  79. package/dist/client/daemon-client.d.ts +31 -0
  80. package/dist/client/daemon-client.js +203 -0
  81. package/dist/client/daemon-client.js.map +1 -0
  82. package/dist/client/mode-router.d.ts +9 -0
  83. package/dist/client/mode-router.js +28 -0
  84. package/dist/client/mode-router.js.map +1 -0
  85. package/dist/client/search-policy.d.ts +11 -0
  86. package/dist/client/search-policy.js +15 -0
  87. package/dist/client/search-policy.js.map +1 -0
  88. package/dist/daemon/backend.d.ts +78 -0
  89. package/dist/daemon/backend.js +1118 -0
  90. package/dist/daemon/backend.js.map +1 -0
  91. package/dist/daemon/change-set.d.ts +30 -0
  92. package/dist/daemon/change-set.js +163 -0
  93. package/dist/daemon/change-set.js.map +1 -0
  94. package/dist/daemon/config.d.ts +27 -0
  95. package/dist/daemon/config.js +107 -0
  96. package/dist/daemon/config.js.map +1 -0
  97. package/dist/daemon/errors.d.ts +5 -0
  98. package/dist/daemon/errors.js +11 -0
  99. package/dist/daemon/errors.js.map +1 -0
  100. package/dist/daemon/http-server.d.ts +30 -0
  101. package/dist/daemon/http-server.js +307 -0
  102. package/dist/daemon/http-server.js.map +1 -0
  103. package/dist/daemon/index-coordinator.d.ts +22 -0
  104. package/dist/daemon/index-coordinator.js +55 -0
  105. package/dist/daemon/index-coordinator.js.map +1 -0
  106. package/dist/daemon/job-scheduler.d.ts +85 -0
  107. package/dist/daemon/job-scheduler.js +492 -0
  108. package/dist/daemon/job-scheduler.js.map +1 -0
  109. package/dist/daemon/logger.d.ts +10 -0
  110. package/dist/daemon/logger.js +101 -0
  111. package/dist/daemon/logger.js.map +1 -0
  112. package/dist/daemon/model-pool.d.ts +48 -0
  113. package/dist/daemon/model-pool.js +190 -0
  114. package/dist/daemon/model-pool.js.map +1 -0
  115. package/dist/daemon/root-lease.d.ts +14 -0
  116. package/dist/daemon/root-lease.js +197 -0
  117. package/dist/daemon/root-lease.js.map +1 -0
  118. package/dist/daemon/root-runtime.d.ts +80 -0
  119. package/dist/daemon/root-runtime.js +340 -0
  120. package/dist/daemon/root-runtime.js.map +1 -0
  121. package/dist/daemon/runtime-manager.d.ts +47 -0
  122. package/dist/daemon/runtime-manager.js +252 -0
  123. package/dist/daemon/runtime-manager.js.map +1 -0
  124. package/dist/daemon/runtime.d.ts +12 -0
  125. package/dist/daemon/runtime.js +105 -0
  126. package/dist/daemon/runtime.js.map +1 -0
  127. package/dist/daemon/server-controller.d.ts +38 -0
  128. package/dist/daemon/server-controller.js +327 -0
  129. package/dist/daemon/server-controller.js.map +1 -0
  130. package/dist/daemon/watch-manager.d.ts +62 -0
  131. package/dist/daemon/watch-manager.js +415 -0
  132. package/dist/daemon/watch-manager.js.map +1 -0
  133. package/dist/daemon/workspace-read-session-cache.d.ts +31 -0
  134. package/dist/daemon/workspace-read-session-cache.js +115 -0
  135. package/dist/daemon/workspace-read-session-cache.js.map +1 -0
  136. package/dist/engine/code-formats.d.ts +3 -0
  137. package/dist/engine/code-formats.js +14 -0
  138. package/dist/engine/code-formats.js.map +1 -0
  139. package/dist/engine/config.d.ts +57 -0
  140. package/dist/engine/config.js +351 -0
  141. package/dist/engine/config.js.map +1 -0
  142. package/dist/engine/errors.d.ts +20 -0
  143. package/dist/engine/errors.js +51 -0
  144. package/dist/engine/errors.js.map +1 -0
  145. package/dist/engine/extraction/code/adapter.d.ts +19 -0
  146. package/dist/engine/extraction/code/adapter.js +24 -0
  147. package/dist/engine/extraction/code/adapter.js.map +1 -0
  148. package/dist/engine/extraction/code/extractor.d.ts +13 -0
  149. package/dist/engine/extraction/code/extractor.js +696 -0
  150. package/dist/engine/extraction/code/extractor.js.map +1 -0
  151. package/dist/engine/extraction/code/families/c-family.d.ts +2 -0
  152. package/dist/engine/extraction/code/families/c-family.js +122 -0
  153. package/dist/engine/extraction/code/families/c-family.js.map +1 -0
  154. package/dist/engine/extraction/code/families/js-ts.d.ts +12 -0
  155. package/dist/engine/extraction/code/families/js-ts.js +118 -0
  156. package/dist/engine/extraction/code/families/js-ts.js.map +1 -0
  157. package/dist/engine/extraction/code/families/metadata.d.ts +8 -0
  158. package/dist/engine/extraction/code/families/metadata.js +86 -0
  159. package/dist/engine/extraction/code/families/metadata.js.map +1 -0
  160. package/dist/engine/extraction/code/families/name-field.d.ts +3 -0
  161. package/dist/engine/extraction/code/families/name-field.js +22 -0
  162. package/dist/engine/extraction/code/families/name-field.js.map +1 -0
  163. package/dist/engine/extraction/code/languages/c.d.ts +1 -0
  164. package/dist/engine/extraction/code/languages/c.js +12 -0
  165. package/dist/engine/extraction/code/languages/c.js.map +1 -0
  166. package/dist/engine/extraction/code/languages/cpp.d.ts +1 -0
  167. package/dist/engine/extraction/code/languages/cpp.js +18 -0
  168. package/dist/engine/extraction/code/languages/cpp.js.map +1 -0
  169. package/dist/engine/extraction/code/languages/go.d.ts +2 -0
  170. package/dist/engine/extraction/code/languages/go.js +70 -0
  171. package/dist/engine/extraction/code/languages/go.js.map +1 -0
  172. package/dist/engine/extraction/code/languages/java.d.ts +1 -0
  173. package/dist/engine/extraction/code/languages/java.js +17 -0
  174. package/dist/engine/extraction/code/languages/java.js.map +1 -0
  175. package/dist/engine/extraction/code/languages/javascript.d.ts +2 -0
  176. package/dist/engine/extraction/code/languages/javascript.js +23 -0
  177. package/dist/engine/extraction/code/languages/javascript.js.map +1 -0
  178. package/dist/engine/extraction/code/languages/python.d.ts +2 -0
  179. package/dist/engine/extraction/code/languages/python.js +52 -0
  180. package/dist/engine/extraction/code/languages/python.js.map +1 -0
  181. package/dist/engine/extraction/code/languages/rust.d.ts +2 -0
  182. package/dist/engine/extraction/code/languages/rust.js +25 -0
  183. package/dist/engine/extraction/code/languages/rust.js.map +1 -0
  184. package/dist/engine/extraction/code/languages/typescript.d.ts +2 -0
  185. package/dist/engine/extraction/code/languages/typescript.js +37 -0
  186. package/dist/engine/extraction/code/languages/typescript.js.map +1 -0
  187. package/dist/engine/extraction/code/tree-sitter/grammar.d.ts +4 -0
  188. package/dist/engine/extraction/code/tree-sitter/grammar.js +49 -0
  189. package/dist/engine/extraction/code/tree-sitter/grammar.js.map +1 -0
  190. package/dist/engine/extraction/code/tree-sitter/nodes.d.ts +3 -0
  191. package/dist/engine/extraction/code/tree-sitter/nodes.js +27 -0
  192. package/dist/engine/extraction/code/tree-sitter/nodes.js.map +1 -0
  193. package/dist/engine/extraction/code/tree-sitter/parser.d.ts +2 -0
  194. package/dist/engine/extraction/code/tree-sitter/parser.js +31 -0
  195. package/dist/engine/extraction/code/tree-sitter/parser.js.map +1 -0
  196. package/dist/engine/extraction/ids.d.ts +1 -0
  197. package/dist/engine/extraction/ids.js +5 -0
  198. package/dist/engine/extraction/ids.js.map +1 -0
  199. package/dist/engine/extraction/image/extractor.d.ts +5 -0
  200. package/dist/engine/extraction/image/extractor.js +35 -0
  201. package/dist/engine/extraction/image/extractor.js.map +1 -0
  202. package/dist/engine/extraction/index.d.ts +4 -0
  203. package/dist/engine/extraction/index.js +3 -0
  204. package/dist/engine/extraction/index.js.map +1 -0
  205. package/dist/engine/extraction/markdown/extractor.d.ts +7 -0
  206. package/dist/engine/extraction/markdown/extractor.js +362 -0
  207. package/dist/engine/extraction/markdown/extractor.js.map +1 -0
  208. package/dist/engine/extraction/runtime.d.ts +9 -0
  209. package/dist/engine/extraction/runtime.js +37 -0
  210. package/dist/engine/extraction/runtime.js.map +1 -0
  211. package/dist/engine/extraction/source.d.ts +18 -0
  212. package/dist/engine/extraction/source.js +22 -0
  213. package/dist/engine/extraction/source.js.map +1 -0
  214. package/dist/engine/extraction/text/extractor.d.ts +7 -0
  215. package/dist/engine/extraction/text/extractor.js +168 -0
  216. package/dist/engine/extraction/text/extractor.js.map +1 -0
  217. package/dist/engine/extraction/types.d.ts +4 -0
  218. package/dist/engine/extraction/types.js +2 -0
  219. package/dist/engine/extraction/types.js.map +1 -0
  220. package/dist/engine/extraction/vector-content.d.ts +5 -0
  221. package/dist/engine/extraction/vector-content.js +83 -0
  222. package/dist/engine/extraction/vector-content.js.map +1 -0
  223. package/dist/engine/file-size-policy.d.ts +6 -0
  224. package/dist/engine/file-size-policy.js +20 -0
  225. package/dist/engine/file-size-policy.js.map +1 -0
  226. package/dist/engine/file-type.d.ts +15 -0
  227. package/dist/engine/file-type.js +152 -0
  228. package/dist/engine/file-type.js.map +1 -0
  229. package/dist/engine/index-status.d.ts +9 -0
  230. package/dist/engine/index-status.js +38 -0
  231. package/dist/engine/index-status.js.map +1 -0
  232. package/dist/engine/manifest.d.ts +13 -0
  233. package/dist/engine/manifest.js +124 -0
  234. package/dist/engine/manifest.js.map +1 -0
  235. package/dist/engine/models/artifact-cache-lock.d.ts +16 -0
  236. package/dist/engine/models/artifact-cache-lock.js +228 -0
  237. package/dist/engine/models/artifact-cache-lock.js.map +1 -0
  238. package/dist/engine/models/artifact-downloader.d.ts +77 -0
  239. package/dist/engine/models/artifact-downloader.js +766 -0
  240. package/dist/engine/models/artifact-downloader.js.map +1 -0
  241. package/dist/engine/models/backends/llama-cpp.d.ts +100 -0
  242. package/dist/engine/models/backends/llama-cpp.js +567 -0
  243. package/dist/engine/models/backends/llama-cpp.js.map +1 -0
  244. package/dist/engine/models/backends/model2vec-runtime.d.ts +63 -0
  245. package/dist/engine/models/backends/model2vec-runtime.js +112 -0
  246. package/dist/engine/models/backends/model2vec-runtime.js.map +1 -0
  247. package/dist/engine/models/backends/model2vec-tokenizer.d.ts +2 -0
  248. package/dist/engine/models/backends/model2vec-tokenizer.js +42 -0
  249. package/dist/engine/models/backends/model2vec-tokenizer.js.map +1 -0
  250. package/dist/engine/models/backends/model2vec-worker-pool.d.ts +25 -0
  251. package/dist/engine/models/backends/model2vec-worker-pool.js +264 -0
  252. package/dist/engine/models/backends/model2vec-worker-pool.js.map +1 -0
  253. package/dist/engine/models/backends/model2vec-worker.d.ts +1 -0
  254. package/dist/engine/models/backends/model2vec-worker.js +41 -0
  255. package/dist/engine/models/backends/model2vec-worker.js.map +1 -0
  256. package/dist/engine/models/backends/model2vec.d.ts +38 -0
  257. package/dist/engine/models/backends/model2vec.js +344 -0
  258. package/dist/engine/models/backends/model2vec.js.map +1 -0
  259. package/dist/engine/models/backends/qwen.d.ts +44 -0
  260. package/dist/engine/models/backends/qwen.js +411 -0
  261. package/dist/engine/models/backends/qwen.js.map +1 -0
  262. package/dist/engine/models/backends/transformers-js.d.ts +90 -0
  263. package/dist/engine/models/backends/transformers-js.js +411 -0
  264. package/dist/engine/models/backends/transformers-js.js.map +1 -0
  265. package/dist/engine/models/catalog.d.ts +490 -0
  266. package/dist/engine/models/catalog.js +522 -0
  267. package/dist/engine/models/catalog.js.map +1 -0
  268. package/dist/engine/models/download-progress.d.ts +16 -0
  269. package/dist/engine/models/download-progress.js +47 -0
  270. package/dist/engine/models/download-progress.js.map +1 -0
  271. package/dist/engine/models/embeddings.d.ts +73 -0
  272. package/dist/engine/models/embeddings.js +125 -0
  273. package/dist/engine/models/embeddings.js.map +1 -0
  274. package/dist/engine/models/factory.d.ts +2 -0
  275. package/dist/engine/models/factory.js +51 -0
  276. package/dist/engine/models/factory.js.map +1 -0
  277. package/dist/engine/models/index.d.ts +5 -0
  278. package/dist/engine/models/index.js +6 -0
  279. package/dist/engine/models/index.js.map +1 -0
  280. package/dist/engine/models/local-embedding-parallelism.d.ts +15 -0
  281. package/dist/engine/models/local-embedding-parallelism.js +63 -0
  282. package/dist/engine/models/local-embedding-parallelism.js.map +1 -0
  283. package/dist/engine/models/local-embedding-queue.d.ts +12 -0
  284. package/dist/engine/models/local-embedding-queue.js +65 -0
  285. package/dist/engine/models/local-embedding-queue.js.map +1 -0
  286. package/dist/engine/models/ranking.d.ts +19 -0
  287. package/dist/engine/models/ranking.js +2 -0
  288. package/dist/engine/models/ranking.js.map +1 -0
  289. package/dist/engine/models/resolution.d.ts +8 -0
  290. package/dist/engine/models/resolution.js +22 -0
  291. package/dist/engine/models/resolution.js.map +1 -0
  292. package/dist/engine/pipeline/indexing/index.d.ts +14 -0
  293. package/dist/engine/pipeline/indexing/index.js +1219 -0
  294. package/dist/engine/pipeline/indexing/index.js.map +1 -0
  295. package/dist/engine/pipeline/indexing/input-budget.d.ts +2 -0
  296. package/dist/engine/pipeline/indexing/input-budget.js +51 -0
  297. package/dist/engine/pipeline/indexing/input-budget.js.map +1 -0
  298. package/dist/engine/pipeline/indexing/root-paths.d.ts +7 -0
  299. package/dist/engine/pipeline/indexing/root-paths.js +148 -0
  300. package/dist/engine/pipeline/indexing/root-paths.js.map +1 -0
  301. package/dist/engine/pipeline/indexing/scanner/index.d.ts +14 -0
  302. package/dist/engine/pipeline/indexing/scanner/index.js +776 -0
  303. package/dist/engine/pipeline/indexing/scanner/index.js.map +1 -0
  304. package/dist/engine/pipeline/search/index.d.ts +12 -0
  305. package/dist/engine/pipeline/search/index.js +811 -0
  306. package/dist/engine/pipeline/search/index.js.map +1 -0
  307. package/dist/engine/service/index.d.ts +3 -0
  308. package/dist/engine/service/index.js +2 -0
  309. package/dist/engine/service/index.js.map +1 -0
  310. package/dist/engine/service/lexical.d.ts +28 -0
  311. package/dist/engine/service/lexical.js +446 -0
  312. package/dist/engine/service/lexical.js.map +1 -0
  313. package/dist/engine/service/root.d.ts +15 -0
  314. package/dist/engine/service/root.js +57 -0
  315. package/dist/engine/service/root.js.map +1 -0
  316. package/dist/engine/service/structure-enrichment.d.ts +8 -0
  317. package/dist/engine/service/structure-enrichment.js +178 -0
  318. package/dist/engine/service/structure-enrichment.js.map +1 -0
  319. package/dist/engine/service/types.d.ts +212 -0
  320. package/dist/engine/service/types.js +2 -0
  321. package/dist/engine/service/types.js.map +1 -0
  322. package/dist/engine/service/workspace-index.d.ts +26 -0
  323. package/dist/engine/service/workspace-index.js +157 -0
  324. package/dist/engine/service/workspace-index.js.map +1 -0
  325. package/dist/engine/service/zvec-grep.d.ts +15 -0
  326. package/dist/engine/service/zvec-grep.js +1520 -0
  327. package/dist/engine/service/zvec-grep.js.map +1 -0
  328. package/dist/engine/storage/index.d.ts +53 -0
  329. package/dist/engine/storage/index.js +3 -0
  330. package/dist/engine/storage/index.js.map +1 -0
  331. package/dist/engine/storage/layout.d.ts +10 -0
  332. package/dist/engine/storage/layout.js +29 -0
  333. package/dist/engine/storage/layout.js.map +1 -0
  334. package/dist/engine/storage/zvec.d.ts +4 -0
  335. package/dist/engine/storage/zvec.js +976 -0
  336. package/dist/engine/storage/zvec.js.map +1 -0
  337. package/dist/engine/types.d.ts +303 -0
  338. package/dist/engine/types.js +14 -0
  339. package/dist/engine/types.js.map +1 -0
  340. package/dist/engine/utils/atomic-file.d.ts +3 -0
  341. package/dist/engine/utils/atomic-file.js +39 -0
  342. package/dist/engine/utils/atomic-file.js.map +1 -0
  343. package/dist/engine/utils/daemon-lease.d.ts +17 -0
  344. package/dist/engine/utils/daemon-lease.js +175 -0
  345. package/dist/engine/utils/daemon-lease.js.map +1 -0
  346. package/dist/engine/utils/file-selection.d.ts +10 -0
  347. package/dist/engine/utils/file-selection.js +157 -0
  348. package/dist/engine/utils/file-selection.js.map +1 -0
  349. package/dist/engine/utils/glob.d.ts +9 -0
  350. package/dist/engine/utils/glob.js +208 -0
  351. package/dist/engine/utils/glob.js.map +1 -0
  352. package/dist/engine/utils/hash.d.ts +2 -0
  353. package/dist/engine/utils/hash.js +8 -0
  354. package/dist/engine/utils/hash.js.map +1 -0
  355. package/dist/engine/utils/json.d.ts +8 -0
  356. package/dist/engine/utils/json.js +83 -0
  357. package/dist/engine/utils/json.js.map +1 -0
  358. package/dist/engine/utils/lock.d.ts +19 -0
  359. package/dist/engine/utils/lock.js +254 -0
  360. package/dist/engine/utils/lock.js.map +1 -0
  361. package/dist/engine/utils/path.d.ts +4 -0
  362. package/dist/engine/utils/path.js +18 -0
  363. package/dist/engine/utils/path.js.map +1 -0
  364. package/dist/engine/utils/timing.d.ts +19 -0
  365. package/dist/engine/utils/timing.js +87 -0
  366. package/dist/engine/utils/timing.js.map +1 -0
  367. package/dist/index-progress.d.ts +10 -0
  368. package/dist/index-progress.js +159 -0
  369. package/dist/index-progress.js.map +1 -0
  370. package/dist/index.d.ts +6 -0
  371. package/dist/index.js +4 -0
  372. package/dist/index.js.map +1 -0
  373. package/dist/mcp/http-transport.d.ts +27 -0
  374. package/dist/mcp/http-transport.js +182 -0
  375. package/dist/mcp/http-transport.js.map +1 -0
  376. package/dist/mcp/input-normalization.d.ts +38 -0
  377. package/dist/mcp/input-normalization.js +104 -0
  378. package/dist/mcp/input-normalization.js.map +1 -0
  379. package/dist/mcp/progress-heartbeat.d.ts +17 -0
  380. package/dist/mcp/progress-heartbeat.js +46 -0
  381. package/dist/mcp/progress-heartbeat.js.map +1 -0
  382. package/dist/mcp/request-metadata.d.ts +2 -0
  383. package/dist/mcp/request-metadata.js +11 -0
  384. package/dist/mcp/request-metadata.js.map +1 -0
  385. package/dist/mcp/request-state.d.ts +40 -0
  386. package/dist/mcp/request-state.js +210 -0
  387. package/dist/mcp/request-state.js.map +1 -0
  388. package/dist/mcp/result-format.d.ts +13 -0
  389. package/dist/mcp/result-format.js +75 -0
  390. package/dist/mcp/result-format.js.map +1 -0
  391. package/dist/mcp/schemas.d.ts +668 -0
  392. package/dist/mcp/schemas.js +512 -0
  393. package/dist/mcp/schemas.js.map +1 -0
  394. package/dist/mcp/stdio-bridge.d.ts +15 -0
  395. package/dist/mcp/stdio-bridge.js +190 -0
  396. package/dist/mcp/stdio-bridge.js.map +1 -0
  397. package/dist/mcp/tools.d.ts +153 -0
  398. package/dist/mcp/tools.js +488 -0
  399. package/dist/mcp/tools.js.map +1 -0
  400. package/dist/mcp/toolset.d.ts +5 -0
  401. package/dist/mcp/toolset.js +12 -0
  402. package/dist/mcp/toolset.js.map +1 -0
  403. package/dist/observability/trace-context.d.ts +11 -0
  404. package/dist/observability/trace-context.js +138 -0
  405. package/dist/observability/trace-context.js.map +1 -0
  406. package/dist/prompts/zvec-grep-guidance.d.ts +2 -0
  407. package/dist/prompts/zvec-grep-guidance.js +9 -0
  408. package/dist/prompts/zvec-grep-guidance.js.map +1 -0
  409. package/docs/01-agents.md +217 -0
  410. package/docs/02-cli.md +259 -0
  411. package/docs/03-mcp.md +185 -0
  412. package/docs/04-pipeline.md +234 -0
  413. package/docs/05-architecture.md +86 -0
  414. package/docs/06-server.md +279 -0
  415. package/docs/07-embedding.md +227 -0
  416. package/docs/08-roadmap.md +85 -0
  417. package/docs/README.md +53 -0
  418. package/package.json +87 -0
@@ -0,0 +1,522 @@
1
+ const DEFAULT_QWEN_TEXT_EMBEDDING_ENDPOINT = "https://dashscope.aliyuncs.com/compatible-mode/v1/embeddings";
2
+ const DEFAULT_QWEN3_VL_EMBEDDING_ENDPOINT = "https://dashscope.aliyuncs.com/api/v1/services/embeddings/multimodal-embedding/multimodal-embedding";
3
+ export const EMBEDDING_MODEL_CATALOG = {
4
+ "local/embeddinggemma-300m": {
5
+ backend: "llama-cpp",
6
+ reference: "local/embeddinggemma-300m",
7
+ provider: "local",
8
+ model: "embeddinggemma-300m",
9
+ uri: "hf:ggml-org/embeddinggemma-300M-GGUF/embeddinggemma-300M-Q8_0.gguf#0f741b5a6585bd53aeb15cd1372c56f2a0f65e12",
10
+ cacheFile: "hf_ggml-org_embeddinggemma-300M-Q8_0.gguf",
11
+ sources: {
12
+ huggingFace: {
13
+ repo: "ggml-org/embeddinggemma-300M-GGUF",
14
+ revision: "0f741b5a6585bd53aeb15cd1372c56f2a0f65e12",
15
+ },
16
+ modelScope: {
17
+ repo: "ggml-org/embeddinggemma-300M-GGUF",
18
+ revision: "e2fab2963ac0943b1dc320cf0e267bd0e06e2e97",
19
+ },
20
+ },
21
+ artifacts: [
22
+ {
23
+ path: "embeddinggemma-300M-Q8_0.gguf",
24
+ size: 333590944,
25
+ sha256: "b5ce9d77a3fc4b3b39ccb5643c36777911cc4eb46a66962eadfa3f5f60490d63",
26
+ },
27
+ ],
28
+ dimension: 768,
29
+ metric: "cosine",
30
+ format: "embeddinggemma",
31
+ contextSize: 2048,
32
+ maxBatchSize: 16,
33
+ },
34
+ "local/qwen3-embedding-0.6b": {
35
+ backend: "llama-cpp",
36
+ reference: "local/qwen3-embedding-0.6b",
37
+ provider: "local",
38
+ model: "qwen3-embedding-0.6b",
39
+ uri: "hf:Qwen/Qwen3-Embedding-0.6B-GGUF/Qwen3-Embedding-0.6B-Q8_0.gguf#370f27d7550e0def9b39c1f16d3fbaa13aa67728",
40
+ cacheFile: "hf_Qwen_Qwen3-Embedding-0.6B-Q8_0.gguf",
41
+ sources: {
42
+ huggingFace: {
43
+ repo: "Qwen/Qwen3-Embedding-0.6B-GGUF",
44
+ revision: "370f27d7550e0def9b39c1f16d3fbaa13aa67728",
45
+ },
46
+ modelScope: {
47
+ repo: "Qwen/Qwen3-Embedding-0.6B-GGUF",
48
+ revision: "61ae123505fc6fa7f36d372fae1a30bc384241ea",
49
+ },
50
+ },
51
+ artifacts: [
52
+ {
53
+ path: "Qwen3-Embedding-0.6B-Q8_0.gguf",
54
+ size: 639150592,
55
+ sha256: "06507c7b42688469c4e7298b0a1e16deff06caf291cf0a5b278c308249c3e439",
56
+ },
57
+ ],
58
+ dimension: 1024,
59
+ metric: "cosine",
60
+ format: "qwen3",
61
+ contextSize: 8192,
62
+ maxBatchSize: 8,
63
+ },
64
+ "qwen/text-embedding-v4": {
65
+ backend: "qwen",
66
+ kind: "text",
67
+ reference: "qwen/text-embedding-v4",
68
+ provider: "qwen",
69
+ model: "text-embedding-v4",
70
+ dimension: 1024,
71
+ metric: "cosine",
72
+ defaultEndpoint: DEFAULT_QWEN_TEXT_EMBEDDING_ENDPOINT,
73
+ maxBatchSize: 10,
74
+ maxInputTokens: 8192,
75
+ },
76
+ "qwen/qwen3.7-text-embedding": {
77
+ backend: "qwen",
78
+ kind: "text",
79
+ reference: "qwen/qwen3.7-text-embedding",
80
+ provider: "qwen",
81
+ model: "qwen3.7-text-embedding",
82
+ dimension: 1024,
83
+ metric: "cosine",
84
+ defaultEndpoint: DEFAULT_QWEN_TEXT_EMBEDDING_ENDPOINT,
85
+ maxBatchSize: 20,
86
+ maxInputTokens: 128000,
87
+ },
88
+ "qwen/qwen3-vl-embedding": {
89
+ backend: "qwen",
90
+ kind: "multimodal",
91
+ reference: "qwen/qwen3-vl-embedding",
92
+ provider: "qwen",
93
+ model: "qwen3-vl-embedding",
94
+ dimension: 2560,
95
+ metric: "cosine",
96
+ defaultEndpoint: DEFAULT_QWEN3_VL_EMBEDDING_ENDPOINT,
97
+ maxBatchSize: 20,
98
+ maxInputTokens: 32000,
99
+ maxImageBytes: 10 * 1024 * 1024,
100
+ },
101
+ "local/bge-small-en-v1.5": {
102
+ backend: "transformers-js",
103
+ reference: "local/bge-small-en-v1.5",
104
+ provider: "local",
105
+ model: "bge-small-en-v1.5",
106
+ repo: "onnx-community/bge-small-en-v1.5-ONNX",
107
+ revision: "4a9a46c7b88fa408e650a571a1800243f26309bd",
108
+ sources: {
109
+ huggingFace: {
110
+ repo: "onnx-community/bge-small-en-v1.5-ONNX",
111
+ revision: "4a9a46c7b88fa408e650a571a1800243f26309bd",
112
+ },
113
+ modelScope: {
114
+ repo: "onnx-community/bge-small-en-v1.5-ONNX",
115
+ revision: "f246b360b061b613fe0b449f2e14de12f875dad7",
116
+ },
117
+ },
118
+ artifacts: [
119
+ {
120
+ path: "config.json",
121
+ size: 867,
122
+ sha256: "26ad1d93a1ba37422fc25472191cfa230010631fa6a01f9d9f81fa13df2d0917",
123
+ },
124
+ {
125
+ path: "tokenizer.json",
126
+ size: 533603,
127
+ sha256: "ea77de727ef7fd34d177b83b4b1f1d3bb8884c95c90b6554a0adb0b3b65350a9",
128
+ },
129
+ {
130
+ path: "tokenizer_config.json",
131
+ size: 1271,
132
+ sha256: "eebe14d184cfbd65f6a11d2a5ff39385c4044c8a670a89acf1a13331e04faa60",
133
+ },
134
+ {
135
+ path: "onnx/model_q4.onnx",
136
+ size: 132562,
137
+ sha256: "266acb3edd98a1932f876d15bd8f7881a4955d0d53a6d5e79900f788b432de09",
138
+ },
139
+ {
140
+ path: "onnx/model_q4.onnx_data",
141
+ size: 61185024,
142
+ sha256: "77aff1dfb1e0591a40b61a91e5a97796f14c2aff56706e345bef5cbb3613b8cc",
143
+ },
144
+ ],
145
+ dtype: "q4",
146
+ dimension: 384,
147
+ metric: "cosine",
148
+ pooling: "cls",
149
+ normalize: true,
150
+ queryPrefix: "Represent this sentence for searching relevant passages: ",
151
+ maxInputTokens: 512,
152
+ maxBatchSize: 4,
153
+ },
154
+ "local/all-minilm-l6-v2": {
155
+ backend: "transformers-js",
156
+ reference: "local/all-minilm-l6-v2",
157
+ provider: "local",
158
+ model: "all-minilm-l6-v2",
159
+ repo: "onnx-community/all-MiniLM-L6-v2-ONNX",
160
+ revision: "aff7a1dc4e8a1ea593e6ea21e95c22ef0a25966f",
161
+ sources: {
162
+ huggingFace: {
163
+ repo: "onnx-community/all-MiniLM-L6-v2-ONNX",
164
+ revision: "aff7a1dc4e8a1ea593e6ea21e95c22ef0a25966f",
165
+ },
166
+ modelScope: {
167
+ repo: "onnx-community/all-MiniLM-L6-v2-ONNX",
168
+ revision: "e1da369847063d70f2fd772226551865bcab1c2d",
169
+ },
170
+ },
171
+ artifacts: [
172
+ {
173
+ path: "config.json",
174
+ size: 794,
175
+ sha256: "fe5da868b77bdb104140822a5af0837cb6450ad6de8ff3dfcc8dd44ddd3e3ae7",
176
+ },
177
+ {
178
+ path: "tokenizer.json",
179
+ size: 533808,
180
+ sha256: "07805d116826679de90b4edeb2222269c4b8753bc0981be4399f732b2708e904",
181
+ },
182
+ {
183
+ path: "tokenizer_config.json",
184
+ size: 1463,
185
+ sha256: "e10bb633ba0d7f69ed342ae7de607f36b39ce53b455fbda69c71700bf57e6f66",
186
+ },
187
+ {
188
+ path: "onnx/model_q4.onnx",
189
+ size: 69663,
190
+ sha256: "e4dcb918111189b7686147e309379832fce83d4ecbf17c395961749b5788c786",
191
+ },
192
+ {
193
+ path: "onnx/model_q4.onnx_data",
194
+ size: 54429696,
195
+ sha256: "56fb7a55115e900196115a74e399beb45c2f41ae00b99525d46fb52935c4ee2a",
196
+ },
197
+ ],
198
+ dtype: "q4",
199
+ dimension: 384,
200
+ metric: "cosine",
201
+ pooling: "mean",
202
+ normalize: true,
203
+ maxInputTokens: 256,
204
+ maxBatchSize: 4,
205
+ },
206
+ "local/potion-retrieval-32m": {
207
+ backend: "model2vec",
208
+ reference: "local/potion-retrieval-32m",
209
+ provider: "local",
210
+ model: "potion-retrieval-32m",
211
+ repo: "minishlab/potion-retrieval-32M",
212
+ revision: "6fc8051fab2a1e0ee76689cf08c853792ac285e7",
213
+ sources: {
214
+ huggingFace: {
215
+ repo: "minishlab/potion-retrieval-32M",
216
+ revision: "6fc8051fab2a1e0ee76689cf08c853792ac285e7",
217
+ },
218
+ modelScope: {
219
+ repo: "minishlab/potion-retrieval-32M",
220
+ revision: "33da23fc75cb732b5370bf25adde3db74b0d65b3",
221
+ },
222
+ },
223
+ artifacts: [
224
+ {
225
+ path: "model.safetensors",
226
+ size: 129210456,
227
+ sha256: "07609e5bd33aad37900b3fd62f4ec96f6daec88ca4d46b9d8b928bfababf6ea0",
228
+ },
229
+ {
230
+ path: "tokenizer.json",
231
+ size: 1493150,
232
+ sha256: "7d75cbc54318138807c401b0f0c9721117c628b39de8e8e0edb6cb17e0ee7d18",
233
+ },
234
+ ],
235
+ modelFile: "model.safetensors",
236
+ embeddingTensor: "embeddings",
237
+ tokenizerFile: "tokenizer.json",
238
+ dimension: 512,
239
+ metric: "cosine",
240
+ normalize: true,
241
+ maxInputTokens: 1024,
242
+ maxBatchSize: 256,
243
+ defaultConcurrency: 2,
244
+ },
245
+ "local/potion-multilingual-128m": {
246
+ backend: "model2vec",
247
+ reference: "local/potion-multilingual-128m",
248
+ provider: "local",
249
+ model: "potion-multilingual-128m",
250
+ repo: "minishlab/potion-multilingual-128M",
251
+ revision: "73908c3438cf03b6a01bcb9611d62b23d0726f08",
252
+ sources: {
253
+ huggingFace: {
254
+ repo: "minishlab/potion-multilingual-128M",
255
+ revision: "73908c3438cf03b6a01bcb9611d62b23d0726f08",
256
+ },
257
+ modelScope: {
258
+ repo: "minishlab/potion-multilingual-128M",
259
+ revision: "e8524678123f281add99f9745ac33d6604434dd7",
260
+ },
261
+ },
262
+ artifacts: [
263
+ {
264
+ path: "model.safetensors",
265
+ size: 512361560,
266
+ sha256: "14b5eb39cb4ce5666da8ad1f3dc6be4346e9b2d601c073302fa0a31bf7943397",
267
+ },
268
+ {
269
+ path: "tokenizer.json",
270
+ size: 18616131,
271
+ sha256: "19f1909063da3cfe3bd83a782381f040dccea475f4816de11116444a73e1b6a1",
272
+ },
273
+ ],
274
+ modelFile: "model.safetensors",
275
+ embeddingTensor: "embeddings",
276
+ tokenizerFile: "tokenizer.json",
277
+ dimension: 256,
278
+ metric: "cosine",
279
+ normalize: true,
280
+ maxInputTokens: 1024,
281
+ maxBatchSize: 256,
282
+ defaultConcurrency: 2,
283
+ },
284
+ "local/potion-code-16m-v2": {
285
+ backend: "model2vec",
286
+ reference: "local/potion-code-16m-v2",
287
+ provider: "local",
288
+ model: "potion-code-16m-v2",
289
+ repo: "minishlab/potion-code-16M-v2",
290
+ revision: "e9d2a44ca6a05ac6685f3b23709ea57eb7352d5b",
291
+ sources: {
292
+ huggingFace: {
293
+ repo: "minishlab/potion-code-16M-v2",
294
+ revision: "e9d2a44ca6a05ac6685f3b23709ea57eb7352d5b",
295
+ },
296
+ modelScope: {
297
+ repo: "minishlab/potion-code-16M-v2",
298
+ revision: "3e922fde18f43b8db69f3381b6d468738d3dd2d7",
299
+ },
300
+ },
301
+ artifacts: [
302
+ {
303
+ path: "model.safetensors",
304
+ size: 32490072,
305
+ sha256: "75cf7a6c2171b230ad19b1e7d8e0b1aee86da5a02af8e7cacedd9921d227623c",
306
+ },
307
+ {
308
+ path: "tokenizer.json",
309
+ size: 1024340,
310
+ sha256: "107bbdcbad4bff1d299b7a4c3a2fb17c52890688b7dd0e4c9deab79d3c4f3d45",
311
+ },
312
+ ],
313
+ modelFile: "model.safetensors",
314
+ embeddingTensor: "embeddings",
315
+ tokenizerFile: "tokenizer.json",
316
+ dimension: 256,
317
+ metric: "cosine",
318
+ normalize: true,
319
+ maxInputTokens: 1024,
320
+ maxBatchSize: 256,
321
+ defaultConcurrency: 2,
322
+ },
323
+ "local/multilingual-e5-small": {
324
+ backend: "transformers-js",
325
+ reference: "local/multilingual-e5-small",
326
+ provider: "local",
327
+ model: "multilingual-e5-small",
328
+ repo: "Xenova/multilingual-e5-small",
329
+ revision: "761b726dd34fb83930e26aab4e9ac3899aa1fa78",
330
+ sources: {
331
+ huggingFace: {
332
+ repo: "Xenova/multilingual-e5-small",
333
+ revision: "761b726dd34fb83930e26aab4e9ac3899aa1fa78",
334
+ },
335
+ modelScope: {
336
+ repo: "Xenova/multilingual-e5-small",
337
+ revision: "252d0dcb679dda2c7b6fd5bbfed15df3c7feaebf",
338
+ },
339
+ },
340
+ artifacts: [
341
+ {
342
+ path: "config.json",
343
+ size: 658,
344
+ sha256: "cb99455288675345e1a4f411438d5d0adbba5fbd3a67ea4fb03c015433b996c1",
345
+ },
346
+ {
347
+ path: "tokenizer.json",
348
+ size: 17082730,
349
+ sha256: "0b44a9d7b51c3c62626640cda0e2c2f70fdacdc25bbbd68038369d14ebdf4c39",
350
+ },
351
+ {
352
+ path: "tokenizer_config.json",
353
+ size: 443,
354
+ sha256: "a1d6bc8734a6f635dc158508bef000f8e2e5a759c7d92f984b2c86e5ff53425b",
355
+ },
356
+ {
357
+ path: "onnx/model_quantized.onnx",
358
+ size: 118308185,
359
+ sha256: "f80102d3f2a1229f387d3c81909990d8945513e347b0eab049f7de3c6f98c193",
360
+ },
361
+ ],
362
+ dtype: "q8",
363
+ dimension: 384,
364
+ metric: "cosine",
365
+ pooling: "mean",
366
+ normalize: true,
367
+ queryPrefix: "query: ",
368
+ documentPrefix: "passage: ",
369
+ maxInputTokens: 512,
370
+ maxBatchSize: 4,
371
+ },
372
+ "local/jina-embeddings-v2-base-code": {
373
+ backend: "transformers-js",
374
+ reference: "local/jina-embeddings-v2-base-code",
375
+ provider: "local",
376
+ model: "jina-embeddings-v2-base-code",
377
+ repo: "jinaai/jina-embeddings-v2-base-code",
378
+ revision: "516f4baf13dec4ddddda8631e019b5737c8bc250",
379
+ sources: {
380
+ huggingFace: {
381
+ repo: "jinaai/jina-embeddings-v2-base-code",
382
+ revision: "516f4baf13dec4ddddda8631e019b5737c8bc250",
383
+ },
384
+ modelScope: {
385
+ repo: "jinaai/jina-embeddings-v2-base-code",
386
+ revision: "91aa0a6aa801c408149324e32e8cd43f8502da8f",
387
+ },
388
+ },
389
+ artifacts: [
390
+ {
391
+ path: "config.json",
392
+ size: 1216,
393
+ sha256: "e426aa684c7f9a95c5f020aa855faf93a24f065f5fad0c9e17b124670cabdea6",
394
+ },
395
+ {
396
+ path: "tokenizer.json",
397
+ size: 2561316,
398
+ sha256: "b01c78a902aa4facb2f47f95449f48e2f7bbfea5d2472ee2f6ce92323c6f86e5",
399
+ },
400
+ {
401
+ path: "tokenizer_config.json",
402
+ size: 493,
403
+ sha256: "f477aeb15ff9f78d3c1ddf2361d2b0b8b20cf55220f839f29a37f3a18efddd89",
404
+ },
405
+ {
406
+ path: "onnx/model_quantized.onnx",
407
+ size: 161895621,
408
+ sha256: "ed45870251c9f0cf656e78aab0d37a23489066df8a222bb1c8caf8a45f2cb16d",
409
+ },
410
+ ],
411
+ dtype: "q8",
412
+ dimension: 768,
413
+ metric: "cosine",
414
+ pooling: "mean",
415
+ normalize: true,
416
+ maxInputTokens: 8192,
417
+ maxBatchSize: 2,
418
+ },
419
+ "local/gte-modernbert-base": {
420
+ backend: "transformers-js",
421
+ reference: "local/gte-modernbert-base",
422
+ provider: "local",
423
+ model: "gte-modernbert-base",
424
+ repo: "Alibaba-NLP/gte-modernbert-base",
425
+ revision: "e7f32e3c00f91d699e8c43b53106206bcc72bb22",
426
+ sources: {
427
+ huggingFace: {
428
+ repo: "Alibaba-NLP/gte-modernbert-base",
429
+ revision: "e7f32e3c00f91d699e8c43b53106206bcc72bb22",
430
+ },
431
+ modelScope: {
432
+ repo: "iic/gte-modernbert-base",
433
+ revision: "678f4ed93760af288132f4f9dc5b6daebdc48777",
434
+ },
435
+ },
436
+ artifacts: [
437
+ {
438
+ path: "config.json",
439
+ size: 1184,
440
+ sha256: "8ba54dc3d35d7194f5178a4194b649f146753e02dabd22bdca5c5cbac15069ed",
441
+ },
442
+ {
443
+ path: "tokenizer.json",
444
+ size: 3583228,
445
+ sha256: "6c8aaa9a542084f2457eab775d4eeb51f92a70c0fd9de28d5edb0ddec3c08d30",
446
+ },
447
+ {
448
+ path: "tokenizer_config.json",
449
+ size: 20867,
450
+ sha256: "9654072f7c873161814043cf08cb5ed72f71d0b935abcd4e267935cb34352c21",
451
+ },
452
+ {
453
+ path: "onnx/model_q4.onnx",
454
+ size: 224152761,
455
+ sha256: "5d1278a1ba749c06b82f9a2f65c2c1c5765d36f2eb88b4888de62ff12b0724a2",
456
+ },
457
+ ],
458
+ dtype: "q4",
459
+ dimension: 768,
460
+ metric: "cosine",
461
+ pooling: "cls",
462
+ normalize: true,
463
+ maxInputTokens: 8192,
464
+ maxBatchSize: 2,
465
+ },
466
+ "local/nomic-embed-text-v1.5": {
467
+ backend: "transformers-js",
468
+ reference: "local/nomic-embed-text-v1.5",
469
+ provider: "local",
470
+ model: "nomic-embed-text-v1.5",
471
+ repo: "nomic-ai/nomic-embed-text-v1.5",
472
+ revision: "e9b6763023c676ca8431644204f50c2b100d9aab",
473
+ sources: {
474
+ huggingFace: {
475
+ repo: "nomic-ai/nomic-embed-text-v1.5",
476
+ revision: "e9b6763023c676ca8431644204f50c2b100d9aab",
477
+ },
478
+ modelScope: {
479
+ repo: "nomic-ai/nomic-embed-text-v1.5",
480
+ revision: "c6fb77fdf73531ee8319b34e46f7e749b59e74e8",
481
+ },
482
+ },
483
+ artifacts: [
484
+ {
485
+ path: "config.json",
486
+ size: 2538,
487
+ sha256: "9ab00bd92cee80a569f708140b7b6c1661a65891ff3765b1519e181ba2f2c92b",
488
+ },
489
+ {
490
+ path: "tokenizer.json",
491
+ size: 711396,
492
+ sha256: "d241a60d5e8f04cc1b2b3e9ef7a4921b27bf526d9f6050ab90f9267a1f9e5c66",
493
+ },
494
+ {
495
+ path: "tokenizer_config.json",
496
+ size: 1191,
497
+ sha256: "d7e0000bcc80134debd2222220427e6bf5fa20a669f40a0d0d1409cc18e0a9bc",
498
+ },
499
+ {
500
+ path: "onnx/model_q4.onnx",
501
+ size: 165113221,
502
+ sha256: "314976b7b9fba83283f9c8a29ee680a159fa485f52104e3fa39d3d5858337003",
503
+ },
504
+ ],
505
+ dtype: "q4",
506
+ dimension: 768,
507
+ metric: "cosine",
508
+ pooling: "mean",
509
+ normalize: true,
510
+ queryPrefix: "search_query: ",
511
+ documentPrefix: "search_document: ",
512
+ maxInputTokens: 8192,
513
+ maxBatchSize: 2,
514
+ },
515
+ };
516
+ export function listEmbeddingModels() {
517
+ return Object.values(EMBEDDING_MODEL_CATALOG);
518
+ }
519
+ export function getEmbeddingModelCatalogEntry(reference) {
520
+ return EMBEDDING_MODEL_CATALOG[reference];
521
+ }
522
+ //# sourceMappingURL=catalog.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"catalog.js","sourceRoot":"","sources":["../../../src/engine/models/catalog.ts"],"names":[],"mappings":"AAAA,MAAM,oCAAoC,GACxC,8DAA8D,CAAC;AACjE,MAAM,mCAAmC,GACvC,qGAAqG,CAAC;AAExG,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,2BAA2B,EAAE;QAC3B,OAAO,EAAE,WAAW;QACpB,SAAS,EAAE,2BAA2B;QACtC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,qBAAqB;QAC5B,GAAG,EAAE,6GAA6G;QAClH,SAAS,EAAE,2CAA2C;QACtD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,mCAAmC;gBACzC,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,mCAAmC;gBACzC,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,+BAA+B;gBACrC,IAAI,EAAE,SAAS;gBACf,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,MAAM,EAAE,gBAAgB;QACxB,WAAW,EAAE,IAAI;QACjB,YAAY,EAAE,EAAE;KACjB;IAED,4BAA4B,EAAE;QAC5B,OAAO,EAAE,WAAW;QACpB,SAAS,EAAE,4BAA4B;QACvC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,sBAAsB;QAC7B,GAAG,EAAE,2GAA2G;QAChH,SAAS,EAAE,wCAAwC;QACnD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,gCAAgC;gBACtC,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,gCAAgC;gBACtC,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,gCAAgC;gBACtC,IAAI,EAAE,SAAS;gBACf,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,SAAS,EAAE,IAAI;QACf,MAAM,EAAE,QAAQ;QAChB,MAAM,EAAE,OAAO;QACf,WAAW,EAAE,IAAI;QACjB,YAAY,EAAE,CAAC;KAChB;IAED,wBAAwB,EAAE;QACxB,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,MAAM;QACZ,SAAS,EAAE,wBAAwB;QACnC,QAAQ,EAAE,MAAM;QAChB,KAAK,EAAE,mBAAmB;QAC1B,SAAS,EAAE,IAAI;QACf,MAAM,EAAE,QAAQ;QAChB,eAAe,EAAE,oCAAoC;QACrD,YAAY,EAAE,EAAE;QAChB,cAAc,EAAE,IAAI;KACrB;IAED,6BAA6B,EAAE;QAC7B,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,MAAM;QACZ,SAAS,EAAE,6BAA6B;QACxC,QAAQ,EAAE,MAAM;QAChB,KAAK,EAAE,wBAAwB;QAC/B,SAAS,EAAE,IAAI;QACf,MAAM,EAAE,QAAQ;QAChB,eAAe,EAAE,oCAAoC;QACrD,YAAY,EAAE,EAAE;QAChB,cAAc,EAAE,MAAM;KACvB;IAED,yBAAyB,EAAE;QACzB,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,YAAY;QAClB,SAAS,EAAE,yBAAyB;QACpC,QAAQ,EAAE,MAAM;QAChB,KAAK,EAAE,oBAAoB;QAC3B,SAAS,EAAE,IAAI;QACf,MAAM,EAAE,QAAQ;QAChB,eAAe,EAAE,mCAAmC;QACpD,YAAY,EAAE,EAAE;QAChB,cAAc,EAAE,KAAK;QACrB,aAAa,EAAE,EAAE,GAAG,IAAI,GAAG,IAAI;KAChC;IAED,yBAAyB,EAAE;QACzB,OAAO,EAAE,iBAAiB;QAC1B,SAAS,EAAE,yBAAyB;QACpC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,mBAAmB;QAC1B,IAAI,EAAE,uCAAuC;QAC7C,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,uCAAuC;gBAC7C,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,uCAAuC;gBAC7C,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,GAAG;gBACT,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,MAAM;gBACZ,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE,IAAI;gBACV,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,oBAAoB;gBAC1B,IAAI,EAAE,MAAM;gBACZ,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,yBAAyB;gBAC/B,IAAI,EAAE,QAAQ;gBACd,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,IAAI;QACf,WAAW,EAAE,2DAA2D;QACxE,cAAc,EAAE,GAAG;QACnB,YAAY,EAAE,CAAC;KAChB;IAED,wBAAwB,EAAE;QACxB,OAAO,EAAE,iBAAiB;QAC1B,SAAS,EAAE,wBAAwB;QACnC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,kBAAkB;QACzB,IAAI,EAAE,sCAAsC;QAC5C,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,sCAAsC;gBAC5C,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,sCAAsC;gBAC5C,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,GAAG;gBACT,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,MAAM;gBACZ,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE,IAAI;gBACV,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,oBAAoB;gBAC1B,IAAI,EAAE,KAAK;gBACX,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,yBAAyB;gBAC/B,IAAI,EAAE,QAAQ;gBACd,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,OAAO,EAAE,MAAM;QACf,SAAS,EAAE,IAAI;QACf,cAAc,EAAE,GAAG;QACnB,YAAY,EAAE,CAAC;KAChB;IAED,4BAA4B,EAAE;QAC5B,OAAO,EAAE,WAAW;QACpB,SAAS,EAAE,4BAA4B;QACvC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,sBAAsB;QAC7B,IAAI,EAAE,gCAAgC;QACtC,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,gCAAgC;gBACtC,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,gCAAgC;gBACtC,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,mBAAmB;gBACzB,IAAI,EAAE,SAAS;gBACf,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,OAAO;gBACb,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,SAAS,EAAE,mBAAmB;QAC9B,eAAe,EAAE,YAAY;QAC7B,aAAa,EAAE,gBAAgB;QAC/B,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,IAAI;QACf,cAAc,EAAE,IAAI;QACpB,YAAY,EAAE,GAAG;QACjB,kBAAkB,EAAE,CAAC;KACtB;IAED,gCAAgC,EAAE;QAChC,OAAO,EAAE,WAAW;QACpB,SAAS,EAAE,gCAAgC;QAC3C,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,0BAA0B;QACjC,IAAI,EAAE,oCAAoC;QAC1C,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,oCAAoC;gBAC1C,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,oCAAoC;gBAC1C,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,mBAAmB;gBACzB,IAAI,EAAE,SAAS;gBACf,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,QAAQ;gBACd,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,SAAS,EAAE,mBAAmB;QAC9B,eAAe,EAAE,YAAY;QAC7B,aAAa,EAAE,gBAAgB;QAC/B,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,IAAI;QACf,cAAc,EAAE,IAAI;QACpB,YAAY,EAAE,GAAG;QACjB,kBAAkB,EAAE,CAAC;KACtB;IAED,0BAA0B,EAAE;QAC1B,OAAO,EAAE,WAAW;QACpB,SAAS,EAAE,0BAA0B;QACrC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,oBAAoB;QAC3B,IAAI,EAAE,8BAA8B;QACpC,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,8BAA8B;gBACpC,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,8BAA8B;gBACpC,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,mBAAmB;gBACzB,IAAI,EAAE,QAAQ;gBACd,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,OAAO;gBACb,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,SAAS,EAAE,mBAAmB;QAC9B,eAAe,EAAE,YAAY;QAC7B,aAAa,EAAE,gBAAgB;QAC/B,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,IAAI;QACf,cAAc,EAAE,IAAI;QACpB,YAAY,EAAE,GAAG;QACjB,kBAAkB,EAAE,CAAC;KACtB;IAED,6BAA6B,EAAE;QAC7B,OAAO,EAAE,iBAAiB;QAC1B,SAAS,EAAE,6BAA6B;QACxC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,uBAAuB;QAC9B,IAAI,EAAE,8BAA8B;QACpC,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,8BAA8B;gBACpC,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,8BAA8B;gBACpC,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,GAAG;gBACT,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,QAAQ;gBACd,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE,GAAG;gBACT,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,2BAA2B;gBACjC,IAAI,EAAE,SAAS;gBACf,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,OAAO,EAAE,MAAM;QACf,SAAS,EAAE,IAAI;QACf,WAAW,EAAE,SAAS;QACtB,cAAc,EAAE,WAAW;QAC3B,cAAc,EAAE,GAAG;QACnB,YAAY,EAAE,CAAC;KAChB;IAED,oCAAoC,EAAE;QACpC,OAAO,EAAE,iBAAiB;QAC1B,SAAS,EAAE,oCAAoC;QAC/C,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,8BAA8B;QACrC,IAAI,EAAE,qCAAqC;QAC3C,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,qCAAqC;gBAC3C,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,qCAAqC;gBAC3C,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,IAAI;gBACV,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,OAAO;gBACb,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE,GAAG;gBACT,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,2BAA2B;gBACjC,IAAI,EAAE,SAAS;gBACf,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,OAAO,EAAE,MAAM;QACf,SAAS,EAAE,IAAI;QACf,cAAc,EAAE,IAAI;QACpB,YAAY,EAAE,CAAC;KAChB;IAED,2BAA2B,EAAE;QAC3B,OAAO,EAAE,iBAAiB;QAC1B,SAAS,EAAE,2BAA2B;QACtC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,qBAAqB;QAC5B,IAAI,EAAE,iCAAiC;QACvC,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,iCAAiC;gBACvC,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,yBAAyB;gBAC/B,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,IAAI;gBACV,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,OAAO;gBACb,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE,KAAK;gBACX,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,oBAAoB;gBAC1B,IAAI,EAAE,SAAS;gBACf,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,OAAO,EAAE,KAAK;QACd,SAAS,EAAE,IAAI;QACf,cAAc,EAAE,IAAI;QACpB,YAAY,EAAE,CAAC;KAChB;IAED,6BAA6B,EAAE;QAC7B,OAAO,EAAE,iBAAiB;QAC1B,SAAS,EAAE,6BAA6B;QACxC,QAAQ,EAAE,OAAO;QACjB,KAAK,EAAE,uBAAuB;QAC9B,IAAI,EAAE,gCAAgC;QACtC,QAAQ,EAAE,0CAA0C;QACpD,OAAO,EAAE;YACP,WAAW,EAAE;gBACX,IAAI,EAAE,gCAAgC;gBACtC,QAAQ,EAAE,0CAA0C;aACrD;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,gCAAgC;gBACtC,QAAQ,EAAE,0CAA0C;aACrD;SACF;QACD,SAAS,EAAE;YACT;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,IAAI;gBACV,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,gBAAgB;gBACtB,IAAI,EAAE,MAAM;gBACZ,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE,IAAI;gBACV,MAAM,EACJ,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,oBAAoB;gBAC1B,IAAI,EAAE,SAAS;gBACf,MAAM,EACJ,kEAAkE;aACrE;SACF;QACD,KAAK,EAAE,IAAI;QACX,SAAS,EAAE,GAAG;QACd,MAAM,EAAE,QAAQ;QAChB,OAAO,EAAE,MAAM;QACf,SAAS,EAAE,IAAI;QACf,WAAW,EAAE,gBAAgB;QAC7B,cAAc,EAAE,mBAAmB;QACnC,cAAc,EAAE,IAAI;QACpB,YAAY,EAAE,CAAC;KAChB;CACO,CAAC;AAqCX,MAAM,UAAU,mBAAmB;IACjC,OAAO,MAAM,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC3C,SAAiB;IAEjB,OAAO,uBAAuB,CAAC,SAAoC,CAAC,CAAC;AACvE,CAAC"}
@@ -0,0 +1,16 @@
1
+ import type { EmbeddingModelProgress } from "./embeddings.js";
2
+ export type ModelArtifactDownloadProgress = {
3
+ artifact: string;
4
+ downloadedBytes: number;
5
+ };
6
+ export type ModelDownloadProgressReporter = {
7
+ start(): void;
8
+ setDownloadPlan(artifacts: readonly {
9
+ path: string;
10
+ size: number;
11
+ }[]): void;
12
+ report(progress: ModelArtifactDownloadProgress): void;
13
+ warning(message: string): boolean;
14
+ finish(): void;
15
+ };
16
+ export declare function createModelDownloadProgressReporter(model: string, onProgress?: (progress: EmbeddingModelProgress) => void): ModelDownloadProgressReporter;
@@ -0,0 +1,47 @@
1
+ export function createModelDownloadProgressReporter(model, onProgress) {
2
+ const artifacts = new Map();
3
+ const reportDownload = () => {
4
+ const values = [...artifacts.values()];
5
+ const downloadedBytes = values.reduce((sum, artifact) => sum + artifact.downloadedBytes, 0);
6
+ const totalBytes = values.reduce((sum, artifact) => sum + artifact.totalBytes, 0);
7
+ onProgress?.({
8
+ stage: "downloading",
9
+ model,
10
+ downloadedBytes,
11
+ totalBytes,
12
+ });
13
+ };
14
+ return {
15
+ start() {
16
+ onProgress?.({ stage: "preparing", model });
17
+ },
18
+ setDownloadPlan(plannedArtifacts) {
19
+ artifacts.clear();
20
+ for (const artifact of plannedArtifacts) {
21
+ artifacts.set(artifact.path, {
22
+ downloadedBytes: 0,
23
+ totalBytes: artifact.size,
24
+ });
25
+ }
26
+ },
27
+ report(progress) {
28
+ const artifact = artifacts.get(progress.artifact);
29
+ if (!artifact) {
30
+ return;
31
+ }
32
+ artifact.downloadedBytes = progress.downloadedBytes;
33
+ reportDownload();
34
+ },
35
+ warning(message) {
36
+ if (!onProgress) {
37
+ return false;
38
+ }
39
+ onProgress({ stage: "warning", model, message });
40
+ return true;
41
+ },
42
+ finish() {
43
+ onProgress?.({ stage: "ready", model });
44
+ },
45
+ };
46
+ }
47
+ //# sourceMappingURL=download-progress.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"download-progress.js","sourceRoot":"","sources":["../../../src/engine/models/download-progress.ts"],"names":[],"mappings":"AAeA,MAAM,UAAU,mCAAmC,CACjD,KAAa,EACb,UAAuD;IAEvD,MAAM,SAAS,GAAG,IAAI,GAAG,EAGtB,CAAC;IAEJ,MAAM,cAAc,GAAG,GAAS,EAAE;QAChC,MAAM,MAAM,GAAG,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;QACvC,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CACnC,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE,CAAC,GAAG,GAAG,QAAQ,CAAC,eAAe,EACjD,CAAC,CACF,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAC9B,CAAC,GAAG,EAAE,QAAQ,EAAE,EAAE,CAAC,GAAG,GAAG,QAAQ,CAAC,UAAU,EAC5C,CAAC,CACF,CAAC;QACF,UAAU,EAAE,CAAC;YACX,KAAK,EAAE,aAAa;YACpB,KAAK;YACL,eAAe;YACf,UAAU;SACX,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,OAAO;QACL,KAAK;YACH,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,eAAe,CAAC,gBAAgB;YAC9B,SAAS,CAAC,KAAK,EAAE,CAAC;YAClB,KAAK,MAAM,QAAQ,IAAI,gBAAgB,EAAE,CAAC;gBACxC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE;oBAC3B,eAAe,EAAE,CAAC;oBAClB,UAAU,EAAE,QAAQ,CAAC,IAAI;iBAC1B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,MAAM,CAAC,QAAQ;YACb,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAClD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;YACT,CAAC;YACD,QAAQ,CAAC,eAAe,GAAG,QAAQ,CAAC,eAAe,CAAC;YACpD,cAAc,EAAE,CAAC;QACnB,CAAC;QACD,OAAO,CAAC,OAAO;YACb,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO,KAAK,CAAC;YACf,CAAC;YACD,UAAU,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACjD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,MAAM;YACJ,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1C,CAAC;KACF,CAAC;AACJ,CAAC"}
@@ -0,0 +1,73 @@
1
+ import type { Content, ContentKind } from "../types.js";
2
+ export type CreateEmbeddingModelOptions = {
3
+ apiKey?: string;
4
+ endpoint?: string;
5
+ modelCacheDir?: string;
6
+ device?: "auto" | "cpu" | "metal" | "vulkan" | "cuda";
7
+ /** Explicit local Transformer/llama instance limit; index callers resolve configuration. */
8
+ embeddingConcurrency?: number;
9
+ };
10
+ export declare const EmbeddingPurpose: {
11
+ readonly Document: "document";
12
+ readonly Query: "query";
13
+ };
14
+ export type EmbeddingPurpose = (typeof EmbeddingPurpose)[keyof typeof EmbeddingPurpose];
15
+ export type EmbeddingOptions = {
16
+ purpose?: EmbeddingPurpose;
17
+ signal?: AbortSignal;
18
+ onProgress?: (progress: EmbeddingModelProgress) => void;
19
+ };
20
+ export type EmbeddingModelProgress = {
21
+ stage: "preparing" | "ready";
22
+ model: string;
23
+ } | {
24
+ stage: "downloading";
25
+ model: string;
26
+ downloadedBytes?: number;
27
+ totalBytes?: number;
28
+ } | {
29
+ stage: "warning";
30
+ model: string;
31
+ message: string;
32
+ };
33
+ export type EmbeddingResult = {
34
+ vectors: number[][];
35
+ truncated: number[];
36
+ };
37
+ export type EmbeddingModelInfo = Readonly<{
38
+ reference: string;
39
+ provider: string;
40
+ name: string;
41
+ dimension: number;
42
+ metric: "cosine" | "dot" | "euclidean";
43
+ endpoint?: string;
44
+ defaultConcurrency?: number;
45
+ inputKinds: readonly ContentKind[];
46
+ limits: Readonly<{
47
+ maxBatchSize: number;
48
+ /** Maximum embed() batches the runtime executes at once. */
49
+ maxConcurrentBatches?: number;
50
+ maxInputTokens?: number;
51
+ maxImageBytes?: number;
52
+ }>;
53
+ }>;
54
+ export interface EmbeddingModel {
55
+ readonly info: EmbeddingModelInfo;
56
+ /** Prepare local resources before indexing queues any embedding batches. */
57
+ prepare?(options?: Pick<EmbeddingOptions, "signal" | "onProgress">): Promise<void>;
58
+ embed(contents: readonly Content[], options?: EmbeddingOptions): Promise<EmbeddingResult>;
59
+ dispose(): Promise<void>;
60
+ }
61
+ export type NormalizedEmbeddingOptions = {
62
+ purpose: EmbeddingPurpose;
63
+ signal?: AbortSignal;
64
+ onProgress?: (progress: EmbeddingModelProgress) => void;
65
+ };
66
+ export declare abstract class BaseEmbeddingModel implements EmbeddingModel {
67
+ abstract readonly info: EmbeddingModelInfo;
68
+ embed(contents: readonly Content[], options?: EmbeddingOptions): Promise<EmbeddingResult>;
69
+ dispose(): Promise<void>;
70
+ protected abstract doEmbed(contents: readonly Content[], options: NormalizedEmbeddingOptions): Promise<EmbeddingResult>;
71
+ private validateContents;
72
+ private validateResult;
73
+ }