mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-14 02:42:24 +03:00
65e81158ab
Landed with the design call resolved per the owner's pick — **option 1**: the synced store is now endpoint-agnostic (persistDiscoveredModels and managedModelImport no longer drop non-chat models at write time), and chat selectability moved to read time (auto-pool expansion in autoStrategy applies filterChatSelectableModels; the models-route projection already had its chatOnly filter). Your discovery test now passes end-to-end (3/3): /api/show capabilities persist per connection and image/embedding requests route through the advertising host. Reconciliation notes: conflicted areas merged onto the current tip (adobe discovery import, requestedModel preflight signature, resolvedProvider fast-path coexists with the synced-route override — explicit resolution wins); carried base-red drains (#10055 memoization, #11071 test variants) dropped as already-landed; the managed-model-import exclusion test was propagated to the new contract (image/video models persist; the read filter still hides them from chat pickers — pinned by a new assertion). Full battery: 205/206 focused (the one red is a confirmed periodic-timer timing flake on the loaded devbox — 20/20 isolated), autoCombo vitest 30/30, combo suites 46/46, gates + typecheck clean. Thank you @yourspraveen — the capability probe + routing design was right; it just needed the store contract opened up. Fixes #11087.
80 lines
2.9 KiB
JavaScript
80 lines
2.9 KiB
JavaScript
#!/usr/bin/env node
|
|
|
|
import { spawn } from "node:child_process";
|
|
import { existsSync } from "node:fs";
|
|
import { dirname, join } from "node:path";
|
|
import { fileURLToPath, pathToFileURL } from "node:url";
|
|
|
|
const __filename = fileURLToPath(import.meta.url);
|
|
const __dirname = dirname(__filename);
|
|
const ROOT = join(__dirname, "..");
|
|
|
|
/**
|
|
* @param {string} [rootDir] - project root directory (defaults to package root)
|
|
* @param {(path: string) => boolean} [existsSyncFn] - injectable fs.existsSync
|
|
* for testing; defaults to the real existsSync
|
|
* @returns {string|null} resolved absolute path to the MCP server entry, or null
|
|
*/
|
|
export function resolveMcpEntry(rootDir = ROOT, existsSyncFn = existsSync) {
|
|
const candidates = [
|
|
// Preferred distributable JS entry (npm publish artifact, built by prepublish.ts)
|
|
join(rootDir, "dist", "open-sse", "mcp-server", "server.js"),
|
|
// Local workspace TypeScript source fallback
|
|
join(rootDir, "open-sse", "mcp-server", "server.ts"),
|
|
];
|
|
|
|
for (const entry of candidates) {
|
|
if (existsSyncFn(entry)) return entry;
|
|
}
|
|
return null;
|
|
}
|
|
|
|
function formatSpawnError(exitCode, signal) {
|
|
if (signal) return `MCP server exited by signal ${signal}`;
|
|
return `MCP server exited with code ${exitCode ?? 1}`;
|
|
}
|
|
|
|
export async function startMcpCli(rootDir = ROOT) {
|
|
const mcpEntry = resolveMcpEntry(rootDir);
|
|
if (!mcpEntry) {
|
|
throw new Error(
|
|
"MCP server entrypoint not found. Expected dist/open-sse/mcp-server/server.js or open-sse/mcp-server/server.ts."
|
|
);
|
|
}
|
|
|
|
// `tsx` loader is only required for local `.ts` fallback; JS entry works without it.
|
|
const tsxLoaderArgs = mcpEntry.endsWith(".ts") ? ["--import", "tsx"] : [];
|
|
// Preload the stdout/stderr console guard before mcpEntry's own module graph evaluates —
|
|
// DB init (a side effect of createMcpServer()'s tool registration) logs via plain
|
|
// console.log, and by the time any code inside mcpEntry itself could redirect it, that
|
|
// module's own (hoisted) imports have already run. Loading the guard first, in a separate
|
|
// module, is the only point early enough to guarantee it never leaks into the JSON-RPC
|
|
// stream on stdout.
|
|
const consoleGuard = pathToFileURL(join(__dirname, "mcpStdioConsoleGuard.mjs")).href;
|
|
const loaderArgs = ["--import", consoleGuard, ...tsxLoaderArgs];
|
|
|
|
await new Promise((resolve, reject) => {
|
|
const child = spawn(process.execPath, [...loaderArgs, mcpEntry], {
|
|
cwd: rootDir,
|
|
env: process.env,
|
|
stdio: "inherit",
|
|
});
|
|
|
|
child.once("error", reject);
|
|
child.once("exit", (code, signal) => {
|
|
if ((code ?? 0) === 0 && !signal) {
|
|
resolve(undefined);
|
|
return;
|
|
}
|
|
reject(new Error(formatSpawnError(code, signal)));
|
|
});
|
|
});
|
|
}
|
|
|
|
if (process.argv[1] && fileURLToPath(import.meta.url) === process.argv[1]) {
|
|
startMcpCli().catch((err) => {
|
|
console.error("\x1b[31m✖ Failed to start MCP server:\x1b[0m", err?.message || err);
|
|
process.exit(1);
|
|
});
|
|
}
|