pudu-ai 0.2.21 → 0.2.23

package/dist/cli/{render-3UBPZNF4.js → render-Y4DNH7FW.js}

@@ -16,21 +16,28 @@ import {
   resultText,
   runBenchmark,
   tasksText
-} from "./chunk-6C5EPKRI.js";
+} from "./chunk-6BGYI7XY.js";
+import "./chunk-47LK5I2X.js";
+import {
+  buildRepoGraph
+} from "./chunk-XEEGLMVT.js";
 import {
   formatNumber,
   formatPercent,
-  formatTokensPerSec,
+  formatTokensPerSec
+} from "./chunk-473MNTMS.js";
+import "./chunk-7KPNGK6A.js";
+import {
   t
-} from "./chunk-QGEXKMQ5.js";
+} from "./chunk-4VEDGI4L.js";
 import "./chunk-SS7OXXQQ.js";
 
 // src/tui/render.tsx
 import { render } from "ink";
 
 // src/tui/App.tsx
-import { useApp as useApp2, useInput as useInput4 } from "ink";
-import { useMemo as useMemo2, useRef as useRef2, useState as useState4 } from "react";
+import { useApp as useApp2, useInput as useInput5 } from "ink";
+import { useMemo as useMemo3, useRef as useRef2, useState as useState5 } from "react";
 
 // src/tui/views/Dashboard.tsx
 import { Box as Box2, Text as Text2 } from "ink";
@@ -84,6 +91,7 @@ var NAV = [
   { key: "M", label: "Models", color: "cyan" },
   { key: "R", label: "Rec", color: "magenta" },
   { key: "T", label: "Tasks", color: "blue" },
+  { key: "G", label: "Graph", color: "magenta" },
   { key: "H", label: "HW", color: "white" },
   { key: "C", label: "Cmp", color: "green" },
   { key: "L", label: "Log", color: "cyan" },
@@ -729,6 +737,221 @@ function TasksView({ session }) {
   ] });
 }
 
