@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.
- package/LICENSE +201 -0
- package/README.md +426 -0
- package/README.zh-CN.md +326 -0
- package/cli.js +79 -0
- package/package.json +119 -0
- package/scripts/postinstall.js +18 -0
- package/src/ProjectOnboarding.tsx +198 -0
- package/src/Tool.ts +82 -0
- package/src/commands/agents.tsx +3410 -0
- package/src/commands/approvedTools.ts +53 -0
- package/src/commands/bug.tsx +20 -0
- package/src/commands/clear.ts +43 -0
- package/src/commands/compact.ts +120 -0
- package/src/commands/config.tsx +19 -0
- package/src/commands/cost.ts +18 -0
- package/src/commands/ctx_viz.ts +209 -0
- package/src/commands/doctor.ts +24 -0
- package/src/commands/help.tsx +19 -0
- package/src/commands/init.ts +37 -0
- package/src/commands/listen.ts +42 -0
- package/src/commands/login.tsx +51 -0
- package/src/commands/logout.tsx +40 -0
- package/src/commands/mcp.ts +41 -0
- package/src/commands/model.tsx +40 -0
- package/src/commands/modelstatus.tsx +20 -0
- package/src/commands/onboarding.tsx +34 -0
- package/src/commands/pr_comments.ts +59 -0
- package/src/commands/refreshCommands.ts +54 -0
- package/src/commands/release-notes.ts +34 -0
- package/src/commands/resume.tsx +31 -0
- package/src/commands/review.ts +49 -0
- package/src/commands/terminalSetup.ts +221 -0
- package/src/commands.ts +139 -0
- package/src/components/ApproveApiKey.tsx +93 -0
- package/src/components/AsciiLogo.tsx +13 -0
- package/src/components/AutoUpdater.tsx +148 -0
- package/src/components/Bug.tsx +367 -0
- package/src/components/Config.tsx +293 -0
- package/src/components/ConsoleOAuthFlow.tsx +327 -0
- package/src/components/Cost.tsx +23 -0
- package/src/components/CostThresholdDialog.tsx +46 -0
- package/src/components/CustomSelect/option-map.ts +42 -0
- package/src/components/CustomSelect/select-option.tsx +78 -0
- package/src/components/CustomSelect/select.tsx +152 -0
- package/src/components/CustomSelect/theme.ts +45 -0
- package/src/components/CustomSelect/use-select-state.ts +414 -0
- package/src/components/CustomSelect/use-select.ts +35 -0
- package/src/components/FallbackToolUseRejectedMessage.tsx +15 -0
- package/src/components/FileEditToolUpdatedMessage.tsx +66 -0
- package/src/components/Help.tsx +215 -0
- package/src/components/HighlightedCode.tsx +33 -0
- package/src/components/InvalidConfigDialog.tsx +113 -0
- package/src/components/Link.tsx +32 -0
- package/src/components/LogSelector.tsx +86 -0
- package/src/components/Logo.tsx +170 -0
- package/src/components/MCPServerApprovalDialog.tsx +100 -0
- package/src/components/MCPServerDialogCopy.tsx +25 -0
- package/src/components/MCPServerMultiselectDialog.tsx +109 -0
- package/src/components/Message.tsx +221 -0
- package/src/components/MessageResponse.tsx +15 -0
- package/src/components/MessageSelector.tsx +211 -0
- package/src/components/ModeIndicator.tsx +88 -0
- package/src/components/ModelConfig.tsx +301 -0
- package/src/components/ModelListManager.tsx +227 -0
- package/src/components/ModelSelector.tsx +3387 -0
- package/src/components/ModelStatusDisplay.tsx +230 -0
- package/src/components/Onboarding.tsx +274 -0
- package/src/components/PressEnterToContinue.tsx +11 -0
- package/src/components/PromptInput.tsx +760 -0
- package/src/components/SentryErrorBoundary.ts +39 -0
- package/src/components/Spinner.tsx +129 -0
- package/src/components/StickerRequestForm.tsx +16 -0
- package/src/components/StructuredDiff.tsx +191 -0
- package/src/components/TextInput.tsx +259 -0
- package/src/components/TodoItem.tsx +47 -0
- package/src/components/TokenWarning.tsx +31 -0
- package/src/components/ToolUseLoader.tsx +40 -0
- package/src/components/TrustDialog.tsx +106 -0
- package/src/components/binary-feedback/BinaryFeedback.tsx +63 -0
- package/src/components/binary-feedback/BinaryFeedbackOption.tsx +111 -0
- package/src/components/binary-feedback/BinaryFeedbackView.tsx +172 -0
- package/src/components/binary-feedback/utils.ts +220 -0
- package/src/components/messages/AssistantBashOutputMessage.tsx +22 -0
- package/src/components/messages/AssistantLocalCommandOutputMessage.tsx +49 -0
- package/src/components/messages/AssistantRedactedThinkingMessage.tsx +19 -0
- package/src/components/messages/AssistantTextMessage.tsx +144 -0
- package/src/components/messages/AssistantThinkingMessage.tsx +40 -0
- package/src/components/messages/AssistantToolUseMessage.tsx +133 -0
- package/src/components/messages/TaskProgressMessage.tsx +32 -0
- package/src/components/messages/TaskToolMessage.tsx +58 -0
- package/src/components/messages/UserBashInputMessage.tsx +28 -0
- package/src/components/messages/UserCommandMessage.tsx +30 -0
- package/src/components/messages/UserKodingInputMessage.tsx +28 -0
- package/src/components/messages/UserPromptMessage.tsx +35 -0
- package/src/components/messages/UserTextMessage.tsx +39 -0
- package/src/components/messages/UserToolResultMessage/UserToolCanceledMessage.tsx +12 -0
- package/src/components/messages/UserToolResultMessage/UserToolErrorMessage.tsx +36 -0
- package/src/components/messages/UserToolResultMessage/UserToolRejectMessage.tsx +31 -0
- package/src/components/messages/UserToolResultMessage/UserToolResultMessage.tsx +57 -0
- package/src/components/messages/UserToolResultMessage/UserToolSuccessMessage.tsx +35 -0
- package/src/components/messages/UserToolResultMessage/utils.tsx +56 -0
- package/src/components/permissions/BashPermissionRequest/BashPermissionRequest.tsx +121 -0
- package/src/components/permissions/FallbackPermissionRequest.tsx +153 -0
- package/src/components/permissions/FileEditPermissionRequest/FileEditPermissionRequest.tsx +182 -0
- package/src/components/permissions/FileEditPermissionRequest/FileEditToolDiff.tsx +77 -0
- package/src/components/permissions/FileWritePermissionRequest/FileWritePermissionRequest.tsx +164 -0
- package/src/components/permissions/FileWritePermissionRequest/FileWriteToolDiff.tsx +83 -0
- package/src/components/permissions/FilesystemPermissionRequest/FilesystemPermissionRequest.tsx +240 -0
- package/src/components/permissions/PermissionRequest.tsx +101 -0
- package/src/components/permissions/PermissionRequestTitle.tsx +69 -0
- package/src/components/permissions/hooks.ts +44 -0
- package/src/components/permissions/toolUseOptions.ts +59 -0
- package/src/components/permissions/utils.ts +23 -0
- package/src/constants/betas.ts +5 -0
- package/src/constants/claude-asterisk-ascii-art.tsx +238 -0
- package/src/constants/figures.ts +4 -0
- package/src/constants/keys.ts +3 -0
- package/src/constants/macros.ts +8 -0
- package/src/constants/modelCapabilities.ts +179 -0
- package/src/constants/models.ts +1025 -0
- package/src/constants/oauth.ts +18 -0
- package/src/constants/product.ts +17 -0
- package/src/constants/prompts.ts +168 -0
- package/src/constants/releaseNotes.ts +7 -0
- package/src/context/PermissionContext.tsx +149 -0
- package/src/context.ts +278 -0
- package/src/cost-tracker.ts +84 -0
- package/src/entrypoints/cli.tsx +1542 -0
- package/src/entrypoints/mcp.ts +176 -0
- package/src/history.ts +25 -0
- package/src/hooks/useApiKeyVerification.ts +59 -0
- package/src/hooks/useArrowKeyHistory.ts +55 -0
- package/src/hooks/useCanUseTool.ts +138 -0
- package/src/hooks/useCancelRequest.ts +39 -0
- package/src/hooks/useDoublePress.ts +42 -0
- package/src/hooks/useExitOnCtrlCD.ts +31 -0
- package/src/hooks/useInterval.ts +25 -0
- package/src/hooks/useLogMessages.ts +16 -0
- package/src/hooks/useLogStartupTime.ts +12 -0
- package/src/hooks/useNotifyAfterTimeout.ts +65 -0
- package/src/hooks/usePermissionRequestLogging.ts +44 -0
- package/src/hooks/useTerminalSize.ts +49 -0
- package/src/hooks/useTextInput.ts +318 -0
- package/src/hooks/useUnifiedCompletion.ts +1405 -0
- package/src/messages.ts +38 -0
- package/src/permissions.ts +268 -0
- package/src/query.ts +715 -0
- package/src/screens/ConfigureNpmPrefix.tsx +197 -0
- package/src/screens/Doctor.tsx +219 -0
- package/src/screens/LogList.tsx +68 -0
- package/src/screens/REPL.tsx +809 -0
- package/src/screens/ResumeConversation.tsx +68 -0
- package/src/services/adapters/base.ts +38 -0
- package/src/services/adapters/chatCompletions.ts +90 -0
- package/src/services/adapters/responsesAPI.ts +170 -0
- package/src/services/browserMocks.ts +66 -0
- package/src/services/claude.ts +2197 -0
- package/src/services/customCommands.ts +704 -0
- package/src/services/fileFreshness.ts +377 -0
- package/src/services/gpt5ConnectionTest.ts +340 -0
- package/src/services/mcpClient.ts +564 -0
- package/src/services/mcpServerApproval.tsx +50 -0
- package/src/services/mentionProcessor.ts +273 -0
- package/src/services/modelAdapterFactory.ts +69 -0
- package/src/services/notifier.ts +40 -0
- package/src/services/oauth.ts +357 -0
- package/src/services/openai.ts +1338 -0
- package/src/services/responseStateManager.ts +90 -0
- package/src/services/sentry.ts +3 -0
- package/src/services/statsig.ts +172 -0
- package/src/services/statsigStorage.ts +86 -0
- package/src/services/systemReminder.ts +507 -0
- package/src/services/vcr.ts +161 -0
- package/src/test/testAdapters.ts +96 -0
- package/src/tools/ArchitectTool/ArchitectTool.tsx +122 -0
- package/src/tools/ArchitectTool/prompt.ts +15 -0
- package/src/tools/AskExpertModelTool/AskExpertModelTool.tsx +569 -0
- package/src/tools/BashTool/BashTool.tsx +243 -0
- package/src/tools/BashTool/BashToolResultMessage.tsx +38 -0
- package/src/tools/BashTool/OutputLine.tsx +49 -0
- package/src/tools/BashTool/prompt.ts +174 -0
- package/src/tools/BashTool/utils.ts +56 -0
- package/src/tools/FileEditTool/FileEditTool.tsx +315 -0
- package/src/tools/FileEditTool/prompt.ts +51 -0
- package/src/tools/FileEditTool/utils.ts +58 -0
- package/src/tools/FileReadTool/FileReadTool.tsx +404 -0
- package/src/tools/FileReadTool/prompt.ts +7 -0
- package/src/tools/FileWriteTool/FileWriteTool.tsx +297 -0
- package/src/tools/FileWriteTool/prompt.ts +10 -0
- package/src/tools/GlobTool/GlobTool.tsx +119 -0
- package/src/tools/GlobTool/prompt.ts +8 -0
- package/src/tools/GrepTool/GrepTool.tsx +147 -0
- package/src/tools/GrepTool/prompt.ts +11 -0
- package/src/tools/MCPTool/MCPTool.tsx +107 -0
- package/src/tools/MCPTool/prompt.ts +3 -0
- package/src/tools/MemoryReadTool/MemoryReadTool.tsx +127 -0
- package/src/tools/MemoryReadTool/prompt.ts +3 -0
- package/src/tools/MemoryWriteTool/MemoryWriteTool.tsx +89 -0
- package/src/tools/MemoryWriteTool/prompt.ts +3 -0
- package/src/tools/MultiEditTool/MultiEditTool.tsx +366 -0
- package/src/tools/MultiEditTool/prompt.ts +45 -0
- package/src/tools/NotebookEditTool/NotebookEditTool.tsx +298 -0
- package/src/tools/NotebookEditTool/prompt.ts +3 -0
- package/src/tools/NotebookReadTool/NotebookReadTool.tsx +258 -0
- package/src/tools/NotebookReadTool/prompt.ts +3 -0
- package/src/tools/StickerRequestTool/StickerRequestTool.tsx +93 -0
- package/src/tools/StickerRequestTool/prompt.ts +19 -0
- package/src/tools/TaskTool/TaskTool.tsx +466 -0
- package/src/tools/TaskTool/constants.ts +1 -0
- package/src/tools/TaskTool/prompt.ts +92 -0
- package/src/tools/ThinkTool/ThinkTool.tsx +54 -0
- package/src/tools/ThinkTool/prompt.ts +12 -0
- package/src/tools/TodoWriteTool/TodoWriteTool.tsx +313 -0
- package/src/tools/TodoWriteTool/prompt.ts +63 -0
- package/src/tools/URLFetcherTool/URLFetcherTool.tsx +178 -0
- package/src/tools/URLFetcherTool/cache.ts +55 -0
- package/src/tools/URLFetcherTool/htmlToMarkdown.ts +55 -0
- package/src/tools/URLFetcherTool/prompt.ts +17 -0
- package/src/tools/WebSearchTool/WebSearchTool.tsx +103 -0
- package/src/tools/WebSearchTool/prompt.ts +13 -0
- package/src/tools/WebSearchTool/searchProviders.ts +66 -0
- package/src/tools/lsTool/lsTool.tsx +272 -0
- package/src/tools/lsTool/prompt.ts +2 -0
- package/src/tools.ts +67 -0
- package/src/types/PermissionMode.ts +120 -0
- package/src/types/RequestContext.ts +72 -0
- package/src/types/conversation.ts +51 -0
- package/src/types/logs.ts +58 -0
- package/src/types/modelCapabilities.ts +64 -0
- package/src/types/notebook.ts +87 -0
- package/src/utils/Cursor.ts +436 -0
- package/src/utils/PersistentShell.ts +552 -0
- package/src/utils/advancedFuzzyMatcher.ts +290 -0
- package/src/utils/agentLoader.ts +278 -0
- package/src/utils/agentStorage.ts +97 -0
- package/src/utils/array.ts +3 -0
- package/src/utils/ask.tsx +99 -0
- package/src/utils/auth.ts +13 -0
- package/src/utils/autoCompactCore.ts +223 -0
- package/src/utils/autoUpdater.ts +458 -0
- package/src/utils/betas.ts +20 -0
- package/src/utils/browser.ts +14 -0
- package/src/utils/cleanup.ts +72 -0
- package/src/utils/commands.ts +261 -0
- package/src/utils/commonUnixCommands.ts +161 -0
- package/src/utils/config.ts +945 -0
- package/src/utils/conversationRecovery.ts +55 -0
- package/src/utils/debugLogger.ts +1235 -0
- package/src/utils/diff.ts +42 -0
- package/src/utils/env.ts +57 -0
- package/src/utils/errors.ts +21 -0
- package/src/utils/exampleCommands.ts +109 -0
- package/src/utils/execFileNoThrow.ts +51 -0
- package/src/utils/expertChatStorage.ts +136 -0
- package/src/utils/file.ts +405 -0
- package/src/utils/fileRecoveryCore.ts +71 -0
- package/src/utils/format.tsx +44 -0
- package/src/utils/fuzzyMatcher.ts +328 -0
- package/src/utils/generators.ts +62 -0
- package/src/utils/git.ts +92 -0
- package/src/utils/globalLogger.ts +77 -0
- package/src/utils/http.ts +10 -0
- package/src/utils/imagePaste.ts +38 -0
- package/src/utils/json.ts +13 -0
- package/src/utils/log.ts +382 -0
- package/src/utils/markdown.ts +213 -0
- package/src/utils/messageContextManager.ts +289 -0
- package/src/utils/messages.tsx +939 -0
- package/src/utils/model.ts +914 -0
- package/src/utils/permissions/filesystem.ts +127 -0
- package/src/utils/responseState.ts +23 -0
- package/src/utils/ripgrep.ts +167 -0
- package/src/utils/secureFile.ts +564 -0
- package/src/utils/sessionState.ts +49 -0
- package/src/utils/state.ts +25 -0
- package/src/utils/style.ts +29 -0
- package/src/utils/terminal.ts +50 -0
- package/src/utils/theme.ts +127 -0
- package/src/utils/thinking.ts +144 -0
- package/src/utils/todoStorage.ts +431 -0
- package/src/utils/tokens.ts +43 -0
- package/src/utils/toolExecutionController.ts +163 -0
- package/src/utils/unaryLogging.ts +26 -0
- package/src/utils/user.ts +37 -0
- package/src/utils/validate.ts +165 -0
- 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
|
+
}
|