march-cli 0.1.24 → 0.1.26
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/bin/march.mjs +13 -13
- package/package.json +49 -49
- package/src/agent/command-exec-tool.mjs +172 -172
- package/src/agent/context-stats-tool.mjs +57 -57
- package/src/agent/editing/diff-apply.mjs +28 -28
- package/src/agent/editing/diff-format.mjs +57 -57
- package/src/agent/editing/lsp-report.mjs +69 -69
- package/src/agent/file-edit-tool.mjs +250 -250
- package/src/agent/file-tools/read-file-tool.mjs +112 -112
- package/src/agent/file-tools/read-image-tool.mjs +76 -76
- package/src/agent/model-payload-dumper.mjs +208 -208
- package/src/agent/pi-session/pi-session-sidecar-failure.mjs +10 -10
- package/src/agent/provider/payload-messages.mjs +138 -138
- package/src/agent/runner/codex-large-context-guard.mjs +87 -87
- package/src/agent/runner/codex-transport-compression.mjs +180 -180
- package/src/agent/runner/codex-transport-debug.mjs +113 -113
- package/src/agent/runner/codex-websocket-event-debug.mjs +130 -130
- package/src/agent/runner/fast-model.mjs +36 -36
- package/src/agent/runner/runner-cleanup.mjs +12 -12
- package/src/agent/runner/runner-init.mjs +15 -15
- package/src/agent/runner/runner-session-state.mjs +40 -40
- package/src/agent/runner/runner-utils.mjs +24 -24
- package/src/agent/runner.mjs +299 -299
- package/src/agent/runtime/ipc/ipc-peer.mjs +99 -99
- package/src/agent/runtime/ipc/process-ipc-transport.mjs +16 -16
- package/src/agent/runtime/remote-runner-client.mjs +73 -73
- package/src/agent/runtime/remote-ui-client.mjs +20 -20
- package/src/agent/runtime/runner-ipc-target.mjs +125 -125
- package/src/agent/runtime/runner-process-client.mjs +47 -47
- package/src/agent/runtime/runner-process-entry.mjs +11 -11
- package/src/agent/runtime/runner-process-factory.mjs +111 -108
- package/src/agent/runtime/runner-runtime-host.mjs +79 -79
- package/src/agent/runtime/runtime-factory.mjs +42 -42
- package/src/agent/runtime/runtime-host.mjs +34 -34
- package/src/agent/runtime/ui-event-bridge.mjs +95 -95
- package/src/agent/screen-tools/list-windows-tool.mjs +39 -39
- package/src/agent/screen-tools/screen-tool.mjs +49 -49
- package/src/agent/screen-tools/windows-screen.mjs +133 -133
- package/src/agent/session/session-auto-name.mjs +41 -41
- package/src/agent/session/session-binding.mjs +12 -12
- package/src/agent/session/session-options.mjs +47 -47
- package/src/agent/tool-names.mjs +1 -1
- package/src/agent/tool-result.mjs +3 -3
- package/src/agent/tool-summary.mjs +112 -112
- package/src/agent/tools.mjs +58 -58
- package/src/agent/turn/turn-events.mjs +111 -111
- package/src/agent/turn/turn-logging.mjs +30 -30
- package/src/agent/turn/turn-runner.mjs +196 -196
- package/src/agent/vision-capability.mjs +14 -14
- package/src/auth/login-command.mjs +90 -90
- package/src/auth/storage.mjs +34 -34
- package/src/cli/args.mjs +96 -79
- package/src/cli/commands/copy-command.mjs +87 -87
- package/src/cli/commands/export-command.mjs +206 -206
- package/src/cli/commands/extensions-command.mjs +53 -53
- package/src/cli/commands/help-command.mjs +7 -7
- package/src/cli/commands/model-command.mjs +141 -141
- package/src/cli/commands/paste-image-command.mjs +43 -43
- package/src/cli/commands/provider-command.mjs +59 -59
- package/src/cli/commands/status-command.mjs +196 -194
- package/src/cli/commands/thinking-command.mjs +87 -87
- package/src/cli/fallback-ui.mjs +156 -156
- package/src/cli/input/attachment-tokens.mjs +20 -20
- package/src/cli/input/autocomplete.mjs +74 -74
- package/src/cli/input/external-editor.mjs +39 -39
- package/src/cli/input/file-search/index.mjs +160 -160
- package/src/cli/input/history-store.mjs +35 -35
- package/src/cli/input/image-clipboard.mjs +55 -55
- package/src/cli/input/keybinding-dispatch.mjs +76 -76
- package/src/cli/input/keybindings.mjs +96 -96
- package/src/cli/input/mode-state.mjs +43 -43
- package/src/cli/input/prompt-templates.mjs +84 -84
- package/src/cli/input/select-with-keyboard.mjs +86 -86
- package/src/cli/permissions.mjs +103 -103
- package/src/cli/repl-commands.mjs +86 -86
- package/src/cli/repl-loop.mjs +183 -183
- package/src/cli/selector-list.mjs +21 -21
- package/src/cli/session/pi-session-switch-command.mjs +41 -41
- package/src/cli/session/session-command.mjs +23 -23
- package/src/cli/session/session-list-command.mjs +68 -68
- package/src/cli/session/session-name-command.mjs +26 -26
- package/src/cli/session/session-source-command.mjs +89 -89
- package/src/cli/session/session-switch-command.mjs +1 -1
- package/src/cli/shell/shell-command.mjs +55 -55
- package/src/cli/shell/shell-drawer-controls.mjs +33 -33
- package/src/cli/shell/shell-drawer.mjs +192 -192
- package/src/cli/shell/shell-split-layout.mjs +70 -70
- package/src/cli/slash-commands.mjs +192 -192
- package/src/cli/startup/create-runtime-runner.mjs +61 -61
- package/src/cli/startup/runtime-close.mjs +23 -23
- package/src/cli/startup/startup-banner.mjs +71 -71
- package/src/cli/startup/startup-session.mjs +51 -51
- package/src/cli/status-line-updater.mjs +75 -75
- package/src/cli/tool-output.mjs +9 -9
- package/src/cli/tui/editor/external-editor-runner.mjs +24 -24
- package/src/cli/tui/input/mouse-selection-controller.mjs +91 -91
- package/src/cli/tui/input/mouse-tracking.mjs +20 -20
- package/src/cli/tui/layout/main-pane-layout.mjs +47 -47
- package/src/cli/tui/layout/safe-render-boundary.mjs +46 -46
- package/src/cli/tui/markdown-renderer.mjs +285 -285
- package/src/cli/tui/output/scroll-state.mjs +79 -79
- package/src/cli/tui/output/text-line-renderer.mjs +50 -50
- package/src/cli/tui/output/tool-card-renderer.mjs +59 -59
- package/src/cli/tui/output/visible-lines.mjs +8 -8
- package/src/cli/tui/output-buffer.mjs +293 -293
- package/src/cli/tui/permission-request-ui.mjs +18 -18
- package/src/cli/tui/recall-rendering.mjs +28 -25
- package/src/cli/tui/render/render-scheduler.mjs +26 -26
- package/src/cli/tui/render/stream-delta-buffer.mjs +46 -46
- package/src/cli/tui/select/editor-select-list.mjs +111 -111
- package/src/cli/tui/selection-screen.mjs +269 -269
- package/src/cli/tui/status/retry-status.mjs +72 -72
- package/src/cli/tui/status/spinner-status.mjs +42 -42
- package/src/cli/tui/status/status-bar.mjs +225 -225
- package/src/cli/tui/syntax/highlighting.mjs +260 -260
- package/src/cli/tui/syntax/languages.mjs +91 -91
- package/src/cli/tui/syntax/tree-sitter/bash.highlights.scm +261 -261
- package/src/cli/tui/syntax/tree-sitter/c.highlights.scm +341 -341
- package/src/cli/tui/syntax/tree-sitter/cpp.highlights.scm +268 -268
- package/src/cli/tui/syntax/tree-sitter/csharp.highlights.scm +577 -577
- package/src/cli/tui/syntax/tree-sitter/css.highlights.scm +109 -109
- package/src/cli/tui/syntax/tree-sitter/diff.highlights.scm +49 -49
- package/src/cli/tui/syntax/tree-sitter/go.highlights.scm +254 -254
- package/src/cli/tui/syntax/tree-sitter/html.highlights.scm +13 -13
- package/src/cli/tui/syntax/tree-sitter/java.highlights.scm +330 -330
- package/src/cli/tui/syntax/tree-sitter/json.highlights.scm +38 -38
- package/src/cli/tui/syntax/tree-sitter/php.highlights.scm +203 -203
- package/src/cli/tui/syntax/tree-sitter/python.highlights.scm +137 -137
- package/src/cli/tui/syntax/tree-sitter/ruby.highlights.scm +309 -309
- package/src/cli/tui/syntax/tree-sitter/rust.highlights.scm +531 -531
- package/src/cli/tui/syntax/tree-sitter/toml.highlights.scm +39 -39
- package/src/cli/tui/syntax/tree-sitter/tsx.highlights.scm +35 -35
- package/src/cli/tui/syntax/tree-sitter/typescript.highlights.scm +35 -35
- package/src/cli/tui/syntax/tree-sitter/yaml.highlights.scm +99 -99
- package/src/cli/tui/tool-rendering.mjs +87 -87
- package/src/cli/tui/tui-diff-rendering.mjs +157 -157
- package/src/cli/tui/tui-handlers.mjs +111 -111
- package/src/cli/tui/tui-input-controller.mjs +61 -61
- package/src/cli/tui/ui-theme.mjs +157 -157
- package/src/cli/ui.mjs +297 -297
- package/src/config/config-json.mjs +108 -84
- package/src/config/dotenv.mjs +20 -20
- package/src/config/features.mjs +75 -75
- package/src/config/loader.mjs +156 -143
- package/src/config/settings-command.mjs +97 -97
- package/src/context/engine.mjs +199 -198
- package/src/context/injections.mjs +26 -26
- package/src/context/profiles.mjs +39 -39
- package/src/context/project-context.mjs +20 -20
- package/src/context/session-status.mjs +25 -17
- package/src/context/shell-layers.mjs +23 -23
- package/src/context/system-core/base.md +51 -50
- package/src/context/system-core/prompts/deepseek-v4-pro.md +3 -3
- package/src/context/system-core/prompts/default.md +3 -3
- package/src/context/system-core.mjs +35 -35
- package/src/debug/logger.mjs +141 -141
- package/src/debug/model-context-dumper.mjs +52 -52
- package/src/extensions/discovery.mjs +40 -40
- package/src/extensions/lifecycle-adapter.mjs +210 -210
- package/src/extensions/lifecycle-manifest.mjs +69 -69
- package/src/image-gen/index.mjs +7 -7
- package/src/image-gen/provider.mjs +231 -231
- package/src/image-gen/tool.mjs +84 -84
- package/src/lsp/client.mjs +257 -257
- package/src/lsp/diagnostic-store.mjs +42 -42
- package/src/lsp/diagnostics-format.mjs +72 -72
- package/src/lsp/managed-node-server.mjs +99 -99
- package/src/lsp/path-match.mjs +10 -10
- package/src/lsp/server-definitions.mjs +188 -188
- package/src/lsp/servers.mjs +165 -165
- package/src/lsp/service.mjs +110 -110
- package/src/lsp/status-message.mjs +9 -9
- package/src/lsp/typescript-project-resolver.mjs +186 -186
- package/src/main.mjs +294 -299
- package/src/mcp/client.mjs +195 -195
- package/src/mcp/config.mjs +130 -130
- package/src/mcp/index.mjs +48 -48
- package/src/mcp/tools.mjs +98 -98
- package/src/memory/command.mjs +120 -0
- package/src/memory/markdown/markdown-delete.mjs +23 -23
- package/src/memory/markdown/markdown-format.mjs +128 -128
- package/src/memory/markdown/markdown-recall.mjs +28 -28
- package/src/memory/markdown/ripgrep.mjs +16 -16
- package/src/memory/markdown/sqlite-index.mjs +87 -87
- package/src/memory/markdown-store.mjs +272 -286
- package/src/memory/markdown-tools.mjs +174 -103
- package/src/memory/remote/client.mjs +68 -0
- package/src/memory/remote/config.mjs +52 -0
- package/src/memory/remote/server.mjs +99 -0
- package/src/memory/search.mjs +183 -0
- package/src/network/environment.mjs +131 -131
- package/src/notification/desktop-notifier.mjs +262 -262
- package/src/platform/open-file.mjs +28 -28
- package/src/platform/spawn-command.mjs +27 -27
- package/src/provider/accept-command.mjs +89 -89
- package/src/provider/command.mjs +21 -21
- package/src/provider/config-command.mjs +129 -129
- package/src/provider/custom-provider.mjs +113 -113
- package/src/provider/hosted-tools.mjs +111 -111
- package/src/provider/presets.mjs +72 -72
- package/src/provider/share-command.mjs +79 -79
- package/src/provider/share-payload.mjs +52 -52
- package/src/session/attachment-display.mjs +16 -16
- package/src/session/attachment-references.mjs +65 -65
- package/src/session/attachments.mjs +140 -140
- package/src/session/persist.mjs +1 -1
- package/src/session/pi-manager.mjs +34 -34
- package/src/session/session-utils.mjs +16 -16
- package/src/session/sidecar-sync.mjs +19 -19
- package/src/session/sidecar.mjs +69 -69
- package/src/session/transcript.mjs +83 -83
- package/src/session/tree.mjs +42 -42
- package/src/shell/cli-runtime.mjs +11 -11
- package/src/shell/hints.mjs +12 -12
- package/src/shell/node-pty-adapter.mjs +81 -81
- package/src/shell/runtime-state.mjs +126 -126
- package/src/shell/runtime.mjs +252 -252
- package/src/shell/screen-buffer.mjs +136 -136
- package/src/shell/tool-read.mjs +74 -74
- package/src/shell/tools.mjs +299 -299
- package/src/supergrok/actions/image-generate.mjs +60 -60
- package/src/supergrok/actions/search.mjs +78 -78
- package/src/supergrok/auth.mjs +36 -36
- package/src/supergrok/constants.mjs +18 -18
- package/src/supergrok/oauth-provider.mjs +278 -278
- package/src/supergrok/provider.mjs +35 -35
- package/src/supergrok/response.mjs +76 -76
- package/src/supergrok/tool.mjs +61 -61
- package/src/text/ansi.mjs +3 -3
- package/src/web/config-command.mjs +43 -43
- package/src/web/fetch.mjs +78 -78
- package/src/web/presets.mjs +16 -16
- package/src/web/search.mjs +83 -83
- package/src/web/tools.mjs +107 -107
package/src/agent/runner.mjs
CHANGED
|
@@ -1,299 +1,299 @@
|
|
|
1
|
-
import { createAgentSession, ModelRegistry, SettingsManager } from "@earendil-works/pi-coding-agent";
|
|
2
|
-
import { createMarchAuthStorage } from "../auth/storage.mjs";
|
|
3
|
-
import { ContextEngine } from "../context/engine.mjs";
|
|
4
|
-
import { createMarchLifecycleAdapter } from "../extensions/lifecycle-adapter.mjs";
|
|
5
|
-
import { syncPiSessionSidecar } from "../session/sidecar-sync.mjs";
|
|
6
|
-
import { LspService } from "../lsp/service.mjs";
|
|
7
|
-
import { formatLspServiceEvent } from "../lsp/status-message.mjs";
|
|
8
|
-
import { estimateProviderPayloadTokens, installModelPayloadDumper, replaceProviderContextMessages } from "./model-payload-dumper.mjs";
|
|
9
|
-
import { resolveInitialModel, resolveRunnerSessionManager } from "./runner/runner-init.mjs";
|
|
10
|
-
import { runRunnerCleanup } from "./runner/runner-cleanup.mjs";
|
|
11
|
-
import { createRunnerRuntimeHost } from "./runtime/runner-runtime-host.mjs";
|
|
12
|
-
import { createRuntimeUiBridge } from "./runtime/ui-event-bridge.mjs";
|
|
13
|
-
import { getRunnerSessionStats, syncEngineSessionState } from "./runner/runner-session-state.mjs";
|
|
14
|
-
import { notifyTurnEndBestEffort, notifyTurnEndDetached, providerContextToPayload } from "./runner/runner-utils.mjs";
|
|
15
|
-
import { dumpCodexTransportDebug, getCodexTransportDebugSnapshot } from "./runner/codex-transport-debug.mjs";
|
|
16
|
-
import { installCodexWebSocketEventDebug } from "./runner/codex-websocket-event-debug.mjs";
|
|
17
|
-
import { installCodexTransportCompression } from "./runner/codex-transport-compression.mjs";
|
|
18
|
-
import { applyCodexLargeContextGuardToPayload, installCodexLargeContextGuard } from "./runner/codex-large-context-guard.mjs";
|
|
19
|
-
import { resolveRunnerSessionOptions } from "./session/session-options.mjs";
|
|
20
|
-
import { createSessionBinding } from "./session/session-binding.mjs";
|
|
21
|
-
import { maybeAutoNameSession } from "./session/session-auto-name.mjs";
|
|
22
|
-
import { MARCH_BASE_TOOL_NAMES } from "./tool-names.mjs";
|
|
23
|
-
import { runRunnerTurn } from "./turn/turn-runner.mjs";
|
|
24
|
-
import { beginLoggedTurn } from "./turn/turn-logging.mjs";
|
|
25
|
-
import { appendFastVariants, createFastModelEntry, fromFastEntryModel, isFastProvider } from "./runner/fast-model.mjs";
|
|
26
|
-
import { registerSuperGrokProvider } from "../supergrok/provider.mjs";
|
|
27
|
-
import { registerCustomProviders } from "../provider/custom-provider.mjs";
|
|
28
|
-
import { injectHostedTools } from "../provider/hosted-tools.mjs";
|
|
29
|
-
export { MARCH_BASE_TOOL_NAMES, installModelPayloadDumper };
|
|
30
|
-
export { createDefaultSessionManager, resolveRunnerSessionManager } from "./runner/runner-init.mjs";
|
|
31
|
-
export { getRunnerSessionStats, syncEngineSessionState } from "./runner/runner-session-state.mjs";
|
|
32
|
-
export async function createRunner({ cwd, modelId = null, provider = null, providers = {}, stateRoot, ui, memoryRoot = null, profilePaths = null, memoryStore = null, memoryTools = [], shellRuntime = null, mcpTools = [], mcpInjections = [], mcpClientManager = null, webTools = [], namespace = "", sessionManager = null, useRuntimeHost = false, projectMarchDir = null, syncPiSidecar = false, extensionPaths = [], lifecycleHooks = [], lifecycleDiagnostics = [], authStorage = null, permissionController = null, modelContextDumper = null, turnNotifier = null, logger = null, onModelPayload = null, createAgentSessionImpl = createAgentSession, createAgentSessionRuntimeImpl, createRuntimeServices, createRuntimeSessionFromServices, maxTurns, trimBatch, serviceTier = null, hostedTools = {} }) {
|
|
33
|
-
installCodexLargeContextGuard();
|
|
34
|
-
installCodexTransportCompression();
|
|
35
|
-
installCodexWebSocketEventDebug();
|
|
36
|
-
if (!useRuntimeHost && extensionPaths.length > 0) {
|
|
37
|
-
throw new Error("--extension requires the default pi runtime host path");
|
|
38
|
-
}
|
|
39
|
-
const authConfig = authStorage
|
|
40
|
-
? { authStorage, hasAuth: true }
|
|
41
|
-
: createMarchAuthStorage({ provider: provider ?? "deepseek", providers, cwd });
|
|
42
|
-
if (!authConfig.hasAuth) throw new Error("No providers configured. Run: march provider --config");
|
|
43
|
-
const resolvedAuth = authConfig.authStorage;
|
|
44
|
-
const modelRegistry = ModelRegistry.create(resolvedAuth);
|
|
45
|
-
registerSuperGrokProvider(modelRegistry);
|
|
46
|
-
registerCustomProviders(modelRegistry, providers);
|
|
47
|
-
const selectedModel = resolveInitialModel({ modelRegistry, provider, modelId });
|
|
48
|
-
if (!selectedModel) throw new Error("No authenticated models available. Run: march provider --config");
|
|
49
|
-
provider = selectedModel.provider;
|
|
50
|
-
modelId = selectedModel.id;
|
|
51
|
-
const settingsManager = SettingsManager.inMemory({
|
|
52
|
-
compaction: { enabled: false },
|
|
53
|
-
retry: { enabled: true, maxRetries: 3, baseDelayMs: 2000 },
|
|
54
|
-
});
|
|
55
|
-
const { ui: runtimeUi, eventBus: runtimeUiEvents, detach: detachRuntimeUi } = createRuntimeUiBridge(ui);
|
|
56
|
-
const lspService = new LspService({ cwd, onEvent: (event) => runtimeUi.status?.(formatLspServiceEvent(event)) });
|
|
57
|
-
const engine = new ContextEngine({ cwd, modelId, provider, namespace, memoryRoot, profilePaths, shellRuntime, lspService, injections: mcpInjections, maxTurns, trimBatch });
|
|
58
|
-
const resolvedSessionManager = resolveRunnerSessionManager(cwd, sessionManager);
|
|
59
|
-
const sessionBinding = createSessionBinding(null);
|
|
60
|
-
let currentModelCallKind = "model", currentTurnId = null, currentPromptForContext = "";
|
|
61
|
-
let currentTurnContextMode = "rebuild";
|
|
62
|
-
let nextTurnContextMode = "rebuild";
|
|
63
|
-
let lastNotificationResult = null, runtimeHost = null, lifecycleAdapter = null;
|
|
64
|
-
let _currentFastEntry = null;
|
|
65
|
-
if (useRuntimeHost) {
|
|
66
|
-
runtimeHost = await createRunnerRuntimeHost({
|
|
67
|
-
cwd, stateRoot, provider, modelId,
|
|
68
|
-
authStorage: resolvedAuth, settingsManager, modelRegistry,
|
|
69
|
-
providers,
|
|
70
|
-
sessionManager: resolvedSessionManager, sessionBinding, engine, ui: runtimeUi,
|
|
71
|
-
projectMarchDir,
|
|
72
|
-
memoryTools, memoryStore, shellRuntime, lspService, mcpTools, webTools,
|
|
73
|
-
permissionController, extensionPaths, hostedTools,
|
|
74
|
-
onRebind: (session) => {
|
|
75
|
-
installModelPayloadDumper(session, modelContextDumper, () => currentModelCallKind, onLoggedModelPayload, injectMarchSystemContext);
|
|
76
|
-
syncEngineSessionState(engine, session);
|
|
77
|
-
},
|
|
78
|
-
createAgentSessionRuntimeImpl,
|
|
79
|
-
createServices: createRuntimeServices,
|
|
80
|
-
createFromServices: createRuntimeSessionFromServices,
|
|
81
|
-
});
|
|
82
|
-
} else {
|
|
83
|
-
const sessionOptions = resolveRunnerSessionOptions({
|
|
84
|
-
cwd, provider, modelId, modelRegistry, engine, ui: runtimeUi,
|
|
85
|
-
memoryTools, shellRuntime, lspService, mcpTools, webTools, permissionController,
|
|
86
|
-
authStorage: resolvedAuth, projectMarchDir,
|
|
87
|
-
getCurrentModel: () => sessionBinding.get()?.model ?? selectedModel,
|
|
88
|
-
});
|
|
89
|
-
const { session } = await createAgentSessionImpl({
|
|
90
|
-
cwd, agentDir: stateRoot, ...sessionOptions,
|
|
91
|
-
authStorage: resolvedAuth, modelRegistry,
|
|
92
|
-
sessionManager: resolvedSessionManager, settingsManager,
|
|
93
|
-
});
|
|
94
|
-
sessionBinding.set(session);
|
|
95
|
-
installModelPayloadDumper(session, modelContextDumper, () => currentModelCallKind, onLoggedModelPayload, injectMarchSystemContext);
|
|
96
|
-
}
|
|
97
|
-
syncEngineSessionState(engine, sessionBinding.get());
|
|
98
|
-
lifecycleAdapter = createMarchLifecycleAdapter({
|
|
99
|
-
cwd, projectMarchDir, extensionPaths, sessionBinding, engine,
|
|
100
|
-
getSessionStats: () => getRunnerSessionStats(sessionBinding.get(), runtimeHost),
|
|
101
|
-
getRuntimeDiagnostics: () => runtimeHost?.getDiagnostics?.() ?? [],
|
|
102
|
-
manifestHooks: lifecycleHooks,
|
|
103
|
-
manifestDiagnostics: lifecycleDiagnostics,
|
|
104
|
-
});
|
|
105
|
-
if (serviceTier === "priority" && selectedModel && isFastProvider(selectedModel.provider)) {
|
|
106
|
-
_currentFastEntry = createFastModelEntry(selectedModel).model;
|
|
107
|
-
}
|
|
108
|
-
return {
|
|
109
|
-
engine,
|
|
110
|
-
get session() { return sessionBinding.get(); },
|
|
111
|
-
shellRuntime,
|
|
112
|
-
runtimeUiEvents,
|
|
113
|
-
async runTurn(prompt, userMessage, { userRecallHints = [], currentProject = "" } = {}) {
|
|
114
|
-
currentPromptForContext = prompt;
|
|
115
|
-
const contextMode = nextTurnContextMode;
|
|
116
|
-
currentTurnContextMode = contextMode;
|
|
117
|
-
nextTurnContextMode = "rebuild";
|
|
118
|
-
const turnStartedAt = Date.now();
|
|
119
|
-
const codexTransportStatsBefore = getCodexTransportDebugSnapshot(sessionBinding.get());
|
|
120
|
-
const turnLog = beginLoggedTurn({ logger, engine, modelId, provider, contextMode, userMessage, userRecallHints, startedAt: turnStartedAt }); currentTurnId = turnLog.turnId;
|
|
121
|
-
try {
|
|
122
|
-
const result = await runRunnerTurn({
|
|
123
|
-
prompt, userMessage, options: { userRecallHints, currentProject },
|
|
124
|
-
sessionBinding, engine, ui: runtimeUi, projectMarchDir, memoryStore,
|
|
125
|
-
setModelCallKind: (kind) => { currentModelCallKind = kind; },
|
|
126
|
-
logger: turnLog.logger,
|
|
127
|
-
setPhase: turnLog.setPhase,
|
|
128
|
-
syncCurrentPiSidecar,
|
|
129
|
-
autoNameSession,
|
|
130
|
-
contextMode,
|
|
131
|
-
});
|
|
132
|
-
notifyTurnEndDetached(turnNotifier, {
|
|
133
|
-
status: "success",
|
|
134
|
-
sessionName: engine.sessionName,
|
|
135
|
-
draft: result?.draft ?? "",
|
|
136
|
-
durationMs: Date.now() - turnStartedAt,
|
|
137
|
-
}, (notificationResult) => { lastNotificationResult = notificationResult; });
|
|
138
|
-
turnLog.endSuccess(result);
|
|
139
|
-
return result;
|
|
140
|
-
} catch (err) {
|
|
141
|
-
notifyTurnEndDetached(turnNotifier, {
|
|
142
|
-
status: "error",
|
|
143
|
-
sessionName: engine.sessionName,
|
|
144
|
-
errorMessage: err?.message ?? String(err),
|
|
145
|
-
durationMs: Date.now() - turnStartedAt,
|
|
146
|
-
}, (notificationResult) => { lastNotificationResult = notificationResult; });
|
|
147
|
-
turnLog.endError(err);
|
|
148
|
-
throw err;
|
|
149
|
-
} finally {
|
|
150
|
-
dumpCodexTransportDebug({ before: codexTransportStatsBefore, session: sessionBinding.get(), ui: runtimeUi, logger });
|
|
151
|
-
currentTurnId = null;
|
|
152
|
-
currentTurnContextMode = "rebuild";
|
|
153
|
-
}
|
|
154
|
-
},
|
|
155
|
-
abort() {
|
|
156
|
-
const activeSession = sessionBinding.get();
|
|
157
|
-
activeSession.abortRetry?.();
|
|
158
|
-
const result = activeSession.abort();
|
|
159
|
-
nextTurnContextMode = "continueExistingPiTranscript";
|
|
160
|
-
return result;
|
|
161
|
-
},
|
|
162
|
-
async cycleModel() {
|
|
163
|
-
const result = await sessionBinding.get().cycleModel();
|
|
164
|
-
_currentFastEntry = null;
|
|
165
|
-
if (result?.model) {
|
|
166
|
-
engine.setRuntimeState({
|
|
167
|
-
modelId: result.model.id, provider: result.model.provider,
|
|
168
|
-
thinkingLevel: result.thinkingLevel,
|
|
169
|
-
});
|
|
170
|
-
}
|
|
171
|
-
return result;
|
|
172
|
-
},
|
|
173
|
-
getCurrentModel() { return _currentFastEntry ?? sessionBinding.get().model; },
|
|
174
|
-
async setModel(model) {
|
|
175
|
-
const activeSession = sessionBinding.get();
|
|
176
|
-
const { baseId, isFast } = fromFastEntryModel(model);
|
|
177
|
-
const baseEntry = sessionBinding.get().scopedModels.find((e) => e.model.id === baseId);
|
|
178
|
-
if (!baseEntry) throw new Error(`Model not found: ${baseId}`);
|
|
179
|
-
await activeSession.setModel(baseEntry.model);
|
|
180
|
-
_currentFastEntry = isFast ? model : null;
|
|
181
|
-
engine.setRuntimeState({
|
|
182
|
-
modelId: activeSession.model?.id ?? baseId,
|
|
183
|
-
provider: activeSession.model?.provider ?? model.provider,
|
|
184
|
-
thinkingLevel: activeSession.thinkingLevel,
|
|
185
|
-
});
|
|
186
|
-
return model;
|
|
187
|
-
},
|
|
188
|
-
getScopedModels() { return appendFastVariants(sessionBinding.get().scopedModels); },
|
|
189
|
-
getConfiguredProviders() {
|
|
190
|
-
const configured = Object.values(providers ?? {}).map((profile) => profile?.type).filter(Boolean);
|
|
191
|
-
const available = (modelRegistry.getAvailable?.() ?? []).map((model) => model.provider);
|
|
192
|
-
return [...new Set([...configured, ...available])];
|
|
193
|
-
},
|
|
194
|
-
getSessionStats() { return getRunnerSessionStats(sessionBinding.get(), runtimeHost); },
|
|
195
|
-
getLastNotificationResult() { return lastNotificationResult; },
|
|
196
|
-
async notifyTest({ title = "March", message = "If you see this, March runtime notifications work." } = {}) {
|
|
197
|
-
lastNotificationResult = await notifyTurnEndBestEffort(turnNotifier, {
|
|
198
|
-
status: "success",
|
|
199
|
-
sessionName: engine.sessionName,
|
|
200
|
-
title,
|
|
201
|
-
message,
|
|
202
|
-
durationMs: 0,
|
|
203
|
-
});
|
|
204
|
-
return lastNotificationResult;
|
|
205
|
-
},
|
|
206
|
-
estimateContextTokens(userMessage = "") {
|
|
207
|
-
return estimateProviderPayloadTokens(providerContextToPayload(engine.buildProviderContext(userMessage)));
|
|
208
|
-
},
|
|
209
|
-
setSessionName(name) {
|
|
210
|
-
const activeSession = sessionBinding.get();
|
|
211
|
-
activeSession.setSessionName?.(name);
|
|
212
|
-
engine.setSessionName(name);
|
|
213
|
-
syncCurrentPiSidecar();
|
|
214
|
-
return engine.sessionName;
|
|
215
|
-
},
|
|
216
|
-
canSwitchPiSession() { return Boolean(runtimeHost); },
|
|
217
|
-
async startNewSession() {
|
|
218
|
-
if (!runtimeHost) throw new Error("pi runtime host is not enabled");
|
|
219
|
-
nextTurnContextMode = "rebuild";
|
|
220
|
-
syncCurrentPiSidecar();
|
|
221
|
-
const result = await runtimeHost.newSession();
|
|
222
|
-
if (result?.cancelled) return { cancelled: true };
|
|
223
|
-
engine.restoreSession({}, [], { replace: true });
|
|
224
|
-
shellRuntime?.killAll?.();
|
|
225
|
-
const stats = getRunnerSessionStats(sessionBinding.get(), runtimeHost);
|
|
226
|
-
return { sessionId: stats.sessionId, sessionFile: stats.sessionFile };
|
|
227
|
-
},
|
|
228
|
-
getExtensionDiagnostics() { return runtimeHost?.getDiagnostics?.() ?? []; },
|
|
229
|
-
getExtensionLifecycleState() { return lifecycleAdapter.getState(); },
|
|
230
|
-
getLspStatus() { return lspService.snapshot(); },
|
|
231
|
-
async switchPiSession(sessionPath, restoreState = null) {
|
|
232
|
-
if (!runtimeHost) throw new Error("pi runtime host is not enabled");
|
|
233
|
-
nextTurnContextMode = "rebuild";
|
|
234
|
-
const result = await runtimeHost.switchSession(sessionPath);
|
|
235
|
-
if (!result?.cancelled && restoreState) engine.restoreSession(restoreState, null, { replace: true });
|
|
236
|
-
return result;
|
|
237
|
-
},
|
|
238
|
-
cycleThinkingLevel() {
|
|
239
|
-
const level = sessionBinding.get().cycleThinkingLevel();
|
|
240
|
-
engine.setRuntimeState({ thinkingLevel: level });
|
|
241
|
-
return level;
|
|
242
|
-
},
|
|
243
|
-
getThinkingLevel() { return sessionBinding.get().thinkingLevel; },
|
|
244
|
-
setThinkingLevel(level) {
|
|
245
|
-
const activeSession = sessionBinding.get();
|
|
246
|
-
activeSession.setThinkingLevel(level);
|
|
247
|
-
engine.setRuntimeState({ thinkingLevel: activeSession.thinkingLevel });
|
|
248
|
-
return activeSession.thinkingLevel;
|
|
249
|
-
},
|
|
250
|
-
getAvailableThinkingLevels() { return sessionBinding.get().getAvailableThinkingLevels(); },
|
|
251
|
-
async dispose() {
|
|
252
|
-
await runRunnerCleanup([
|
|
253
|
-
() => runtimeHost?.dispose() ?? sessionBinding.get().dispose(),
|
|
254
|
-
() => shellRuntime?.dispose?.() ?? shellRuntime?.killAll?.(),
|
|
255
|
-
() => lspService.dispose(),
|
|
256
|
-
() => mcpClientManager?.disconnectAll?.(),
|
|
257
|
-
() => detachRuntimeUi(),
|
|
258
|
-
]);
|
|
259
|
-
},
|
|
260
|
-
};
|
|
261
|
-
function syncCurrentPiSidecar() {
|
|
262
|
-
return syncPiSessionSidecar({
|
|
263
|
-
enabled: syncPiSidecar, projectMarchDir, engine,
|
|
264
|
-
sessionStats: getRunnerSessionStats(sessionBinding.get(), runtimeHost),
|
|
265
|
-
});
|
|
266
|
-
}
|
|
267
|
-
function autoNameSession() {
|
|
268
|
-
return maybeAutoNameSession({
|
|
269
|
-
engine,
|
|
270
|
-
session: sessionBinding.get(),
|
|
271
|
-
setSessionName: (name) => {
|
|
272
|
-
const activeSession = sessionBinding.get();
|
|
273
|
-
activeSession.setSessionName?.(name);
|
|
274
|
-
engine.setSessionName(name);
|
|
275
|
-
return engine.sessionName;
|
|
276
|
-
},
|
|
277
|
-
});
|
|
278
|
-
}
|
|
279
|
-
function onLoggedModelPayload(event) {
|
|
280
|
-
logger?.event("model.payload", {
|
|
281
|
-
kind: event.kind,
|
|
282
|
-
provider: event.model?.provider,
|
|
283
|
-
model: event.model?.id,
|
|
284
|
-
estimatedTokens: event.estimatedTokens,
|
|
285
|
-
turnId: currentTurnId,
|
|
286
|
-
});
|
|
287
|
-
onModelPayload?.(event);
|
|
288
|
-
}
|
|
289
|
-
function injectMarchSystemContext(payload, { kind, model } = {}) {
|
|
290
|
-
if (kind !== "user") return payload;
|
|
291
|
-
let nextPayload = currentTurnContextMode === "continueExistingPiTranscript"
|
|
292
|
-
? payload
|
|
293
|
-
: replaceProviderContextMessages(payload, engine.buildProviderContext(currentPromptForContext));
|
|
294
|
-
nextPayload = injectHostedTools(nextPayload, model, hostedTools);
|
|
295
|
-
nextPayload = applyCodexLargeContextGuardToPayload(nextPayload, { model, session: sessionBinding.get() });
|
|
296
|
-
if (_currentFastEntry) nextPayload = { ...nextPayload, service_tier: "priority" };
|
|
297
|
-
return nextPayload;
|
|
298
|
-
}
|
|
299
|
-
}
|
|
1
|
+
import { createAgentSession, ModelRegistry, SettingsManager } from "@earendil-works/pi-coding-agent";
|
|
2
|
+
import { createMarchAuthStorage } from "../auth/storage.mjs";
|
|
3
|
+
import { ContextEngine } from "../context/engine.mjs";
|
|
4
|
+
import { createMarchLifecycleAdapter } from "../extensions/lifecycle-adapter.mjs";
|
|
5
|
+
import { syncPiSessionSidecar } from "../session/sidecar-sync.mjs";
|
|
6
|
+
import { LspService } from "../lsp/service.mjs";
|
|
7
|
+
import { formatLspServiceEvent } from "../lsp/status-message.mjs";
|
|
8
|
+
import { estimateProviderPayloadTokens, installModelPayloadDumper, replaceProviderContextMessages } from "./model-payload-dumper.mjs";
|
|
9
|
+
import { resolveInitialModel, resolveRunnerSessionManager } from "./runner/runner-init.mjs";
|
|
10
|
+
import { runRunnerCleanup } from "./runner/runner-cleanup.mjs";
|
|
11
|
+
import { createRunnerRuntimeHost } from "./runtime/runner-runtime-host.mjs";
|
|
12
|
+
import { createRuntimeUiBridge } from "./runtime/ui-event-bridge.mjs";
|
|
13
|
+
import { getRunnerSessionStats, syncEngineSessionState } from "./runner/runner-session-state.mjs";
|
|
14
|
+
import { notifyTurnEndBestEffort, notifyTurnEndDetached, providerContextToPayload } from "./runner/runner-utils.mjs";
|
|
15
|
+
import { dumpCodexTransportDebug, getCodexTransportDebugSnapshot } from "./runner/codex-transport-debug.mjs";
|
|
16
|
+
import { installCodexWebSocketEventDebug } from "./runner/codex-websocket-event-debug.mjs";
|
|
17
|
+
import { installCodexTransportCompression } from "./runner/codex-transport-compression.mjs";
|
|
18
|
+
import { applyCodexLargeContextGuardToPayload, installCodexLargeContextGuard } from "./runner/codex-large-context-guard.mjs";
|
|
19
|
+
import { resolveRunnerSessionOptions } from "./session/session-options.mjs";
|
|
20
|
+
import { createSessionBinding } from "./session/session-binding.mjs";
|
|
21
|
+
import { maybeAutoNameSession } from "./session/session-auto-name.mjs";
|
|
22
|
+
import { MARCH_BASE_TOOL_NAMES } from "./tool-names.mjs";
|
|
23
|
+
import { runRunnerTurn } from "./turn/turn-runner.mjs";
|
|
24
|
+
import { beginLoggedTurn } from "./turn/turn-logging.mjs";
|
|
25
|
+
import { appendFastVariants, createFastModelEntry, fromFastEntryModel, isFastProvider } from "./runner/fast-model.mjs";
|
|
26
|
+
import { registerSuperGrokProvider } from "../supergrok/provider.mjs";
|
|
27
|
+
import { registerCustomProviders } from "../provider/custom-provider.mjs";
|
|
28
|
+
import { injectHostedTools } from "../provider/hosted-tools.mjs";
|
|
29
|
+
export { MARCH_BASE_TOOL_NAMES, installModelPayloadDumper };
|
|
30
|
+
export { createDefaultSessionManager, resolveRunnerSessionManager } from "./runner/runner-init.mjs";
|
|
31
|
+
export { getRunnerSessionStats, syncEngineSessionState } from "./runner/runner-session-state.mjs";
|
|
32
|
+
export async function createRunner({ cwd, modelId = null, provider = null, providers = {}, stateRoot, ui, memoryRoot = null, profilePaths = null, memoryStore = null, memoryTools = [], remoteMemorySources = [], shellRuntime = null, mcpTools = [], mcpInjections = [], mcpClientManager = null, webTools = [], namespace = "", sessionManager = null, useRuntimeHost = false, projectMarchDir = null, syncPiSidecar = false, extensionPaths = [], lifecycleHooks = [], lifecycleDiagnostics = [], authStorage = null, permissionController = null, modelContextDumper = null, turnNotifier = null, logger = null, onModelPayload = null, createAgentSessionImpl = createAgentSession, createAgentSessionRuntimeImpl, createRuntimeServices, createRuntimeSessionFromServices, maxTurns, trimBatch, serviceTier = null, hostedTools = {} }) {
|
|
33
|
+
installCodexLargeContextGuard();
|
|
34
|
+
installCodexTransportCompression();
|
|
35
|
+
installCodexWebSocketEventDebug();
|
|
36
|
+
if (!useRuntimeHost && extensionPaths.length > 0) {
|
|
37
|
+
throw new Error("--extension requires the default pi runtime host path");
|
|
38
|
+
}
|
|
39
|
+
const authConfig = authStorage
|
|
40
|
+
? { authStorage, hasAuth: true }
|
|
41
|
+
: createMarchAuthStorage({ provider: provider ?? "deepseek", providers, cwd });
|
|
42
|
+
if (!authConfig.hasAuth) throw new Error("No providers configured. Run: march provider --config");
|
|
43
|
+
const resolvedAuth = authConfig.authStorage;
|
|
44
|
+
const modelRegistry = ModelRegistry.create(resolvedAuth);
|
|
45
|
+
registerSuperGrokProvider(modelRegistry);
|
|
46
|
+
registerCustomProviders(modelRegistry, providers);
|
|
47
|
+
const selectedModel = resolveInitialModel({ modelRegistry, provider, modelId });
|
|
48
|
+
if (!selectedModel) throw new Error("No authenticated models available. Run: march provider --config");
|
|
49
|
+
provider = selectedModel.provider;
|
|
50
|
+
modelId = selectedModel.id;
|
|
51
|
+
const settingsManager = SettingsManager.inMemory({
|
|
52
|
+
compaction: { enabled: false },
|
|
53
|
+
retry: { enabled: true, maxRetries: 3, baseDelayMs: 2000 },
|
|
54
|
+
});
|
|
55
|
+
const { ui: runtimeUi, eventBus: runtimeUiEvents, detach: detachRuntimeUi } = createRuntimeUiBridge(ui);
|
|
56
|
+
const lspService = new LspService({ cwd, onEvent: (event) => runtimeUi.status?.(formatLspServiceEvent(event)) });
|
|
57
|
+
const engine = new ContextEngine({ cwd, modelId, provider, namespace, memoryRoot, profilePaths, remoteMemorySources, shellRuntime, lspService, injections: mcpInjections, maxTurns, trimBatch });
|
|
58
|
+
const resolvedSessionManager = resolveRunnerSessionManager(cwd, sessionManager);
|
|
59
|
+
const sessionBinding = createSessionBinding(null);
|
|
60
|
+
let currentModelCallKind = "model", currentTurnId = null, currentPromptForContext = "";
|
|
61
|
+
let currentTurnContextMode = "rebuild";
|
|
62
|
+
let nextTurnContextMode = "rebuild";
|
|
63
|
+
let lastNotificationResult = null, runtimeHost = null, lifecycleAdapter = null;
|
|
64
|
+
let _currentFastEntry = null;
|
|
65
|
+
if (useRuntimeHost) {
|
|
66
|
+
runtimeHost = await createRunnerRuntimeHost({
|
|
67
|
+
cwd, stateRoot, provider, modelId,
|
|
68
|
+
authStorage: resolvedAuth, settingsManager, modelRegistry,
|
|
69
|
+
providers,
|
|
70
|
+
sessionManager: resolvedSessionManager, sessionBinding, engine, ui: runtimeUi,
|
|
71
|
+
projectMarchDir,
|
|
72
|
+
memoryTools, memoryStore, shellRuntime, lspService, mcpTools, webTools,
|
|
73
|
+
permissionController, extensionPaths, hostedTools,
|
|
74
|
+
onRebind: (session) => {
|
|
75
|
+
installModelPayloadDumper(session, modelContextDumper, () => currentModelCallKind, onLoggedModelPayload, injectMarchSystemContext);
|
|
76
|
+
syncEngineSessionState(engine, session);
|
|
77
|
+
},
|
|
78
|
+
createAgentSessionRuntimeImpl,
|
|
79
|
+
createServices: createRuntimeServices,
|
|
80
|
+
createFromServices: createRuntimeSessionFromServices,
|
|
81
|
+
});
|
|
82
|
+
} else {
|
|
83
|
+
const sessionOptions = resolveRunnerSessionOptions({
|
|
84
|
+
cwd, provider, modelId, modelRegistry, engine, ui: runtimeUi,
|
|
85
|
+
memoryTools, shellRuntime, lspService, mcpTools, webTools, permissionController,
|
|
86
|
+
authStorage: resolvedAuth, projectMarchDir,
|
|
87
|
+
getCurrentModel: () => sessionBinding.get()?.model ?? selectedModel,
|
|
88
|
+
});
|
|
89
|
+
const { session } = await createAgentSessionImpl({
|
|
90
|
+
cwd, agentDir: stateRoot, ...sessionOptions,
|
|
91
|
+
authStorage: resolvedAuth, modelRegistry,
|
|
92
|
+
sessionManager: resolvedSessionManager, settingsManager,
|
|
93
|
+
});
|
|
94
|
+
sessionBinding.set(session);
|
|
95
|
+
installModelPayloadDumper(session, modelContextDumper, () => currentModelCallKind, onLoggedModelPayload, injectMarchSystemContext);
|
|
96
|
+
}
|
|
97
|
+
syncEngineSessionState(engine, sessionBinding.get());
|
|
98
|
+
lifecycleAdapter = createMarchLifecycleAdapter({
|
|
99
|
+
cwd, projectMarchDir, extensionPaths, sessionBinding, engine,
|
|
100
|
+
getSessionStats: () => getRunnerSessionStats(sessionBinding.get(), runtimeHost),
|
|
101
|
+
getRuntimeDiagnostics: () => runtimeHost?.getDiagnostics?.() ?? [],
|
|
102
|
+
manifestHooks: lifecycleHooks,
|
|
103
|
+
manifestDiagnostics: lifecycleDiagnostics,
|
|
104
|
+
});
|
|
105
|
+
if (serviceTier === "priority" && selectedModel && isFastProvider(selectedModel.provider)) {
|
|
106
|
+
_currentFastEntry = createFastModelEntry(selectedModel).model;
|
|
107
|
+
}
|
|
108
|
+
return {
|
|
109
|
+
engine,
|
|
110
|
+
get session() { return sessionBinding.get(); },
|
|
111
|
+
shellRuntime,
|
|
112
|
+
runtimeUiEvents,
|
|
113
|
+
async runTurn(prompt, userMessage, { userRecallHints = [], currentProject = "" } = {}) {
|
|
114
|
+
currentPromptForContext = prompt;
|
|
115
|
+
const contextMode = nextTurnContextMode;
|
|
116
|
+
currentTurnContextMode = contextMode;
|
|
117
|
+
nextTurnContextMode = "rebuild";
|
|
118
|
+
const turnStartedAt = Date.now();
|
|
119
|
+
const codexTransportStatsBefore = getCodexTransportDebugSnapshot(sessionBinding.get());
|
|
120
|
+
const turnLog = beginLoggedTurn({ logger, engine, modelId, provider, contextMode, userMessage, userRecallHints, startedAt: turnStartedAt }); currentTurnId = turnLog.turnId;
|
|
121
|
+
try {
|
|
122
|
+
const result = await runRunnerTurn({
|
|
123
|
+
prompt, userMessage, options: { userRecallHints, currentProject },
|
|
124
|
+
sessionBinding, engine, ui: runtimeUi, projectMarchDir, memoryStore,
|
|
125
|
+
setModelCallKind: (kind) => { currentModelCallKind = kind; },
|
|
126
|
+
logger: turnLog.logger,
|
|
127
|
+
setPhase: turnLog.setPhase,
|
|
128
|
+
syncCurrentPiSidecar,
|
|
129
|
+
autoNameSession,
|
|
130
|
+
contextMode,
|
|
131
|
+
});
|
|
132
|
+
notifyTurnEndDetached(turnNotifier, {
|
|
133
|
+
status: "success",
|
|
134
|
+
sessionName: engine.sessionName,
|
|
135
|
+
draft: result?.draft ?? "",
|
|
136
|
+
durationMs: Date.now() - turnStartedAt,
|
|
137
|
+
}, (notificationResult) => { lastNotificationResult = notificationResult; });
|
|
138
|
+
turnLog.endSuccess(result);
|
|
139
|
+
return result;
|
|
140
|
+
} catch (err) {
|
|
141
|
+
notifyTurnEndDetached(turnNotifier, {
|
|
142
|
+
status: "error",
|
|
143
|
+
sessionName: engine.sessionName,
|
|
144
|
+
errorMessage: err?.message ?? String(err),
|
|
145
|
+
durationMs: Date.now() - turnStartedAt,
|
|
146
|
+
}, (notificationResult) => { lastNotificationResult = notificationResult; });
|
|
147
|
+
turnLog.endError(err);
|
|
148
|
+
throw err;
|
|
149
|
+
} finally {
|
|
150
|
+
dumpCodexTransportDebug({ before: codexTransportStatsBefore, session: sessionBinding.get(), ui: runtimeUi, logger });
|
|
151
|
+
currentTurnId = null;
|
|
152
|
+
currentTurnContextMode = "rebuild";
|
|
153
|
+
}
|
|
154
|
+
},
|
|
155
|
+
abort() {
|
|
156
|
+
const activeSession = sessionBinding.get();
|
|
157
|
+
activeSession.abortRetry?.();
|
|
158
|
+
const result = activeSession.abort();
|
|
159
|
+
nextTurnContextMode = "continueExistingPiTranscript";
|
|
160
|
+
return result;
|
|
161
|
+
},
|
|
162
|
+
async cycleModel() {
|
|
163
|
+
const result = await sessionBinding.get().cycleModel();
|
|
164
|
+
_currentFastEntry = null;
|
|
165
|
+
if (result?.model) {
|
|
166
|
+
engine.setRuntimeState({
|
|
167
|
+
modelId: result.model.id, provider: result.model.provider,
|
|
168
|
+
thinkingLevel: result.thinkingLevel,
|
|
169
|
+
});
|
|
170
|
+
}
|
|
171
|
+
return result;
|
|
172
|
+
},
|
|
173
|
+
getCurrentModel() { return _currentFastEntry ?? sessionBinding.get().model; },
|
|
174
|
+
async setModel(model) {
|
|
175
|
+
const activeSession = sessionBinding.get();
|
|
176
|
+
const { baseId, isFast } = fromFastEntryModel(model);
|
|
177
|
+
const baseEntry = sessionBinding.get().scopedModels.find((e) => e.model.id === baseId);
|
|
178
|
+
if (!baseEntry) throw new Error(`Model not found: ${baseId}`);
|
|
179
|
+
await activeSession.setModel(baseEntry.model);
|
|
180
|
+
_currentFastEntry = isFast ? model : null;
|
|
181
|
+
engine.setRuntimeState({
|
|
182
|
+
modelId: activeSession.model?.id ?? baseId,
|
|
183
|
+
provider: activeSession.model?.provider ?? model.provider,
|
|
184
|
+
thinkingLevel: activeSession.thinkingLevel,
|
|
185
|
+
});
|
|
186
|
+
return model;
|
|
187
|
+
},
|
|
188
|
+
getScopedModels() { return appendFastVariants(sessionBinding.get().scopedModels); },
|
|
189
|
+
getConfiguredProviders() {
|
|
190
|
+
const configured = Object.values(providers ?? {}).map((profile) => profile?.type).filter(Boolean);
|
|
191
|
+
const available = (modelRegistry.getAvailable?.() ?? []).map((model) => model.provider);
|
|
192
|
+
return [...new Set([...configured, ...available])];
|
|
193
|
+
},
|
|
194
|
+
getSessionStats() { return getRunnerSessionStats(sessionBinding.get(), runtimeHost); },
|
|
195
|
+
getLastNotificationResult() { return lastNotificationResult; },
|
|
196
|
+
async notifyTest({ title = "March", message = "If you see this, March runtime notifications work." } = {}) {
|
|
197
|
+
lastNotificationResult = await notifyTurnEndBestEffort(turnNotifier, {
|
|
198
|
+
status: "success",
|
|
199
|
+
sessionName: engine.sessionName,
|
|
200
|
+
title,
|
|
201
|
+
message,
|
|
202
|
+
durationMs: 0,
|
|
203
|
+
});
|
|
204
|
+
return lastNotificationResult;
|
|
205
|
+
},
|
|
206
|
+
estimateContextTokens(userMessage = "") {
|
|
207
|
+
return estimateProviderPayloadTokens(providerContextToPayload(engine.buildProviderContext(userMessage)));
|
|
208
|
+
},
|
|
209
|
+
setSessionName(name) {
|
|
210
|
+
const activeSession = sessionBinding.get();
|
|
211
|
+
activeSession.setSessionName?.(name);
|
|
212
|
+
engine.setSessionName(name);
|
|
213
|
+
syncCurrentPiSidecar();
|
|
214
|
+
return engine.sessionName;
|
|
215
|
+
},
|
|
216
|
+
canSwitchPiSession() { return Boolean(runtimeHost); },
|
|
217
|
+
async startNewSession() {
|
|
218
|
+
if (!runtimeHost) throw new Error("pi runtime host is not enabled");
|
|
219
|
+
nextTurnContextMode = "rebuild";
|
|
220
|
+
syncCurrentPiSidecar();
|
|
221
|
+
const result = await runtimeHost.newSession();
|
|
222
|
+
if (result?.cancelled) return { cancelled: true };
|
|
223
|
+
engine.restoreSession({}, [], { replace: true });
|
|
224
|
+
shellRuntime?.killAll?.();
|
|
225
|
+
const stats = getRunnerSessionStats(sessionBinding.get(), runtimeHost);
|
|
226
|
+
return { sessionId: stats.sessionId, sessionFile: stats.sessionFile };
|
|
227
|
+
},
|
|
228
|
+
getExtensionDiagnostics() { return runtimeHost?.getDiagnostics?.() ?? []; },
|
|
229
|
+
getExtensionLifecycleState() { return lifecycleAdapter.getState(); },
|
|
230
|
+
getLspStatus() { return lspService.snapshot(); },
|
|
231
|
+
async switchPiSession(sessionPath, restoreState = null) {
|
|
232
|
+
if (!runtimeHost) throw new Error("pi runtime host is not enabled");
|
|
233
|
+
nextTurnContextMode = "rebuild";
|
|
234
|
+
const result = await runtimeHost.switchSession(sessionPath);
|
|
235
|
+
if (!result?.cancelled && restoreState) engine.restoreSession(restoreState, null, { replace: true });
|
|
236
|
+
return result;
|
|
237
|
+
},
|
|
238
|
+
cycleThinkingLevel() {
|
|
239
|
+
const level = sessionBinding.get().cycleThinkingLevel();
|
|
240
|
+
engine.setRuntimeState({ thinkingLevel: level });
|
|
241
|
+
return level;
|
|
242
|
+
},
|
|
243
|
+
getThinkingLevel() { return sessionBinding.get().thinkingLevel; },
|
|
244
|
+
setThinkingLevel(level) {
|
|
245
|
+
const activeSession = sessionBinding.get();
|
|
246
|
+
activeSession.setThinkingLevel(level);
|
|
247
|
+
engine.setRuntimeState({ thinkingLevel: activeSession.thinkingLevel });
|
|
248
|
+
return activeSession.thinkingLevel;
|
|
249
|
+
},
|
|
250
|
+
getAvailableThinkingLevels() { return sessionBinding.get().getAvailableThinkingLevels(); },
|
|
251
|
+
async dispose() {
|
|
252
|
+
await runRunnerCleanup([
|
|
253
|
+
() => runtimeHost?.dispose() ?? sessionBinding.get().dispose(),
|
|
254
|
+
() => shellRuntime?.dispose?.() ?? shellRuntime?.killAll?.(),
|
|
255
|
+
() => lspService.dispose(),
|
|
256
|
+
() => mcpClientManager?.disconnectAll?.(),
|
|
257
|
+
() => detachRuntimeUi(),
|
|
258
|
+
]);
|
|
259
|
+
},
|
|
260
|
+
};
|
|
261
|
+
function syncCurrentPiSidecar() {
|
|
262
|
+
return syncPiSessionSidecar({
|
|
263
|
+
enabled: syncPiSidecar, projectMarchDir, engine,
|
|
264
|
+
sessionStats: getRunnerSessionStats(sessionBinding.get(), runtimeHost),
|
|
265
|
+
});
|
|
266
|
+
}
|
|
267
|
+
function autoNameSession() {
|
|
268
|
+
return maybeAutoNameSession({
|
|
269
|
+
engine,
|
|
270
|
+
session: sessionBinding.get(),
|
|
271
|
+
setSessionName: (name) => {
|
|
272
|
+
const activeSession = sessionBinding.get();
|
|
273
|
+
activeSession.setSessionName?.(name);
|
|
274
|
+
engine.setSessionName(name);
|
|
275
|
+
return engine.sessionName;
|
|
276
|
+
},
|
|
277
|
+
});
|
|
278
|
+
}
|
|
279
|
+
function onLoggedModelPayload(event) {
|
|
280
|
+
logger?.event("model.payload", {
|
|
281
|
+
kind: event.kind,
|
|
282
|
+
provider: event.model?.provider,
|
|
283
|
+
model: event.model?.id,
|
|
284
|
+
estimatedTokens: event.estimatedTokens,
|
|
285
|
+
turnId: currentTurnId,
|
|
286
|
+
});
|
|
287
|
+
onModelPayload?.(event);
|
|
288
|
+
}
|
|
289
|
+
function injectMarchSystemContext(payload, { kind, model } = {}) {
|
|
290
|
+
if (kind !== "user") return payload;
|
|
291
|
+
let nextPayload = currentTurnContextMode === "continueExistingPiTranscript"
|
|
292
|
+
? payload
|
|
293
|
+
: replaceProviderContextMessages(payload, engine.buildProviderContext(currentPromptForContext));
|
|
294
|
+
nextPayload = injectHostedTools(nextPayload, model, hostedTools);
|
|
295
|
+
nextPayload = applyCodexLargeContextGuardToPayload(nextPayload, { model, session: sessionBinding.get() });
|
|
296
|
+
if (_currentFastEntry) nextPayload = { ...nextPayload, service_tier: "priority" };
|
|
297
|
+
return nextPayload;
|
|
298
|
+
}
|
|
299
|
+
}
|