vectra 0.12.2 → 0.14.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 (392) hide show
  1. package/LICENSE +1 -1
  2. package/README.md +92 -100
  3. package/bin/vectra.js +3 -0
  4. package/lib/BrowserWebFetcher.d.ts +75 -0
  5. package/lib/BrowserWebFetcher.d.ts.map +1 -0
  6. package/lib/BrowserWebFetcher.js +290 -0
  7. package/lib/BrowserWebFetcher.js.map +1 -0
  8. package/lib/FileFetcher.d.ts +5 -0
  9. package/lib/FileFetcher.d.ts.map +1 -0
  10. package/lib/FileFetcher.js +89 -0
  11. package/lib/FileFetcher.js.map +1 -0
  12. package/lib/FileFetcher.spec.d.ts +2 -0
  13. package/lib/FileFetcher.spec.d.ts.map +1 -0
  14. package/lib/FileFetcher.spec.js +244 -0
  15. package/lib/FileFetcher.spec.js.map +1 -0
  16. package/lib/FolderWatcher.d.ts +91 -0
  17. package/lib/FolderWatcher.d.ts.map +1 -0
  18. package/lib/FolderWatcher.js +304 -0
  19. package/lib/FolderWatcher.js.map +1 -0
  20. package/lib/FolderWatcher.spec.d.ts +2 -0
  21. package/lib/FolderWatcher.spec.d.ts.map +1 -0
  22. package/lib/FolderWatcher.spec.js +308 -0
  23. package/lib/FolderWatcher.spec.js.map +1 -0
  24. package/lib/GPT3Tokenizer.d.ts +9 -0
  25. package/lib/GPT3Tokenizer.spec.d.ts +2 -0
  26. package/lib/GPT3Tokenizer.spec.d.ts.map +1 -0
  27. package/lib/GPT3Tokenizer.spec.js +45 -0
  28. package/lib/GPT3Tokenizer.spec.js.map +1 -0
  29. package/lib/ItemSelector.d.ts +41 -0
  30. package/lib/ItemSelector.d.ts.map +1 -0
  31. package/lib/ItemSelector.js +179 -0
  32. package/lib/ItemSelector.js.map +1 -0
  33. package/lib/ItemSelector.spec.d.ts +2 -0
  34. package/lib/ItemSelector.spec.d.ts.map +1 -0
  35. package/lib/ItemSelector.spec.js +204 -0
  36. package/lib/ItemSelector.spec.js.map +1 -0
  37. package/lib/LocalDocument.d.ts +54 -0
  38. package/lib/LocalDocument.d.ts.map +1 -1
  39. package/lib/LocalDocument.js +116 -0
  40. package/lib/LocalDocument.js.map +1 -0
  41. package/lib/LocalDocument.spec.d.ts +2 -0
  42. package/lib/LocalDocument.spec.d.ts.map +1 -0
  43. package/lib/LocalDocument.spec.js +214 -0
  44. package/lib/LocalDocument.spec.js.map +1 -0
  45. package/lib/LocalDocumentIndex.d.ts +152 -0
  46. package/lib/LocalDocumentIndex.d.ts.map +1 -1
  47. package/lib/LocalDocumentIndex.js +420 -0
  48. package/lib/LocalDocumentIndex.js.map +1 -0
  49. package/lib/LocalDocumentIndex.spec.d.ts +2 -0
  50. package/lib/LocalDocumentIndex.spec.d.ts.map +1 -0
  51. package/lib/LocalDocumentIndex.spec.js +494 -0
  52. package/lib/LocalDocumentIndex.spec.js.map +1 -0
  53. package/lib/LocalDocumentResult.d.ts +66 -0
  54. package/lib/LocalDocumentResult.d.ts.map +1 -1
  55. package/lib/LocalDocumentResult.js +376 -0
  56. package/lib/LocalDocumentResult.js.map +1 -0
  57. package/lib/LocalDocumentResult.spec.d.ts +2 -0
  58. package/lib/LocalDocumentResult.spec.d.ts.map +1 -0
  59. package/lib/LocalDocumentResult.spec.js +373 -0
  60. package/lib/LocalDocumentResult.spec.js.map +1 -0
  61. package/lib/LocalEmbeddings.d.ts +59 -0
  62. package/lib/LocalEmbeddings.d.ts.map +1 -0
  63. package/lib/LocalEmbeddings.js +101 -0
  64. package/lib/LocalEmbeddings.js.map +1 -0
  65. package/lib/LocalEmbeddings.spec.d.ts +2 -0
  66. package/lib/LocalEmbeddings.spec.d.ts.map +1 -0
  67. package/lib/LocalEmbeddings.spec.js +155 -0
  68. package/lib/LocalEmbeddings.spec.js.map +1 -0
  69. package/lib/LocalIndex.d.ts +159 -0
  70. package/lib/LocalIndex.d.ts.map +1 -1
  71. package/lib/LocalIndex.js +519 -0
  72. package/lib/LocalIndex.js.map +1 -0
  73. package/lib/LocalIndex.spec.d.ts +2 -0
  74. package/lib/LocalIndex.spec.js +611 -9
  75. package/lib/LocalIndex.spec.js.map +1 -1
  76. package/lib/OpenAIEmbeddings.d.ts +124 -0
  77. package/lib/OpenAIEmbeddings.d.ts.map +1 -0
  78. package/lib/OpenAIEmbeddings.js +166 -0
  79. package/lib/OpenAIEmbeddings.js.map +1 -0
  80. package/lib/OpenAIEmbeddings.spec.d.ts +2 -0
  81. package/lib/OpenAIEmbeddings.spec.d.ts.map +1 -0
  82. package/lib/OpenAIEmbeddings.spec.js +298 -0
  83. package/lib/OpenAIEmbeddings.spec.js.map +1 -0
  84. package/lib/TextSplitter.d.ts +21 -0
  85. package/lib/TextSplitter.d.ts.map +1 -1
  86. package/lib/TextSplitter.js +500 -0
  87. package/lib/TextSplitter.js.map +1 -0
  88. package/lib/TextSplitter.spec.d.ts +2 -0
  89. package/lib/TextSplitter.spec.d.ts.map +1 -0
  90. package/lib/TextSplitter.spec.js +337 -0
  91. package/lib/TextSplitter.spec.js.map +1 -0
  92. package/lib/TransformersEmbeddings.d.ts +121 -0
  93. package/lib/TransformersEmbeddings.d.ts.map +1 -0
  94. package/lib/TransformersEmbeddings.js +176 -0
  95. package/lib/TransformersEmbeddings.js.map +1 -0
  96. package/lib/TransformersEmbeddings.spec.d.ts +2 -0
  97. package/lib/TransformersEmbeddings.spec.d.ts.map +1 -0
  98. package/lib/TransformersEmbeddings.spec.js +198 -0
  99. package/lib/TransformersEmbeddings.spec.js.map +1 -0
  100. package/lib/TransformersTokenizer.d.ts +33 -0
  101. package/lib/TransformersTokenizer.d.ts.map +1 -0
  102. package/lib/TransformersTokenizer.js +44 -0
  103. package/lib/TransformersTokenizer.js.map +1 -0
  104. package/lib/TransformersTokenizer.spec.d.ts +2 -0
  105. package/lib/TransformersTokenizer.spec.d.ts.map +1 -0
  106. package/lib/TransformersTokenizer.spec.js +112 -0
  107. package/lib/TransformersTokenizer.spec.js.map +1 -0
  108. package/lib/WebFetcher.d.ts +14 -0
  109. package/lib/WebFetcher.d.ts.map +1 -0
  110. package/lib/WebFetcher.js +238 -0
  111. package/lib/WebFetcher.js.map +1 -0
  112. package/lib/WebFetcher.spec.d.ts +2 -0
  113. package/lib/WebFetcher.spec.d.ts.map +1 -0
  114. package/lib/WebFetcher.spec.js +263 -0
  115. package/lib/WebFetcher.spec.js.map +1 -0
  116. package/lib/browser.d.ts +30 -0
  117. package/lib/browser.d.ts.map +1 -0
  118. package/lib/browser.js +52 -0
  119. package/lib/browser.js.map +1 -0
  120. package/lib/codecs/IndexCodec.d.ts +37 -0
  121. package/lib/codecs/IndexCodec.d.ts.map +1 -0
  122. package/lib/codecs/IndexCodec.js +3 -0
  123. package/lib/codecs/IndexCodec.js.map +1 -0
  124. package/lib/codecs/JsonCodec.d.ts +19 -0
  125. package/lib/codecs/JsonCodec.d.ts.map +1 -0
  126. package/lib/codecs/JsonCodec.js +35 -0
  127. package/lib/codecs/JsonCodec.js.map +1 -0
  128. package/lib/codecs/JsonCodec.spec.d.ts +2 -0
  129. package/lib/codecs/JsonCodec.spec.d.ts.map +1 -0
  130. package/lib/codecs/JsonCodec.spec.js +66 -0
  131. package/lib/codecs/JsonCodec.spec.js.map +1 -0
  132. package/lib/codecs/LocalIndex.protobuf.spec.d.ts +2 -0
  133. package/lib/codecs/LocalIndex.protobuf.spec.d.ts.map +1 -0
  134. package/lib/codecs/LocalIndex.protobuf.spec.js +108 -0
  135. package/lib/codecs/LocalIndex.protobuf.spec.js.map +1 -0
  136. package/lib/codecs/ProtobufCodec.d.ts +20 -0
  137. package/lib/codecs/ProtobufCodec.d.ts.map +1 -0
  138. package/lib/codecs/ProtobufCodec.js +225 -0
  139. package/lib/codecs/ProtobufCodec.js.map +1 -0
  140. package/lib/codecs/ProtobufCodec.spec.d.ts +2 -0
  141. package/lib/codecs/ProtobufCodec.spec.d.ts.map +1 -0
  142. package/lib/codecs/ProtobufCodec.spec.js +155 -0
  143. package/lib/codecs/ProtobufCodec.spec.js.map +1 -0
  144. package/lib/codecs/index.d.ts +5 -0
  145. package/lib/codecs/index.d.ts.map +1 -0
  146. package/lib/codecs/index.js +21 -0
  147. package/lib/codecs/index.js.map +1 -0
  148. package/lib/codecs/migrateIndex.d.ts +24 -0
  149. package/lib/codecs/migrateIndex.d.ts.map +1 -0
  150. package/lib/codecs/migrateIndex.js +119 -0
  151. package/lib/codecs/migrateIndex.js.map +1 -0
  152. package/lib/codecs/migrateIndex.spec.d.ts +2 -0
  153. package/lib/codecs/migrateIndex.spec.d.ts.map +1 -0
  154. package/lib/codecs/migrateIndex.spec.js +151 -0
  155. package/lib/codecs/migrateIndex.spec.js.map +1 -0
  156. package/lib/codecs/schemas/index.proto +34 -0
  157. package/lib/index.d.ts +20 -0
  158. package/lib/index.d.ts.map +1 -1
  159. package/lib/index.js +36 -0
  160. package/lib/index.js.map +1 -0
  161. package/lib/internals/Colorize.d.ts +14 -0
  162. package/lib/internals/Colorize.d.ts.map +1 -0
  163. package/lib/internals/Colorize.js +69 -0
  164. package/lib/internals/Colorize.js.map +1 -0
  165. package/lib/internals/index.d.ts +3 -0
  166. package/lib/internals/index.d.ts.map +1 -0
  167. package/lib/internals/index.js +19 -0
  168. package/lib/internals/index.js.map +1 -0
  169. package/lib/internals/types.d.ts +43 -0
  170. package/lib/internals/types.d.ts.map +1 -0
  171. package/lib/internals/types.js +3 -0
  172. package/lib/internals/types.js.map +1 -0
  173. package/lib/server/IndexManager.d.ts +78 -0
  174. package/lib/server/IndexManager.d.ts.map +1 -0
  175. package/lib/server/IndexManager.js +259 -0
  176. package/lib/server/IndexManager.js.map +1 -0
  177. package/lib/server/VectraServer.d.ts +40 -0
  178. package/lib/server/VectraServer.d.ts.map +1 -0
  179. package/lib/server/VectraServer.js +151 -0
  180. package/lib/server/VectraServer.js.map +1 -0
  181. package/lib/server/VectraServer.spec.d.ts +2 -0
  182. package/lib/server/VectraServer.spec.d.ts.map +1 -0
  183. package/lib/server/VectraServer.spec.js +322 -0
  184. package/lib/server/VectraServer.spec.js.map +1 -0
  185. package/lib/server/handlers/documentHandlers.d.ts +15 -0
  186. package/lib/server/handlers/documentHandlers.d.ts.map +1 -0
  187. package/lib/server/handlers/documentHandlers.js +95 -0
  188. package/lib/server/handlers/documentHandlers.js.map +1 -0
  189. package/lib/server/handlers/helpers.d.ts +23 -0
  190. package/lib/server/handlers/helpers.d.ts.map +1 -0
  191. package/lib/server/handlers/helpers.js +138 -0
  192. package/lib/server/handlers/helpers.js.map +1 -0
  193. package/lib/server/handlers/index.d.ts +8 -0
  194. package/lib/server/handlers/index.d.ts.map +1 -0
  195. package/lib/server/handlers/index.js +22 -0
  196. package/lib/server/handlers/index.js.map +1 -0
  197. package/lib/server/handlers/indexHandlers.d.ts +14 -0
  198. package/lib/server/handlers/indexHandlers.d.ts.map +1 -0
  199. package/lib/server/handlers/indexHandlers.js +85 -0
  200. package/lib/server/handlers/indexHandlers.js.map +1 -0
  201. package/lib/server/handlers/itemHandlers.d.ts +34 -0
  202. package/lib/server/handlers/itemHandlers.d.ts.map +1 -0
  203. package/lib/server/handlers/itemHandlers.js +166 -0
  204. package/lib/server/handlers/itemHandlers.js.map +1 -0
  205. package/lib/server/handlers/lifecycleHandlers.d.ts +11 -0
  206. package/lib/server/handlers/lifecycleHandlers.d.ts.map +1 -0
  207. package/lib/server/handlers/lifecycleHandlers.js +31 -0
  208. package/lib/server/handlers/lifecycleHandlers.js.map +1 -0
  209. package/lib/server/handlers/queryHandlers.d.ts +27 -0
  210. package/lib/server/handlers/queryHandlers.d.ts.map +1 -0
  211. package/lib/server/handlers/queryHandlers.js +135 -0
  212. package/lib/server/handlers/queryHandlers.js.map +1 -0
  213. package/lib/server/handlers/statsHandlers.d.ts +17 -0
  214. package/lib/server/handlers/statsHandlers.d.ts.map +1 -0
  215. package/lib/server/handlers/statsHandlers.js +81 -0
  216. package/lib/server/handlers/statsHandlers.js.map +1 -0
  217. package/lib/server/index.d.ts +4 -0
  218. package/lib/server/index.d.ts.map +1 -0
  219. package/lib/server/index.js +23 -0
  220. package/lib/server/index.js.map +1 -0
  221. package/lib/storage/FileStorage.d.ts +92 -0
  222. package/lib/storage/FileStorage.d.ts.map +1 -0
  223. package/lib/storage/FileStorage.js +3 -0
  224. package/lib/storage/FileStorage.js.map +1 -0
  225. package/lib/storage/FileStorageUtilities.d.ts +36 -0
  226. package/lib/storage/FileStorageUtilities.d.ts.map +1 -0
  227. package/lib/storage/FileStorageUtilities.js +91 -0
  228. package/lib/storage/FileStorageUtilities.js.map +1 -0
  229. package/lib/storage/FileStorageUtilities.spec.d.ts +2 -0
  230. package/lib/storage/FileStorageUtilities.spec.d.ts.map +1 -0
  231. package/lib/storage/FileStorageUtilities.spec.js +98 -0
  232. package/lib/storage/FileStorageUtilities.spec.js.map +1 -0
  233. package/lib/storage/FileType.d.ts +29 -0
  234. package/lib/storage/FileType.d.ts.map +1 -0
  235. package/lib/storage/FileType.js +38 -0
  236. package/lib/storage/FileType.js.map +1 -0
  237. package/lib/storage/IndexedDBStorage.d.ts +47 -0
  238. package/lib/storage/IndexedDBStorage.d.ts.map +1 -0
  239. package/lib/storage/IndexedDBStorage.js +347 -0
  240. package/lib/storage/IndexedDBStorage.js.map +1 -0
  241. package/lib/storage/LocalFileStorage.browser.d.ts +19 -0
  242. package/lib/storage/LocalFileStorage.browser.d.ts.map +1 -0
  243. package/lib/storage/LocalFileStorage.browser.js +43 -0
  244. package/lib/storage/LocalFileStorage.browser.js.map +1 -0
  245. package/lib/storage/LocalFileStorage.d.ts +23 -0
  246. package/lib/storage/LocalFileStorage.d.ts.map +1 -0
  247. package/lib/storage/LocalFileStorage.js +152 -0
  248. package/lib/storage/LocalFileStorage.js.map +1 -0
  249. package/lib/storage/LocalFileStorage.spec.d.ts +2 -0
  250. package/lib/storage/LocalFileStorage.spec.d.ts.map +1 -0
  251. package/lib/storage/LocalFileStorage.spec.js +249 -0
  252. package/lib/storage/LocalFileStorage.spec.js.map +1 -0
  253. package/lib/storage/VirtualFileStorage.d.ts +18 -0
  254. package/lib/storage/VirtualFileStorage.d.ts.map +1 -0
  255. package/lib/storage/VirtualFileStorage.js +178 -0
  256. package/lib/storage/VirtualFileStorage.js.map +1 -0
  257. package/lib/storage/VirtualFileStorage.spec.d.ts +2 -0
  258. package/lib/storage/VirtualFileStorage.spec.d.ts.map +1 -0
  259. package/lib/storage/VirtualFileStorage.spec.js +302 -0
  260. package/lib/storage/VirtualFileStorage.spec.js.map +1 -0
  261. package/lib/storage/index.d.ts +6 -0
  262. package/lib/storage/index.d.ts.map +1 -0
  263. package/lib/storage/index.js +22 -0
  264. package/lib/storage/index.js.map +1 -0
  265. package/lib/templates/templates/csharp/README.md +48 -0
  266. package/lib/templates/templates/csharp/VectraClient.cs +234 -0
  267. package/lib/templates/templates/go/README.md +71 -0
  268. package/lib/templates/templates/go/vectra_client.go +322 -0
  269. package/lib/templates/templates/java/README.md +81 -0
  270. package/lib/templates/templates/java/VectraClient.java +232 -0
  271. package/lib/templates/templates/python/README.md +37 -0
  272. package/lib/templates/templates/python/vectra_client.py +279 -0
  273. package/lib/templates/templates/rust/Cargo.toml +14 -0
  274. package/lib/templates/templates/rust/README.md +39 -0
  275. package/lib/templates/templates/rust/build.rs +4 -0
  276. package/lib/templates/templates/rust/lib.rs +284 -0
  277. package/lib/templates/templates/typescript/README.md +96 -0
  278. package/lib/templates/templates/typescript/VectraClient.ts +374 -0
  279. package/lib/templates/typescript/VectraClient.d.ts +114 -0
  280. package/lib/templates/typescript/VectraClient.d.ts.map +1 -0
  281. package/lib/templates/typescript/VectraClient.js +328 -0
  282. package/lib/templates/typescript/VectraClient.js.map +1 -0
  283. package/lib/types.d.ts +153 -0
  284. package/lib/types.d.ts.map +1 -0
  285. package/lib/types.js +3 -0
  286. package/lib/types.js.map +1 -0
  287. package/lib/utils/index.d.ts +2 -0
  288. package/lib/utils/index.d.ts.map +1 -0
  289. package/lib/utils/index.js +18 -0
  290. package/lib/utils/index.js.map +1 -0
  291. package/lib/utils/pathUtils.d.ts +40 -0
  292. package/lib/utils/pathUtils.d.ts.map +1 -0
  293. package/lib/utils/pathUtils.js +98 -0
  294. package/lib/utils/pathUtils.js.map +1 -0
  295. package/lib/vectra-cli.d.ts +2 -0
  296. package/lib/vectra-cli.d.ts.map +1 -1
  297. package/lib/vectra-cli.generate.spec.d.ts +2 -0
  298. package/lib/vectra-cli.generate.spec.d.ts.map +1 -0
  299. package/lib/vectra-cli.generate.spec.js +112 -0
  300. package/lib/vectra-cli.generate.spec.js.map +1 -0
  301. package/lib/vectra-cli.js +760 -0
  302. package/lib/vectra-cli.js.map +1 -0
  303. package/lib/vectra-cli.spec.d.ts +1 -0
  304. package/lib/vectra-cli.spec.d.ts.map +1 -0
  305. package/lib/vectra-cli.spec.js +2 -0
  306. package/lib/vectra-cli.spec.js.map +1 -0
  307. package/package.json +91 -16
  308. package/proto/vectra_service.proto +276 -0
  309. package/src/BrowserWebFetcher.ts +345 -0
  310. package/src/FileFetcher.spec.ts +234 -0
  311. package/src/FileFetcher.ts +37 -25
  312. package/src/FolderWatcher.spec.ts +288 -0
  313. package/src/FolderWatcher.ts +304 -0
  314. package/src/GPT3Tokenizer.spec.ts +50 -0
  315. package/src/ItemSelector.spec.ts +252 -0
  316. package/src/ItemSelector.ts +163 -150
  317. package/src/LocalDocument.spec.ts +211 -0
  318. package/src/LocalDocument.ts +88 -94
  319. package/src/LocalDocumentIndex.spec.ts +481 -0
  320. package/src/LocalDocumentIndex.ts +39 -40
  321. package/src/LocalDocumentResult.spec.ts +373 -0
  322. package/src/LocalDocumentResult.ts +489 -319
  323. package/src/LocalEmbeddings.spec.ts +138 -0
  324. package/src/LocalEmbeddings.ts +120 -0
  325. package/src/LocalIndex.spec.ts +808 -66
  326. package/src/LocalIndex.ts +479 -429
  327. package/src/OpenAIEmbeddings.spec.ts +354 -0
  328. package/src/OpenAIEmbeddings.ts +26 -27
  329. package/src/TextSplitter.spec.ts +342 -0
  330. package/src/TextSplitter.ts +517 -532
  331. package/src/TransformersEmbeddings.spec.ts +188 -0
  332. package/src/TransformersEmbeddings.ts +232 -0
  333. package/src/TransformersTokenizer.spec.ts +143 -0
  334. package/src/TransformersTokenizer.ts +45 -0
  335. package/src/WebFetcher.spec.ts +288 -0
  336. package/src/WebFetcher.ts +184 -186
  337. package/src/browser.ts +69 -0
  338. package/src/codecs/IndexCodec.ts +40 -0
  339. package/src/codecs/JsonCodec.spec.ts +70 -0
  340. package/src/codecs/JsonCodec.ts +37 -0
  341. package/src/codecs/LocalIndex.protobuf.spec.ts +115 -0
  342. package/src/codecs/ProtobufCodec.spec.ts +166 -0
  343. package/src/codecs/ProtobufCodec.ts +193 -0
  344. package/src/codecs/index.ts +4 -0
  345. package/src/codecs/migrateIndex.spec.ts +176 -0
  346. package/src/codecs/migrateIndex.ts +125 -0
  347. package/src/codecs/schemas/index.proto +34 -0
  348. package/src/index.ts +9 -1
  349. package/src/internals/Colorize.ts +19 -16
  350. package/src/server/IndexManager.ts +243 -0
  351. package/src/server/VectraServer.spec.ts +303 -0
  352. package/src/server/VectraServer.ts +156 -0
  353. package/src/server/handlers/documentHandlers.ts +59 -0
  354. package/src/server/handlers/helpers.ts +93 -0
  355. package/src/server/handlers/index.ts +7 -0
  356. package/src/server/handlers/indexHandlers.ts +44 -0
  357. package/src/server/handlers/itemHandlers.ts +140 -0
  358. package/src/server/handlers/lifecycleHandlers.ts +26 -0
  359. package/src/server/handlers/queryHandlers.ts +96 -0
  360. package/src/server/handlers/statsHandlers.ts +38 -0
  361. package/src/server/index.ts +3 -0
  362. package/src/storage/FileStorage.ts +105 -0
  363. package/src/storage/FileStorageUtilities.spec.ts +106 -0
  364. package/src/storage/FileStorageUtilities.ts +77 -0
  365. package/src/storage/FileType.ts +61 -0
  366. package/src/storage/IndexedDBStorage.ts +365 -0
  367. package/src/storage/LocalFileStorage.browser.ts +52 -0
  368. package/src/storage/LocalFileStorage.spec.ts +292 -0
  369. package/src/storage/LocalFileStorage.ts +98 -0
  370. package/src/storage/VirtualFileStorage.spec.ts +307 -0
  371. package/src/storage/VirtualFileStorage.ts +169 -0
  372. package/src/storage/index.ts +5 -0
  373. package/src/templates/csharp/README.md +48 -0
  374. package/src/templates/csharp/VectraClient.cs +234 -0
  375. package/src/templates/go/README.md +71 -0
  376. package/src/templates/go/vectra_client.go +322 -0
  377. package/src/templates/java/README.md +81 -0
  378. package/src/templates/java/VectraClient.java +232 -0
  379. package/src/templates/python/README.md +37 -0
  380. package/src/templates/python/vectra_client.py +279 -0
  381. package/src/templates/rust/Cargo.toml +14 -0
  382. package/src/templates/rust/README.md +39 -0
  383. package/src/templates/rust/build.rs +4 -0
  384. package/src/templates/rust/lib.rs +284 -0
  385. package/src/templates/typescript/README.md +96 -0
  386. package/src/templates/typescript/VectraClient.ts +374 -0
  387. package/src/types.ts +131 -123
  388. package/src/utils/index.ts +1 -0
  389. package/src/utils/pathUtils.ts +106 -0
  390. package/src/vectra-cli.generate.spec.ts +72 -0
  391. package/src/vectra-cli.spec.ts +0 -0
  392. package/src/vectra-cli.ts +687 -246
