@renxqoo/renx-code 0.0.9 → 0.0.10
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/README.md +14 -0
- package/bin/renx.cjs +79 -42
- package/bin/renx.exe +0 -0
- package/package.json +10 -28
- package/src/App.tsx +0 -297
- package/src/agent/runtime/event-format.ts +0 -258
- package/src/agent/runtime/model-types.ts +0 -13
- package/src/agent/runtime/runtime.context-usage.test.ts +0 -192
- package/src/agent/runtime/runtime.error-handling.test.ts +0 -235
- package/src/agent/runtime/runtime.simple.test.ts +0 -16
- package/src/agent/runtime/runtime.test.ts +0 -296
- package/src/agent/runtime/runtime.ts +0 -875
- package/src/agent/runtime/runtime.usage-forwarding.test.ts +0 -228
- package/src/agent/runtime/source-modules.test.ts +0 -38
- package/src/agent/runtime/source-modules.ts +0 -370
- package/src/agent/runtime/tool-call-buffer.test.ts +0 -65
- package/src/agent/runtime/tool-call-buffer.ts +0 -60
- package/src/agent/runtime/tool-confirmation.test.ts +0 -56
- package/src/agent/runtime/tool-confirmation.ts +0 -15
- package/src/agent/runtime/types.ts +0 -99
- package/src/commands/slash-commands.test.ts +0 -216
- package/src/commands/slash-commands.ts +0 -64
- package/src/components/chat/assistant-reply.test.tsx +0 -47
- package/src/components/chat/assistant-reply.tsx +0 -136
- package/src/components/chat/assistant-segment.test.ts +0 -99
- package/src/components/chat/assistant-segment.tsx +0 -125
- package/src/components/chat/assistant-tool-group.tsx +0 -900
- package/src/components/chat/code-block.test.tsx +0 -206
- package/src/components/chat/code-block.tsx +0 -313
- package/src/components/chat/prompt-card.tsx +0 -81
- package/src/components/chat/segment-groups.test.ts +0 -52
- package/src/components/chat/segment-groups.ts +0 -106
- package/src/components/chat/turn-item.tsx +0 -39
- package/src/components/conversation-panel.tsx +0 -43
- package/src/components/file-mention-menu.tsx +0 -77
- package/src/components/file-picker-dialog.tsx +0 -206
- package/src/components/footer-hints.tsx +0 -75
- package/src/components/model-picker-dialog.tsx +0 -248
- package/src/components/prompt.tsx +0 -233
- package/src/components/slash-command-menu.tsx +0 -65
- package/src/components/tool-confirm-dialog-content.test.ts +0 -103
- package/src/components/tool-confirm-dialog-content.ts +0 -186
- package/src/components/tool-confirm-dialog.tsx +0 -187
- package/src/components/tool-display-config.ts +0 -119
- package/src/context-usage-regressions.test.ts +0 -26
- package/src/files/attachment-capabilities.test.ts +0 -30
- package/src/files/attachment-capabilities.ts +0 -50
- package/src/files/attachment-content.ts +0 -153
- package/src/files/file-mention-query.test.ts +0 -34
- package/src/files/file-mention-query.ts +0 -32
- package/src/files/prompt-display.ts +0 -13
- package/src/files/types.ts +0 -5
- package/src/files/workspace-files.ts +0 -61
- package/src/hooks/agent-event-handlers.test.ts +0 -207
- package/src/hooks/agent-event-handlers.ts +0 -196
- package/src/hooks/chat-local-replies.fixed.test.ts +0 -119
- package/src/hooks/chat-local-replies.test.ts +0 -153
- package/src/hooks/chat-local-replies.ts +0 -63
- package/src/hooks/turn-updater.test.ts +0 -70
- package/src/hooks/turn-updater.ts +0 -166
- package/src/hooks/use-agent-chat.context.test.ts +0 -10
- package/src/hooks/use-agent-chat.status.test.ts +0 -14
- package/src/hooks/use-agent-chat.test.ts +0 -80
- package/src/hooks/use-agent-chat.ts +0 -621
- package/src/hooks/use-file-mention-menu.ts +0 -196
- package/src/hooks/use-file-picker.ts +0 -185
- package/src/hooks/use-model-picker.ts +0 -196
- package/src/hooks/use-slash-command-menu.ts +0 -154
- package/src/index.tsx +0 -55
- package/src/runtime/clipboard.test.ts +0 -43
- package/src/runtime/clipboard.ts +0 -89
- package/src/runtime/exit.test.ts +0 -177
- package/src/runtime/exit.ts +0 -98
- package/src/runtime/runtime-support.test.ts +0 -31
- package/src/runtime/terminal-theme.test.ts +0 -55
- package/src/runtime/terminal-theme.ts +0 -196
- package/src/types/chat.ts +0 -32
- package/src/types/message-content.ts +0 -48
- package/src/ui/open-code-theme.ts +0 -176
- package/src/ui/opencode-markdown.ts +0 -211
- package/src/ui/theme.simple.test.ts +0 -52
- package/src/ui/theme.test.ts +0 -151
- package/src/ui/theme.ts +0 -152
- package/src/utils/time.test.ts +0 -144
- package/src/utils/time.ts +0 -7
- package/tsconfig.json +0 -30
- package/vendor/agent-root/src/agent/ENTERPRISE_ACCEPTANCE_CHECKLIST.md +0 -95
- package/vendor/agent-root/src/agent/ENTERPRISE_REALTIME.html +0 -1345
- package/vendor/agent-root/src/agent/ENTERPRISE_REALTIME.md +0 -1353
- package/vendor/agent-root/src/agent/ERROR_CONTRACT.md +0 -60
- package/vendor/agent-root/src/agent/TEST_COVERAGE_ANALYSIS.md +0 -278
- package/vendor/agent-root/src/agent/__test__/error-contract.test.ts +0 -72
- package/vendor/agent-root/src/agent/__test__/types.test.ts +0 -137
- package/vendor/agent-root/src/agent/agent/__test__/abort-runtime.test.ts +0 -83
- package/vendor/agent-root/src/agent/agent/__test__/callback-safety.test.ts +0 -34
- package/vendor/agent-root/src/agent/agent/__test__/compaction.test.ts +0 -323
- package/vendor/agent-root/src/agent/agent/__test__/concurrency.test.ts +0 -290
- package/vendor/agent-root/src/agent/agent/__test__/error-normalizer.test.ts +0 -377
- package/vendor/agent-root/src/agent/agent/__test__/error.test.ts +0 -212
- package/vendor/agent-root/src/agent/agent/__test__/fault-injection.test.ts +0 -295
- package/vendor/agent-root/src/agent/agent/__test__/index.test.ts +0 -3607
- package/vendor/agent-root/src/agent/agent/__test__/logger.test.ts +0 -35
- package/vendor/agent-root/src/agent/agent/__test__/message-utils.test.ts +0 -517
- package/vendor/agent-root/src/agent/agent/__test__/telemetry.test.ts +0 -97
- package/vendor/agent-root/src/agent/agent/__test__/timeout-budget.test.ts +0 -479
- package/vendor/agent-root/src/agent/agent/__test__/tool-call-merge.test.ts +0 -80
- package/vendor/agent-root/src/agent/agent/__test__/tool-execution-ledger.test.ts +0 -76
- package/vendor/agent-root/src/agent/agent/__test__/write-buffer.test.ts +0 -173
- package/vendor/agent-root/src/agent/agent/__test__/write-file-session.test.ts +0 -109
- package/vendor/agent-root/src/agent/agent/abort-runtime.ts +0 -71
- package/vendor/agent-root/src/agent/agent/callback-safety.ts +0 -33
- package/vendor/agent-root/src/agent/agent/compaction.ts +0 -291
- package/vendor/agent-root/src/agent/agent/concurrency.ts +0 -103
- package/vendor/agent-root/src/agent/agent/error-normalizer.ts +0 -190
- package/vendor/agent-root/src/agent/agent/error.ts +0 -198
- package/vendor/agent-root/src/agent/agent/index.ts +0 -1772
- package/vendor/agent-root/src/agent/agent/logger.ts +0 -65
- package/vendor/agent-root/src/agent/agent/message-utils.ts +0 -101
- package/vendor/agent-root/src/agent/agent/stream-events.ts +0 -61
- package/vendor/agent-root/src/agent/agent/telemetry.ts +0 -123
- package/vendor/agent-root/src/agent/agent/timeout-budget.ts +0 -227
- package/vendor/agent-root/src/agent/agent/tool-call-merge.ts +0 -111
- package/vendor/agent-root/src/agent/agent/tool-execution-ledger.ts +0 -164
- package/vendor/agent-root/src/agent/agent/write-buffer.ts +0 -188
- package/vendor/agent-root/src/agent/agent/write-file-session.ts +0 -238
- package/vendor/agent-root/src/agent/app/__test__/agent-app-service.test.ts +0 -1053
- package/vendor/agent-root/src/agent/app/__test__/minimal-agent-application.test.ts +0 -158
- package/vendor/agent-root/src/agent/app/__test__/sqlite-agent-app-store.test.ts +0 -437
- package/vendor/agent-root/src/agent/app/agent-app-service.ts +0 -748
- package/vendor/agent-root/src/agent/app/contracts.ts +0 -109
- package/vendor/agent-root/src/agent/app/index.ts +0 -5
- package/vendor/agent-root/src/agent/app/minimal-agent-application.ts +0 -151
- package/vendor/agent-root/src/agent/app/ports.ts +0 -72
- package/vendor/agent-root/src/agent/app/sqlite-agent-app-store.ts +0 -1182
- package/vendor/agent-root/src/agent/app/sqlite-client.ts +0 -177
- package/vendor/agent-root/src/agent/docs/cli-app-layer/00-README.md +0 -36
- package/vendor/agent-root/src/agent/docs/cli-app-layer/01-scope-and-goals.md +0 -33
- package/vendor/agent-root/src/agent/docs/cli-app-layer/02-architecture-overview.md +0 -40
- package/vendor/agent-root/src/agent/docs/cli-app-layer/03-domain-model-and-contracts.md +0 -91
- package/vendor/agent-root/src/agent/docs/cli-app-layer/04-ports-and-interfaces.md +0 -116
- package/vendor/agent-root/src/agent/docs/cli-app-layer/05-run-orchestration-and-state-machine.md +0 -52
- package/vendor/agent-root/src/agent/docs/cli-app-layer/06-cli-commands-and-ux.md +0 -53
- package/vendor/agent-root/src/agent/docs/cli-app-layer/07-storage-design-local.md +0 -52
- package/vendor/agent-root/src/agent/docs/cli-app-layer/08-error-and-observability.md +0 -40
- package/vendor/agent-root/src/agent/docs/cli-app-layer/09-security-and-policy-boundary.md +0 -19
- package/vendor/agent-root/src/agent/docs/cli-app-layer/10-test-plan-and-acceptance.md +0 -28
- package/vendor/agent-root/src/agent/docs/cli-app-layer/11-implementation-phases.md +0 -26
- package/vendor/agent-root/src/agent/docs/cli-app-layer/12-open-questions-and-risks.md +0 -30
- package/vendor/agent-root/src/agent/docs/cli-app-layer/13-sqlite-schema-fields-and-rationale.md +0 -567
- package/vendor/agent-root/src/agent/docs/cli-app-layer/14-project-flow-mermaid.md +0 -583
- package/vendor/agent-root/src/agent/docs/cli-app-layer/15-openclaw-style-project-blueprint.md +0 -972
- package/vendor/agent-root/src/agent/error-contract.ts +0 -154
- package/vendor/agent-root/src/agent/prompts/system.ts +0 -246
- package/vendor/agent-root/src/agent/prompts/system1.ts +0 -208
- package/vendor/agent-root/src/agent/storage/__test__/file-history-store.test.ts +0 -98
- package/vendor/agent-root/src/agent/storage/file-history-store.ts +0 -313
- package/vendor/agent-root/src/agent/storage/file-storage-config.ts +0 -94
- package/vendor/agent-root/src/agent/storage/file-system.ts +0 -31
- package/vendor/agent-root/src/agent/storage/file-write-service.ts +0 -21
- package/vendor/agent-root/src/agent/tool/__test__/base-tool.test.ts +0 -413
- package/vendor/agent-root/src/agent/tool/__test__/bash-policy.test.ts +0 -356
- package/vendor/agent-root/src/agent/tool/__test__/bash.mocked-coverage.test.ts +0 -375
- package/vendor/agent-root/src/agent/tool/__test__/bash.test.ts +0 -372
- package/vendor/agent-root/src/agent/tool/__test__/error.test.ts +0 -108
- package/vendor/agent-root/src/agent/tool/__test__/file-edit-tool.test.ts +0 -258
- package/vendor/agent-root/src/agent/tool/__test__/file-history-tools.test.ts +0 -121
- package/vendor/agent-root/src/agent/tool/__test__/file-read-tool.test.ts +0 -210
- package/vendor/agent-root/src/agent/tool/__test__/glob.test.ts +0 -139
- package/vendor/agent-root/src/agent/tool/__test__/grep.mocked-coverage.test.ts +0 -456
- package/vendor/agent-root/src/agent/tool/__test__/grep.test.ts +0 -192
- package/vendor/agent-root/src/agent/tool/__test__/lsp.test.ts +0 -300
- package/vendor/agent-root/src/agent/tool/__test__/outside-workspace-confirmation.test.ts +0 -214
- package/vendor/agent-root/src/agent/tool/__test__/path-security.test.ts +0 -336
- package/vendor/agent-root/src/agent/tool/__test__/skill-loader.test.ts +0 -494
- package/vendor/agent-root/src/agent/tool/__test__/skill-parser.test.ts +0 -543
- package/vendor/agent-root/src/agent/tool/__test__/skill-tool.test.ts +0 -172
- package/vendor/agent-root/src/agent/tool/__test__/task-concurrency-and-version.test.ts +0 -116
- package/vendor/agent-root/src/agent/tool/__test__/task-create-get-list-update.test.ts +0 -267
- package/vendor/agent-root/src/agent/tool/__test__/task-create.test.ts +0 -519
- package/vendor/agent-root/src/agent/tool/__test__/task-errors.test.ts +0 -225
- package/vendor/agent-root/src/agent/tool/__test__/task-output-blocking.test.ts +0 -223
- package/vendor/agent-root/src/agent/tool/__test__/task-output.test.ts +0 -184
- package/vendor/agent-root/src/agent/tool/__test__/task-parent-abort.test.ts +0 -287
- package/vendor/agent-root/src/agent/tool/__test__/task-real-runner-adapter.test.ts +0 -190
- package/vendor/agent-root/src/agent/tool/__test__/task-run-lifecycle.test.ts +0 -352
- package/vendor/agent-root/src/agent/tool/__test__/task-store-runner-branches.test.ts +0 -395
- package/vendor/agent-root/src/agent/tool/__test__/task-store.test.ts +0 -391
- package/vendor/agent-root/src/agent/tool/__test__/task-subagent-config-integration.test.ts +0 -176
- package/vendor/agent-root/src/agent/tool/__test__/task-subagent-config.test.ts +0 -68
- package/vendor/agent-root/src/agent/tool/__test__/task-tools-core-edges.test.ts +0 -630
- package/vendor/agent-root/src/agent/tool/__test__/task-tools-runtime-edges.test.ts +0 -732
- package/vendor/agent-root/src/agent/tool/__test__/task-types.test.ts +0 -494
- package/vendor/agent-root/src/agent/tool/__test__/task-utils-branches.test.ts +0 -175
- package/vendor/agent-root/src/agent/tool/__test__/tool-manager.test.ts +0 -505
- package/vendor/agent-root/src/agent/tool/__test__/types.test.ts +0 -55
- package/vendor/agent-root/src/agent/tool/__test__/web-fetch.test.ts +0 -244
- package/vendor/agent-root/src/agent/tool/__test__/web-search.test.ts +0 -290
- package/vendor/agent-root/src/agent/tool/__test__/write-file.test.ts +0 -368
- package/vendor/agent-root/src/agent/tool/base-tool.ts +0 -345
- package/vendor/agent-root/src/agent/tool/bash-policy.ts +0 -636
- package/vendor/agent-root/src/agent/tool/bash.ts +0 -688
- package/vendor/agent-root/src/agent/tool/error.ts +0 -131
- package/vendor/agent-root/src/agent/tool/file-edit-tool.ts +0 -264
- package/vendor/agent-root/src/agent/tool/file-history-list.ts +0 -103
- package/vendor/agent-root/src/agent/tool/file-history-restore.ts +0 -149
- package/vendor/agent-root/src/agent/tool/file-read-tool.ts +0 -211
- package/vendor/agent-root/src/agent/tool/glob.ts +0 -171
- package/vendor/agent-root/src/agent/tool/grep.ts +0 -496
- package/vendor/agent-root/src/agent/tool/lsp.ts +0 -481
- package/vendor/agent-root/src/agent/tool/path-security.ts +0 -117
- package/vendor/agent-root/src/agent/tool/search/common.ts +0 -153
- package/vendor/agent-root/src/agent/tool/skill/index.ts +0 -13
- package/vendor/agent-root/src/agent/tool/skill/loader.ts +0 -229
- package/vendor/agent-root/src/agent/tool/skill/parser.ts +0 -124
- package/vendor/agent-root/src/agent/tool/skill/types.ts +0 -27
- package/vendor/agent-root/src/agent/tool/skill-tool.ts +0 -143
- package/vendor/agent-root/src/agent/tool/task-create.ts +0 -186
- package/vendor/agent-root/src/agent/tool/task-errors.ts +0 -42
- package/vendor/agent-root/src/agent/tool/task-get.ts +0 -116
- package/vendor/agent-root/src/agent/tool/task-graph.ts +0 -78
- package/vendor/agent-root/src/agent/tool/task-list.ts +0 -141
- package/vendor/agent-root/src/agent/tool/task-mock-runner-adapter.ts +0 -232
- package/vendor/agent-root/src/agent/tool/task-output.ts +0 -223
- package/vendor/agent-root/src/agent/tool/task-parent-abort.ts +0 -115
- package/vendor/agent-root/src/agent/tool/task-real-runner-adapter.ts +0 -336
- package/vendor/agent-root/src/agent/tool/task-runner-adapter.ts +0 -55
- package/vendor/agent-root/src/agent/tool/task-stop.ts +0 -187
- package/vendor/agent-root/src/agent/tool/task-store.ts +0 -217
- package/vendor/agent-root/src/agent/tool/task-subagent-config.ts +0 -149
- package/vendor/agent-root/src/agent/tool/task-types.ts +0 -264
- package/vendor/agent-root/src/agent/tool/task-update.ts +0 -315
- package/vendor/agent-root/src/agent/tool/task.ts +0 -209
- package/vendor/agent-root/src/agent/tool/tool-manager.ts +0 -361
- package/vendor/agent-root/src/agent/tool/tool-prompts.ts +0 -242
- package/vendor/agent-root/src/agent/tool/types.ts +0 -116
- package/vendor/agent-root/src/agent/tool/web-fetch.ts +0 -227
- package/vendor/agent-root/src/agent/tool/web-search.ts +0 -208
- package/vendor/agent-root/src/agent/tool/write-file.ts +0 -497
- package/vendor/agent-root/src/agent/types.ts +0 -232
- package/vendor/agent-root/src/agent/utils/__tests__/index.test.ts +0 -18
- package/vendor/agent-root/src/agent/utils/__tests__/message-utils.test.ts +0 -610
- package/vendor/agent-root/src/agent/utils/__tests__/message.test.ts +0 -223
- package/vendor/agent-root/src/agent/utils/__tests__/token.test.ts +0 -42
- package/vendor/agent-root/src/agent/utils/index.ts +0 -16
- package/vendor/agent-root/src/agent/utils/message.ts +0 -171
- package/vendor/agent-root/src/agent/utils/token.ts +0 -28
- package/vendor/agent-root/src/config/__tests__/load-config-to-env.test.ts +0 -238
- package/vendor/agent-root/src/config/__tests__/loader.test.ts +0 -361
- package/vendor/agent-root/src/config/__tests__/runtime.test.ts +0 -88
- package/vendor/agent-root/src/config/index.ts +0 -55
- package/vendor/agent-root/src/config/loader.ts +0 -494
- package/vendor/agent-root/src/config/paths.ts +0 -30
- package/vendor/agent-root/src/config/runtime.ts +0 -163
- package/vendor/agent-root/src/config/types.ts +0 -96
- package/vendor/agent-root/src/logger/index.ts +0 -57
- package/vendor/agent-root/src/logger/logger.ts +0 -819
- package/vendor/agent-root/src/logger/types.ts +0 -150
- package/vendor/agent-root/src/providers/__tests__/errors.test.ts +0 -441
- package/vendor/agent-root/src/providers/__tests__/index.test.ts +0 -16
- package/vendor/agent-root/src/providers/__tests__/openai-compatible.options.test.ts +0 -318
- package/vendor/agent-root/src/providers/__tests__/openai-compatible.test.ts +0 -600
- package/vendor/agent-root/src/providers/__tests__/registry.test.ts +0 -523
- package/vendor/agent-root/src/providers/__tests__/responses-adapter.test.ts +0 -298
- package/vendor/agent-root/src/providers/adapters/__tests__/anthropic.test.ts +0 -354
- package/vendor/agent-root/src/providers/adapters/__tests__/kimi.test.ts +0 -58
- package/vendor/agent-root/src/providers/adapters/__tests__/standard.test.ts +0 -261
- package/vendor/agent-root/src/providers/adapters/anthropic.ts +0 -572
- package/vendor/agent-root/src/providers/adapters/base.ts +0 -131
- package/vendor/agent-root/src/providers/adapters/kimi.ts +0 -48
- package/vendor/agent-root/src/providers/adapters/responses.ts +0 -732
- package/vendor/agent-root/src/providers/adapters/standard.ts +0 -120
- package/vendor/agent-root/src/providers/http/__tests__/client.timeout.test.ts +0 -313
- package/vendor/agent-root/src/providers/http/client.ts +0 -289
- package/vendor/agent-root/src/providers/http/stream-parser.ts +0 -109
- package/vendor/agent-root/src/providers/index.ts +0 -76
- package/vendor/agent-root/src/providers/kimi-headers.ts +0 -177
- package/vendor/agent-root/src/providers/openai-compatible.ts +0 -387
- package/vendor/agent-root/src/providers/registry/model-config.ts +0 -477
- package/vendor/agent-root/src/providers/registry/provider-factory.ts +0 -127
- package/vendor/agent-root/src/providers/registry.ts +0 -135
- package/vendor/agent-root/src/providers/types/api.ts +0 -284
- package/vendor/agent-root/src/providers/types/config.ts +0 -58
- package/vendor/agent-root/src/providers/types/errors.ts +0 -323
- package/vendor/agent-root/src/providers/types/index.ts +0 -72
- package/vendor/agent-root/src/providers/types/provider.ts +0 -45
- package/vendor/agent-root/src/providers/types/registry.ts +0 -68
|
@@ -1,153 +0,0 @@
|
|
|
1
|
-
import { readFile } from 'node:fs/promises';
|
|
2
|
-
import { extname } from 'node:path';
|
|
3
|
-
|
|
4
|
-
import type { InputContentPart, MessageContent } from '../types/message-content';
|
|
5
|
-
import type { PromptFileSelection } from './types';
|
|
6
|
-
import {
|
|
7
|
-
type AttachmentModelCapabilities,
|
|
8
|
-
isAudioSelection,
|
|
9
|
-
isImageSelection,
|
|
10
|
-
isVideoSelection,
|
|
11
|
-
} from './attachment-capabilities';
|
|
12
|
-
|
|
13
|
-
const IMAGE_MIME_BY_EXTENSION: Record<string, string> = {
|
|
14
|
-
'.gif': 'image/gif',
|
|
15
|
-
'.jpeg': 'image/jpeg',
|
|
16
|
-
'.jpg': 'image/jpeg',
|
|
17
|
-
'.png': 'image/png',
|
|
18
|
-
'.webp': 'image/webp',
|
|
19
|
-
};
|
|
20
|
-
|
|
21
|
-
const VIDEO_MIME_BY_EXTENSION: Record<string, string> = {
|
|
22
|
-
'.avi': 'video/x-msvideo',
|
|
23
|
-
'.m4v': 'video/mp4',
|
|
24
|
-
'.mkv': 'video/x-matroska',
|
|
25
|
-
'.mov': 'video/quicktime',
|
|
26
|
-
'.mp4': 'video/mp4',
|
|
27
|
-
'.webm': 'video/webm',
|
|
28
|
-
};
|
|
29
|
-
|
|
30
|
-
const AUDIO_MIME_BY_EXTENSION: Record<string, string> = {
|
|
31
|
-
'.aac': 'audio/aac',
|
|
32
|
-
'.flac': 'audio/flac',
|
|
33
|
-
'.m4a': 'audio/mp4',
|
|
34
|
-
'.mp3': 'audio/mpeg',
|
|
35
|
-
'.ogg': 'audio/ogg',
|
|
36
|
-
'.wav': 'audio/wav',
|
|
37
|
-
};
|
|
38
|
-
|
|
39
|
-
const FALLBACK_FILE_MIME = 'application/octet-stream';
|
|
40
|
-
const MAX_ATTACHMENT_BYTES = 2 * 1024 * 1024;
|
|
41
|
-
const MAX_TEXT_ATTACHMENT_CHARS = 80_000;
|
|
42
|
-
const TEXT_DECODER = new TextDecoder('utf-8', { fatal: false });
|
|
43
|
-
|
|
44
|
-
const formatFileFence = (path: string, content: string, truncated: boolean) => {
|
|
45
|
-
const suffix = truncated ? '\n\n[truncated for prompt size]' : '';
|
|
46
|
-
return `Attached file: ${path}\n\n\`\`\`\n${content}\n\`\`\`${suffix}`;
|
|
47
|
-
};
|
|
48
|
-
|
|
49
|
-
const inferMimeType = (path: string) => {
|
|
50
|
-
const extension = extname(path).toLowerCase();
|
|
51
|
-
return (
|
|
52
|
-
IMAGE_MIME_BY_EXTENSION[extension] ??
|
|
53
|
-
VIDEO_MIME_BY_EXTENSION[extension] ??
|
|
54
|
-
AUDIO_MIME_BY_EXTENSION[extension] ??
|
|
55
|
-
FALLBACK_FILE_MIME
|
|
56
|
-
);
|
|
57
|
-
};
|
|
58
|
-
|
|
59
|
-
const isImageMimeType = (mimeType: string) => mimeType.startsWith('image/');
|
|
60
|
-
const isAudioMimeType = (mimeType: string) => mimeType.startsWith('audio/');
|
|
61
|
-
const isVideoMimeType = (mimeType: string) => mimeType.startsWith('video/');
|
|
62
|
-
|
|
63
|
-
const toImagePart = (file: PromptFileSelection, buffer: Uint8Array): InputContentPart => {
|
|
64
|
-
const mimeType = inferMimeType(file.relativePath);
|
|
65
|
-
const dataUrl = `data:${mimeType};base64,${Buffer.from(buffer).toString('base64')}`;
|
|
66
|
-
return {
|
|
67
|
-
type: 'image_url',
|
|
68
|
-
image_url: {
|
|
69
|
-
url: dataUrl,
|
|
70
|
-
detail: 'auto',
|
|
71
|
-
},
|
|
72
|
-
};
|
|
73
|
-
};
|
|
74
|
-
|
|
75
|
-
const toTextPart = (file: PromptFileSelection, buffer: Uint8Array): InputContentPart => {
|
|
76
|
-
const rawText = TEXT_DECODER.decode(buffer);
|
|
77
|
-
const normalized = rawText.replace(/\r\n/g, '\n').replace(/\r/g, '\n');
|
|
78
|
-
const truncated = normalized.length > MAX_TEXT_ATTACHMENT_CHARS;
|
|
79
|
-
const content = truncated ? normalized.slice(0, MAX_TEXT_ATTACHMENT_CHARS) : normalized;
|
|
80
|
-
return {
|
|
81
|
-
type: 'text',
|
|
82
|
-
text: formatFileFence(file.relativePath, content, truncated),
|
|
83
|
-
};
|
|
84
|
-
};
|
|
85
|
-
|
|
86
|
-
const toAudioTextPart = (file: PromptFileSelection): InputContentPart => {
|
|
87
|
-
return {
|
|
88
|
-
type: 'text',
|
|
89
|
-
text: `Attached audio: ${file.relativePath}`,
|
|
90
|
-
};
|
|
91
|
-
};
|
|
92
|
-
|
|
93
|
-
const toVideoTextPart = (file: PromptFileSelection): InputContentPart => {
|
|
94
|
-
return {
|
|
95
|
-
type: 'text',
|
|
96
|
-
text: `Attached video: ${file.relativePath}`,
|
|
97
|
-
};
|
|
98
|
-
};
|
|
99
|
-
|
|
100
|
-
const toFileParts = async (
|
|
101
|
-
file: PromptFileSelection,
|
|
102
|
-
capabilities: AttachmentModelCapabilities
|
|
103
|
-
): Promise<InputContentPart[]> => {
|
|
104
|
-
const buffer = await readFile(file.absolutePath);
|
|
105
|
-
if (buffer.byteLength > MAX_ATTACHMENT_BYTES) {
|
|
106
|
-
throw new Error(`Attachment too large: ${file.relativePath}`);
|
|
107
|
-
}
|
|
108
|
-
|
|
109
|
-
const mimeType = inferMimeType(file.relativePath);
|
|
110
|
-
if (isImageMimeType(mimeType) && capabilities.image && isImageSelection(file)) {
|
|
111
|
-
return [
|
|
112
|
-
{
|
|
113
|
-
type: 'text',
|
|
114
|
-
text: `Attached image: ${file.relativePath}`,
|
|
115
|
-
},
|
|
116
|
-
toImagePart(file, buffer),
|
|
117
|
-
];
|
|
118
|
-
}
|
|
119
|
-
|
|
120
|
-
if (isAudioMimeType(mimeType) && capabilities.audio && isAudioSelection(file)) {
|
|
121
|
-
return [toAudioTextPart(file)];
|
|
122
|
-
}
|
|
123
|
-
|
|
124
|
-
if (isVideoMimeType(mimeType) && capabilities.video && isVideoSelection(file)) {
|
|
125
|
-
return [toVideoTextPart(file)];
|
|
126
|
-
}
|
|
127
|
-
|
|
128
|
-
return [toTextPart(file, buffer)];
|
|
129
|
-
};
|
|
130
|
-
|
|
131
|
-
export const buildPromptContent = async (
|
|
132
|
-
prompt: string,
|
|
133
|
-
files: PromptFileSelection[],
|
|
134
|
-
capabilities: AttachmentModelCapabilities
|
|
135
|
-
): Promise<MessageContent> => {
|
|
136
|
-
if (files.length === 0) {
|
|
137
|
-
return prompt;
|
|
138
|
-
}
|
|
139
|
-
|
|
140
|
-
const parts: InputContentPart[] = [];
|
|
141
|
-
if (prompt.trim().length > 0) {
|
|
142
|
-
parts.push({
|
|
143
|
-
type: 'text',
|
|
144
|
-
text: prompt,
|
|
145
|
-
});
|
|
146
|
-
}
|
|
147
|
-
|
|
148
|
-
for (const file of files) {
|
|
149
|
-
parts.push(...(await toFileParts(file, capabilities)));
|
|
150
|
-
}
|
|
151
|
-
|
|
152
|
-
return parts;
|
|
153
|
-
};
|
|
@@ -1,34 +0,0 @@
|
|
|
1
|
-
import { describe, expect, it } from 'bun:test';
|
|
2
|
-
|
|
3
|
-
import { findTrailingFileMention, removeTrailingFileMention } from './file-mention-query';
|
|
4
|
-
|
|
5
|
-
describe('file-mention-query', () => {
|
|
6
|
-
it('finds trailing @/ mention at input start', () => {
|
|
7
|
-
expect(findTrailingFileMention('@/src')).toEqual({
|
|
8
|
-
token: '@/src',
|
|
9
|
-
query: 'src',
|
|
10
|
-
start: 0,
|
|
11
|
-
end: 5,
|
|
12
|
-
});
|
|
13
|
-
});
|
|
14
|
-
|
|
15
|
-
it('finds trailing @/ mention after text', () => {
|
|
16
|
-
expect(findTrailingFileMention('check @/src/app')).toEqual({
|
|
17
|
-
token: '@/src/app',
|
|
18
|
-
query: 'src/app',
|
|
19
|
-
start: 6,
|
|
20
|
-
end: 15,
|
|
21
|
-
});
|
|
22
|
-
});
|
|
23
|
-
|
|
24
|
-
it('returns null when mention is not trailing token', () => {
|
|
25
|
-
expect(findTrailingFileMention('use @/src and continue')).toBeNull();
|
|
26
|
-
expect(findTrailingFileMention('plain text')).toBeNull();
|
|
27
|
-
});
|
|
28
|
-
|
|
29
|
-
it('removes trailing mention token and keeps prior text', () => {
|
|
30
|
-
expect(removeTrailingFileMention('@/src')).toBe('');
|
|
31
|
-
expect(removeTrailingFileMention('check @/src/app')).toBe('check ');
|
|
32
|
-
expect(removeTrailingFileMention('plain text')).toBe('plain text');
|
|
33
|
-
});
|
|
34
|
-
});
|
|
@@ -1,32 +0,0 @@
|
|
|
1
|
-
export type FileMentionMatch = {
|
|
2
|
-
token: string;
|
|
3
|
-
query: string;
|
|
4
|
-
start: number;
|
|
5
|
-
end: number;
|
|
6
|
-
};
|
|
7
|
-
|
|
8
|
-
const FILE_MENTION_PATTERN = /(^|\s)(@\/[^\s]*)$/;
|
|
9
|
-
|
|
10
|
-
export const findTrailingFileMention = (value: string): FileMentionMatch | null => {
|
|
11
|
-
const match = value.match(FILE_MENTION_PATTERN);
|
|
12
|
-
const token = match?.[2];
|
|
13
|
-
if (!token) {
|
|
14
|
-
return null;
|
|
15
|
-
}
|
|
16
|
-
|
|
17
|
-
const start = value.length - token.length;
|
|
18
|
-
return {
|
|
19
|
-
token,
|
|
20
|
-
query: token.slice(2),
|
|
21
|
-
start,
|
|
22
|
-
end: value.length,
|
|
23
|
-
};
|
|
24
|
-
};
|
|
25
|
-
|
|
26
|
-
export const removeTrailingFileMention = (value: string): string => {
|
|
27
|
-
const match = findTrailingFileMention(value);
|
|
28
|
-
if (!match) {
|
|
29
|
-
return value;
|
|
30
|
-
}
|
|
31
|
-
return value.slice(0, match.start);
|
|
32
|
-
};
|
|
@@ -1,13 +0,0 @@
|
|
|
1
|
-
import type { PromptFileSelection } from './types';
|
|
2
|
-
|
|
3
|
-
const formatSelectedFiles = (files: PromptFileSelection[]) => {
|
|
4
|
-
return files.map(file => `@${file.relativePath}`).join(' ');
|
|
5
|
-
};
|
|
6
|
-
|
|
7
|
-
export const buildPromptDisplay = (prompt: string, files: PromptFileSelection[]): string => {
|
|
8
|
-
const normalizedPrompt = prompt.trim();
|
|
9
|
-
if (normalizedPrompt.length > 0) {
|
|
10
|
-
return normalizedPrompt;
|
|
11
|
-
}
|
|
12
|
-
return files.length > 0 ? formatSelectedFiles(files) : '';
|
|
13
|
-
};
|
package/src/files/types.ts
DELETED
|
@@ -1,61 +0,0 @@
|
|
|
1
|
-
import { readdir, stat } from 'node:fs/promises';
|
|
2
|
-
import { join, relative } from 'node:path';
|
|
3
|
-
|
|
4
|
-
import { resolveWorkspaceRoot } from '../agent/runtime/source-modules';
|
|
5
|
-
import type { PromptFileSelection } from './types';
|
|
6
|
-
|
|
7
|
-
const IGNORED_DIR_NAMES = new Set([
|
|
8
|
-
'.git',
|
|
9
|
-
'node_modules',
|
|
10
|
-
'dist',
|
|
11
|
-
'build',
|
|
12
|
-
'coverage',
|
|
13
|
-
'.next',
|
|
14
|
-
'.turbo',
|
|
15
|
-
]);
|
|
16
|
-
|
|
17
|
-
const comparePaths = (a: PromptFileSelection, b: PromptFileSelection) => {
|
|
18
|
-
return a.relativePath.localeCompare(b.relativePath);
|
|
19
|
-
};
|
|
20
|
-
|
|
21
|
-
const visitDirectory = async (
|
|
22
|
-
root: string,
|
|
23
|
-
directory: string,
|
|
24
|
-
output: PromptFileSelection[]
|
|
25
|
-
): Promise<void> => {
|
|
26
|
-
const entries = await readdir(directory, { withFileTypes: true }).catch(() => []);
|
|
27
|
-
|
|
28
|
-
const sortedEntries = [...entries].sort((a, b) => a.name.localeCompare(b.name));
|
|
29
|
-
for (const entry of sortedEntries) {
|
|
30
|
-
const absolutePath = join(directory, entry.name);
|
|
31
|
-
if (entry.isDirectory()) {
|
|
32
|
-
if (IGNORED_DIR_NAMES.has(entry.name)) {
|
|
33
|
-
continue;
|
|
34
|
-
}
|
|
35
|
-
await visitDirectory(root, absolutePath, output);
|
|
36
|
-
continue;
|
|
37
|
-
}
|
|
38
|
-
|
|
39
|
-
if (!entry.isFile()) {
|
|
40
|
-
continue;
|
|
41
|
-
}
|
|
42
|
-
|
|
43
|
-
const fileStat = await stat(absolutePath).catch(() => undefined);
|
|
44
|
-
if (!fileStat || !fileStat.isFile()) {
|
|
45
|
-
continue;
|
|
46
|
-
}
|
|
47
|
-
|
|
48
|
-
output.push({
|
|
49
|
-
relativePath: relative(root, absolutePath),
|
|
50
|
-
absolutePath,
|
|
51
|
-
size: fileStat.size,
|
|
52
|
-
});
|
|
53
|
-
}
|
|
54
|
-
};
|
|
55
|
-
|
|
56
|
-
export const listWorkspaceFiles = async (): Promise<PromptFileSelection[]> => {
|
|
57
|
-
const workspaceRoot = resolveWorkspaceRoot();
|
|
58
|
-
const files: PromptFileSelection[] = [];
|
|
59
|
-
await visitDirectory(workspaceRoot, workspaceRoot, files);
|
|
60
|
-
return files.sort(comparePaths);
|
|
61
|
-
};
|
|
@@ -1,207 +0,0 @@
|
|
|
1
|
-
import { describe, expect, it } from 'vitest';
|
|
2
|
-
|
|
3
|
-
import type {
|
|
4
|
-
AgentToolResultEvent,
|
|
5
|
-
AgentToolStreamEvent,
|
|
6
|
-
AgentToolUseEvent,
|
|
7
|
-
} from '../agent/runtime/types';
|
|
8
|
-
import { buildAgentEventHandlers } from './agent-event-handlers';
|
|
9
|
-
import { appendToSegment } from './turn-updater';
|
|
10
|
-
import type { ReplySegment } from '../types/chat';
|
|
11
|
-
|
|
12
|
-
const buildHarness = () => {
|
|
13
|
-
const turnId = 1;
|
|
14
|
-
let segments: ReplySegment[] = [];
|
|
15
|
-
const notes: string[] = [];
|
|
16
|
-
|
|
17
|
-
const handlers = buildAgentEventHandlers({
|
|
18
|
-
turnId,
|
|
19
|
-
isCurrentRequest: () => true,
|
|
20
|
-
appendSegment: (_turnId, segmentId, type, chunk, data) => {
|
|
21
|
-
segments = appendToSegment(segments, segmentId, type, chunk, data);
|
|
22
|
-
},
|
|
23
|
-
appendEventLine: (_turnId, text) => {
|
|
24
|
-
notes.push(text);
|
|
25
|
-
},
|
|
26
|
-
});
|
|
27
|
-
|
|
28
|
-
return {
|
|
29
|
-
turnId,
|
|
30
|
-
handlers,
|
|
31
|
-
readSegments: () => segments,
|
|
32
|
-
notes,
|
|
33
|
-
};
|
|
34
|
-
};
|
|
35
|
-
|
|
36
|
-
const createToolUseEvent = (): AgentToolUseEvent => ({
|
|
37
|
-
id: 'call_1',
|
|
38
|
-
function: {
|
|
39
|
-
name: 'bash',
|
|
40
|
-
arguments: JSON.stringify({ command: 'ls -la' }),
|
|
41
|
-
},
|
|
42
|
-
});
|
|
43
|
-
|
|
44
|
-
const createStdoutStreamEvent = (): AgentToolStreamEvent => ({
|
|
45
|
-
toolCallId: 'call_1',
|
|
46
|
-
toolName: 'bash',
|
|
47
|
-
type: 'stdout',
|
|
48
|
-
sequence: 1,
|
|
49
|
-
timestamp: Date.now(),
|
|
50
|
-
content: 'total 80',
|
|
51
|
-
});
|
|
52
|
-
|
|
53
|
-
const createToolResultEvent = (): AgentToolResultEvent => ({
|
|
54
|
-
toolCall: {
|
|
55
|
-
id: 'call_1',
|
|
56
|
-
function: {
|
|
57
|
-
name: 'bash',
|
|
58
|
-
arguments: JSON.stringify({ command: 'ls -la' }),
|
|
59
|
-
},
|
|
60
|
-
},
|
|
61
|
-
result: {
|
|
62
|
-
success: true,
|
|
63
|
-
data: {
|
|
64
|
-
output: 'total 80',
|
|
65
|
-
},
|
|
66
|
-
},
|
|
67
|
-
});
|
|
68
|
-
|
|
69
|
-
const createEmptyToolResultEvent = (): AgentToolResultEvent => ({
|
|
70
|
-
toolCall: {
|
|
71
|
-
id: 'call_2',
|
|
72
|
-
function: {
|
|
73
|
-
name: 'bash',
|
|
74
|
-
arguments: JSON.stringify({ command: 'find /tmp -name missing' }),
|
|
75
|
-
},
|
|
76
|
-
},
|
|
77
|
-
result: {
|
|
78
|
-
success: true,
|
|
79
|
-
data: {
|
|
80
|
-
summary: 'Command completed successfully with no output.',
|
|
81
|
-
},
|
|
82
|
-
},
|
|
83
|
-
});
|
|
84
|
-
|
|
85
|
-
describe('buildAgentEventHandlers', () => {
|
|
86
|
-
it('keeps ordered stream segments as thinking -> text -> thinking -> tool -> tool result', () => {
|
|
87
|
-
const { handlers, readSegments, turnId } = buildHarness();
|
|
88
|
-
|
|
89
|
-
handlers.onTextDelta?.({
|
|
90
|
-
text: '先想一下要做什么。',
|
|
91
|
-
isReasoning: true,
|
|
92
|
-
});
|
|
93
|
-
handlers.onTextDelta?.({
|
|
94
|
-
text: '当前目录看起来是一个 TypeScript 项目。',
|
|
95
|
-
isReasoning: false,
|
|
96
|
-
});
|
|
97
|
-
handlers.onTextDelta?.({
|
|
98
|
-
text: '我会先执行 ls -la 看目录结构。',
|
|
99
|
-
isReasoning: true,
|
|
100
|
-
});
|
|
101
|
-
handlers.onToolUse?.(createToolUseEvent());
|
|
102
|
-
handlers.onToolStream?.(createStdoutStreamEvent());
|
|
103
|
-
handlers.onToolResult?.(createToolResultEvent());
|
|
104
|
-
handlers.onTextDelta?.({
|
|
105
|
-
text: '当前目录包含以下内容。',
|
|
106
|
-
isReasoning: false,
|
|
107
|
-
});
|
|
108
|
-
handlers.onTextComplete?.('');
|
|
109
|
-
|
|
110
|
-
const segments = readSegments();
|
|
111
|
-
expect(segments.map(segment => segment.id)).toEqual([
|
|
112
|
-
`${turnId}:thinking:1`,
|
|
113
|
-
`${turnId}:text:2`,
|
|
114
|
-
`${turnId}:thinking:3`,
|
|
115
|
-
`${turnId}:tool-use:call_1`,
|
|
116
|
-
`${turnId}:tool:call_1:stdout`,
|
|
117
|
-
`${turnId}:tool-result:call_1`,
|
|
118
|
-
`${turnId}:text:4`,
|
|
119
|
-
]);
|
|
120
|
-
|
|
121
|
-
const firstThinkingIndex = segments.findIndex(segment => segment.id === `${turnId}:thinking:1`);
|
|
122
|
-
const toolUseIndex = segments.findIndex(segment => segment.id === `${turnId}:tool-use:call_1`);
|
|
123
|
-
const toolResultIndex = segments.findIndex(
|
|
124
|
-
segment => segment.id === `${turnId}:tool-result:call_1`
|
|
125
|
-
);
|
|
126
|
-
expect(firstThinkingIndex).toBeGreaterThanOrEqual(0);
|
|
127
|
-
expect(toolUseIndex).toBeGreaterThan(firstThinkingIndex);
|
|
128
|
-
expect(toolResultIndex).toBeGreaterThan(toolUseIndex);
|
|
129
|
-
});
|
|
130
|
-
|
|
131
|
-
it('suppresses duplicated tool output in tool-result when stdout/stderr stream already exists', () => {
|
|
132
|
-
const { handlers, readSegments, turnId } = buildHarness();
|
|
133
|
-
|
|
134
|
-
handlers.onToolUse?.(createToolUseEvent());
|
|
135
|
-
handlers.onToolStream?.(createStdoutStreamEvent());
|
|
136
|
-
handlers.onToolResult?.(createToolResultEvent());
|
|
137
|
-
|
|
138
|
-
const toolResult = readSegments().find(
|
|
139
|
-
segment => segment.id === `${turnId}:tool-result:call_1`
|
|
140
|
-
);
|
|
141
|
-
expect(toolResult?.content).toContain('# Result: bash (call_1) success');
|
|
142
|
-
expect(toolResult?.content).not.toContain('total 80');
|
|
143
|
-
});
|
|
144
|
-
|
|
145
|
-
it('keeps tool-result output when no stdout/stderr stream was emitted', () => {
|
|
146
|
-
const { handlers, readSegments, turnId } = buildHarness();
|
|
147
|
-
|
|
148
|
-
handlers.onToolUse?.(createToolUseEvent());
|
|
149
|
-
handlers.onToolResult?.(createToolResultEvent());
|
|
150
|
-
|
|
151
|
-
const toolResult = readSegments().find(
|
|
152
|
-
segment => segment.id === `${turnId}:tool-result:call_1`
|
|
153
|
-
);
|
|
154
|
-
expect(toolResult?.content).toContain('# Result: bash (call_1) success');
|
|
155
|
-
expect(toolResult?.content).toContain('total 80');
|
|
156
|
-
});
|
|
157
|
-
|
|
158
|
-
it('renders a summary instead of an output wrapper when tool succeeds without output', () => {
|
|
159
|
-
const { handlers, readSegments, turnId } = buildHarness();
|
|
160
|
-
|
|
161
|
-
handlers.onToolUse?.({
|
|
162
|
-
id: 'call_2',
|
|
163
|
-
function: {
|
|
164
|
-
name: 'bash',
|
|
165
|
-
arguments: JSON.stringify({ command: 'find /tmp -name missing' }),
|
|
166
|
-
},
|
|
167
|
-
});
|
|
168
|
-
handlers.onToolResult?.(createEmptyToolResultEvent());
|
|
169
|
-
|
|
170
|
-
const toolResult = readSegments().find(
|
|
171
|
-
segment => segment.id === `${turnId}:tool-result:call_2`
|
|
172
|
-
);
|
|
173
|
-
expect(toolResult?.content).toContain('# Result: bash (call_2) success');
|
|
174
|
-
expect(toolResult?.content).toContain('Command completed successfully with no output.');
|
|
175
|
-
expect(toolResult?.content).not.toContain('{"output":""}');
|
|
176
|
-
});
|
|
177
|
-
|
|
178
|
-
it('stores structured tool event data on tool-use and tool-result segments', () => {
|
|
179
|
-
const { handlers, readSegments, turnId } = buildHarness();
|
|
180
|
-
|
|
181
|
-
const toolUseEvent = createToolUseEvent();
|
|
182
|
-
const toolResultEvent = createToolResultEvent();
|
|
183
|
-
handlers.onToolUse?.(toolUseEvent);
|
|
184
|
-
handlers.onToolResult?.(toolResultEvent);
|
|
185
|
-
|
|
186
|
-
const toolUse = readSegments().find(segment => segment.id === `${turnId}:tool-use:call_1`);
|
|
187
|
-
const toolResult = readSegments().find(
|
|
188
|
-
segment => segment.id === `${turnId}:tool-result:call_1`
|
|
189
|
-
);
|
|
190
|
-
|
|
191
|
-
expect(toolUse?.data).toEqual(toolUseEvent);
|
|
192
|
-
expect(toolResult?.data).toEqual(toolResultEvent);
|
|
193
|
-
});
|
|
194
|
-
|
|
195
|
-
it('deduplicates repeated tool-use events for the same toolCallId', () => {
|
|
196
|
-
const { handlers, readSegments, turnId } = buildHarness();
|
|
197
|
-
|
|
198
|
-
handlers.onToolUse?.(createToolUseEvent());
|
|
199
|
-
handlers.onToolUse?.(createToolUseEvent());
|
|
200
|
-
handlers.onToolUse?.(createToolUseEvent());
|
|
201
|
-
|
|
202
|
-
const toolUseSegments = readSegments().filter(
|
|
203
|
-
segment => segment.id === `${turnId}:tool-use:call_1`
|
|
204
|
-
);
|
|
205
|
-
expect(toolUseSegments.length).toBe(1);
|
|
206
|
-
});
|
|
207
|
-
});
|
|
@@ -1,196 +0,0 @@
|
|
|
1
|
-
import {
|
|
2
|
-
formatLoopEvent,
|
|
3
|
-
formatStepEvent,
|
|
4
|
-
formatStopEvent,
|
|
5
|
-
formatToolConfirmEvent,
|
|
6
|
-
formatToolResultEvent,
|
|
7
|
-
formatToolResultEventCode,
|
|
8
|
-
formatToolStreamEvent,
|
|
9
|
-
formatToolUseEvent,
|
|
10
|
-
formatToolUseEventCode,
|
|
11
|
-
} from '../agent/runtime/event-format';
|
|
12
|
-
import type {
|
|
13
|
-
AgentEventHandlers,
|
|
14
|
-
AgentToolResultEvent,
|
|
15
|
-
AgentToolUseEvent,
|
|
16
|
-
} from '../agent/runtime/types';
|
|
17
|
-
import type { ReplySegmentType } from '../types/chat';
|
|
18
|
-
|
|
19
|
-
type BuildAgentEventHandlersParams = {
|
|
20
|
-
turnId: number;
|
|
21
|
-
isCurrentRequest: () => boolean;
|
|
22
|
-
appendSegment: (
|
|
23
|
-
turnId: number,
|
|
24
|
-
segmentId: string,
|
|
25
|
-
type: ReplySegmentType,
|
|
26
|
-
chunk: string,
|
|
27
|
-
data?: unknown
|
|
28
|
-
) => void;
|
|
29
|
-
appendEventLine: (turnId: number, text: string) => void;
|
|
30
|
-
};
|
|
31
|
-
|
|
32
|
-
const shouldShowEventLog = () => {
|
|
33
|
-
const value = process.env.AGENT_SHOW_EVENTS?.trim().toLowerCase();
|
|
34
|
-
return value === '1' || value === 'true' || value === 'yes' || value === 'on';
|
|
35
|
-
};
|
|
36
|
-
|
|
37
|
-
export const buildAgentEventHandlers = ({
|
|
38
|
-
turnId,
|
|
39
|
-
isCurrentRequest,
|
|
40
|
-
appendSegment,
|
|
41
|
-
appendEventLine,
|
|
42
|
-
}: BuildAgentEventHandlersParams): AgentEventHandlers => {
|
|
43
|
-
const showEvents = shouldShowEventLog();
|
|
44
|
-
const streamedToolCallIds = new Set<string>();
|
|
45
|
-
const renderedToolUseIds = new Set<string>();
|
|
46
|
-
let anonymousToolUseCounter = 0;
|
|
47
|
-
let anonymousToolResultCounter = 0;
|
|
48
|
-
let streamSegmentCursor = 0;
|
|
49
|
-
let activeTextSegment: {
|
|
50
|
-
id: string;
|
|
51
|
-
type: 'thinking' | 'text';
|
|
52
|
-
} | null = null;
|
|
53
|
-
|
|
54
|
-
const createStreamSegmentId = (type: 'thinking' | 'text') => {
|
|
55
|
-
streamSegmentCursor += 1;
|
|
56
|
-
return `${turnId}:${type}:${streamSegmentCursor}`;
|
|
57
|
-
};
|
|
58
|
-
|
|
59
|
-
const appendTextDeltaInOrder = (text: string, isReasoning: boolean) => {
|
|
60
|
-
const type: 'thinking' | 'text' = isReasoning ? 'thinking' : 'text';
|
|
61
|
-
if (!activeTextSegment || activeTextSegment.type !== type) {
|
|
62
|
-
activeTextSegment = {
|
|
63
|
-
id: createStreamSegmentId(type),
|
|
64
|
-
type,
|
|
65
|
-
};
|
|
66
|
-
}
|
|
67
|
-
appendSegment(turnId, activeTextSegment.id, type, text);
|
|
68
|
-
};
|
|
69
|
-
|
|
70
|
-
const breakTextDeltaContinuation = () => {
|
|
71
|
-
activeTextSegment = null;
|
|
72
|
-
};
|
|
73
|
-
|
|
74
|
-
const readToolCallIdFromResult = (event: AgentToolResultEvent): string | undefined => {
|
|
75
|
-
if (!event.toolCall || typeof event.toolCall !== 'object') {
|
|
76
|
-
return undefined;
|
|
77
|
-
}
|
|
78
|
-
const maybeId = (event.toolCall as { id?: unknown }).id;
|
|
79
|
-
return typeof maybeId === 'string' ? maybeId : undefined;
|
|
80
|
-
};
|
|
81
|
-
|
|
82
|
-
const readToolCallIdFromUse = (event: AgentToolUseEvent): string | undefined => {
|
|
83
|
-
if (!event || typeof event !== 'object') {
|
|
84
|
-
return undefined;
|
|
85
|
-
}
|
|
86
|
-
const maybeId = (event as { id?: unknown }).id;
|
|
87
|
-
return typeof maybeId === 'string' ? maybeId : undefined;
|
|
88
|
-
};
|
|
89
|
-
|
|
90
|
-
const logEvent = (text: string) => {
|
|
91
|
-
if (!showEvents) {
|
|
92
|
-
return;
|
|
93
|
-
}
|
|
94
|
-
appendEventLine(turnId, text);
|
|
95
|
-
};
|
|
96
|
-
|
|
97
|
-
return {
|
|
98
|
-
onTextDelta: event => {
|
|
99
|
-
if (!isCurrentRequest() || !event.text) {
|
|
100
|
-
return;
|
|
101
|
-
}
|
|
102
|
-
appendTextDeltaInOrder(event.text, Boolean(event.isReasoning));
|
|
103
|
-
},
|
|
104
|
-
onTextComplete: () => {
|
|
105
|
-
if (!isCurrentRequest()) {
|
|
106
|
-
return;
|
|
107
|
-
}
|
|
108
|
-
breakTextDeltaContinuation();
|
|
109
|
-
logEvent('[text-complete]');
|
|
110
|
-
},
|
|
111
|
-
onToolStream: event => {
|
|
112
|
-
if (!isCurrentRequest()) {
|
|
113
|
-
return;
|
|
114
|
-
}
|
|
115
|
-
breakTextDeltaContinuation();
|
|
116
|
-
const mapped = formatToolStreamEvent(event);
|
|
117
|
-
if (mapped.codeChunk && mapped.segmentKey) {
|
|
118
|
-
appendSegment(turnId, `${turnId}:tool:${mapped.segmentKey}`, 'code', mapped.codeChunk);
|
|
119
|
-
}
|
|
120
|
-
if (
|
|
121
|
-
(event.type === 'stdout' || event.type === 'stderr') &&
|
|
122
|
-
typeof event.toolCallId === 'string' &&
|
|
123
|
-
event.toolCallId.length > 0
|
|
124
|
-
) {
|
|
125
|
-
streamedToolCallIds.add(event.toolCallId);
|
|
126
|
-
}
|
|
127
|
-
if (mapped.note) {
|
|
128
|
-
logEvent(mapped.note);
|
|
129
|
-
}
|
|
130
|
-
},
|
|
131
|
-
onToolConfirm: event => {
|
|
132
|
-
if (!isCurrentRequest()) {
|
|
133
|
-
return;
|
|
134
|
-
}
|
|
135
|
-
logEvent(formatToolConfirmEvent(event));
|
|
136
|
-
},
|
|
137
|
-
onToolUse: event => {
|
|
138
|
-
if (!isCurrentRequest()) {
|
|
139
|
-
return;
|
|
140
|
-
}
|
|
141
|
-
breakTextDeltaContinuation();
|
|
142
|
-
const toolCallId = readToolCallIdFromUse(event);
|
|
143
|
-
if (toolCallId && renderedToolUseIds.has(toolCallId)) {
|
|
144
|
-
return;
|
|
145
|
-
}
|
|
146
|
-
if (toolCallId) {
|
|
147
|
-
renderedToolUseIds.add(toolCallId);
|
|
148
|
-
}
|
|
149
|
-
const segmentSuffix = toolCallId ?? `anonymous_${++anonymousToolUseCounter}`;
|
|
150
|
-
appendSegment(
|
|
151
|
-
turnId,
|
|
152
|
-
`${turnId}:tool-use:${segmentSuffix}`,
|
|
153
|
-
'code',
|
|
154
|
-
`${formatToolUseEventCode(event)}\n`,
|
|
155
|
-
event
|
|
156
|
-
);
|
|
157
|
-
logEvent(formatToolUseEvent(event));
|
|
158
|
-
},
|
|
159
|
-
onToolResult: event => {
|
|
160
|
-
if (!isCurrentRequest()) {
|
|
161
|
-
return;
|
|
162
|
-
}
|
|
163
|
-
breakTextDeltaContinuation();
|
|
164
|
-
const toolCallId = readToolCallIdFromResult(event);
|
|
165
|
-
const suppressOutput = Boolean(toolCallId && streamedToolCallIds.has(toolCallId));
|
|
166
|
-
const segmentSuffix = toolCallId ?? `anonymous_${++anonymousToolResultCounter}`;
|
|
167
|
-
appendSegment(
|
|
168
|
-
turnId,
|
|
169
|
-
`${turnId}:tool-result:${segmentSuffix}`,
|
|
170
|
-
'code',
|
|
171
|
-
`${formatToolResultEventCode(event, { suppressOutput })}\n`,
|
|
172
|
-
event
|
|
173
|
-
);
|
|
174
|
-
logEvent(formatToolResultEvent(event));
|
|
175
|
-
},
|
|
176
|
-
onStep: event => {
|
|
177
|
-
if (!isCurrentRequest()) {
|
|
178
|
-
return;
|
|
179
|
-
}
|
|
180
|
-
logEvent(formatStepEvent(event));
|
|
181
|
-
},
|
|
182
|
-
onLoop: event => {
|
|
183
|
-
if (!isCurrentRequest()) {
|
|
184
|
-
return;
|
|
185
|
-
}
|
|
186
|
-
logEvent(formatLoopEvent(event));
|
|
187
|
-
},
|
|
188
|
-
onStop: event => {
|
|
189
|
-
if (!isCurrentRequest()) {
|
|
190
|
-
return;
|
|
191
|
-
}
|
|
192
|
-
breakTextDeltaContinuation();
|
|
193
|
-
logEvent(formatStopEvent(event));
|
|
194
|
-
},
|
|
195
|
-
};
|
|
196
|
-
};
|