mirror of
https://github.com/Z3Prover/z3
synced 2025-06-23 14:23:40 +00:00
stab at SMTLIB REL mcp server
This commit is contained in:
parent
755f57931b
commit
3f73c8b18f
1 changed files with 62 additions and 0 deletions
62
src/api/mcp/z3mcp.ts
Normal file
62
src/api/mcp/z3mcp.ts
Normal file
|
@ -0,0 +1,62 @@
|
||||||
|
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||||||
|
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
|
||||||
|
|
||||||
|
async function importZ3() {
|
||||||
|
try {
|
||||||
|
const z3 = await import("z3-solver");
|
||||||
|
return await z3.init();
|
||||||
|
} catch (e) {
|
||||||
|
console.error("Failed to import z3-solver:", e?.message);
|
||||||
|
return undefined;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function Z3Run(input) {
|
||||||
|
const z3p = await importZ3();
|
||||||
|
if (!z3p) {
|
||||||
|
return "Z3 not available. Make sure to install the https://www.npmjs.com/package/z3-solver package.";
|
||||||
|
}
|
||||||
|
const { Z3 } = z3p;
|
||||||
|
const cfg = Z3.mk_config();
|
||||||
|
const ctx = Z3.mk_context(cfg);
|
||||||
|
Z3.del_config(cfg);
|
||||||
|
const timeout = 10000;
|
||||||
|
Z3.global_param_set("timeout", String(timeout));
|
||||||
|
let output = "";
|
||||||
|
let error = "";
|
||||||
|
const timeStart = Date.now();
|
||||||
|
try {
|
||||||
|
output = (await Z3.eval_smtlib2_string(ctx, input)) ?? "";
|
||||||
|
} catch (e) {
|
||||||
|
error = e.message ?? "Error message is empty";
|
||||||
|
} finally {
|
||||||
|
Z3.del_context(ctx);
|
||||||
|
}
|
||||||
|
if (/unknown/.test(output)) {
|
||||||
|
const timeEnd = Date.now();
|
||||||
|
if (timeEnd - timeStart >= timeout) {
|
||||||
|
output = output + "\nZ3 timeout\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (!error) return output;
|
||||||
|
else return `error: ${error}\n\n${output || ""}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
const server = new McpServer({
|
||||||
|
name: "z3",
|
||||||
|
version: "1.0.0"
|
||||||
|
});
|
||||||
|
|
||||||
|
server.tool(
|
||||||
|
"eval",
|
||||||
|
{ command: { type: "string", description: "smtlib2 command" } },
|
||||||
|
async ({ command }) => {
|
||||||
|
const result = await Z3Run(command);
|
||||||
|
return { content: [{ type: "text", text: result }] };
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
const transport = new StdioServerTransport();
|
||||||
|
await server.connect(transport);
|
Loading…
Add table
Add a link
Reference in a new issue