@adia-ai/a2ui-mcp 0.6.6 → 0.6.8

package/tools/refine.js

@@ -0,0 +1,151 @@
+import { z } from "zod";
+import { resolveAdapter } from "../server.js";
+const SYSTEM_PROMPT = `You are the A2UI Refiner.
+
+You receive:
+  1. The user's original intent.
+  2. A previously generated A2UI message array (the "previous output").
+  3. A list of validation errors that the previous output failed.
+
+Your job: produce a corrected A2UI message array that fixes ONLY the
+listed errors and preserves everything else about the previous output
+(component types, ids, intent, layout, copy). Do not refactor.
+
+Output ONLY a JSON object of the shape:
+{
+  "messages": [ ...A2UI messages... ]
+}
+
+No prose, no markdown fences. The "messages" array must be valid A2UI
+output \u2014 every component carries an id, a type, and the parent/layout
+fields it had before unless the error explicitly required changing
+them.`;
+function extractJsonObject(text) {
+  const stripped = text.replace(/^\s*```(?:json)?\s*/i, "").replace(/\s*```\s*$/i, "").trim();
+  try {
+    return JSON.parse(stripped);
+  } catch {
+  }
+  const start = stripped.indexOf("{");
+  if (start < 0) throw new Error("No JSON object found in LLM response");
+  let depth = 0;
+  let inStr = false;
+  let esc = false;
+  for (let i = start; i < stripped.length; i++) {
+    const ch = stripped[i];
+    if (inStr) {
+      if (esc) esc = false;
+      else if (ch === "\\") esc = true;
+      else if (ch === '"') inStr = false;
+      continue;
+    }
+    if (ch === '"') inStr = true;
+    else if (ch === "{") depth++;
+    else if (ch === "}") {
+      depth--;
+      if (depth === 0) {
+        return JSON.parse(stripped.slice(start, i + 1));
+      }
+    }
+  }
+  throw new Error("Unbalanced JSON object in LLM response");
+}
+function registerRefineTools(server) {
+  server.tool(
+    "refine_ui",
+    `Refine a previous generate_ui result whose validation failed. Pass the messages from the prior result + the validation errors, and the tool produces a corrected version. For monolithic engine; zettel has refine_composition.`,
+    {
+      intent: z.string().describe("Original intent string"),
+      previousMessages: z.array(z.any()).describe("Messages from the prior generate_ui call"),
+      validationErrors: z.array(z.any()).describe("Errors from the prior result.validation.errors")
+    },
+    async ({ intent, previousMessages, validationErrors }) => {
+      try {
+        const llm = await resolveAdapter();
+        const userPrompt = [
+          `INTENT:`,
+          intent,
+          ``,
+          `PREVIOUS OUTPUT (messages):`,
+          JSON.stringify(previousMessages, null, 2),
+          ``,
+          `VALIDATION ERRORS:`,
+          JSON.stringify(validationErrors, null, 2),
+          ``,
+          `Return ONLY: {"messages": [...]} with the errors above fixed and everything else preserved.`
+        ].join("\n");
+        const result = await llm.complete({
+          messages: [{ role: "user", content: userPrompt }],
+          systemPrompt: SYSTEM_PROMPT
+        });
+        const raw = result?.content ?? "";
+        let parsed;
+        try {
+          parsed = extractJsonObject(raw);
+        } catch (err) {
+          const e = err instanceof Error ? err : new Error(String(err));
+          return {
+            content: [
+              {
+                type: "text",
+                text: JSON.stringify(
+                  {
+                    error: `Refiner returned unparseable output: ${e.message}`,
+                    raw
+                  },
+                  null,
+                  2
+                )
+              }
+            ],
+            isError: true
+          };
+        }
+        const refinedMessages = Array.isArray(parsed?.messages) ? parsed.messages : null;
+        if (!refinedMessages) {
+          return {
+            content: [
+              {
+                type: "text",
+                text: JSON.stringify(
+                  {
+                    error: `Refiner response missing "messages" array`,
+                    parsed
+                  },
+                  null,
+                  2
+                )
+              }
+            ],
+            isError: true
+          };
+        }
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify(
+                {
+                  intent,
+                  messages: refinedMessages,
+                  errorsAddressed: validationErrors.length
+                },
+                null,
+                2
+              )
+            }
+          ]
+        };
+      } catch (err) {
+        const e = err instanceof Error ? err : new Error(String(err));
+        return {
+          content: [{ type: "text", text: `refine_ui error: ${e.message}` }],
+          isError: true
+        };
+      }
+    }
+  );
+}
+export {
+  registerRefineTools
+};

