@adia-ai/a2ui-compose 0.5.18 → 0.5.19

package/CHANGELOG.md

@@ -11,6 +11,10 @@ generator graph.
 
 _No pending changes._
 
+## [0.5.19] - 2026-05-17
+
+_Lockstep ride-along (no source change in this package; companion to web-components v0.5.19 — see root CHANGELOG)._
+
 ## [0.5.18] - 2026-05-16
 
 _Lockstep ride-along (no source change in this package; companion to web-components v0.5.18 — see root CHANGELOG)._

package/core/generator.js

@@ -60,9 +60,10 @@ import { generateThinking } from '../strategies/monolithic/generate-thinking.js'
  *   or from a saved-canvas JSON file. Backward compatible: if omitted, the
  *   store lookup via `executionId` still works.
  * @param {object} [opts.llmAdapter] — LLM adapter (defaults to auto-detected)
+ * @param {object} [opts.context] — Ontology context (domain, tasks) parsed by plan_app_state
  * @returns {Promise<{ executionId: string, messages: object[], validation: object, suggestions: string[] }>}
  */
-export async function generateUI({ intent, engine: engineName = 'monolithic', mode = 'instant', executionId, currentCanvas, llmAdapter, model, sessionId }) {
+export async function generateUI({ intent, engine: engineName = 'monolithic', mode = 'instant', executionId, currentCanvas, llmAdapter, model, sessionId, context }) {
   // Iteration signal: either an executionId (server-stateful path) OR a
   // currentCanvas (client-stateless path). Both mean "refine existing UI".
   const hasClientCanvas = !!(currentCanvas && (currentCanvas.components?.length || currentCanvas.messages?.length));
@@ -138,6 +139,7 @@ export async function generateUI({ intent, engine: engineName = 'monolithic', mo
     sessionId,
     analysis,
     priorComponentsFromPayload,
+    context, // Pass the ontology context to the engine
   });
   const totalMs = Date.now() - t0;
 
@@ -236,9 +238,10 @@ export async function generateUI({ intent, engine: engineName = 'monolithic', mo
  * @param {string} [opts.executionId]
  * @param {object} [opts.llmAdapter]
  * @param {(query: string) => Promise<{ results: { title: string, snippet: string, url: string }[] }>} [opts.search] — Web search function
+ * @param {object} [opts.context] — Ontology context (domain, tasks) parsed by plan_app_state
  * @yields {object}
  */
-export async function* generateUIStream({ intent, executionId, llmAdapter, model, search, currentCanvas }) {
+export async function* generateUIStream({ intent, executionId, llmAdapter, model, search, currentCanvas, context }) {
   // currentCanvas accepted for API parity with generateUI. The streaming
   // path's downstream consumers don't yet read it, but parity prevents the
   // client from having to branch its call shape per mode. Tier-2 will
@@ -320,7 +323,7 @@ export async function* generateUIStream({ intent, executionId, llmAdapter, model
   //       was a tier mismatch. Fixed by running a fast keyword search here.
   //   (b) The synthesizer composes from below-threshold patterns too; "0
   //       patterns" reading as "nothing relevant" actively misleads the user.
-  const context = await getContext(intent, 2);
+  const catalogContext = await getContext(intent, 2);
   const topPatternsRaw = searchBlocks(intent).slice(0, 5);
   const candidateLines = topPatternsRaw.length
     ? topPatternsRaw.map(p => {
@@ -330,14 +333,14 @@ export async function* generateUIStream({ intent, executionId, llmAdapter, model
       })
     : ['  (no keyword match — synthesizer will compose from components)'];
   engine.submitStage(executionId, 'analyze', {
-    context, componentCount: context.components.length, patternCount: topPatternsRaw.length,
+    context: catalogContext, componentCount: catalogContext.components.length, patternCount: topPatternsRaw.length,
     topPatterns: topPatternsRaw.map(p => ({ name: p.name, score: p.score, domain: p.domain })),
-    confidence: context.components.length > 0 ? 0.85 : 0.5,
+    confidence: catalogContext.components.length > 0 ? 0.85 : 0.5,
   });
   yield {
     type: 'status',
     stage: 'analyze',
-    message: `${context.components.length} components inspected · ${topPatternsRaw.length} pattern candidate${topPatternsRaw.length === 1 ? '' : 's'}`,
+    message: `${catalogContext.components.length} components inspected · ${topPatternsRaw.length} pattern candidate${topPatternsRaw.length === 1 ? '' : 's'}`,
     outcomes: ['Top retrievals:', ...candidateLines],
   };
 
@@ -382,7 +385,7 @@ export async function* generateUIStream({ intent, executionId, llmAdapter, model
       try {
         const subResult = await adapter.complete({
           messages: [{ role: 'user', content: subtask.intent }],
-          systemPrompt: await buildSystemPrompt(context, patterns, researchContext, intent),
+          systemPrompt: await buildSystemPrompt(catalogContext, patterns, researchContext, intent, null, context),
         });
         const subMessages = parseA2UIResponse(subResult.content, { executionId, mode: 'stream', intent: subtask.intent });
         subtaskResults.push({ label: subtask.label, messages: subMessages });
@@ -400,7 +403,7 @@ export async function* generateUIStream({ intent, executionId, llmAdapter, model
       subtasks: decomposition.subtasks.length, confidence: 0.8,
     });
 
-    const composedValidation = validateSchema(composed, { intent });
+    const composedValidation = validateSchema(composed, { intent, context });
 
     engine.submitStage(executionId, 'validate', {
       ...composedValidation, confidence: composedValidation.score / 100,
@@ -433,7 +436,7 @@ export async function* generateUIStream({ intent, executionId, llmAdapter, model
   }
 
   // ── Stage 4: Generate via LLM stream ──
-  const systemPrompt = await buildSystemPrompt(context, patterns, researchContext, intent);
+  const systemPrompt = await buildSystemPrompt(catalogContext, patterns, researchContext, intent, null, context);
   const chatMessages = buildChatMessages(intent, previousExecId || executionId);
 
   yield { type: 'status', stage: 'generate', message: 'Streaming from LLM...' };
@@ -473,7 +476,7 @@ export async function* generateUIStream({ intent, executionId, llmAdapter, model
     messages, source: 'llm-stream', confidence: 0.8,
   });
 
-  let validation = validateSchema(messages, { intent });
+  let validation = validateSchema(messages, { intent, context });
 
   // One repair attempt if invalid (non-streaming, quick fix)
   if (!validation.valid) {
@@ -485,7 +488,7 @@ export async function* generateUIStream({ intent, executionId, llmAdapter, model
         systemPrompt,
       });
       messages = parseA2UIResponse(repairResponse.content, { executionId, mode: 'stream', intent });
-      validation = validateSchema(messages, { intent });
+      validation = validateSchema(messages, { intent, context });
     } catch { /* keep original */ }
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adia-ai/a2ui-compose",
-  "version": "0.5.18",
+  "version": "0.5.19",
   "description": "AdiaUI A2UI compose engine — framework-agnostic. Takes natural-language intents + a catalog and produces A2UI protocol messages. Pairs with `@adia-ai/a2ui-retrieval` (intent classification, catalog lookup) and `@adia-ai/a2ui-validator` (schema + semantic checks).",
   "type": "module",
   "exports": {

package/strategies/monolithic/_shared.js

@@ -142,7 +142,7 @@ async function getComponentCatalog() {
   return _componentCatalog;
 }
 
-export async function buildSystemPrompt(context, patterns, researchContext = '', intent = '', composition = null) {
+export async function buildSystemPrompt(context, patterns, researchContext = '', intent = '', composition = null, ontologyContext = null) {
   const parts = [];
 
   // ── Role + output format ──
@@ -154,6 +154,18 @@ Output format: [{ "type": "updateComponents", "surfaceId": "default", "component
 Each component: { "id": "<unique>", "component": "<Type>", "children": ["<childId>", ...], ...props }
 The root must have id "root". Use short, descriptive IDs (e.g., "hdr", "email-field", "submit-btn").`);
 
+  // ── Ontology Context Injection ──
+  if (ontologyContext) {
+    parts.push(`
+CRITICAL CONSTRAINT: You must strictly adhere to the following Domain and Task models.
+Do not hallucinate features, bounds, or data inputs that violate these structures.
+- Domain Entities: ${(ontologyContext.domain?.entities || []).join(', ') || 'N/A'}
+- Domain Metrics: ${(ontologyContext.domain?.metrics || []).join(', ') || 'N/A'}
+- Primary Tasks: ${(ontologyContext.tasks?.primary || []).join(', ') || 'N/A'}
+- Experience Mode: ${ontologyContext.experience?.mode || 'N/A'}
+- Experience Shell: ${ontologyContext.experience?.shell || 'N/A'}`);
+  }
+
   // ── Corpus context (§56, v0.4.5) ──
   // Tells the LLM how the system surrounding it is wired so it doesn't
   // hallucinate component names and so it understands what MATCHED PATTERN /

package/strategies/monolithic/generate-instant.js

@@ -16,7 +16,7 @@ import {
   generateSuggestions,
 } from './_shared.js';
 
-export async function generateInstant({ intent, executionId, storeId, analysis, priorComponentsFromPayload }) {
+export async function generateInstant({ intent, executionId, storeId, analysis, priorComponentsFromPayload, context }) {
   // Forward-compat: instant mode is pattern-match-only and doesn't iterate
   // on a prior canvas yet. Accept the param so callers can pass uniformly.
   void priorComponentsFromPayload;
@@ -56,13 +56,13 @@ export async function generateInstant({ intent, executionId, storeId, analysis,
     }
   }
 
-  // ── Analyze stage ──
-  const context = await getContext(intent, 1);
-  engine.submitStage(execId, 'analyze', {
-    context,
-    componentCount: context.components.length,
-    patternCount: context.patterns.length,
-    confidence: context.patterns.length > 0 ? 0.9 : 0.6,
+  // ── Stage 2: Analyze ──
+  const catalogContext = await getContext(intent, 1);
+  engine.submitStage(executionId, 'analyze', {
+    context: catalogContext,
+    componentCount: catalogContext.components.length,
+    patternCount: catalogContext.patterns.length,
+    confidence: catalogContext.patterns.length > 0 ? 0.9 : 0.6,
   });
 
   // ── Plan stage ──
@@ -168,7 +168,7 @@ export async function generateInstant({ intent, executionId, storeId, analysis,
   });
 
   // ── Validate stage ──
-  const validation = validateSchema(messages, { intent });
+  const validation = validateSchema(messages, { intent, context });
   engine.submitStage(execId, 'validate', {
     ...validation,
     confidence: validation.score / 100,

package/strategies/monolithic/generate-pro.js

@@ -25,7 +25,7 @@ import {
   generateSuggestions,
 } from './_shared.js';
 
-export async function generatePro({ intent, executionId, storeId, llmAdapter, analysis, priorComponentsFromPayload }) {
+export async function generatePro({ intent, executionId, storeId, llmAdapter, analysis, priorComponentsFromPayload, context }) {
   const execId = executionId;
 
   // The steelman is the enriched brief produced by the prompt-analyzer in
@@ -89,9 +89,9 @@ export async function generatePro({ intent, executionId, storeId, llmAdapter, an
   }
 
   // ── Stage 2: Analyze ──
-  const context = await getContext(intent, 2);
+  const catalogContext = await getContext(intent, 2);
   engine.submitStage(execId, 'analyze', {
-    context, componentCount: context.components.length, patternCount: context.patterns.length,
+    context: catalogContext, componentCount: catalogContext.components.length, patternCount: catalogContext.patterns.length,
     confidence: 0.85,
   });
 
@@ -241,7 +241,7 @@ ${chunkRefHtml}
 
   if (bestPattern && bestPattern.template) {
     // PRO PATH: Adapt the matched pattern via LLM
-    systemPrompt = await buildSystemPrompt(context, patterns, '', intent);
+    systemPrompt = await buildSystemPrompt(catalogContext, patterns, '', intent, null, context);
 
     const adaptPrompt = hasPriorCanvas
       ? `Adapt this UI for the intent: "${intent}"${driftGuard}
@@ -291,7 +291,7 @@ Instructions:
     if (isRecording()) { lastRawResponse = response.content; lastTokens = response.usage || null; }
   } else if (hasFragments) {
     // COMPOSE PATH: multiple fragment patterns found for subtasks — compose in a single LLM call
-    systemPrompt = await buildSystemPrompt(context, patterns, '', intent, { fragmentPatterns, decomposition });
+    systemPrompt = await buildSystemPrompt(catalogContext, patterns, '', intent, { fragmentPatterns, decomposition }, context);
 
     const fragmentContext = fragmentPatterns.map(f =>
       `## ${f.label} (from "${f.pattern.name}" pattern)\n${JSON.stringify(f.pattern.template, null, 1)}`
@@ -332,7 +332,7 @@ Instructions:
     if (isRecording()) { lastRawResponse = response.content; lastTokens = response.usage || null; }
   } else {
     // No pattern — fall through to full thinking-style generation
-    systemPrompt = await buildSystemPrompt(context, patterns, '', intent);
+    systemPrompt = await buildSystemPrompt(catalogContext, patterns, '', intent, null, context);
 
     if (hasPriorCanvas) {
       // Multi-turn: inject current canvas as explicit iteration context
@@ -404,7 +404,7 @@ ${buildCanvasDiffPrompt(intent, priorComponents, { originalIntent })}`;
   });
 
   // ── Stage 5: Validate + repair ──
-  let validation = validateSchema(messages, { intent });
+  let validation = validateSchema(messages, { intent, context });
   if (!validation.valid) {
     try {
       const failedChecks = validation.checks.filter(c => !c.passed);
@@ -413,7 +413,7 @@ ${buildCanvasDiffPrompt(intent, priorComponents, { originalIntent })}`;
         systemPrompt,
       });
       messages = parseA2UIResponse(repairResponse.content, { executionId: execId, intent, mode: 'pro', stopReason: repairResponse.stopReason });
-      validation = validateSchema(messages, { intent });
+      validation = validateSchema(messages, { intent, context });
     } catch { /* keep original */ }
   }
   engine.submitStage(execId, 'validate', { ...validation, confidence: validation.score / 100 });

package/strategies/monolithic/generate-thinking.js

@@ -21,7 +21,7 @@ import {
   generateSuggestions,
 } from './_shared.js';
 
-export async function generateThinking({ intent, executionId, storeId, llmAdapter, analysis, priorComponentsFromPayload }) {
+export async function generateThinking({ intent, executionId, storeId, llmAdapter, analysis, priorComponentsFromPayload, context }) {
   // Note: thinking mode currently composes from scratch each turn. The
   // priorComponentsFromPayload param is accepted for forward-compatibility
   // with multi-turn thinking iterations; not used in this pass yet.
@@ -72,12 +72,12 @@ export async function generateThinking({ intent, executionId, storeId, llmAdapte
   }
 
   // ── Stage 2: Analyze (tier 2 = 5000 token budget) ──
-  const context = await getContext(intent, 2);
+  const catalogContext = await getContext(intent, 2);
   engine.submitStage(execId, 'analyze', {
-    context,
-    componentCount: context.components.length,
-    patternCount: context.patterns.length,
-    confidence: context.components.length > 0 ? 0.85 : 0.5,
+    context: catalogContext,
+    componentCount: catalogContext.components.length,
+    patternCount: catalogContext.patterns.length,
+    confidence: catalogContext.components.length > 0 ? 0.85 : 0.5,
   });
 
   // ── Stage 3: Plan ──
@@ -91,7 +91,7 @@ export async function generateThinking({ intent, executionId, storeId, llmAdapte
   engine.submitStage(execId, 'plan', plan);
 
   // ── Stage 4: Generate via LLM ──
-  const systemPrompt = await buildSystemPrompt(context, patterns, '', intent);
+  const systemPrompt = await buildSystemPrompt(catalogContext, patterns, '', intent, null, context);
   const chatMessages = buildChatMessages(intent, storeId || execId);
   // Prepend the analyzer's enriched signals to the first user message so the
   // LLM treats them as ground truth instead of re-deriving from a terse prompt.
@@ -129,7 +129,7 @@ export async function generateThinking({ intent, executionId, storeId, llmAdapte
   // Either failing triggers a repair attempt. Catalog failures are typically
   // small (unknown prop, wrong enum, missing required id) and fix cleanly in
   // one repair round. Three-attempt cap preserved from the original loop.
-  let validation = validateSchema(messages, { intent });
+  let validation = validateSchema(messages, { intent, context });
   let catalogValidation = await validateCatalog(messages);
   let attempts = 0;
 
@@ -146,7 +146,7 @@ export async function generateThinking({ intent, executionId, storeId, llmAdapte
       });
       messages = parseA2UIResponse(repairResponse.content, { executionId: execId, mode: 'thinking', intent, stopReason: repairResponse.stopReason });
       if (isRecording()) { lastRawResponse = repairResponse.content; lastTokens = repairResponse.usage || lastTokens; }
-      validation = validateSchema(messages, { intent });
+      validation = validateSchema(messages, { intent, context });
       catalogValidation = await validateCatalog(messages);
     } catch {
       break; // stop repair loop on adapter error