@shareai-lab/kode 1.0.9

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (286) hide show
  1. package/LICENSE +201 -0
  2. package/README.md +426 -0
  3. package/README.zh-CN.md +326 -0
  4. package/cli.js +79 -0
  5. package/package.json +119 -0
  6. package/scripts/postinstall.js +18 -0
  7. package/src/ProjectOnboarding.tsx +198 -0
  8. package/src/Tool.ts +82 -0
  9. package/src/commands/agents.tsx +3410 -0
  10. package/src/commands/approvedTools.ts +53 -0
  11. package/src/commands/bug.tsx +20 -0
  12. package/src/commands/clear.ts +43 -0
  13. package/src/commands/compact.ts +120 -0
  14. package/src/commands/config.tsx +19 -0
  15. package/src/commands/cost.ts +18 -0
  16. package/src/commands/ctx_viz.ts +209 -0
  17. package/src/commands/doctor.ts +24 -0
  18. package/src/commands/help.tsx +19 -0
  19. package/src/commands/init.ts +37 -0
  20. package/src/commands/listen.ts +42 -0
  21. package/src/commands/login.tsx +51 -0
  22. package/src/commands/logout.tsx +40 -0
  23. package/src/commands/mcp.ts +41 -0
  24. package/src/commands/model.tsx +40 -0
  25. package/src/commands/modelstatus.tsx +20 -0
  26. package/src/commands/onboarding.tsx +34 -0
  27. package/src/commands/pr_comments.ts +59 -0
  28. package/src/commands/refreshCommands.ts +54 -0
  29. package/src/commands/release-notes.ts +34 -0
  30. package/src/commands/resume.tsx +31 -0
  31. package/src/commands/review.ts +49 -0
  32. package/src/commands/terminalSetup.ts +221 -0
  33. package/src/commands.ts +139 -0
  34. package/src/components/ApproveApiKey.tsx +93 -0
  35. package/src/components/AsciiLogo.tsx +13 -0
  36. package/src/components/AutoUpdater.tsx +148 -0
  37. package/src/components/Bug.tsx +367 -0
  38. package/src/components/Config.tsx +293 -0
  39. package/src/components/ConsoleOAuthFlow.tsx +327 -0
  40. package/src/components/Cost.tsx +23 -0
  41. package/src/components/CostThresholdDialog.tsx +46 -0
  42. package/src/components/CustomSelect/option-map.ts +42 -0
  43. package/src/components/CustomSelect/select-option.tsx +78 -0
  44. package/src/components/CustomSelect/select.tsx +152 -0
  45. package/src/components/CustomSelect/theme.ts +45 -0
  46. package/src/components/CustomSelect/use-select-state.ts +414 -0
  47. package/src/components/CustomSelect/use-select.ts +35 -0
  48. package/src/components/FallbackToolUseRejectedMessage.tsx +15 -0
  49. package/src/components/FileEditToolUpdatedMessage.tsx +66 -0
  50. package/src/components/Help.tsx +215 -0
  51. package/src/components/HighlightedCode.tsx +33 -0
  52. package/src/components/InvalidConfigDialog.tsx +113 -0
  53. package/src/components/Link.tsx +32 -0
  54. package/src/components/LogSelector.tsx +86 -0
  55. package/src/components/Logo.tsx +170 -0
  56. package/src/components/MCPServerApprovalDialog.tsx +100 -0
  57. package/src/components/MCPServerDialogCopy.tsx +25 -0
  58. package/src/components/MCPServerMultiselectDialog.tsx +109 -0
  59. package/src/components/Message.tsx +221 -0
  60. package/src/components/MessageResponse.tsx +15 -0
  61. package/src/components/MessageSelector.tsx +211 -0
  62. package/src/components/ModeIndicator.tsx +88 -0
  63. package/src/components/ModelConfig.tsx +301 -0
  64. package/src/components/ModelListManager.tsx +227 -0
  65. package/src/components/ModelSelector.tsx +3387 -0
  66. package/src/components/ModelStatusDisplay.tsx +230 -0
  67. package/src/components/Onboarding.tsx +274 -0
  68. package/src/components/PressEnterToContinue.tsx +11 -0
  69. package/src/components/PromptInput.tsx +760 -0
  70. package/src/components/SentryErrorBoundary.ts +39 -0
  71. package/src/components/Spinner.tsx +129 -0
  72. package/src/components/StickerRequestForm.tsx +16 -0
  73. package/src/components/StructuredDiff.tsx +191 -0
  74. package/src/components/TextInput.tsx +259 -0
  75. package/src/components/TodoItem.tsx +47 -0
  76. package/src/components/TokenWarning.tsx +31 -0
  77. package/src/components/ToolUseLoader.tsx +40 -0
  78. package/src/components/TrustDialog.tsx +106 -0
  79. package/src/components/binary-feedback/BinaryFeedback.tsx +63 -0
  80. package/src/components/binary-feedback/BinaryFeedbackOption.tsx +111 -0
  81. package/src/components/binary-feedback/BinaryFeedbackView.tsx +172 -0
  82. package/src/components/binary-feedback/utils.ts +220 -0
  83. package/src/components/messages/AssistantBashOutputMessage.tsx +22 -0
  84. package/src/components/messages/AssistantLocalCommandOutputMessage.tsx +49 -0
  85. package/src/components/messages/AssistantRedactedThinkingMessage.tsx +19 -0
  86. package/src/components/messages/AssistantTextMessage.tsx +144 -0
  87. package/src/components/messages/AssistantThinkingMessage.tsx +40 -0
  88. package/src/components/messages/AssistantToolUseMessage.tsx +133 -0
  89. package/src/components/messages/TaskProgressMessage.tsx +32 -0
  90. package/src/components/messages/TaskToolMessage.tsx +58 -0
  91. package/src/components/messages/UserBashInputMessage.tsx +28 -0
  92. package/src/components/messages/UserCommandMessage.tsx +30 -0
  93. package/src/components/messages/UserKodingInputMessage.tsx +28 -0
  94. package/src/components/messages/UserPromptMessage.tsx +35 -0
  95. package/src/components/messages/UserTextMessage.tsx +39 -0
  96. package/src/components/messages/UserToolResultMessage/UserToolCanceledMessage.tsx +12 -0
  97. package/src/components/messages/UserToolResultMessage/UserToolErrorMessage.tsx +36 -0
  98. package/src/components/messages/UserToolResultMessage/UserToolRejectMessage.tsx +31 -0
  99. package/src/components/messages/UserToolResultMessage/UserToolResultMessage.tsx +57 -0
  100. package/src/components/messages/UserToolResultMessage/UserToolSuccessMessage.tsx +35 -0
  101. package/src/components/messages/UserToolResultMessage/utils.tsx +56 -0
  102. package/src/components/permissions/BashPermissionRequest/BashPermissionRequest.tsx +121 -0
  103. package/src/components/permissions/FallbackPermissionRequest.tsx +153 -0
  104. package/src/components/permissions/FileEditPermissionRequest/FileEditPermissionRequest.tsx +182 -0
  105. package/src/components/permissions/FileEditPermissionRequest/FileEditToolDiff.tsx +77 -0
  106. package/src/components/permissions/FileWritePermissionRequest/FileWritePermissionRequest.tsx +164 -0
  107. package/src/components/permissions/FileWritePermissionRequest/FileWriteToolDiff.tsx +83 -0
  108. package/src/components/permissions/FilesystemPermissionRequest/FilesystemPermissionRequest.tsx +240 -0
  109. package/src/components/permissions/PermissionRequest.tsx +101 -0
  110. package/src/components/permissions/PermissionRequestTitle.tsx +69 -0
  111. package/src/components/permissions/hooks.ts +44 -0
  112. package/src/components/permissions/toolUseOptions.ts +59 -0
  113. package/src/components/permissions/utils.ts +23 -0
  114. package/src/constants/betas.ts +5 -0
  115. package/src/constants/claude-asterisk-ascii-art.tsx +238 -0
  116. package/src/constants/figures.ts +4 -0
  117. package/src/constants/keys.ts +3 -0
  118. package/src/constants/macros.ts +8 -0
  119. package/src/constants/modelCapabilities.ts +179 -0
  120. package/src/constants/models.ts +1025 -0
  121. package/src/constants/oauth.ts +18 -0
  122. package/src/constants/product.ts +17 -0
  123. package/src/constants/prompts.ts +168 -0
  124. package/src/constants/releaseNotes.ts +7 -0
  125. package/src/context/PermissionContext.tsx +149 -0
  126. package/src/context.ts +278 -0
  127. package/src/cost-tracker.ts +84 -0
  128. package/src/entrypoints/cli.tsx +1542 -0
  129. package/src/entrypoints/mcp.ts +176 -0
  130. package/src/history.ts +25 -0
  131. package/src/hooks/useApiKeyVerification.ts +59 -0
  132. package/src/hooks/useArrowKeyHistory.ts +55 -0
  133. package/src/hooks/useCanUseTool.ts +138 -0
  134. package/src/hooks/useCancelRequest.ts +39 -0
  135. package/src/hooks/useDoublePress.ts +42 -0
  136. package/src/hooks/useExitOnCtrlCD.ts +31 -0
  137. package/src/hooks/useInterval.ts +25 -0
  138. package/src/hooks/useLogMessages.ts +16 -0
  139. package/src/hooks/useLogStartupTime.ts +12 -0
  140. package/src/hooks/useNotifyAfterTimeout.ts +65 -0
  141. package/src/hooks/usePermissionRequestLogging.ts +44 -0
  142. package/src/hooks/useTerminalSize.ts +49 -0
  143. package/src/hooks/useTextInput.ts +318 -0
  144. package/src/hooks/useUnifiedCompletion.ts +1405 -0
  145. package/src/messages.ts +38 -0
  146. package/src/permissions.ts +268 -0
  147. package/src/query.ts +715 -0
  148. package/src/screens/ConfigureNpmPrefix.tsx +197 -0
  149. package/src/screens/Doctor.tsx +219 -0
  150. package/src/screens/LogList.tsx +68 -0
  151. package/src/screens/REPL.tsx +809 -0
  152. package/src/screens/ResumeConversation.tsx +68 -0
  153. package/src/services/adapters/base.ts +38 -0
  154. package/src/services/adapters/chatCompletions.ts +90 -0
  155. package/src/services/adapters/responsesAPI.ts +170 -0
  156. package/src/services/browserMocks.ts +66 -0
  157. package/src/services/claude.ts +2197 -0
  158. package/src/services/customCommands.ts +704 -0
  159. package/src/services/fileFreshness.ts +377 -0
  160. package/src/services/gpt5ConnectionTest.ts +340 -0
  161. package/src/services/mcpClient.ts +564 -0
  162. package/src/services/mcpServerApproval.tsx +50 -0
  163. package/src/services/mentionProcessor.ts +273 -0
  164. package/src/services/modelAdapterFactory.ts +69 -0
  165. package/src/services/notifier.ts +40 -0
  166. package/src/services/oauth.ts +357 -0
  167. package/src/services/openai.ts +1338 -0
  168. package/src/services/responseStateManager.ts +90 -0
  169. package/src/services/sentry.ts +3 -0
  170. package/src/services/statsig.ts +172 -0
  171. package/src/services/statsigStorage.ts +86 -0
  172. package/src/services/systemReminder.ts +507 -0
  173. package/src/services/vcr.ts +161 -0
  174. package/src/test/testAdapters.ts +96 -0
  175. package/src/tools/ArchitectTool/ArchitectTool.tsx +122 -0
  176. package/src/tools/ArchitectTool/prompt.ts +15 -0
  177. package/src/tools/AskExpertModelTool/AskExpertModelTool.tsx +569 -0
  178. package/src/tools/BashTool/BashTool.tsx +243 -0
  179. package/src/tools/BashTool/BashToolResultMessage.tsx +38 -0
  180. package/src/tools/BashTool/OutputLine.tsx +49 -0
  181. package/src/tools/BashTool/prompt.ts +174 -0
  182. package/src/tools/BashTool/utils.ts +56 -0
  183. package/src/tools/FileEditTool/FileEditTool.tsx +315 -0
  184. package/src/tools/FileEditTool/prompt.ts +51 -0
  185. package/src/tools/FileEditTool/utils.ts +58 -0
  186. package/src/tools/FileReadTool/FileReadTool.tsx +404 -0
  187. package/src/tools/FileReadTool/prompt.ts +7 -0
  188. package/src/tools/FileWriteTool/FileWriteTool.tsx +297 -0
  189. package/src/tools/FileWriteTool/prompt.ts +10 -0
  190. package/src/tools/GlobTool/GlobTool.tsx +119 -0
  191. package/src/tools/GlobTool/prompt.ts +8 -0
  192. package/src/tools/GrepTool/GrepTool.tsx +147 -0
  193. package/src/tools/GrepTool/prompt.ts +11 -0
  194. package/src/tools/MCPTool/MCPTool.tsx +107 -0
  195. package/src/tools/MCPTool/prompt.ts +3 -0
  196. package/src/tools/MemoryReadTool/MemoryReadTool.tsx +127 -0
  197. package/src/tools/MemoryReadTool/prompt.ts +3 -0
  198. package/src/tools/MemoryWriteTool/MemoryWriteTool.tsx +89 -0
  199. package/src/tools/MemoryWriteTool/prompt.ts +3 -0
  200. package/src/tools/MultiEditTool/MultiEditTool.tsx +366 -0
  201. package/src/tools/MultiEditTool/prompt.ts +45 -0
  202. package/src/tools/NotebookEditTool/NotebookEditTool.tsx +298 -0
  203. package/src/tools/NotebookEditTool/prompt.ts +3 -0
  204. package/src/tools/NotebookReadTool/NotebookReadTool.tsx +258 -0
  205. package/src/tools/NotebookReadTool/prompt.ts +3 -0
  206. package/src/tools/StickerRequestTool/StickerRequestTool.tsx +93 -0
  207. package/src/tools/StickerRequestTool/prompt.ts +19 -0
  208. package/src/tools/TaskTool/TaskTool.tsx +466 -0
  209. package/src/tools/TaskTool/constants.ts +1 -0
  210. package/src/tools/TaskTool/prompt.ts +92 -0
  211. package/src/tools/ThinkTool/ThinkTool.tsx +54 -0
  212. package/src/tools/ThinkTool/prompt.ts +12 -0
  213. package/src/tools/TodoWriteTool/TodoWriteTool.tsx +313 -0
  214. package/src/tools/TodoWriteTool/prompt.ts +63 -0
  215. package/src/tools/URLFetcherTool/URLFetcherTool.tsx +178 -0
  216. package/src/tools/URLFetcherTool/cache.ts +55 -0
  217. package/src/tools/URLFetcherTool/htmlToMarkdown.ts +55 -0
  218. package/src/tools/URLFetcherTool/prompt.ts +17 -0
  219. package/src/tools/WebSearchTool/WebSearchTool.tsx +103 -0
  220. package/src/tools/WebSearchTool/prompt.ts +13 -0
  221. package/src/tools/WebSearchTool/searchProviders.ts +66 -0
  222. package/src/tools/lsTool/lsTool.tsx +272 -0
  223. package/src/tools/lsTool/prompt.ts +2 -0
  224. package/src/tools.ts +67 -0
  225. package/src/types/PermissionMode.ts +120 -0
  226. package/src/types/RequestContext.ts +72 -0
  227. package/src/types/conversation.ts +51 -0
  228. package/src/types/logs.ts +58 -0
  229. package/src/types/modelCapabilities.ts +64 -0
  230. package/src/types/notebook.ts +87 -0
  231. package/src/utils/Cursor.ts +436 -0
  232. package/src/utils/PersistentShell.ts +552 -0
  233. package/src/utils/advancedFuzzyMatcher.ts +290 -0
  234. package/src/utils/agentLoader.ts +278 -0
  235. package/src/utils/agentStorage.ts +97 -0
  236. package/src/utils/array.ts +3 -0
  237. package/src/utils/ask.tsx +99 -0
  238. package/src/utils/auth.ts +13 -0
  239. package/src/utils/autoCompactCore.ts +223 -0
  240. package/src/utils/autoUpdater.ts +458 -0
  241. package/src/utils/betas.ts +20 -0
  242. package/src/utils/browser.ts +14 -0
  243. package/src/utils/cleanup.ts +72 -0
  244. package/src/utils/commands.ts +261 -0
  245. package/src/utils/commonUnixCommands.ts +161 -0
  246. package/src/utils/config.ts +945 -0
  247. package/src/utils/conversationRecovery.ts +55 -0
  248. package/src/utils/debugLogger.ts +1235 -0
  249. package/src/utils/diff.ts +42 -0
  250. package/src/utils/env.ts +57 -0
  251. package/src/utils/errors.ts +21 -0
  252. package/src/utils/exampleCommands.ts +109 -0
  253. package/src/utils/execFileNoThrow.ts +51 -0
  254. package/src/utils/expertChatStorage.ts +136 -0
  255. package/src/utils/file.ts +405 -0
  256. package/src/utils/fileRecoveryCore.ts +71 -0
  257. package/src/utils/format.tsx +44 -0
  258. package/src/utils/fuzzyMatcher.ts +328 -0
  259. package/src/utils/generators.ts +62 -0
  260. package/src/utils/git.ts +92 -0
  261. package/src/utils/globalLogger.ts +77 -0
  262. package/src/utils/http.ts +10 -0
  263. package/src/utils/imagePaste.ts +38 -0
  264. package/src/utils/json.ts +13 -0
  265. package/src/utils/log.ts +382 -0
  266. package/src/utils/markdown.ts +213 -0
  267. package/src/utils/messageContextManager.ts +289 -0
  268. package/src/utils/messages.tsx +939 -0
  269. package/src/utils/model.ts +914 -0
  270. package/src/utils/permissions/filesystem.ts +127 -0
  271. package/src/utils/responseState.ts +23 -0
  272. package/src/utils/ripgrep.ts +167 -0
  273. package/src/utils/secureFile.ts +564 -0
  274. package/src/utils/sessionState.ts +49 -0
  275. package/src/utils/state.ts +25 -0
  276. package/src/utils/style.ts +29 -0
  277. package/src/utils/terminal.ts +50 -0
  278. package/src/utils/theme.ts +127 -0
  279. package/src/utils/thinking.ts +144 -0
  280. package/src/utils/todoStorage.ts +431 -0
  281. package/src/utils/tokens.ts +43 -0
  282. package/src/utils/toolExecutionController.ts +163 -0
  283. package/src/utils/unaryLogging.ts +26 -0
  284. package/src/utils/user.ts +37 -0
  285. package/src/utils/validate.ts +165 -0
  286. package/yoga.wasm +0 -0