package/tools/corpus.ts

@@ -1,112 +0,0 @@
-/**
- * Corpus tools — gen-UI training-chunk search and retrieval.
- *
- * Extracted from server.ts. Registers:
- *   search_chunks, get_chunk, lookup_chunk
- */
-
-import { z } from 'zod';
-import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-
-import {
-  getChunk as getGenUIChunk,
-  lookupChunksByPrimary,
-  searchChunks as searchGenUIChunks,
-} from '../../corpus/scripts/chunk-library.js';
-
-export function registerCorpusTools(server: McpServer): void {
-  server.tool(
-    'search_chunks',
-    `Search the gen-UI training-chunk corpus by keyword.
-
-The chunk corpus comes from \`packages/a2ui/corpus/chunks/\` — JSON records
-extracted from every \`[data-chunk]\` element in site/pages/* and the corpus
-exemplars. There are three kinds:
-  - block (default): atomic UI fragment (KPI grid, sign-in form, table)
-  - panel: tab-panel fragment of a page (e.g. dashboard-overview-panel)
-  - page: full-page composition (e.g. dashboard-admin-page)
-
-Returns ranked candidates with chunk name, kind, primary tag, and a relevance
-score. Use \`get_chunk\` to fetch the full record (HTML + slot bindings + nested
-chunks) for a specific name.`,
-    {
-      query: z.string().describe('Keyword query — chunk name fragment, intent words, primary-tag name'),
-      kind: z.enum(['block', 'panel', 'page']).optional().describe('Filter by chunk kind'),
-      limit: z.number().int().min(1).max(50).default(20).describe('Max results'),
-    },
-    async ({ query, kind, limit }) => {
-      const results = searchGenUIChunks(query, { kind, limit });
-      return {
-        content: [{
-          type: 'text',
-          text: JSON.stringify({ query, kind: kind ?? 'any', count: results.length, results }, null, 2),
-        }],
-      };
-    },
-  );
-
-  server.tool(
-    'get_chunk',
-    `Fetch the full record for a single gen-UI training chunk by name.
-
-Returns the chunk's bounding HTML, slot annotations, nested chunk names, and
-metadata (primary tag, kind, source page). For chunks that appear on multiple
-pages (reusable slot chunks like \`auth-card-header\`, \`reg-step-header\`),
-returns an \`instances\` array — one entry per page where the chunk appears.
-
-The HTML is suitable for direct rendering / inclusion in an A2UI message
-construction prompt.`,
-    {
-      name: z.string().describe('The chunk name, e.g. "dashboard-kpi-grid", "auth-signin-card-email", "code-language"'),
-    },
-    async ({ name }) => {
-      const rec = getGenUIChunk(name);
-      if (!rec) {
-        return {
-          isError: true,
-          content: [{ type: 'text', text: JSON.stringify({ error: 'chunk not found', name }, null, 2) }],
-        };
-      }
-      return { content: [{ type: 'text', text: JSON.stringify(rec, null, 2) }] };
-    },
-  );
-
-  server.tool(
-    'lookup_chunk',
-    `List every chunk whose primary element is \`<component_name>\`.
-
-Useful for "show me every page that opens with a \`<card-ui raw>\`" or "every
-chunk built around a \`<grid-ui>\` root." Returns chunk names + kinds + sources.
-
-Pair with \`get_chunk\` to fetch full records for any of the returned names.`,
-    {
-      component_name: z.string().describe('Component tag name, e.g. "card-ui", "grid-ui", "drawer-ui"'),
-    },
-    async ({ component_name }) => {
-      const recs = lookupChunksByPrimary(component_name);
-      return {
-        content: [{
-          type: 'text',
-          text: JSON.stringify({
-            component: component_name,
-            count: recs.length,
-            chunks: recs.map((r) => {
-              const rec = r as Record<string, unknown>;
-              const instances = rec['instances'] as Array<Record<string, unknown>> | undefined;
-              const firstInstance = instances?.[0];
-              const slots = (rec['slots'] ?? firstInstance?.['slots'] ?? []) as Array<{ name: string }>;
-              const nested = (rec['nested'] ?? firstInstance?.['nested'] ?? []) as unknown[];
-              return {
-                name: rec['name'],
-                kind: rec['kind'],
-                page: rec['page'] ?? firstInstance?.['page'],
-                slots: slots.map((s) => s.name),
-                nested,
-              };
-            }),
-          }, null, 2),
-        }],
-      };
-    },
-  );
-}

