@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.
- package/.vscode/launch.json +14 -0
- package/CONFIG.md +336 -0
- package/README.md +49 -0
- package/autodoc/chat.mdx +20 -0
- package/autodoc/cli.mdx +11 -0
- package/autodoc/plugins/asana.mdx +47 -0
- package/autodoc/plugins/downloader/downloader.mdx +38 -0
- package/autodoc/plugins/downloader/plugin.mdx +37 -0
- package/autodoc/plugins/downloader/types.mdx +42 -0
- package/autodoc/plugins/embedding.mdx +41 -0
- package/autodoc/plugins/figma.mdx +45 -0
- package/autodoc/plugins/github.mdx +40 -0
- package/autodoc/plugins/jira.mdx +46 -0
- package/autodoc/plugins/language.mdx +37 -0
- package/autodoc/plugins/linear.mdx +35 -0
- package/autodoc/plugins/notion.mdx +38 -0
- package/autodoc/plugins/plugins.mdx +59 -0
- package/autodoc/plugins/types.mdx +51 -0
- package/autodoc/plugins/vim.mdx +39 -0
- package/autodoc/tools/addInternalTools.mdx +1 -0
- package/autodoc/tools/agentCall.mdx +1 -0
- package/autodoc/tools/asana/definitions.mdx +10 -0
- package/autodoc/tools/asana/index.mdx +12 -0
- package/autodoc/tools/askHuman.mdx +1 -0
- package/autodoc/tools/callPlugin.mdx +1 -0
- package/autodoc/tools/embeddingSearch.mdx +1 -0
- package/autodoc/tools/execCommand.mdx +1 -0
- package/autodoc/tools/fileSearch.mdx +1 -0
- package/autodoc/tools/finalAnswer.mdx +1 -0
- package/autodoc/tools/github/definitions.mdx +6 -0
- package/autodoc/tools/github/index.mdx +8 -0
- package/autodoc/tools/index.mdx +14 -0
- package/autodoc/tools/lintFile.mdx +7 -0
- package/autodoc/tools/list.mdx +16 -0
- package/autodoc/tools/modifyFile.mdx +7 -0
- package/autodoc/tools/patch.mdx +9 -0
- package/autodoc/tools/readBlocks.mdx +1 -0
- package/autodoc/tools/readFile.mdx +1 -0
- package/autodoc/tools/scanFile.mdx +1 -0
- package/autodoc/tools/textSearch.mdx +6 -0
- package/autodoc/tools/types/fileblock.mdx +1 -0
- package/autodoc/tools/visionTool.mdx +1 -0
- package/autodoc/tools/writeFile.mdx +1 -0
- package/jest.config.js +18 -0
- package/package.json +89 -0
- package/src/agents/base/base.ts +619 -0
- package/src/agents/base/prompt.ts +26 -0
- package/src/agents/configurable/ConfigAgent.ts +23 -0
- package/src/agents/developer/developer.ts +69 -0
- package/src/agents/index.ts +8 -0
- package/src/agents/interface.ts +11 -0
- package/src/agents/patcher/codebase.md +27 -0
- package/src/agents/patcher/patcher.ts +110 -0
- package/src/agents/researcher/researcher.ts +109 -0
- package/src/agents/tools/addInternalTools.ts +21 -0
- package/src/agents/tools/agentCall.ts +18 -0
- package/src/agents/tools/aiClient.ts +36 -0
- package/src/agents/tools/asana/definitions.ts +199 -0
- package/src/agents/tools/asana/index.ts +108 -0
- package/src/agents/tools/askHuman.ts +8 -0
- package/src/agents/tools/callPlugin.ts +4 -0
- package/src/agents/tools/embeddingSearch.ts +5 -0
- package/src/agents/tools/execCommand.ts +26 -0
- package/src/agents/tools/fileSearch.ts +33 -0
- package/src/agents/tools/finalAnswer.ts +4 -0
- package/src/agents/tools/github/definitions.ts +89 -0
- package/src/agents/tools/github/index.ts +67 -0
- package/src/agents/tools/googleSearch.ts +242 -0
- package/src/agents/tools/index.ts +24 -0
- package/src/agents/tools/language/definitions.ts +97 -0
- package/src/agents/tools/language/index.ts +19 -0
- package/src/agents/tools/lintFile.ts +17 -0
- package/src/agents/tools/list.ts +514 -0
- package/src/agents/tools/loadWebpage.ts +129 -0
- package/src/agents/tools/modifyFile.ts +53 -0
- package/src/agents/tools/patch.ts +672 -0
- package/src/agents/tools/readBlocks.ts +41 -0
- package/src/agents/tools/readFile.ts +39 -0
- package/src/agents/tools/scanFile.ts +16 -0
- package/src/agents/tools/textSearch.ts +21 -0
- package/src/agents/tools/types/fileblock.ts +5 -0
- package/src/agents/tools/visionTool.ts +5 -0
- package/src/agents/tools/writeFile.ts +46 -0
- package/src/agents/vim/vim.ts +153 -0
- package/src/ai.ts +167 -0
- package/src/chat.ts +412 -0
- package/src/cli.ts +85 -0
- package/src/clients/anthropic.ts +399 -0
- package/src/clients/gemini.ts +486 -0
- package/src/clients/http.ts +107 -0
- package/src/clients/index.ts +183 -0
- package/src/clients/knowhow.ts +33 -0
- package/src/clients/openai.ts +253 -0
- package/src/clients/types.ts +91 -0
- package/src/clients/xai.ts +132 -0
- package/src/config.ts +211 -0
- package/src/conversion.ts +140 -0
- package/src/dataset/diffs/README.md +12 -0
- package/src/dataset/diffs/debug-errors.ts +29 -0
- package/src/dataset/diffs/debug.ts +64 -0
- package/src/dataset/diffs/generate.ts +71 -0
- package/src/dataset/diffs/jsonl.ts +54 -0
- package/src/dataset/diffs/test.ts +193 -0
- package/src/embeddings.ts +411 -0
- package/src/hashes.ts +67 -0
- package/src/index.ts +376 -0
- package/src/login.ts +78 -0
- package/src/microphone.ts +136 -0
- package/src/modules/index.ts +37 -0
- package/src/modules/types.ts +48 -0
- package/src/plugins/asana.ts +134 -0
- package/src/plugins/downloader/downloader.ts +204 -0
- package/src/plugins/downloader/plugin.ts +85 -0
- package/src/plugins/downloader/types.ts +85 -0
- package/src/plugins/embedding.ts +42 -0
- package/src/plugins/figma.ts +137 -0
- package/src/plugins/github.ts +153 -0
- package/src/plugins/jira.ts +104 -0
- package/src/plugins/language.ts +100 -0
- package/src/plugins/linear.ts +219 -0
- package/src/plugins/notion.ts +170 -0
- package/src/plugins/plugins.ts +56 -0
- package/src/plugins/types.ts +5 -0
- package/src/plugins/url.ts +63 -0
- package/src/plugins/vim.ts +73 -0
- package/src/prompts/BasicCodeDocumenter.ts +1 -0
- package/src/prompts/BasicProjectDocumenter.ts +1 -0
- package/src/prompts/EmbeddingSeachHelper.ts +1 -0
- package/src/prompts/index.ts +9 -0
- package/src/server/index.ts +14 -0
- package/src/services/AgentService.ts +98 -0
- package/src/services/EmbeddingService.ts +410 -0
- package/src/services/EventService.ts +20 -0
- package/src/services/GitHub.ts +60 -0
- package/src/services/KnowhowClient.ts +113 -0
- package/src/services/Mcp.ts +270 -0
- package/src/services/McpServer.ts +119 -0
- package/src/services/McpWebsocketTransport.ts +64 -0
- package/src/services/S3.ts +126 -0
- package/src/services/Tools.ts +65 -0
- package/src/services/flags.ts +52 -0
- package/src/services/index.ts +8 -0
- package/src/types.ts +225 -0
- package/src/utils/index.ts +108 -0
- package/src/worker.ts +80 -0
- package/tests/integration/figma.test.ts +70 -0
- package/tests/integration/fileblocks/readwrite.test.ts +95 -0
- package/tests/integration/patching/input.txt +145 -0
- package/tests/integration/patching/output.txt +145 -0
- package/tests/integration/patching/patch.txt +15 -0
- package/tests/integration/patching/unseen.txt +80 -0
- package/tests/integration/patching.test.ts +136 -0
- package/tests/languagePlugin.test.ts +74 -0
- package/tests/patching/corrupted.test.ts +78 -0
- package/tests/patching/imports.patch.txt +11 -0
- package/tests/patching/imports.test.ts +43 -0
- package/tests/patching/imports.txt +11 -0
- package/tests/patching/interface.patch.txt +15 -0
- package/tests/patching/interface.txt +7 -0
- package/tests/test.spec.ts +132 -0
- package/ts_build/src/agents/base/base.d.ts +88 -0
- package/ts_build/src/agents/base/base.js +464 -0
- package/ts_build/src/agents/base/base.js.map +1 -0
- package/ts_build/src/agents/base/prompt.d.ts +1 -0
- package/ts_build/src/agents/base/prompt.js +30 -0
- package/ts_build/src/agents/base/prompt.js.map +1 -0
- package/ts_build/src/agents/configurable/ConfigAgent.d.ts +10 -0
- package/ts_build/src/agents/configurable/ConfigAgent.js +25 -0
- package/ts_build/src/agents/configurable/ConfigAgent.js.map +1 -0
- package/ts_build/src/agents/configurable/OpenAIAgent.d.ts +0 -0
- package/ts_build/src/agents/configurable/OpenAIAgent.js +1 -0
- package/ts_build/src/agents/configurable/OpenAIAgent.js.map +1 -0
- package/ts_build/src/agents/developer/developer.d.ts +9 -0
- package/ts_build/src/agents/developer/developer.js +69 -0
- package/ts_build/src/agents/developer/developer.js.map +1 -0
- package/ts_build/src/agents/index.d.ts +7 -0
- package/ts_build/src/agents/index.js +38 -0
- package/ts_build/src/agents/index.js.map +1 -0
- package/ts_build/src/agents/interface.d.ts +10 -0
- package/ts_build/src/agents/interface.js +3 -0
- package/ts_build/src/agents/interface.js.map +1 -0
- package/ts_build/src/agents/patcher/patcher.d.ts +9 -0
- package/ts_build/src/agents/patcher/patcher.js +106 -0
- package/ts_build/src/agents/patcher/patcher.js.map +1 -0
- package/ts_build/src/agents/researcher/researcher.d.ts +9 -0
- package/ts_build/src/agents/researcher/researcher.js +107 -0
- package/ts_build/src/agents/researcher/researcher.js.map +1 -0
- package/ts_build/src/agents/tools/addInternalTools.d.ts +5 -0
- package/ts_build/src/agents/tools/addInternalTools.js +18 -0
- package/ts_build/src/agents/tools/addInternalTools.js.map +1 -0
- package/ts_build/src/agents/tools/agentCall.d.ts +1 -0
- package/ts_build/src/agents/tools/agentCall.js +21 -0
- package/ts_build/src/agents/tools/agentCall.js.map +1 -0
- package/ts_build/src/agents/tools/aiClient.d.ts +6 -0
- package/ts_build/src/agents/tools/aiClient.js +25 -0
- package/ts_build/src/agents/tools/aiClient.js.map +1 -0
- package/ts_build/src/agents/tools/asana/definitions.d.ts +202 -0
- package/ts_build/src/agents/tools/asana/definitions.js +197 -0
- package/ts_build/src/agents/tools/asana/definitions.js.map +1 -0
- package/ts_build/src/agents/tools/asana/index.d.ts +8 -0
- package/ts_build/src/agents/tools/asana/index.js +98 -0
- package/ts_build/src/agents/tools/asana/index.js.map +1 -0
- package/ts_build/src/agents/tools/askHuman.d.ts +1 -0
- package/ts_build/src/agents/tools/askHuman.js +15 -0
- package/ts_build/src/agents/tools/askHuman.js.map +1 -0
- package/ts_build/src/agents/tools/callPlugin.d.ts +1 -0
- package/ts_build/src/agents/tools/callPlugin.js +9 -0
- package/ts_build/src/agents/tools/callPlugin.js.map +1 -0
- package/ts_build/src/agents/tools/client.d.ts +5 -0
- package/ts_build/src/agents/tools/client.js +21 -0
- package/ts_build/src/agents/tools/client.js.map +1 -0
- package/ts_build/src/agents/tools/embeddingSearch.d.ts +1 -0
- package/ts_build/src/agents/tools/embeddingSearch.js +9 -0
- package/ts_build/src/agents/tools/embeddingSearch.js.map +1 -0
- package/ts_build/src/agents/tools/execCommand.d.ts +1 -0
- package/ts_build/src/agents/tools/execCommand.js +25 -0
- package/ts_build/src/agents/tools/execCommand.js.map +1 -0
- package/ts_build/src/agents/tools/fileSearch.d.ts +1 -0
- package/ts_build/src/agents/tools/fileSearch.js +31 -0
- package/ts_build/src/agents/tools/fileSearch.js.map +1 -0
- package/ts_build/src/agents/tools/finalAnswer.d.ts +1 -0
- package/ts_build/src/agents/tools/finalAnswer.js +8 -0
- package/ts_build/src/agents/tools/finalAnswer.js.map +1 -0
- package/ts_build/src/agents/tools/github/definitions.d.ts +47 -0
- package/ts_build/src/agents/tools/github/definitions.js +86 -0
- package/ts_build/src/agents/tools/github/definitions.js.map +1 -0
- package/ts_build/src/agents/tools/github/index.d.ts +789 -0
- package/ts_build/src/agents/tools/github/index.js +56 -0
- package/ts_build/src/agents/tools/github/index.js.map +1 -0
- package/ts_build/src/agents/tools/googleSearch.d.ts +93 -0
- package/ts_build/src/agents/tools/googleSearch.js +117 -0
- package/ts_build/src/agents/tools/googleSearch.js.map +1 -0
- package/ts_build/src/agents/tools/googleSearchTypes.d.ts +74 -0
- package/ts_build/src/agents/tools/googleSearchTypes.js +3 -0
- package/ts_build/src/agents/tools/googleSearchTypes.js.map +1 -0
- package/ts_build/src/agents/tools/index.d.ts +24 -0
- package/ts_build/src/agents/tools/index.js +41 -0
- package/ts_build/src/agents/tools/index.js.map +1 -0
- package/ts_build/src/agents/tools/language/definitions.d.ts +89 -0
- package/ts_build/src/agents/tools/language/definitions.js +95 -0
- package/ts_build/src/agents/tools/language/definitions.js.map +1 -0
- package/ts_build/src/agents/tools/language/index.d.ts +4 -0
- package/ts_build/src/agents/tools/language/index.js +22 -0
- package/ts_build/src/agents/tools/language/index.js.map +1 -0
- package/ts_build/src/agents/tools/lintFile.d.ts +1 -0
- package/ts_build/src/agents/tools/lintFile.js +22 -0
- package/ts_build/src/agents/tools/lintFile.js.map +1 -0
- package/ts_build/src/agents/tools/list.d.ts +2 -0
- package/ts_build/src/agents/tools/list.js +505 -0
- package/ts_build/src/agents/tools/list.js.map +1 -0
- package/ts_build/src/agents/tools/loadWebpage.d.ts +8 -0
- package/ts_build/src/agents/tools/loadWebpage.js +97 -0
- package/ts_build/src/agents/tools/loadWebpage.js.map +1 -0
- package/ts_build/src/agents/tools/modifyFile.d.ts +2 -0
- package/ts_build/src/agents/tools/modifyFile.js +74 -0
- package/ts_build/src/agents/tools/modifyFile.js.map +1 -0
- package/ts_build/src/agents/tools/patch.d.ts +21 -0
- package/ts_build/src/agents/tools/patch.js +443 -0
- package/ts_build/src/agents/tools/patch.js.map +1 -0
- package/ts_build/src/agents/tools/readBlocks.d.ts +2 -0
- package/ts_build/src/agents/tools/readBlocks.js +59 -0
- package/ts_build/src/agents/tools/readBlocks.js.map +1 -0
- package/ts_build/src/agents/tools/readFile.d.ts +1 -0
- package/ts_build/src/agents/tools/readFile.js +46 -0
- package/ts_build/src/agents/tools/readFile.js.map +1 -0
- package/ts_build/src/agents/tools/scanFile.d.ts +1 -0
- package/ts_build/src/agents/tools/scanFile.js +36 -0
- package/ts_build/src/agents/tools/scanFile.js.map +1 -0
- package/ts_build/src/agents/tools/textSearch.d.ts +1 -0
- package/ts_build/src/agents/tools/textSearch.js +22 -0
- package/ts_build/src/agents/tools/textSearch.js.map +1 -0
- package/ts_build/src/agents/tools/types/fileblock.d.ts +5 -0
- package/ts_build/src/agents/tools/types/fileblock.js +3 -0
- package/ts_build/src/agents/tools/types/fileblock.js.map +1 -0
- package/ts_build/src/agents/tools/visionTool.d.ts +1 -0
- package/ts_build/src/agents/tools/visionTool.js +9 -0
- package/ts_build/src/agents/tools/visionTool.js.map +1 -0
- package/ts_build/src/agents/tools/writeFile.d.ts +2 -0
- package/ts_build/src/agents/tools/writeFile.js +61 -0
- package/ts_build/src/agents/tools/writeFile.js.map +1 -0
- package/ts_build/src/agents/vim/vim.d.ts +14 -0
- package/ts_build/src/agents/vim/vim.js +171 -0
- package/ts_build/src/agents/vim/vim.js.map +1 -0
- package/ts_build/src/ai.d.ts +10 -0
- package/ts_build/src/ai.js +102 -0
- package/ts_build/src/ai.js.map +1 -0
- package/ts_build/src/chat.d.ts +13 -0
- package/ts_build/src/chat.js +319 -0
- package/ts_build/src/chat.js.map +1 -0
- package/ts_build/src/cli.d.ts +2 -0
- package/ts_build/src/cli.js +98 -0
- package/ts_build/src/cli.js.map +1 -0
- package/ts_build/src/clients/anthropic.d.ts +31 -0
- package/ts_build/src/clients/anthropic.js +344 -0
- package/ts_build/src/clients/anthropic.js.map +1 -0
- package/ts_build/src/clients/gemini.d.ts +20 -0
- package/ts_build/src/clients/gemini.js +347 -0
- package/ts_build/src/clients/gemini.js.map +1 -0
- package/ts_build/src/clients/http.d.ts +12 -0
- package/ts_build/src/clients/http.js +91 -0
- package/ts_build/src/clients/http.js.map +1 -0
- package/ts_build/src/clients/index.d.ts +47 -0
- package/ts_build/src/clients/index.js +143 -0
- package/ts_build/src/clients/index.js.map +1 -0
- package/ts_build/src/clients/knowhow.d.ts +10 -0
- package/ts_build/src/clients/knowhow.js +24 -0
- package/ts_build/src/clients/knowhow.js.map +1 -0
- package/ts_build/src/clients/openai.d.ts +20 -0
- package/ts_build/src/clients/openai.js +221 -0
- package/ts_build/src/clients/openai.js.map +1 -0
- package/ts_build/src/clients/types.d.ts +89 -0
- package/ts_build/src/clients/types.js +3 -0
- package/ts_build/src/clients/types.js.map +1 -0
- package/ts_build/src/clients/xai.d.ts +22 -0
- package/ts_build/src/clients/xai.js +108 -0
- package/ts_build/src/clients/xai.js.map +1 -0
- package/ts_build/src/config.d.ts +9 -0
- package/ts_build/src/config.js +216 -0
- package/ts_build/src/config.js.map +1 -0
- package/ts_build/src/conversion.d.ts +7 -0
- package/ts_build/src/conversion.js +114 -0
- package/ts_build/src/conversion.js.map +1 -0
- package/ts_build/src/dataset/diffs/debug-errors.d.ts +1 -0
- package/ts_build/src/dataset/diffs/debug-errors.js +28 -0
- package/ts_build/src/dataset/diffs/debug-errors.js.map +1 -0
- package/ts_build/src/dataset/diffs/debug.d.ts +1 -0
- package/ts_build/src/dataset/diffs/debug.js +50 -0
- package/ts_build/src/dataset/diffs/debug.js.map +1 -0
- package/ts_build/src/dataset/diffs/generate.d.ts +1 -0
- package/ts_build/src/dataset/diffs/generate.js +70 -0
- package/ts_build/src/dataset/diffs/generate.js.map +1 -0
- package/ts_build/src/dataset/diffs/jsonl.d.ts +1 -0
- package/ts_build/src/dataset/diffs/jsonl.js +47 -0
- package/ts_build/src/dataset/diffs/jsonl.js.map +1 -0
- package/ts_build/src/dataset/diffs/test.d.ts +1 -0
- package/ts_build/src/dataset/diffs/test.js +114 -0
- package/ts_build/src/dataset/diffs/test.js.map +1 -0
- package/ts_build/src/embeddings.d.ts +19 -0
- package/ts_build/src/embeddings.js +322 -0
- package/ts_build/src/embeddings.js.map +1 -0
- package/ts_build/src/hashes.d.ts +6 -0
- package/ts_build/src/hashes.js +78 -0
- package/ts_build/src/hashes.js.map +1 -0
- package/ts_build/src/index.d.ts +14 -0
- package/ts_build/src/index.js +258 -0
- package/ts_build/src/index.js.map +1 -0
- package/ts_build/src/login.d.ts +2 -0
- package/ts_build/src/login.js +63 -0
- package/ts_build/src/login.js.map +1 -0
- package/ts_build/src/microphone.d.ts +9 -0
- package/ts_build/src/microphone.js +131 -0
- package/ts_build/src/microphone.js.map +1 -0
- package/ts_build/src/modules/index.d.ts +3 -0
- package/ts_build/src/modules/index.js +34 -0
- package/ts_build/src/modules/index.js.map +1 -0
- package/ts_build/src/modules/types.d.ts +37 -0
- package/ts_build/src/modules/types.js +3 -0
- package/ts_build/src/modules/types.js.map +1 -0
- package/ts_build/src/plugins/asana.d.ts +16 -0
- package/ts_build/src/plugins/asana.js +114 -0
- package/ts_build/src/plugins/asana.js.map +1 -0
- package/ts_build/src/plugins/downloader/downloader.d.ts +12 -0
- package/ts_build/src/plugins/downloader/downloader.js +174 -0
- package/ts_build/src/plugins/downloader/downloader.js.map +1 -0
- package/ts_build/src/plugins/downloader/index.d.ts +3 -0
- package/ts_build/src/plugins/downloader/index.js +41 -0
- package/ts_build/src/plugins/downloader/index.js.map +1 -0
- package/ts_build/src/plugins/downloader/plugin.d.ts +8 -0
- package/ts_build/src/plugins/downloader/plugin.js +83 -0
- package/ts_build/src/plugins/downloader/plugin.js.map +1 -0
- package/ts_build/src/plugins/downloader/types.d.ts +82 -0
- package/ts_build/src/plugins/downloader/types.js +79 -0
- package/ts_build/src/plugins/downloader/types.js.map +1 -0
- package/ts_build/src/plugins/embedding.d.ts +5 -0
- package/ts_build/src/plugins/embedding.js +28 -0
- package/ts_build/src/plugins/embedding.js.map +1 -0
- package/ts_build/src/plugins/figma.d.ts +20 -0
- package/ts_build/src/plugins/figma.js +94 -0
- package/ts_build/src/plugins/figma.js.map +1 -0
- package/ts_build/src/plugins/github.d.ts +1223 -0
- package/ts_build/src/plugins/github.js +115 -0
- package/ts_build/src/plugins/github.js.map +1 -0
- package/ts_build/src/plugins/jira.d.ts +15 -0
- package/ts_build/src/plugins/jira.js +94 -0
- package/ts_build/src/plugins/jira.js.map +1 -0
- package/ts_build/src/plugins/language.d.ts +8 -0
- package/ts_build/src/plugins/language.js +69 -0
- package/ts_build/src/plugins/language.js.map +1 -0
- package/ts_build/src/plugins/linear.d.ts +23 -0
- package/ts_build/src/plugins/linear.js +184 -0
- package/ts_build/src/plugins/linear.js.map +1 -0
- package/ts_build/src/plugins/notion.d.ts +32 -0
- package/ts_build/src/plugins/notion.js +131 -0
- package/ts_build/src/plugins/notion.js.map +1 -0
- package/ts_build/src/plugins/plugins.d.ts +11 -0
- package/ts_build/src/plugins/plugins.js +51 -0
- package/ts_build/src/plugins/plugins.js.map +1 -0
- package/ts_build/src/plugins/types.d.ts +5 -0
- package/ts_build/src/plugins/types.js +3 -0
- package/ts_build/src/plugins/types.js.map +1 -0
- package/ts_build/src/plugins/url.d.ts +8 -0
- package/ts_build/src/plugins/url.js +50 -0
- package/ts_build/src/plugins/url.js.map +1 -0
- package/ts_build/src/plugins/vim.d.ts +11 -0
- package/ts_build/src/plugins/vim.js +63 -0
- package/ts_build/src/plugins/vim.js.map +1 -0
- package/ts_build/src/prompts/BasicCodeDocumenter.d.ts +2 -0
- package/ts_build/src/prompts/BasicCodeDocumenter.js +4 -0
- package/ts_build/src/prompts/BasicCodeDocumenter.js.map +1 -0
- package/ts_build/src/prompts/BasicProjectDocumenter.d.ts +2 -0
- package/ts_build/src/prompts/BasicProjectDocumenter.js +4 -0
- package/ts_build/src/prompts/BasicProjectDocumenter.js.map +1 -0
- package/ts_build/src/prompts/EmbeddingSeachHelper.d.ts +2 -0
- package/ts_build/src/prompts/EmbeddingSeachHelper.js +4 -0
- package/ts_build/src/prompts/EmbeddingSeachHelper.js.map +1 -0
- package/ts_build/src/prompts/index.d.ts +13 -0
- package/ts_build/src/prompts/index.js +15 -0
- package/ts_build/src/prompts/index.js.map +1 -0
- package/ts_build/src/server/index.d.ts +1 -0
- package/ts_build/src/server/index.js +17 -0
- package/ts_build/src/server/index.js.map +1 -0
- package/ts_build/src/services/AgentService.d.ts +14 -0
- package/ts_build/src/services/AgentService.js +87 -0
- package/ts_build/src/services/AgentService.js.map +1 -0
- package/ts_build/src/services/EmbeddingService.d.ts +39 -0
- package/ts_build/src/services/EmbeddingService.js +307 -0
- package/ts_build/src/services/EmbeddingService.js.map +1 -0
- package/ts_build/src/services/EventService.d.ts +9 -0
- package/ts_build/src/services/EventService.js +20 -0
- package/ts_build/src/services/EventService.js.map +1 -0
- package/ts_build/src/services/GitHub.d.ts +11 -0
- package/ts_build/src/services/GitHub.js +52 -0
- package/ts_build/src/services/GitHub.js.map +1 -0
- package/ts_build/src/services/KnowhowClient.d.ts +18 -0
- package/ts_build/src/services/KnowhowClient.js +85 -0
- package/ts_build/src/services/KnowhowClient.js.map +1 -0
- package/ts_build/src/services/Mcp.d.ts +377 -0
- package/ts_build/src/services/Mcp.js +221 -0
- package/ts_build/src/services/Mcp.js.map +1 -0
- package/ts_build/src/services/McpServer.d.ts +19 -0
- package/ts_build/src/services/McpServer.js +104 -0
- package/ts_build/src/services/McpServer.js.map +1 -0
- package/ts_build/src/services/McpWebsocketTransport.d.ts +13 -0
- package/ts_build/src/services/McpWebsocketTransport.js +59 -0
- package/ts_build/src/services/McpWebsocketTransport.js.map +1 -0
- package/ts_build/src/services/S3.d.ts +9 -0
- package/ts_build/src/services/S3.js +116 -0
- package/ts_build/src/services/S3.js.map +1 -0
- package/ts_build/src/services/Tools.d.ts +19 -0
- package/ts_build/src/services/Tools.js +51 -0
- package/ts_build/src/services/Tools.js.map +1 -0
- package/ts_build/src/services/flags.d.ts +13 -0
- package/ts_build/src/services/flags.js +48 -0
- package/ts_build/src/services/flags.js.map +1 -0
- package/ts_build/src/services/index.d.ts +8 -0
- package/ts_build/src/services/index.js +38 -0
- package/ts_build/src/services/index.js.map +1 -0
- package/ts_build/src/terminal.d.ts +1 -0
- package/ts_build/src/terminal.js +35 -0
- package/ts_build/src/terminal.js.map +1 -0
- package/ts_build/src/types.d.ts +181 -0
- package/ts_build/src/types.js +94 -0
- package/ts_build/src/types.js.map +1 -0
- package/ts_build/src/utils/index.d.ts +24 -0
- package/ts_build/src/utils/index.js +116 -0
- package/ts_build/src/utils/index.js.map +1 -0
- package/ts_build/src/worker.d.ts +1 -0
- package/ts_build/src/worker.js +68 -0
- package/ts_build/src/worker.js.map +1 -0
- package/ts_build/tests/integration/figma.test.d.ts +1 -0
- package/ts_build/tests/integration/figma.test.js +47 -0
- package/ts_build/tests/integration/figma.test.js.map +1 -0
- package/ts_build/tests/integration/fileblocks/readwrite.test.d.ts +1 -0
- package/ts_build/tests/integration/fileblocks/readwrite.test.js +95 -0
- package/ts_build/tests/integration/fileblocks/readwrite.test.js.map +1 -0
- package/ts_build/tests/integration/patching.test.d.ts +1 -0
- package/ts_build/tests/integration/patching.test.js +117 -0
- package/ts_build/tests/integration/patching.test.js.map +1 -0
- package/ts_build/tests/languagePlugin.test.d.ts +1 -0
- package/ts_build/tests/languagePlugin.test.js +78 -0
- package/ts_build/tests/languagePlugin.test.js.map +1 -0
- package/ts_build/tests/patching/corrupted.test.d.ts +1 -0
- package/ts_build/tests/patching/corrupted.test.js +53 -0
- package/ts_build/tests/patching/corrupted.test.js.map +1 -0
- package/ts_build/tests/patching/imports.test.d.ts +1 -0
- package/ts_build/tests/patching/imports.test.js +61 -0
- package/ts_build/tests/patching/imports.test.js.map +1 -0
- package/ts_build/tests/test.spec.d.ts +1 -0
- package/ts_build/tests/test.spec.js +101 -0
- package/ts_build/tests/test.spec.js.map +1 -0
- package/tsconfig.json +31 -0
- 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 @@
|
|
|
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[]>;
|