@zhixuan92/multi-model-agent 5.0.1 → 5.0.3
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 +8 -9
- package/dist/cli/index.d.ts +62 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +345 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/info.d.ts +22 -0
- package/dist/cli/info.d.ts.map +1 -0
- package/dist/cli/info.js +100 -0
- package/dist/cli/info.js.map +1 -0
- package/dist/cli/logs.d.ts +15 -0
- package/dist/cli/logs.d.ts.map +1 -0
- package/dist/cli/logs.js +102 -0
- package/dist/cli/logs.js.map +1 -0
- package/dist/cli/print-token.d.ts +18 -0
- package/dist/cli/print-token.d.ts.map +1 -0
- package/dist/cli/print-token.js +60 -0
- package/dist/cli/print-token.js.map +1 -0
- package/dist/cli/serve.d.ts +28 -0
- package/dist/cli/serve.d.ts.map +1 -0
- package/dist/cli/serve.js +405 -0
- package/dist/cli/serve.js.map +1 -0
- package/dist/cli/status.d.ts +49 -0
- package/dist/cli/status.d.ts.map +1 -0
- package/dist/cli/status.js +155 -0
- package/dist/cli/status.js.map +1 -0
- package/dist/cli/sync-skills.d.ts +58 -0
- package/dist/cli/sync-skills.d.ts.map +1 -0
- package/dist/cli/sync-skills.js +266 -0
- package/dist/cli/sync-skills.js.map +1 -0
- package/dist/cli/telemetry.d.ts +10 -0
- package/dist/cli/telemetry.d.ts.map +1 -0
- package/dist/cli/telemetry.js +161 -0
- package/dist/cli/telemetry.js.map +1 -0
- package/dist/cli/toggle.d.ts +26 -0
- package/dist/cli/toggle.d.ts.map +1 -0
- package/dist/cli/toggle.js +185 -0
- package/dist/cli/toggle.js.map +1 -0
- package/dist/http/async-dispatch.d.ts +44 -0
- package/dist/http/async-dispatch.d.ts.map +1 -0
- package/dist/http/async-dispatch.js +175 -0
- package/dist/http/async-dispatch.js.map +1 -0
- package/dist/http/auth.d.ts +20 -0
- package/dist/http/auth.d.ts.map +1 -0
- package/dist/http/auth.js +56 -0
- package/dist/http/auth.js.map +1 -0
- package/dist/http/canonicalize-file-paths.d.ts +8 -0
- package/dist/http/canonicalize-file-paths.d.ts.map +1 -0
- package/dist/http/canonicalize-file-paths.js +43 -0
- package/dist/http/canonicalize-file-paths.js.map +1 -0
- package/dist/http/cwd-validator.d.ts +11 -0
- package/dist/http/cwd-validator.d.ts.map +1 -0
- package/dist/http/cwd-validator.js +130 -0
- package/dist/http/cwd-validator.js.map +1 -0
- package/dist/http/errors.d.ts +4 -0
- package/dist/http/errors.d.ts.map +1 -0
- package/dist/http/errors.js +9 -0
- package/dist/http/errors.js.map +1 -0
- package/dist/http/execution-context.d.ts +18 -0
- package/dist/http/execution-context.d.ts.map +1 -0
- package/dist/http/execution-context.js +61 -0
- package/dist/http/execution-context.js.map +1 -0
- package/dist/http/handler-deps.d.ts +19 -0
- package/dist/http/handler-deps.d.ts.map +1 -0
- package/dist/http/handler-deps.js +2 -0
- package/dist/http/handler-deps.js.map +1 -0
- package/dist/http/handlers/control/batch-slice.d.ts +4 -0
- package/dist/http/handlers/control/batch-slice.d.ts.map +1 -0
- package/dist/http/handlers/control/batch-slice.js +40 -0
- package/dist/http/handlers/control/batch-slice.js.map +1 -0
- package/dist/http/handlers/control/batch.d.ts +23 -0
- package/dist/http/handlers/control/batch.d.ts.map +1 -0
- package/dist/http/handlers/control/batch.js +332 -0
- package/dist/http/handlers/control/batch.js.map +1 -0
- package/dist/http/handlers/control/context-blocks.d.ts +22 -0
- package/dist/http/handlers/control/context-blocks.d.ts.map +1 -0
- package/dist/http/handlers/control/context-blocks.js +111 -0
- package/dist/http/handlers/control/context-blocks.js.map +1 -0
- package/dist/http/handlers/introspection/health.d.ts +20 -0
- package/dist/http/handlers/introspection/health.d.ts.map +1 -0
- package/dist/http/handlers/introspection/health.js +18 -0
- package/dist/http/handlers/introspection/health.js.map +1 -0
- package/dist/http/handlers/introspection/status.d.ts +26 -0
- package/dist/http/handlers/introspection/status.d.ts.map +1 -0
- package/dist/http/handlers/introspection/status.js +136 -0
- package/dist/http/handlers/introspection/status.js.map +1 -0
- package/dist/http/handlers/tools/audit.d.ts +4 -0
- package/dist/http/handlers/tools/audit.d.ts.map +1 -0
- package/dist/http/handlers/tools/audit.js +43 -0
- package/dist/http/handlers/tools/audit.js.map +1 -0
- package/dist/http/handlers/tools/debug.d.ts +4 -0
- package/dist/http/handlers/tools/debug.d.ts.map +1 -0
- package/dist/http/handlers/tools/debug.js +43 -0
- package/dist/http/handlers/tools/debug.js.map +1 -0
- package/dist/http/handlers/tools/delegate.d.ts +4 -0
- package/dist/http/handlers/tools/delegate.d.ts.map +1 -0
- package/dist/http/handlers/tools/delegate.js +43 -0
- package/dist/http/handlers/tools/delegate.js.map +1 -0
- package/dist/http/handlers/tools/execute-plan.d.ts +4 -0
- package/dist/http/handlers/tools/execute-plan.d.ts.map +1 -0
- package/dist/http/handlers/tools/execute-plan.js +45 -0
- package/dist/http/handlers/tools/execute-plan.js.map +1 -0
- package/dist/http/handlers/tools/investigate.d.ts +4 -0
- package/dist/http/handlers/tools/investigate.d.ts.map +1 -0
- package/dist/http/handlers/tools/investigate.js +64 -0
- package/dist/http/handlers/tools/investigate.js.map +1 -0
- package/dist/http/handlers/tools/journal-recall.d.ts +4 -0
- package/dist/http/handlers/tools/journal-recall.d.ts.map +1 -0
- package/dist/http/handlers/tools/journal-recall.js +40 -0
- package/dist/http/handlers/tools/journal-recall.js.map +1 -0
- package/dist/http/handlers/tools/journal-record.d.ts +8 -0
- package/dist/http/handlers/tools/journal-record.d.ts.map +1 -0
- package/dist/http/handlers/tools/journal-record.js +40 -0
- package/dist/http/handlers/tools/journal-record.js.map +1 -0
- package/dist/http/handlers/tools/research.d.ts +4 -0
- package/dist/http/handlers/tools/research.d.ts.map +1 -0
- package/dist/http/handlers/tools/research.js +64 -0
- package/dist/http/handlers/tools/research.js.map +1 -0
- package/dist/http/handlers/tools/retry.d.ts +4 -0
- package/dist/http/handlers/tools/retry.d.ts.map +1 -0
- package/dist/http/handlers/tools/retry.js +73 -0
- package/dist/http/handlers/tools/retry.js.map +1 -0
- package/dist/http/handlers/tools/review.d.ts +4 -0
- package/dist/http/handlers/tools/review.d.ts.map +1 -0
- package/dist/http/handlers/tools/review.js +43 -0
- package/dist/http/handlers/tools/review.js.map +1 -0
- package/dist/http/journal-lock.d.ts +4 -0
- package/dist/http/journal-lock.d.ts.map +1 -0
- package/dist/http/journal-lock.js +34 -0
- package/dist/http/journal-lock.js.map +1 -0
- package/dist/http/middleware/body-reader.d.ts +16 -0
- package/dist/http/middleware/body-reader.d.ts.map +1 -0
- package/dist/http/middleware/body-reader.js +44 -0
- package/dist/http/middleware/body-reader.js.map +1 -0
- package/dist/http/middleware/caller-identity.d.ts +16 -0
- package/dist/http/middleware/caller-identity.d.ts.map +1 -0
- package/dist/http/middleware/caller-identity.js +16 -0
- package/dist/http/middleware/caller-identity.js.map +1 -0
- package/dist/http/middleware/decompress.d.ts +14 -0
- package/dist/http/middleware/decompress.d.ts.map +1 -0
- package/dist/http/middleware/decompress.js +51 -0
- package/dist/http/middleware/decompress.js.map +1 -0
- package/dist/http/project-registry.d.ts +54 -0
- package/dist/http/project-registry.d.ts.map +1 -0
- package/dist/http/project-registry.js +130 -0
- package/dist/http/project-registry.js.map +1 -0
- package/dist/http/request-observability.d.ts +8 -0
- package/dist/http/request-observability.d.ts.map +1 -0
- package/dist/http/request-observability.js +20 -0
- package/dist/http/request-observability.js.map +1 -0
- package/dist/http/request-pipeline.d.ts +16 -0
- package/dist/http/request-pipeline.d.ts.map +1 -0
- package/dist/http/request-pipeline.js +144 -0
- package/dist/http/request-pipeline.js.map +1 -0
- package/dist/http/server.d.ts +17 -0
- package/dist/http/server.d.ts.map +1 -0
- package/dist/http/server.js +300 -0
- package/dist/http/server.js.map +1 -0
- package/dist/http/types.d.ts +20 -0
- package/dist/http/types.d.ts.map +1 -0
- package/dist/http/types.js +2 -0
- package/dist/http/types.js.map +1 -0
- package/dist/skill-install/disabled-state.d.ts +35 -0
- package/dist/skill-install/disabled-state.d.ts.map +1 -0
- package/dist/skill-install/disabled-state.js +96 -0
- package/dist/skill-install/disabled-state.js.map +1 -0
- package/dist/skill-install/discover.d.ts +29 -0
- package/dist/skill-install/discover.d.ts.map +1 -0
- package/dist/skill-install/discover.js +104 -0
- package/dist/skill-install/discover.js.map +1 -0
- package/dist/skill-install/include-utils.d.ts +27 -0
- package/dist/skill-install/include-utils.d.ts.map +1 -0
- package/dist/skill-install/include-utils.js +90 -0
- package/dist/skill-install/include-utils.js.map +1 -0
- package/dist/skill-install/manifest.d.ts +82 -0
- package/dist/skill-install/manifest.d.ts.map +1 -0
- package/dist/skill-install/manifest.js +215 -0
- package/dist/skill-install/manifest.js.map +1 -0
- package/dist/skill-install/skill-installer-common.d.ts +26 -0
- package/dist/skill-install/skill-installer-common.d.ts.map +1 -0
- package/dist/skill-install/skill-installer-common.js +139 -0
- package/dist/skill-install/skill-installer-common.js.map +1 -0
- package/dist/skill-install/skill-installers/claude-code.d.ts +43 -0
- package/dist/skill-install/skill-installers/claude-code.d.ts.map +1 -0
- package/dist/skill-install/skill-installers/claude-code.js +65 -0
- package/dist/skill-install/skill-installers/claude-code.js.map +1 -0
- package/dist/skill-install/skill-installers/codex-cli.d.ts +27 -0
- package/dist/skill-install/skill-installers/codex-cli.d.ts.map +1 -0
- package/dist/skill-install/skill-installers/codex-cli.js +84 -0
- package/dist/skill-install/skill-installers/codex-cli.js.map +1 -0
- package/dist/skill-install/skill-installers/cursor.d.ts +72 -0
- package/dist/skill-install/skill-installers/cursor.d.ts.map +1 -0
- package/dist/skill-install/skill-installers/cursor.js +81 -0
- package/dist/skill-install/skill-installers/cursor.js.map +1 -0
- package/dist/skill-install/skill-installers/gemini-cli.d.ts +50 -0
- package/dist/skill-install/skill-installers/gemini-cli.d.ts.map +1 -0
- package/dist/skill-install/skill-installers/gemini-cli.js +72 -0
- package/dist/skill-install/skill-installers/gemini-cli.js.map +1 -0
- package/dist/skill-install/skill-manifest-sync.d.ts +11 -0
- package/dist/skill-install/skill-manifest-sync.d.ts.map +1 -0
- package/dist/skill-install/skill-manifest-sync.js +65 -0
- package/dist/skill-install/skill-manifest-sync.js.map +1 -0
- package/dist/skills/_shared/auth.md +41 -0
- package/dist/skills/_shared/error-handling.md +31 -0
- package/dist/skills/_shared/polling.md +88 -0
- package/dist/skills/_shared/response-shape.md +55 -0
- package/dist/skills/_shared/review-policy.md +15 -0
- package/dist/skills/mma-audit/SKILL.md +270 -0
- package/dist/skills/mma-context-blocks/SKILL.md +148 -0
- package/dist/skills/mma-debug/SKILL.md +208 -0
- package/dist/skills/mma-delegate/SKILL.md +216 -0
- package/dist/skills/mma-execute-plan/SKILL.md +214 -0
- package/dist/skills/mma-explore/SKILL.md +190 -0
- package/dist/skills/mma-investigate/SKILL.md +258 -0
- package/dist/skills/mma-journal-recall/SKILL.md +242 -0
- package/dist/skills/mma-journal-record/SKILL.md +202 -0
- package/dist/skills/mma-research/SKILL.md +223 -0
- package/dist/skills/mma-retry/SKILL.md +221 -0
- package/dist/skills/mma-review/SKILL.md +209 -0
- package/dist/skills/multi-model-agent/SKILL.md +206 -0
- package/dist/telemetry/consent.d.ts +4 -0
- package/dist/telemetry/consent.d.ts.map +1 -0
- package/dist/telemetry/consent.js +40 -0
- package/dist/telemetry/consent.js.map +1 -0
- package/dist/telemetry/flusher.d.ts +19 -0
- package/dist/telemetry/flusher.d.ts.map +1 -0
- package/dist/telemetry/flusher.js +277 -0
- package/dist/telemetry/flusher.js.map +1 -0
- package/dist/telemetry/generation.d.ts +9 -0
- package/dist/telemetry/generation.d.ts.map +1 -0
- package/dist/telemetry/generation.js +33 -0
- package/dist/telemetry/generation.js.map +1 -0
- package/dist/telemetry/identity.d.ts +9 -0
- package/dist/telemetry/identity.d.ts.map +1 -0
- package/dist/telemetry/identity.js +35 -0
- package/dist/telemetry/identity.js.map +1 -0
- package/dist/telemetry/install-id.d.ts +13 -0
- package/dist/telemetry/install-id.d.ts.map +1 -0
- package/dist/telemetry/install-id.js +49 -0
- package/dist/telemetry/install-id.js.map +1 -0
- package/dist/telemetry/install-meta.d.ts +10 -0
- package/dist/telemetry/install-meta.d.ts.map +1 -0
- package/dist/telemetry/install-meta.js +15 -0
- package/dist/telemetry/install-meta.js.map +1 -0
- package/dist/telemetry/queue.d.ts +35 -0
- package/dist/telemetry/queue.d.ts.map +1 -0
- package/dist/telemetry/queue.js +287 -0
- package/dist/telemetry/queue.js.map +1 -0
- package/dist/telemetry/recorder.d.ts +39 -0
- package/dist/telemetry/recorder.d.ts.map +1 -0
- package/dist/telemetry/recorder.js +173 -0
- package/dist/telemetry/recorder.js.map +1 -0
- package/package.json +43 -24
- package/scripts/postinstall.js +36 -0
- package/bin/mmagent.mjs +0 -47
- package/postinstall.mjs +0 -8
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
import { createProjectContext } from '@zhixuan92/multi-model-agent-core';
|
|
2
|
+
import { validateCwd } from './cwd-validator.js';
|
|
3
|
+
export class ProjectRegistry {
|
|
4
|
+
map = new Map();
|
|
5
|
+
cap;
|
|
6
|
+
idleEvictionMs;
|
|
7
|
+
evictionIntervalMs;
|
|
8
|
+
evictionTimer = null;
|
|
9
|
+
onProjectCreated;
|
|
10
|
+
onProjectEvicted;
|
|
11
|
+
constructor(options) {
|
|
12
|
+
this.cap = options.cap;
|
|
13
|
+
this.idleEvictionMs = options.idleEvictionMs;
|
|
14
|
+
this.evictionIntervalMs = options.evictionIntervalMs;
|
|
15
|
+
this.onProjectCreated = options.onProjectCreated;
|
|
16
|
+
this.onProjectEvicted = options.onProjectEvicted;
|
|
17
|
+
}
|
|
18
|
+
startEvictionTimer() {
|
|
19
|
+
if (this.evictionTimer)
|
|
20
|
+
return;
|
|
21
|
+
this.evictionTimer = setInterval(() => this.evictIdle(), this.evictionIntervalMs);
|
|
22
|
+
this.evictionTimer.unref?.();
|
|
23
|
+
}
|
|
24
|
+
stopEvictionTimer() {
|
|
25
|
+
if (this.evictionTimer)
|
|
26
|
+
clearInterval(this.evictionTimer);
|
|
27
|
+
this.evictionTimer = null;
|
|
28
|
+
}
|
|
29
|
+
/** Synchronous lookup-or-create with cap enforcement. Increments pendingReservations on success. */
|
|
30
|
+
reserveProject(cwd) {
|
|
31
|
+
const v = validateCwd(cwd);
|
|
32
|
+
if (!v.ok)
|
|
33
|
+
return { ok: false, error: v.error, message: v.message };
|
|
34
|
+
const key = v.canonicalCwd;
|
|
35
|
+
const existing = this.map.get(key);
|
|
36
|
+
if (existing) {
|
|
37
|
+
existing.pendingReservations += 1;
|
|
38
|
+
existing.lastActivityAt = Date.now();
|
|
39
|
+
return { ok: true, projectContext: existing, created: false };
|
|
40
|
+
}
|
|
41
|
+
if (this.map.size >= this.cap) {
|
|
42
|
+
return { ok: false, error: 'project_cap', message: `server at ${this.cap} projects; close some connections and retry` };
|
|
43
|
+
}
|
|
44
|
+
const pc = createProjectContext(key);
|
|
45
|
+
pc.pendingReservations = 1;
|
|
46
|
+
this.map.set(key, pc);
|
|
47
|
+
this.onProjectCreated?.(key);
|
|
48
|
+
return { ok: true, projectContext: pc, created: true };
|
|
49
|
+
}
|
|
50
|
+
/**
|
|
51
|
+
* Called from onsessioninitialized. Decrements pendingReservations and registers the session.
|
|
52
|
+
* `canonicalCwd` MUST be the `.cwd` value returned from a prior `reserveProject` call.
|
|
53
|
+
*/
|
|
54
|
+
attachSession(canonicalCwd, sessionId) {
|
|
55
|
+
const pc = this.map.get(canonicalCwd);
|
|
56
|
+
if (!pc)
|
|
57
|
+
throw new Error(`attachSession: no project for ${canonicalCwd} (did you pass the raw cwd instead of projectContext.cwd?)`);
|
|
58
|
+
pc.activeSessions.add(sessionId);
|
|
59
|
+
if (pc.pendingReservations > 0)
|
|
60
|
+
pc.pendingReservations -= 1;
|
|
61
|
+
pc.lastActivityAt = Date.now();
|
|
62
|
+
}
|
|
63
|
+
/** Detach a session. `canonicalCwd` must match `projectContext.cwd`. No-op if unknown. */
|
|
64
|
+
detachSession(canonicalCwd, sessionId) {
|
|
65
|
+
const pc = this.map.get(canonicalCwd);
|
|
66
|
+
if (!pc)
|
|
67
|
+
return;
|
|
68
|
+
pc.activeSessions.delete(sessionId);
|
|
69
|
+
pc.lastActivityAt = Date.now();
|
|
70
|
+
}
|
|
71
|
+
/** Called if attachSession never fires. `canonicalCwd` must match `projectContext.cwd`. */
|
|
72
|
+
cancelReservation(canonicalCwd) {
|
|
73
|
+
const pc = this.map.get(canonicalCwd);
|
|
74
|
+
if (!pc)
|
|
75
|
+
return;
|
|
76
|
+
if (pc.pendingReservations > 0)
|
|
77
|
+
pc.pendingReservations -= 1;
|
|
78
|
+
pc.lastActivityAt = Date.now();
|
|
79
|
+
}
|
|
80
|
+
/** Look up a project by its canonical cwd. */
|
|
81
|
+
get(canonicalCwd) {
|
|
82
|
+
return this.map.get(canonicalCwd);
|
|
83
|
+
}
|
|
84
|
+
get size() {
|
|
85
|
+
return this.map.size;
|
|
86
|
+
}
|
|
87
|
+
*entries() {
|
|
88
|
+
yield* this.map.entries();
|
|
89
|
+
}
|
|
90
|
+
/**
|
|
91
|
+
* Returns true if the project context is idle and eligible for eviction.
|
|
92
|
+
* Gates on: no active HTTP requests, no active batches in the registry,
|
|
93
|
+
* no active sessions, no pending reservations, and idle for at least idleTimeoutMs.
|
|
94
|
+
*/
|
|
95
|
+
isIdleFor(pc, now, idleTimeoutMs, batchRegistry) {
|
|
96
|
+
return (pc.activeRequests === 0 &&
|
|
97
|
+
batchRegistry.countActiveForProject(pc.cwd) === 0 &&
|
|
98
|
+
pc.activeSessions.size === 0 &&
|
|
99
|
+
pc.pendingReservations === 0 &&
|
|
100
|
+
(now - pc.lastActivityAt) > idleTimeoutMs);
|
|
101
|
+
}
|
|
102
|
+
evictIdle(batchRegistry) {
|
|
103
|
+
const now = Date.now();
|
|
104
|
+
const victims = [];
|
|
105
|
+
for (const [cwd, pc] of this.map.entries()) {
|
|
106
|
+
if (pc.activeSessions.size === 0 &&
|
|
107
|
+
pc.activeRequests === 0 &&
|
|
108
|
+
pc.pendingReservations === 0 &&
|
|
109
|
+
(batchRegistry === undefined || batchRegistry.countActiveForProject(cwd) === 0) &&
|
|
110
|
+
now - pc.lastActivityAt > this.idleEvictionMs) {
|
|
111
|
+
victims.push(cwd);
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
for (const cwd of victims) {
|
|
115
|
+
const pc = this.map.get(cwd);
|
|
116
|
+
pc.contextBlocks.clear();
|
|
117
|
+
pc.batchCache.clear();
|
|
118
|
+
this.map.delete(cwd);
|
|
119
|
+
this.onProjectEvicted?.(cwd, now - pc.lastActivityAt);
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
clear() {
|
|
123
|
+
for (const pc of this.map.values()) {
|
|
124
|
+
pc.contextBlocks.clear();
|
|
125
|
+
pc.batchCache.clear();
|
|
126
|
+
}
|
|
127
|
+
this.map.clear();
|
|
128
|
+
}
|
|
129
|
+
}
|
|
130
|
+
//# sourceMappingURL=project-registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"project-registry.js","sourceRoot":"","sources":["../../src/http/project-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAsC,MAAM,mCAAmC,CAAC;AAC7G,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAgBjD,MAAM,OAAO,eAAe;IACT,GAAG,GAAG,IAAI,GAAG,EAA0B,CAAC;IACxC,GAAG,CAAS;IACZ,cAAc,CAAS;IACvB,kBAAkB,CAAS;IACpC,aAAa,GAA0B,IAAI,CAAC;IACnC,gBAAgB,CAAyB;IACzC,gBAAgB,CAAyC;IAE1E,YAAY,OAA+B;QACzC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC7C,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,CAAC;QACrD,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACjD,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IACnD,CAAC;IAED,kBAAkB;QAChB,IAAI,IAAI,CAAC,aAAa;YAAE,OAAO;QAC/B,IAAI,CAAC,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAClF,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,EAAE,CAAC;IAC/B,CAAC;IAED,iBAAiB;QACf,IAAI,IAAI,CAAC,aAAa;YAAE,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;IAC5B,CAAC;IAED,oGAAoG;IACpG,cAAc,CAAC,GAAW;QACxB,MAAM,CAAC,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;QAC3B,IAAI,CAAC,CAAC,CAAC,EAAE;YAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QACpE,MAAM,GAAG,GAAG,CAAC,CAAC,YAAY,CAAC;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,mBAAmB,IAAI,CAAC,CAAC;YAClC,QAAQ,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACrC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAChE,CAAC;QACD,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YAC9B,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,aAAa,IAAI,CAAC,GAAG,6CAA6C,EAAE,CAAC;QAC1H,CAAC;QACD,MAAM,EAAE,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;QACrC,EAAE,CAAC,mBAAmB,GAAG,CAAC,CAAC;QAC3B,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACtB,IAAI,CAAC,gBAAgB,EAAE,CAAC,GAAG,CAAC,CAAC;QAC7B,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IACzD,CAAC;IAED;;;OAGG;IACH,aAAa,CAAC,YAAoB,EAAE,SAAiB;QACnD,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtC,IAAI,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,YAAY,4DAA4D,CAAC,CAAC;QACpI,EAAE,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjC,IAAI,EAAE,CAAC,mBAAmB,GAAG,CAAC;YAAE,EAAE,CAAC,mBAAmB,IAAI,CAAC,CAAC;QAC5D,EAAE,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACjC,CAAC;IAED,0FAA0F;IAC1F,aAAa,CAAC,YAAoB,EAAE,SAAiB;QACnD,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtC,IAAI,CAAC,EAAE;YAAE,OAAO;QAChB,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACpC,EAAE,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACjC,CAAC;IAED,2FAA2F;IAC3F,iBAAiB,CAAC,YAAoB;QACpC,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACtC,IAAI,CAAC,EAAE;YAAE,OAAO;QAChB,IAAI,EAAE,CAAC,mBAAmB,GAAG,CAAC;YAAE,EAAE,CAAC,mBAAmB,IAAI,CAAC,CAAC;QAC5D,EAAE,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACjC,CAAC;IAED,8CAA8C;IAC9C,GAAG,CAAC,YAAoB;QACtB,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACpC,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;IACvB,CAAC;IAED,CAAC,OAAO;QACN,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAC5B,CAAC;IAED;;;;OAIG;IACH,SAAS,CAAC,EAAkB,EAAE,GAAW,EAAE,aAAqB,EAAE,aAA4B;QAC5F,OAAO,CACL,EAAE,CAAC,cAAc,KAAK,CAAC;YACvB,aAAa,CAAC,qBAAqB,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC;YACjD,EAAE,CAAC,cAAc,CAAC,IAAI,KAAK,CAAC;YAC5B,EAAE,CAAC,mBAAmB,KAAK,CAAC;YAC5B,CAAC,GAAG,GAAG,EAAE,CAAC,cAAc,CAAC,GAAG,aAAa,CAC1C,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,aAA6B;QACrC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC;YAC3C,IACE,EAAE,CAAC,cAAc,CAAC,IAAI,KAAK,CAAC;gBAC5B,EAAE,CAAC,cAAc,KAAK,CAAC;gBACvB,EAAE,CAAC,mBAAmB,KAAK,CAAC;gBAC5B,CAAC,aAAa,KAAK,SAAS,IAAI,aAAa,CAAC,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAC/E,GAAG,GAAG,EAAE,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,EAC7C,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACpB,CAAC;QACH,CAAC;QACD,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;YAC9B,EAAE,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;YACzB,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACtB,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACrB,IAAI,CAAC,gBAAgB,EAAE,CAAC,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC,cAAc,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED,KAAK;QACH,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC;YACnC,EAAE,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;YACzB,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;IACnB,CAAC;CACF"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { EnvelopeBus } from '@zhixuan92/multi-model-agent-core/events/envelope-bus';
|
|
2
|
+
import type { LogWriter } from '@zhixuan92/multi-model-agent-core/events/log-writer';
|
|
3
|
+
export interface EmitRequestReceivedDeps {
|
|
4
|
+
bus: EnvelopeBus;
|
|
5
|
+
logWriter: LogWriter;
|
|
6
|
+
}
|
|
7
|
+
export declare function emitRequestReceived(deps: EmitRequestReceivedDeps, batchId: string, route: string, parsed: unknown): Promise<void>;
|
|
8
|
+
//# sourceMappingURL=request-observability.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"request-observability.d.ts","sourceRoot":"","sources":["../../src/http/request-observability.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,uDAAuD,CAAC;AACzF,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,qDAAqD,CAAC;AAErF,MAAM,WAAW,uBAAuB;IACtC,GAAG,EAAE,WAAW,CAAC;IACjB,SAAS,EAAE,SAAS,CAAC;CACtB;AAED,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,uBAAuB,EAC7B,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,OAAO,GACd,OAAO,CAAC,IAAI,CAAC,CAmBf"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
export async function emitRequestReceived(deps, batchId, route, parsed) {
|
|
2
|
+
const json = JSON.stringify(parsed);
|
|
3
|
+
const bodyBytes = Buffer.byteLength(json, 'utf8');
|
|
4
|
+
const ts = new Date().toISOString();
|
|
5
|
+
// Emit batch_created plain entry
|
|
6
|
+
deps.bus.emitPlainEntry({ ts, kind: 'batch_created', fields: { batch_id: batchId, route } });
|
|
7
|
+
// Emit request_received plain entry
|
|
8
|
+
const inline = bodyBytes <= deps.logWriter.inlineBodyLimit();
|
|
9
|
+
const fields = { batch_id: batchId, route, body_bytes: bodyBytes };
|
|
10
|
+
if (inline) {
|
|
11
|
+
fields.body = json;
|
|
12
|
+
}
|
|
13
|
+
else {
|
|
14
|
+
const spilled = await deps.logWriter.spillRequestBody({ batchId, body: parsed });
|
|
15
|
+
fields.body_path = spilled.path;
|
|
16
|
+
fields.body_bytes = spilled.bytes;
|
|
17
|
+
}
|
|
18
|
+
deps.bus.emitPlainEntry({ ts, kind: 'request_received', fields });
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=request-observability.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"request-observability.js","sourceRoot":"","sources":["../../src/http/request-observability.ts"],"names":[],"mappings":"AAQA,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,IAA6B,EAC7B,OAAe,EACf,KAAa,EACb,MAAe;IAEf,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACpC,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAClD,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAEpC,iCAAiC;IACjC,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;IAE7F,oCAAoC;IACpC,MAAM,MAAM,GAAG,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;IAC7D,MAAM,MAAM,GAAqD,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;IACrH,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;IACrB,CAAC;SAAM,CAAC;QACN,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QACjF,MAAM,CAAC,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;QAChC,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC;IACpC,CAAC;IACD,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,CAAC,CAAC;AACpE,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { IncomingMessage, ServerResponse } from 'node:http';
|
|
2
|
+
import type { ServerConfig } from '@zhixuan92/multi-model-agent-core';
|
|
3
|
+
import { RouteDispatcher } from '@zhixuan92/multi-model-agent-core';
|
|
4
|
+
import type { RawHandler } from './types.js';
|
|
5
|
+
export declare function setDraining(d: boolean): void;
|
|
6
|
+
export declare function isDraining(): boolean;
|
|
7
|
+
export interface PipelineConfig {
|
|
8
|
+
loopbackOnlyPaths: ReadonlySet<string>;
|
|
9
|
+
authExemptPaths: ReadonlySet<string>;
|
|
10
|
+
cwdRequiredPaths: ReadonlySet<string>;
|
|
11
|
+
/** Routes that REQUIRE X-MMA-Main-Model header. Tool dispatches must
|
|
12
|
+
* attribute to a main model so wire telemetry's main_model is never null. */
|
|
13
|
+
mainModelRequiredPaths: ReadonlySet<string>;
|
|
14
|
+
}
|
|
15
|
+
export declare function handleRequest(router: RouteDispatcher<RawHandler>, token: string, req: IncomingMessage, res: ServerResponse, cfg: ServerConfig, pipelineCfg: PipelineConfig): Promise<void>;
|
|
16
|
+
//# sourceMappingURL=request-pipeline.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"request-pipeline.d.ts","sourceRoot":"","sources":["../../src/http/request-pipeline.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AACjE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAC;AACtE,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AAQpE,OAAO,KAAK,EAAkB,UAAU,EAAE,MAAM,YAAY,CAAC;AAS7D,wBAAgB,WAAW,CAAC,CAAC,EAAE,OAAO,GAAG,IAAI,CAAsB;AACnE,wBAAgB,UAAU,IAAI,OAAO,CAAyB;AAE9D,MAAM,WAAW,cAAc;IAC7B,iBAAiB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IACvC,eAAe,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IACrC,gBAAgB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IACtC;kFAC8E;IAC9E,sBAAsB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;CAC7C;AAED,wBAAsB,aAAa,CACjC,MAAM,EAAE,eAAe,CAAC,UAAU,CAAC,EACnC,KAAK,EAAE,MAAM,EACb,GAAG,EAAE,eAAe,EACpB,GAAG,EAAE,cAAc,EACnB,GAAG,EAAE,YAAY,EACjB,WAAW,EAAE,cAAc,GAC1B,OAAO,CAAC,IAAI,CAAC,CAuJf"}
|
|
@@ -0,0 +1,144 @@
|
|
|
1
|
+
import { sendError } from './errors.js';
|
|
2
|
+
import { readBody } from './middleware/body-reader.js';
|
|
3
|
+
import { decompressBody } from './middleware/decompress.js';
|
|
4
|
+
import { validateAuthHeader } from './auth.js';
|
|
5
|
+
import { validateCwd } from './cwd-validator.js';
|
|
6
|
+
import { isLoopbackAddress, isAllowedHostHeader } from '@zhixuan92/multi-model-agent-core';
|
|
7
|
+
import { resolveCallerIdentity } from './middleware/caller-identity.js';
|
|
8
|
+
const BODY_METHODS = new Set(['POST', 'PUT', 'PATCH', 'DELETE']);
|
|
9
|
+
// Daemon-wide draining flag — set true by the cleanupSignal handler when
|
|
10
|
+
// SIGTERM/SIGINT fires. New dispatches refuse with 503 service_unavailable
|
|
11
|
+
// while shutdown drains in-flight tasks. /health is allowed through so
|
|
12
|
+
// operators can confirm the daemon is winding down.
|
|
13
|
+
let drainingMode = false;
|
|
14
|
+
export function setDraining(d) { drainingMode = d; }
|
|
15
|
+
export function isDraining() { return drainingMode; }
|
|
16
|
+
export async function handleRequest(router, token, req, res, cfg, pipelineCfg) {
|
|
17
|
+
const method = req.method ?? 'GET';
|
|
18
|
+
const rawUrl = req.url ?? '/';
|
|
19
|
+
// ── Step 0: draining check ────────────────────────────────────────────────
|
|
20
|
+
// Once SIGTERM/SIGINT fires and cleanupSignal calls setDraining(true), new
|
|
21
|
+
// dispatches refuse with 503 so in-flight tasks can drain cleanly. /health
|
|
22
|
+
// stays available so operators can confirm the daemon is winding down.
|
|
23
|
+
if (drainingMode && !rawUrl.startsWith('/health')) {
|
|
24
|
+
sendError(res, 503, 'service_unavailable', 'daemon is draining; retry after restart');
|
|
25
|
+
return;
|
|
26
|
+
}
|
|
27
|
+
// ── Step 1: Body size cap ──────────────────────────────────────────────────
|
|
28
|
+
let rawBody;
|
|
29
|
+
if (BODY_METHODS.has(method)) {
|
|
30
|
+
const result = await readBody(req, cfg.server.limits.maxBodyBytes);
|
|
31
|
+
if (!result.ok) {
|
|
32
|
+
res.writeHead(413, { 'content-type': 'application/json', 'connection': 'close' });
|
|
33
|
+
res.end(JSON.stringify({ error: { code: 'payload_too_large', message: `Request body exceeds the ${cfg.server.limits.maxBodyBytes}-byte limit` } }), () => { req.socket?.destroy(); });
|
|
34
|
+
return;
|
|
35
|
+
}
|
|
36
|
+
rawBody = result.body;
|
|
37
|
+
}
|
|
38
|
+
// ── Step 2: Route match ────────────────────────────────────────────────────
|
|
39
|
+
const match = router.match(method, rawUrl);
|
|
40
|
+
if (!match) {
|
|
41
|
+
const allowed = router.methodsFor(rawUrl);
|
|
42
|
+
if (allowed.length > 0) {
|
|
43
|
+
sendError(res, 405, 'method_not_allowed', `Method ${method} not allowed`, { allowed });
|
|
44
|
+
}
|
|
45
|
+
else {
|
|
46
|
+
sendError(res, 404, 'not_found', `Unknown path ${rawUrl.split('?')[0]}`);
|
|
47
|
+
}
|
|
48
|
+
return;
|
|
49
|
+
}
|
|
50
|
+
// ── Step 3: Loopback & rebinding guard ─────────────────────────────────────
|
|
51
|
+
const pathname = rawUrl.split('?')[0];
|
|
52
|
+
// (a) Host-header rebinding check — ALL routes. Defends against DNS rebinding:
|
|
53
|
+
// the connection is loopback (IP check passes) but the Host header is an
|
|
54
|
+
// attacker-controlled domain. Only literal loopback host forms are allowed.
|
|
55
|
+
if (!isAllowedHostHeader(req.headers.host)) {
|
|
56
|
+
sendError(res, 403, 'forbidden_host', 'Request Host header is not an allowed loopback host');
|
|
57
|
+
return;
|
|
58
|
+
}
|
|
59
|
+
// (b) IP-level loopback check — loopbackOnlyPaths only.
|
|
60
|
+
if (pipelineCfg.loopbackOnlyPaths.has(pathname)) {
|
|
61
|
+
const remoteAddr = req.socket?.remoteAddress;
|
|
62
|
+
if (!isLoopbackAddress(remoteAddr)) {
|
|
63
|
+
sendError(res, 403, 'loopback_only', 'This endpoint is only accessible from the loopback interface');
|
|
64
|
+
return;
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
// ── Step 4: Bearer auth ────────────────────────────────────────────────────
|
|
68
|
+
if (!pipelineCfg.authExemptPaths.has(pathname)) {
|
|
69
|
+
const header = req.headers['authorization'];
|
|
70
|
+
const authResult = validateAuthHeader(header, token);
|
|
71
|
+
if (!authResult.ok) {
|
|
72
|
+
sendError(res, 401, 'unauthorized', 'Valid Bearer token required');
|
|
73
|
+
return;
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
// ── Step 5: Decompress body ─────────────────────────────────────────────────
|
|
77
|
+
if (rawBody !== undefined && rawBody.length > 0) {
|
|
78
|
+
const enc = req.headers['content-encoding'];
|
|
79
|
+
if (enc !== undefined) {
|
|
80
|
+
const result = await decompressBody(rawBody, enc, {
|
|
81
|
+
maxDecompressedBytes: cfg.server.limits.maxBodyBytes,
|
|
82
|
+
});
|
|
83
|
+
if (!result.ok) {
|
|
84
|
+
sendError(res, result.statusCode, result.reason, result.message);
|
|
85
|
+
return;
|
|
86
|
+
}
|
|
87
|
+
rawBody = result.body;
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
// ── Step 6: JSON body parse ────────────────────────────────────────────────
|
|
91
|
+
let parsedBody;
|
|
92
|
+
if (rawBody !== undefined && rawBody.length > 0) {
|
|
93
|
+
try {
|
|
94
|
+
parsedBody = JSON.parse(rawBody.toString('utf8'));
|
|
95
|
+
}
|
|
96
|
+
catch {
|
|
97
|
+
sendError(res, 400, 'invalid_json', 'Request body is not valid JSON');
|
|
98
|
+
return;
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
// ── Step 7: cwd query param validation ─────────────────────────────────────
|
|
102
|
+
let cwdValue;
|
|
103
|
+
const urlObj = new URL(rawUrl, 'http://localhost');
|
|
104
|
+
const requiresCwd = pipelineCfg.cwdRequiredPaths.has(pathname) ||
|
|
105
|
+
pathname === '/context-blocks' ||
|
|
106
|
+
/^\/context-blocks\//.test(pathname);
|
|
107
|
+
if (requiresCwd) {
|
|
108
|
+
const cwdParam = urlObj.searchParams.get('cwd') ?? undefined;
|
|
109
|
+
const cwdResult = validateCwd(cwdParam);
|
|
110
|
+
if (!cwdResult.ok) {
|
|
111
|
+
const statusCode = cwdResult.error === 'forbidden_cwd' ? 403 : 400;
|
|
112
|
+
sendError(res, statusCode, cwdResult.error, cwdResult.message);
|
|
113
|
+
return;
|
|
114
|
+
}
|
|
115
|
+
cwdValue = cwdResult.canonicalCwd;
|
|
116
|
+
}
|
|
117
|
+
// ── Step 8: Caller identity from headers ────────────────────────────────────
|
|
118
|
+
const identity = resolveCallerIdentity(req);
|
|
119
|
+
if (pipelineCfg.mainModelRequiredPaths.has(pathname)) {
|
|
120
|
+
if (identity.callerClient === 'other') {
|
|
121
|
+
sendError(res, 400, 'client_required', 'X-MMA-Client header is required on tool routes. Set it to one of: claude-code, cursor, codex-cli, gemini-cli.');
|
|
122
|
+
return;
|
|
123
|
+
}
|
|
124
|
+
// Auto-detect was unreliable: the claude-agent-sdk used by our own
|
|
125
|
+
// claude-tier workers writes JSONL files into the same project slug,
|
|
126
|
+
// and the resolver would pick those up (e.g. haiku) as the "main"
|
|
127
|
+
// model. The header must come from the calling client.
|
|
128
|
+
if (!identity.mainModel) {
|
|
129
|
+
sendError(res, 400, 'main_model_required', 'X-MMA-Main-Model header is required on tool routes. Set it to the calling agent\'s model id (e.g. claude-opus-4-7, gpt-5.4).');
|
|
130
|
+
return;
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
// ── Step 9: Hand off to matched handler ────────────────────────────────────
|
|
134
|
+
const ctx = {
|
|
135
|
+
url: urlObj,
|
|
136
|
+
cwd: cwdValue,
|
|
137
|
+
body: parsedBody,
|
|
138
|
+
authed: !pipelineCfg.authExemptPaths.has(pathname),
|
|
139
|
+
callerClient: identity.callerClient,
|
|
140
|
+
mainModel: identity.mainModel,
|
|
141
|
+
};
|
|
142
|
+
await match.handler(req, res, match.params, ctx);
|
|
143
|
+
}
|
|
144
|
+
//# sourceMappingURL=request-pipeline.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"request-pipeline.js","sourceRoot":"","sources":["../../src/http/request-pipeline.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,mCAAmC,CAAC;AAC3F,OAAO,EAAE,qBAAqB,EAAE,MAAM,iCAAiC,CAAC;AAGxE,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;AAEjE,yEAAyE;AACzE,2EAA2E;AAC3E,uEAAuE;AACvE,oDAAoD;AACpD,IAAI,YAAY,GAAG,KAAK,CAAC;AACzB,MAAM,UAAU,WAAW,CAAC,CAAU,IAAU,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC;AACnE,MAAM,UAAU,UAAU,KAAc,OAAO,YAAY,CAAC,CAAC,CAAC;AAW9D,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAmC,EACnC,KAAa,EACb,GAAoB,EACpB,GAAmB,EACnB,GAAiB,EACjB,WAA2B;IAE3B,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,KAAK,CAAC;IACnC,MAAM,MAAM,GAAG,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC;IAE9B,6EAA6E;IAC7E,2EAA2E;IAC3E,2EAA2E;IAC3E,uEAAuE;IACvE,IAAI,YAAY,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAClD,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,qBAAqB,EAAE,yCAAyC,CAAC,CAAC;QACtF,OAAO;IACT,CAAC;IAED,8EAA8E;IAC9E,IAAI,OAA2B,CAAC;IAChC,IAAI,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;YACf,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;YAClF,GAAG,CAAC,GAAG,CACL,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,4BAA4B,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,aAAa,EAAE,EAAE,CAAC,EAC1I,GAAG,EAAE,GAAG,GAAG,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CACjC,CAAC;YACF,OAAO;QACT,CAAC;QACD,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC;IACxB,CAAC;IAED,8EAA8E;IAC9E,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3C,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,oBAAoB,EAAE,UAAU,MAAM,cAAc,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACzF,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,WAAW,EAAE,gBAAgB,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QACD,OAAO;IACT,CAAC;IAED,8EAA8E;IAC9E,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC;IACvC,+EAA+E;IAC/E,yEAAyE;IACzE,4EAA4E;IAC5E,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3C,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,gBAAgB,EAAE,qDAAqD,CAAC,CAAC;QAC7F,OAAO;IACT,CAAC;IACD,wDAAwD;IACxD,IAAI,WAAW,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChD,MAAM,UAAU,GAAG,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC;QAC7C,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE,CAAC;YACnC,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,eAAe,EAAE,8DAA8D,CAAC,CAAC;YACrG,OAAO;QACT,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,IAAI,CAAC,WAAW,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC/C,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAC5C,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QACrD,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC;YACnB,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,cAAc,EAAE,6BAA6B,CAAC,CAAC;YACnE,OAAO;QACT,CAAC;IACH,CAAC;IAED,+EAA+E;IAC/E,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,MAAM,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QAC5C,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,GAAG,EAAE;gBAChD,oBAAoB,EAAE,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY;aACrD,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;gBACf,SAAS,CAAC,GAAG,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;gBACjE,OAAO;YACT,CAAC;YACD,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,IAAI,UAAmB,CAAC;IACxB,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,IAAI,CAAC;YACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QACpD,CAAC;QAAC,MAAM,CAAC;YACP,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,cAAc,EAAE,gCAAgC,CAAC,CAAC;YACtE,OAAO;QACT,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,IAAI,QAA4B,CAAC;IACjC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;IACnD,MAAM,WAAW,GAAG,WAAW,CAAC,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC;QAC5D,QAAQ,KAAK,iBAAiB;QAC9B,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAEvC,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,QAAQ,GAAG,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,SAAS,CAAC;QAC7D,MAAM,SAAS,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC;YAClB,MAAM,UAAU,GAAG,SAAS,CAAC,KAAK,KAAK,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;YACnE,SAAS,CAAC,GAAG,EAAE,UAAU,EAAE,SAAS,CAAC,KAAK,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;YAC/D,OAAO;QACT,CAAC;QACD,QAAQ,GAAG,SAAS,CAAC,YAAY,CAAC;IACpC,CAAC;IAED,+EAA+E;IAC/E,MAAM,QAAQ,GAAG,qBAAqB,CAAC,GAAG,CAAC,CAAC;IAE5C,IAAI,WAAW,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;QACrD,IAAI,QAAQ,CAAC,YAAY,KAAK,OAAO,EAAE,CAAC;YACtC,SAAS,CACP,GAAG,EACH,GAAG,EACH,iBAAiB,EACjB,+GAA+G,CAChH,CAAC;YACF,OAAO;QACT,CAAC;QACD,mEAAmE;QACnE,qEAAqE;QACrE,kEAAkE;QAClE,uDAAuD;QACvD,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YACxB,SAAS,CACP,GAAG,EACH,GAAG,EACH,qBAAqB,EACrB,8HAA8H,CAC/H,CAAC;YACF,OAAO;QACT,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,MAAM,GAAG,GAAmB;QAC1B,GAAG,EAAE,MAAM;QACX,GAAG,EAAE,QAAQ;QACb,IAAI,EAAE,UAAU;QAChB,MAAM,EAAE,CAAC,WAAW,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC;QAClD,YAAY,EAAE,QAAQ,CAAC,YAAY;QACnC,SAAS,EAAE,QAAQ,CAAC,SAAS;KAC9B,CAAC;IAEF,MAAM,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AACnD,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import type { ServerConfig, BatchRegistry } from '@zhixuan92/multi-model-agent-core';
|
|
2
|
+
import type { ProjectRegistry } from './project-registry.js';
|
|
3
|
+
export declare const SERVER_VERSION: string;
|
|
4
|
+
export interface RunningServer {
|
|
5
|
+
port: number;
|
|
6
|
+
/** The resolved address the server is bound to. Used by CLI to log the actual listen address. */
|
|
7
|
+
serverAddress: string | null;
|
|
8
|
+
stop(): Promise<void>;
|
|
9
|
+
/** Shared BatchRegistry — exposed for testing and introspection handlers. */
|
|
10
|
+
batchRegistry: BatchRegistry;
|
|
11
|
+
/** Shared ProjectRegistry — exposed for testing and introspection handlers. */
|
|
12
|
+
projectRegistry: ProjectRegistry;
|
|
13
|
+
/** Wall-clock ms when the server finished starting (Date.now()). Used for uptimeMs in /status. */
|
|
14
|
+
serverStartedAt: number;
|
|
15
|
+
}
|
|
16
|
+
export declare function startServer(config: ServerConfig, injectedManifestSync?: import('../skill-install/skill-manifest-sync.js').SkillManifestSync): Promise<RunningServer>;
|
|
17
|
+
//# sourceMappingURL=server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/http/server.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAWrF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAiB7D,eAAO,MAAM,cAAc,QAAsB,CAAC;AAElD,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,CAAC;IACb,iGAAiG;IACjG,aAAa,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,6EAA6E;IAC7E,aAAa,EAAE,aAAa,CAAC;IAC7B,+EAA+E;IAC/E,eAAe,EAAE,eAAe,CAAC;IACjC,kGAAkG;IAClG,eAAe,EAAE,MAAM,CAAC;CACzB;AAoND,wBAAsB,WAAW,CAC/B,MAAM,EAAE,YAAY,EACpB,oBAAoB,CAAC,EAAE,OAAO,yCAAyC,EAAE,iBAAiB,GACzF,OAAO,CAAC,aAAa,CAAC,CA+ExB"}
|