@zvec/zvec-grep 0.1.5 → 0.1.7

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 (358) hide show
  1. package/README.md +154 -236
  2. package/README_CN.md +142 -233
  3. package/dist/authorization/index.d.ts +7 -0
  4. package/dist/authorization/index.js +7 -0
  5. package/dist/authorization/index.js.map +1 -0
  6. package/dist/authorization/manager.d.ts +8 -0
  7. package/dist/authorization/manager.js +19 -0
  8. package/dist/authorization/manager.js.map +1 -0
  9. package/dist/authorization/operation.d.ts +7 -0
  10. package/dist/authorization/operation.js +50 -0
  11. package/dist/authorization/operation.js.map +1 -0
  12. package/dist/authorization/planner.d.ts +21 -0
  13. package/dist/authorization/planner.js +97 -0
  14. package/dist/authorization/planner.js.map +1 -0
  15. package/dist/authorization/prompt.d.ts +11 -0
  16. package/dist/authorization/prompt.js +62 -0
  17. package/dist/authorization/prompt.js.map +1 -0
  18. package/dist/authorization/store.d.ts +20 -0
  19. package/dist/authorization/store.js +251 -0
  20. package/dist/authorization/store.js.map +1 -0
  21. package/dist/authorization/target.d.ts +15 -0
  22. package/dist/authorization/target.js +49 -0
  23. package/dist/authorization/target.js.map +1 -0
  24. package/dist/authorization/types.d.ts +61 -0
  25. package/dist/authorization/types.js +2 -0
  26. package/dist/authorization/types.js.map +1 -0
  27. package/dist/cli/args.d.ts +1 -1
  28. package/dist/cli/args.js +338 -55
  29. package/dist/cli/args.js.map +1 -1
  30. package/dist/cli/auth.d.ts +21 -0
  31. package/dist/cli/auth.js +209 -0
  32. package/dist/cli/auth.js.map +1 -0
  33. package/dist/cli/commands.js +469 -625
  34. package/dist/cli/commands.js.map +1 -1
  35. package/dist/cli/errors.js +2 -2
  36. package/dist/cli/errors.js.map +1 -1
  37. package/dist/cli/format/context.d.ts +4 -0
  38. package/dist/cli/format/context.js +311 -44
  39. package/dist/cli/format/context.js.map +1 -1
  40. package/dist/cli/format/debug.js +12 -6
  41. package/dist/cli/format/debug.js.map +1 -1
  42. package/dist/cli/format/progress.d.ts +4 -0
  43. package/dist/cli/format/progress.js +219 -46
  44. package/dist/cli/format/progress.js.map +1 -1
  45. package/dist/cli/format/status.d.ts +79 -6
  46. package/dist/cli/format/status.js +341 -111
  47. package/dist/cli/format/status.js.map +1 -1
  48. package/dist/cli/help.js +395 -63
  49. package/dist/cli/help.js.map +1 -1
  50. package/dist/cli/index.d.ts +1 -1
  51. package/dist/cli/index.js +1 -1
  52. package/dist/cli/install.d.ts +4 -0
  53. package/dist/cli/install.js +963 -0
  54. package/dist/cli/install.js.map +1 -0
  55. package/dist/cli/managed-rg.d.ts +10 -0
  56. package/dist/cli/managed-rg.js +329 -0
  57. package/dist/cli/managed-rg.js.map +1 -0
  58. package/dist/cli/server-search.d.ts +3 -0
  59. package/dist/cli/server-search.js +12 -0
  60. package/dist/cli/server-search.js.map +1 -0
  61. package/dist/cli/types.d.ts +23 -6
  62. package/dist/cli/types.js.map +1 -1
  63. package/dist/client/daemon-client.d.ts +31 -0
  64. package/dist/client/daemon-client.js +203 -0
  65. package/dist/client/daemon-client.js.map +1 -0
  66. package/dist/client/mode-router.d.ts +9 -0
  67. package/dist/client/mode-router.js +28 -0
  68. package/dist/client/mode-router.js.map +1 -0
  69. package/dist/client/search-policy.d.ts +11 -0
  70. package/dist/client/search-policy.js +15 -0
  71. package/dist/client/search-policy.js.map +1 -0
  72. package/dist/daemon/backend.d.ts +77 -0
  73. package/dist/daemon/backend.js +1087 -0
  74. package/dist/daemon/backend.js.map +1 -0
  75. package/dist/daemon/change-set.d.ts +25 -0
  76. package/dist/daemon/change-set.js +101 -0
  77. package/dist/daemon/change-set.js.map +1 -0
  78. package/dist/daemon/config.d.ts +24 -0
  79. package/dist/daemon/config.js +88 -0
  80. package/dist/daemon/config.js.map +1 -0
  81. package/dist/daemon/errors.d.ts +5 -0
  82. package/dist/daemon/errors.js +11 -0
  83. package/dist/daemon/errors.js.map +1 -0
  84. package/dist/daemon/http-server.d.ts +30 -0
  85. package/dist/daemon/http-server.js +288 -0
  86. package/dist/daemon/http-server.js.map +1 -0
  87. package/dist/daemon/index-coordinator.d.ts +24 -0
  88. package/dist/daemon/index-coordinator.js +68 -0
  89. package/dist/daemon/index-coordinator.js.map +1 -0
  90. package/dist/daemon/job-scheduler.d.ts +70 -0
  91. package/dist/daemon/job-scheduler.js +366 -0
  92. package/dist/daemon/job-scheduler.js.map +1 -0
  93. package/dist/daemon/logger.d.ts +9 -0
  94. package/dist/daemon/logger.js +53 -0
  95. package/dist/daemon/logger.js.map +1 -0
  96. package/dist/daemon/model-pool.d.ts +45 -0
  97. package/dist/daemon/model-pool.js +187 -0
  98. package/dist/daemon/model-pool.js.map +1 -0
  99. package/dist/daemon/root-lease.d.ts +14 -0
  100. package/dist/daemon/root-lease.js +188 -0
  101. package/dist/daemon/root-lease.js.map +1 -0
  102. package/dist/daemon/root-runtime.d.ts +78 -0
  103. package/dist/daemon/root-runtime.js +333 -0
  104. package/dist/daemon/root-runtime.js.map +1 -0
  105. package/dist/daemon/runtime-manager.d.ts +46 -0
  106. package/dist/daemon/runtime-manager.js +246 -0
  107. package/dist/daemon/runtime-manager.js.map +1 -0
  108. package/dist/daemon/runtime.d.ts +12 -0
  109. package/dist/daemon/runtime.js +102 -0
  110. package/dist/daemon/runtime.js.map +1 -0
  111. package/dist/daemon/server-controller.d.ts +39 -0
  112. package/dist/daemon/server-controller.js +254 -0
  113. package/dist/daemon/server-controller.js.map +1 -0
  114. package/dist/daemon/watch-manager.d.ts +50 -0
  115. package/dist/daemon/watch-manager.js +293 -0
  116. package/dist/daemon/watch-manager.js.map +1 -0
  117. package/dist/daemon/workspace-read-session-cache.d.ts +31 -0
  118. package/dist/daemon/workspace-read-session-cache.js +115 -0
  119. package/dist/daemon/workspace-read-session-cache.js.map +1 -0
  120. package/dist/engine/code-formats.d.ts +3 -0
  121. package/dist/engine/code-formats.js +14 -0
  122. package/dist/engine/code-formats.js.map +1 -0
  123. package/dist/engine/config.d.ts +30 -8
  124. package/dist/engine/config.js +171 -45
  125. package/dist/engine/config.js.map +1 -1
  126. package/dist/engine/{errors/index.d.ts → errors.d.ts} +6 -1
  127. package/dist/engine/{errors/details.js → errors.js} +17 -4
  128. package/dist/engine/errors.js.map +1 -0
  129. package/dist/engine/extraction/code/adapter.d.ts +1 -1
  130. package/dist/engine/extraction/code/adapter.js +1 -1
  131. package/dist/engine/extraction/code/adapter.js.map +1 -1
  132. package/dist/engine/extraction/code/extractor.d.ts +5 -11
  133. package/dist/engine/extraction/code/extractor.js +116 -48
  134. package/dist/engine/extraction/code/extractor.js.map +1 -1
  135. package/dist/engine/extraction/code/families/c-family.js +1 -1
  136. package/dist/engine/extraction/code/families/c-family.js.map +1 -1
  137. package/dist/engine/extraction/code/families/js-ts.d.ts +1 -1
  138. package/dist/engine/extraction/code/families/metadata.d.ts +1 -1
  139. package/dist/engine/extraction/code/tree-sitter/grammar.js.map +1 -0
  140. package/dist/engine/extraction/code/tree-sitter/nodes.js.map +1 -0
  141. package/dist/engine/extraction/code/tree-sitter/parser.js.map +1 -0
  142. package/dist/engine/extraction/image/extractor.d.ts +5 -0
  143. package/dist/engine/extraction/image/extractor.js +35 -0
  144. package/dist/engine/extraction/image/extractor.js.map +1 -0
  145. package/dist/engine/extraction/index.d.ts +4 -44
  146. package/dist/engine/extraction/index.js +2 -280
  147. package/dist/engine/extraction/index.js.map +1 -1
  148. package/dist/engine/extraction/markdown/extractor.d.ts +5 -11
  149. package/dist/engine/extraction/markdown/extractor.js +38 -17
  150. package/dist/engine/extraction/markdown/extractor.js.map +1 -1
  151. package/dist/engine/extraction/runtime.d.ts +4 -0
  152. package/dist/engine/extraction/runtime.js +26 -0
  153. package/dist/engine/extraction/runtime.js.map +1 -0
  154. package/dist/engine/extraction/source.d.ts +18 -0
  155. package/dist/engine/extraction/source.js +22 -0
  156. package/dist/engine/extraction/source.js.map +1 -0
  157. package/dist/engine/extraction/text/extractor.d.ts +7 -0
  158. package/dist/engine/extraction/text/extractor.js +168 -0
  159. package/dist/engine/extraction/text/extractor.js.map +1 -0
  160. package/dist/engine/extraction/types.d.ts +4 -0
  161. package/dist/engine/{models → extraction}/types.js.map +1 -1
  162. package/dist/engine/extraction/vector-content.d.ts +5 -0
  163. package/dist/engine/extraction/vector-content.js +95 -0
  164. package/dist/engine/extraction/vector-content.js.map +1 -0
  165. package/dist/engine/file-size-policy.d.ts +6 -0
  166. package/dist/engine/file-size-policy.js +20 -0
  167. package/dist/engine/file-size-policy.js.map +1 -0
  168. package/dist/engine/file-type.d.ts +15 -0
  169. package/dist/engine/file-type.js +152 -0
  170. package/dist/engine/file-type.js.map +1 -0
  171. package/dist/engine/index-status.d.ts +9 -0
  172. package/dist/engine/index-status.js +38 -0
  173. package/dist/engine/index-status.js.map +1 -0
  174. package/dist/engine/manifest.d.ts +13 -0
  175. package/dist/engine/manifest.js +124 -0
  176. package/dist/engine/manifest.js.map +1 -0
  177. package/dist/engine/models/{providers/llama-cpp/embedding.d.ts → backends/llama-cpp.d.ts} +25 -20
  178. package/dist/engine/models/{providers/llama-cpp/embedding.js → backends/llama-cpp.js} +127 -106
  179. package/dist/engine/models/backends/llama-cpp.js.map +1 -0
  180. package/dist/engine/models/backends/model2vec-runtime.d.ts +63 -0
  181. package/dist/engine/models/backends/model2vec-runtime.js +112 -0
  182. package/dist/engine/models/backends/model2vec-runtime.js.map +1 -0
  183. package/dist/engine/models/backends/model2vec-tokenizer.d.ts +2 -0
  184. package/dist/engine/models/backends/model2vec-tokenizer.js +42 -0
  185. package/dist/engine/models/backends/model2vec-tokenizer.js.map +1 -0
  186. package/dist/engine/models/backends/model2vec-worker-pool.d.ts +25 -0
  187. package/dist/engine/models/backends/model2vec-worker-pool.js +264 -0
  188. package/dist/engine/models/backends/model2vec-worker-pool.js.map +1 -0
  189. package/dist/engine/models/backends/model2vec-worker.d.ts +1 -0
  190. package/dist/engine/models/backends/model2vec-worker.js +41 -0
  191. package/dist/engine/models/backends/model2vec-worker.js.map +1 -0
  192. package/dist/engine/models/backends/model2vec.d.ts +40 -0
  193. package/dist/engine/models/backends/model2vec.js +311 -0
  194. package/dist/engine/models/backends/model2vec.js.map +1 -0
  195. package/dist/engine/models/backends/qwen.d.ts +44 -0
  196. package/dist/engine/models/{providers/qwen/embedding.js → backends/qwen.js} +155 -94
  197. package/dist/engine/models/backends/qwen.js.map +1 -0
  198. package/dist/engine/models/backends/transformers-js.d.ts +87 -0
  199. package/dist/engine/models/backends/transformers-js.js +316 -0
  200. package/dist/engine/models/backends/transformers-js.js.map +1 -0
  201. package/dist/engine/models/catalog.d.ts +207 -23
  202. package/dist/engine/models/catalog.js +191 -25
  203. package/dist/engine/models/catalog.js.map +1 -1
  204. package/dist/engine/models/download-progress.d.ts +15 -0
  205. package/dist/engine/models/download-progress.js +48 -0
  206. package/dist/engine/models/download-progress.js.map +1 -0
  207. package/dist/engine/models/embeddings.d.ts +59 -18
  208. package/dist/engine/models/embeddings.js +52 -28
  209. package/dist/engine/models/embeddings.js.map +1 -1
  210. package/dist/engine/models/factory.d.ts +2 -5
  211. package/dist/engine/models/factory.js +41 -56
  212. package/dist/engine/models/factory.js.map +1 -1
  213. package/dist/engine/models/index.d.ts +4 -6
  214. package/dist/engine/models/index.js +4 -4
  215. package/dist/engine/models/index.js.map +1 -1
  216. package/dist/engine/models/ranking.d.ts +7 -3
  217. package/dist/engine/models/resolution.d.ts +8 -0
  218. package/dist/engine/models/resolution.js +22 -0
  219. package/dist/engine/models/resolution.js.map +1 -0
  220. package/dist/engine/pipeline/indexing/index.d.ts +9 -7
  221. package/dist/engine/pipeline/indexing/index.js +409 -165
  222. package/dist/engine/pipeline/indexing/index.js.map +1 -1
  223. package/dist/engine/pipeline/indexing/input-budget.d.ts +2 -0
  224. package/dist/engine/pipeline/indexing/input-budget.js +51 -0
  225. package/dist/engine/pipeline/indexing/input-budget.js.map +1 -0
  226. package/dist/engine/pipeline/indexing/root-paths.js +6 -6
  227. package/dist/engine/pipeline/indexing/root-paths.js.map +1 -1
  228. package/dist/engine/pipeline/indexing/scanner/index.d.ts +9 -2
  229. package/dist/engine/pipeline/indexing/scanner/index.js +291 -35
  230. package/dist/engine/pipeline/indexing/scanner/index.js.map +1 -1
  231. package/dist/engine/pipeline/search/index.d.ts +6 -6
  232. package/dist/engine/pipeline/search/index.js +8 -10
  233. package/dist/engine/pipeline/search/index.js.map +1 -1
  234. package/dist/engine/service/index.d.ts +3 -2
  235. package/dist/engine/service/index.js +1 -1
  236. package/dist/engine/service/index.js.map +1 -1
  237. package/dist/engine/service/root.d.ts +10 -12
  238. package/dist/engine/service/root.js +29 -25
  239. package/dist/engine/service/root.js.map +1 -1
  240. package/dist/engine/service/structure-enrichment.d.ts +1 -1
  241. package/dist/engine/service/structure-enrichment.js +20 -22
  242. package/dist/engine/service/structure-enrichment.js.map +1 -1
  243. package/dist/engine/service/types.d.ts +40 -38
  244. package/dist/engine/service/workspace-index.d.ts +26 -0
  245. package/dist/engine/service/workspace-index.js +157 -0
  246. package/dist/engine/service/workspace-index.js.map +1 -0
  247. package/dist/engine/service/zvec-grep.d.ts +12 -1
  248. package/dist/engine/service/zvec-grep.js +698 -489
  249. package/dist/engine/service/zvec-grep.js.map +1 -1
  250. package/dist/engine/storage/index.d.ts +30 -20
  251. package/dist/engine/storage/index.js +2 -1
  252. package/dist/engine/storage/index.js.map +1 -1
  253. package/dist/engine/storage/layout.d.ts +10 -0
  254. package/dist/engine/storage/layout.js +29 -0
  255. package/dist/engine/storage/layout.js.map +1 -0
  256. package/dist/engine/storage/zvec.d.ts +4 -67
  257. package/dist/engine/storage/zvec.js +115 -85
  258. package/dist/engine/storage/zvec.js.map +1 -1
  259. package/dist/engine/types.d.ts +29 -23
  260. package/dist/engine/types.js +1 -1
  261. package/dist/engine/types.js.map +1 -1
  262. package/dist/engine/utils/daemon-lease.d.ts +17 -0
  263. package/dist/engine/utils/daemon-lease.js +175 -0
  264. package/dist/engine/utils/daemon-lease.js.map +1 -0
  265. package/dist/engine/utils/file-selection.js +38 -1
  266. package/dist/engine/utils/file-selection.js.map +1 -1
  267. package/dist/engine/utils/lock.js +1 -1
  268. package/dist/engine/utils/lock.js.map +1 -1
  269. package/dist/index-progress.d.ts +10 -0
  270. package/dist/index-progress.js +159 -0
  271. package/dist/index-progress.js.map +1 -0
  272. package/dist/index.d.ts +4 -4
  273. package/dist/index.js +1 -1
  274. package/dist/index.js.map +1 -1
  275. package/dist/mcp/http-transport.d.ts +27 -0
  276. package/dist/mcp/http-transport.js +182 -0
  277. package/dist/mcp/http-transport.js.map +1 -0
  278. package/dist/mcp/input-normalization.d.ts +38 -0
  279. package/dist/mcp/input-normalization.js +104 -0
  280. package/dist/mcp/input-normalization.js.map +1 -0
  281. package/dist/mcp/progress-heartbeat.d.ts +17 -0
  282. package/dist/mcp/progress-heartbeat.js +46 -0
  283. package/dist/mcp/progress-heartbeat.js.map +1 -0
  284. package/dist/mcp/request-metadata.d.ts +2 -0
  285. package/dist/mcp/request-metadata.js +11 -0
  286. package/dist/mcp/request-metadata.js.map +1 -0
  287. package/dist/mcp/request-state.d.ts +40 -0
  288. package/dist/mcp/request-state.js +210 -0
  289. package/dist/mcp/request-state.js.map +1 -0
  290. package/dist/mcp/result-format.d.ts +13 -0
  291. package/dist/mcp/result-format.js +75 -0
  292. package/dist/mcp/result-format.js.map +1 -0
  293. package/dist/mcp/schemas.d.ts +672 -0
  294. package/dist/mcp/schemas.js +506 -0
  295. package/dist/mcp/schemas.js.map +1 -0
  296. package/dist/mcp/stdio-bridge.d.ts +9 -0
  297. package/dist/mcp/stdio-bridge.js +131 -0
  298. package/dist/mcp/stdio-bridge.js.map +1 -0
  299. package/dist/mcp/tools.d.ts +153 -0
  300. package/dist/mcp/tools.js +482 -0
  301. package/dist/mcp/tools.js.map +1 -0
  302. package/dist/mcp/toolset.d.ts +5 -0
  303. package/dist/mcp/toolset.js +12 -0
  304. package/dist/mcp/toolset.js.map +1 -0
  305. package/dist/observability/trace-context.d.ts +11 -0
  306. package/dist/observability/trace-context.js +138 -0
  307. package/dist/observability/trace-context.js.map +1 -0
  308. package/dist/prompts/zvec-grep-guidance.d.ts +2 -0
  309. package/dist/prompts/zvec-grep-guidance.js +9 -0
  310. package/dist/prompts/zvec-grep-guidance.js.map +1 -0
  311. package/docs/01-agents.md +132 -0
  312. package/docs/02-cli.md +239 -0
  313. package/docs/03-mcp.md +185 -0
  314. package/docs/04-pipeline.md +226 -0
  315. package/docs/05-architecture.md +86 -0
  316. package/docs/06-server.md +215 -0
  317. package/docs/07-embedding.md +182 -0
  318. package/docs/08-roadmap.md +85 -0
  319. package/docs/README.md +53 -0
  320. package/package.json +20 -11
  321. package/dist/cli/mcp.d.ts +0 -2
  322. package/dist/cli/mcp.js +0 -354
  323. package/dist/cli/mcp.js.map +0 -1
  324. package/dist/engine/collection/index.d.ts +0 -55
  325. package/dist/engine/collection/index.js +0 -710
  326. package/dist/engine/collection/index.js.map +0 -1
  327. package/dist/engine/errors/details.d.ts +0 -6
  328. package/dist/engine/errors/details.js.map +0 -1
  329. package/dist/engine/errors/index.js +0 -16
  330. package/dist/engine/errors/index.js.map +0 -1
  331. package/dist/engine/extraction/tree-sitter/grammar.js.map +0 -1
  332. package/dist/engine/extraction/tree-sitter/nodes.js.map +0 -1
  333. package/dist/engine/extraction/tree-sitter/parser.js.map +0 -1
  334. package/dist/engine/files/file-type.d.ts +0 -5
  335. package/dist/engine/files/file-type.js +0 -119
  336. package/dist/engine/files/file-type.js.map +0 -1
  337. package/dist/engine/models/providers/index.d.ts +0 -2
  338. package/dist/engine/models/providers/index.js +0 -3
  339. package/dist/engine/models/providers/index.js.map +0 -1
  340. package/dist/engine/models/providers/llama-cpp/embedding.js.map +0 -1
  341. package/dist/engine/models/providers/llama-cpp/index.d.ts +0 -1
  342. package/dist/engine/models/providers/llama-cpp/index.js +0 -2
  343. package/dist/engine/models/providers/llama-cpp/index.js.map +0 -1
  344. package/dist/engine/models/providers/qwen/embedding.d.ts +0 -42
  345. package/dist/engine/models/providers/qwen/embedding.js.map +0 -1
  346. package/dist/engine/models/providers/qwen/index.d.ts +0 -1
  347. package/dist/engine/models/providers/qwen/index.js +0 -2
  348. package/dist/engine/models/providers/qwen/index.js.map +0 -1
  349. package/dist/engine/models/types.d.ts +0 -30
  350. package/skills/zvec-grep/SKILL.md +0 -151
  351. package/skills/zvec-grep/agents/openai.yaml +0 -4
  352. /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/grammar.d.ts +0 -0
  353. /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/grammar.js +0 -0
  354. /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/nodes.d.ts +0 -0
  355. /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/nodes.js +0 -0
  356. /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/parser.d.ts +0 -0
  357. /package/dist/engine/extraction/{tree-sitter → code/tree-sitter}/parser.js +0 -0
  358. /package/dist/engine/{models → extraction}/types.js +0 -0
