@travisennis/acai 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/.acai/acai.json +9 -0
- package/.acai/prompts/add-openrouter-model.md +13 -0
- package/.acai/prompts/project-status.md +4 -0
- package/.acai/prompts/update-architecture-document.md +9 -0
- package/.acai/rules/learned-rules.md +9 -0
- package/.ai/docs/available-tools.txt +3 -0
- package/.ai/docs/cognitive_complexity_refactoring_progress.md +65 -0
- package/.ai/docs/deleted_tools.md +168 -0
- package/.ai/docs/deleted_tools_88ced9ef.md +56 -0
- package/.ai/docs/image-pasting.md +46 -0
- package/.ai/docs/initialize-app.md +117 -0
- package/.ai/docs/issue-4-plan.md +44 -0
- package/.ai/docs/marked-renderer-debug.md +15 -0
- package/.ai/docs/marked-renderer-refactor-plan.md +64 -0
- package/.ai/docs/memory-use-cases.md +55 -0
- package/.ai/docs/prompt-consistency.md +31 -0
- package/.ai/docs/refactoring-tools.md +98 -0
- package/.ai/docs/system-prompt-update.md +174 -0
- package/.ai/docs/system_prompt.txt +210 -0
- package/.ai/docs/tasks.md +49 -0
- package/.ai/plan.md +131 -0
- package/.ai/prompt.md +1 -0
- package/.ai/scripts/fetch_models.js +27 -0
- package/.ai/scripts/generateSystemPrompt.ts +15 -0
- package/.ai/scripts/list-tools.mjs +4 -0
- package/.ai/scripts/p5_geometric_shapes.js +149 -0
- package/.husky/commit-msg +1 -0
- package/.husky/pre-commit +3 -0
- package/.husky/pre-push +1 -0
- package/.ignore +4 -0
- package/AGENTS.md +25 -0
- package/ARCHITECTURE.md +304 -0
- package/LICENSE +21 -0
- package/README.md +392 -0
- package/TODO.md +2 -0
- package/biome.json +61 -0
- package/commitlint.config.js +3 -0
- package/dist/cli.d.ts +19 -0
- package/dist/cli.js +116 -0
- package/dist/commands/application-log-command.d.ts +2 -0
- package/dist/commands/application-log-command.js +43 -0
- package/dist/commands/clear-command.d.ts +2 -0
- package/dist/commands/clear-command.js +12 -0
- package/dist/commands/compact-command.d.ts +2 -0
- package/dist/commands/compact-command.js +51 -0
- package/dist/commands/copy-command.d.ts +2 -0
- package/dist/commands/copy-command.js +51 -0
- package/dist/commands/edit-command.d.ts +2 -0
- package/dist/commands/edit-command.js +53 -0
- package/dist/commands/edit-prompt-command.d.ts +2 -0
- package/dist/commands/edit-prompt-command.js +25 -0
- package/dist/commands/exit-command.d.ts +2 -0
- package/dist/commands/exit-command.js +14 -0
- package/dist/commands/files-command.d.ts +2 -0
- package/dist/commands/files-command.js +63 -0
- package/dist/commands/generate-rules-command.d.ts +2 -0
- package/dist/commands/generate-rules-command.js +61 -0
- package/dist/commands/help-command.d.ts +2 -0
- package/dist/commands/help-command.js +19 -0
- package/dist/commands/init-command.d.ts +2 -0
- package/dist/commands/init-command.js +40 -0
- package/dist/commands/last-log-command.d.ts +2 -0
- package/dist/commands/last-log-command.js +76 -0
- package/dist/commands/manager.d.ts +22 -0
- package/dist/commands/manager.js +123 -0
- package/dist/commands/model-command.d.ts +2 -0
- package/dist/commands/model-command.js +84 -0
- package/dist/commands/paste-command.d.ts +2 -0
- package/dist/commands/paste-command.js +40 -0
- package/dist/commands/prompt-command.d.ts +2 -0
- package/dist/commands/prompt-command.js +111 -0
- package/dist/commands/reset-command.d.ts +2 -0
- package/dist/commands/reset-command.js +16 -0
- package/dist/commands/rules-command.d.ts +2 -0
- package/dist/commands/rules-command.js +68 -0
- package/dist/commands/save-command.d.ts +2 -0
- package/dist/commands/save-command.js +14 -0
- package/dist/commands/types.d.ts +26 -0
- package/dist/commands/types.js +1 -0
- package/dist/commands/usage-command.d.ts +2 -0
- package/dist/commands/usage-command.js +21 -0
- package/dist/config.d.ts +60 -0
- package/dist/config.js +193 -0
- package/dist/conversation-analyzer.d.ts +10 -0
- package/dist/conversation-analyzer.js +88 -0
- package/dist/dedent.d.ts +3 -0
- package/dist/dedent.js +38 -0
- package/dist/formatting.d.ts +17 -0
- package/dist/formatting.js +103 -0
- package/dist/index.d.ts +18 -0
- package/dist/index.js +213 -0
- package/dist/logger.d.ts +2 -0
- package/dist/logger.js +24 -0
- package/dist/mentions.d.ts +9 -0
- package/dist/mentions.js +182 -0
- package/dist/messages.d.ts +69 -0
- package/dist/messages.js +261 -0
- package/dist/middleware/audit-message.d.ts +5 -0
- package/dist/middleware/audit-message.js +95 -0
- package/dist/middleware/index.d.ts +2 -0
- package/dist/middleware/index.js +2 -0
- package/dist/middleware/rate-limit.d.ts +4 -0
- package/dist/middleware/rate-limit.js +17 -0
- package/dist/models/ai-config.d.ts +12 -0
- package/dist/models/ai-config.js +87 -0
- package/dist/models/anthropic-provider.d.ts +25 -0
- package/dist/models/anthropic-provider.js +184 -0
- package/dist/models/deepseek-provider.d.ts +20 -0
- package/dist/models/deepseek-provider.js +42 -0
- package/dist/models/google-provider.d.ts +19 -0
- package/dist/models/google-provider.js +56 -0
- package/dist/models/manager.d.ts +15 -0
- package/dist/models/manager.js +48 -0
- package/dist/models/openai-provider.d.ts +22 -0
- package/dist/models/openai-provider.js +70 -0
- package/dist/models/openrouter-provider.d.ts +36 -0
- package/dist/models/openrouter-provider.js +276 -0
- package/dist/models/providers.d.ts +33 -0
- package/dist/models/providers.js +116 -0
- package/dist/models/xai-provider.d.ts +20 -0
- package/dist/models/xai-provider.js +47 -0
- package/dist/parsing.d.ts +2 -0
- package/dist/parsing.js +18 -0
- package/dist/prompts/manager.d.ts +19 -0
- package/dist/prompts/manager.js +71 -0
- package/dist/prompts.d.ts +4 -0
- package/dist/prompts.js +158 -0
- package/dist/repl-prompt.d.ts +14 -0
- package/dist/repl-prompt.js +147 -0
- package/dist/repl.d.ts +27 -0
- package/dist/repl.js +431 -0
- package/dist/source/cli.d.ts +19 -0
- package/dist/source/cli.js +116 -0
- package/dist/source/commands/application-log-command.d.ts +2 -0
- package/dist/source/commands/application-log-command.js +43 -0
- package/dist/source/commands/clear-command.d.ts +2 -0
- package/dist/source/commands/clear-command.js +12 -0
- package/dist/source/commands/compact-command.d.ts +2 -0
- package/dist/source/commands/compact-command.js +51 -0
- package/dist/source/commands/copy-command.d.ts +2 -0
- package/dist/source/commands/copy-command.js +51 -0
- package/dist/source/commands/edit-command.d.ts +2 -0
- package/dist/source/commands/edit-command.js +53 -0
- package/dist/source/commands/edit-prompt-command.d.ts +2 -0
- package/dist/source/commands/edit-prompt-command.js +25 -0
- package/dist/source/commands/exit-command.d.ts +2 -0
- package/dist/source/commands/exit-command.js +14 -0
- package/dist/source/commands/files-command.d.ts +2 -0
- package/dist/source/commands/files-command.js +63 -0
- package/dist/source/commands/generate-rules-command.d.ts +2 -0
- package/dist/source/commands/generate-rules-command.js +61 -0
- package/dist/source/commands/help-command.d.ts +2 -0
- package/dist/source/commands/help-command.js +19 -0
- package/dist/source/commands/init-command.d.ts +2 -0
- package/dist/source/commands/init-command.js +40 -0
- package/dist/source/commands/last-log-command.d.ts +2 -0
- package/dist/source/commands/last-log-command.js +76 -0
- package/dist/source/commands/manager.d.ts +22 -0
- package/dist/source/commands/manager.js +123 -0
- package/dist/source/commands/model-command.d.ts +2 -0
- package/dist/source/commands/model-command.js +84 -0
- package/dist/source/commands/paste-command.d.ts +2 -0
- package/dist/source/commands/paste-command.js +40 -0
- package/dist/source/commands/prompt-command.d.ts +2 -0
- package/dist/source/commands/prompt-command.js +111 -0
- package/dist/source/commands/reset-command.d.ts +2 -0
- package/dist/source/commands/reset-command.js +16 -0
- package/dist/source/commands/rules-command.d.ts +2 -0
- package/dist/source/commands/rules-command.js +68 -0
- package/dist/source/commands/save-command.d.ts +2 -0
- package/dist/source/commands/save-command.js +14 -0
- package/dist/source/commands/types.d.ts +26 -0
- package/dist/source/commands/types.js +1 -0
- package/dist/source/commands/usage-command.d.ts +2 -0
- package/dist/source/commands/usage-command.js +21 -0
- package/dist/source/config.d.ts +60 -0
- package/dist/source/config.js +193 -0
- package/dist/source/conversation-analyzer.d.ts +10 -0
- package/dist/source/conversation-analyzer.js +88 -0
- package/dist/source/dedent.d.ts +3 -0
- package/dist/source/dedent.js +38 -0
- package/dist/source/formatting.d.ts +17 -0
- package/dist/source/formatting.js +103 -0
- package/dist/source/index.d.ts +18 -0
- package/dist/source/index.js +213 -0
- package/dist/source/logger.d.ts +2 -0
- package/dist/source/logger.js +24 -0
- package/dist/source/mentions.d.ts +9 -0
- package/dist/source/mentions.js +182 -0
- package/dist/source/messages.d.ts +69 -0
- package/dist/source/messages.js +261 -0
- package/dist/source/middleware/audit-message.d.ts +5 -0
- package/dist/source/middleware/audit-message.js +95 -0
- package/dist/source/middleware/index.d.ts +2 -0
- package/dist/source/middleware/index.js +2 -0
- package/dist/source/middleware/rate-limit.d.ts +4 -0
- package/dist/source/middleware/rate-limit.js +17 -0
- package/dist/source/models/ai-config.d.ts +12 -0
- package/dist/source/models/ai-config.js +87 -0
- package/dist/source/models/anthropic-provider.d.ts +25 -0
- package/dist/source/models/anthropic-provider.js +184 -0
- package/dist/source/models/deepseek-provider.d.ts +20 -0
- package/dist/source/models/deepseek-provider.js +42 -0
- package/dist/source/models/google-provider.d.ts +19 -0
- package/dist/source/models/google-provider.js +56 -0
- package/dist/source/models/manager.d.ts +15 -0
- package/dist/source/models/manager.js +48 -0
- package/dist/source/models/openai-provider.d.ts +22 -0
- package/dist/source/models/openai-provider.js +70 -0
- package/dist/source/models/openrouter-provider.d.ts +36 -0
- package/dist/source/models/openrouter-provider.js +276 -0
- package/dist/source/models/providers.d.ts +33 -0
- package/dist/source/models/providers.js +116 -0
- package/dist/source/models/xai-provider.d.ts +20 -0
- package/dist/source/models/xai-provider.js +47 -0
- package/dist/source/parsing.d.ts +2 -0
- package/dist/source/parsing.js +18 -0
- package/dist/source/prompts/manager.d.ts +19 -0
- package/dist/source/prompts/manager.js +71 -0
- package/dist/source/prompts.d.ts +4 -0
- package/dist/source/prompts.js +158 -0
- package/dist/source/repl-prompt.d.ts +14 -0
- package/dist/source/repl-prompt.js +147 -0
- package/dist/source/repl.d.ts +27 -0
- package/dist/source/repl.js +431 -0
- package/dist/source/terminal/formatting.d.ts +37 -0
- package/dist/source/terminal/formatting.js +106 -0
- package/dist/source/terminal/index.d.ts +94 -0
- package/dist/source/terminal/index.js +420 -0
- package/dist/source/terminal/markdown-utils.d.ts +2 -0
- package/dist/source/terminal/markdown-utils.js +81 -0
- package/dist/source/terminal/markdown.d.ts +1 -0
- package/dist/source/terminal/markdown.js +111 -0
- package/dist/source/terminal/types.d.ts +71 -0
- package/dist/source/terminal/types.js +1 -0
- package/dist/source/terminal-output.d.ts +8 -0
- package/dist/source/terminal-output.js +213 -0
- package/dist/source/terminal-output.test.d.ts +8 -0
- package/dist/source/terminal-output.test.js +213 -0
- package/dist/source/token-tracker.d.ts +14 -0
- package/dist/source/token-tracker.js +53 -0
- package/dist/source/token-utils.d.ts +7 -0
- package/dist/source/token-utils.js +13 -0
- package/dist/source/tools/agent.d.ts +17 -0
- package/dist/source/tools/agent.js +87 -0
- package/dist/source/tools/bash.d.ts +19 -0
- package/dist/source/tools/bash.js +294 -0
- package/dist/source/tools/code-interpreter.d.ts +12 -0
- package/dist/source/tools/code-interpreter.js +131 -0
- package/dist/source/tools/command-validation.d.ts +8 -0
- package/dist/source/tools/command-validation.js +69 -0
- package/dist/source/tools/delete-file.d.ts +12 -0
- package/dist/source/tools/delete-file.js +56 -0
- package/dist/source/tools/directory-tree.d.ts +12 -0
- package/dist/source/tools/directory-tree.js +38 -0
- package/dist/source/tools/edit-file.d.ts +19 -0
- package/dist/source/tools/edit-file.js +107 -0
- package/dist/source/tools/filesystem-utils.d.ts +22 -0
- package/dist/source/tools/filesystem-utils.js +191 -0
- package/dist/source/tools/git-utils.d.ts +14 -0
- package/dist/source/tools/git-utils.js +64 -0
- package/dist/source/tools/grep.d.ts +17 -0
- package/dist/source/tools/grep.js +138 -0
- package/dist/source/tools/index.d.ts +161 -0
- package/dist/source/tools/index.js +209 -0
- package/dist/source/tools/memory-read.d.ts +13 -0
- package/dist/source/tools/memory-read.js +135 -0
- package/dist/source/tools/memory-write.d.ts +12 -0
- package/dist/source/tools/memory-write.js +83 -0
- package/dist/source/tools/move-file.d.ts +13 -0
- package/dist/source/tools/move-file.js +44 -0
- package/dist/source/tools/read-file.d.ts +17 -0
- package/dist/source/tools/read-file.js +86 -0
- package/dist/source/tools/read-multiple-files.d.ts +14 -0
- package/dist/source/tools/read-multiple-files.js +55 -0
- package/dist/source/tools/save-file.d.ts +17 -0
- package/dist/source/tools/save-file.js +98 -0
- package/dist/source/tools/think.d.ts +11 -0
- package/dist/source/tools/think.js +45 -0
- package/dist/source/tools/types.d.ts +29 -0
- package/dist/source/tools/types.js +14 -0
- package/dist/source/tools/web-fetch.d.ts +47 -0
- package/dist/source/tools/web-fetch.js +246 -0
- package/dist/source/tools/web-search.d.ts +13 -0
- package/dist/source/tools/web-search.js +80 -0
- package/dist/source/utils/process.d.ts +36 -0
- package/dist/source/utils/process.js +75 -0
- package/dist/source/version.d.ts +1 -0
- package/dist/source/version.js +21 -0
- package/dist/terminal/formatting.d.ts +37 -0
- package/dist/terminal/formatting.js +106 -0
- package/dist/terminal/index.d.ts +94 -0
- package/dist/terminal/index.js +420 -0
- package/dist/terminal/markdown-utils.d.ts +2 -0
- package/dist/terminal/markdown-utils.js +81 -0
- package/dist/terminal/markdown.d.ts +1 -0
- package/dist/terminal/markdown.js +111 -0
- package/dist/terminal/types.d.ts +71 -0
- package/dist/terminal/types.js +1 -0
- package/dist/terminal-output.d.ts +8 -0
- package/dist/terminal-output.js +213 -0
- package/dist/token-tracker.d.ts +14 -0
- package/dist/token-tracker.js +53 -0
- package/dist/token-utils.d.ts +7 -0
- package/dist/token-utils.js +13 -0
- package/dist/tools/agent.d.ts +17 -0
- package/dist/tools/agent.js +87 -0
- package/dist/tools/bash.d.ts +19 -0
- package/dist/tools/bash.js +294 -0
- package/dist/tools/code-interpreter.d.ts +12 -0
- package/dist/tools/code-interpreter.js +131 -0
- package/dist/tools/command-validation.d.ts +8 -0
- package/dist/tools/command-validation.js +69 -0
- package/dist/tools/delete-file.d.ts +12 -0
- package/dist/tools/delete-file.js +56 -0
- package/dist/tools/directory-tree.d.ts +12 -0
- package/dist/tools/directory-tree.js +38 -0
- package/dist/tools/edit-file.d.ts +19 -0
- package/dist/tools/edit-file.js +107 -0
- package/dist/tools/filesystem-utils.d.ts +22 -0
- package/dist/tools/filesystem-utils.js +191 -0
- package/dist/tools/git-utils.d.ts +14 -0
- package/dist/tools/git-utils.js +64 -0
- package/dist/tools/grep.d.ts +17 -0
- package/dist/tools/grep.js +138 -0
- package/dist/tools/index.d.ts +161 -0
- package/dist/tools/index.js +209 -0
- package/dist/tools/memory-read.d.ts +13 -0
- package/dist/tools/memory-read.js +135 -0
- package/dist/tools/memory-write.d.ts +12 -0
- package/dist/tools/memory-write.js +83 -0
- package/dist/tools/move-file.d.ts +13 -0
- package/dist/tools/move-file.js +44 -0
- package/dist/tools/read-file.d.ts +17 -0
- package/dist/tools/read-file.js +86 -0
- package/dist/tools/read-multiple-files.d.ts +14 -0
- package/dist/tools/read-multiple-files.js +55 -0
- package/dist/tools/save-file.d.ts +17 -0
- package/dist/tools/save-file.js +98 -0
- package/dist/tools/think.d.ts +11 -0
- package/dist/tools/think.js +45 -0
- package/dist/tools/types.d.ts +29 -0
- package/dist/tools/types.js +14 -0
- package/dist/tools/web-fetch.d.ts +47 -0
- package/dist/tools/web-fetch.js +246 -0
- package/dist/tools/web-search.d.ts +13 -0
- package/dist/tools/web-search.js +80 -0
- package/dist/utils/process.d.ts +36 -0
- package/dist/utils/process.js +75 -0
- package/dist/version.d.ts +1 -0
- package/dist/version.js +21 -0
- package/knip.json +5 -0
- package/package.json +83 -0
- package/source/cli.ts +172 -0
- package/source/commands/application-log-command.ts +53 -0
- package/source/commands/clear-command.ts +14 -0
- package/source/commands/compact-command.ts +64 -0
- package/source/commands/copy-command.ts +55 -0
- package/source/commands/edit-command.ts +63 -0
- package/source/commands/edit-prompt-command.ts +31 -0
- package/source/commands/exit-command.ts +18 -0
- package/source/commands/files-command.ts +85 -0
- package/source/commands/generate-rules-command.ts +82 -0
- package/source/commands/help-command.ts +27 -0
- package/source/commands/init-command.ts +48 -0
- package/source/commands/last-log-command.ts +88 -0
- package/source/commands/manager.ts +151 -0
- package/source/commands/model-command.ts +123 -0
- package/source/commands/paste-command.ts +62 -0
- package/source/commands/prompt-command.ts +150 -0
- package/source/commands/reset-command.ts +22 -0
- package/source/commands/rules-command.ts +76 -0
- package/source/commands/save-command.ts +20 -0
- package/source/commands/types.ts +28 -0
- package/source/commands/usage-command.ts +26 -0
- package/source/config.ts +223 -0
- package/source/conversation-analyzer.ts +115 -0
- package/source/dedent.ts +53 -0
- package/source/formatting.ts +132 -0
- package/source/index.ts +240 -0
- package/source/logger.ts +29 -0
- package/source/mentions.ts +227 -0
- package/source/messages.ts +360 -0
- package/source/middleware/audit-message.ts +133 -0
- package/source/middleware/index.ts +2 -0
- package/source/middleware/rate-limit.ts +24 -0
- package/source/models/ai-config.ts +109 -0
- package/source/models/anthropic-provider.ts +199 -0
- package/source/models/deepseek-provider.ts +53 -0
- package/source/models/google-provider.ts +68 -0
- package/source/models/manager.ts +84 -0
- package/source/models/openai-provider.ts +81 -0
- package/source/models/openrouter-provider.ts +288 -0
- package/source/models/providers.ts +197 -0
- package/source/models/xai-provider.ts +59 -0
- package/source/parsing.ts +20 -0
- package/source/prompts/manager.ts +90 -0
- package/source/prompts.ts +172 -0
- package/source/repl-prompt.ts +196 -0
- package/source/repl.ts +572 -0
- package/source/terminal/formatting.ts +121 -0
- package/source/terminal/index.ts +518 -0
- package/source/terminal/markdown-utils.ts +89 -0
- package/source/terminal/markdown.ts +155 -0
- package/source/terminal/types.ts +84 -0
- package/source/terminal-output.test.ts +266 -0
- package/source/token-tracker.ts +78 -0
- package/source/token-utils.ts +17 -0
- package/source/tools/agent.ts +107 -0
- package/source/tools/bash.ts +367 -0
- package/source/tools/code-interpreter.ts +172 -0
- package/source/tools/command-validation.ts +81 -0
- package/source/tools/delete-file.ts +71 -0
- package/source/tools/directory-tree.ts +54 -0
- package/source/tools/edit-file.ts +155 -0
- package/source/tools/filesystem-utils.ts +265 -0
- package/source/tools/git-utils.ts +70 -0
- package/source/tools/grep.ts +184 -0
- package/source/tools/index.ts +278 -0
- package/source/tools/memory-read.ts +174 -0
- package/source/tools/memory-write.ts +105 -0
- package/source/tools/move-file.ts +59 -0
- package/source/tools/read-file.ts +129 -0
- package/source/tools/read-multiple-files.ts +80 -0
- package/source/tools/save-file.ts +147 -0
- package/source/tools/think.ts +51 -0
- package/source/tools/types.ts +58 -0
- package/source/tools/web-fetch.ts +327 -0
- package/source/tools/web-search.ts +101 -0
- package/source/utils/process.ts +121 -0
- package/source/version.ts +21 -0
- package/test/commands/copy-command.test.ts +69 -0
- package/test/config.test.ts +200 -0
- package/test/terminal/markdown-utils.test.ts +124 -0
- package/test/tools/bash-tool.test.ts +58 -0
- package/test/tools/code-interpreter.test.ts +91 -0
- package/test/tools/command-validation.test.ts +48 -0
- package/tsconfig.build.json +9 -0
- package/tsconfig.json +30 -0
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
const THINKING_TIERS = [
|
|
2
|
+
{
|
|
3
|
+
pattern: /\b(ultrathink|think super hard|think really hard|think intensely)\b/i,
|
|
4
|
+
budget: 31999,
|
|
5
|
+
effort: "high",
|
|
6
|
+
},
|
|
7
|
+
{
|
|
8
|
+
pattern: /\b(megathink|think (very )?hard|think (a lot|more|about it))\b/i,
|
|
9
|
+
budget: 10000,
|
|
10
|
+
effort: "medium",
|
|
11
|
+
},
|
|
12
|
+
{
|
|
13
|
+
pattern: /\bthink\b/i, // Catch-all for standalone "think"
|
|
14
|
+
budget: 4000,
|
|
15
|
+
effort: "low",
|
|
16
|
+
},
|
|
17
|
+
];
|
|
18
|
+
function calculateThinkingLevel(userInput) {
|
|
19
|
+
let tokenBudget = 0; // Default
|
|
20
|
+
let effort = "none";
|
|
21
|
+
for (const tier of THINKING_TIERS) {
|
|
22
|
+
if (tier.pattern.test(userInput)) {
|
|
23
|
+
tokenBudget = tier.budget;
|
|
24
|
+
effort = tier.effort;
|
|
25
|
+
break; // Use highest priority match
|
|
26
|
+
}
|
|
27
|
+
}
|
|
28
|
+
return { tokenBudget, effort };
|
|
29
|
+
}
|
|
30
|
+
export class AiConfig {
|
|
31
|
+
modelMetadata;
|
|
32
|
+
prompt;
|
|
33
|
+
constructor({ modelMetadata, prompt, }) {
|
|
34
|
+
this.modelMetadata = modelMetadata;
|
|
35
|
+
this.prompt = prompt;
|
|
36
|
+
}
|
|
37
|
+
getMaxTokens() {
|
|
38
|
+
const modelConfig = this.modelMetadata;
|
|
39
|
+
const thinkingLevel = calculateThinkingLevel(this.prompt);
|
|
40
|
+
const maxTokens = modelConfig.provider === "anthropic" && modelConfig.supportsReasoning
|
|
41
|
+
? modelConfig.maxOutputTokens - thinkingLevel.tokenBudget
|
|
42
|
+
: modelConfig.maxOutputTokens;
|
|
43
|
+
return maxTokens;
|
|
44
|
+
}
|
|
45
|
+
getProviderOptions() {
|
|
46
|
+
const modelConfig = this.modelMetadata;
|
|
47
|
+
const thinkingLevel = calculateThinkingLevel(this.prompt);
|
|
48
|
+
if (modelConfig.supportsReasoning && thinkingLevel.effort !== "none") {
|
|
49
|
+
switch (modelConfig.provider) {
|
|
50
|
+
case "anthropic":
|
|
51
|
+
return {
|
|
52
|
+
anthropic: {
|
|
53
|
+
thinking: {
|
|
54
|
+
type: "enabled",
|
|
55
|
+
budgetTokens: thinkingLevel.tokenBudget,
|
|
56
|
+
},
|
|
57
|
+
},
|
|
58
|
+
};
|
|
59
|
+
case "openai":
|
|
60
|
+
return { openai: { reasoningEffort: thinkingLevel.effort } };
|
|
61
|
+
case "google": {
|
|
62
|
+
return {
|
|
63
|
+
google: {
|
|
64
|
+
thinkingConfig: {
|
|
65
|
+
thinkingBudget: thinkingLevel.tokenBudget,
|
|
66
|
+
},
|
|
67
|
+
},
|
|
68
|
+
};
|
|
69
|
+
}
|
|
70
|
+
case "openrouter": {
|
|
71
|
+
return {
|
|
72
|
+
openrouter: {
|
|
73
|
+
reasoning: {
|
|
74
|
+
enabled: true,
|
|
75
|
+
effort: thinkingLevel.effort,
|
|
76
|
+
},
|
|
77
|
+
},
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
default:
|
|
81
|
+
return {};
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
// If supportsReasoning is false, or no provider case matched
|
|
85
|
+
return {};
|
|
86
|
+
}
|
|
87
|
+
}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import type { ModelMetadata } from "./providers.ts";
|
|
2
|
+
declare const anthropicModels: {
|
|
3
|
+
readonly opus: import("@ai-sdk/provider").LanguageModelV2;
|
|
4
|
+
readonly sonnet: import("@ai-sdk/provider").LanguageModelV2;
|
|
5
|
+
readonly sonnet37: import("@ai-sdk/provider").LanguageModelV2;
|
|
6
|
+
readonly "sonnet37-token-efficient-tools": import("@ai-sdk/provider").LanguageModelV2;
|
|
7
|
+
readonly "sonnet37-128k": import("@ai-sdk/provider").LanguageModelV2;
|
|
8
|
+
readonly sonnet35: import("@ai-sdk/provider").LanguageModelV2;
|
|
9
|
+
readonly haiku: import("@ai-sdk/provider").LanguageModelV2;
|
|
10
|
+
};
|
|
11
|
+
type ModelName = `anthropic:${keyof typeof anthropicModels}`;
|
|
12
|
+
export declare const anthropicModelNames: ModelName[];
|
|
13
|
+
export declare const anthropicProvider: {
|
|
14
|
+
anthropic: import("@ai-sdk/provider").ProviderV2 & {
|
|
15
|
+
languageModel(modelId: "opus" | "sonnet" | "sonnet37" | "sonnet37-token-efficient-tools" | "sonnet37-128k" | "sonnet35" | "haiku"): import("@ai-sdk/provider").LanguageModelV2;
|
|
16
|
+
textEmbeddingModel(modelId: string): import("@ai-sdk/provider").EmbeddingModelV2<string>;
|
|
17
|
+
imageModel(modelId: string): import("@ai-sdk/provider").ImageModelV2;
|
|
18
|
+
transcriptionModel(modelId: string): import("@ai-sdk/provider").TranscriptionModelV2;
|
|
19
|
+
speechModel(modelId: string): import("@ai-sdk/provider").SpeechModelV2;
|
|
20
|
+
};
|
|
21
|
+
};
|
|
22
|
+
export declare const anthropicModelRegistry: {
|
|
23
|
+
[K in ModelName]: ModelMetadata<ModelName>;
|
|
24
|
+
};
|
|
25
|
+
export {};
|
|
@@ -0,0 +1,184 @@
|
|
|
1
|
+
import { createAnthropic, anthropic as originalAnthropic, } from "@ai-sdk/anthropic";
|
|
2
|
+
import { objectKeys } from "@travisennis/stdlib/object";
|
|
3
|
+
import { isRecord } from "@travisennis/stdlib/typeguards";
|
|
4
|
+
import { customProvider } from "ai";
|
|
5
|
+
// Helper function copied from original providers.ts
|
|
6
|
+
function addCacheControlToTools(body) {
|
|
7
|
+
const parsedBody = JSON.parse(body);
|
|
8
|
+
if (isRecord(parsedBody)) {
|
|
9
|
+
const tools = parsedBody["tools"];
|
|
10
|
+
if (Array.isArray(tools)) {
|
|
11
|
+
tools.at(-1).cache_control = { type: "ephemeral" };
|
|
12
|
+
}
|
|
13
|
+
}
|
|
14
|
+
return JSON.stringify(parsedBody);
|
|
15
|
+
}
|
|
16
|
+
const anthropicModels = {
|
|
17
|
+
opus: createAnthropic({
|
|
18
|
+
fetch(input, init) {
|
|
19
|
+
const body = init?.body;
|
|
20
|
+
if (body && typeof body === "string") {
|
|
21
|
+
init.body = addCacheControlToTools(body);
|
|
22
|
+
}
|
|
23
|
+
return fetch(input, init);
|
|
24
|
+
},
|
|
25
|
+
})("claude-4-opus-20250514"),
|
|
26
|
+
sonnet: createAnthropic({
|
|
27
|
+
fetch(input, init) {
|
|
28
|
+
const body = init?.body;
|
|
29
|
+
if (body && typeof body === "string") {
|
|
30
|
+
init.body = addCacheControlToTools(body);
|
|
31
|
+
}
|
|
32
|
+
return fetch(input, init);
|
|
33
|
+
},
|
|
34
|
+
})("claude-4-sonnet-20250514"),
|
|
35
|
+
sonnet37: createAnthropic({
|
|
36
|
+
fetch(input, init) {
|
|
37
|
+
const body = init?.body;
|
|
38
|
+
if (body && typeof body === "string") {
|
|
39
|
+
init.body = addCacheControlToTools(body);
|
|
40
|
+
}
|
|
41
|
+
return fetch(input, init);
|
|
42
|
+
},
|
|
43
|
+
})("claude-3-7-sonnet-20250219"),
|
|
44
|
+
"sonnet37-token-efficient-tools": createAnthropic({
|
|
45
|
+
headers: {
|
|
46
|
+
"anthropic-version": "2023-06-01",
|
|
47
|
+
"anthropic-beta": "token-efficient-tools-2025-02-19",
|
|
48
|
+
},
|
|
49
|
+
fetch(input, init) {
|
|
50
|
+
const body = init?.body;
|
|
51
|
+
if (body && typeof body === "string") {
|
|
52
|
+
init.body = addCacheControlToTools(body);
|
|
53
|
+
}
|
|
54
|
+
return fetch(input, init);
|
|
55
|
+
},
|
|
56
|
+
})("claude-3-7-sonnet-20250219"),
|
|
57
|
+
"sonnet37-128k": createAnthropic({
|
|
58
|
+
headers: {
|
|
59
|
+
"anthropic-version": "2023-06-01",
|
|
60
|
+
"anthropic-beta": "output-128k-2025-02-19",
|
|
61
|
+
},
|
|
62
|
+
fetch(input, init) {
|
|
63
|
+
const body = init?.body;
|
|
64
|
+
if (body && typeof body === "string") {
|
|
65
|
+
init.body = addCacheControlToTools(body);
|
|
66
|
+
}
|
|
67
|
+
return fetch(input, init);
|
|
68
|
+
},
|
|
69
|
+
})("claude-3-7-sonnet-20250219"),
|
|
70
|
+
sonnet35: createAnthropic({
|
|
71
|
+
headers: {
|
|
72
|
+
"anthropic-version": "2023-06-01",
|
|
73
|
+
"anthropic-beta": "max-tokens-3-5-sonnet-2024-07-15",
|
|
74
|
+
},
|
|
75
|
+
fetch(input, init) {
|
|
76
|
+
const body = init?.body;
|
|
77
|
+
if (body && typeof body === "string") {
|
|
78
|
+
init.body = addCacheControlToTools(body);
|
|
79
|
+
}
|
|
80
|
+
return fetch(input, init);
|
|
81
|
+
},
|
|
82
|
+
})("claude-3-5-sonnet-20241022"),
|
|
83
|
+
haiku: originalAnthropic("claude-3-5-haiku-20241022"),
|
|
84
|
+
};
|
|
85
|
+
export const anthropicModelNames = objectKeys(anthropicModels).map((key) => `anthropic:${key}`);
|
|
86
|
+
export const anthropicProvider = {
|
|
87
|
+
anthropic: customProvider({
|
|
88
|
+
languageModels: anthropicModels,
|
|
89
|
+
fallbackProvider: originalAnthropic,
|
|
90
|
+
}),
|
|
91
|
+
};
|
|
92
|
+
export const anthropicModelRegistry = {
|
|
93
|
+
"anthropic:opus": {
|
|
94
|
+
id: "anthropic:opus",
|
|
95
|
+
provider: "anthropic",
|
|
96
|
+
contextWindow: 200000,
|
|
97
|
+
maxOutputTokens: 64000,
|
|
98
|
+
defaultTemperature: 0.3,
|
|
99
|
+
promptFormat: "xml",
|
|
100
|
+
supportsReasoning: true,
|
|
101
|
+
supportsToolCalling: true,
|
|
102
|
+
costPerInputToken: 0.000015,
|
|
103
|
+
costPerOutputToken: 0.000075,
|
|
104
|
+
category: "powerful",
|
|
105
|
+
},
|
|
106
|
+
"anthropic:sonnet": {
|
|
107
|
+
id: "anthropic:sonnet",
|
|
108
|
+
provider: "anthropic",
|
|
109
|
+
contextWindow: 200000,
|
|
110
|
+
maxOutputTokens: 64000,
|
|
111
|
+
defaultTemperature: 0.3,
|
|
112
|
+
promptFormat: "xml",
|
|
113
|
+
supportsReasoning: true,
|
|
114
|
+
supportsToolCalling: true,
|
|
115
|
+
costPerInputToken: 0.000003,
|
|
116
|
+
costPerOutputToken: 0.000015,
|
|
117
|
+
category: "balanced",
|
|
118
|
+
},
|
|
119
|
+
"anthropic:sonnet37": {
|
|
120
|
+
id: "anthropic:sonnet37",
|
|
121
|
+
provider: "anthropic",
|
|
122
|
+
contextWindow: 200000,
|
|
123
|
+
maxOutputTokens: 64000,
|
|
124
|
+
defaultTemperature: 0.3,
|
|
125
|
+
promptFormat: "xml",
|
|
126
|
+
supportsReasoning: true,
|
|
127
|
+
supportsToolCalling: true,
|
|
128
|
+
costPerInputToken: 0.000003,
|
|
129
|
+
costPerOutputToken: 0.000015,
|
|
130
|
+
category: "balanced",
|
|
131
|
+
},
|
|
132
|
+
"anthropic:sonnet37-token-efficient-tools": {
|
|
133
|
+
id: "anthropic:sonnet37-token-efficient-tools",
|
|
134
|
+
provider: "anthropic",
|
|
135
|
+
contextWindow: 200000,
|
|
136
|
+
maxOutputTokens: 64000,
|
|
137
|
+
defaultTemperature: 0.3,
|
|
138
|
+
promptFormat: "xml",
|
|
139
|
+
supportsReasoning: true,
|
|
140
|
+
supportsToolCalling: true,
|
|
141
|
+
costPerInputToken: 0.000003,
|
|
142
|
+
costPerOutputToken: 0.000015,
|
|
143
|
+
category: "balanced",
|
|
144
|
+
},
|
|
145
|
+
"anthropic:sonnet37-128k": {
|
|
146
|
+
id: "anthropic:sonnet37-128k",
|
|
147
|
+
provider: "anthropic",
|
|
148
|
+
contextWindow: 200000,
|
|
149
|
+
maxOutputTokens: 128000,
|
|
150
|
+
defaultTemperature: 0.3,
|
|
151
|
+
promptFormat: "xml",
|
|
152
|
+
supportsReasoning: true,
|
|
153
|
+
supportsToolCalling: true,
|
|
154
|
+
costPerInputToken: 0.000003,
|
|
155
|
+
costPerOutputToken: 0.000015,
|
|
156
|
+
category: "powerful",
|
|
157
|
+
},
|
|
158
|
+
"anthropic:sonnet35": {
|
|
159
|
+
id: "anthropic:sonnet35",
|
|
160
|
+
provider: "anthropic",
|
|
161
|
+
contextWindow: 200000,
|
|
162
|
+
maxOutputTokens: 8096,
|
|
163
|
+
defaultTemperature: 0.3,
|
|
164
|
+
promptFormat: "xml",
|
|
165
|
+
supportsReasoning: false,
|
|
166
|
+
supportsToolCalling: true,
|
|
167
|
+
costPerInputToken: 0.000003,
|
|
168
|
+
costPerOutputToken: 0.000015,
|
|
169
|
+
category: "balanced",
|
|
170
|
+
},
|
|
171
|
+
"anthropic:haiku": {
|
|
172
|
+
id: "anthropic:haiku",
|
|
173
|
+
provider: "anthropic",
|
|
174
|
+
contextWindow: 200000,
|
|
175
|
+
maxOutputTokens: 4096,
|
|
176
|
+
defaultTemperature: 0.3,
|
|
177
|
+
promptFormat: "xml",
|
|
178
|
+
supportsReasoning: false,
|
|
179
|
+
supportsToolCalling: true,
|
|
180
|
+
costPerInputToken: 0.0000008,
|
|
181
|
+
costPerOutputToken: 0.000004,
|
|
182
|
+
category: "fast",
|
|
183
|
+
},
|
|
184
|
+
};
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { ModelMetadata } from "./providers.ts";
|
|
2
|
+
declare const deepseekModels: {
|
|
3
|
+
readonly "deepseek-chat": import("@ai-sdk/provider").LanguageModelV2;
|
|
4
|
+
readonly "deepseek-reasoner": import("@ai-sdk/provider").LanguageModelV2;
|
|
5
|
+
};
|
|
6
|
+
type ModelName = `deepseek:${keyof typeof deepseekModels}`;
|
|
7
|
+
export declare const deepseekModelNames: ModelName[];
|
|
8
|
+
export declare const deepseekProvider: {
|
|
9
|
+
deepseek: import("@ai-sdk/provider").ProviderV2 & {
|
|
10
|
+
languageModel(modelId: "deepseek-chat" | "deepseek-reasoner"): import("@ai-sdk/provider").LanguageModelV2;
|
|
11
|
+
textEmbeddingModel(modelId: string): import("@ai-sdk/provider").EmbeddingModelV2<string>;
|
|
12
|
+
imageModel(modelId: string): import("@ai-sdk/provider").ImageModelV2;
|
|
13
|
+
transcriptionModel(modelId: string): import("@ai-sdk/provider").TranscriptionModelV2;
|
|
14
|
+
speechModel(modelId: string): import("@ai-sdk/provider").SpeechModelV2;
|
|
15
|
+
};
|
|
16
|
+
};
|
|
17
|
+
export declare const deepseekModelRegistry: {
|
|
18
|
+
[K in ModelName]: ModelMetadata<ModelName>;
|
|
19
|
+
};
|
|
20
|
+
export {};
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import { deepseek as originalDeepseek } from "@ai-sdk/deepseek";
|
|
2
|
+
import { objectKeys } from "@travisennis/stdlib/object";
|
|
3
|
+
import { customProvider } from "ai";
|
|
4
|
+
const deepseekModels = {
|
|
5
|
+
"deepseek-chat": originalDeepseek("deepseek-chat"),
|
|
6
|
+
"deepseek-reasoner": originalDeepseek("deepseek-reasoner"),
|
|
7
|
+
};
|
|
8
|
+
export const deepseekModelNames = objectKeys(deepseekModels).map((key) => `deepseek:${key}`);
|
|
9
|
+
export const deepseekProvider = {
|
|
10
|
+
deepseek: customProvider({
|
|
11
|
+
languageModels: deepseekModels,
|
|
12
|
+
fallbackProvider: originalDeepseek,
|
|
13
|
+
}),
|
|
14
|
+
};
|
|
15
|
+
export const deepseekModelRegistry = {
|
|
16
|
+
"deepseek:deepseek-chat": {
|
|
17
|
+
id: "deepseek:deepseek-chat",
|
|
18
|
+
provider: "deepseek",
|
|
19
|
+
contextWindow: 128000,
|
|
20
|
+
maxOutputTokens: 8000,
|
|
21
|
+
defaultTemperature: 0.3,
|
|
22
|
+
promptFormat: "bracket",
|
|
23
|
+
supportsReasoning: false,
|
|
24
|
+
supportsToolCalling: true,
|
|
25
|
+
costPerInputToken: 0.0000012, // Check official pricing
|
|
26
|
+
costPerOutputToken: 0.0000012, // Check official pricing
|
|
27
|
+
category: "balanced",
|
|
28
|
+
},
|
|
29
|
+
"deepseek:deepseek-reasoner": {
|
|
30
|
+
id: "deepseek:deepseek-reasoner",
|
|
31
|
+
provider: "deepseek",
|
|
32
|
+
contextWindow: 128000,
|
|
33
|
+
maxOutputTokens: 32768,
|
|
34
|
+
defaultTemperature: 0.6,
|
|
35
|
+
promptFormat: "bracket",
|
|
36
|
+
supportsReasoning: true,
|
|
37
|
+
supportsToolCalling: true, // Check if this model supports tools
|
|
38
|
+
costPerInputToken: 0.00000055, // Check official pricing
|
|
39
|
+
costPerOutputToken: 0.00000219, // Check official pricing
|
|
40
|
+
category: "balanced",
|
|
41
|
+
},
|
|
42
|
+
};
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { ModelMetadata } from "./providers.ts";
|
|
2
|
+
declare const googleModels: {
|
|
3
|
+
readonly flash25lite: import("@ai-sdk/provider").LanguageModelV2;
|
|
4
|
+
readonly pro25: import("@ai-sdk/provider").LanguageModelV2;
|
|
5
|
+
readonly flash25: import("@ai-sdk/provider").LanguageModelV2;
|
|
6
|
+
};
|
|
7
|
+
type ModelName = `google:${keyof typeof googleModels}`;
|
|
8
|
+
export declare const googleModelNames: ModelName[];
|
|
9
|
+
export declare const googleProvider: {
|
|
10
|
+
google: import("@ai-sdk/provider").ProviderV2 & {
|
|
11
|
+
languageModel(modelId: "flash25lite" | "pro25" | "flash25"): import("@ai-sdk/provider").LanguageModelV2;
|
|
12
|
+
textEmbeddingModel(modelId: string): import("@ai-sdk/provider").EmbeddingModelV2<string>;
|
|
13
|
+
imageModel(modelId: string): import("@ai-sdk/provider").ImageModelV2;
|
|
14
|
+
transcriptionModel(modelId: string): import("@ai-sdk/provider").TranscriptionModelV2;
|
|
15
|
+
speechModel(modelId: string): import("@ai-sdk/provider").SpeechModelV2;
|
|
16
|
+
};
|
|
17
|
+
};
|
|
18
|
+
export declare const googleModelRegistry: Record<ModelName, ModelMetadata<ModelName>>;
|
|
19
|
+
export {};
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
import { google as originalGoogle } from "@ai-sdk/google";
|
|
2
|
+
import { objectKeys } from "@travisennis/stdlib/object";
|
|
3
|
+
import { customProvider } from "ai";
|
|
4
|
+
const googleModels = {
|
|
5
|
+
flash25lite: originalGoogle("gemini-2.5-flash-lite-preview-06-17"),
|
|
6
|
+
pro25: originalGoogle("gemini-2.5-pro"),
|
|
7
|
+
flash25: originalGoogle("gemini-2.5-flash"),
|
|
8
|
+
};
|
|
9
|
+
export const googleModelNames = objectKeys(googleModels).map((key) => `google:${key}`);
|
|
10
|
+
export const googleProvider = {
|
|
11
|
+
google: customProvider({
|
|
12
|
+
languageModels: googleModels,
|
|
13
|
+
fallbackProvider: originalGoogle,
|
|
14
|
+
}),
|
|
15
|
+
};
|
|
16
|
+
export const googleModelRegistry = {
|
|
17
|
+
"google:flash25lite": {
|
|
18
|
+
id: "google:flash25lite",
|
|
19
|
+
provider: "google",
|
|
20
|
+
contextWindow: 1000000,
|
|
21
|
+
maxOutputTokens: 8192,
|
|
22
|
+
defaultTemperature: 0.3,
|
|
23
|
+
promptFormat: "markdown",
|
|
24
|
+
supportsReasoning: false,
|
|
25
|
+
supportsToolCalling: true,
|
|
26
|
+
costPerInputToken: 0,
|
|
27
|
+
costPerOutputToken: 0,
|
|
28
|
+
category: "fast",
|
|
29
|
+
},
|
|
30
|
+
"google:pro25": {
|
|
31
|
+
id: "google:pro25",
|
|
32
|
+
provider: "google",
|
|
33
|
+
contextWindow: 1000000,
|
|
34
|
+
maxOutputTokens: 64000,
|
|
35
|
+
defaultTemperature: 0.7,
|
|
36
|
+
promptFormat: "markdown",
|
|
37
|
+
supportsReasoning: true,
|
|
38
|
+
supportsToolCalling: true,
|
|
39
|
+
costPerInputToken: 0,
|
|
40
|
+
costPerOutputToken: 0,
|
|
41
|
+
category: "powerful",
|
|
42
|
+
},
|
|
43
|
+
"google:flash25": {
|
|
44
|
+
id: "google:flash25",
|
|
45
|
+
provider: "google",
|
|
46
|
+
contextWindow: 1048576,
|
|
47
|
+
maxOutputTokens: 65536,
|
|
48
|
+
defaultTemperature: 0.3,
|
|
49
|
+
promptFormat: "markdown",
|
|
50
|
+
supportsReasoning: true,
|
|
51
|
+
supportsToolCalling: true,
|
|
52
|
+
costPerInputToken: 0.00000015,
|
|
53
|
+
costPerOutputToken: 0.0000035,
|
|
54
|
+
category: "balanced",
|
|
55
|
+
},
|
|
56
|
+
};
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { LanguageModelV2 } from "@ai-sdk/provider";
|
|
2
|
+
import { type ModelMetadata, type ModelName } from "./providers.ts";
|
|
3
|
+
type App = "repl" | "cli" | "title-conversation" | "conversation-summarizer" | "conversation-analyzer" | "tool-repair" | "init-project" | "task-agent";
|
|
4
|
+
export declare class ModelManager {
|
|
5
|
+
private modelMap;
|
|
6
|
+
private modelMetadataMap;
|
|
7
|
+
private stateDir;
|
|
8
|
+
constructor({ stateDir }: {
|
|
9
|
+
stateDir: string;
|
|
10
|
+
});
|
|
11
|
+
setModel(app: App, model: ModelName): void;
|
|
12
|
+
getModel(app: App): LanguageModelV2;
|
|
13
|
+
getModelMetadata(app: App): ModelMetadata<ModelName>;
|
|
14
|
+
}
|
|
15
|
+
export {};
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
import { wrapLanguageModel } from "ai";
|
|
2
|
+
import { auditMessage, createRateLimitMiddleware, } from "../middleware/index.js";
|
|
3
|
+
import { languageModel, modelRegistry, } from "./providers.js";
|
|
4
|
+
function getLanguageModel({ model, app, stateDir, }) {
|
|
5
|
+
const langModel = wrapLanguageModel({
|
|
6
|
+
model: languageModel(model),
|
|
7
|
+
middleware: [
|
|
8
|
+
createRateLimitMiddleware({ requestsPerMinute: 30 }),
|
|
9
|
+
auditMessage({ filePath: stateDir, app }),
|
|
10
|
+
],
|
|
11
|
+
});
|
|
12
|
+
return langModel;
|
|
13
|
+
}
|
|
14
|
+
export class ModelManager {
|
|
15
|
+
modelMap;
|
|
16
|
+
modelMetadataMap;
|
|
17
|
+
stateDir;
|
|
18
|
+
constructor({ stateDir }) {
|
|
19
|
+
this.modelMap = new Map();
|
|
20
|
+
this.modelMetadataMap = new Map();
|
|
21
|
+
this.stateDir = stateDir;
|
|
22
|
+
}
|
|
23
|
+
setModel(app, model) {
|
|
24
|
+
this.modelMap.set(app, getLanguageModel({
|
|
25
|
+
model,
|
|
26
|
+
app,
|
|
27
|
+
stateDir: this.stateDir,
|
|
28
|
+
}));
|
|
29
|
+
const modelMetadata = modelRegistry[model];
|
|
30
|
+
if (modelMetadata) {
|
|
31
|
+
this.modelMetadataMap.set(app, modelMetadata);
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
getModel(app) {
|
|
35
|
+
const model = this.modelMap.get(app);
|
|
36
|
+
if (!model) {
|
|
37
|
+
throw new Error("Model not initialized.");
|
|
38
|
+
}
|
|
39
|
+
return model;
|
|
40
|
+
}
|
|
41
|
+
getModelMetadata(app) {
|
|
42
|
+
const metadata = this.modelMetadataMap.get(app);
|
|
43
|
+
if (!metadata) {
|
|
44
|
+
throw new Error("Model not initialized.");
|
|
45
|
+
}
|
|
46
|
+
return metadata;
|
|
47
|
+
}
|
|
48
|
+
}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { ModelMetadata } from "./providers.ts";
|
|
2
|
+
declare const openaiModels: {
|
|
3
|
+
readonly "gpt-4.1": import("@ai-sdk/provider").LanguageModelV2;
|
|
4
|
+
readonly o3: import("@ai-sdk/provider").LanguageModelV2;
|
|
5
|
+
readonly "o4-mini": import("@ai-sdk/provider").LanguageModelV2;
|
|
6
|
+
readonly "codex-mini": import("@ai-sdk/provider").LanguageModelV2;
|
|
7
|
+
};
|
|
8
|
+
type ModelName = `openai:${keyof typeof openaiModels}`;
|
|
9
|
+
export declare const openaiModelNames: ModelName[];
|
|
10
|
+
export declare const openaiProvider: {
|
|
11
|
+
openai: import("@ai-sdk/provider").ProviderV2 & {
|
|
12
|
+
languageModel(modelId: "o3" | "gpt-4.1" | "o4-mini" | "codex-mini"): import("@ai-sdk/provider").LanguageModelV2;
|
|
13
|
+
textEmbeddingModel(modelId: string): import("@ai-sdk/provider").EmbeddingModelV2<string>;
|
|
14
|
+
imageModel(modelId: string): import("@ai-sdk/provider").ImageModelV2;
|
|
15
|
+
transcriptionModel(modelId: string): import("@ai-sdk/provider").TranscriptionModelV2;
|
|
16
|
+
speechModel(modelId: string): import("@ai-sdk/provider").SpeechModelV2;
|
|
17
|
+
};
|
|
18
|
+
};
|
|
19
|
+
export declare const openaiModelRegistry: {
|
|
20
|
+
[K in ModelName]: ModelMetadata<ModelName>;
|
|
21
|
+
};
|
|
22
|
+
export {};
|
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
import { openai as originalOpenAi } from "@ai-sdk/openai";
|
|
2
|
+
import { objectKeys } from "@travisennis/stdlib/object";
|
|
3
|
+
import { customProvider } from "ai";
|
|
4
|
+
const openaiModels = {
|
|
5
|
+
"gpt-4.1": originalOpenAi("gpt-4.1"),
|
|
6
|
+
o3: originalOpenAi.responses("o3"),
|
|
7
|
+
"o4-mini": originalOpenAi.responses("o4-mini"),
|
|
8
|
+
"codex-mini": originalOpenAi("codex-mini-latest"),
|
|
9
|
+
};
|
|
10
|
+
export const openaiModelNames = objectKeys(openaiModels).map((key) => `openai:${key}`);
|
|
11
|
+
export const openaiProvider = {
|
|
12
|
+
openai: customProvider({
|
|
13
|
+
languageModels: openaiModels,
|
|
14
|
+
fallbackProvider: originalOpenAi,
|
|
15
|
+
}),
|
|
16
|
+
};
|
|
17
|
+
export const openaiModelRegistry = {
|
|
18
|
+
"openai:gpt-4.1": {
|
|
19
|
+
id: "openai:gpt-4.1",
|
|
20
|
+
provider: "openai",
|
|
21
|
+
contextWindow: 1000000,
|
|
22
|
+
maxOutputTokens: 32768,
|
|
23
|
+
defaultTemperature: 0.3,
|
|
24
|
+
promptFormat: "markdown",
|
|
25
|
+
supportsReasoning: false,
|
|
26
|
+
supportsToolCalling: true,
|
|
27
|
+
costPerInputToken: 0.000002,
|
|
28
|
+
costPerOutputToken: 0.000008,
|
|
29
|
+
category: "balanced",
|
|
30
|
+
},
|
|
31
|
+
"openai:o3": {
|
|
32
|
+
id: "openai:o3",
|
|
33
|
+
provider: "openai",
|
|
34
|
+
contextWindow: 200000,
|
|
35
|
+
maxOutputTokens: 100000,
|
|
36
|
+
defaultTemperature: 0.3,
|
|
37
|
+
promptFormat: "markdown",
|
|
38
|
+
supportsReasoning: true,
|
|
39
|
+
supportsToolCalling: true,
|
|
40
|
+
costPerInputToken: 0.00001,
|
|
41
|
+
costPerOutputToken: 0.00004,
|
|
42
|
+
category: "powerful",
|
|
43
|
+
},
|
|
44
|
+
"openai:o4-mini": {
|
|
45
|
+
id: "openai:o4-mini",
|
|
46
|
+
provider: "openai",
|
|
47
|
+
contextWindow: 200000,
|
|
48
|
+
maxOutputTokens: 100000,
|
|
49
|
+
defaultTemperature: 0.3,
|
|
50
|
+
promptFormat: "markdown",
|
|
51
|
+
supportsReasoning: true,
|
|
52
|
+
supportsToolCalling: true,
|
|
53
|
+
costPerInputToken: 0.0000011,
|
|
54
|
+
costPerOutputToken: 0.0000044,
|
|
55
|
+
category: "balanced",
|
|
56
|
+
},
|
|
57
|
+
"openai:codex-mini": {
|
|
58
|
+
id: "openai:codex-mini",
|
|
59
|
+
provider: "openai",
|
|
60
|
+
contextWindow: 200000,
|
|
61
|
+
maxOutputTokens: 100000,
|
|
62
|
+
defaultTemperature: 0.3,
|
|
63
|
+
promptFormat: "markdown",
|
|
64
|
+
supportsReasoning: false,
|
|
65
|
+
supportsToolCalling: true,
|
|
66
|
+
costPerInputToken: 0.0000015,
|
|
67
|
+
costPerOutputToken: 0.000006,
|
|
68
|
+
category: "balanced",
|
|
69
|
+
},
|
|
70
|
+
};
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import type { ModelMetadata } from "./providers.ts";
|
|
2
|
+
declare const openrouterModels: {
|
|
3
|
+
readonly "deepseek-v3": import("@ai-sdk/provider").LanguageModelV2;
|
|
4
|
+
readonly "deepseek-r1": import("@ai-sdk/provider").LanguageModelV2;
|
|
5
|
+
readonly "deepseek-v3-free": import("@ai-sdk/provider").LanguageModelV2;
|
|
6
|
+
readonly "deepseek-r1-free": import("@ai-sdk/provider").LanguageModelV2;
|
|
7
|
+
readonly "gemini-flash25": import("@ai-sdk/provider").LanguageModelV2;
|
|
8
|
+
readonly "gemini-pro25": import("@ai-sdk/provider").LanguageModelV2;
|
|
9
|
+
readonly sonnet4: import("@ai-sdk/provider").LanguageModelV2;
|
|
10
|
+
readonly opus4: import("@ai-sdk/provider").LanguageModelV2;
|
|
11
|
+
readonly "opus-4.1": import("@ai-sdk/provider").LanguageModelV2;
|
|
12
|
+
readonly "gpt-4.1": import("@ai-sdk/provider").LanguageModelV2;
|
|
13
|
+
readonly "kimi-k2": import("@ai-sdk/provider").LanguageModelV2;
|
|
14
|
+
readonly "kimi-k2-free": import("@ai-sdk/provider").LanguageModelV2;
|
|
15
|
+
readonly "devstral-medium": import("@ai-sdk/provider").LanguageModelV2;
|
|
16
|
+
readonly "qwen3-coder": import("@ai-sdk/provider").LanguageModelV2;
|
|
17
|
+
readonly "qwen3-coder-free": import("@ai-sdk/provider").LanguageModelV2;
|
|
18
|
+
readonly "glm-4.5": import("@ai-sdk/provider").LanguageModelV2;
|
|
19
|
+
readonly "gpt-5": import("@ai-sdk/provider").LanguageModelV2;
|
|
20
|
+
readonly "gpt-5-mini": import("@ai-sdk/provider").LanguageModelV2;
|
|
21
|
+
};
|
|
22
|
+
type ModelName = `openrouter:${keyof typeof openrouterModels}`;
|
|
23
|
+
export declare const openrouterModelNames: ModelName[];
|
|
24
|
+
export declare const openrouterProvider: {
|
|
25
|
+
openrouter: import("@ai-sdk/provider").ProviderV2 & {
|
|
26
|
+
languageModel(modelId: "gpt-5" | "gpt-5-mini" | "gpt-4.1" | "deepseek-v3" | "deepseek-r1" | "deepseek-v3-free" | "deepseek-r1-free" | "gemini-flash25" | "gemini-pro25" | "sonnet4" | "opus4" | "opus-4.1" | "kimi-k2" | "kimi-k2-free" | "devstral-medium" | "qwen3-coder" | "qwen3-coder-free" | "glm-4.5"): import("@ai-sdk/provider").LanguageModelV2;
|
|
27
|
+
textEmbeddingModel(modelId: string): import("@ai-sdk/provider").EmbeddingModelV2<string>;
|
|
28
|
+
imageModel(modelId: string): import("@ai-sdk/provider").ImageModelV2;
|
|
29
|
+
transcriptionModel(modelId: string): import("@ai-sdk/provider").TranscriptionModelV2;
|
|
30
|
+
speechModel(modelId: string): import("@ai-sdk/provider").SpeechModelV2;
|
|
31
|
+
};
|
|
32
|
+
};
|
|
33
|
+
export declare const openrouterModelRegistry: {
|
|
34
|
+
[K in ModelName]: ModelMetadata<ModelName>;
|
|
35
|
+
};
|
|
36
|
+
export {};
|