@@ -0,0 +1,151 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || (function () {
19
+ var ownKeys = function(o) {
20
+ ownKeys = Object.getOwnPropertyNames || function (o) {
21
+ var ar = [];
22
+ for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
23
+ return ar;
24
+ };
25
+ return ownKeys(o);
26
+ };
27
+ return function (mod) {
28
+ if (mod && mod.__esModule) return mod;
29
+ var result = {};
30
+ if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
31
+ __setModuleDefault(result, mod);
32
+ return result;
33
+ };
34
+ })();
35
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
36
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
37
+ return new (P || (P = Promise))(function (resolve, reject) {
38
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
39
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
40
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
41
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
42
+ });
43
+ };
44
+ Object.defineProperty(exports, "__esModule", { value: true });
45
+ exports.VectraServer = void 0;
46
+ const path = __importStar(require("path"));
47
+ const grpc = __importStar(require("@grpc/grpc-js"));
48
+ const protoLoader = __importStar(require("@grpc/proto-loader"));
49
+ const IndexManager_1 = require("./IndexManager");
50
+ const handlers_1 = require("./handlers");
51
+ const DRAIN_TIMEOUT_MS = 5000;
52
+ /**
53
+ * gRPC server that exposes Vectra indexes over the VectraService proto.
54
+ */
55
+ class VectraServer {
56
+ constructor(config) {
57
+ this._startTime = 0;
58
+ this._config = config;
59
+ const managerConfig = {
60
+ indexPath: config.indexPath,
61
+ rootDir: config.rootDir,
62
+ embeddings: config.embeddings,
63
+ scanInterval: config.scanInterval,
64
+ };
65
+ this._indexManager = new IndexManager_1.IndexManager(managerConfig);
66
+ }
67
+ /** The underlying IndexManager. */
68
+ get indexManager() {
69
+ return this._indexManager;
70
+ }
71
+ /** The underlying gRPC server instance. */
72
+ get server() {
73
+ return this._server;
74
+ }
75
+ /**
76
+ * Starts the gRPC server and loads indexes.
77
+ * @returns The port the server is listening on.
78
+ */
79
+ start() {
80
+ return __awaiter(this, void 0, void 0, function* () {
81
+ var _a;
82
+ this._startTime = Date.now();
83
+ // Load proto definition
84
+ const protoPath = path.resolve(__dirname, '..', '..', 'proto', 'vectra_service.proto');
85
+ const packageDefinition = protoLoader.loadSync(protoPath, {
86
+ keepCase: true,
87
+ longs: Number,
88
+ enums: String,
89
+ defaults: true,
90
+ oneofs: true,
91
+ });
92
+ const protoDescriptor = grpc.loadPackageDefinition(packageDefinition);
93
+ const vectraService = protoDescriptor.vectra.VectraService;
94
+ // Initialize index manager
95
+ yield this._indexManager.initialize();
96
+ // Create gRPC server
97
+ this._server = new grpc.Server();
98
+ // Register all handler groups
99
+ const indexHandlers = (0, handlers_1.createIndexHandlers)(this._indexManager);
100
+ const itemHandlers = (0, handlers_1.createItemHandlers)(this._indexManager, this._config.embeddings);
101
+ const queryHandlers = (0, handlers_1.createQueryHandlers)(this._indexManager, this._config.embeddings);
102
+ const documentHandlers = (0, handlers_1.createDocumentHandlers)(this._indexManager);
103
+ const statsHandlers = (0, handlers_1.createStatsHandlers)(this._indexManager);
104
+ const lifecycleHandlers = (0, handlers_1.createLifecycleHandlers)(this._indexManager, this._startTime, () => this.shutdown());
105
+ this._server.addService(vectraService.service, Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({}, indexHandlers), itemHandlers), queryHandlers), documentHandlers), statsHandlers), lifecycleHandlers));
106
+ // Bind to localhost only
107
+ const port = (_a = this._config.port) !== null && _a !== void 0 ? _a : 50051;
108
+ const boundPort = yield new Promise((resolve, reject) => {
109
+ this._server.bindAsync(`127.0.0.1:${port}`, grpc.ServerCredentials.createInsecure(), (err, actualPort) => {
110
+ if (err) {
111
+ reject(err);
112
+ }
113
+ else {
114
+ resolve(actualPort);
115
+ }
116
+ });
117
+ });
118
+ return boundPort;
119
+ });
120
+ }
121
+ /**
122
+ * Gracefully shuts down the server with a 5s draining timeout.
123
+ */
124
+ shutdown() {
125
+ return __awaiter(this, void 0, void 0, function* () {
126
+ if (this._shutdownPromise)
127
+ return this._shutdownPromise;
128
+ this._shutdownPromise = (() => __awaiter(this, void 0, void 0, function* () {
129
+ // Stop index manager scanning
130
+ yield this._indexManager.shutdown();
131
+ if (!this._server)
132
+ return;
133
+ // Try graceful shutdown with timeout
134
+ yield new Promise((resolve) => {
135
+ const timer = setTimeout(() => {
136
+ this._server.forceShutdown();
137
+ resolve();
138
+ }, DRAIN_TIMEOUT_MS);
139
+ this._server.tryShutdown((err) => {
140
+ clearTimeout(timer);
141
+ resolve();
142
+ });
143
+ });
144
+ this._server = undefined;
145
+ }))();
146
+ return this._shutdownPromise;
147
+ });
148
+ }
149
+ }
150
+ exports.VectraServer = VectraServer;
151
+ //# sourceMappingURL=VectraServer.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"VectraServer.js","sourceRoot":"","sources":["../../src/server/VectraServer.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CAA6B;AAC7B,oDAAsC;AACtC,gEAAkD;AAClD,iDAAkE;AAElE,yCAOoB;AAepB,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAE9B;;GAEG;AACH,MAAa,YAAY;IAOrB,YAAY,MAA0B;QAH9B,eAAU,GAAW,CAAC,CAAC;QAI3B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,MAAM,aAAa,GAAuB;YACtC,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,YAAY,EAAE,MAAM,CAAC,YAAY;SACpC,CAAC;QACF,IAAI,CAAC,aAAa,GAAG,IAAI,2BAAY,CAAC,aAAa,CAAC,CAAC;IACzD,CAAC;IAED,mCAAmC;IACnC,IAAW,YAAY;QACnB,OAAO,IAAI,CAAC,aAAa,CAAC;IAC9B,CAAC;IAED,2CAA2C;IAC3C,IAAW,MAAM;QACb,OAAO,IAAI,CAAC,OAAO,CAAC;IACxB,CAAC;IAED;;;OAGG;IACU,KAAK;;;YACd,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,wBAAwB;YACxB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;YACvF,MAAM,iBAAiB,GAAG,WAAW,CAAC,QAAQ,CAAC,SAAS,EAAE;gBACtD,QAAQ,EAAE,IAAI;gBACd,KAAK,EAAE,MAAM;gBACb,KAAK,EAAE,MAAM;gBACb,QAAQ,EAAE,IAAI;gBACd,MAAM,EAAE,IAAI;aACf,CAAC,CAAC;YACH,MAAM,eAAe,GAAG,IAAI,CAAC,qBAAqB,CAAC,iBAAiB,CAAQ,CAAC;YAC7E,MAAM,aAAa,GAAG,eAAe,CAAC,MAAM,CAAC,aAAa,CAAC;YAE3D,2BAA2B;YAC3B,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;YAEtC,qBAAqB;YACrB,IAAI,CAAC,OAAO,GAAG,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAEjC,8BAA8B;YAC9B,MAAM,aAAa,GAAG,IAAA,8BAAmB,EAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9D,MAAM,YAAY,GAAG,IAAA,6BAAkB,EAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YACrF,MAAM,aAAa,GAAG,IAAA,8BAAmB,EAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YACvF,MAAM,gBAAgB,GAAG,IAAA,iCAAsB,EAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACpE,MAAM,aAAa,GAAG,IAAA,8BAAmB,EAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9D,MAAM,iBAAiB,GAAG,IAAA,kCAAuB,EAC7C,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,UAAU,EACf,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,CACxB,CAAC;YAEF,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,OAAO,0FACtC,aAAa,GACb,YAAY,GACZ,aAAa,GACb,gBAAgB,GAChB,aAAa,GACb,iBAAiB,EACtB,CAAC;YAEH,yBAAyB;YACzB,MAAM,IAAI,GAAG,MAAA,IAAI,CAAC,OAAO,CAAC,IAAI,mCAAI,KAAK,CAAC;YACxC,MAAM,SAAS,GAAG,MAAM,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC5D,IAAI,CAAC,OAAQ,CAAC,SAAS,CACnB,aAAa,IAAI,EAAE,EACnB,IAAI,CAAC,iBAAiB,CAAC,cAAc,EAAE,EACvC,CAAC,GAAG,EAAE,UAAU,EAAE,EAAE;oBAChB,IAAI,GAAG,EAAE,CAAC;wBACN,MAAM,CAAC,GAAG,CAAC,CAAC;oBAChB,CAAC;yBAAM,CAAC;wBACJ,OAAO,CAAC,UAAU,CAAC,CAAC;oBACxB,CAAC;gBACL,CAAC,CACJ,CAAC;YACN,CAAC,CAAC,CAAC;YAEH,OAAO,SAAS,CAAC;QACrB,CAAC;KAAA;IAED;;OAEG;IACU,QAAQ;;YACjB,IAAI,IAAI,CAAC,gBAAgB;gBAAE,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAExD,IAAI,CAAC,gBAAgB,GAAG,CAAC,GAAS,EAAE;gBAChC,8BAA8B;gBAC9B,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;gBAEpC,IAAI,CAAC,IAAI,CAAC,OAAO;oBAAE,OAAO;gBAE1B,qCAAqC;gBACrC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;oBAChC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;wBAC1B,IAAI,CAAC,OAAQ,CAAC,aAAa,EAAE,CAAC;wBAC9B,OAAO,EAAE,CAAC;oBACd,CAAC,EAAE,gBAAgB,CAAC,CAAC;oBAErB,IAAI,CAAC,OAAQ,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE;wBAC9B,YAAY,CAAC,KAAK,CAAC,CAAC;wBACpB,OAAO,EAAE,CAAC;oBACd,CAAC,CAAC,CAAC;gBACP,CAAC,CAAC,CAAC;gBAEH,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;YAC7B,CAAC,CAAA,CAAC,EAAE,CAAC;YAEL,OAAO,IAAI,CAAC,gBAAgB,CAAC;QACjC,CAAC;KAAA;CACJ;AA3HD,oCA2HC"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=VectraServer.spec.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"VectraServer.spec.d.ts","sourceRoot":"","sources":["../../src/server/VectraServer.spec.ts"],"names":[],"mappings":""}
@@ -0,0 +1,322 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || (function () {
19
+ var ownKeys = function(o) {
20
+ ownKeys = Object.getOwnPropertyNames || function (o) {
21
+ var ar = [];
22
+ for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
23
+ return ar;
24
+ };
25
+ return ownKeys(o);
26
+ };
27
+ return function (mod) {
28
+ if (mod && mod.__esModule) return mod;
29
+ var result = {};
30
+ if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
31
+ __setModuleDefault(result, mod);
32
+ return result;
33
+ };
34
+ })();
35
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
36
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
37
+ return new (P || (P = Promise))(function (resolve, reject) {
38
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
39
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
40
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
41
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
42
+ });
43
+ };
44
+ Object.defineProperty(exports, "__esModule", { value: true });
45
+ const assert = __importStar(require("assert"));
46
+ const path = __importStar(require("path"));
47
+ const fs = __importStar(require("fs"));
48
+ const grpc = __importStar(require("@grpc/grpc-js"));
49
+ const protoLoader = __importStar(require("@grpc/proto-loader"));
50
+ const VectraServer_1 = require("./VectraServer");
51
+ const TEST_ROOT = path.join(__dirname, '..', '..', '.test-server-root');
52
+ const PROTO_PATH = path.join(__dirname, '..', '..', 'proto', 'vectra_service.proto');
53
+ function loadClient(port) {
54
+ const packageDefinition = protoLoader.loadSync(PROTO_PATH, {
55
+ keepCase: true,
56
+ longs: Number,
57
+ enums: String,
58
+ defaults: true,
59
+ oneofs: true,
60
+ });
61
+ const proto = grpc.loadPackageDefinition(packageDefinition);
62
+ return new proto.vectra.VectraService(`127.0.0.1:${port}`, grpc.credentials.createInsecure());
63
+ }
64
+ function rpc(client, method, request) {
65
+ return new Promise((resolve, reject) => {
66
+ client[method](request, (err, response) => {
67
+ if (err)
68
+ reject(err);
69
+ else
70
+ resolve(response);
71
+ });
72
+ });
73
+ }
74
+ function cleanTestDir() {
75
+ if (fs.existsSync(TEST_ROOT)) {
76
+ fs.rmSync(TEST_ROOT, { recursive: true });
77
+ }
78
+ fs.mkdirSync(TEST_ROOT, { recursive: true });
79
+ }
80
+ describe('VectraServer', function () {
81
+ this.timeout(30000);
82
+ let server;
83
+ let client;
84
+ let port;
85
+ before(() => __awaiter(this, void 0, void 0, function* () {
86
+ cleanTestDir();
87
+ server = new VectraServer_1.VectraServer({
88
+ port: 0, // random available port
89
+ rootDir: TEST_ROOT,
90
+ scanInterval: 500,
91
+ });
92
+ port = yield server.start();
93
+ client = loadClient(port);
94
+ }));
95
+ after(() => __awaiter(this, void 0, void 0, function* () {
96
+ if (client) {
97
+ client.close();
98
+ }
99
+ if (server) {
100
+ yield server.shutdown();
101
+ }
102
+ if (fs.existsSync(TEST_ROOT)) {
103
+ fs.rmSync(TEST_ROOT, { recursive: true });
104
+ }
105
+ }));
106
+ describe('Lifecycle RPCs', () => {
107
+ it('should respond to Healthcheck', () => __awaiter(this, void 0, void 0, function* () {
108
+ const res = yield rpc(client, 'Healthcheck', {});
109
+ assert.strictEqual(res.status, 'ok');
110
+ assert.strictEqual(typeof res.uptime_seconds, 'number');
111
+ assert.strictEqual(typeof res.loaded_indexes, 'number');
112
+ }));
113
+ });
114
+ describe('Index Management RPCs', () => {
115
+ it('should create an index', () => __awaiter(this, void 0, void 0, function* () {
116
+ yield rpc(client, 'CreateIndex', {
117
+ index_name: 'test-index',
118
+ format: 'json',
119
+ is_document_index: false,
120
+ });
121
+ const list = yield rpc(client, 'ListIndexes', {});
122
+ const names = list.indexes.map((i) => i.name);
123
+ assert.ok(names.includes('test-index'), `Expected test-index in ${JSON.stringify(names)}`);
124
+ }));
125
+ it('should fail to create a duplicate index', () => __awaiter(this, void 0, void 0, function* () {
126
+ try {
127
+ yield rpc(client, 'CreateIndex', {
128
+ index_name: 'test-index',
129
+ format: 'json',
130
+ is_document_index: false,
131
+ });
132
+ assert.fail('Expected ALREADY_EXISTS error');
133
+ }
134
+ catch (err) {
135
+ assert.strictEqual(err.code, grpc.status.ALREADY_EXISTS);
136
+ }
137
+ }));
138
+ it('should list indexes', () => __awaiter(this, void 0, void 0, function* () {
139
+ const res = yield rpc(client, 'ListIndexes', {});
140
+ assert.ok(Array.isArray(res.indexes));
141
+ assert.ok(res.indexes.length >= 1);
142
+ }));
143
+ });
144
+ describe('Item Operations RPCs', () => {
145
+ const testVector = new Array(3).fill(0).map((_, i) => (i + 1) * 0.1);
146
+ it('should insert an item', () => __awaiter(this, void 0, void 0, function* () {
147
+ const res = yield rpc(client, 'InsertItem', {
148
+ index_name: 'test-index',
149
+ vector: testVector,
150
+ metadata: {
151
+ category: { string_value: 'test' },
152
+ score: { number_value: 0.95 },
153
+ },
154
+ id: 'item-1',
155
+ });
156
+ assert.strictEqual(res.id, 'item-1');
157
+ }));
158
+ it('should get an item by id', () => __awaiter(this, void 0, void 0, function* () {
159
+ const res = yield rpc(client, 'GetItem', {
160
+ index_name: 'test-index',
161
+ id: 'item-1',
162
+ });
163
+ assert.ok(res.item);
164
+ assert.strictEqual(res.item.id, 'item-1');
165
+ assert.ok(res.item.vector.length > 0);
166
+ }));
167
+ it('should return NOT_FOUND for missing item', () => __awaiter(this, void 0, void 0, function* () {
168
+ try {
169
+ yield rpc(client, 'GetItem', {
170
+ index_name: 'test-index',
171
+ id: 'nonexistent',
172
+ });
173
+ assert.fail('Expected NOT_FOUND error');
174
+ }
175
+ catch (err) {
176
+ assert.strictEqual(err.code, grpc.status.NOT_FOUND);
177
+ }
178
+ }));
179
+ it('should upsert an item', () => __awaiter(this, void 0, void 0, function* () {
180
+ const newVector = new Array(3).fill(0).map((_, i) => (i + 1) * 0.2);
181
+ const res = yield rpc(client, 'UpsertItem', {
182
+ index_name: 'test-index',
183
+ id: 'item-1',
184
+ vector: newVector,
185
+ metadata: {
186
+ category: { string_value: 'updated' },
187
+ },
188
+ });
189
+ assert.strictEqual(res.id, 'item-1');
190
+ // Verify update
191
+ const getRes = yield rpc(client, 'GetItem', {
192
+ index_name: 'test-index',
193
+ id: 'item-1',
194
+ });
195
+ assert.strictEqual(getRes.item.metadata.category.string_value, 'updated');
196
+ }));
197
+ it('should batch insert items', () => __awaiter(this, void 0, void 0, function* () {
198
+ const res = yield rpc(client, 'BatchInsertItems', {
199
+ index_name: 'test-index',
200
+ items: [
201
+ { vector: [0.1, 0.2, 0.3], metadata: { tag: { string_value: 'a' } }, id: 'batch-1' },
202
+ { vector: [0.4, 0.5, 0.6], metadata: { tag: { string_value: 'b' } }, id: 'batch-2' },
203
+ ],
204
+ });
205
+ assert.strictEqual(res.ids.length, 2);
206
+ assert.ok(res.ids.includes('batch-1'));
207
+ assert.ok(res.ids.includes('batch-2'));
208
+ }));
209
+ it('should list items', () => __awaiter(this, void 0, void 0, function* () {
210
+ const res = yield rpc(client, 'ListItems', {
211
+ index_name: 'test-index',
212
+ });
213
+ assert.ok(res.items.length >= 3); // item-1, batch-1, batch-2
214
+ }));
215
+ it('should delete an item', () => __awaiter(this, void 0, void 0, function* () {
216
+ yield rpc(client, 'DeleteItem', {
217
+ index_name: 'test-index',
218
+ id: 'batch-2',
219
+ });
220
+ // Verify it's gone
221
+ try {
222
+ yield rpc(client, 'GetItem', {
223
+ index_name: 'test-index',
224
+ id: 'batch-2',
225
+ });
226
+ assert.fail('Expected NOT_FOUND');
227
+ }
228
+ catch (err) {
229
+ assert.strictEqual(err.code, grpc.status.NOT_FOUND);
230
+ }
231
+ }));
232
+ });
233
+ describe('Query RPCs', () => {
234
+ it('should query items by vector', () => __awaiter(this, void 0, void 0, function* () {
235
+ const res = yield rpc(client, 'QueryItems', {
236
+ index_name: 'test-index',
237
+ vector: [0.1, 0.2, 0.3],
238
+ top_k: 5,
239
+ });
240
+ assert.ok(Array.isArray(res.results));
241
+ assert.ok(res.results.length > 0);
242
+ // Results should have scores
243
+ assert.ok(typeof res.results[0].score === 'number');
244
+ }));
245
+ });
246
+ describe('Stats RPCs', () => {
247
+ it('should get index stats', () => __awaiter(this, void 0, void 0, function* () {
248
+ const res = yield rpc(client, 'GetIndexStats', {
249
+ index_name: 'test-index',
250
+ });
251
+ assert.strictEqual(res.version, 1);
252
+ assert.strictEqual(res.format, 'json');
253
+ assert.ok(typeof res.item_count === 'number');
254
+ }));
255
+ });
256
+ describe('Error handling', () => {
257
+ it('should return NOT_FOUND for unknown index', () => __awaiter(this, void 0, void 0, function* () {
258
+ try {
259
+ yield rpc(client, 'GetIndexStats', {
260
+ index_name: 'no-such-index',
261
+ });
262
+ assert.fail('Expected NOT_FOUND');
263
+ }
264
+ catch (err) {
265
+ assert.strictEqual(err.code, grpc.status.NOT_FOUND);
266
+ }
267
+ }));
268
+ it('should return INVALID_ARGUMENT for missing required fields', () => __awaiter(this, void 0, void 0, function* () {
269
+ try {
270
+ yield rpc(client, 'InsertItem', {
271
+ index_name: 'test-index',
272
+ // no text or vector
273
+ });
274
+ assert.fail('Expected INVALID_ARGUMENT');
275
+ }
276
+ catch (err) {
277
+ assert.strictEqual(err.code, grpc.status.INVALID_ARGUMENT);
278
+ }
279
+ }));
280
+ it('should return FAILED_PRECONDITION for non-document index on document ops', () => __awaiter(this, void 0, void 0, function* () {
281
+ try {
282
+ yield rpc(client, 'ListDocuments', {
283
+ index_name: 'test-index',
284
+ });
285
+ assert.fail('Expected FAILED_PRECONDITION');
286
+ }
287
+ catch (err) {
288
+ assert.strictEqual(err.code, grpc.status.FAILED_PRECONDITION);
289
+ }
290
+ }));
291
+ });
292
+ describe('Index deletion', () => {
293
+ it('should delete an index', () => __awaiter(this, void 0, void 0, function* () {
294
+ yield rpc(client, 'DeleteIndex', {
295
+ index_name: 'test-index',
296
+ });
297
+ const list = yield rpc(client, 'ListIndexes', {});
298
+ const names = list.indexes.map((i) => i.name);
299
+ assert.ok(!names.includes('test-index'));
300
+ }));
301
+ });
302
+ describe('Auto-detect new indexes', () => {
303
+ it('should detect indexes created on disk', function () {
304
+ return __awaiter(this, void 0, void 0, function* () {
305
+ // Create an index directly on disk (outside the gRPC API)
306
+ const indexDir = path.join(TEST_ROOT, 'auto-detected');
307
+ fs.mkdirSync(indexDir, { recursive: true });
308
+ fs.writeFileSync(path.join(indexDir, 'index.json'), JSON.stringify({
309
+ version: 1,
310
+ metadata_config: {},
311
+ items: [],
312
+ }));
313
+ // Wait for the scanner to pick it up
314
+ yield new Promise(r => setTimeout(r, 1500));
315
+ const list = yield rpc(client, 'ListIndexes', {});
316
+ const names = list.indexes.map((i) => i.name);
317
+ assert.ok(names.includes('auto-detected'), `Expected auto-detected in ${JSON.stringify(names)}`);
318
+ });
319
+ });
320
+ });
321
+ });
322
+ //# sourceMappingURL=VectraServer.spec.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"VectraServer.spec.js","sourceRoot":"","sources":["../../src/server/VectraServer.spec.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,+CAAiC;AACjC,2CAA6B;AAC7B,uCAAyB;AACzB,oDAAsC;AACtC,gEAAkD;AAClD,iDAA8C;AAE9C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,mBAAmB,CAAC,CAAC;AACxE,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;AAErF,SAAS,UAAU,CAAC,IAAY;IAC5B,MAAM,iBAAiB,GAAG,WAAW,CAAC,QAAQ,CAAC,UAAU,EAAE;QACvD,QAAQ,EAAE,IAAI;QACd,KAAK,EAAE,MAAM;QACb,KAAK,EAAE,MAAM;QACb,QAAQ,EAAE,IAAI;QACd,MAAM,EAAE,IAAI;KACf,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC,iBAAiB,CAAQ,CAAC;IACnE,OAAO,IAAI,KAAK,CAAC,MAAM,CAAC,aAAa,CACjC,aAAa,IAAI,EAAE,EACnB,IAAI,CAAC,WAAW,CAAC,cAAc,EAAE,CACpC,CAAC;AACN,CAAC;AAED,SAAS,GAAG,CAAC,MAAW,EAAE,MAAc,EAAE,OAAY;IAClD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACnC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,CAAC,GAAQ,EAAE,QAAa,EAAE,EAAE;YAChD,IAAI,GAAG;gBAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;gBAChB,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,YAAY;IACjB,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9C,CAAC;IACD,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;AACjD,CAAC;AAED,QAAQ,CAAC,cAAc,EAAE;IACrB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAEpB,IAAI,MAAoB,CAAC;IACzB,IAAI,MAAW,CAAC;IAChB,IAAI,IAAY,CAAC;IAEjB,MAAM,CAAC,GAAS,EAAE;QACd,YAAY,EAAE,CAAC;QACf,MAAM,GAAG,IAAI,2BAAY,CAAC;YACtB,IAAI,EAAE,CAAC,EAAE,wBAAwB;YACjC,OAAO,EAAE,SAAS;YAClB,YAAY,EAAE,GAAG;SACpB,CAAC,CAAC;QACH,IAAI,GAAG,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QAC5B,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC,CAAA,CAAC,CAAC;IAEH,KAAK,CAAC,GAAS,EAAE;QACb,IAAI,MAAM,EAAE,CAAC;YACT,MAAM,CAAC,KAAK,EAAE,CAAC;QACnB,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACT,MAAM,MAAM,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC;QACD,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3B,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC;IACL,CAAC,CAAA,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,+BAA+B,EAAE,GAAS,EAAE;YAC3C,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;YACjD,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACrC,MAAM,CAAC,WAAW,CAAC,OAAO,GAAG,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;YACxD,MAAM,CAAC,WAAW,CAAC,OAAO,GAAG,CAAC,cAAc,EAAE,QAAQ,CAAC,CAAC;QAC5D,CAAC,CAAA,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,wBAAwB,EAAE,GAAS,EAAE;YACpC,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE;gBAC7B,UAAU,EAAE,YAAY;gBACxB,MAAM,EAAE,MAAM;gBACd,iBAAiB,EAAE,KAAK;aAC3B,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACnD,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,0BAA0B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/F,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAS,EAAE;YACrD,IAAI,CAAC;gBACD,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE;oBAC7B,UAAU,EAAE,YAAY;oBACxB,MAAM,EAAE,MAAM;oBACd,iBAAiB,EAAE,KAAK;iBAC3B,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YAC7D,CAAC;QACL,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,qBAAqB,EAAE,GAAS,EAAE;YACjC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;YACjD,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC;QACvC,CAAC,CAAA,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAClC,MAAM,UAAU,GAAG,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC;QAErE,EAAE,CAAC,uBAAuB,EAAE,GAAS,EAAE;YACnC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE;gBACxC,UAAU,EAAE,YAAY;gBACxB,MAAM,EAAE,UAAU;gBAClB,QAAQ,EAAE;oBACN,QAAQ,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE;oBAClC,KAAK,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;iBAChC;gBACD,EAAE,EAAE,QAAQ;aACf,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QACzC,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAS,EAAE;YACtC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE;gBACrC,UAAU,EAAE,YAAY;gBACxB,EAAE,EAAE,QAAQ;aACf,CAAC,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACpB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YAC1C,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAS,EAAE;YACtD,IAAI,CAAC;gBACD,MAAM,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE;oBACzB,UAAU,EAAE,YAAY;oBACxB,EAAE,EAAE,aAAa;iBACpB,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAC5C,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YACxD,CAAC;QACL,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,uBAAuB,EAAE,GAAS,EAAE;YACnC,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC;YACpE,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE;gBACxC,UAAU,EAAE,YAAY;gBACxB,EAAE,EAAE,QAAQ;gBACZ,MAAM,EAAE,SAAS;gBACjB,QAAQ,EAAE;oBACN,QAAQ,EAAE,EAAE,YAAY,EAAE,SAAS,EAAE;iBACxC;aACJ,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YAErC,gBAAgB;YAChB,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE;gBACxC,UAAU,EAAE,YAAY;gBACxB,EAAE,EAAE,QAAQ;aACf,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;QAC9E,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAS,EAAE;YACvC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,kBAAkB,EAAE;gBAC9C,UAAU,EAAE,YAAY;gBACxB,KAAK,EAAE;oBACH,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,EAAE,GAAG,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE;oBACpF,EAAE,MAAM,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,EAAE,GAAG,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE;iBACvF;aACJ,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;YACvC,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,mBAAmB,EAAE,GAAS,EAAE;YAC/B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,WAAW,EAAE;gBACvC,UAAU,EAAE,YAAY;aAC3B,CAAC,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,2BAA2B;QACjE,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,uBAAuB,EAAE,GAAS,EAAE;YACnC,MAAM,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE;gBAC5B,UAAU,EAAE,YAAY;gBACxB,EAAE,EAAE,SAAS;aAChB,CAAC,CAAC;YAEH,mBAAmB;YACnB,IAAI,CAAC;gBACD,MAAM,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE;oBACzB,UAAU,EAAE,YAAY;oBACxB,EAAE,EAAE,SAAS;iBAChB,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACtC,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YACxD,CAAC;QACL,CAAC,CAAA,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,8BAA8B,EAAE,GAAS,EAAE;YAC1C,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE;gBACxC,UAAU,EAAE,YAAY;gBACxB,MAAM,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;gBACvB,KAAK,EAAE,CAAC;aACX,CAAC,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAClC,6BAA6B;YAC7B,MAAM,CAAC,EAAE,CAAC,OAAO,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC;QACxD,CAAC,CAAA,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,wBAAwB,EAAE,GAAS,EAAE;YACpC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,eAAe,EAAE;gBAC3C,UAAU,EAAE,YAAY;aAC3B,CAAC,CAAC;YACH,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvC,MAAM,CAAC,EAAE,CAAC,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC;QAClD,CAAC,CAAA,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,2CAA2C,EAAE,GAAS,EAAE;YACvD,IAAI,CAAC;gBACD,MAAM,GAAG,CAAC,MAAM,EAAE,eAAe,EAAE;oBAC/B,UAAU,EAAE,eAAe;iBAC9B,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACtC,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YACxD,CAAC;QACL,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAS,EAAE;YACxE,IAAI,CAAC;gBACD,MAAM,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE;oBAC5B,UAAU,EAAE,YAAY;oBACxB,oBAAoB;iBACvB,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAC/D,CAAC;QACL,CAAC,CAAA,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAS,EAAE;YACtF,IAAI,CAAC;gBACD,MAAM,GAAG,CAAC,MAAM,EAAE,eAAe,EAAE;oBAC/B,UAAU,EAAE,YAAY;iBAC3B,CAAC,CAAC;gBACH,MAAM,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;YAChD,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAChB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;YAClE,CAAC;QACL,CAAC,CAAA,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,wBAAwB,EAAE,GAAS,EAAE;YACpC,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE;gBAC7B,UAAU,EAAE,YAAY;aAC3B,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACnD,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAA,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,uCAAuC,EAAE;;gBACxC,0DAA0D;gBAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;gBACvD,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC5C,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC;oBAC/D,OAAO,EAAE,CAAC;oBACV,eAAe,EAAE,EAAE;oBACnB,KAAK,EAAE,EAAE;iBACZ,CAAC,CAAC,CAAC;gBAEJ,qCAAqC;gBACrC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;gBAE5C,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACnD,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,6BAA6B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACrG,CAAC;SAAA,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}
@@ -0,0 +1,15 @@
1
+ import * as grpc from '@grpc/grpc-js';
2
+ import { IndexManager } from '../IndexManager';
3
+ export declare function createDocumentHandlers(manager: IndexManager): {
4
+ UpsertDocument: grpc.handleUnaryCall<any, {
5
+ document_id: string;
6
+ }>;
7
+ DeleteDocument: grpc.handleUnaryCall<any, {}>;
8
+ ListDocuments: grpc.handleUnaryCall<any, {
9
+ documents: {
10
+ uri: string;
11
+ document_id: string;
12
+ }[];
13
+ }>;
14
+ };
15
+ //# sourceMappingURL=documentHandlers.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"documentHandlers.d.ts","sourceRoot":"","sources":["../../../src/server/handlers/documentHandlers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAG/C,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,YAAY;;;;;;;;;;;EAsD3D"}
@@ -0,0 +1,95 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
15
+ }) : function(o, v) {
16
+ o["default"] = v;
17
+ });
18
+ var __importStar = (this && this.__importStar) || (function () {
19
+ var ownKeys = function(o) {
20
+ ownKeys = Object.getOwnPropertyNames || function (o) {
21
+ var ar = [];
22
+ for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
23
+ return ar;
24
+ };
25
+ return ownKeys(o);
26
+ };
27
+ return function (mod) {
28
+ if (mod && mod.__esModule) return mod;
29
+ var result = {};
30
+ if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
31
+ __setModuleDefault(result, mod);
32
+ return result;
33
+ };
34
+ })();
35
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
36
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
37
+ return new (P || (P = Promise))(function (resolve, reject) {
38
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
39
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
40
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
41
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
42
+ });
43
+ };
44
+ Object.defineProperty(exports, "__esModule", { value: true });
45
+ exports.createDocumentHandlers = createDocumentHandlers;
46
+ const grpc = __importStar(require("@grpc/grpc-js"));
47
+ const helpers_1 = require("./helpers");
48
+ function createDocumentHandlers(manager) {
49
+ return {
50
+ UpsertDocument: (0, helpers_1.wrapHandler)((call) => __awaiter(this, void 0, void 0, function* () {
51
+ const req = call.request;
52
+ if (!req.index_name) {
53
+ throw (0, helpers_1.grpcError)(grpc.status.INVALID_ARGUMENT, 'index_name is required');
54
+ }
55
+ if (!req.uri) {
56
+ throw (0, helpers_1.grpcError)(grpc.status.INVALID_ARGUMENT, 'uri is required');
57
+ }
58
+ if (!req.text) {
59
+ throw (0, helpers_1.grpcError)(grpc.status.INVALID_ARGUMENT, 'text is required');
60
+ }
61
+ const { docIndex } = manager.requireDocumentIndex(req.index_name);
62
+ const metadata = (0, helpers_1.fromProtoMetadata)(req.metadata);
63
+ const metadataArg = Object.keys(metadata).length > 0 ? metadata : undefined;
64
+ const doc = yield docIndex.upsertDocument(req.uri, req.text, req.doc_type || undefined, metadataArg);
65
+ return { document_id: doc.id };
66
+ })),
67
+ DeleteDocument: (0, helpers_1.wrapHandler)((call) => __awaiter(this, void 0, void 0, function* () {
68
+ const req = call.request;
69
+ if (!req.index_name) {
70
+ throw (0, helpers_1.grpcError)(grpc.status.INVALID_ARGUMENT, 'index_name is required');
71
+ }
72
+ if (!req.uri) {
73
+ throw (0, helpers_1.grpcError)(grpc.status.INVALID_ARGUMENT, 'uri is required');
74
+ }
75
+ const { docIndex } = manager.requireDocumentIndex(req.index_name);
76
+ yield docIndex.deleteDocument(req.uri);
77
+ return {};
78
+ })),
79
+ ListDocuments: (0, helpers_1.wrapHandler)((call) => __awaiter(this, void 0, void 0, function* () {
80
+ const req = call.request;
81
+ if (!req.index_name) {
82
+ throw (0, helpers_1.grpcError)(grpc.status.INVALID_ARGUMENT, 'index_name is required');
83
+ }
84
+ const { docIndex } = manager.requireDocumentIndex(req.index_name);
85
+ const docs = yield docIndex.listDocuments();
86
+ return {
87
+ documents: docs.map(doc => ({
88
+ uri: doc.uri,
89
+ document_id: doc.id,
90
+ })),
91
+ };
92
+ })),
93
+ };
94
+ }
95
+ //# sourceMappingURL=documentHandlers.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"documentHandlers.js","sourceRoot":"","sources":["../../../src/server/handlers/documentHandlers.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAIA,wDAsDC;AA1DD,oDAAsC;AAEtC,uCAAsE;AAEtE,SAAgB,sBAAsB,CAAC,OAAqB;IACxD,OAAO;QACH,cAAc,EAAE,IAAA,qBAAW,EAAC,CAAO,IAAoC,EAAE,EAAE;YACvE,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC;YACzB,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;gBAClB,MAAM,IAAA,mBAAS,EAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,wBAAwB,CAAC,CAAC;YAC5E,CAAC;YACD,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;gBACX,MAAM,IAAA,mBAAS,EAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,iBAAiB,CAAC,CAAC;YACrE,CAAC;YACD,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;gBACZ,MAAM,IAAA,mBAAS,EAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;YACtE,CAAC;YACD,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,oBAAoB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAClE,MAAM,QAAQ,GAAG,IAAA,2BAAiB,EAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;YAE5E,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,cAAc,CACrC,GAAG,CAAC,GAAG,EACP,GAAG,CAAC,IAAI,EACR,GAAG,CAAC,QAAQ,IAAI,SAAS,EACzB,WAAW,CACd,CAAC;YACF,OAAO,EAAE,WAAW,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC;QACnC,CAAC,CAAA,CAAC;QAEF,cAAc,EAAE,IAAA,qBAAW,EAAC,CAAO,IAAoC,EAAE,EAAE;YACvE,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC;YACzB,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;gBAClB,MAAM,IAAA,mBAAS,EAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,wBAAwB,CAAC,CAAC;YAC5E,CAAC;YACD,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;gBACX,MAAM,IAAA,mBAAS,EAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,iBAAiB,CAAC,CAAC;YACrE,CAAC;YACD,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,oBAAoB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAClE,MAAM,QAAQ,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACvC,OAAO,EAAE,CAAC;QACd,CAAC,CAAA,CAAC;QAEF,aAAa,EAAE,IAAA,qBAAW,EAAC,CAAO,IAAoC,EAAE,EAAE;YACtE,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC;YACzB,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;gBAClB,MAAM,IAAA,mBAAS,EAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,wBAAwB,CAAC,CAAC;YAC5E,CAAC;YACD,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,oBAAoB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAClE,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC5C,OAAO;gBACH,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACxB,GAAG,EAAE,GAAG,CAAC,GAAG;oBACZ,WAAW,EAAE,GAAG,CAAC,EAAE;iBACtB,CAAC,CAAC;aACN,CAAC;QACN,CAAC,CAAA,CAAC;KACL,CAAC;AACN,CAAC"}
@@ -0,0 +1,23 @@
1
+ import * as grpc from '@grpc/grpc-js';
2
+ import { MetadataTypes } from '../../types';
3
+ /**
4
+ * Converts proto MetadataValue map to a plain JS record.
5
+ */
6
+ export declare function fromProtoMetadata(protoMeta: Record<string, any> | undefined): Record<string, MetadataTypes>;
7
+ /**
8
+ * Converts a plain JS metadata record to proto MetadataValue map.
9
+ */
10
+ export declare function toProtoMetadata(meta: Record<string, MetadataTypes> | undefined): Record<string, any>;
11
+ /**
12
+ * Parses the filter_json field from a MetadataFilter proto message.
13
+ */
14
+ export declare function parseFilterJson(filter: any): Record<string, any> | undefined;
15
+ /**
16
+ * Creates a gRPC ServiceError with the given status code and message.
17
+ */
18
+ export declare function grpcError(code: grpc.status, message: string): grpc.ServiceError;
19
+ /**
20
+ * Wraps an async handler function with standard error mapping.
21
+ */
22
+ export declare function wrapHandler<TReq, TRes>(handler: (call: grpc.ServerUnaryCall<TReq, TRes>) => Promise<TRes>): grpc.handleUnaryCall<TReq, TRes>;
23
+ //# sourceMappingURL=helpers.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"helpers.d.ts","sourceRoot":"","sources":["../../../src/server/handlers/helpers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAe3G;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,GAAG,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAapG;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,GAAG,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS,CAS5E;AAED;;GAEG;AACH,wBAAgB,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC,YAAY,CAM/E;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,IAAI,EAClC,OAAO,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,GACnE,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,CAqBlC"}