package/tools/feedback.ts

@@ -1,73 +0,0 @@
-/**
- * Feedback tools — generation feedback collection and quality metrics.
- *
- * Extracted from server.ts. Registers:
- *   submit_feedback, get_quality_metrics, get_training_gaps
- */
-
-import { z } from 'zod';
-import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-
-import { FeedbackCollector } from '../../retrieval/feedback/feedback.js';
-import { feedbackStore } from '../../retrieval/feedback/feedback-store.js';
-
-const feedbackCollector = new FeedbackCollector();
-
-export function registerFeedbackTools(server: McpServer): void {
-  server.tool(
-    'submit_feedback',
-    'Submit structured feedback for a generation execution. Used by the evolution engine to learn from each generation.',
-    {
-      executionId: z.string().describe('Execution ID from generate_ui'),
-      rating: z.number().min(1).max(5).describe('Overall quality 1-5'),
-      intent: z.string().optional(),
-      domain: z.string().optional(),
-      intentAlignment: z.number().min(1).max(5).optional(),
-      visualQuality: z.number().min(1).max(5).optional(),
-      componentChoice: z.number().min(1).max(5).optional(),
-      userEdited: z.boolean().optional(),
-      editSummary: z.string().optional(),
-      notes: z.string().optional(),
-      shouldBePattern: z.boolean().optional(),
-      suggestedName: z.string().optional(),
-    },
-    async (args) => {
-      feedbackCollector.collectFeedback(args.executionId, {
-        rating: args.rating,
-        intentAlignment: args.intentAlignment,
-        visualQuality: args.visualQuality,
-        componentChoice: args.componentChoice,
-        userEdited: args.userEdited,
-        editSummary: args.editSummary,
-        notes: args.notes,
-      });
-      if (args.shouldBePattern != null) {
-        feedbackCollector.collectPatternFeedback(args.executionId, {
-          shouldBePattern: args.shouldBePattern,
-          suggestedName: args.suggestedName,
-        });
-      }
-      return { content: [{ type: 'text', text: JSON.stringify({ recorded: true, executionId: args.executionId, totalEntries: feedbackCollector.size }) }] };
-    }
-  );
-
-  server.tool(
-    'get_quality_metrics',
-    'Get aggregated quality metrics from the feedback store: avg score, thumb-up rate, per-domain breakdown, training gaps.',
-    {},
-    async () => {
-      const metrics = await feedbackStore.getQualityMetrics();
-      return { content: [{ type: 'text', text: JSON.stringify(metrics, null, 2) }] };
-    }
-  );
-
-  server.tool(
-    'get_training_gaps',
-    'Get training gap signals from LLM self-critique: missing patterns, weak domain keywords, component gaps.',
-    {},
-    async () => {
-      const gaps = await feedbackStore.getGapSummary();
-      return { content: [{ type: 'text', text: JSON.stringify(gaps, null, 2) }] };
-    }
-  );
-}