@tyvm/knowhow 0.0.1

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 (492) hide show
  1. package/.vscode/launch.json +14 -0
  2. package/CONFIG.md +336 -0
  3. package/README.md +49 -0
  4. package/autodoc/chat.mdx +20 -0
  5. package/autodoc/cli.mdx +11 -0
  6. package/autodoc/plugins/asana.mdx +47 -0
  7. package/autodoc/plugins/downloader/downloader.mdx +38 -0
  8. package/autodoc/plugins/downloader/plugin.mdx +37 -0
  9. package/autodoc/plugins/downloader/types.mdx +42 -0
  10. package/autodoc/plugins/embedding.mdx +41 -0
  11. package/autodoc/plugins/figma.mdx +45 -0
  12. package/autodoc/plugins/github.mdx +40 -0
  13. package/autodoc/plugins/jira.mdx +46 -0
  14. package/autodoc/plugins/language.mdx +37 -0
  15. package/autodoc/plugins/linear.mdx +35 -0
  16. package/autodoc/plugins/notion.mdx +38 -0
  17. package/autodoc/plugins/plugins.mdx +59 -0
  18. package/autodoc/plugins/types.mdx +51 -0
  19. package/autodoc/plugins/vim.mdx +39 -0
  20. package/autodoc/tools/addInternalTools.mdx +1 -0
  21. package/autodoc/tools/agentCall.mdx +1 -0
  22. package/autodoc/tools/asana/definitions.mdx +10 -0
  23. package/autodoc/tools/asana/index.mdx +12 -0
  24. package/autodoc/tools/askHuman.mdx +1 -0
  25. package/autodoc/tools/callPlugin.mdx +1 -0
  26. package/autodoc/tools/embeddingSearch.mdx +1 -0
  27. package/autodoc/tools/execCommand.mdx +1 -0
  28. package/autodoc/tools/fileSearch.mdx +1 -0
  29. package/autodoc/tools/finalAnswer.mdx +1 -0
  30. package/autodoc/tools/github/definitions.mdx +6 -0
  31. package/autodoc/tools/github/index.mdx +8 -0
  32. package/autodoc/tools/index.mdx +14 -0
  33. package/autodoc/tools/lintFile.mdx +7 -0
  34. package/autodoc/tools/list.mdx +16 -0
  35. package/autodoc/tools/modifyFile.mdx +7 -0
  36. package/autodoc/tools/patch.mdx +9 -0
  37. package/autodoc/tools/readBlocks.mdx +1 -0
  38. package/autodoc/tools/readFile.mdx +1 -0
  39. package/autodoc/tools/scanFile.mdx +1 -0
  40. package/autodoc/tools/textSearch.mdx +6 -0
  41. package/autodoc/tools/types/fileblock.mdx +1 -0
  42. package/autodoc/tools/visionTool.mdx +1 -0
  43. package/autodoc/tools/writeFile.mdx +1 -0
  44. package/jest.config.js +18 -0
  45. package/package.json +89 -0
  46. package/src/agents/base/base.ts +619 -0
  47. package/src/agents/base/prompt.ts +26 -0
  48. package/src/agents/configurable/ConfigAgent.ts +23 -0
  49. package/src/agents/developer/developer.ts +69 -0
  50. package/src/agents/index.ts +8 -0
  51. package/src/agents/interface.ts +11 -0
  52. package/src/agents/patcher/codebase.md +27 -0
  53. package/src/agents/patcher/patcher.ts +110 -0
  54. package/src/agents/researcher/researcher.ts +109 -0
  55. package/src/agents/tools/addInternalTools.ts +21 -0
  56. package/src/agents/tools/agentCall.ts +18 -0
  57. package/src/agents/tools/aiClient.ts +36 -0
  58. package/src/agents/tools/asana/definitions.ts +199 -0
  59. package/src/agents/tools/asana/index.ts +108 -0
  60. package/src/agents/tools/askHuman.ts +8 -0
  61. package/src/agents/tools/callPlugin.ts +4 -0
  62. package/src/agents/tools/embeddingSearch.ts +5 -0
  63. package/src/agents/tools/execCommand.ts +26 -0
  64. package/src/agents/tools/fileSearch.ts +33 -0
  65. package/src/agents/tools/finalAnswer.ts +4 -0
  66. package/src/agents/tools/github/definitions.ts +89 -0
  67. package/src/agents/tools/github/index.ts +67 -0
  68. package/src/agents/tools/googleSearch.ts +242 -0
  69. package/src/agents/tools/index.ts +24 -0
  70. package/src/agents/tools/language/definitions.ts +97 -0
  71. package/src/agents/tools/language/index.ts +19 -0
  72. package/src/agents/tools/lintFile.ts +17 -0
  73. package/src/agents/tools/list.ts +514 -0
  74. package/src/agents/tools/loadWebpage.ts +129 -0
  75. package/src/agents/tools/modifyFile.ts +53 -0
  76. package/src/agents/tools/patch.ts +672 -0
  77. package/src/agents/tools/readBlocks.ts +41 -0
  78. package/src/agents/tools/readFile.ts +39 -0
  79. package/src/agents/tools/scanFile.ts +16 -0
  80. package/src/agents/tools/textSearch.ts +21 -0
  81. package/src/agents/tools/types/fileblock.ts +5 -0
  82. package/src/agents/tools/visionTool.ts +5 -0
  83. package/src/agents/tools/writeFile.ts +46 -0
  84. package/src/agents/vim/vim.ts +153 -0
  85. package/src/ai.ts +167 -0
  86. package/src/chat.ts +412 -0
  87. package/src/cli.ts +85 -0
  88. package/src/clients/anthropic.ts +399 -0
  89. package/src/clients/gemini.ts +486 -0
  90. package/src/clients/http.ts +107 -0
  91. package/src/clients/index.ts +183 -0
  92. package/src/clients/knowhow.ts +33 -0
  93. package/src/clients/openai.ts +253 -0
  94. package/src/clients/types.ts +91 -0
  95. package/src/clients/xai.ts +132 -0
  96. package/src/config.ts +211 -0
  97. package/src/conversion.ts +140 -0
  98. package/src/dataset/diffs/README.md +12 -0
  99. package/src/dataset/diffs/debug-errors.ts +29 -0
  100. package/src/dataset/diffs/debug.ts +64 -0
  101. package/src/dataset/diffs/generate.ts +71 -0
  102. package/src/dataset/diffs/jsonl.ts +54 -0
  103. package/src/dataset/diffs/test.ts +193 -0
  104. package/src/embeddings.ts +411 -0
  105. package/src/hashes.ts +67 -0
  106. package/src/index.ts +376 -0
  107. package/src/login.ts +78 -0
  108. package/src/microphone.ts +136 -0
  109. package/src/modules/index.ts +37 -0
  110. package/src/modules/types.ts +48 -0
  111. package/src/plugins/asana.ts +134 -0
  112. package/src/plugins/downloader/downloader.ts +204 -0
  113. package/src/plugins/downloader/plugin.ts +85 -0
  114. package/src/plugins/downloader/types.ts +85 -0
  115. package/src/plugins/embedding.ts +42 -0
  116. package/src/plugins/figma.ts +137 -0
  117. package/src/plugins/github.ts +153 -0
  118. package/src/plugins/jira.ts +104 -0
  119. package/src/plugins/language.ts +100 -0
  120. package/src/plugins/linear.ts +219 -0
  121. package/src/plugins/notion.ts +170 -0
  122. package/src/plugins/plugins.ts +56 -0
  123. package/src/plugins/types.ts +5 -0
  124. package/src/plugins/url.ts +63 -0
  125. package/src/plugins/vim.ts +73 -0
  126. package/src/prompts/BasicCodeDocumenter.ts +1 -0
  127. package/src/prompts/BasicProjectDocumenter.ts +1 -0
  128. package/src/prompts/EmbeddingSeachHelper.ts +1 -0
  129. package/src/prompts/index.ts +9 -0
  130. package/src/server/index.ts +14 -0
  131. package/src/services/AgentService.ts +98 -0
  132. package/src/services/EmbeddingService.ts +410 -0
  133. package/src/services/EventService.ts +20 -0
  134. package/src/services/GitHub.ts +60 -0
  135. package/src/services/KnowhowClient.ts +113 -0
  136. package/src/services/Mcp.ts +270 -0
  137. package/src/services/McpServer.ts +119 -0
  138. package/src/services/McpWebsocketTransport.ts +64 -0
  139. package/src/services/S3.ts +126 -0
  140. package/src/services/Tools.ts +65 -0
  141. package/src/services/flags.ts +52 -0
  142. package/src/services/index.ts +8 -0
  143. package/src/types.ts +225 -0
  144. package/src/utils/index.ts +108 -0
  145. package/src/worker.ts +80 -0
  146. package/tests/integration/figma.test.ts +70 -0
  147. package/tests/integration/fileblocks/readwrite.test.ts +95 -0
  148. package/tests/integration/patching/input.txt +145 -0
  149. package/tests/integration/patching/output.txt +145 -0
  150. package/tests/integration/patching/patch.txt +15 -0
  151. package/tests/integration/patching/unseen.txt +80 -0
  152. package/tests/integration/patching.test.ts +136 -0
  153. package/tests/languagePlugin.test.ts +74 -0
  154. package/tests/patching/corrupted.test.ts +78 -0
  155. package/tests/patching/imports.patch.txt +11 -0
  156. package/tests/patching/imports.test.ts +43 -0
  157. package/tests/patching/imports.txt +11 -0
  158. package/tests/patching/interface.patch.txt +15 -0
  159. package/tests/patching/interface.txt +7 -0
  160. package/tests/test.spec.ts +132 -0
  161. package/ts_build/src/agents/base/base.d.ts +88 -0
  162. package/ts_build/src/agents/base/base.js +464 -0
  163. package/ts_build/src/agents/base/base.js.map +1 -0
  164. package/ts_build/src/agents/base/prompt.d.ts +1 -0
  165. package/ts_build/src/agents/base/prompt.js +30 -0
  166. package/ts_build/src/agents/base/prompt.js.map +1 -0
  167. package/ts_build/src/agents/configurable/ConfigAgent.d.ts +10 -0
  168. package/ts_build/src/agents/configurable/ConfigAgent.js +25 -0
  169. package/ts_build/src/agents/configurable/ConfigAgent.js.map +1 -0
  170. package/ts_build/src/agents/configurable/OpenAIAgent.d.ts +0 -0
  171. package/ts_build/src/agents/configurable/OpenAIAgent.js +1 -0
  172. package/ts_build/src/agents/configurable/OpenAIAgent.js.map +1 -0
  173. package/ts_build/src/agents/developer/developer.d.ts +9 -0
  174. package/ts_build/src/agents/developer/developer.js +69 -0
  175. package/ts_build/src/agents/developer/developer.js.map +1 -0
  176. package/ts_build/src/agents/index.d.ts +7 -0
  177. package/ts_build/src/agents/index.js +38 -0
  178. package/ts_build/src/agents/index.js.map +1 -0
  179. package/ts_build/src/agents/interface.d.ts +10 -0
  180. package/ts_build/src/agents/interface.js +3 -0
  181. package/ts_build/src/agents/interface.js.map +1 -0
  182. package/ts_build/src/agents/patcher/patcher.d.ts +9 -0
  183. package/ts_build/src/agents/patcher/patcher.js +106 -0
  184. package/ts_build/src/agents/patcher/patcher.js.map +1 -0
  185. package/ts_build/src/agents/researcher/researcher.d.ts +9 -0
  186. package/ts_build/src/agents/researcher/researcher.js +107 -0
  187. package/ts_build/src/agents/researcher/researcher.js.map +1 -0
  188. package/ts_build/src/agents/tools/addInternalTools.d.ts +5 -0
  189. package/ts_build/src/agents/tools/addInternalTools.js +18 -0
  190. package/ts_build/src/agents/tools/addInternalTools.js.map +1 -0
  191. package/ts_build/src/agents/tools/agentCall.d.ts +1 -0
  192. package/ts_build/src/agents/tools/agentCall.js +21 -0
  193. package/ts_build/src/agents/tools/agentCall.js.map +1 -0
  194. package/ts_build/src/agents/tools/aiClient.d.ts +6 -0
  195. package/ts_build/src/agents/tools/aiClient.js +25 -0
  196. package/ts_build/src/agents/tools/aiClient.js.map +1 -0
  197. package/ts_build/src/agents/tools/asana/definitions.d.ts +202 -0
  198. package/ts_build/src/agents/tools/asana/definitions.js +197 -0
  199. package/ts_build/src/agents/tools/asana/definitions.js.map +1 -0
  200. package/ts_build/src/agents/tools/asana/index.d.ts +8 -0
  201. package/ts_build/src/agents/tools/asana/index.js +98 -0
  202. package/ts_build/src/agents/tools/asana/index.js.map +1 -0
  203. package/ts_build/src/agents/tools/askHuman.d.ts +1 -0
  204. package/ts_build/src/agents/tools/askHuman.js +15 -0
  205. package/ts_build/src/agents/tools/askHuman.js.map +1 -0
  206. package/ts_build/src/agents/tools/callPlugin.d.ts +1 -0
  207. package/ts_build/src/agents/tools/callPlugin.js +9 -0
  208. package/ts_build/src/agents/tools/callPlugin.js.map +1 -0
  209. package/ts_build/src/agents/tools/client.d.ts +5 -0
  210. package/ts_build/src/agents/tools/client.js +21 -0
  211. package/ts_build/src/agents/tools/client.js.map +1 -0
  212. package/ts_build/src/agents/tools/embeddingSearch.d.ts +1 -0
  213. package/ts_build/src/agents/tools/embeddingSearch.js +9 -0
  214. package/ts_build/src/agents/tools/embeddingSearch.js.map +1 -0
  215. package/ts_build/src/agents/tools/execCommand.d.ts +1 -0
  216. package/ts_build/src/agents/tools/execCommand.js +25 -0
  217. package/ts_build/src/agents/tools/execCommand.js.map +1 -0
  218. package/ts_build/src/agents/tools/fileSearch.d.ts +1 -0
  219. package/ts_build/src/agents/tools/fileSearch.js +31 -0
  220. package/ts_build/src/agents/tools/fileSearch.js.map +1 -0
  221. package/ts_build/src/agents/tools/finalAnswer.d.ts +1 -0
  222. package/ts_build/src/agents/tools/finalAnswer.js +8 -0
  223. package/ts_build/src/agents/tools/finalAnswer.js.map +1 -0
  224. package/ts_build/src/agents/tools/github/definitions.d.ts +47 -0
  225. package/ts_build/src/agents/tools/github/definitions.js +86 -0
  226. package/ts_build/src/agents/tools/github/definitions.js.map +1 -0
  227. package/ts_build/src/agents/tools/github/index.d.ts +789 -0
  228. package/ts_build/src/agents/tools/github/index.js +56 -0
  229. package/ts_build/src/agents/tools/github/index.js.map +1 -0
  230. package/ts_build/src/agents/tools/googleSearch.d.ts +93 -0
  231. package/ts_build/src/agents/tools/googleSearch.js +117 -0
  232. package/ts_build/src/agents/tools/googleSearch.js.map +1 -0
  233. package/ts_build/src/agents/tools/googleSearchTypes.d.ts +74 -0
  234. package/ts_build/src/agents/tools/googleSearchTypes.js +3 -0
  235. package/ts_build/src/agents/tools/googleSearchTypes.js.map +1 -0
  236. package/ts_build/src/agents/tools/index.d.ts +24 -0
  237. package/ts_build/src/agents/tools/index.js +41 -0
  238. package/ts_build/src/agents/tools/index.js.map +1 -0
  239. package/ts_build/src/agents/tools/language/definitions.d.ts +89 -0
  240. package/ts_build/src/agents/tools/language/definitions.js +95 -0
  241. package/ts_build/src/agents/tools/language/definitions.js.map +1 -0
  242. package/ts_build/src/agents/tools/language/index.d.ts +4 -0
  243. package/ts_build/src/agents/tools/language/index.js +22 -0
  244. package/ts_build/src/agents/tools/language/index.js.map +1 -0
  245. package/ts_build/src/agents/tools/lintFile.d.ts +1 -0
  246. package/ts_build/src/agents/tools/lintFile.js +22 -0
  247. package/ts_build/src/agents/tools/lintFile.js.map +1 -0
  248. package/ts_build/src/agents/tools/list.d.ts +2 -0
  249. package/ts_build/src/agents/tools/list.js +505 -0
  250. package/ts_build/src/agents/tools/list.js.map +1 -0
  251. package/ts_build/src/agents/tools/loadWebpage.d.ts +8 -0
  252. package/ts_build/src/agents/tools/loadWebpage.js +97 -0
  253. package/ts_build/src/agents/tools/loadWebpage.js.map +1 -0
  254. package/ts_build/src/agents/tools/modifyFile.d.ts +2 -0
  255. package/ts_build/src/agents/tools/modifyFile.js +74 -0
  256. package/ts_build/src/agents/tools/modifyFile.js.map +1 -0
  257. package/ts_build/src/agents/tools/patch.d.ts +21 -0
  258. package/ts_build/src/agents/tools/patch.js +443 -0
  259. package/ts_build/src/agents/tools/patch.js.map +1 -0
  260. package/ts_build/src/agents/tools/readBlocks.d.ts +2 -0
  261. package/ts_build/src/agents/tools/readBlocks.js +59 -0
  262. package/ts_build/src/agents/tools/readBlocks.js.map +1 -0
  263. package/ts_build/src/agents/tools/readFile.d.ts +1 -0
  264. package/ts_build/src/agents/tools/readFile.js +46 -0
  265. package/ts_build/src/agents/tools/readFile.js.map +1 -0
  266. package/ts_build/src/agents/tools/scanFile.d.ts +1 -0
  267. package/ts_build/src/agents/tools/scanFile.js +36 -0
  268. package/ts_build/src/agents/tools/scanFile.js.map +1 -0
  269. package/ts_build/src/agents/tools/textSearch.d.ts +1 -0
  270. package/ts_build/src/agents/tools/textSearch.js +22 -0
  271. package/ts_build/src/agents/tools/textSearch.js.map +1 -0
  272. package/ts_build/src/agents/tools/types/fileblock.d.ts +5 -0
  273. package/ts_build/src/agents/tools/types/fileblock.js +3 -0
  274. package/ts_build/src/agents/tools/types/fileblock.js.map +1 -0
  275. package/ts_build/src/agents/tools/visionTool.d.ts +1 -0
  276. package/ts_build/src/agents/tools/visionTool.js +9 -0
  277. package/ts_build/src/agents/tools/visionTool.js.map +1 -0
  278. package/ts_build/src/agents/tools/writeFile.d.ts +2 -0
  279. package/ts_build/src/agents/tools/writeFile.js +61 -0
  280. package/ts_build/src/agents/tools/writeFile.js.map +1 -0
  281. package/ts_build/src/agents/vim/vim.d.ts +14 -0
  282. package/ts_build/src/agents/vim/vim.js +171 -0
  283. package/ts_build/src/agents/vim/vim.js.map +1 -0
  284. package/ts_build/src/ai.d.ts +10 -0
  285. package/ts_build/src/ai.js +102 -0
  286. package/ts_build/src/ai.js.map +1 -0
  287. package/ts_build/src/chat.d.ts +13 -0
  288. package/ts_build/src/chat.js +319 -0
  289. package/ts_build/src/chat.js.map +1 -0
  290. package/ts_build/src/cli.d.ts +2 -0
  291. package/ts_build/src/cli.js +98 -0
  292. package/ts_build/src/cli.js.map +1 -0
  293. package/ts_build/src/clients/anthropic.d.ts +31 -0
  294. package/ts_build/src/clients/anthropic.js +344 -0
  295. package/ts_build/src/clients/anthropic.js.map +1 -0
  296. package/ts_build/src/clients/gemini.d.ts +20 -0
  297. package/ts_build/src/clients/gemini.js +347 -0
  298. package/ts_build/src/clients/gemini.js.map +1 -0
  299. package/ts_build/src/clients/http.d.ts +12 -0
  300. package/ts_build/src/clients/http.js +91 -0
  301. package/ts_build/src/clients/http.js.map +1 -0
  302. package/ts_build/src/clients/index.d.ts +47 -0
  303. package/ts_build/src/clients/index.js +143 -0
  304. package/ts_build/src/clients/index.js.map +1 -0
  305. package/ts_build/src/clients/knowhow.d.ts +10 -0
  306. package/ts_build/src/clients/knowhow.js +24 -0
  307. package/ts_build/src/clients/knowhow.js.map +1 -0
  308. package/ts_build/src/clients/openai.d.ts +20 -0
  309. package/ts_build/src/clients/openai.js +221 -0
  310. package/ts_build/src/clients/openai.js.map +1 -0
  311. package/ts_build/src/clients/types.d.ts +89 -0
  312. package/ts_build/src/clients/types.js +3 -0
  313. package/ts_build/src/clients/types.js.map +1 -0
  314. package/ts_build/src/clients/xai.d.ts +22 -0
  315. package/ts_build/src/clients/xai.js +108 -0
  316. package/ts_build/src/clients/xai.js.map +1 -0
  317. package/ts_build/src/config.d.ts +9 -0
  318. package/ts_build/src/config.js +216 -0
  319. package/ts_build/src/config.js.map +1 -0
  320. package/ts_build/src/conversion.d.ts +7 -0
  321. package/ts_build/src/conversion.js +114 -0
  322. package/ts_build/src/conversion.js.map +1 -0
  323. package/ts_build/src/dataset/diffs/debug-errors.d.ts +1 -0
  324. package/ts_build/src/dataset/diffs/debug-errors.js +28 -0
  325. package/ts_build/src/dataset/diffs/debug-errors.js.map +1 -0
  326. package/ts_build/src/dataset/diffs/debug.d.ts +1 -0
  327. package/ts_build/src/dataset/diffs/debug.js +50 -0
  328. package/ts_build/src/dataset/diffs/debug.js.map +1 -0
  329. package/ts_build/src/dataset/diffs/generate.d.ts +1 -0
  330. package/ts_build/src/dataset/diffs/generate.js +70 -0
  331. package/ts_build/src/dataset/diffs/generate.js.map +1 -0
  332. package/ts_build/src/dataset/diffs/jsonl.d.ts +1 -0
  333. package/ts_build/src/dataset/diffs/jsonl.js +47 -0
  334. package/ts_build/src/dataset/diffs/jsonl.js.map +1 -0
  335. package/ts_build/src/dataset/diffs/test.d.ts +1 -0
  336. package/ts_build/src/dataset/diffs/test.js +114 -0
  337. package/ts_build/src/dataset/diffs/test.js.map +1 -0
  338. package/ts_build/src/embeddings.d.ts +19 -0
  339. package/ts_build/src/embeddings.js +322 -0
  340. package/ts_build/src/embeddings.js.map +1 -0
  341. package/ts_build/src/hashes.d.ts +6 -0
  342. package/ts_build/src/hashes.js +78 -0
  343. package/ts_build/src/hashes.js.map +1 -0
  344. package/ts_build/src/index.d.ts +14 -0
  345. package/ts_build/src/index.js +258 -0
  346. package/ts_build/src/index.js.map +1 -0
  347. package/ts_build/src/login.d.ts +2 -0
  348. package/ts_build/src/login.js +63 -0
  349. package/ts_build/src/login.js.map +1 -0
  350. package/ts_build/src/microphone.d.ts +9 -0
  351. package/ts_build/src/microphone.js +131 -0
  352. package/ts_build/src/microphone.js.map +1 -0
  353. package/ts_build/src/modules/index.d.ts +3 -0
  354. package/ts_build/src/modules/index.js +34 -0
  355. package/ts_build/src/modules/index.js.map +1 -0
  356. package/ts_build/src/modules/types.d.ts +37 -0
  357. package/ts_build/src/modules/types.js +3 -0
  358. package/ts_build/src/modules/types.js.map +1 -0
  359. package/ts_build/src/plugins/asana.d.ts +16 -0
  360. package/ts_build/src/plugins/asana.js +114 -0
  361. package/ts_build/src/plugins/asana.js.map +1 -0
  362. package/ts_build/src/plugins/downloader/downloader.d.ts +12 -0
  363. package/ts_build/src/plugins/downloader/downloader.js +174 -0
  364. package/ts_build/src/plugins/downloader/downloader.js.map +1 -0
  365. package/ts_build/src/plugins/downloader/index.d.ts +3 -0
  366. package/ts_build/src/plugins/downloader/index.js +41 -0
  367. package/ts_build/src/plugins/downloader/index.js.map +1 -0
  368. package/ts_build/src/plugins/downloader/plugin.d.ts +8 -0
  369. package/ts_build/src/plugins/downloader/plugin.js +83 -0
  370. package/ts_build/src/plugins/downloader/plugin.js.map +1 -0
  371. package/ts_build/src/plugins/downloader/types.d.ts +82 -0
  372. package/ts_build/src/plugins/downloader/types.js +79 -0
  373. package/ts_build/src/plugins/downloader/types.js.map +1 -0
  374. package/ts_build/src/plugins/embedding.d.ts +5 -0
  375. package/ts_build/src/plugins/embedding.js +28 -0
  376. package/ts_build/src/plugins/embedding.js.map +1 -0
  377. package/ts_build/src/plugins/figma.d.ts +20 -0
  378. package/ts_build/src/plugins/figma.js +94 -0
  379. package/ts_build/src/plugins/figma.js.map +1 -0
  380. package/ts_build/src/plugins/github.d.ts +1223 -0
  381. package/ts_build/src/plugins/github.js +115 -0
  382. package/ts_build/src/plugins/github.js.map +1 -0
  383. package/ts_build/src/plugins/jira.d.ts +15 -0
  384. package/ts_build/src/plugins/jira.js +94 -0
  385. package/ts_build/src/plugins/jira.js.map +1 -0
  386. package/ts_build/src/plugins/language.d.ts +8 -0
  387. package/ts_build/src/plugins/language.js +69 -0
  388. package/ts_build/src/plugins/language.js.map +1 -0
  389. package/ts_build/src/plugins/linear.d.ts +23 -0
  390. package/ts_build/src/plugins/linear.js +184 -0
  391. package/ts_build/src/plugins/linear.js.map +1 -0
  392. package/ts_build/src/plugins/notion.d.ts +32 -0
  393. package/ts_build/src/plugins/notion.js +131 -0
  394. package/ts_build/src/plugins/notion.js.map +1 -0
  395. package/ts_build/src/plugins/plugins.d.ts +11 -0
  396. package/ts_build/src/plugins/plugins.js +51 -0
  397. package/ts_build/src/plugins/plugins.js.map +1 -0
  398. package/ts_build/src/plugins/types.d.ts +5 -0
  399. package/ts_build/src/plugins/types.js +3 -0
  400. package/ts_build/src/plugins/types.js.map +1 -0
  401. package/ts_build/src/plugins/url.d.ts +8 -0
  402. package/ts_build/src/plugins/url.js +50 -0
  403. package/ts_build/src/plugins/url.js.map +1 -0
  404. package/ts_build/src/plugins/vim.d.ts +11 -0
  405. package/ts_build/src/plugins/vim.js +63 -0
  406. package/ts_build/src/plugins/vim.js.map +1 -0
  407. package/ts_build/src/prompts/BasicCodeDocumenter.d.ts +2 -0
  408. package/ts_build/src/prompts/BasicCodeDocumenter.js +4 -0
  409. package/ts_build/src/prompts/BasicCodeDocumenter.js.map +1 -0
  410. package/ts_build/src/prompts/BasicProjectDocumenter.d.ts +2 -0
  411. package/ts_build/src/prompts/BasicProjectDocumenter.js +4 -0
  412. package/ts_build/src/prompts/BasicProjectDocumenter.js.map +1 -0
  413. package/ts_build/src/prompts/EmbeddingSeachHelper.d.ts +2 -0
  414. package/ts_build/src/prompts/EmbeddingSeachHelper.js +4 -0
  415. package/ts_build/src/prompts/EmbeddingSeachHelper.js.map +1 -0
  416. package/ts_build/src/prompts/index.d.ts +13 -0
  417. package/ts_build/src/prompts/index.js +15 -0
  418. package/ts_build/src/prompts/index.js.map +1 -0
  419. package/ts_build/src/server/index.d.ts +1 -0
  420. package/ts_build/src/server/index.js +17 -0
  421. package/ts_build/src/server/index.js.map +1 -0
  422. package/ts_build/src/services/AgentService.d.ts +14 -0
  423. package/ts_build/src/services/AgentService.js +87 -0
  424. package/ts_build/src/services/AgentService.js.map +1 -0
  425. package/ts_build/src/services/EmbeddingService.d.ts +39 -0
  426. package/ts_build/src/services/EmbeddingService.js +307 -0
  427. package/ts_build/src/services/EmbeddingService.js.map +1 -0
  428. package/ts_build/src/services/EventService.d.ts +9 -0
  429. package/ts_build/src/services/EventService.js +20 -0
  430. package/ts_build/src/services/EventService.js.map +1 -0
  431. package/ts_build/src/services/GitHub.d.ts +11 -0
  432. package/ts_build/src/services/GitHub.js +52 -0
  433. package/ts_build/src/services/GitHub.js.map +1 -0
  434. package/ts_build/src/services/KnowhowClient.d.ts +18 -0
  435. package/ts_build/src/services/KnowhowClient.js +85 -0
  436. package/ts_build/src/services/KnowhowClient.js.map +1 -0
  437. package/ts_build/src/services/Mcp.d.ts +377 -0
  438. package/ts_build/src/services/Mcp.js +221 -0
  439. package/ts_build/src/services/Mcp.js.map +1 -0
  440. package/ts_build/src/services/McpServer.d.ts +19 -0
  441. package/ts_build/src/services/McpServer.js +104 -0
  442. package/ts_build/src/services/McpServer.js.map +1 -0
  443. package/ts_build/src/services/McpWebsocketTransport.d.ts +13 -0
  444. package/ts_build/src/services/McpWebsocketTransport.js +59 -0
  445. package/ts_build/src/services/McpWebsocketTransport.js.map +1 -0
  446. package/ts_build/src/services/S3.d.ts +9 -0
  447. package/ts_build/src/services/S3.js +116 -0
  448. package/ts_build/src/services/S3.js.map +1 -0
  449. package/ts_build/src/services/Tools.d.ts +19 -0
  450. package/ts_build/src/services/Tools.js +51 -0
  451. package/ts_build/src/services/Tools.js.map +1 -0
  452. package/ts_build/src/services/flags.d.ts +13 -0
  453. package/ts_build/src/services/flags.js +48 -0
  454. package/ts_build/src/services/flags.js.map +1 -0
  455. package/ts_build/src/services/index.d.ts +8 -0
  456. package/ts_build/src/services/index.js +38 -0
  457. package/ts_build/src/services/index.js.map +1 -0
  458. package/ts_build/src/terminal.d.ts +1 -0
  459. package/ts_build/src/terminal.js +35 -0
  460. package/ts_build/src/terminal.js.map +1 -0
  461. package/ts_build/src/types.d.ts +181 -0
  462. package/ts_build/src/types.js +94 -0
  463. package/ts_build/src/types.js.map +1 -0
  464. package/ts_build/src/utils/index.d.ts +24 -0
  465. package/ts_build/src/utils/index.js +116 -0
  466. package/ts_build/src/utils/index.js.map +1 -0
  467. package/ts_build/src/worker.d.ts +1 -0
  468. package/ts_build/src/worker.js +68 -0
  469. package/ts_build/src/worker.js.map +1 -0
  470. package/ts_build/tests/integration/figma.test.d.ts +1 -0
  471. package/ts_build/tests/integration/figma.test.js +47 -0
  472. package/ts_build/tests/integration/figma.test.js.map +1 -0
  473. package/ts_build/tests/integration/fileblocks/readwrite.test.d.ts +1 -0
  474. package/ts_build/tests/integration/fileblocks/readwrite.test.js +95 -0
  475. package/ts_build/tests/integration/fileblocks/readwrite.test.js.map +1 -0
  476. package/ts_build/tests/integration/patching.test.d.ts +1 -0
  477. package/ts_build/tests/integration/patching.test.js +117 -0
  478. package/ts_build/tests/integration/patching.test.js.map +1 -0
  479. package/ts_build/tests/languagePlugin.test.d.ts +1 -0
  480. package/ts_build/tests/languagePlugin.test.js +78 -0
  481. package/ts_build/tests/languagePlugin.test.js.map +1 -0
  482. package/ts_build/tests/patching/corrupted.test.d.ts +1 -0
  483. package/ts_build/tests/patching/corrupted.test.js +53 -0
  484. package/ts_build/tests/patching/corrupted.test.js.map +1 -0
  485. package/ts_build/tests/patching/imports.test.d.ts +1 -0
  486. package/ts_build/tests/patching/imports.test.js +61 -0
  487. package/ts_build/tests/patching/imports.test.js.map +1 -0
  488. package/ts_build/tests/test.spec.d.ts +1 -0
  489. package/ts_build/tests/test.spec.js +101 -0
  490. package/ts_build/tests/test.spec.js.map +1 -0
  491. package/tsconfig.json +31 -0
  492. package/tslint.json +11 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/clients/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAOA,qCAA+C;AAC/C,2CAAqD;AACrD,qCAA+C;AAC/C,iCAAoC;AACpC,oCAAkC;AAClC,sCAAsC;AACtC,+BAAyC;AAEzC,SAAS,QAAQ,CAAC,GAAW;IAC3B,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,KAAK,CAAC;KACd;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAa,QAAQ;IACnB,OAAO,GAAG;QACR,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,4BAAmB,EAAE,EAAE,CAAC;QAEpE,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI;YACnC,SAAS,EAAE,IAAI,kCAAsB,EAAE;SACxC,CAAC;QAEF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,4BAAmB,EAAE,EAAE,CAAC;QACxE,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,IAAI,sBAAgB,EAAE,EAAE,CAAC;KAChE,CAAC;IAEF,YAAY,GAAG;QACb,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QACvE,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI;YACnC,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,SAAS,CAAC;SAC3C,CAAC;QACF,GAAG,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3E,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,cAAM,CAAC,GAAG,CAAC,EAAE,CAAC;KACnE,CAAC;IAEF,SAAS,CAAC,QAAgB,EAAE,KAAc;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;YAC1B,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;SAC5D;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE3D,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;QAC7B,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAEvB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;YAC3B,MAAM,IAAI,KAAK,CACb,YAAY,QAAQ,cAAc,KAAK,yCAAyC,MAAM,CAAC,IAAI,CACzF,IAAI,CAAC,OAAO,CACb,EAAE,CACJ,CAAC;SACH;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAExD,IAAI,CAAC,QAAQ,EAAE;YACb,MAAM,IAAI,KAAK,CACb,SAAS,KAAK,gCAAgC,QAAQ,GAAG,CAC1D,CAAC;SACH;QAED,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC7D,CAAC;IAED,cAAc,CAAC,QAAgB,EAAE,MAAqB;QACpD,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,MAAM,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,wBAAwB;QAC5B,MAAM,MAAM,GAAG,MAAM,IAAA,kBAAS,GAAE,CAAC;QACjC,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC;QAEnD,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE;YAC1C,MAAM,MAAM,GAAG,IAAI,iBAAU,CAAC,aAAa,CAAC,GAAG,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;YAExE,IAAI,aAAa,CAAC,OAAO,EAAE;gBACzB,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;aAC3C;YAED,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAEpD,IAAI;gBACF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;gBACxC,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAC5C,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;aAClD;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CACX,0CAA0C,aAAa,CAAC,QAAQ,GAAG,EACnE,KAAK,CAAC,OAAO,CACd,CAAC;aACH;SACF;IACH,CAAC;IAED,cAAc,CAAC,QAAgB,EAAE,MAAgB;QAC/C,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxD,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,GAAG,KAAK,CAAC,IAAI,CACtC,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CACtC,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,QAAgB,EAAE,KAAa;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC1B,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAEO,mBAAmB,CAAC,QAAgB,EAAE,KAAa;QACzD,IAAI,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE;YAC1C,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;SAC5B;QAED,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE;YACvB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAE/B,MAAM,gBAAgB,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE/C,IAAI,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,EAAE,aAAa,CAAC,EAAE;gBAC1D,OAAO,EAAE,QAAQ,EAAE,gBAAgB,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;aAC7D;SACF;QAED,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACjD,MAAM,aAAa,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACzC,IAAI,CAAC,gBAAgB,CAAC,CAAC,EAAE,KAAK,CAAC,CAChC,CAAC;QAEF,IAAI,aAAa,EAAE;YACjB,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;SAC3C;QAED,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,QAAgB,EAChB,OAA0B;QAE1B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QAClE,OAAO,MAAM,CAAC,oBAAoB,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,QAAgB,EAChB,OAAyB;QAEzB,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QAClE,OAAO,MAAM,CAAC,eAAe,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,mBAAmB,CAAC,QAAgB;QAClC,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC3C,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,gBAAgB;QACd,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACxC,CAAC;CACF;AApJD,4BAoJC;AAEY,QAAA,OAAO,GAAG,IAAI,QAAQ,EAAE,CAAC;AAEtC,0CAAwB;AAExB,yCAAuB;AACvB,2CAAyB;AACzB,8CAA4B;AAC5B,4CAA0B;AAC1B,2CAAyB;AACzB,wCAAsB"}