@@ -0,0 +1,50 @@
1
+ import { safeParseJSON } from './json'
2
+ import { logError } from './log'
3
+ import { queryQuick } from '../services/claude'
4
+
5
+ export function setTerminalTitle(title: string): void {
6
+ if (process.platform === 'win32') {
7
+ process.title = title ? `✳ ${title}` : title
8
+ } else {
9
+ process.stdout.write(`\x1b]0;${title ? `✳ ${title}` : ''}\x07`)
10
+ }
11
+ }
12
+
13
+ export async function updateTerminalTitle(message: string): Promise<void> {
14
+ try {
15
+ const result = await queryQuick({
16
+ systemPrompt: [
17
+ "Analyze if this message indicates a new conversation topic. If it does, extract a 2-3 word title that captures the new topic. Format your response as a JSON object with two fields: 'isNewTopic' (boolean) and 'title' (string, or null if isNewTopic is false). Only include these fields, no other text.",
18
+ ],
19
+ userPrompt: message,
20
+ enablePromptCaching: true,
21
+ })
22
+
23
+ const content = result.message.content
24
+ .filter(_ => _.type === 'text')
25
+ .map(_ => _.text)
26
+ .join('')
27
+
28
+ const response = safeParseJSON(content)
29
+ if (
30
+ response &&
31
+ typeof response === 'object' &&
32
+ 'isNewTopic' in response &&
33
+ 'title' in response
34
+ ) {
35
+ if (response.isNewTopic && response.title) {
36
+ setTerminalTitle(response.title as string)
37
+ }
38
+ }
39
+ } catch (error) {
40
+ logError(error)
41
+ }
42
+ }
43
+
44
+ export function clearTerminal(): Promise<void> {
45
+ return new Promise(resolve => {
46
+ process.stdout.write('\x1b[2J\x1b[3J\x1b[H', () => {
47
+ resolve()
48
+ })
49
+ })
50
+ }
@@ -0,0 +1,127 @@
1
+ import { getGlobalConfig } from './config'
2
+
3
+ export interface Theme {
4
+ bashBorder: string
5
+ kode: string
6
+ noting: string
7
+ permission: string
8
+ secondaryBorder: string
9
+ text: string
10
+ secondaryText: string
11
+ suggestion: string
12
+ success: string
13
+ error: string
14
+ warning: string
15
+ primary: string
16
+ secondary: string
17
+ diff: {
18
+ added: string
19
+ removed: string
20
+ addedDimmed: string
21
+ removedDimmed: string
22
+ }
23
+ }
24
+
25
+ const lightTheme: Theme = {
26
+ bashBorder: '#FF6E57',
27
+ kode: '#FFC233',
28
+ noting: '#222222',
29
+ permission: '#e9c61aff',
30
+ secondaryBorder: '#999',
31
+ text: '#000',
32
+ secondaryText: '#666',
33
+ suggestion: '#32e98aff',
34
+ success: '#2c7a39',
35
+ error: '#ab2b3f',
36
+ warning: '#966c1e',
37
+ primary: '#000',
38
+ secondary: '#666',
39
+ diff: {
40
+ added: '#69db7c',
41
+ removed: '#ffa8b4',
42
+ addedDimmed: '#c7e1cb',
43
+ removedDimmed: '#fdd2d8',
44
+ },
45
+ }
46
+
47
+ const lightDaltonizedTheme: Theme = {
48
+ bashBorder: '#FF6E57',
49
+ kode: '#FFC233',
50
+ noting: '#222222',
51
+ permission: '#3366ff',
52
+ secondaryBorder: '#999',
53
+ text: '#000',
54
+ secondaryText: '#666',
55
+ suggestion: '#3366ff',
56
+ success: '#006699',
57
+ error: '#cc0000',
58
+ warning: '#ff9900',
59
+ primary: '#000',
60
+ secondary: '#666',
61
+ diff: {
62
+ added: '#99ccff',
63
+ removed: '#ffcccc',
64
+ addedDimmed: '#d1e7fd',
65
+ removedDimmed: '#ffe9e9',
66
+ },
67
+ }
68
+
69
+ const darkTheme: Theme = {
70
+ bashBorder: '#FF6E57',
71
+ kode: '#FFC233',
72
+ noting: '#222222',
73
+ permission: '#b1b9f9',
74
+ secondaryBorder: '#888',
75
+ text: '#fff',
76
+ secondaryText: '#999',
77
+ suggestion: '#b1b9f9',
78
+ success: '#4eba65',
79
+ error: '#ff6b80',
80
+ warning: '#ffc107',
81
+ primary: '#fff',
82
+ secondary: '#999',
83
+ diff: {
84
+ added: '#225c2b',
85
+ removed: '#7a2936',
86
+ addedDimmed: '#47584a',
87
+ removedDimmed: '#69484d',
88
+ },
89
+ }
90
+
91
+ const darkDaltonizedTheme: Theme = {
92
+ bashBorder: '#FF6E57',
93
+ kode: '#FFC233',
94
+ noting: '#222222',
95
+ permission: '#99ccff',
96
+ secondaryBorder: '#888',
97
+ text: '#fff',
98
+ secondaryText: '#999',
99
+ suggestion: '#99ccff',
100
+ success: '#3399ff',
101
+ error: '#ff6666',
102
+ warning: '#ffcc00',
103
+ primary: '#fff',
104
+ secondary: '#999',
105
+ diff: {
106
+ added: '#004466',
107
+ removed: '#660000',
108
+ addedDimmed: '#3e515b',
109
+ removedDimmed: '#3e2c2c',
110
+ },
111
+ }
112
+
113
+ export type ThemeNames = 'dark' | 'light' | 'light-daltonized' | 'dark-daltonized'
114
+
115
+ export function getTheme(overrideTheme?: ThemeNames): Theme {
116
+ const config = getGlobalConfig()
117
+ switch (overrideTheme ?? config.theme) {
118
+ case 'light':
119
+ return lightTheme
120
+ case 'light-daltonized':
121
+ return lightDaltonizedTheme
122
+ case 'dark-daltonized':
123
+ return darkDaltonizedTheme
124
+ default:
125
+ return darkTheme
126
+ }
127
+ }
@@ -0,0 +1,144 @@
1
+ import { last } from 'lodash-es'
2
+ import type { Message } from '../query'
3
+ import { logEvent } from '../services/statsig'
4
+ import { getLastAssistantMessageId } from './messages'
5
+ import { ThinkTool } from '../tools/ThinkTool/ThinkTool'
6
+ import { USE_BEDROCK, USE_VERTEX, getModelManager } from './model'
7
+ import { getGlobalConfig } from './config'
8
+
9
+ export async function getMaxThinkingTokens(
10
+ messages: Message[],
11
+ ): Promise<number> {
12
+ if (process.env.MAX_THINKING_TOKENS) {
13
+ const tokens = parseInt(process.env.MAX_THINKING_TOKENS, 10)
14
+ logEvent('tengu_thinking', {
15
+ method: 'scratchpad',
16
+ tokenCount: tokens.toString(),
17
+ messageId: getLastAssistantMessageId(messages),
18
+ provider: USE_BEDROCK ? 'bedrock' : USE_VERTEX ? 'vertex' : '1p',
19
+ })
20
+ return tokens
21
+ }
22
+
23
+ if (await ThinkTool.isEnabled()) {
24
+ logEvent('tengu_thinking', {
25
+ method: 'scratchpad',
26
+ tokenCount: '0',
27
+ messageId: getLastAssistantMessageId(messages),
28
+ provider: USE_BEDROCK ? 'bedrock' : USE_VERTEX ? 'vertex' : '1p',
29
+ })
30
+ return 0
31
+ }
32
+
33
+ const lastMessage = last(messages)
34
+ if (
35
+ lastMessage?.type !== 'user' ||
36
+ typeof lastMessage.message.content !== 'string'
37
+ ) {
38
+ logEvent('tengu_thinking', {
39
+ method: 'scratchpad',
40
+ tokenCount: '0',
41
+ messageId: getLastAssistantMessageId(messages),
42
+ provider: USE_BEDROCK ? 'bedrock' : USE_VERTEX ? 'vertex' : '1p',
43
+ })
44
+ return 0
45
+ }
46
+
47
+ const content = lastMessage.message.content.toLowerCase()
48
+ if (
49
+ content.includes('think harder') ||
50
+ content.includes('think intensely') ||
51
+ content.includes('think longer') ||
52
+ content.includes('think really hard') ||
53
+ content.includes('think super hard') ||
54
+ content.includes('think very hard') ||
55
+ content.includes('ultrathink')
56
+ ) {
57
+ logEvent('tengu_thinking', {
58
+ method: 'scratchpad',
59
+ tokenCount: '31999',
60
+ messageId: getLastAssistantMessageId(messages),
61
+ provider: USE_BEDROCK ? 'bedrock' : USE_VERTEX ? 'vertex' : '1p',
62
+ })
63
+ return 32_000 - 1
64
+ }
65
+
66
+ if (
67
+ content.includes('think about it') ||
68
+ content.includes('think a lot') ||
69
+ content.includes('think hard') ||
70
+ content.includes('think more') ||
71
+ content.includes('megathink')
72
+ ) {
73
+ logEvent('tengu_thinking', {
74
+ method: 'scratchpad',
75
+ tokenCount: '10000',
76
+ messageId: getLastAssistantMessageId(messages),
77
+ provider: USE_BEDROCK ? 'bedrock' : USE_VERTEX ? 'vertex' : '1p',
78
+ })
79
+ return 10_000
80
+ }
81
+
82
+ if (content.includes('think')) {
83
+ logEvent('tengu_thinking', {
84
+ method: 'scratchpad',
85
+ tokenCount: '4000',
86
+ messageId: getLastAssistantMessageId(messages),
87
+ provider: USE_BEDROCK ? 'bedrock' : USE_VERTEX ? 'vertex' : '1p',
88
+ })
89
+ return 4_000
90
+ }
91
+
92
+ logEvent('tengu_thinking', {
93
+ method: 'scratchpad',
94
+ tokenCount: '0',
95
+ messageId: getLastAssistantMessageId(messages),
96
+ provider: USE_BEDROCK ? 'bedrock' : USE_VERTEX ? 'vertex' : '1p',
97
+ })
98
+ return 0
99
+ }
100
+
101
+ export async function getReasoningEffort(
102
+ modelProfile: any,
103
+ messages: Message[],
104
+ ): Promise<'low' | 'medium' | 'high' | null> {
105
+ const thinkingTokens = await getMaxThinkingTokens(messages)
106
+
107
+ // Get reasoning effort from ModelProfile first, then fallback to config
108
+ let reasoningEffort: 'low' | 'medium' | 'high' | undefined
109
+ if (modelProfile?.reasoningEffort) {
110
+ reasoningEffort = modelProfile.reasoningEffort
111
+ } else {
112
+ // 🔧 Fix: Use ModelManager fallback instead of legacy config
113
+ const modelManager = getModelManager()
114
+ const fallbackProfile = modelManager.getModel('main')
115
+ reasoningEffort = fallbackProfile?.reasoningEffort || 'medium'
116
+ }
117
+
118
+ const maxEffort =
119
+ reasoningEffort === 'high'
120
+ ? 2
121
+ : reasoningEffort === 'medium'
122
+ ? 1
123
+ : reasoningEffort === 'low'
124
+ ? 0
125
+ : null
126
+ if (!maxEffort) {
127
+ return null
128
+ }
129
+
130
+ let effort = 0
131
+ if (thinkingTokens < 10_000) {
132
+ effort = 0
133
+ } else if (thinkingTokens >= 10_000 && thinkingTokens < 30_000) {
134
+ effort = 1
135
+ } else {
136
+ effort = 2
137
+ }
138
+
139
+ if (effort > maxEffort) {
140
+ return maxEffort === 2 ? 'high' : maxEffort === 1 ? 'medium' : 'low'
141
+ }
142
+
143
+ return effort === 2 ? 'high' : effort === 1 ? 'medium' : 'low'
144
+ }