+// src/tui/views/Graph.tsx
+import { Box as Box10, Text as Text11, useInput as useInput4 } from "ink";
+import { useEffect as useEffect2, useMemo as useMemo2, useState as useState4 } from "react";
+
+// src/agent-lab/harness.ts
+var GRADE_RANK = { S: 0, A: 1, B: 2, C: 3, D: 4, F: 5 };
+function isCodeish(useCase) {
+  const cases = useCase ?? ["chat", "code"];
+  return cases.some((item) => item.includes("code") || item.includes("chat"));
+}
+function pickHarnessModel(session, effort) {
+  const rows = session.rows.filter((row) => isCodeish(row.catalog?.useCase));
+  const pool = rows.length ? rows : session.rows;
+  if (!pool.length) {
+    return {
+      modelId: null,
+      modelName: null,
+      ollamaTag: null,
+      grade: null,
+      measuredTps: null,
+      reason: "No installed models. N/A",
+      origin: null
+    };
+  }
+  const scored = pool.map((row) => ({
+    row,
+    grade: row.compatibility?.grade ?? "C",
+    tps: row.lastBenchmark?.benchmark.generationTokensPerSecond ?? null,
+    params: row.catalog?.paramsBillions ?? null
+  }));
+  scored.sort((a, b) => {
+    if (effort.label === "LOW") {
+      if (a.tps != null && b.tps != null && a.tps !== b.tps) return b.tps - a.tps;
+      if (a.tps != null && b.tps == null) return -1;
+      if (a.tps == null && b.tps != null) return 1;
+      return (GRADE_RANK[a.grade] ?? 9) - (GRADE_RANK[b.grade] ?? 9);
+    }
+    if (effort.label === "HIGH" || effort.label === "VERY_HIGH") {
+      if (a.params != null && b.params != null && a.params !== b.params) return b.params - a.params;
+      if (a.params != null && b.params == null) return -1;
+      if (a.params == null && b.params != null) return 1;
+      return (GRADE_RANK[a.grade] ?? 9) - (GRADE_RANK[b.grade] ?? 9);
+    }
+    const grade = (GRADE_RANK[a.grade] ?? 9) - (GRADE_RANK[b.grade] ?? 9);
+    if (grade !== 0) return grade;
+    if (a.tps != null && b.tps != null) return b.tps - a.tps;
+    return 0;
+  });
+  const best = scored[0];
+  const reasons = {
+    LOW: "LOW effort: prefer fastest measured local coding model; else best grade.",
+    MEDIUM: "MEDIUM effort: prefer best hardware grade, then measured t/s.",
+    HIGH: "HIGH effort: prefer larger catalog params when known; else best grade.",
+    VERY_HIGH: "VERY_HIGH effort: prefer larger catalog params when known; else best grade."
+  };
+  return {
+    modelId: best.row.local.id,
+    modelName: best.row.local.name,
+    ollamaTag: resolveOllamaTag(best.row.local.id, best.row.local.name, best.row.catalog?.id, best.row.catalog?.name) ?? (best.row.local.id.includes(":") ? best.row.local.id : null),
+    grade: best.grade,
+    measuredTps: best.tps,
+    reason: reasons[effort.label],
+    origin: "DERIVED"
+  };
+}
+
+// src/tui/views/Graph.tsx
+import { Fragment as Fragment6, jsx as jsx11, jsxs as jsxs11 } from "react/jsx-runtime";
+var TOOLS2 = [
+  { key: "1", id: "opencode", label: "OpenCode" },
+  { key: "2", id: "hermes", label: "Hermes" },
+  { key: "3", id: "openclaw", label: "OpenClaw" }
+];
+function GraphView(props) {
+  const [phase, setPhase] = useState4("build");
+  const [graph, setGraph] = useState4();
+  const [error, setError] = useState4("");
+  const [prompt, setPrompt] = useState4("");
+  const repo = process.cwd();
+  useEffect2(() => {
+    let live = true;
+    void (async () => {
+      try {
+        const result = await buildRepoGraph(repo);
+        if (!live) return;
+        setGraph(result);
+        setPhase("prompt");
+      } catch (err) {
+        if (!live) return;
+        setError(err instanceof Error ? err.message : String(err));
+        setPhase("error");
+      }
+    })();
+    return () => {
+      live = false;
+    };
+  }, [repo]);
+  const pick = useMemo2(() => graph ? pickHarnessModel(props.session, graph.effort) : void 0, [graph, props.session]);
+  useInput4((input, key) => {
+    if (phase === "prompt") {
+      if (key.return) {
+        setPhase("pick");
+        return;
+      }
+      if (key.backspace || key.delete) {
+        setPrompt((value) => value.slice(0, -1));
+        return;
+      }
+      if (input && !key.ctrl && !key.meta && input !== "	") {
+        setPrompt((value) => `${value}${input}`.slice(0, 240));
+      }
+      return;
+    }
+    if (phase !== "pick" || !pick?.modelId) return;
+    const tool = TOOLS2.find((item) => item.key === input)?.id ?? (key.return ? "opencode" : void 0);
+    if (!tool) return;
+    if (!pick.ollamaTag) {
+      setError(t("repoHarnessNoTag"));
+      setPhase("error");
+      return;
+    }
+    props.onLaunch({
+      tool,
+      repo,
+      task: prompt,
+      modelId: pick.modelId,
+      ollamaTag: pick.ollamaTag
+    });
+  });
+  if (phase === "build") {
+    return /* @__PURE__ */ jsxs11(Panel, { title: t("repoGraphTitle"), color: "magenta", children: [
+      /* @__PURE__ */ jsx11(Text11, { children: t("repoGraphBuilding") }),
+      /* @__PURE__ */ jsx11(Text11, { dimColor: true, children: repo })
+    ] });
+  }
+  if (phase === "error") {
+    return /* @__PURE__ */ jsx11(Panel, { title: t("repoGraphTitle"), color: "magenta", children: /* @__PURE__ */ jsx11(Text11, { color: "red", children: error }) });
+  }
+  if (!graph) return /* @__PURE__ */ jsx11(Text11, { children: " " });
+  const degree = /* @__PURE__ */ new Map();
+  for (const edge of graph.edges) {
+    degree.set(edge.source, (degree.get(edge.source) ?? 0) + 1);
+  }
+  const top = [...graph.nodes].filter((node) => node.type === "file" || node.type === "function" || node.type === "test").sort((a, b) => (degree.get(b.id) ?? 0) - (degree.get(a.id) ?? 0)).slice(0, 8);
+  return /* @__PURE__ */ jsxs11(Fragment6, { children: [
+    /* @__PURE__ */ jsxs11(Panel, { title: t("repoGraphTitle"), color: "magenta", children: [
+      /* @__PURE__ */ jsx11(Text11, { dimColor: true, children: fit(graph.repo, 80) }),
+      /* @__PURE__ */ jsxs11(Text11, { children: [
+        t("repoGraphFiles"),
+        " ",
+        graph.metrics.fileCount,
+        "  ",
+        t("repoGraphNodes"),
+        " ",
+        graph.metrics.nodeCount,
+        "  ",
+        t("repoGraphEdges"),
+        " ",
+        graph.metrics.edgeCount,
+        "  (",
+        graph.metrics.origin,
+        ")"
+      ] }),
+      top.map((node) => /* @__PURE__ */ jsxs11(Box10, { children: [
+        /* @__PURE__ */ jsx11(Text11, { color: "cyan", children: fit(node.file ?? node.name, 36) }),
+        /* @__PURE__ */ jsxs11(Text11, { children: [
+          "  ",
+          fit(node.name, 18),
+          "  "
+        ] }),
+        /* @__PURE__ */ jsx11(Text11, { dimColor: true, children: degree.get(node.id) ?? 0 })
+      ] }, node.id)),
+      !top.length ? /* @__PURE__ */ jsx11(Text11, { dimColor: true, children: t("repoGraphEmpty") }) : null
+    ] }),
+    /* @__PURE__ */ jsxs11(Panel, { title: t("repoEffortTitle"), color: "yellow", children: [
+      /* @__PURE__ */ jsxs11(Text11, { children: [
+        graph.effort.label,
+        "  ",
+        graph.effort.score0to100,
+        "/100  (",
+        graph.effort.origin,
+        ")"
+      ] }),
+      /* @__PURE__ */ jsxs11(Text11, { dimColor: true, children: [
+        t("repoTokens"),
+        " ",
+        graph.metrics.tokenEstimate ?? "N/A",
+        " (",
+        graph.metrics.tokenOrigin ?? "N/A",
+        ")"
+      ] })
+    ] }),
+    phase === "prompt" ? /* @__PURE__ */ jsxs11(Panel, { title: t("repoHarnessPrompt"), color: "blue", children: [
+      /* @__PURE__ */ jsxs11(Text11, { children: [
+        prompt || t("repoHarnessPromptHint"),
+        /* @__PURE__ */ jsx11(Text11, { color: "cyan", children: "\u258D" })
+      ] }),
+      /* @__PURE__ */ jsx11(Text11, { dimColor: true, children: t("repoHarnessEnter") })
+    ] }) : /* @__PURE__ */ jsxs11(Panel, { title: t("repoHarnessModel"), color: "green", children: [
+      /* @__PURE__ */ jsx11(Text11, { dimColor: true, children: prompt || t("repoHarnessNoPrompt") }),
+      pick?.modelName ? /* @__PURE__ */ jsxs11(Text11, { children: [
+        pick.modelName,
+        "  ",
+        pick.ollamaTag ?? "\u2014",
+        "  ",
+        pick.grade ?? "\u2014",
+        "  ",
+        pick.measuredTps != null ? `${pick.measuredTps} t/s` : t("notTested")
+      ] }) : /* @__PURE__ */ jsx11(Text11, { color: "yellow", children: pick?.reason ?? "N/A" }),
+      /* @__PURE__ */ jsx11(Text11, { dimColor: true, children: pick?.reason }),
+      pick?.ollamaTag ? /* @__PURE__ */ jsx11(Text11, { color: "cyan", children: t("repoHarnessKeys") }) : /* @__PURE__ */ jsx11(Text11, { color: "yellow", children: t("repoHarnessNoTag") })
+    ] })
+  ] });
+}
+
 // src/tui/keys.ts
 function globalNav(letter) {
   if (letter === "b") return { type: "screen", screen: "benchmark" };
@@ -739,11 +962,16 @@ function globalNav(letter) {
   if (letter === "l") return { type: "screen", screen: "history" };
   if (letter === "t") return { type: "screen", screen: "tasks" };
   if (letter === "s") return { type: "screen", screen: "setup" };
+  if (letter === "g") return { type: "screen", screen: "graph" };
   return void 0;
 }
 function handleAppKey(screen, input, key) {
   const letter = input.toLowerCase();
   if (screen === "benchmark") return { type: "ignore" };
+  if (screen === "graph") {
+    if (key.escape) return { type: "home" };
+    return { type: "ignore" };
+  }
   if (letter === "q") return { type: "quit" };
   if (key.escape) return screen === "home" ? { type: "quit" } : { type: "home" };
   if (screen === "setup") {
@@ -757,34 +985,44 @@ function handleAppKey(screen, input, key) {
 }
 
 // src/tui/App.tsx
-import { jsx as jsx11, jsxs as jsxs11 } from "react/jsx-runtime";
+import { jsx as jsx12, jsxs as jsxs12 } from "react/jsx-runtime";
 function App(props) {
   const { exit } = useApp2();
-  const [screen, setScreen] = useState4(props.start ?? "home");
+  const [screen, setScreen] = useState5(props.start ?? "home");
   const screenRef = useRef2(screen);
   screenRef.current = screen;
-  const [selected, setSelected] = useState4(0);
-  const benchable = useMemo2(
+  const [selected, setSelected] = useState5(0);
+  const benchable = useMemo3(
     () => props.session.models.filter((m) => Boolean(m.artifactPath)),
     [props.session.models]
   );
-  useInput4((input, key) => {
+  useInput5((input, key) => {
     const result = handleAppKey(screenRef.current, input, key);
     if (result.type === "quit") exit();
     if (result.type === "home") setScreen("home");
     if (result.type === "screen") setScreen(result.screen);
   });
-  return /* @__PURE__ */ jsxs11(Frame, { children: [
-    screen !== "benchmark" && /* @__PURE__ */ jsx11(Header, {}),
-    screen === "home" && /* @__PURE__ */ jsx11(Dashboard, { session: props.session }),
-    screen === "models" && /* @__PURE__ */ jsx11(ModelsView, { session: props.session }),
-    screen === "hardware" && /* @__PURE__ */ jsx11(HardwareView, { session: props.session }),
-    screen === "recommend" && /* @__PURE__ */ jsx11(RecommendView, { session: props.session }),
-    screen === "setup" && /* @__PURE__ */ jsx11(SetupView, { session: props.session }),
-    screen === "compare" && /* @__PURE__ */ jsx11(CompareView, { session: props.session }),
-    screen === "history" && /* @__PURE__ */ jsx11(HistoryView, { session: props.session }),
-    screen === "tasks" && /* @__PURE__ */ jsx11(TasksView, { session: props.session }),
-    screen === "benchmark" && /* @__PURE__ */ jsx11(
+  return /* @__PURE__ */ jsxs12(Frame, { children: [
+    screen !== "benchmark" && /* @__PURE__ */ jsx12(Header, {}),
+    screen === "home" && /* @__PURE__ */ jsx12(Dashboard, { session: props.session }),
+    screen === "models" && /* @__PURE__ */ jsx12(ModelsView, { session: props.session }),
+    screen === "hardware" && /* @__PURE__ */ jsx12(HardwareView, { session: props.session }),
+    screen === "recommend" && /* @__PURE__ */ jsx12(RecommendView, { session: props.session }),
+    screen === "setup" && /* @__PURE__ */ jsx12(SetupView, { session: props.session }),
+    screen === "compare" && /* @__PURE__ */ jsx12(CompareView, { session: props.session }),
+    screen === "history" && /* @__PURE__ */ jsx12(HistoryView, { session: props.session }),
+    screen === "tasks" && /* @__PURE__ */ jsx12(TasksView, { session: props.session }),
+    screen === "graph" && /* @__PURE__ */ jsx12(
+      GraphView,
+      {
+        session: props.session,
+        onLaunch: (req) => {
+          props.onLaunch?.(req);
+          exit();
+        }
+      }
+    ),
+    screen === "benchmark" && /* @__PURE__ */ jsx12(
       BenchmarkView,
       {
         session: props.session,
@@ -799,13 +1037,28 @@ function App(props) {
 }
 
 // src/tui/render.tsx
-import { jsx as jsx12 } from "react/jsx-runtime";
+import { jsx as jsx13 } from "react/jsx-runtime";
 async function renderDashboard(session, preset, start) {
-  const instance = render(/* @__PURE__ */ jsx12(App, { session, preset, start }), {
-    exitOnCtrlC: true,
-    patchConsole: true
-  });
+  let pending;
+  const instance = render(
+    /* @__PURE__ */ jsx13(
+      App,
+      {
+        session,
+        preset,
+        start,
+        onLaunch: (req) => {
+          pending = req;
+        }
+      }
+    ),
+    {
+      exitOnCtrlC: true,
+      patchConsole: true
+    }
+  );
   await instance.waitUntilExit();
+  return pending;
 }
 export {
   renderDashboard

package/dist/cli/{repo-cli-4GSG5MD3.js → repo-cli-BK76EVBZ.js}

@@ -1,64 +1,22 @@
 import {
-  na,
-  t
-} from "./chunk-QGEXKMQ5.js";
+  buildRepoGraph,
+  runRepoHarness
+} from "./chunk-XEEGLMVT.js";
+import {
+  na
+} from "./chunk-473MNTMS.js";
+import {
+  findPython,
+  invokePuduAgent,
+  parseJsonStdout
+} from "./chunk-7KPNGK6A.js";
 import {
-  commandExists,
-  runCommand
-} from "./chunk-SS7OXXQQ.js";
+  t
+} from "./chunk-4VEDGI4L.js";
+import "./chunk-SS7OXXQQ.js";
 
 // src/agent-lab/repo-cli.ts
-import path2 from "node:path";
-
-// src/agent-lab/python-bridge.ts
-import { existsSync } from "node:fs";
 import path from "node:path";
-import { fileURLToPath } from "node:url";
-function hasEngine(dir) {
-  return existsSync(path.join(dir, "pudu_agent", "__main__.py"));
-}
-function pythonRoot() {
-  const starts = [path.dirname(fileURLToPath(import.meta.url)), process.cwd()];
-  for (const start of starts) {
-    let dir = start;
-    for (let i = 0; i < 8; i++) {
-      const candidate = path.join(dir, "python");
-      if (hasEngine(candidate)) return candidate;
-      const parent = path.dirname(dir);
-      if (parent === dir) break;
-      dir = parent;
-    }
-  }
-  return path.resolve(path.dirname(fileURLToPath(import.meta.url)), "..", "..", "python");
-}
-async function findPython() {
-  return await commandExists("python3") ?? await commandExists("python");
-}
-async function invokePuduAgent(request, timeout = 3e4) {
-  const python = await findPython();
-  if (!python) {
-    throw new Error("python3 not found on PATH");
-  }
-  const root = pythonRoot();
-  const payload = `${JSON.stringify(request)}
-`;
-  const result = await runCommand(python, ["-m", "pudu_agent"], {
-    cwd: root,
-    env: { ...process.env, PYTHONPATH: root, PYTHONUNBUFFERED: "1" },
-    input: payload,
-    timeout
-  });
-  return { ...result, python };
-}
-function parseJsonStdout(stdout) {
-  const text = stdout.trim();
-  if (!text) {
-    throw new Error("Python engine returned empty stdout");
-  }
-  const start = text.indexOf("{");
-  const slice = start >= 0 ? text.slice(start) : text;
-  return JSON.parse(slice);
-}
 
 // src/agent-lab/schema.ts
 import { z } from "zod";
@@ -83,7 +41,7 @@ var searchMatchSchema = z.object({
   endLine: z.number().nullable(),
   endColumn: z.number().nullable(),
   text: z.string(),
-  strategy: z.enum(["rg", "ast-grep"]),
+  strategy: z.enum(["rg", "ast-grep", "graph"]),
   language: z.string().nullable(),
   metavariables: z.record(z.string())
 });
@@ -205,16 +163,41 @@ async function runRepoCommand(args) {
     print(t("repoHelp"), args.json);
     return 0;
   }
+  const python = await findPython();
+  if (!python) {
+    process.stderr.write(`${t("repoPythonMissing")}
+`);
+    return 1;
+  }
+  const repo = path.resolve(args.repo ?? process.cwd());
+  if (sub === "graph") {
+    try {
+      const graph = await buildRepoGraph(repo);
+      print(args.json ? graph : formatGraphText(graph), args.json);
+      return graph.ok ? 0 : 1;
+    } catch (error) {
+      return fail(error, args.json);
+    }
+  }
+  if (sub === "harness") {
+    const task = args.task ?? args.positional.slice(1).join(" ").trim() ?? "";
+    if (!task) {
+      process.stderr.write(`${t("repoNeedTask")}
+`);
+      return 1;
+    }
+    try {
+      const harness = await runRepoHarness(repo, task);
+      print(args.json ? harness : formatGraphText(harness.graph, task), args.json);
+      return harness.ok ? 0 : 1;
+    } catch (error) {
+      return fail(error, args.json);
+    }
+  }
   if (sub !== "search") {
     process.stderr.write(`${t("repoUnknownSubcommand", { sub })}
 `);
     process.stderr.write(`${t("repoHelp")}
-`);
-    return 1;
-  }
-  const python = await findPython();
-  if (!python) {
-    process.stderr.write(`${t("repoPythonMissing")}
 `);
     return 1;
   }
@@ -226,7 +209,6 @@ async function runRepoCommand(args) {
     return 1;
   }
   const intent = asIntent(args.intent);
-  const repo = path2.resolve(args.repo ?? process.cwd());
   try {
     const result = await searchRepo({
       repo,
@@ -239,17 +221,54 @@ async function runRepoCommand(args) {
     print(args.json ? result : formatSearchText(result), args.json);
     return result.ok ? 0 : 1;
   } catch (error) {
-    const message = error instanceof Error ? error.message : String(error);
-    if (args.json) {
-      print({ ok: false, error: message }, true);
-    } else {
-      process.stderr.write(`${message}
+    return fail(error, args.json);
+  }
+}
+function fail(error, json) {
+  const message = error instanceof Error ? error.message : String(error);
+  if (json) print({ ok: false, error: message }, true);
+  else process.stderr.write(`${message}
 `);
-    }
-    return 1;
+  return 1;
+}
+function formatGraphText(graph, task) {
+  const e = graph.effort;
+  const m = graph.metrics;
+  const lines = [
+    t("repoGraphTitle"),
+    "",
+    t("aboutRepoGraph"),
+    "",
+    `Repo       ${graph.repo}`,
+    `Backend    ${graph.backend}`,
+    `Files      ${m.fileCount}  (${m.origin})`,
+    `Nodes      ${m.nodeCount}  (${m.origin})`,
+    `Edges      ${m.edgeCount}  (${m.origin})`,
+    `Tokens     ${m.tokenEstimate ?? "N/A"}  (${m.tokenOrigin ?? "N/A"})`,
+    `Effort     ${e.label}  ${e.score0to100}/100  (${e.origin})`
+  ];
+  if (task) lines.push(`Task       ${task}`);
+  if (graph.writtenTo) lines.push(`Wrote      ${graph.writtenTo}`);
+  for (const err of graph.errors) {
+    lines.push(`Error      ${err.tool}: ${err.message}  (${err.origin})`);
+  }
+  lines.push("");
+  const degree = /* @__PURE__ */ new Map();
+  for (const edge of graph.edges) {
+    degree.set(edge.source, (degree.get(edge.source) ?? 0) + 1);
+    degree.set(edge.target, (degree.get(edge.target) ?? 0) + 1);
   }
+  const top = [...graph.nodes].filter((n) => n.type === "file" || n.type === "function" || n.type === "test").sort((a, b) => (degree.get(b.id) ?? 0) - (degree.get(a.id) ?? 0)).slice(0, 12);
+  for (const node of top) {
+    const related = graph.edges.filter((edge) => edge.source === node.id).slice(0, 3).map((edge) => `${edge.type}\u2192${edge.symbol ?? edge.target} [${edge.confidence}]`);
+    lines.push(`${node.file ?? node.name}  ${node.name}  (${degree.get(node.id) ?? 0})`);
+    for (const rel of related) lines.push(`  ${rel}`);
+  }
+  if (!top.length) lines.push(t("repoGraphEmpty"));
+  return lines.join("\n");
 }
 export {
+  formatGraphText,
   formatSearchText,
   runRepoCommand
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pudu-ai",
-  "version": "0.2.21",
+  "version": "0.2.23",
   "description": "Pudu-AI — local AI hardware & benchmark lab for the terminal",
   "keywords": [
     "ai",

package/python/pudu_agent/__main__.py

@@ -1,4 +1,10 @@
-from pudu_agent.search import main
+import sys
 
 if __name__ == "__main__":
+    if len(sys.argv) > 1 and sys.argv[1] == "mcp":
+        from pudu_agent.mcp_server import main as mcp_main
+
+        raise SystemExit(mcp_main())
+    from pudu_agent.search import main
+
     raise SystemExit(main())