@wanshi-kg/wanshi 0.1.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 (443) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +458 -0
  3. package/dist/__tests__/helpers.js +27 -0
  4. package/dist/__tests__/helpers.js.map +1 -0
  5. package/dist/cli/commands/export.command.js +99 -0
  6. package/dist/cli/commands/export.command.js.map +1 -0
  7. package/dist/cli/commands/index.js +22 -0
  8. package/dist/cli/commands/index.js.map +1 -0
  9. package/dist/cli/commands/inspectMerges.command.js +84 -0
  10. package/dist/cli/commands/inspectMerges.command.js.map +1 -0
  11. package/dist/cli/commands/metrics.command.js +196 -0
  12. package/dist/cli/commands/metrics.command.js.map +1 -0
  13. package/dist/cli/commands/process.command.js +82 -0
  14. package/dist/cli/commands/process.command.js.map +1 -0
  15. package/dist/cli/commands/watch.command.js +91 -0
  16. package/dist/cli/commands/watch.command.js.map +1 -0
  17. package/dist/cli/index.js +269 -0
  18. package/dist/cli/index.js.map +1 -0
  19. package/dist/cli/optionsToConfig.js +160 -0
  20. package/dist/cli/optionsToConfig.js.map +1 -0
  21. package/dist/config/index.js +59 -0
  22. package/dist/config/index.js.map +1 -0
  23. package/dist/config/legacyHints.js +113 -0
  24. package/dist/config/legacyHints.js.map +1 -0
  25. package/dist/config/schema.js +803 -0
  26. package/dist/config/schema.js.map +1 -0
  27. package/dist/config/ui.js +221 -0
  28. package/dist/config/ui.js.map +1 -0
  29. package/dist/core/DirectoryProcessor.js +725 -0
  30. package/dist/core/DirectoryProcessor.js.map +1 -0
  31. package/dist/core/adapters/IStructuredAdapter.js +3 -0
  32. package/dist/core/adapters/IStructuredAdapter.js.map +1 -0
  33. package/dist/core/adapters/SqliteAdapter.js +267 -0
  34. package/dist/core/adapters/SqliteAdapter.js.map +1 -0
  35. package/dist/core/adapters/StructuredAdapterRegistry.js +31 -0
  36. package/dist/core/adapters/StructuredAdapterRegistry.js.map +1 -0
  37. package/dist/core/adapters/index.js +20 -0
  38. package/dist/core/adapters/index.js.map +1 -0
  39. package/dist/core/checkpoint/CheckpointService.js +188 -0
  40. package/dist/core/checkpoint/CheckpointService.js.map +1 -0
  41. package/dist/core/checkpoint/index.js +18 -0
  42. package/dist/core/checkpoint/index.js.map +1 -0
  43. package/dist/core/corpus/CorpusAnalyzer.js +266 -0
  44. package/dist/core/corpus/CorpusAnalyzer.js.map +1 -0
  45. package/dist/core/corpus/CorpusProfileStore.js +92 -0
  46. package/dist/core/corpus/CorpusProfileStore.js.map +1 -0
  47. package/dist/core/corpus/index.js +21 -0
  48. package/dist/core/corpus/index.js.map +1 -0
  49. package/dist/core/corpus/normalizeGlossary.js +60 -0
  50. package/dist/core/corpus/normalizeGlossary.js.map +1 -0
  51. package/dist/core/corpus/relPath.js +52 -0
  52. package/dist/core/corpus/relPath.js.map +1 -0
  53. package/dist/core/corpus/termFrequency.js +86 -0
  54. package/dist/core/corpus/termFrequency.js.map +1 -0
  55. package/dist/core/cost/CostMeter.js +235 -0
  56. package/dist/core/cost/CostMeter.js.map +1 -0
  57. package/dist/core/cost/index.js +19 -0
  58. package/dist/core/cost/index.js.map +1 -0
  59. package/dist/core/cost/prices.js +38 -0
  60. package/dist/core/cost/prices.js.map +1 -0
  61. package/dist/core/cv/ObjectDetectionService.js +119 -0
  62. package/dist/core/cv/ObjectDetectionService.js.map +1 -0
  63. package/dist/core/di/ContainerFactory.js +670 -0
  64. package/dist/core/di/ContainerFactory.js.map +1 -0
  65. package/dist/core/di/DIContainer.js +103 -0
  66. package/dist/core/di/DIContainer.js.map +1 -0
  67. package/dist/core/di/index.js +19 -0
  68. package/dist/core/di/index.js.map +1 -0
  69. package/dist/core/errors/CustomErrors.js +342 -0
  70. package/dist/core/errors/CustomErrors.js.map +1 -0
  71. package/dist/core/errors/index.js +18 -0
  72. package/dist/core/errors/index.js.map +1 -0
  73. package/dist/core/export/KnowledgeGraphExportService.js +56 -0
  74. package/dist/core/export/KnowledgeGraphExportService.js.map +1 -0
  75. package/dist/core/export/index.js +19 -0
  76. package/dist/core/export/index.js.map +1 -0
  77. package/dist/core/export/strategies/GraphitiExportStrategy.js +115 -0
  78. package/dist/core/export/strategies/GraphitiExportStrategy.js.map +1 -0
  79. package/dist/core/export/strategies/GraphvizDotExportStrategy.js +331 -0
  80. package/dist/core/export/strategies/GraphvizDotExportStrategy.js.map +1 -0
  81. package/dist/core/export/strategies/IExportStrategy.js +3 -0
  82. package/dist/core/export/strategies/IExportStrategy.js.map +1 -0
  83. package/dist/core/export/strategies/JsonExportStrategy.js +19 -0
  84. package/dist/core/export/strategies/JsonExportStrategy.js.map +1 -0
  85. package/dist/core/export/strategies/JsonlExportStrategy.js +69 -0
  86. package/dist/core/export/strategies/JsonlExportStrategy.js.map +1 -0
  87. package/dist/core/export/strategies/KblamExportStrategy.js +36 -0
  88. package/dist/core/export/strategies/KblamExportStrategy.js.map +1 -0
  89. package/dist/core/export/strategies/LoraExportStrategy.js +46 -0
  90. package/dist/core/export/strategies/LoraExportStrategy.js.map +1 -0
  91. package/dist/core/export/strategies/McpExportStrategy.js +67 -0
  92. package/dist/core/export/strategies/McpExportStrategy.js.map +1 -0
  93. package/dist/core/export/strategies/index.js +25 -0
  94. package/dist/core/export/strategies/index.js.map +1 -0
  95. package/dist/core/export/strategies/kbTriples.js +60 -0
  96. package/dist/core/export/strategies/kbTriples.js.map +1 -0
  97. package/dist/core/index.js +22 -0
  98. package/dist/core/index.js.map +1 -0
  99. package/dist/core/knowledge/KnowledgeGraphBuilder.js +627 -0
  100. package/dist/core/knowledge/KnowledgeGraphBuilder.js.map +1 -0
  101. package/dist/core/knowledge/MergeRecord.js +3 -0
  102. package/dist/core/knowledge/MergeRecord.js.map +1 -0
  103. package/dist/core/knowledge/canon/Canonicalizer.js +414 -0
  104. package/dist/core/knowledge/canon/Canonicalizer.js.map +1 -0
  105. package/dist/core/knowledge/canon/index.js +18 -0
  106. package/dist/core/knowledge/canon/index.js.map +1 -0
  107. package/dist/core/knowledge/contradiction/HeuristicContradictionChecker.js +92 -0
  108. package/dist/core/knowledge/contradiction/HeuristicContradictionChecker.js.map +1 -0
  109. package/dist/core/knowledge/contradiction/LlmContradictionChecker.js +52 -0
  110. package/dist/core/knowledge/contradiction/LlmContradictionChecker.js.map +1 -0
  111. package/dist/core/knowledge/contradiction/index.js +19 -0
  112. package/dist/core/knowledge/contradiction/index.js.map +1 -0
  113. package/dist/core/knowledge/grounding/KeywordGroundingChecker.js +33 -0
  114. package/dist/core/knowledge/grounding/KeywordGroundingChecker.js.map +1 -0
  115. package/dist/core/knowledge/grounding/MiniCheckGroundingChecker.js +82 -0
  116. package/dist/core/knowledge/grounding/MiniCheckGroundingChecker.js.map +1 -0
  117. package/dist/core/knowledge/grounding/index.js +20 -0
  118. package/dist/core/knowledge/grounding/index.js.map +1 -0
  119. package/dist/core/knowledge/grounding/verbalize.js +38 -0
  120. package/dist/core/knowledge/grounding/verbalize.js.map +1 -0
  121. package/dist/core/knowledge/images/imageMetaGraph.js +136 -0
  122. package/dist/core/knowledge/images/imageMetaGraph.js.map +1 -0
  123. package/dist/core/knowledge/index.js +20 -0
  124. package/dist/core/knowledge/index.js.map +1 -0
  125. package/dist/core/knowledge/merging/KnowledgeMerger.js +624 -0
  126. package/dist/core/knowledge/merging/KnowledgeMerger.js.map +1 -0
  127. package/dist/core/knowledge/references/ReferenceResolver.js +184 -0
  128. package/dist/core/knowledge/references/ReferenceResolver.js.map +1 -0
  129. package/dist/core/knowledge/references/citations/CitationEvidenceProcessor.js +401 -0
  130. package/dist/core/knowledge/references/citations/CitationEvidenceProcessor.js.map +1 -0
  131. package/dist/core/knowledge/references/citations/CitationResolver.js +95 -0
  132. package/dist/core/knowledge/references/citations/CitationResolver.js.map +1 -0
  133. package/dist/core/knowledge/references/citations/GrobidClient.js +143 -0
  134. package/dist/core/knowledge/references/citations/GrobidClient.js.map +1 -0
  135. package/dist/core/knowledge/references/citations/TitleIdResolver.js +101 -0
  136. package/dist/core/knowledge/references/citations/TitleIdResolver.js.map +1 -0
  137. package/dist/core/knowledge/references/web/FetchCacheService.js +114 -0
  138. package/dist/core/knowledge/references/web/FetchCacheService.js.map +1 -0
  139. package/dist/core/knowledge/references/web/GatedFetcher.js +228 -0
  140. package/dist/core/knowledge/references/web/GatedFetcher.js.map +1 -0
  141. package/dist/core/knowledge/references/web/WebReferenceProcessor.js +164 -0
  142. package/dist/core/knowledge/references/web/WebReferenceProcessor.js.map +1 -0
  143. package/dist/core/knowledge/search/KnowledgeGraphSearch.js +261 -0
  144. package/dist/core/knowledge/search/KnowledgeGraphSearch.js.map +1 -0
  145. package/dist/core/knowledge/vocabulary.js +162 -0
  146. package/dist/core/knowledge/vocabulary.js.map +1 -0
  147. package/dist/core/llm/EmbeddingService.js +113 -0
  148. package/dist/core/llm/EmbeddingService.js.map +1 -0
  149. package/dist/core/llm/OllamaService.js +146 -0
  150. package/dist/core/llm/OllamaService.js.map +1 -0
  151. package/dist/core/llm/OpenAICompatibleService.js +190 -0
  152. package/dist/core/llm/OpenAICompatibleService.js.map +1 -0
  153. package/dist/core/llm/OpenAIEmbeddingService.js +129 -0
  154. package/dist/core/llm/OpenAIEmbeddingService.js.map +1 -0
  155. package/dist/core/llm/embeddingUtils.js +25 -0
  156. package/dist/core/llm/embeddingUtils.js.map +1 -0
  157. package/dist/core/llm/index.js +23 -0
  158. package/dist/core/llm/index.js.map +1 -0
  159. package/dist/core/llm/prompts/PromptManager.js +388 -0
  160. package/dist/core/llm/prompts/PromptManager.js.map +1 -0
  161. package/dist/core/llm/prompts/PromptTemplateEngine.js +257 -0
  162. package/dist/core/llm/prompts/PromptTemplateEngine.js.map +1 -0
  163. package/dist/core/llm/prompts/templates/partials/examples/EXAMPLE_STYLE_GUIDE.md +84 -0
  164. package/dist/core/llm/prompts/templates/partials/examples/article.md +187 -0
  165. package/dist/core/llm/prompts/templates/partials/examples/code.md +229 -0
  166. package/dist/core/llm/prompts/templates/partials/examples/communication.md +205 -0
  167. package/dist/core/llm/prompts/templates/partials/examples/documentation.md +262 -0
  168. package/dist/core/llm/prompts/templates/partials/examples/financial.md +157 -0
  169. package/dist/core/llm/prompts/templates/partials/examples/legal.md +153 -0
  170. package/dist/core/llm/prompts/templates/partials/examples/logs.md +127 -0
  171. package/dist/core/llm/prompts/templates/partials/examples/medical.md +218 -0
  172. package/dist/core/llm/prompts/templates/partials/examples/notes.md +201 -0
  173. package/dist/core/llm/prompts/templates/partials/examples/research.md +208 -0
  174. package/dist/core/llm/prompts/templates/partials/examples/tabular.md +178 -0
  175. package/dist/core/llm/prompts/templates/partials/examples/transcript.md +204 -0
  176. package/dist/core/llm/prompts/templates/partials/retrieved-context.hbs +18 -0
  177. package/dist/core/llm/prompts/templates/v1/system.hbs +371 -0
  178. package/dist/core/llm/prompts/templates/v1/user.hbs +20 -0
  179. package/dist/core/llm/prompts/templates/v2/system.hbs +573 -0
  180. package/dist/core/llm/prompts/templates/v2/user.hbs +20 -0
  181. package/dist/core/llm/prompts/templates/v3/system.hbs +861 -0
  182. package/dist/core/llm/prompts/templates/v3/user.hbs +16 -0
  183. package/dist/core/llm/prompts/templates/v4/system.hbs +800 -0
  184. package/dist/core/llm/prompts/templates/v4/user.hbs +40 -0
  185. package/dist/core/llm/prompts/templates/v4.5/system.hbs +71 -0
  186. package/dist/core/llm/prompts/templates/v4.5/user.hbs +46 -0
  187. package/dist/core/llm/prompts/templates/v5/glossary/system.hbs +40 -0
  188. package/dist/core/llm/prompts/templates/v5/glossary/user.hbs +11 -0
  189. package/dist/core/llm/prompts/templates/v5/system.hbs +163 -0
  190. package/dist/core/llm/prompts/templates/v5/user.hbs +55 -0
  191. package/dist/core/pipeline/GroundingTransform.js +52 -0
  192. package/dist/core/pipeline/GroundingTransform.js.map +1 -0
  193. package/dist/core/pipeline/PipelineRunner.js +51 -0
  194. package/dist/core/pipeline/PipelineRunner.js.map +1 -0
  195. package/dist/core/pipeline/RelationFilterTransform.js +72 -0
  196. package/dist/core/pipeline/RelationFilterTransform.js.map +1 -0
  197. package/dist/core/pipeline/index.js +20 -0
  198. package/dist/core/pipeline/index.js.map +1 -0
  199. package/dist/core/processor/FileProcessor.js +184 -0
  200. package/dist/core/processor/FileProcessor.js.map +1 -0
  201. package/dist/core/processor/ProcessedRegistry.js +38 -0
  202. package/dist/core/processor/ProcessedRegistry.js.map +1 -0
  203. package/dist/core/processor/ast/AstSeedService.js +0 -0
  204. package/dist/core/processor/ast/AstSeedService.js.map +1 -0
  205. package/dist/core/processor/ast/AstSymbolStore.js +110 -0
  206. package/dist/core/processor/ast/AstSymbolStore.js.map +1 -0
  207. package/dist/core/processor/ast/index.js +19 -0
  208. package/dist/core/processor/ast/index.js.map +1 -0
  209. package/dist/core/processor/chunking/TextChunker.js +98 -0
  210. package/dist/core/processor/chunking/TextChunker.js.map +1 -0
  211. package/dist/core/processor/chunking/index.js +18 -0
  212. package/dist/core/processor/chunking/index.js.map +1 -0
  213. package/dist/core/processor/classifier/CONTENT_CLASSES.js +294 -0
  214. package/dist/core/processor/classifier/CONTENT_CLASSES.js.map +1 -0
  215. package/dist/core/processor/classifier/CascadeContentClassifier.js +107 -0
  216. package/dist/core/processor/classifier/CascadeContentClassifier.js.map +1 -0
  217. package/dist/core/processor/classifier/HeuristicContentClassifier.js +113 -0
  218. package/dist/core/processor/classifier/HeuristicContentClassifier.js.map +1 -0
  219. package/dist/core/processor/classifier/IContentTypeClassifier.js +3 -0
  220. package/dist/core/processor/classifier/IContentTypeClassifier.js.map +1 -0
  221. package/dist/core/processor/classifier/LlmContentClassifier.js +107 -0
  222. package/dist/core/processor/classifier/LlmContentClassifier.js.map +1 -0
  223. package/dist/core/processor/classifier/NER_DOMAIN_EXAMPLES.js +498 -0
  224. package/dist/core/processor/classifier/NER_DOMAIN_EXAMPLES.js.map +1 -0
  225. package/dist/core/processor/classifier/index.js +21 -0
  226. package/dist/core/processor/classifier/index.js.map +1 -0
  227. package/dist/core/processor/classifier/mergeClassifications.js +32 -0
  228. package/dist/core/processor/classifier/mergeClassifications.js.map +1 -0
  229. package/dist/core/processor/index.js +20 -0
  230. package/dist/core/processor/index.js.map +1 -0
  231. package/dist/core/processor/readers/AudioReader.js +462 -0
  232. package/dist/core/processor/readers/AudioReader.js.map +1 -0
  233. package/dist/core/processor/readers/BinaryReader.js +90 -0
  234. package/dist/core/processor/readers/BinaryReader.js.map +1 -0
  235. package/dist/core/processor/readers/ChandraPdfReader.js +187 -0
  236. package/dist/core/processor/readers/ChandraPdfReader.js.map +1 -0
  237. package/dist/core/processor/readers/ChatExportReader.js +365 -0
  238. package/dist/core/processor/readers/ChatExportReader.js.map +1 -0
  239. package/dist/core/processor/readers/DoclingReader.js +445 -0
  240. package/dist/core/processor/readers/DoclingReader.js.map +1 -0
  241. package/dist/core/processor/readers/EmailReader.js +259 -0
  242. package/dist/core/processor/readers/EmailReader.js.map +1 -0
  243. package/dist/core/processor/readers/EpubReader.js +175 -0
  244. package/dist/core/processor/readers/EpubReader.js.map +1 -0
  245. package/dist/core/processor/readers/FileReader.js +90 -0
  246. package/dist/core/processor/readers/FileReader.js.map +1 -0
  247. package/dist/core/processor/readers/FileReaderFactory.js +49 -0
  248. package/dist/core/processor/readers/FileReaderFactory.js.map +1 -0
  249. package/dist/core/processor/readers/HtmlReader.js +371 -0
  250. package/dist/core/processor/readers/HtmlReader.js.map +1 -0
  251. package/dist/core/processor/readers/ImageReader.js +162 -0
  252. package/dist/core/processor/readers/ImageReader.js.map +1 -0
  253. package/dist/core/processor/readers/JsonFileReader.js +232 -0
  254. package/dist/core/processor/readers/JsonFileReader.js.map +1 -0
  255. package/dist/core/processor/readers/JupyterReader.js +178 -0
  256. package/dist/core/processor/readers/JupyterReader.js.map +1 -0
  257. package/dist/core/processor/readers/LatexReader.js +176 -0
  258. package/dist/core/processor/readers/LatexReader.js.map +1 -0
  259. package/dist/core/processor/readers/MarkdownReader.js +289 -0
  260. package/dist/core/processor/readers/MarkdownReader.js.map +1 -0
  261. package/dist/core/processor/readers/MarkerPdfReader.js +193 -0
  262. package/dist/core/processor/readers/MarkerPdfReader.js.map +1 -0
  263. package/dist/core/processor/readers/MistralOcrReader.js +198 -0
  264. package/dist/core/processor/readers/MistralOcrReader.js.map +1 -0
  265. package/dist/core/processor/readers/OfficeReader.js +174 -0
  266. package/dist/core/processor/readers/OfficeReader.js.map +1 -0
  267. package/dist/core/processor/readers/PdfReader.js +116 -0
  268. package/dist/core/processor/readers/PdfReader.js.map +1 -0
  269. package/dist/core/processor/readers/RtfReader.js +107 -0
  270. package/dist/core/processor/readers/RtfReader.js.map +1 -0
  271. package/dist/core/processor/readers/SubtitleReader.js +145 -0
  272. package/dist/core/processor/readers/SubtitleReader.js.map +1 -0
  273. package/dist/core/processor/readers/TesseractPdfReader.js +183 -0
  274. package/dist/core/processor/readers/TesseractPdfReader.js.map +1 -0
  275. package/dist/core/processor/readers/TextReader.js +129 -0
  276. package/dist/core/processor/readers/TextReader.js.map +1 -0
  277. package/dist/core/processor/readers/TranscriptReader.js +234 -0
  278. package/dist/core/processor/readers/TranscriptReader.js.map +1 -0
  279. package/dist/core/processor/readers/image/imageMetadata.js +155 -0
  280. package/dist/core/processor/readers/image/imageMetadata.js.map +1 -0
  281. package/dist/core/processor/readers/index.js +41 -0
  282. package/dist/core/processor/readers/index.js.map +1 -0
  283. package/dist/core/processor/readers/referenceExtraction.js +198 -0
  284. package/dist/core/processor/readers/referenceExtraction.js.map +1 -0
  285. package/dist/core/processor/readers/stripReferences.js +59 -0
  286. package/dist/core/processor/readers/stripReferences.js.map +1 -0
  287. package/dist/core/processor/readers/transcript/turnPacking.js +81 -0
  288. package/dist/core/processor/readers/transcript/turnPacking.js.map +1 -0
  289. package/dist/core/progress/NdjsonProgressEmitter.js +30 -0
  290. package/dist/core/progress/NdjsonProgressEmitter.js.map +1 -0
  291. package/dist/core/progress/NoopProgressEmitter.js +15 -0
  292. package/dist/core/progress/NoopProgressEmitter.js.map +1 -0
  293. package/dist/core/progress/index.js +19 -0
  294. package/dist/core/progress/index.js.map +1 -0
  295. package/dist/core/trace/TraceWriter.js +100 -0
  296. package/dist/core/trace/TraceWriter.js.map +1 -0
  297. package/dist/core/trace/events.js +13 -0
  298. package/dist/core/trace/events.js.map +1 -0
  299. package/dist/core/trace/index.js +20 -0
  300. package/dist/core/trace/index.js.map +1 -0
  301. package/dist/core/trace/lineage.js +97 -0
  302. package/dist/core/trace/lineage.js.map +1 -0
  303. package/dist/evaluation/BenchmarkRunner.js +171 -0
  304. package/dist/evaluation/BenchmarkRunner.js.map +1 -0
  305. package/dist/evaluation/classifier/ClassifierAccuracy.js +185 -0
  306. package/dist/evaluation/classifier/ClassifierAccuracy.js.map +1 -0
  307. package/dist/evaluation/classifier/labeledSamples.js +379 -0
  308. package/dist/evaluation/classifier/labeledSamples.js.map +1 -0
  309. package/dist/evaluation/compare/goldCompare.js +126 -0
  310. package/dist/evaluation/compare/goldCompare.js.map +1 -0
  311. package/dist/evaluation/crossre/compareScoring.js +30 -0
  312. package/dist/evaluation/crossre/compareScoring.js.map +1 -0
  313. package/dist/evaluation/datasets/CrossREDataset.js +170 -0
  314. package/dist/evaluation/datasets/CrossREDataset.js.map +1 -0
  315. package/dist/evaluation/datasets/IDataset.js +3 -0
  316. package/dist/evaluation/datasets/IDataset.js.map +1 -0
  317. package/dist/evaluation/datasets/RebelDataset.js +117 -0
  318. package/dist/evaluation/datasets/RebelDataset.js.map +1 -0
  319. package/dist/evaluation/datasets/RedocredDataset.js +218 -0
  320. package/dist/evaluation/datasets/RedocredDataset.js.map +1 -0
  321. package/dist/evaluation/datasets/SemEval2010Dataset.js +150 -0
  322. package/dist/evaluation/datasets/SemEval2010Dataset.js.map +1 -0
  323. package/dist/evaluation/index.js +33 -0
  324. package/dist/evaluation/index.js.map +1 -0
  325. package/dist/evaluation/matching/ExactMatcher.js +75 -0
  326. package/dist/evaluation/matching/ExactMatcher.js.map +1 -0
  327. package/dist/evaluation/matching/SemanticMatcher.js +143 -0
  328. package/dist/evaluation/matching/SemanticMatcher.js.map +1 -0
  329. package/dist/evaluation/metrics/TripleMetrics.js +64 -0
  330. package/dist/evaluation/metrics/TripleMetrics.js.map +1 -0
  331. package/dist/evaluation/mine/MineCheckpoint.js +114 -0
  332. package/dist/evaluation/mine/MineCheckpoint.js.map +1 -0
  333. package/dist/evaluation/mine/MineDataset.js +208 -0
  334. package/dist/evaluation/mine/MineDataset.js.map +1 -0
  335. package/dist/evaluation/mine/MineReporter.js +98 -0
  336. package/dist/evaluation/mine/MineReporter.js.map +1 -0
  337. package/dist/evaluation/mine/MineRunner.js +148 -0
  338. package/dist/evaluation/mine/MineRunner.js.map +1 -0
  339. package/dist/evaluation/mine/MineScorer.js +127 -0
  340. package/dist/evaluation/mine/MineScorer.js.map +1 -0
  341. package/dist/evaluation/mine/types.js +12 -0
  342. package/dist/evaluation/mine/types.js.map +1 -0
  343. package/dist/evaluation/reporters/ConsoleReporter.js +55 -0
  344. package/dist/evaluation/reporters/ConsoleReporter.js.map +1 -0
  345. package/dist/evaluation/reporters/JsonReporter.js +50 -0
  346. package/dist/evaluation/reporters/JsonReporter.js.map +1 -0
  347. package/dist/index.js +28 -0
  348. package/dist/index.js.map +1 -0
  349. package/dist/quality/CompositeScore.js +61 -0
  350. package/dist/quality/CompositeScore.js.map +1 -0
  351. package/dist/quality/ConsistencyMetrics.js +70 -0
  352. package/dist/quality/ConsistencyMetrics.js.map +1 -0
  353. package/dist/quality/FactualMetrics.js +76 -0
  354. package/dist/quality/FactualMetrics.js.map +1 -0
  355. package/dist/quality/GraphHealthMetrics.js +68 -0
  356. package/dist/quality/GraphHealthMetrics.js.map +1 -0
  357. package/dist/quality/SemanticMetrics.js +102 -0
  358. package/dist/quality/SemanticMetrics.js.map +1 -0
  359. package/dist/quality/StructuralMetrics.js +60 -0
  360. package/dist/quality/StructuralMetrics.js.map +1 -0
  361. package/dist/quality/index.js +23 -0
  362. package/dist/quality/index.js.map +1 -0
  363. package/dist/shared/index.js +20 -0
  364. package/dist/shared/index.js.map +1 -0
  365. package/dist/shared/logger/Logger.js +3 -0
  366. package/dist/shared/logger/Logger.js.map +1 -0
  367. package/dist/shared/logger/LoggerFactory.js +75 -0
  368. package/dist/shared/logger/LoggerFactory.js.map +1 -0
  369. package/dist/shared/logger/index.js +19 -0
  370. package/dist/shared/logger/index.js.map +1 -0
  371. package/dist/shared/shutdown.js +30 -0
  372. package/dist/shared/shutdown.js.map +1 -0
  373. package/dist/shared/utils/agglomerativeCluster.js +269 -0
  374. package/dist/shared/utils/agglomerativeCluster.js.map +1 -0
  375. package/dist/shared/utils/astSymbols.js +69 -0
  376. package/dist/shared/utils/astSymbols.js.map +1 -0
  377. package/dist/shared/utils/cosineSimilarity.js +18 -0
  378. package/dist/shared/utils/cosineSimilarity.js.map +1 -0
  379. package/dist/shared/utils/directoryTree.js +184 -0
  380. package/dist/shared/utils/directoryTree.js.map +1 -0
  381. package/dist/shared/utils/documentOutline.js +74 -0
  382. package/dist/shared/utils/documentOutline.js.map +1 -0
  383. package/dist/shared/utils/index.js +24 -0
  384. package/dist/shared/utils/index.js.map +1 -0
  385. package/dist/shared/utils/jaroWinklerSimilarity.js +60 -0
  386. package/dist/shared/utils/jaroWinklerSimilarity.js.map +1 -0
  387. package/dist/shared/utils/parseJsonLenient.js +27 -0
  388. package/dist/shared/utils/parseJsonLenient.js.map +1 -0
  389. package/dist/shared/utils/readConfig.js +42 -0
  390. package/dist/shared/utils/readConfig.js.map +1 -0
  391. package/dist/shared/utils/readRtf.js +216 -0
  392. package/dist/shared/utils/readRtf.js.map +1 -0
  393. package/dist/shared/utils/softmax.js +26 -0
  394. package/dist/shared/utils/softmax.js.map +1 -0
  395. package/dist/types/ContentClass.js +3 -0
  396. package/dist/types/ContentClass.js.map +1 -0
  397. package/dist/types/CorpusProfile.js +3 -0
  398. package/dist/types/CorpusProfile.js.map +1 -0
  399. package/dist/types/IContradictionChecker.js +3 -0
  400. package/dist/types/IContradictionChecker.js.map +1 -0
  401. package/dist/types/ICorpusAnalyzer.js +3 -0
  402. package/dist/types/ICorpusAnalyzer.js.map +1 -0
  403. package/dist/types/IDirectoryProcessor.js +3 -0
  404. package/dist/types/IDirectoryProcessor.js.map +1 -0
  405. package/dist/types/IEmbeddingProvider.js +3 -0
  406. package/dist/types/IEmbeddingProvider.js.map +1 -0
  407. package/dist/types/IEmbeddingService.js +6 -0
  408. package/dist/types/IEmbeddingService.js.map +1 -0
  409. package/dist/types/IFileProcessor.js +3 -0
  410. package/dist/types/IFileProcessor.js.map +1 -0
  411. package/dist/types/IGroundingChecker.js +3 -0
  412. package/dist/types/IGroundingChecker.js.map +1 -0
  413. package/dist/types/IKnowledgeGraphBuilder.js +3 -0
  414. package/dist/types/IKnowledgeGraphBuilder.js.map +1 -0
  415. package/dist/types/IKnowledgeGraphExporter.js +3 -0
  416. package/dist/types/IKnowledgeGraphExporter.js.map +1 -0
  417. package/dist/types/IKnowledgeGraphMerger.js +3 -0
  418. package/dist/types/IKnowledgeGraphMerger.js.map +1 -0
  419. package/dist/types/IKnowledgeGraphSearch.js +3 -0
  420. package/dist/types/IKnowledgeGraphSearch.js.map +1 -0
  421. package/dist/types/ILLMProvider.js +3 -0
  422. package/dist/types/ILLMProvider.js.map +1 -0
  423. package/dist/types/ILLMService.js +3 -0
  424. package/dist/types/ILLMService.js.map +1 -0
  425. package/dist/types/IObjectDetector.js +3 -0
  426. package/dist/types/IObjectDetector.js.map +1 -0
  427. package/dist/types/IProcessingService.js +3 -0
  428. package/dist/types/IProcessingService.js.map +1 -0
  429. package/dist/types/IProgressEmitter.js +3 -0
  430. package/dist/types/IProgressEmitter.js.map +1 -0
  431. package/dist/types/IPromptManager.js +3 -0
  432. package/dist/types/IPromptManager.js.map +1 -0
  433. package/dist/types/KnowledgeGraph.js +3 -0
  434. package/dist/types/KnowledgeGraph.js.map +1 -0
  435. package/dist/types/MCPKnowledgeGraph.js +3 -0
  436. package/dist/types/MCPKnowledgeGraph.js.map +1 -0
  437. package/dist/types/Observation.js +21 -0
  438. package/dist/types/Observation.js.map +1 -0
  439. package/dist/types/ProcessingOptions.js +3 -0
  440. package/dist/types/ProcessingOptions.js.map +1 -0
  441. package/dist/types/index.js +40 -0
  442. package/dist/types/index.js.map +1 -0
  443. package/package.json +122 -0