@@ -0,0 +1,10 @@
1
+ import { GenericClient, CompletionOptions, CompletionResponse, EmbeddingOptions, EmbeddingResponse } from "./types";
2
+ export declare class KnowhowGenericClient implements GenericClient {
3
+ private client;
4
+ constructor(baseUrl: string, jwt?: string);
5
+ createChatCompletion(options: CompletionOptions): Promise<CompletionResponse>;
6
+ createEmbedding(options: EmbeddingOptions): Promise<EmbeddingResponse>;
7
+ getModels(): Promise<{
8
+ id: string;
9
+ }[]>;
10
+ }
@@ -0,0 +1,24 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.KnowhowGenericClient = void 0;
4
+ const KnowhowClient_1 = require("../services/KnowhowClient");
5
+ class KnowhowGenericClient {
6
+ client;
7
+ constructor(baseUrl, jwt) {
8
+ this.client = new KnowhowClient_1.KnowhowSimpleClient(baseUrl, jwt);
9
+ }
10
+ async createChatCompletion(options) {
11
+ const response = await this.client.createChatCompletion(options);
12
+ return response.data;
13
+ }
14
+ async createEmbedding(options) {
15
+ const response = await this.client.createEmbedding(options);
16
+ return response.data;
17
+ }
18
+ async getModels() {
19
+ const response = await this.client.getModels();
20
+ return response.data;
21
+ }
22
+ }
23
+ exports.KnowhowGenericClient = KnowhowGenericClient;
24
+ //# sourceMappingURL=knowhow.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"knowhow.js","sourceRoot":"","sources":["../../../src/clients/knowhow.ts"],"names":[],"mappings":";;;AAOA,6DAAgE;AAEhE,MAAa,oBAAoB;IACvB,MAAM,CAAsB;IAEpC,YAAY,OAAe,EAAE,GAAY;QACvC,IAAI,CAAC,MAAM,GAAG,IAAI,mCAAmB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,OAA0B;QAE1B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACjE,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAyB;QAC7C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAC5D,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QAC/C,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;CACF;AAvBD,oDAuBC"}
@@ -0,0 +1,20 @@
1
+ import OpenAI from "openai";
2
+ import { GenericClient, CompletionOptions, CompletionResponse, EmbeddingOptions, EmbeddingResponse } from "./types";
3
+ export declare class GenericOpenAiClient extends OpenAI implements GenericClient {
4
+ constructor();
5
+ createChatCompletion(options: CompletionOptions): Promise<CompletionResponse>;
6
+ pricesPerMillion(): {
7
+ [x: string]: {
8
+ input: number;
9
+ cached_input: number;
10
+ output: number;
11
+ };
12
+ };
13
+ calculateCost(model: string, usage: OpenAI.ChatCompletion["usage"] | OpenAI.CreateEmbeddingResponse["usage"]): number | undefined;
14
+ getModels(): Promise<{
15
+ id: string;
16
+ object: "model";
17
+ owned_by: string;
18
+ }[]>;
19
+ createEmbedding(options: EmbeddingOptions): Promise<EmbeddingResponse>;
20
+ }
@@ -0,0 +1,221 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.GenericOpenAiClient = void 0;
7
+ const openai_1 = __importDefault(require("openai"));
8
+ const config_1 = require("../config");
9
+ const types_1 = require("../types");
10
+ const config = (0, config_1.getConfigSync)();
11
+ class GenericOpenAiClient extends openai_1.default {
12
+ constructor() {
13
+ super({
14
+ apiKey: process.env.OPENAI_KEY,
15
+ ...(config.openaiBaseUrl && { baseURL: config.openaiBaseUrl }),
16
+ });
17
+ }
18
+ async createChatCompletion(options) {
19
+ const openaiMessages = options.messages.map((msg) => {
20
+ if (msg.role === "tool") {
21
+ return {
22
+ ...msg,
23
+ content: msg.content || "",
24
+ role: "tool",
25
+ tool_call_id: msg.tool_call_id,
26
+ };
27
+ }
28
+ return msg;
29
+ });
30
+ const response = await this.chat.completions.create({
31
+ model: options.model,
32
+ messages: openaiMessages,
33
+ max_tokens: options.max_tokens,
34
+ ...(types_1.OpenAiReasoningModels.includes(options.model) && {
35
+ max_tokens: undefined,
36
+ max_completion_tokens: options.max_tokens,
37
+ }),
38
+ ...(options.tools && {
39
+ tools: options.tools,
40
+ tool_choice: "auto",
41
+ }),
42
+ });
43
+ console.log(JSON.stringify({ response }, null, 2));
44
+ const usdCost = this.calculateCost(options.model, response.usage);
45
+ return {
46
+ choices: response.choices.map((choice) => ({
47
+ message: {
48
+ role: choice.message?.role || "assistant",
49
+ content: choice.message?.content || null,
50
+ tool_calls: choice.message?.tool_calls
51
+ ? choice.message.tool_calls
52
+ : undefined,
53
+ },
54
+ })),
55
+ model: options.model,
56
+ usage: response.usage,
57
+ usd_cost: usdCost,
58
+ };
59
+ }
60
+ pricesPerMillion() {
61
+ return {
62
+ [types_1.Models.openai.GPT_4o]: {
63
+ input: 2.5,
64
+ cached_input: 1.25,
65
+ output: 10.0,
66
+ },
67
+ [types_1.Models.openai.GPT_4o_Mini]: {
68
+ input: 0.15,
69
+ cached_input: 0.075,
70
+ output: 0.6,
71
+ },
72
+ [types_1.Models.openai.o1]: {
73
+ input: 15.0,
74
+ cached_input: 7.5,
75
+ output: 60.0,
76
+ },
77
+ [types_1.Models.openai.o1_Mini]: {
78
+ input: 1.1,
79
+ cached_input: 0.55,
80
+ output: 4.4,
81
+ },
82
+ [types_1.Models.openai.o3_Mini]: {
83
+ input: 1.1,
84
+ cached_input: 0.55,
85
+ output: 4.4,
86
+ },
87
+ [types_1.Models.openai.GPT_41]: {
88
+ input: 2.0,
89
+ cached_input: 0.5,
90
+ output: 8.0,
91
+ },
92
+ [types_1.Models.openai.GPT_41_Mini]: {
93
+ input: 0.4,
94
+ cached_input: 0.1,
95
+ output: 1.6,
96
+ },
97
+ [types_1.Models.openai.GPT_41_Nano]: {
98
+ input: 0.1,
99
+ cached_input: 0.025,
100
+ output: 0.4,
101
+ },
102
+ [types_1.Models.openai.GPT_45]: {
103
+ input: 75.0,
104
+ cached_input: 37.5,
105
+ output: 150.0,
106
+ },
107
+ [types_1.Models.openai.GPT_4o_Audio]: {
108
+ input: 2.5,
109
+ cached_input: 0,
110
+ output: 10.0,
111
+ },
112
+ [types_1.Models.openai.GPT_4o_Realtime]: {
113
+ input: 5.0,
114
+ cached_input: 2.5,
115
+ output: 20.0,
116
+ },
117
+ [types_1.Models.openai.GPT_4o_Mini_Audio]: {
118
+ input: 0.15,
119
+ cached_input: 0,
120
+ output: 0.6,
121
+ },
122
+ [types_1.Models.openai.GPT_4o_Mini_Realtime]: {
123
+ input: 0.6,
124
+ cached_input: 0.3,
125
+ output: 2.4,
126
+ },
127
+ [types_1.Models.openai.o1_Pro]: {
128
+ input: 150.0,
129
+ cached_input: 0,
130
+ output: 600.0,
131
+ },
132
+ [types_1.Models.openai.o3]: {
133
+ input: 10.0,
134
+ cached_input: 2.5,
135
+ output: 40.0,
136
+ },
137
+ [types_1.Models.openai.o4_Mini]: {
138
+ input: 1.1,
139
+ cached_input: 0.275,
140
+ output: 4.4,
141
+ },
142
+ [types_1.Models.openai.GPT_4o_Mini_Search]: {
143
+ input: 0.15,
144
+ cached_input: 0,
145
+ output: 0.6,
146
+ },
147
+ [types_1.Models.openai.GPT_4o_Search]: {
148
+ input: 2.5,
149
+ cached_input: 0,
150
+ output: 10.0,
151
+ },
152
+ [types_1.Models.openai.Computer_Use]: {
153
+ input: 3.0,
154
+ cached_input: 0,
155
+ output: 12.0,
156
+ },
157
+ [types_1.Models.openai.Codex_Mini]: {
158
+ input: 1.5,
159
+ cached_input: 0.375,
160
+ output: 6.0,
161
+ },
162
+ [types_1.Models.openai.EmbeddingAda2]: {
163
+ input: 0.1,
164
+ cached_input: 0,
165
+ output: 0,
166
+ },
167
+ [types_1.Models.openai.EmbeddingLarge3]: {
168
+ input: 0.13,
169
+ cached_input: 0,
170
+ output: 0,
171
+ },
172
+ [types_1.Models.openai.EmbeddingLarge3]: {
173
+ input: 0.02,
174
+ cached_input: 0,
175
+ output: 0,
176
+ },
177
+ };
178
+ }
179
+ calculateCost(model, usage) {
180
+ const pricing = this.pricesPerMillion()[model];
181
+ console.log({ pricing });
182
+ if (!pricing) {
183
+ return undefined;
184
+ }
185
+ const cachedInputTokens = ("prompt_tokens_details" in usage &&
186
+ usage.prompt_tokens_details?.cached_tokens) ||
187
+ 0;
188
+ const cachedInputCost = (cachedInputTokens * pricing.cached_input) / 1e6;
189
+ const inputTokens = usage.prompt_tokens;
190
+ const inputCost = ((inputTokens - cachedInputCost) * pricing.input) / 1e6;
191
+ const outputTokens = ("completion_tokens" in usage && usage?.completion_tokens) || 0;
192
+ const outputCost = (outputTokens * pricing.output) / 1e6;
193
+ const total = cachedInputCost + inputCost + outputCost;
194
+ console.log({ total });
195
+ return total;
196
+ }
197
+ async getModels() {
198
+ const models = await this.models.list();
199
+ return models.data.map((m) => {
200
+ return {
201
+ id: m.id,
202
+ object: m.object,
203
+ owned_by: m.owned_by,
204
+ };
205
+ });
206
+ }
207
+ async createEmbedding(options) {
208
+ const openAiEmbedding = await this.embeddings.create({
209
+ input: options.input,
210
+ model: options.model,
211
+ });
212
+ return {
213
+ data: openAiEmbedding.data,
214
+ model: options.model,
215
+ usage: openAiEmbedding.usage,
216
+ usd_cost: this.calculateCost(options.model, openAiEmbedding.usage),
217
+ };
218
+ }
219
+ }
220
+ exports.GenericOpenAiClient = GenericOpenAiClient;
221
+ //# sourceMappingURL=openai.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/clients/openai.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAC5B,sCAA0C;AAc1C,oCAAyD;AAEzD,MAAM,MAAM,GAAG,IAAA,sBAAa,GAAE,CAAC;AAE/B,MAAa,mBAAoB,SAAQ,gBAAM;IAC7C;QACE,KAAK,CAAC;YACJ,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU;YAC9B,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,OAA0B;QAE1B,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAClD,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE;gBACvB,OAAO;oBACL,GAAG,GAAG;oBACN,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;oBAC1B,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,GAAG,CAAC,YAAY;iBACG,CAAC;aACrC;YACD,OAAO,GAAiC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YAClD,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,cAAc;YACxB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,GAAG,CAAC,6BAAqB,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI;gBACnD,UAAU,EAAE,SAAS;gBACrB,qBAAqB,EAAE,OAAO,CAAC,UAAU;aAC1C,CAAC;YAEF,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,WAAW,EAAE,MAAM;aACpB,CAAC;SACH,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;QAElE,OAAO;YACL,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACzC,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,WAAW;oBACzC,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,IAAI,IAAI;oBACxC,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,UAAU;wBACpC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU;wBAC3B,CAAC,CAAC,SAAS;iBACd;aACF,CAAC,CAAC;YAEH,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,QAAQ,EAAE,OAAO;SAClB,CAAC;IACJ,CAAC;IAED,gBAAgB;QACd,OAAO;YACL,CAAC,cAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE;gBACtB,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE;gBAC3B,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE,KAAK;gBACnB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE;gBAClB,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE,GAAG;gBACjB,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;gBACvB,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;gBACvB,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE;gBACtB,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,GAAG;gBACjB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE;gBAC3B,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,GAAG;gBACjB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE;gBAC3B,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,KAAK;gBACnB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE;gBACtB,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE,IAAI;gBAClB,MAAM,EAAE,KAAK;aACd;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE;gBAC5B,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE;gBAC/B,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,GAAG;gBACjB,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE;gBACjC,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,EAAE;gBACpC,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,GAAG;gBACjB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE;gBACtB,KAAK,EAAE,KAAK;gBACZ,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,KAAK;aACd;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE;gBAClB,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE,GAAG;gBACjB,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;gBACvB,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,KAAK;gBACnB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE;gBAClC,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE;gBAC7B,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE;gBAC5B,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE;gBAC1B,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,KAAK;gBACnB,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE;gBAC7B,KAAK,EAAE,GAAG;gBACV,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,CAAC;aACV;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE;gBAC/B,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,CAAC;aACV;YACD,CAAC,cAAM,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE;gBAC/B,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE,CAAC;gBACf,MAAM,EAAE,CAAC;aACV;SACF,CAAC;IACJ,CAAC;IAED,aAAa,CACX,KAAa,EACb,KAE2C;QAE3C,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC,KAAK,CAAC,CAAC;QAE/C,OAAO,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QACzB,IAAI,CAAC,OAAO,EAAE;YACZ,OAAO,SAAS,CAAC;SAClB;QAED,MAAM,iBAAiB,GACrB,CAAC,uBAAuB,IAAI,KAAK;YAC/B,KAAK,CAAC,qBAAqB,EAAE,aAAa,CAAC;YAC7C,CAAC,CAAC;QACJ,MAAM,eAAe,GAAG,CAAC,iBAAiB,GAAG,OAAO,CAAC,YAAY,CAAC,GAAG,GAAG,CAAC;QAEzE,MAAM,WAAW,GAAG,KAAK,CAAC,aAAa,CAAC;QACxC,MAAM,SAAS,GAAG,CAAC,CAAC,WAAW,GAAG,eAAe,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC;QAE1E,MAAM,YAAY,GAChB,CAAC,mBAAmB,IAAI,KAAK,IAAI,KAAK,EAAE,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;QAEzD,MAAM,KAAK,GAAG,eAAe,GAAG,SAAS,GAAG,UAAU,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,SAAS;QACb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACxC,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YAC3B,OAAO;gBACL,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,MAAM,EAAE,CAAC,CAAC,MAAM;gBAChB,QAAQ,EAAE,CAAC,CAAC,QAAQ;aACrB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAyB;QAC7C,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;YACnD,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,EAAE,eAAe,CAAC,IAAI;YAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,KAAK,EAAE,eAAe,CAAC,KAAK;YAC5B,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,eAAe,CAAC,KAAK,CAAC;SACnE,CAAC;IACJ,CAAC;CACF;AAzOD,kDAyOC"}
@@ -0,0 +1,89 @@
1
+ export type MessageContent = {
2
+ type: "text";
3
+ text: string;
4
+ } | {
5
+ type: "image_url";
6
+ image_url: {
7
+ url: string;
8
+ };
9
+ };
10
+ export interface Message {
11
+ role: "system" | "user" | "assistant" | "tool";
12
+ content: string | MessageContent[];
13
+ name?: string;
14
+ tool_call_id?: string;
15
+ tool_calls?: ToolCall[];
16
+ }
17
+ export interface OutputMessage extends Message {
18
+ content: string;
19
+ }
20
+ export interface ToolProp {
21
+ type: string;
22
+ description: string;
23
+ properties?: {
24
+ [key: string]: ToolProp;
25
+ };
26
+ items?: ToolProp;
27
+ }
28
+ export interface Tool {
29
+ type: "function";
30
+ function: {
31
+ name: string;
32
+ description: string;
33
+ parameters: {
34
+ type: string;
35
+ positional?: boolean;
36
+ properties: {
37
+ [key: string]: ToolProp;
38
+ };
39
+ required: string[];
40
+ };
41
+ };
42
+ }
43
+ export interface ToolCall {
44
+ id: string;
45
+ type: "function";
46
+ function: {
47
+ name: string;
48
+ arguments: string;
49
+ };
50
+ }
51
+ export interface CompletionOptions {
52
+ model: string;
53
+ messages: Message[];
54
+ tools?: Tool[];
55
+ tool_choice?: "auto" | "none";
56
+ max_tokens?: number;
57
+ }
58
+ export interface CompletionResponse {
59
+ choices: {
60
+ message: OutputMessage;
61
+ }[];
62
+ model: string;
63
+ usage: any;
64
+ usd_cost?: number;
65
+ }
66
+ export interface EmbeddingOptions {
67
+ input: string;
68
+ model?: string;
69
+ }
70
+ export interface EmbeddingResponse {
71
+ data: {
72
+ object: string;
73
+ embedding: number[];
74
+ index: number;
75
+ }[];
76
+ model: string;
77
+ usage: {
78
+ prompt_tokens: number;
79
+ total_tokens: number;
80
+ };
81
+ usd_cost?: number;
82
+ }
83
+ export interface GenericClient {
84
+ createChatCompletion(options: CompletionOptions): Promise<CompletionResponse>;
85
+ createEmbedding(options: EmbeddingOptions): Promise<EmbeddingResponse>;
86
+ getModels(): Promise<{
87
+ id: string;
88
+ }[]>;
89
+ }
@@ -0,0 +1,3 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/clients/types.ts"],"names":[],"mappings":""}
@@ -0,0 +1,22 @@
1
+ import OpenAI from "openai";
2
+ import { GenericClient, CompletionOptions, CompletionResponse, EmbeddingOptions, EmbeddingResponse } from "./types";
3
+ export declare class GenericXAIClient extends OpenAI implements GenericClient {
4
+ constructor();
5
+ createChatCompletion(options: CompletionOptions): Promise<CompletionResponse>;
6
+ pricesPerMillion(): {
7
+ [x: string]: {
8
+ input: number;
9
+ output: number;
10
+ image_input?: undefined;
11
+ } | {
12
+ input: number;
13
+ output: number;
14
+ image_input: number;
15
+ };
16
+ };
17
+ calculateCost(model: string, usage: OpenAI.ChatCompletion["usage"]): number | undefined;
18
+ getModels(): Promise<{
19
+ id: any;
20
+ }[]>;
21
+ createEmbedding(options: EmbeddingOptions): Promise<EmbeddingResponse>;
22
+ }
@@ -0,0 +1,108 @@
1
+ "use strict";
2
+ var __importDefault = (this && this.__importDefault) || function (mod) {
3
+ return (mod && mod.__esModule) ? mod : { "default": mod };
4
+ };
5
+ Object.defineProperty(exports, "__esModule", { value: true });
6
+ exports.GenericXAIClient = void 0;
7
+ const openai_1 = __importDefault(require("openai"));
8
+ const config_1 = require("../config");
9
+ const types_1 = require("../types");
10
+ const config = (0, config_1.getConfigSync)();
11
+ class GenericXAIClient extends openai_1.default {
12
+ constructor() {
13
+ super({
14
+ apiKey: process.env.XAI_API_KEY,
15
+ baseURL: "https://api.x.ai/v1",
16
+ });
17
+ }
18
+ async createChatCompletion(options) {
19
+ const xaiMessages = options.messages.map((msg) => {
20
+ if (msg.role === "tool") {
21
+ return {
22
+ ...msg,
23
+ content: msg.content || "",
24
+ role: "tool",
25
+ tool_call_id: msg.tool_call_id,
26
+ };
27
+ }
28
+ return msg;
29
+ });
30
+ const response = await this.chat.completions.create({
31
+ model: options.model,
32
+ messages: xaiMessages,
33
+ max_tokens: options.max_tokens,
34
+ ...(options.tools && {
35
+ tools: options.tools,
36
+ tool_choice: "auto",
37
+ }),
38
+ });
39
+ console.log(JSON.stringify({ response }, null, 2));
40
+ const usdCost = this.calculateCost(options.model, response.usage);
41
+ return {
42
+ choices: response.choices.map((choice) => ({
43
+ message: {
44
+ role: choice.message?.role || "assistant",
45
+ content: choice.message?.content || null,
46
+ tool_calls: choice.message?.tool_calls
47
+ ? choice.message.tool_calls
48
+ : undefined,
49
+ },
50
+ })),
51
+ model: options.model,
52
+ usage: response.usage,
53
+ usd_cost: usdCost,
54
+ };
55
+ }
56
+ pricesPerMillion() {
57
+ return {
58
+ [types_1.Models.xai.Grok3Beta]: {
59
+ input: 3.0,
60
+ output: 15.0,
61
+ },
62
+ [types_1.Models.xai.Grok3MiniBeta]: {
63
+ input: 0.3,
64
+ output: 0.5,
65
+ },
66
+ [types_1.Models.xai.Grok3FastBeta]: {
67
+ input: 5.0,
68
+ output: 25.0,
69
+ },
70
+ [types_1.Models.xai.Grok3MiniFastBeta]: {
71
+ input: 0.6,
72
+ output: 4.0,
73
+ },
74
+ [types_1.Models.xai.Grok21212]: {
75
+ input: 2.0,
76
+ output: 10.0,
77
+ },
78
+ [types_1.Models.xai.Grok2Vision1212]: {
79
+ input: 2.0,
80
+ output: 10.0,
81
+ image_input: 2.0,
82
+ },
83
+ };
84
+ }
85
+ calculateCost(model, usage) {
86
+ const pricing = this.pricesPerMillion()[model];
87
+ if (!pricing) {
88
+ return undefined;
89
+ }
90
+ const inputTokens = usage.prompt_tokens;
91
+ const inputCost = (inputTokens * pricing.input) / 1e6;
92
+ const outputTokens = usage.completion_tokens || 0;
93
+ const outputCost = (outputTokens * pricing.output) / 1e6;
94
+ const total = inputCost + outputCost;
95
+ console.log({ total });
96
+ return total;
97
+ }
98
+ async getModels() {
99
+ return Object.keys(types_1.Models.xai).map((key) => ({
100
+ id: types_1.Models.xai[key],
101
+ }));
102
+ }
103
+ async createEmbedding(options) {
104
+ throw new Error("XAI provider does not support embeddings");
105
+ }
106
+ }
107
+ exports.GenericXAIClient = GenericXAIClient;
108
+ //# sourceMappingURL=xai.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"xai.js","sourceRoot":"","sources":["../../../src/clients/xai.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAC5B,sCAA0C;AAa1C,oCAAkC;AAElC,MAAM,MAAM,GAAG,IAAA,sBAAa,GAAE,CAAC;AAE/B,MAAa,gBAAiB,SAAQ,gBAAM;IAC1C;QACE,KAAK,CAAC;YACJ,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW;YAC/B,OAAO,EAAE,qBAAqB;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,oBAAoB,CACxB,OAA0B;QAE1B,MAAM,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/C,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE;gBACvB,OAAO;oBACL,GAAG,GAAG;oBACN,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,EAAE;oBAC1B,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,GAAG,CAAC,YAAY;iBACG,CAAC;aACrC;YACD,OAAO,GAAiC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YAClD,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,WAAW;YACrB,UAAU,EAAE,OAAO,CAAC,UAAU;YAC9B,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,WAAW,EAAE,MAAM;aACpB,CAAC;SACH,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;QAClE,OAAO;YACL,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACzC,OAAO,EAAE;oBACP,IAAI,EAAE,MAAM,CAAC,OAAO,EAAE,IAAI,IAAI,WAAW;oBACzC,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,IAAI,IAAI;oBACxC,UAAU,EAAE,MAAM,CAAC,OAAO,EAAE,UAAU;wBACpC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU;wBAC3B,CAAC,CAAC,SAAS;iBACd;aACF,CAAC,CAAC;YAEH,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,QAAQ,EAAE,OAAO;SAClB,CAAC;IACJ,CAAC;IAED,gBAAgB;QACd,OAAO;YACL,CAAC,cAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;gBACtB,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE;gBAC1B,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE;gBAC1B,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,EAAE;gBAC9B,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,GAAG;aACZ;YACD,CAAC,cAAM,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;gBACtB,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,IAAI;aACb;YACD,CAAC,cAAM,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE;gBAC5B,KAAK,EAAE,GAAG;gBACV,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,GAAG;aACjB;SACF,CAAC;IACJ,CAAC;IAED,aAAa,CACX,KAAa,EACb,KAAqC;QAErC,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC,KAAK,CAAC,CAAC;QAE/C,IAAI,CAAC,OAAO,EAAE;YACZ,OAAO,SAAS,CAAC;SAClB;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,aAAa,CAAC;QACxC,MAAM,SAAS,GAAG,CAAC,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC;QAEtD,MAAM,YAAY,GAAG,KAAK,CAAC,iBAAiB,IAAI,CAAC,CAAC;QAClD,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;QAEzD,MAAM,KAAK,GAAG,SAAS,GAAG,UAAU,CAAC;QACrC,OAAO,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,SAAS;QAEb,OAAO,MAAM,CAAC,IAAI,CAAC,cAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC3C,EAAE,EAAE,cAAM,CAAC,GAAG,CAAC,GAAG,CAAC;SACpB,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,OAAyB;QAC7C,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC;CACF;AAjHD,4CAiHC"}
@@ -0,0 +1,9 @@
1
+ import { Config, Language } from "./types";
2
+ export declare function init(): Promise<void>;
3
+ export declare function getLanguageConfig(): Promise<Language>;
4
+ export declare function updateLanguageConfig(language: Language): Promise<void>;
5
+ export declare function updateConfig(config: Config): Promise<void>;
6
+ export declare function getConfigSync(): Config;
7
+ export declare function getConfig(): Promise<Config>;
8
+ export declare function loadPrompt(promptName: string): Promise<string>;
9
+ export declare function getIgnorePattern(): Promise<string[]>;