march-cli 0.1.18 → 0.1.20
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/package.json
CHANGED
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import { getOpenAICodexWebSocketDebugStats } from "@earendil-works/pi-ai/openai-codex-responses";
|
|
2
|
+
import { getCodexWebSocketLastEvent } from "./codex-websocket-event-debug.mjs";
|
|
2
3
|
|
|
3
4
|
export function getCodexTransportDebugSnapshot(session) {
|
|
4
5
|
if (!isCodexTransportDebugEnabled()) return null;
|
|
@@ -12,7 +13,7 @@ export function dumpCodexTransportDebug({ before, session, ui, logger }) {
|
|
|
12
13
|
const after = sessionId ? (getOpenAICodexWebSocketDebugStats(sessionId) ?? null) : null;
|
|
13
14
|
const fields = formatCodexTransportDebugFields(sessionId, before, after);
|
|
14
15
|
logger?.event("codex.transport", fields);
|
|
15
|
-
writeCodexTransportDebug(ui, formatCodexTransportDebugLines(fields));
|
|
16
|
+
writeCodexTransportDebug(ui, formatCodexTransportDebugLines(fields, logger?.path));
|
|
16
17
|
}
|
|
17
18
|
|
|
18
19
|
function isCodexTransportDebugEnabled() {
|
|
@@ -20,6 +21,11 @@ function isCodexTransportDebugEnabled() {
|
|
|
20
21
|
return value === "1" || value === "true" || value === "yes";
|
|
21
22
|
}
|
|
22
23
|
|
|
24
|
+
function isCodexTransportDebugEventsVisible() {
|
|
25
|
+
const value = process.env.MARCH_CODEX_TRANSPORT_DEBUG_EVENTS;
|
|
26
|
+
return value === "1" || value === "true" || value === "yes";
|
|
27
|
+
}
|
|
28
|
+
|
|
23
29
|
function formatCodexTransportDebugFields(sessionId, before, after) {
|
|
24
30
|
const delta = (key) => (after?.[key] ?? 0) - (before?.[key] ?? 0);
|
|
25
31
|
return {
|
|
@@ -38,21 +44,60 @@ function formatCodexTransportDebugFields(sessionId, before, after) {
|
|
|
38
44
|
lastInputItems: after?.lastInputItems ?? 0,
|
|
39
45
|
lastDeltaInputItems: after?.lastDeltaInputItems ?? 0,
|
|
40
46
|
lastWebSocketError: after?.lastWebSocketError ?? null,
|
|
47
|
+
lastWebSocketEvent: getCodexWebSocketLastEvent(sessionId),
|
|
41
48
|
hasStats: Boolean(after),
|
|
42
49
|
};
|
|
43
50
|
}
|
|
44
51
|
|
|
45
|
-
function formatCodexTransportDebugLines(fields) {
|
|
52
|
+
function formatCodexTransportDebugLines(fields, logPath) {
|
|
46
53
|
if (!fields.hasStats) return [`[codex-transport] sessionId=${fields.sessionId} no Codex WebSocket stats`];
|
|
47
|
-
|
|
54
|
+
const lines = [
|
|
48
55
|
`[codex-transport] sessionId=${fields.sessionId}`,
|
|
49
56
|
` requests=${fields.requests} totalRequests=${fields.totalRequests}`,
|
|
50
57
|
` wsConnections created=${fields.connectionsCreated} reused=${fields.connectionsReused}`,
|
|
51
58
|
` modes full=${fields.fullContextRequests} delta=${fields.deltaRequests} cached=${fields.cachedContextRequests} storeTrue=${fields.storeTrueRequests}`,
|
|
52
59
|
` fallback websocketFailures=${fields.websocketFailures} sseFallbacks=${fields.sseFallbacks} active=${fields.websocketFallbackActive}`,
|
|
60
|
+
` error lastWebSocketError=${formatDebugValue(fields.lastWebSocketError)}`,
|
|
53
61
|
` lastInputItems=${fields.lastInputItems} lastDeltaInputItems=${fields.lastDeltaInputItems}`,
|
|
54
|
-
...(fields.lastWebSocketError ? [` lastWebSocketError=${fields.lastWebSocketError}`] : []),
|
|
55
62
|
];
|
|
63
|
+
if (isCodexTransportDebugEventsVisible()) {
|
|
64
|
+
lines.push(...formatWebSocketEventLines(fields.lastWebSocketEvent));
|
|
65
|
+
} else if (fields.lastWebSocketEvent && logPath) {
|
|
66
|
+
lines.push(` rawWebSocketEventLog=${formatDebugValue(logPath)}`);
|
|
67
|
+
}
|
|
68
|
+
return lines;
|
|
69
|
+
}
|
|
70
|
+
|
|
71
|
+
function formatWebSocketEventLines(events) {
|
|
72
|
+
if (!events) return [" wsEvent none"];
|
|
73
|
+
return [
|
|
74
|
+
...formatSingleWebSocketEvent("wsEvent", events.lastEvent),
|
|
75
|
+
...formatSingleWebSocketEvent("wsError", events.lastErrorEvent),
|
|
76
|
+
...formatSingleWebSocketCloseEvent(events.lastCloseEvent),
|
|
77
|
+
];
|
|
78
|
+
}
|
|
79
|
+
|
|
80
|
+
function formatSingleWebSocketEvent(prefix, event) {
|
|
81
|
+
if (!event) return [` ${prefix} none`];
|
|
82
|
+
return [
|
|
83
|
+
` ${prefix} phase=${formatDebugValue(event.phase)} type=${formatDebugValue(event.type)} keys=${formatDebugValue(event.eventKeys)} readyState=${formatDebugValue(event.readyState)}`,
|
|
84
|
+
` ${prefix} errorName=${formatDebugValue(event.errorName)} errorMessage=${formatDebugValue(event.errorMessage)} errorCode=${formatDebugValue(event.errorCode)} errorString=${formatDebugValue(event.errorString)} eventMessage=${formatDebugValue(event.eventMessage)}`,
|
|
85
|
+
];
|
|
86
|
+
}
|
|
87
|
+
|
|
88
|
+
function formatSingleWebSocketCloseEvent(event) {
|
|
89
|
+
if (!event) return [" wsClose none"];
|
|
90
|
+
return [` wsClose code=${formatDebugValue(event.closeCode)} reason=${formatDebugValue(event.closeReason)} wasClean=${formatDebugValue(event.closeWasClean)} readyState=${formatDebugValue(event.readyState)}`];
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
function formatDebugValue(value) {
|
|
94
|
+
if (value === null || value === undefined || value === "") return "none";
|
|
95
|
+
if (typeof value === "string") return JSON.stringify(value);
|
|
96
|
+
try {
|
|
97
|
+
return JSON.stringify(value);
|
|
98
|
+
} catch {
|
|
99
|
+
return JSON.stringify(String(value));
|
|
100
|
+
}
|
|
56
101
|
}
|
|
57
102
|
|
|
58
103
|
function writeCodexTransportDebug(ui, lines) {
|
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
const SOCKET_SESSION_ID = Symbol.for("march.codex.websocket.sessionId");
|
|
2
|
+
const SOCKET_LISTENERS = Symbol.for("march.codex.websocket.listeners");
|
|
3
|
+
const DEBUG_EVENTS_KEY = Symbol.for("march.codex.websocket.debugEvents");
|
|
4
|
+
const ORIGINAL_WEBSOCKET_KEY = Symbol.for("march.codex.websocket.originalConstructor");
|
|
5
|
+
|
|
6
|
+
export function installCodexWebSocketEventDebug() {
|
|
7
|
+
if (!isCodexTransportDebugEnabled()) return;
|
|
8
|
+
const OriginalWebSocket = globalThis.WebSocket;
|
|
9
|
+
if (typeof OriginalWebSocket !== "function") return;
|
|
10
|
+
if (globalThis[ORIGINAL_WEBSOCKET_KEY]) return;
|
|
11
|
+
|
|
12
|
+
globalThis[DEBUG_EVENTS_KEY] = globalThis[DEBUG_EVENTS_KEY] ?? new Map();
|
|
13
|
+
globalThis[ORIGINAL_WEBSOCKET_KEY] = OriginalWebSocket;
|
|
14
|
+
|
|
15
|
+
class MarchDebugWebSocket extends OriginalWebSocket {
|
|
16
|
+
constructor(...args) {
|
|
17
|
+
super(...args);
|
|
18
|
+
this[SOCKET_SESSION_ID] = extractSessionId(args[1]) ?? extractSessionId(args[2]) ?? null;
|
|
19
|
+
this[SOCKET_LISTENERS] = new Map();
|
|
20
|
+
}
|
|
21
|
+
|
|
22
|
+
addEventListener(type, listener, options) {
|
|
23
|
+
if ((type !== "error" && type !== "close") || !listener) {
|
|
24
|
+
return super.addEventListener(type, listener, options);
|
|
25
|
+
}
|
|
26
|
+
const wrapped = wrapListener(this, type, listener);
|
|
27
|
+
rememberWrappedListener(this, type, listener, wrapped);
|
|
28
|
+
return super.addEventListener(type, wrapped, options);
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
removeEventListener(type, listener, options) {
|
|
32
|
+
const wrapped = takeWrappedListener(this, type, listener) ?? listener;
|
|
33
|
+
return super.removeEventListener(type, wrapped, options);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
|
|
37
|
+
globalThis.WebSocket = MarchDebugWebSocket;
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
export function getCodexWebSocketLastEvent(sessionId) {
|
|
41
|
+
if (!sessionId) return null;
|
|
42
|
+
return globalThis[DEBUG_EVENTS_KEY]?.get(sessionId) ?? null;
|
|
43
|
+
}
|
|
44
|
+
|
|
45
|
+
function isCodexTransportDebugEnabled() {
|
|
46
|
+
const value = process.env.MARCH_CODEX_TRANSPORT_DEBUG;
|
|
47
|
+
return value === "1" || value === "true" || value === "yes";
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
function extractSessionId(options) {
|
|
51
|
+
const headers = options?.headers;
|
|
52
|
+
if (!headers || typeof headers !== "object") return null;
|
|
53
|
+
const value = headers.session_id ?? headers["session_id"] ?? headers["x-client-request-id"] ?? headers["X-Client-Request-Id"];
|
|
54
|
+
return typeof value === "string" && value.length > 0 ? value : null;
|
|
55
|
+
}
|
|
56
|
+
|
|
57
|
+
function wrapListener(socket, type, listener) {
|
|
58
|
+
return function marchCodexWebSocketDebugListener(event) {
|
|
59
|
+
recordWebSocketEvent(socket, type, event);
|
|
60
|
+
if (typeof listener === "function") return listener.call(this, event);
|
|
61
|
+
return listener.handleEvent?.(event);
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
function rememberWrappedListener(socket, type, listener, wrapped) {
|
|
66
|
+
let byType = socket[SOCKET_LISTENERS].get(type);
|
|
67
|
+
if (!byType) {
|
|
68
|
+
byType = new Map();
|
|
69
|
+
socket[SOCKET_LISTENERS].set(type, byType);
|
|
70
|
+
}
|
|
71
|
+
byType.set(listener, wrapped);
|
|
72
|
+
}
|
|
73
|
+
|
|
74
|
+
function takeWrappedListener(socket, type, listener) {
|
|
75
|
+
const byType = socket[SOCKET_LISTENERS]?.get(type);
|
|
76
|
+
if (!byType) return null;
|
|
77
|
+
const wrapped = byType.get(listener);
|
|
78
|
+
byType.delete(listener);
|
|
79
|
+
return wrapped ?? null;
|
|
80
|
+
}
|
|
81
|
+
|
|
82
|
+
function recordWebSocketEvent(socket, type, event) {
|
|
83
|
+
const sessionId = socket[SOCKET_SESSION_ID];
|
|
84
|
+
if (!sessionId) return;
|
|
85
|
+
const events = globalThis[DEBUG_EVENTS_KEY].get(sessionId) ?? {};
|
|
86
|
+
const summary = summarizeWebSocketEvent(socket, type, event);
|
|
87
|
+
events.lastEvent = summary;
|
|
88
|
+
if (type === "error") events.lastErrorEvent = summary;
|
|
89
|
+
if (type === "close") events.lastCloseEvent = summary;
|
|
90
|
+
globalThis[DEBUG_EVENTS_KEY].set(sessionId, events);
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
function summarizeWebSocketEvent(socket, type, event) {
|
|
94
|
+
const nestedError = event && typeof event === "object" && "error" in event ? event.error : null;
|
|
95
|
+
return {
|
|
96
|
+
phase: type,
|
|
97
|
+
type: readString(event, "type") ?? type,
|
|
98
|
+
eventKeys: getEventKeys(event),
|
|
99
|
+
eventMessage: readString(event, "message"),
|
|
100
|
+
errorName: readString(nestedError, "name"),
|
|
101
|
+
errorMessage: readString(nestedError, "message"),
|
|
102
|
+
errorCode: readString(nestedError, "code") ?? readNumber(nestedError, "code"),
|
|
103
|
+
errorString: nestedError ? String(nestedError) : undefined,
|
|
104
|
+
closeCode: readNumber(event, "code"),
|
|
105
|
+
closeReason: readString(event, "reason"),
|
|
106
|
+
closeWasClean: readBoolean(event, "wasClean"),
|
|
107
|
+
readyState: typeof socket.readyState === "number" ? socket.readyState : undefined,
|
|
108
|
+
};
|
|
109
|
+
}
|
|
110
|
+
|
|
111
|
+
function getEventKeys(event) {
|
|
112
|
+
if (!event || typeof event !== "object") return [];
|
|
113
|
+
return [...new Set([...Object.keys(event), "type", "message", "error", "code", "reason", "wasClean"].filter((key) => key in event))];
|
|
114
|
+
}
|
|
115
|
+
|
|
116
|
+
function readString(value, key) {
|
|
117
|
+
const field = value && typeof value === "object" ? value[key] : undefined;
|
|
118
|
+
return typeof field === "string" && field.length > 0 ? field : undefined;
|
|
119
|
+
}
|
|
120
|
+
|
|
121
|
+
function readNumber(value, key) {
|
|
122
|
+
const field = value && typeof value === "object" ? value[key] : undefined;
|
|
123
|
+
return typeof field === "number" ? field : undefined;
|
|
124
|
+
}
|
|
125
|
+
|
|
126
|
+
function readBoolean(value, key) {
|
|
127
|
+
const field = value && typeof value === "object" ? value[key] : undefined;
|
|
128
|
+
return typeof field === "boolean" ? field : undefined;
|
|
129
|
+
}
|
|
130
|
+
|
package/src/agent/runner.mjs
CHANGED
|
@@ -13,6 +13,7 @@ import { createRuntimeUiBridge } from "./runtime/ui-event-bridge.mjs";
|
|
|
13
13
|
import { getRunnerSessionStats, syncEngineSessionState } from "./runner/runner-session-state.mjs";
|
|
14
14
|
import { notifyTurnEndBestEffort, notifyTurnEndDetached, providerContextToPayload } from "./runner/runner-utils.mjs";
|
|
15
15
|
import { dumpCodexTransportDebug, getCodexTransportDebugSnapshot } from "./runner/codex-transport-debug.mjs";
|
|
16
|
+
import { installCodexWebSocketEventDebug } from "./runner/codex-websocket-event-debug.mjs";
|
|
16
17
|
import { resolveRunnerSessionOptions } from "./session/session-options.mjs";
|
|
17
18
|
import { createSessionBinding } from "./session/session-binding.mjs";
|
|
18
19
|
import { maybeAutoNameSession } from "./session/session-auto-name.mjs";
|
|
@@ -28,6 +29,7 @@ export { installModelPayloadDumper } from "./model-payload-dumper.mjs";
|
|
|
28
29
|
export { createDefaultSessionManager, resolveRunnerSessionManager } from "./runner/runner-init.mjs";
|
|
29
30
|
export { getRunnerSessionStats, syncEngineSessionState } from "./runner/runner-session-state.mjs";
|
|
30
31
|
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 = {} }) {
|
|
32
|
+
installCodexWebSocketEventDebug();
|
|
31
33
|
if (!useRuntimeHost && extensionPaths.length > 0) {
|
|
32
34
|
throw new Error("--extension requires the default pi runtime host path");
|
|
33
35
|
}
|