@@ -0,0 +1,46 @@
1
+ import { type ZvecGrepInfoResult } from "../engine/service/index.js";
2
+ import type { CreateZvecGrepOptions } from "../engine/service/types.js";
3
+ import type { EmbeddingModelLoadRequest, EmbeddingModelPool } from "./model-pool.js";
4
+ import { RootRuntime } from "./root-runtime.js";
5
+ import { RootLeaseManager } from "./root-lease.js";
6
+ export type RuntimeManagerOptions = {
7
+ modelPool: EmbeddingModelPool;
8
+ serviceOptions?: CreateZvecGrepOptions;
9
+ readSessionIdleTtlMs?: number;
10
+ runtimeIdleTtlMs?: number;
11
+ onRuntimeEvicted?: (canonicalRoot: string) => void | Promise<void>;
12
+ rootLeaseManager?: RootLeaseManager;
13
+ createRuntime?: (input: {
14
+ canonicalRoot: string;
15
+ modelLoadRequest?: EmbeddingModelLoadRequest;
16
+ modelPool: EmbeddingModelPool;
17
+ }) => RootRuntime | Promise<RootRuntime>;
18
+ };
19
+ export type RuntimeManagerSnapshot = {
20
+ activeRuntimes: number;
21
+ };
22
+ export declare class RuntimeManager {
23
+ private readonly options;
24
+ private readonly runtimes;
25
+ private readonly creating;
26
+ private readonly aliases;
27
+ private readonly rootLeaseManager;
28
+ private readonly idleTimers;
29
+ private closed;
30
+ constructor(options: RuntimeManagerOptions);
31
+ get instanceToken(): string;
32
+ activate(requestedRoot: string): Promise<RootRuntime>;
33
+ activateForIndex(requestedRoot: string): Promise<RootRuntime>;
34
+ peek(requestedRoot: string): Promise<RootRuntime | undefined>;
35
+ getByCanonicalRoot(canonicalRoot: string): RootRuntime | undefined;
36
+ getByRequestedRoot(requestedRoot: string): RootRuntime | undefined;
37
+ snapshot(): RuntimeManagerSnapshot;
38
+ evict(canonicalRoot: string): Promise<boolean>;
39
+ close(): Promise<void>;
40
+ private getOrCreate;
41
+ private createRuntime;
42
+ private touchRuntime;
43
+ private evictIfIdle;
44
+ }
45
+ export declare function inspectRoot(requestedRoot: string, serviceOptions?: CreateZvecGrepOptions, includeStatus?: boolean): Promise<ZvecGrepInfoResult>;
46
+ export declare function resolveRequestedRoot(requestedRoot: string, writable: boolean): Promise<string>;
@@ -0,0 +1,246 @@
1
+ import { access, realpath, stat } from "node:fs/promises";
2
+ import { constants as fsConstants } from "node:fs";
3
+ import { isAbsolute } from "node:path";
4
+ import { createZvecGrep, } from "../engine/service/index.js";
5
+ import { DaemonError } from "./errors.js";
6
+ import { RootRuntime } from "./root-runtime.js";
7
+ import { RootLeaseManager } from "./root-lease.js";
8
+ export class RuntimeManager {
9
+ options;
10
+ runtimes = new Map();
11
+ creating = new Map();
12
+ aliases = new Map();
13
+ rootLeaseManager;
14
+ idleTimers = new Map();
15
+ closed = false;
16
+ constructor(options) {
17
+ this.options = options;
18
+ this.rootLeaseManager = options.rootLeaseManager ?? new RootLeaseManager();
19
+ }
20
+ get instanceToken() {
21
+ return this.rootLeaseManager.instanceToken;
22
+ }
23
+ async activate(requestedRoot) {
24
+ if (this.closed) {
25
+ throw new DaemonError("DAEMON_SHUTTING_DOWN", "The daemon is shutting down.", true);
26
+ }
27
+ const canonicalRequestedRoot = await resolveRequestedRoot(requestedRoot, false);
28
+ const activeRequestedRoot = this.runtimes.get(this.aliases.get(canonicalRequestedRoot) ?? canonicalRequestedRoot);
29
+ if (activeRequestedRoot &&
30
+ (activeRequestedRoot.snapshot().writerPending ||
31
+ activeRequestedRoot.needsReconciliation())) {
32
+ return activeRequestedRoot;
33
+ }
34
+ const creatingRequestedRoot = this.creating.get(this.aliases.get(canonicalRequestedRoot) ?? canonicalRequestedRoot);
35
+ if (creatingRequestedRoot) {
36
+ return creatingRequestedRoot;
37
+ }
38
+ const info = await inspectRoot(requestedRoot, this.options.serviceOptions, false);
39
+ if (!info.indexed || !info.workspaceIndex?.embedding) {
40
+ throw new DaemonError("INDEX_MISSING", `Indexed search requires a built zvec-grep index for ${info.root}. Use an available exact-search fallback when it is sufficient. Creating or rebuilding a persistent index requires explicit user authorization.`);
41
+ }
42
+ const canonicalRoot = await realpath(info.root);
43
+ this.aliases.set(canonicalRequestedRoot, canonicalRoot);
44
+ return this.getOrCreate(canonicalRoot, {
45
+ model: {
46
+ provider: info.workspaceIndex.embedding.provider,
47
+ name: info.workspaceIndex.embedding.model,
48
+ },
49
+ });
50
+ }
51
+ async activateForIndex(requestedRoot) {
52
+ if (this.closed) {
53
+ throw new DaemonError("DAEMON_SHUTTING_DOWN", "The daemon is shutting down.", true);
54
+ }
55
+ const canonicalRequestedRoot = await resolveRequestedRoot(requestedRoot, true);
56
+ const activeRequestedRoot = this.runtimes.get(this.aliases.get(canonicalRequestedRoot) ?? canonicalRequestedRoot);
57
+ if (activeRequestedRoot) {
58
+ return activeRequestedRoot;
59
+ }
60
+ const creatingRequestedRoot = this.creating.get(this.aliases.get(canonicalRequestedRoot) ?? canonicalRequestedRoot);
61
+ if (creatingRequestedRoot) {
62
+ return creatingRequestedRoot;
63
+ }
64
+ const info = await inspectRoot(canonicalRequestedRoot, this.options.serviceOptions, false);
65
+ const canonicalRoot = await resolveRequestedRoot(info.root, true);
66
+ this.aliases.set(canonicalRequestedRoot, canonicalRoot);
67
+ return this.getOrCreate(canonicalRoot);
68
+ }
69
+ async peek(requestedRoot) {
70
+ const info = await inspectRoot(requestedRoot, this.options.serviceOptions, false);
71
+ const canonicalRoot = await realpath(info.root);
72
+ return this.runtimes.get(canonicalRoot);
73
+ }
74
+ getByCanonicalRoot(canonicalRoot) {
75
+ return this.runtimes.get(canonicalRoot);
76
+ }
77
+ getByRequestedRoot(requestedRoot) {
78
+ return this.runtimes.get(this.aliases.get(requestedRoot) ?? requestedRoot);
79
+ }
80
+ snapshot() {
81
+ return { activeRuntimes: this.runtimes.size };
82
+ }
83
+ async evict(canonicalRoot) {
84
+ const creating = this.creating.get(canonicalRoot);
85
+ if (creating) {
86
+ await creating;
87
+ }
88
+ const timer = this.idleTimers.get(canonicalRoot);
89
+ if (timer)
90
+ clearTimeout(timer);
91
+ this.idleTimers.delete(canonicalRoot);
92
+ const runtime = this.runtimes.get(canonicalRoot);
93
+ this.runtimes.delete(canonicalRoot);
94
+ for (const [alias, target] of this.aliases) {
95
+ if (alias === canonicalRoot || target === canonicalRoot) {
96
+ this.aliases.delete(alias);
97
+ }
98
+ }
99
+ if (!runtime)
100
+ return false;
101
+ try {
102
+ await this.options.onRuntimeEvicted?.(canonicalRoot);
103
+ }
104
+ finally {
105
+ await runtime.close();
106
+ }
107
+ return true;
108
+ }
109
+ async close() {
110
+ if (this.closed) {
111
+ return;
112
+ }
113
+ this.closed = true;
114
+ await Promise.allSettled(this.creating.values());
115
+ const runtimes = [...this.runtimes.values()];
116
+ for (const timer of this.idleTimers.values())
117
+ clearTimeout(timer);
118
+ this.idleTimers.clear();
119
+ this.runtimes.clear();
120
+ this.aliases.clear();
121
+ await Promise.all(runtimes.map((runtime) => runtime.close()));
122
+ await this.rootLeaseManager.close();
123
+ }
124
+ async getOrCreate(canonicalRoot, modelLoadRequest) {
125
+ const existing = this.runtimes.get(canonicalRoot);
126
+ if (existing) {
127
+ if (modelLoadRequest) {
128
+ existing.updateModelLoadRequest(modelLoadRequest);
129
+ }
130
+ this.touchRuntime(canonicalRoot);
131
+ return existing;
132
+ }
133
+ let pending = this.creating.get(canonicalRoot);
134
+ if (!pending) {
135
+ pending = this.createRuntime(canonicalRoot, modelLoadRequest);
136
+ this.creating.set(canonicalRoot, pending);
137
+ }
138
+ try {
139
+ const runtime = await pending;
140
+ if (modelLoadRequest) {
141
+ runtime.updateModelLoadRequest(modelLoadRequest);
142
+ }
143
+ this.touchRuntime(canonicalRoot);
144
+ return runtime;
145
+ }
146
+ finally {
147
+ this.creating.delete(canonicalRoot);
148
+ }
149
+ }
150
+ async createRuntime(canonicalRoot, modelLoadRequest) {
151
+ let runtime;
152
+ if (this.options.createRuntime) {
153
+ runtime = await this.options.createRuntime({
154
+ canonicalRoot,
155
+ modelLoadRequest,
156
+ modelPool: this.options.modelPool,
157
+ });
158
+ }
159
+ else {
160
+ const rootLease = await this.rootLeaseManager.acquire(canonicalRoot);
161
+ runtime = new RootRuntime({
162
+ canonicalRoot,
163
+ modelPool: this.options.modelPool,
164
+ modelLoadRequest,
165
+ rootLease,
166
+ readSessionIdleTtlMs: this.options.readSessionIdleTtlMs,
167
+ onActivity: () => this.touchRuntime(canonicalRoot),
168
+ });
169
+ }
170
+ if (this.closed) {
171
+ await runtime.close();
172
+ throw new DaemonError("DAEMON_SHUTTING_DOWN", "The daemon is shutting down.", true);
173
+ }
174
+ this.runtimes.set(canonicalRoot, runtime);
175
+ this.touchRuntime(canonicalRoot);
176
+ return runtime;
177
+ }
178
+ touchRuntime(canonicalRoot) {
179
+ const idleTtlMs = this.options.runtimeIdleTtlMs ?? 30 * 60_000;
180
+ const existing = this.idleTimers.get(canonicalRoot);
181
+ if (existing)
182
+ clearTimeout(existing);
183
+ if (idleTtlMs <= 0 || this.closed) {
184
+ return;
185
+ }
186
+ const timer = setTimeout(() => void this.evictIfIdle(canonicalRoot), idleTtlMs);
187
+ timer.unref?.();
188
+ this.idleTimers.set(canonicalRoot, timer);
189
+ }
190
+ async evictIfIdle(canonicalRoot) {
191
+ this.idleTimers.delete(canonicalRoot);
192
+ const runtime = this.runtimes.get(canonicalRoot);
193
+ if (!runtime) {
194
+ return;
195
+ }
196
+ const snapshot = runtime.snapshot();
197
+ if (snapshot.activeReaders > 0 ||
198
+ snapshot.activeOperations > 0 ||
199
+ snapshot.writerPending ||
200
+ snapshot.watcherPending) {
201
+ this.touchRuntime(canonicalRoot);
202
+ return;
203
+ }
204
+ await this.evict(canonicalRoot);
205
+ }
206
+ }
207
+ export async function inspectRoot(requestedRoot, serviceOptions = {}, includeStatus = true) {
208
+ const canonicalRequestedRoot = await resolveRequestedRoot(requestedRoot, false);
209
+ const service = await createZvecGrep({
210
+ ...serviceOptions,
211
+ root: canonicalRequestedRoot,
212
+ });
213
+ try {
214
+ return await service.info({ root: canonicalRequestedRoot, includeStatus });
215
+ }
216
+ finally {
217
+ await service.close();
218
+ }
219
+ }
220
+ export async function resolveRequestedRoot(requestedRoot, writable) {
221
+ if (!isAbsolute(requestedRoot)) {
222
+ throw new DaemonError("ROOT_NOT_ABSOLUTE", "root must be an absolute path.");
223
+ }
224
+ let rootStat;
225
+ try {
226
+ rootStat = await stat(requestedRoot);
227
+ }
228
+ catch (error) {
229
+ const code = error.code;
230
+ if (code === "EACCES" || code === "EPERM") {
231
+ throw new DaemonError("ROOT_PERMISSION_DENIED", "root is not readable.");
232
+ }
233
+ throw new DaemonError("ROOT_NOT_FOUND", "root does not exist.");
234
+ }
235
+ if (!rootStat.isDirectory()) {
236
+ throw new DaemonError("ROOT_NOT_FOUND", "root is not a directory.");
237
+ }
238
+ try {
239
+ await access(requestedRoot, writable ? fsConstants.R_OK | fsConstants.W_OK : fsConstants.R_OK);
240
+ }
241
+ catch {
242
+ throw new DaemonError("ROOT_PERMISSION_DENIED", writable ? "root is not writable." : "root is not readable.");
243
+ }
244
+ return realpath(requestedRoot);
245
+ }
246
+ //# sourceMappingURL=runtime-manager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"runtime-manager.js","sourceRoot":"","sources":["../../src/daemon/runtime-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,SAAS,IAAI,WAAW,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,EACL,cAAc,GAEf,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAK1C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAoBnD,MAAM,OAAO,cAAc;IAWI;IAVZ,QAAQ,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC1C,QAAQ,GAAG,IAAI,GAAG,EAAgC,CAAC;IACnD,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IACpC,gBAAgB,CAAmB;IACnC,UAAU,GAAG,IAAI,GAAG,EAGlC,CAAC;IACI,MAAM,GAAG,KAAK,CAAC;IAEvB,YAA6B,OAA8B;QAA9B,YAAO,GAAP,OAAO,CAAuB;QACzD,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,IAAI,gBAAgB,EAAE,CAAC;IAC7E,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,gBAAgB,CAAC,aAAa,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,aAAqB;QAClC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,WAAW,CACnB,sBAAsB,EACtB,8BAA8B,EAC9B,IAAI,CACL,CAAC;QACJ,CAAC;QACD,MAAM,sBAAsB,GAAG,MAAM,oBAAoB,CACvD,aAAa,EACb,KAAK,CACN,CAAC;QACF,MAAM,mBAAmB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAC3C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,sBAAsB,CACnE,CAAC;QACF,IACE,mBAAmB;YACnB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,CAAC,aAAa;gBAC3C,mBAAmB,CAAC,mBAAmB,EAAE,CAAC,EAC5C,CAAC;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QACD,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAC7C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,sBAAsB,CACnE,CAAC;QACF,IAAI,qBAAqB,EAAE,CAAC;YAC1B,OAAO,qBAAqB,CAAC;QAC/B,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,WAAW,CAC5B,aAAa,EACb,IAAI,CAAC,OAAO,CAAC,cAAc,EAC3B,KAAK,CACN,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,SAAS,EAAE,CAAC;YACrD,MAAM,IAAI,WAAW,CACnB,eAAe,EACf,uDAAuD,IAAI,CAAC,IAAI,iJAAiJ,CAClN,CAAC;QACJ,CAAC;QACD,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,aAAa,CAAC,CAAC;QACxD,OAAO,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE;YACrC,KAAK,EAAE;gBACL,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,QAAQ;gBAChD,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,KAAK;aAC1C;SACF,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,aAAqB;QAC1C,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,IAAI,WAAW,CACnB,sBAAsB,EACtB,8BAA8B,EAC9B,IAAI,CACL,CAAC;QACJ,CAAC;QACD,MAAM,sBAAsB,GAAG,MAAM,oBAAoB,CACvD,aAAa,EACb,IAAI,CACL,CAAC;QACF,MAAM,mBAAmB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAC3C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,sBAAsB,CACnE,CAAC;QACF,IAAI,mBAAmB,EAAE,CAAC;YACxB,OAAO,mBAAmB,CAAC;QAC7B,CAAC;QACD,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAC7C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,IAAI,sBAAsB,CACnE,CAAC;QACF,IAAI,qBAAqB,EAAE,CAAC;YAC1B,OAAO,qBAAqB,CAAC;QAC/B,CAAC;QACD,MAAM,IAAI,GAAG,MAAM,WAAW,CAC5B,sBAAsB,EACtB,IAAI,CAAC,OAAO,CAAC,cAAc,EAC3B,KAAK,CACN,CAAC;QACF,MAAM,aAAa,GAAG,MAAM,oBAAoB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAClE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,aAAa,CAAC,CAAC;QACxD,OAAO,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,aAAqB;QAC9B,MAAM,IAAI,GAAG,MAAM,WAAW,CAC5B,aAAa,EACb,IAAI,CAAC,OAAO,CAAC,cAAc,EAC3B,KAAK,CACN,CAAC;QACF,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC1C,CAAC;IAED,kBAAkB,CAAC,aAAqB;QACtC,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC1C,CAAC;IAED,kBAAkB,CAAC,aAAqB;QACtC,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,CAAC;IAC7E,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,aAAqB;QAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAClD,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,CAAC;QACjB,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACjD,IAAI,KAAK;YAAE,YAAY,CAAC,KAAK,CAAC,CAAC;QAC/B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACjD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACpC,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC3C,IAAI,KAAK,KAAK,aAAa,IAAI,MAAM,KAAK,aAAa,EAAE,CAAC;gBACxD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC,aAAa,CAAC,CAAC;QACvD,CAAC;gBAAS,CAAC;YACT,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;QACxB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7C,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE;YAAE,YAAY,CAAC,KAAK,CAAC,CAAC;QAClE,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAC9D,MAAM,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,aAAqB,EACrB,gBAA4C;QAE5C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAClD,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,gBAAgB,EAAE,CAAC;gBACrB,QAAQ,CAAC,sBAAsB,CAAC,gBAAgB,CAAC,CAAC;YACpD,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YACjC,OAAO,QAAQ,CAAC;QAClB,CAAC;QACD,IAAI,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;YAC9D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC;YAC9B,IAAI,gBAAgB,EAAE,CAAC;gBACrB,OAAO,CAAC,sBAAsB,CAAC,gBAAgB,CAAC,CAAC;YACnD,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YACjC,OAAO,OAAO,CAAC;QACjB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,aAAqB,EACrB,gBAA4C;QAE5C,IAAI,OAAoB,CAAC;QACzB,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YAC/B,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;gBACzC,aAAa;gBACb,gBAAgB;gBAChB,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;aAClC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YACrE,OAAO,GAAG,IAAI,WAAW,CAAC;gBACxB,aAAa;gBACb,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;gBACjC,gBAAgB;gBAChB,SAAS;gBACT,oBAAoB,EAAE,IAAI,CAAC,OAAO,CAAC,oBAAoB;gBACvD,UAAU,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC;aACnD,CAAC,CAAC;QACL,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;YACtB,MAAM,IAAI,WAAW,CACnB,sBAAsB,EACtB,8BAA8B,EAC9B,IAAI,CACL,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAC1C,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;QACjC,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,YAAY,CAAC,aAAqB;QACxC,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,gBAAgB,IAAI,EAAE,GAAG,MAAM,CAAC;QAC/D,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACpD,IAAI,QAAQ;YAAE,YAAY,CAAC,QAAQ,CAAC,CAAC;QACrC,IAAI,SAAS,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAClC,OAAO;QACT,CAAC;QACD,MAAM,KAAK,GAAG,UAAU,CACtB,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,EAC1C,SAAS,CACV,CAAC;QACF,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;QAChB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;IAC5C,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,aAAqB;QAC7C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QACD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QACpC,IACE,QAAQ,CAAC,aAAa,GAAG,CAAC;YAC1B,QAAQ,CAAC,gBAAgB,GAAG,CAAC;YAC7B,QAAQ,CAAC,aAAa;YACtB,QAAQ,CAAC,cAAc,EACvB,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;YACjC,OAAO;QACT,CAAC;QACD,MAAM,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IAClC,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,aAAqB,EACrB,iBAAwC,EAAE,EAC1C,aAAa,GAAG,IAAI;IAEpB,MAAM,sBAAsB,GAAG,MAAM,oBAAoB,CACvD,aAAa,EACb,KAAK,CACN,CAAC;IACF,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC;QACnC,GAAG,cAAc;QACjB,IAAI,EAAE,sBAAsB;KAC7B,CAAC,CAAC;IACH,IAAI,CAAC;QACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,aAAa,EAAE,CAAC,CAAC;IAC7E,CAAC;YAAS,CAAC;QACT,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,aAAqB,EACrB,QAAiB;IAEjB,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,WAAW,CACnB,mBAAmB,EACnB,gCAAgC,CACjC,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ,CAAC;IACb,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,CAAC;IACvC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,GAAI,KAA+B,CAAC,IAAI,CAAC;QACnD,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YAC1C,MAAM,IAAI,WAAW,CAAC,wBAAwB,EAAE,uBAAuB,CAAC,CAAC;QAC3E,CAAC;QACD,MAAM,IAAI,WAAW,CAAC,gBAAgB,EAAE,sBAAsB,CAAC,CAAC;IAClE,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;QAC5B,MAAM,IAAI,WAAW,CAAC,gBAAgB,EAAE,0BAA0B,CAAC,CAAC;IACtE,CAAC;IACD,IAAI,CAAC;QACH,MAAM,MAAM,CACV,aAAa,EACb,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAClE,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,WAAW,CACnB,wBAAwB,EACxB,QAAQ,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAC7D,CAAC;IACJ,CAAC;IAED,OAAO,QAAQ,CAAC,aAAa,CAAC,CAAC;AACjC,CAAC"}
@@ -0,0 +1,12 @@
1
+ import type { CreateZvecGrepOptions } from "../engine/service/types.js";
2
+ import { type McpToolset } from "../mcp/toolset.js";
3
+ export type RunDaemonOptions = {
4
+ version: string;
5
+ listen?: string;
6
+ token?: string;
7
+ tokenFile?: string;
8
+ home?: string;
9
+ mcpToolset?: McpToolset;
10
+ serviceOptions?: CreateZvecGrepOptions;
11
+ };
12
+ export declare function runDaemonForeground(options: RunDaemonOptions): Promise<void>;
@@ -0,0 +1,102 @@
1
+ import { DaemonBackend } from "./backend.js";
2
+ import { configuredListenAddress, resolveServerToken } from "./config.js";
3
+ import { DaemonHttpServer } from "./http-server.js";
4
+ import { DaemonInstanceLock } from "./server-controller.js";
5
+ import { createDaemonLogger } from "./logger.js";
6
+ import { MCP_TOOLSET_ENV, resolveMcpToolset, } from "../mcp/toolset.js";
7
+ import { loadOrCreateMcpRequestStateKey, PersistentRemoteEmbeddingRequestStateReplayGuard, } from "../mcp/request-state.js";
8
+ export async function runDaemonForeground(options) {
9
+ const mcpToolset = resolveMcpToolset(options.mcpToolset, process.env[MCP_TOOLSET_ENV]);
10
+ const listen = configuredListenAddress(options.listen);
11
+ const displayAddress = `http://${displayHost(listen.host)}:${listen.port}/mcp`;
12
+ const instanceLock = await DaemonInstanceLock.acquire(options.home, displayAddress, mcpToolset);
13
+ const logger = createDaemonLogger(options.home);
14
+ let auth;
15
+ try {
16
+ auth = await resolveServerToken({
17
+ token: options.token,
18
+ tokenFile: options.tokenFile,
19
+ home: options.home,
20
+ });
21
+ }
22
+ catch (error) {
23
+ await instanceLock.release();
24
+ throw error;
25
+ }
26
+ let requestStateKey;
27
+ try {
28
+ requestStateKey = await loadOrCreateMcpRequestStateKey(options.home);
29
+ }
30
+ catch (error) {
31
+ await instanceLock.release();
32
+ throw error;
33
+ }
34
+ const backend = new DaemonBackend({
35
+ version: options.version,
36
+ serviceOptions: options.serviceOptions,
37
+ logger,
38
+ });
39
+ let requestStop;
40
+ const httpServer = new DaemonHttpServer({
41
+ ...listen,
42
+ token: auth.token,
43
+ version: options.version,
44
+ mcpToolset,
45
+ backend,
46
+ requestStateKey,
47
+ requestStateReplayGuard: new PersistentRemoteEmbeddingRequestStateReplayGuard(options.home),
48
+ logger,
49
+ onShutdown: () => requestStop?.(),
50
+ });
51
+ let address;
52
+ try {
53
+ address = await httpServer.start();
54
+ }
55
+ catch (error) {
56
+ await backend.close();
57
+ await instanceLock.release();
58
+ throw error;
59
+ }
60
+ const stopped = new Promise((resolve) => {
61
+ let stopping = false;
62
+ const stop = () => {
63
+ if (stopping)
64
+ return;
65
+ stopping = true;
66
+ void (async () => {
67
+ await backend.close();
68
+ await httpServer.close();
69
+ await instanceLock.release();
70
+ logger.event("server.stopped", { pid: process.pid });
71
+ await logger.flush();
72
+ })().finally(resolve);
73
+ };
74
+ requestStop = stop;
75
+ process.once("SIGINT", stop);
76
+ process.once("SIGTERM", stop);
77
+ });
78
+ try {
79
+ await instanceLock.markReady();
80
+ }
81
+ catch (error) {
82
+ requestStop?.();
83
+ await stopped;
84
+ throw error;
85
+ }
86
+ logger.event("server.ready", {
87
+ host: listen.host,
88
+ port: address.port,
89
+ pid: process.pid,
90
+ mcp_toolset: mcpToolset,
91
+ });
92
+ console.log(`zvec-grep server listening on http://${displayHost(address.address)}:${address.port}/mcp`);
93
+ console.log(`MCP toolset: ${mcpToolset}`);
94
+ if (auth.tokenFile) {
95
+ console.log(`Bearer token file: ${auth.tokenFile}`);
96
+ }
97
+ await stopped;
98
+ }
99
+ function displayHost(host) {
100
+ return host.includes(":") ? `[${host}]` : host;
101
+ }
102
+ //# sourceMappingURL=runtime.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"runtime.js","sourceRoot":"","sources":["../../src/daemon/runtime.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,uBAAuB,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAC1E,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EACL,eAAe,EACf,iBAAiB,GAElB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,8BAA8B,EAC9B,gDAAgD,GACjD,MAAM,yBAAyB,CAAC;AAYjC,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,OAAyB;IAEzB,MAAM,UAAU,GAAG,iBAAiB,CAClC,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAC7B,CAAC;IACF,MAAM,MAAM,GAAG,uBAAuB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACvD,MAAM,cAAc,GAAG,UAAU,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,IAAI,MAAM,CAAC;IAC/E,MAAM,YAAY,GAAG,MAAM,kBAAkB,CAAC,OAAO,CACnD,OAAO,CAAC,IAAI,EACZ,cAAc,EACd,UAAU,CACX,CAAC;IACF,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,IAAI,IAAI,CAAC;IACT,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,kBAAkB,CAAC;YAC9B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,CAAC,OAAO,EAAE,CAAC;QAC7B,MAAM,KAAK,CAAC;IACd,CAAC;IACD,IAAI,eAAe,CAAC;IACpB,IAAI,CAAC;QACH,eAAe,GAAG,MAAM,8BAA8B,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACvE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,CAAC,OAAO,EAAE,CAAC;QAC7B,MAAM,KAAK,CAAC;IACd,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC;QAChC,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,cAAc,EAAE,OAAO,CAAC,cAAc;QACtC,MAAM;KACP,CAAC,CAAC;IACH,IAAI,WAAqC,CAAC;IAC1C,MAAM,UAAU,GAAG,IAAI,gBAAgB,CAAC;QACtC,GAAG,MAAM;QACT,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,UAAU;QACV,OAAO;QACP,eAAe;QACf,uBAAuB,EACrB,IAAI,gDAAgD,CAAC,OAAO,CAAC,IAAI,CAAC;QACpE,MAAM;QACN,UAAU,EAAE,GAAG,EAAE,CAAC,WAAW,EAAE,EAAE;KAClC,CAAC,CAAC;IACH,IAAI,OAAO,CAAC;IACZ,IAAI,CAAC;QACH,OAAO,GAAG,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;IACrC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;QACtB,MAAM,YAAY,CAAC,OAAO,EAAE,CAAC;QAC7B,MAAM,KAAK,CAAC;IACd,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;QAC5C,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,MAAM,IAAI,GAAG,GAAG,EAAE;YAChB,IAAI,QAAQ;gBAAE,OAAO;YACrB,QAAQ,GAAG,IAAI,CAAC;YAChB,KAAK,CAAC,KAAK,IAAI,EAAE;gBACf,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;gBACtB,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC;gBACzB,MAAM,YAAY,CAAC,OAAO,EAAE,CAAC;gBAC7B,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;gBACrD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACvB,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACxB,CAAC,CAAC;QACF,WAAW,GAAG,IAAI,CAAC;QACnB,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC7B,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IACH,IAAI,CAAC;QACH,MAAM,YAAY,CAAC,SAAS,EAAE,CAAC;IACjC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,WAAW,EAAE,EAAE,CAAC;QAChB,MAAM,OAAO,CAAC;QACd,MAAM,KAAK,CAAC;IACd,CAAC;IACD,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;QAC3B,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,WAAW,EAAE,UAAU;KACxB,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CACT,wCAAwC,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,IAAI,MAAM,CAC3F,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,gBAAgB,UAAU,EAAE,CAAC,CAAC;IAC1C,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;IACtD,CAAC;IAED,MAAM,OAAO,CAAC;AAChB,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;AACjD,CAAC"}
@@ -0,0 +1,39 @@
1
+ import { type McpToolset } from "../mcp/toolset.js";
2
+ export type DaemonInstanceRecord = {
3
+ pid: number;
4
+ hostname: string;
5
+ instanceToken: string;
6
+ startedAt: number;
7
+ updatedAt: number;
8
+ serverUrl: string;
9
+ ready: boolean;
10
+ mcpToolset: McpToolset;
11
+ };
12
+ export type DaemonControlStatus = {
13
+ running: boolean;
14
+ ready: boolean;
15
+ pid?: number;
16
+ serverUrl?: string;
17
+ mcpToolset?: McpToolset;
18
+ };
19
+ export declare class DaemonInstanceLock {
20
+ readonly path: string;
21
+ readonly record: DaemonInstanceRecord;
22
+ private heartbeat?;
23
+ private constructor();
24
+ static acquire(home: string | undefined, serverUrl: string, mcpToolset?: McpToolset): Promise<DaemonInstanceLock>;
25
+ markReady(): Promise<void>;
26
+ release(): Promise<void>;
27
+ private write;
28
+ }
29
+ export declare function readInstanceRecord(home?: string): Promise<DaemonInstanceRecord | undefined>;
30
+ export declare function serverStatus(home?: string): Promise<DaemonControlStatus>;
31
+ export declare function startServer(options: {
32
+ cliPath: string;
33
+ listen?: string;
34
+ tokenFile?: string;
35
+ home?: string;
36
+ timeoutMs?: number;
37
+ mcpToolset?: McpToolset;
38
+ }): Promise<DaemonControlStatus>;
39
+ export declare function stopServer(home?: string, timeoutMs?: number, tokenFile?: string): Promise<DaemonControlStatus>;