@@ -0,0 +1,146 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ Object.defineProperty(exports, "__esModule", { value: true });
12
+ exports.OllamaService = void 0;
13
+ const ollama_1 = require("ollama");
14
+ const zod_to_json_schema_1 = require("zod-to-json-schema");
15
+ const utils_1 = require("../../shared/utils");
16
+ const cost_1 = require("../cost");
17
+ /**
18
+ * Service for interacting with Ollama LLMs
19
+ */
20
+ class OllamaService {
21
+ constructor(options, logger) {
22
+ this.logger = logger;
23
+ this.options = Object.assign({ temperature: 0.1, contextLength: 8192, repeatPenalty: 1.1 }, options);
24
+ this.ollama = new ollama_1.Ollama({ host: this.options.host });
25
+ }
26
+ stripCodeFence(content) {
27
+ if (content.startsWith("```")) {
28
+ return content.slice(content.indexOf("\n") + 1, content.lastIndexOf("\n"));
29
+ }
30
+ return content;
31
+ }
32
+ /**
33
+ * Generate a completion with structured output
34
+ */
35
+ generateStructured(messages_1, schema_1) {
36
+ return __awaiter(this, arguments, void 0, function* (messages, schema, retries = 3) {
37
+ const jsonSchema = (0, zod_to_json_schema_1.zodToJsonSchema)(schema);
38
+ // Convert our messages to Ollama format
39
+ const ollamaMessages = yield this.toOllamaMessages(messages);
40
+ // Ollama options
41
+ const chatRequest = {
42
+ model: this.options.model,
43
+ messages: ollamaMessages,
44
+ format: jsonSchema,
45
+ think: false,
46
+ options: Object.assign(Object.assign(Object.assign({ num_ctx: Number(this.options.contextLength || 8192) }, (this.options.maxTokens
47
+ ? { num_predict: Number(this.options.maxTokens) }
48
+ : {})), { temperature: Number(this.options.temperature), repeat_penalty: Number(this.options.repeatPenalty) }), (this.options.seed !== undefined
49
+ ? { seed: Number(this.options.seed) }
50
+ : {})),
51
+ };
52
+ for (let attempt = 0; attempt < retries; attempt++) {
53
+ try {
54
+ this.logger.debug(`Generating structured output (attempt ${attempt + 1}/${retries})`);
55
+ const response = yield this.ollama.chat(chatRequest);
56
+ this.logResponseStats(response);
57
+ // A "length" done_reason means output hit num_predict / the model's
58
+ // limit and the JSON is truncated — give an actionable hint.
59
+ if (response.done_reason === "length") {
60
+ this.logger.warn("LLM output was truncated at the output-token limit — the JSON is incomplete. " +
61
+ "Increase --max-tokens or reduce --chunk-size.");
62
+ }
63
+ // Parse the response
64
+ const responseContent = response.message.content.trim();
65
+ this.logger.debug(`Raw LLM response: ${responseContent.substring(0, 200)}...`);
66
+ const parsed = (0, utils_1.parseJsonLenient)(this.stripCodeFence(responseContent), () => this.logger.warn("Response JSON was malformed; recovered with jsonrepair"));
67
+ return schema.parse(parsed);
68
+ }
69
+ catch (error) {
70
+ this.logger.error(`LLM generation attempt ${attempt + 1} failed: ${error}`);
71
+ if (attempt === retries - 1) {
72
+ throw new Error(`Failed to generate valid output after ${retries} attempts: ${error}`);
73
+ }
74
+ // Wait before retry
75
+ yield this.delay(1000 * (attempt + 1));
76
+ }
77
+ }
78
+ throw new Error("Should not reach here");
79
+ });
80
+ }
81
+ toOllamaMessages(messages) {
82
+ return __awaiter(this, void 0, void 0, function* () {
83
+ const modelCaps = yield this.getModelCapabilities(this.options.model);
84
+ const supportsVision = modelCaps.includes("vision");
85
+ if (this.options.images && !supportsVision) {
86
+ this.logger.warn("Model does not supports vision. Skipping images.");
87
+ }
88
+ const images = this.options.images && supportsVision;
89
+ const ollamaMessages = messages.map((msg) => (Object.assign({ role: msg.role, content: msg.content }, (images && msg.images && { images: msg.images }))));
90
+ return ollamaMessages;
91
+ });
92
+ }
93
+ getModelCapabilities(modelName) {
94
+ return __awaiter(this, void 0, void 0, function* () {
95
+ const modelInfo = yield this.ollama.show({ model: modelName });
96
+ return modelInfo.capabilities;
97
+ });
98
+ }
99
+ /**
100
+ * Check if a model is available
101
+ */
102
+ isModelAvailable(modelName) {
103
+ return __awaiter(this, void 0, void 0, function* () {
104
+ try {
105
+ const models = yield this.ollama.list();
106
+ return models.models.some((m) => m.name === modelName);
107
+ }
108
+ catch (error) {
109
+ this.logger.error(`Failed to check model availability: ${error}`);
110
+ return false;
111
+ }
112
+ });
113
+ }
114
+ /**
115
+ * Log response statistics
116
+ */
117
+ logResponseStats(response) {
118
+ var _a, _b;
119
+ this.lastUsage = {
120
+ promptTokens: response.prompt_eval_count,
121
+ completionTokens: response.eval_count,
122
+ totalTokens: ((_a = response.prompt_eval_count) !== null && _a !== void 0 ? _a : 0) + ((_b = response.eval_count) !== null && _b !== void 0 ? _b : 0) || undefined,
123
+ };
124
+ if (cost_1.meter.enabled)
125
+ cost_1.meter.record(this.options.model, this.lastUsage);
126
+ const stats = {
127
+ eval_count: response.eval_count,
128
+ prompt_eval_count: response.prompt_eval_count,
129
+ total_duration_ms: response.total_duration / 1000000,
130
+ eval_speed_tps: (60000000 * (response.prompt_eval_count + response.eval_count)) /
131
+ response.total_duration,
132
+ };
133
+ this.logger.info(`LLM stats: ${JSON.stringify(stats)}`);
134
+ }
135
+ getLastUsage() {
136
+ return this.lastUsage;
137
+ }
138
+ /**
139
+ * Simple delay helper
140
+ */
141
+ delay(ms) {
142
+ return new Promise((resolve) => setTimeout(resolve, ms));
143
+ }
144
+ }
145
+ exports.OllamaService = OllamaService;
146
+ //# sourceMappingURL=OllamaService.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"OllamaService.js","sourceRoot":"","sources":["../../../src/core/llm/OllamaService.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,mCAAoE;AAEpE,2DAAqD;AAErD,8CAAsD;AAEtD,kCAAgC;AAKhC;;GAEG;AACH,MAAa,aAAa;IAMxB,YAAY,OAAmB,EAAE,MAAc;QAC7C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,mBACV,WAAW,EAAE,GAAG,EAChB,aAAa,EAAE,IAAI,EACnB,aAAa,EAAE,GAAG,IACf,OAAO,CACX,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,IAAI,eAAM,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC;IAEO,cAAc,CAAC,OAAe;QACpC,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,OAAO,CAAC,KAAK,CAClB,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EACzB,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAC1B,CAAC;QACJ,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACG,kBAAkB;6DACtB,QAAsB,EACtB,MAAoB,EACpB,UAAkB,CAAC;YAEnB,MAAM,UAAU,GAAG,IAAA,oCAAe,EAAC,MAAM,CAAC,CAAC;YAC3C,wCAAwC;YACxC,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAE7D,iBAAiB;YACjB,MAAM,WAAW,GAAG;gBAClB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;gBACzB,QAAQ,EAAE,cAAc;gBACxB,MAAM,EAAE,UAAU;gBAClB,KAAK,EAAE,KAAK;gBACZ,OAAO,8CACL,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,IAAI,IAAI,CAAC,IAChD,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS;oBACxB,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;oBACjD,CAAC,CAAC,EAAE,CAAC,KACP,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAC7C,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,KAE/C,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,SAAS;oBACjC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;oBACrC,CAAC,CAAC,EAAE,CAAC,CACR;aACF,CAAC;YAEF,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;gBACnD,IAAI,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,yCAAyC,OAAO,GAAG,CAAC,IAAI,OAAO,GAAG,CACnE,CAAC;oBAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;oBAErD,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;oBAEhC,oEAAoE;oBACpE,6DAA6D;oBAC7D,IAAK,QAAqC,CAAC,WAAW,KAAK,QAAQ,EAAE,CAAC;wBACpE,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,+EAA+E;4BAC7E,+CAA+C,CAClD,CAAC;oBACJ,CAAC;oBAED,qBAAqB;oBACrB,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;oBACxD,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,qBAAqB,eAAe,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAC5D,CAAC;oBAEF,MAAM,MAAM,GAAG,IAAA,wBAAgB,EAAC,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,EAAE,GAAG,EAAE,CACzE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAC3E,CAAC;oBAEF,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC9B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,0BAA0B,OAAO,GAAG,CAAC,YAAY,KAAK,EAAE,CACzD,CAAC;oBAEF,IAAI,OAAO,KAAK,OAAO,GAAG,CAAC,EAAE,CAAC;wBAC5B,MAAM,IAAI,KAAK,CACb,yCAAyC,OAAO,cAAc,KAAK,EAAE,CACtE,CAAC;oBACJ,CAAC;oBAED,oBAAoB;oBACpB,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YAED,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;KAAA;IAEa,gBAAgB,CAAC,QAAsB;;YACnD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtE,MAAM,cAAc,GAAG,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAEpD,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;gBAC3C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;YACvE,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,cAAc,CAAC;YAErD,MAAM,cAAc,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,iBAC3C,IAAI,EAAE,GAAG,CAAC,IAAI,EACd,OAAO,EAAE,GAAG,CAAC,OAAO,IACjB,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,EACnD,CAAC,CAAC;YAEJ,OAAO,cAAc,CAAC;QACxB,CAAC;KAAA;IAEK,oBAAoB,CAAC,SAAiB;;YAC1C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;YAC/D,OAAO,SAAS,CAAC,YAAY,CAAC;QAChC,CAAC;KAAA;IAED;;OAEG;IACG,gBAAgB,CAAC,SAAiB;;YACtC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;gBACxC,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;YACzD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAuC,KAAK,EAAE,CAAC,CAAC;gBAClE,OAAO,KAAK,CAAC;YACf,CAAC;QACH,CAAC;KAAA;IAED;;OAEG;IACK,gBAAgB,CAAC,QAAsB;;QAC7C,IAAI,CAAC,SAAS,GAAG;YACf,YAAY,EAAE,QAAQ,CAAC,iBAAiB;YACxC,gBAAgB,EAAE,QAAQ,CAAC,UAAU;YACrC,WAAW,EACT,CAAC,MAAA,QAAQ,CAAC,iBAAiB,mCAAI,CAAC,CAAC,GAAG,CAAC,MAAA,QAAQ,CAAC,UAAU,mCAAI,CAAC,CAAC,IAAI,SAAS;SAC9E,CAAC;QACF,IAAI,YAAK,CAAC,OAAO;YAAE,YAAK,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACpE,MAAM,KAAK,GAAG;YACZ,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,iBAAiB,EAAE,QAAQ,CAAC,iBAAiB;YAC7C,iBAAiB,EAAE,QAAQ,CAAC,cAAc,GAAG,OAAO;YACpD,cAAc,EACZ,CAAC,QAAQ,GAAG,CAAC,QAAQ,CAAC,iBAAiB,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;gBAC/D,QAAQ,CAAC,cAAc;SAC1B,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,EAAU;QACtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;CACF;AAlLD,sCAkLC"}
@@ -0,0 +1,190 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ var __importDefault = (this && this.__importDefault) || function (mod) {
12
+ return (mod && mod.__esModule) ? mod : { "default": mod };
13
+ };
14
+ Object.defineProperty(exports, "__esModule", { value: true });
15
+ exports.OpenAICompatibleService = void 0;
16
+ const openai_1 = __importDefault(require("openai"));
17
+ const zod_to_json_schema_1 = require("zod-to-json-schema");
18
+ const utils_1 = require("../../shared/utils");
19
+ const cost_1 = require("../cost");
20
+ /**
21
+ * LLM provider for any OpenAI-compatible chat-completions endpoint
22
+ * (OpenAI, OpenRouter, Together, local vLLM, Ollama's OpenAI shim, ...).
23
+ *
24
+ * `options.host` is used as the base URL and `options.apiKey` as the bearer
25
+ * token. Structured output is requested via `response_format: json_schema`,
26
+ * with a graceful fallback to `json_object` + schema-in-prompt for providers
27
+ * or models (e.g. some Gemma deployments) that reject json_schema.
28
+ */
29
+ class OpenAICompatibleService {
30
+ constructor(options, logger) {
31
+ // Flips to false for the rest of the process once a provider rejects json_schema.
32
+ this.useJsonSchema = true;
33
+ this.logger = logger;
34
+ this.options = Object.assign({ temperature: 0.1, contextLength: 8192 }, options);
35
+ this.client = new openai_1.default({
36
+ apiKey: this.options.apiKey || "not-needed",
37
+ baseURL: this.options.host,
38
+ });
39
+ }
40
+ generateStructured(messages_1, schema_1) {
41
+ return __awaiter(this, arguments, void 0, function* (messages, schema, retries = 3) {
42
+ var _a, _b, _c, _d;
43
+ const jsonSchema = this.toPlainJsonSchema(schema);
44
+ for (let attempt = 0; attempt < retries; attempt++) {
45
+ try {
46
+ this.logger.debug(`Generating structured output (attempt ${attempt + 1}/${retries}, ` +
47
+ `mode=${this.useJsonSchema ? "json_schema" : "json_object"})`);
48
+ const completion = yield this.client.chat.completions.create(Object.assign(Object.assign({ model: this.options.model, messages: this.toOpenAIMessages(messages, jsonSchema), temperature: Number((_a = this.options.temperature) !== null && _a !== void 0 ? _a : 0.1) }, (this.options.maxTokens
49
+ ? { max_tokens: Number(this.options.maxTokens) }
50
+ : {})), { response_format: this.useJsonSchema
51
+ ? {
52
+ type: "json_schema",
53
+ json_schema: {
54
+ name: "knowledge_graph",
55
+ schema: jsonSchema,
56
+ strict: false,
57
+ },
58
+ }
59
+ : { type: "json_object" } }));
60
+ this.logUsage(completion);
61
+ // A "length" finish means the model ran out of output budget and the
62
+ // JSON is truncated — surface an actionable hint instead of a cryptic
63
+ // SyntaxError on the partial body.
64
+ if (((_b = completion.choices[0]) === null || _b === void 0 ? void 0 : _b.finish_reason) === "length") {
65
+ this.logger.warn("LLM output was truncated at the max output-token limit — the JSON is incomplete. " +
66
+ "Increase --max-tokens or reduce --chunk-size.");
67
+ }
68
+ const responseContent = (((_d = (_c = completion.choices[0]) === null || _c === void 0 ? void 0 : _c.message) === null || _d === void 0 ? void 0 : _d.content) || "").trim();
69
+ this.logger.debug(`Raw LLM response: ${responseContent.substring(0, 200)}...`);
70
+ const parsed = (0, utils_1.parseJsonLenient)(this.stripCodeFence(responseContent), () => this.logger.warn("Response JSON was malformed; recovered with jsonrepair"));
71
+ return schema.parse(parsed);
72
+ }
73
+ catch (error) {
74
+ this.logger.error(`LLM generation attempt ${attempt + 1} failed: ${error}`);
75
+ // Degrade to json_object if the provider/model rejects json_schema.
76
+ if (this.useJsonSchema && this.isResponseFormatError(error)) {
77
+ this.logger.warn("Provider rejected json_schema response_format; falling back to json_object.");
78
+ this.useJsonSchema = false;
79
+ }
80
+ if (attempt === retries - 1) {
81
+ throw new Error(`Failed to generate valid output after ${retries} attempts: ${error}`);
82
+ }
83
+ yield this.delay(this.backoffMs(error, attempt));
84
+ }
85
+ }
86
+ throw new Error("Should not reach here");
87
+ });
88
+ }
89
+ /**
90
+ * OpenAI-compatible endpoints have no model-introspection equivalent of
91
+ * Ollama's `show`. Vision support cannot be probed, so report none and let
92
+ * the caller decide whether to attach images.
93
+ */
94
+ getModelCapabilities(_modelName) {
95
+ return __awaiter(this, void 0, void 0, function* () {
96
+ return [];
97
+ });
98
+ }
99
+ toOpenAIMessages(messages, jsonSchema) {
100
+ return messages.map((msg) => {
101
+ // In json_object mode the schema must travel in the prompt itself.
102
+ const content = !this.useJsonSchema && msg.role === "system"
103
+ ? `${msg.content}\n\nRespond ONLY with JSON matching this JSON schema:\n${JSON.stringify(jsonSchema)}`
104
+ : msg.content;
105
+ const hasImages = this.options.images && msg.images && msg.images.length > 0;
106
+ if (hasImages && msg.role === "user") {
107
+ return {
108
+ role: "user",
109
+ content: [
110
+ { type: "text", text: content },
111
+ ...msg.images.map((b64) => ({
112
+ type: "image_url",
113
+ image_url: { url: `data:image/png;base64,${b64}` },
114
+ })),
115
+ ],
116
+ };
117
+ }
118
+ return { role: msg.role, content };
119
+ });
120
+ }
121
+ toPlainJsonSchema(schema) {
122
+ const json = (0, zod_to_json_schema_1.zodToJsonSchema)(schema);
123
+ // OpenAI's json_schema does not want the meta $schema key.
124
+ delete json["$schema"];
125
+ return json;
126
+ }
127
+ stripCodeFence(content) {
128
+ if (content.startsWith("```")) {
129
+ return content.slice(content.indexOf("\n") + 1, content.lastIndexOf("\n"));
130
+ }
131
+ return content;
132
+ }
133
+ /**
134
+ * Backoff before the next retry. Rate-limit / overload responses (429, 503)
135
+ * back off much longer — honoring a `Retry-After` header when present, else
136
+ * exponential (2s, 4s, 8s … capped at 30s) — so sustained cloud throttling
137
+ * recovers within the retry budget instead of failing the request. Other
138
+ * errors keep the original linear 1s/2s/3s.
139
+ */
140
+ backoffMs(error, attempt) {
141
+ const status = error === null || error === void 0 ? void 0 : error.status;
142
+ if (status === 429 || status === 503) {
143
+ const retryAfter = this.retryAfterMs(error);
144
+ return retryAfter !== null && retryAfter !== void 0 ? retryAfter : Math.min(2000 * Math.pow(2, attempt), 30000);
145
+ }
146
+ return 1000 * (attempt + 1);
147
+ }
148
+ /** Parse a `Retry-After` header (seconds) into ms, when the provider sends one. */
149
+ retryAfterMs(error) {
150
+ const headers = error === null || error === void 0 ? void 0 : error.headers;
151
+ const raw = headers === null || headers === void 0 ? void 0 : headers["retry-after"];
152
+ if (!raw)
153
+ return undefined;
154
+ const secs = Number(raw);
155
+ return Number.isFinite(secs) ? secs * 1000 : undefined;
156
+ }
157
+ isResponseFormatError(error) {
158
+ var _a;
159
+ const message = String((_a = error === null || error === void 0 ? void 0 : error.message) !== null && _a !== void 0 ? _a : error);
160
+ // Only a genuine response_format / json_schema rejection should latch the
161
+ // downgrade for the rest of the process. A bare HTTP 400 (a malformed chunk,
162
+ // a rate limit, a content-policy block) must NOT permanently disable
163
+ // json_schema — that was the KG-18 bug; the message must name the feature.
164
+ return /response[_\s-]?format|json[_\s-]?schema/i.test(message);
165
+ }
166
+ logUsage(completion) {
167
+ if (completion.usage) {
168
+ this.lastUsage = {
169
+ promptTokens: completion.usage.prompt_tokens,
170
+ completionTokens: completion.usage.completion_tokens,
171
+ totalTokens: completion.usage.total_tokens,
172
+ };
173
+ if (cost_1.meter.enabled)
174
+ cost_1.meter.record(this.options.model, this.lastUsage);
175
+ this.logger.info(`LLM stats: ${JSON.stringify({
176
+ prompt_tokens: completion.usage.prompt_tokens,
177
+ completion_tokens: completion.usage.completion_tokens,
178
+ total_tokens: completion.usage.total_tokens,
179
+ })}`);
180
+ }
181
+ }
182
+ getLastUsage() {
183
+ return this.lastUsage;
184
+ }
185
+ delay(ms) {
186
+ return new Promise((resolve) => setTimeout(resolve, ms));
187
+ }
188
+ }
189
+ exports.OpenAICompatibleService = OpenAICompatibleService;
190
+ //# sourceMappingURL=OpenAICompatibleService.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"OpenAICompatibleService.js","sourceRoot":"","sources":["../../../src/core/llm/OpenAICompatibleService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,oDAA4B;AAE5B,2DAAqD;AAErD,8CAAsD;AAEtD,kCAAgC;AAEhC;;;;;;;;GAQG;AACH,MAAa,uBAAuB;IASlC,YAAY,OAAmB,EAAE,MAAc;QAL/C,kFAAkF;QAC1E,kBAAa,GAAG,IAAI,CAAC;QAK3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,mBACV,WAAW,EAAE,GAAG,EAChB,aAAa,EAAE,IAAI,IAChB,OAAO,CACX,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAM,CAAC;YACvB,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,YAAY;YAC3C,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;SAC3B,CAAC,CAAC;IACL,CAAC;IAEK,kBAAkB;6DACtB,QAAsB,EACtB,MAAoB,EACpB,UAAkB,CAAC;;YAEnB,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAElD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;gBACnD,IAAI,CAAC;oBACH,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,yCAAyC,OAAO,GAAG,CAAC,IAAI,OAAO,IAAI;wBACjE,QAAQ,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,GAAG,CAChE,CAAC;oBAEF,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,+BAC1D,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,EACzB,QAAQ,EAAE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,CAAC,EACrD,WAAW,EAAE,MAAM,CAAC,MAAA,IAAI,CAAC,OAAO,CAAC,WAAW,mCAAI,GAAG,CAAC,IACjD,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS;wBACxB,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;wBAChD,CAAC,CAAC,EAAE,CAAC,KACP,eAAe,EAAE,IAAI,CAAC,aAAa;4BACjC,CAAC,CAAC;gCACE,IAAI,EAAE,aAAa;gCACnB,WAAW,EAAE;oCACX,IAAI,EAAE,iBAAiB;oCACvB,MAAM,EAAE,UAAU;oCAClB,MAAM,EAAE,KAAK;iCACd;6BACF;4BACH,CAAC,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAC3B,CAAC;oBAEH,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;oBAE1B,qEAAqE;oBACrE,sEAAsE;oBACtE,mCAAmC;oBACnC,IAAI,CAAA,MAAA,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,0CAAE,aAAa,MAAK,QAAQ,EAAE,CAAC;wBACtD,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,mFAAmF;4BACjF,+CAA+C,CAClD,CAAC;oBACJ,CAAC;oBAED,MAAM,eAAe,GAAG,CACtB,CAAA,MAAA,MAAA,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,0CAAE,OAAO,0CAAE,OAAO,KAAI,EAAE,CAC9C,CAAC,IAAI,EAAE,CAAC;oBACT,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,qBAAqB,eAAe,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAC5D,CAAC;oBAEF,MAAM,MAAM,GAAG,IAAA,wBAAgB,EAAC,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,EAAE,GAAG,EAAE,CACzE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAC3E,CAAC;oBACF,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC9B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,0BAA0B,OAAO,GAAG,CAAC,YAAY,KAAK,EAAE,CACzD,CAAC;oBAEF,oEAAoE;oBACpE,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAAC;wBAC5D,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,6EAA6E,CAC9E,CAAC;wBACF,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;oBAC7B,CAAC;oBAED,IAAI,OAAO,KAAK,OAAO,GAAG,CAAC,EAAE,CAAC;wBAC5B,MAAM,IAAI,KAAK,CACb,yCAAyC,OAAO,cAAc,KAAK,EAAE,CACtE,CAAC;oBACJ,CAAC;oBAED,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;YAED,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;KAAA;IAED;;;;OAIG;IACG,oBAAoB,CAAC,UAAkB;;YAC3C,OAAO,EAAE,CAAC;QACZ,CAAC;KAAA;IAEO,gBAAgB,CACtB,QAAsB,EACtB,UAAmC;QAEnC,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAC1B,mEAAmE;YACnE,MAAM,OAAO,GACX,CAAC,IAAI,CAAC,aAAa,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;gBAC1C,CAAC,CAAC,GAAG,GAAG,CAAC,OAAO,0DAA0D,IAAI,CAAC,SAAS,CACpF,UAAU,CACX,EAAE;gBACL,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;YAElB,MAAM,SAAS,GACb,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;YAE7D,IAAI,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACrC,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE;wBAC/B,GAAG,GAAG,CAAC,MAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;4BAC3B,IAAI,EAAE,WAAoB;4BAC1B,SAAS,EAAE,EAAE,GAAG,EAAE,yBAAyB,GAAG,EAAE,EAAE;yBACnD,CAAC,CAAC;qBACJ;iBACF,CAAC;YACJ,CAAC;YAED,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAwD,CAAC;QAC3F,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CAAC,MAA0B;QAClD,MAAM,IAAI,GAAG,IAAA,oCAAe,EAAC,MAAM,CAA4B,CAAC;QAChE,2DAA2D;QAC3D,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,cAAc,CAAC,OAAe;QACpC,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,OAAO,CAAC,KAAK,CAClB,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,EACzB,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAC1B,CAAC;QACJ,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;;;;OAMG;IACK,SAAS,CAAC,KAAc,EAAE,OAAe;QAC/C,MAAM,MAAM,GAAI,KAA6B,aAA7B,KAAK,uBAAL,KAAK,CAA0B,MAAM,CAAC;QACtD,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;YACrC,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YAC5C,OAAO,UAAU,aAAV,UAAU,cAAV,UAAU,GAAI,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,SAAA,CAAC,EAAI,OAAO,CAAA,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC;QACD,OAAO,IAAI,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,mFAAmF;IAC3E,YAAY,CAAC,KAAc;QACjC,MAAM,OAAO,GAAI,KAA8C,aAA9C,KAAK,uBAAL,KAAK,CAA2C,OAAO,CAAC;QACzE,MAAM,GAAG,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAG,aAAa,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG;YAAE,OAAO,SAAS,CAAC;QAC3B,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QACzB,OAAO,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACzD,CAAC;IAEO,qBAAqB,CAAC,KAAc;;QAC1C,MAAM,OAAO,GAAG,MAAM,CAAC,MAAC,KAA8B,aAA9B,KAAK,uBAAL,KAAK,CAA2B,OAAO,mCAAI,KAAK,CAAC,CAAC;QAC1E,0EAA0E;QAC1E,6EAA6E;QAC7E,qEAAqE;QACrE,2EAA2E;QAC3E,OAAO,0CAA0C,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClE,CAAC;IAEO,QAAQ,CAAC,UAAkD;QACjE,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,SAAS,GAAG;gBACf,YAAY,EAAE,UAAU,CAAC,KAAK,CAAC,aAAa;gBAC5C,gBAAgB,EAAE,UAAU,CAAC,KAAK,CAAC,iBAAiB;gBACpD,WAAW,EAAE,UAAU,CAAC,KAAK,CAAC,YAAY;aAC3C,CAAC;YACF,IAAI,YAAK,CAAC,OAAO;gBAAE,YAAK,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YACpE,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,cAAc,IAAI,CAAC,SAAS,CAAC;gBAC3B,aAAa,EAAE,UAAU,CAAC,KAAK,CAAC,aAAa;gBAC7C,iBAAiB,EAAE,UAAU,CAAC,KAAK,CAAC,iBAAiB;gBACrD,YAAY,EAAE,UAAU,CAAC,KAAK,CAAC,YAAY;aAC5C,CAAC,EAAE,CACL,CAAC;QACJ,CAAC;IACH,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,EAAU;QACtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;CACF;AA9ND,0DA8NC"}
@@ -0,0 +1,129 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ var __importDefault = (this && this.__importDefault) || function (mod) {
12
+ return (mod && mod.__esModule) ? mod : { "default": mod };
13
+ };
14
+ Object.defineProperty(exports, "__esModule", { value: true });
15
+ exports.OpenAIEmbeddingService = void 0;
16
+ const openai_1 = __importDefault(require("openai"));
17
+ const embeddingUtils_1 = require("./embeddingUtils");
18
+ /**
19
+ * Embedding provider for any OpenAI-compatible embeddings endpoint.
20
+ * Mirrors EmbeddingService (same in-memory cache contract) but batches
21
+ * natively via the array `input` form to cut request count on metered APIs.
22
+ */
23
+ class OpenAIEmbeddingService {
24
+ constructor(options, logger) {
25
+ this.options = Object.assign({ maxInputChars: embeddingUtils_1.DEFAULT_MAX_EMBED_CHARS }, options);
26
+ this.cache = new Map();
27
+ this.logger = logger;
28
+ this.client = new openai_1.default({
29
+ apiKey: options.apiKey || "not-needed",
30
+ baseURL: options.host,
31
+ });
32
+ }
33
+ /** Trim input to the configured char budget before embedding. */
34
+ truncate(text) {
35
+ const max = this.options.maxInputChars;
36
+ if (max && max > 0 && text.length > max) {
37
+ this.logger.debug(`Truncating embedding input from ${text.length} to ${max} chars`);
38
+ return text.slice(0, max);
39
+ }
40
+ return text;
41
+ }
42
+ embed(text) {
43
+ return __awaiter(this, void 0, void 0, function* () {
44
+ const cached = this.cache.get(text);
45
+ if (cached)
46
+ return cached;
47
+ const embedding = yield this.embedWithFallback(this.truncate(text));
48
+ this.cache.set(text, embedding);
49
+ return embedding;
50
+ });
51
+ }
52
+ /** Embed one input, halving + retrying if the model rejects it as too long. */
53
+ embedWithFallback(input) {
54
+ return __awaiter(this, void 0, void 0, function* () {
55
+ let current = input;
56
+ while (true) {
57
+ try {
58
+ const response = yield this.client.embeddings.create({
59
+ model: this.options.model,
60
+ input: current,
61
+ });
62
+ return response.data[0].embedding;
63
+ }
64
+ catch (error) {
65
+ if ((0, embeddingUtils_1.isContextLengthError)(error) && current.length > embeddingUtils_1.MIN_EMBED_CHARS) {
66
+ const next = Math.max(embeddingUtils_1.MIN_EMBED_CHARS, Math.floor(current.length / 2));
67
+ this.logger.warn(`Embedding input too long (${current.length} chars); retrying with ${next}`);
68
+ current = current.slice(0, next);
69
+ continue;
70
+ }
71
+ this.logger.error(`Failed to generate embedding: ${error}`);
72
+ throw new Error(`Failed to generate embedding: ${error}`);
73
+ }
74
+ }
75
+ });
76
+ }
77
+ embedBatch(texts) {
78
+ return __awaiter(this, void 0, void 0, function* () {
79
+ this.logger.debug(`Generating embeddings for ${texts.length} texts`);
80
+ const results = new Array(texts.length);
81
+ const uncachedIndexes = [];
82
+ texts.forEach((text, i) => {
83
+ const cached = this.cache.get(text);
84
+ if (cached) {
85
+ results[i] = cached;
86
+ }
87
+ else {
88
+ uncachedIndexes.push(i);
89
+ }
90
+ });
91
+ const batchSize = 100;
92
+ for (let i = 0; i < uncachedIndexes.length; i += batchSize) {
93
+ const batchIndexes = uncachedIndexes.slice(i, i + batchSize);
94
+ const batchInput = batchIndexes.map((idx) => this.truncate(texts[idx]));
95
+ try {
96
+ const response = yield this.client.embeddings.create({
97
+ model: this.options.model,
98
+ input: batchInput,
99
+ });
100
+ response.data.forEach((item, j) => {
101
+ const originalIndex = batchIndexes[j];
102
+ this.cache.set(texts[originalIndex], item.embedding);
103
+ results[originalIndex] = item.embedding;
104
+ });
105
+ }
106
+ catch (error) {
107
+ if (!(0, embeddingUtils_1.isContextLengthError)(error))
108
+ throw error;
109
+ // A too-long item (or too-large request) — fall back to per-item embed,
110
+ // which adaptively shrinks oversized inputs.
111
+ this.logger.warn(`Batch embedding too long; falling back to per-item embedding for ${batchIndexes.length} items`);
112
+ for (const idx of batchIndexes) {
113
+ results[idx] = yield this.embed(texts[idx]);
114
+ }
115
+ }
116
+ }
117
+ return results;
118
+ });
119
+ }
120
+ clearCache() {
121
+ this.cache.clear();
122
+ this.logger.debug("Embedding cache cleared");
123
+ }
124
+ getCacheSize() {
125
+ return this.cache.size;
126
+ }
127
+ }
128
+ exports.OpenAIEmbeddingService = OpenAIEmbeddingService;
129
+ //# sourceMappingURL=OpenAIEmbeddingService.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"OpenAIEmbeddingService.js","sourceRoot":"","sources":["../../../src/core/llm/OpenAIEmbeddingService.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,oDAA4B;AAG5B,qDAAkG;AAUlG;;;;GAIG;AACH,MAAa,sBAAsB;IAMjC,YAAY,OAA+B,EAAE,MAAc;QACzD,IAAI,CAAC,OAAO,mBAAK,aAAa,EAAE,wCAAuB,IAAK,OAAO,CAAE,CAAC;QACtE,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAM,CAAC;YACvB,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,YAAY;YACtC,OAAO,EAAE,OAAO,CAAC,IAAI;SACtB,CAAC,CAAC;IACL,CAAC;IAED,iEAAiE;IACzD,QAAQ,CAAC,IAAY;QAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;QACvC,IAAI,GAAG,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,mCAAmC,IAAI,CAAC,MAAM,OAAO,GAAG,QAAQ,CACjE,CAAC;YACF,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAEK,KAAK,CAAC,IAAY;;YACtB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,MAAM;gBAAE,OAAO,MAAM,CAAC;YAE1B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YACpE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YAChC,OAAO,SAAS,CAAC;QACnB,CAAC;KAAA;IAED,+EAA+E;IACjE,iBAAiB,CAAC,KAAa;;YAC3C,IAAI,OAAO,GAAG,KAAK,CAAC;YACpB,OAAO,IAAI,EAAE,CAAC;gBACZ,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;wBACnD,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;wBACzB,KAAK,EAAE,OAAO;qBACf,CAAC,CAAC;oBACH,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACpC,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,IAAA,qCAAoB,EAAC,KAAK,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,gCAAe,EAAE,CAAC;wBACpE,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,gCAAe,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;wBACvE,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,6BAA6B,OAAO,CAAC,MAAM,0BAA0B,IAAI,EAAE,CAC5E,CAAC;wBACF,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;wBACjC,SAAS;oBACX,CAAC;oBACD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,KAAK,EAAE,CAAC,CAAC;oBAC5D,MAAM,IAAI,KAAK,CAAC,iCAAiC,KAAK,EAAE,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC;QACH,CAAC;KAAA;IAEK,UAAU,CAAC,KAAe;;YAC9B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,KAAK,CAAC,MAAM,QAAQ,CAAC,CAAC;YAErE,MAAM,OAAO,GAAe,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACpD,MAAM,eAAe,GAAa,EAAE,CAAC;YAErC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;gBACxB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBACpC,IAAI,MAAM,EAAE,CAAC;oBACX,OAAO,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC;gBACtB,CAAC;qBAAM,CAAC;oBACN,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,GAAG,CAAC;YACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;gBAC3D,MAAM,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;gBAC7D,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAExE,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;wBACnD,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;wBACzB,KAAK,EAAE,UAAU;qBAClB,CAAC,CAAC;oBAEH,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;wBAChC,MAAM,aAAa,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;wBACtC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;wBACrD,OAAO,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC;oBAC1C,CAAC,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,CAAC,IAAA,qCAAoB,EAAC,KAAK,CAAC;wBAAE,MAAM,KAAK,CAAC;oBAC9C,wEAAwE;oBACxE,6CAA6C;oBAC7C,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,oEAAoE,YAAY,CAAC,MAAM,QAAQ,CAChG,CAAC;oBACF,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;wBAC/B,OAAO,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;oBAC9C,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;KAAA;IAED,UAAU;QACR,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IACzB,CAAC;CACF;AArHD,wDAqHC"}
@@ -0,0 +1,25 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.DEFAULT_MAX_EMBED_CHARS = exports.MIN_EMBED_CHARS = void 0;
4
+ exports.isContextLengthError = isContextLengthError;
5
+ /** Floor for adaptive embedding-input shrinking — don't retry below this. */
6
+ exports.MIN_EMBED_CHARS = 256;
7
+ /** Default embedding input cap (chars). Conservative for 512-token models
8
+ * (e.g. mxbai-embed-large); raise via config for large-context cloud models. */
9
+ exports.DEFAULT_MAX_EMBED_CHARS = 1024;
10
+ /**
11
+ * Heuristic: does this error indicate the input exceeded the model's context?
12
+ * Covers Ollama ("the input length exceeds the context length") and common
13
+ * OpenAI-compatible phrasings.
14
+ */
15
+ function isContextLengthError(error) {
16
+ var _a;
17
+ const msg = String((_a = error === null || error === void 0 ? void 0 : error.message) !== null && _a !== void 0 ? _a : error).toLowerCase();
18
+ return (msg.includes("context length") ||
19
+ msg.includes("context window") ||
20
+ msg.includes("input length") ||
21
+ msg.includes("maximum context") ||
22
+ msg.includes("too long") ||
23
+ msg.includes("exceeds"));
24
+ }
25
+ //# sourceMappingURL=embeddingUtils.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"embeddingUtils.js","sourceRoot":"","sources":["../../../src/core/llm/embeddingUtils.ts"],"names":[],"mappings":";;;AAYA,oDAUC;AAtBD,6EAA6E;AAChE,QAAA,eAAe,GAAG,GAAG,CAAC;AAEnC;gFACgF;AACnE,QAAA,uBAAuB,GAAG,IAAI,CAAC;AAE5C;;;;GAIG;AACH,SAAgB,oBAAoB,CAAC,KAAc;;IACjD,MAAM,GAAG,GAAG,MAAM,CAAC,MAAC,KAA8B,aAA9B,KAAK,uBAAL,KAAK,CAA2B,OAAO,mCAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IACpF,OAAO,CACL,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QAC9B,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QAC9B,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC5B,GAAG,CAAC,QAAQ,CAAC,iBAAiB,CAAC;QAC/B,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC;QACxB,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,CACxB,CAAC;AACJ,CAAC"}
@@ -0,0 +1,23 @@
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 __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
16
+ Object.defineProperty(exports, "__esModule", { value: true });
17
+ __exportStar(require("./OllamaService"), exports);
18
+ __exportStar(require("./OpenAICompatibleService"), exports);
19
+ __exportStar(require("./EmbeddingService"), exports);
20
+ __exportStar(require("./OpenAIEmbeddingService"), exports);
21
+ __exportStar(require("./prompts/PromptManager"), exports);
22
+ __exportStar(require("./prompts/PromptTemplateEngine"), exports);
23
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/llm/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,kDAAgC;AAChC,4DAA0C;AAC1C,qDAAmC;AACnC,2DAAyC;AACzC,0DAAwC;AACxC,iEAA+C"}