@specverse/engines 6.7.8 → 6.16.0

package/libs/instance-factories/services/templates/postgres-native/step-conventions.ts

@@ -0,0 +1,539 @@
+/**
+ * Step Conventions for PostgreSQL native (pg) — mirrors the mongo-native
+ * library's pattern set but emits raw-SQL via the pgClient helpers
+ * (`insertOne` / `findOneByField` / `updateOneById` / etc.) instead of
+ * collection methods.
+ *
+ * Same convention names + same pattern texts, so spec authors can swap
+ * MongoDBNativeDriver for PostgresNativeDriver in the manifest without
+ * touching their .specly. The shared matcher (`matchAgainstConventions`
+ * in `../_shared/step-matching.ts`) drives both libraries — this file
+ * is purely the conventions array + a pg-flavoured `matchPgStep`
+ * delegate. (#43K-D follow-up: third ORM library demonstrating the
+ * abstraction.)
+ */
+
+import {
+  toMethod,
+  toVar,
+  matchAgainstConventions,
+  type SharedConvention,
+  type SharedStepContext,
+} from '../_shared/step-matching.js';
+
+export type PgStepConvention = SharedConvention<PgStepContext>;
+
+export interface PgStepContext extends SharedStepContext {
+  /** Table name for the model (lowercased + 's' or storage.table). */
+  tableName: string;
+  /** Full model registry so conventions can read attributes for default
+   *  values, types, and FK target tables. */
+  models?: Record<string, any>;
+}
+
+/**
+ * Compute sensible default values for a model's attributes — the
+ * postgres-native counterpart of mongodb-native's `deriveModelDefaults`.
+ * Same heuristics; same conventionManaged exclusions.
+ *
+ * Defaults emitted as `key: expression` strings ready to inline into an
+ * object-literal record body (the body is then handed to `insertOne`,
+ * which does the SQL plumbing). Empty if the model isn't in the registry.
+ */
+export function deriveModelDefaults(
+  modelName: string,
+  ctx: PgStepContext,
+): string[] {
+  if (!ctx.models) return [];
+  const model = ctx.models[modelName] || ctx.models[modelName.charAt(0).toUpperCase() + modelName.slice(1)];
+  if (!model) return [];
+  const attrs = model.attributes;
+  if (!attrs) return [];
+  const list = Array.isArray(attrs)
+    ? attrs.map((a: any) => [a.name, a])
+    : Object.entries(attrs);
+
+  const declaredVars = ctx.declaredVars || new Set();
+  const out: string[] = [];
+
+  // The "id" column is database-managed (BIGSERIAL or DEFAULT gen_random_uuid())
+  // so we don't emit it from the convention. createdAt/updatedAt come from
+  // the convention footer.
+  const conventionManaged = new Set(['createdAt', 'updatedAt', 'id']);
+  for (const [name, attr] of list as [string, any][]) {
+    if (!name) continue;
+    if (conventionManaged.has(name)) continue;
+    const required = !!attr.required;
+    const hasDefault = attr.default !== undefined;
+    if (!required && !hasDefault) continue;
+    if (hasDefault) {
+      out.push(`${name}: ${formatDefault(attr.default, attr.type)}`);
+      continue;
+    }
+    const type = (attr.type || 'String').toLowerCase();
+    if (type === 'integer' || type === 'int' || type === 'number' || type === 'float') {
+      const min = attr.min ?? 0;
+      out.push(`${name}: ${min}`);
+    } else if (type === 'boolean') {
+      out.push(`${name}: false`);
+    } else if (type === 'datetime' || type === 'date') {
+      out.push(`${name}: new Date().toISOString()`);
+    } else {
+      // String / UUID / Text — try FK resolution first.
+      const selfVar = modelName.charAt(0).toLowerCase() + modelName.slice(1);
+      const fkMatch = name.match(/^(.+)Id$/);
+      if (fkMatch && fkMatch[1] !== selfVar && declaredVars.has(fkMatch[1])) {
+        out.push(`${name}: (${fkMatch[1]} as any)?.id`);
+      } else {
+        out.push(`${name}: ''`);
+      }
+    }
+  }
+  return out;
+}
+
+function formatDefault(value: any, type?: string): string {
+  if (value === null || value === undefined) return 'null';
+  if (typeof value === 'boolean') return String(value);
+  if (typeof value === 'number') return String(value);
+  if (typeof value === 'string') {
+    if (/^-?\d+(\.\d+)?$/.test(value)) return value;
+    if (value === 'true' || value === 'false') return value;
+    if (value === 'now' && (type === 'DateTime' || type === 'Date')) {
+      return 'new Date().toISOString()';
+    }
+    return JSON.stringify(value);
+  }
+  return JSON.stringify(value);
+}
+
+function toTable(modelName: string): string {
+  return modelName.toLowerCase() + 's';
+}
+
+/** Resolve a value reference in step text to a TS expression. Mirrors
+ *  mongo's resolveValue so spec phrases map identically. */
+function resolveValue(rawValue: string, ctx: PgStepContext): string {
+  const value = rawValue.trim().replace(/^['"]|['"]$/g, '');
+  if (/^(current\s*time|now|timestamp)$/i.test(value)) return 'new Date().toISOString()';
+  if (/^-?\d+(\.\d+)?$/.test(value)) return value;
+  if (value === 'true' || value === 'false') return value;
+  if (value === 'null') return 'null';
+  const declared = ctx.declaredVars || new Set();
+  const params = ctx.parameterNames || [];
+  const head = value.split('.')[0]!;
+  if (declared.has(head)) return value;
+  if (params.includes(head)) return `args.${value}`;
+  return `'${value.replace(/'/g, "\\'")}'`;
+}
+
+function mostRecentStepResult(ctx: PgStepContext): string | null {
+  const declared = Array.from(ctx.declaredVars || []);
+  const stepResults = declared.filter((v) => /^step\d+Result$/.test(v))
+    .sort((a, b) => parseInt(b.slice(4), 10) - parseInt(a.slice(4), 10));
+  return stepResults[0] ?? null;
+}
+
+function pascal(model: string): string {
+  return model.charAt(0).toUpperCase() + model.slice(1);
+}
+
+export const PG_STEP_CONVENTIONS: PgStepConvention[] = [
+  // --- Find / Lookup by single field ---
+  {
+    name: 'find-by-field',
+    pattern: /^(?:look\s+up|find|fetch|get)\s+(\w+)\s+by\s+(\w+)(?:\s+or\s+fail.*)?$/i,
+    generateCall: (m, ctx) => {
+      const model = m[1]!;
+      const field = m[2]!;
+      const modelVar = toVar(model);
+      const table = toTable(model);
+      const params = ctx.parameterNames || [];
+      const declared = ctx.declaredVars || new Set();
+      const idVal = field === 'id'
+        ? (params.includes(`${modelVar}Id`) ? `args.${modelVar}Id` : 'args.id')
+        : `args.${field}`;
+
+      if (declared.has(modelVar)) {
+        return `    // Step ${ctx.stepNum}: Find ${model} by ${field} (already loaded)`;
+      }
+      declared.add(modelVar);
+      const failOnMissing = /or\s+fail/i.test(m[0]);
+      // `let` so subsequent conditional-create can reassign on null lookup.
+      return `    // Step ${ctx.stepNum}: Find ${model} by ${field}
+    let ${modelVar} = await findOneByField('${table}', '${field}', ${idVal}) as any;${failOnMissing ? `
+    if (!${modelVar}) throw new Error('${model} not found');` : ''}`;
+    },
+  },
+
+  // --- Find by composite (two fields) ---
+  {
+    name: 'find-by-composite',
+    pattern: /^(?:look\s+up|find|fetch|get)(?:\s+existing)?\s+(\w+)\s+by\s+(\w+)\s+and\s+(\w+)$/i,
+    generateCall: (m, ctx) => {
+      const model = m[1]!;
+      const f1 = m[2]!;
+      const f2 = m[3]!;
+      const modelVar = toVar(model);
+      const table = toTable(model);
+      const declared = ctx.declaredVars || new Set();
+      if (declared.has(modelVar)) {
+        return `    // Step ${ctx.stepNum}: Find ${model} by ${f1} and ${f2} (already loaded)`;
+      }
+      declared.add(modelVar);
+      const resolveFieldSource = (f: string) => {
+        const stripIdMatch = f.match(/^(.+)Id$/);
+        if (stripIdMatch) {
+          const candidate = stripIdMatch[1]!;
+          if (candidate !== modelVar && declared.has(candidate)) {
+            return `(${candidate} as any)?.id`;
+          }
+        }
+        return `args.${f}`;
+      };
+      const f1Src = resolveFieldSource(f1);
+      const f2Src = resolveFieldSource(f2);
+      return `    // Step ${ctx.stepNum}: Find ${model} by ${f1} and ${f2}
+    let ${modelVar} = await findOneByFields('${table}', { ${f1}: ${f1Src}, ${f2}: ${f2Src} }) as any;`;
+    },
+  },
+
+  // --- Create model record ---
+  {
+    name: 'create',
+    pattern: /^create\s+(?:new\s+)?(\w+)(?:\s+(?:with\s+)?(.+))?/i,
+    generateCall: (m, ctx) => {
+      const model = m[1]!;
+      const modelVar = toVar(model);
+      const table = toTable(model);
+      const params = ctx.parameterNames || [];
+      const declared = ctx.declaredVars || new Set();
+      declared.add(modelVar);
+      const dataExpr = params.length > 0 ? `{ ${params.join(', ')} }` : 'args';
+      return `    // Step ${ctx.stepNum}: Create ${model}
+    const ${modelVar} = await insertOne('${table}', ${dataExpr}) as any;`;
+    },
+  },
+
+  // --- Update specific field on previously-loaded model ---
+  {
+    name: 'update-field',
+    pattern: /^update\s+(\w+)\s+(\w+)\s+to\s+(.+)/i,
+    generateCall: (m, ctx) => {
+      const model = m[1]!;
+      const field = m[2]!;
+      const rawValue = m[3]!;
+      const modelVar = toVar(model);
+      if (!ctx.declaredVars?.has(modelVar)) return '';
+      const table = toTable(model);
+      const val = resolveValue(rawValue, ctx);
+      return `    // Step ${ctx.stepNum}: Update ${model}.${field} to ${rawValue.trim()}
+    await updateOneById('${table}', ${modelVar}.id, { ${field}: ${val} });`;
+    },
+  },
+
+  // --- Update field timestamp ---
+  {
+    name: 'update-field-timestamp',
+    pattern: /^update\s+(\w+)\s+(\w+)\s+timestamp$/i,
+    generateCall: (m, ctx) => {
+      const model = m[1]!;
+      const field = m[2]!;
+      const modelVar = toVar(model);
+      if (!ctx.declaredVars?.has(modelVar)) return '';
+      const table = toTable(model);
+      return `    // Step ${ctx.stepNum}: Update ${model}.${field} timestamp
+    await updateOneById('${table}', ${modelVar}.id, { ${field}: new Date().toISOString() });`;
+    },
+  },
+
+  // --- Generic "Update X" (writes args back) ---
+  {
+    name: 'update',
+    pattern: /^update\s+(\w+)(?:\s+(.+))?$/i,
+    generateCall: (m, ctx) => {
+      const model = m[1]!;
+      const modelVar = toVar(model);
+      if (!ctx.declaredVars?.has(modelVar)) return '';
+      const table = toTable(model);
+      return `    // Step ${ctx.stepNum}: Update ${model}
+    await updateOneById('${table}', ${modelVar}.id, args);`;
+    },
+  },
+
+  // --- Delete model record ---
+  {
+    name: 'delete',
+    pattern: /^delete\s+(\w+)/i,
+    generateCall: (m, ctx) => {
+      const model = m[1]!;
+      const modelVar = toVar(model);
+      if (!ctx.declaredVars?.has(modelVar)) return '';
+      const table = toTable(model);
+      return `    // Step ${ctx.stepNum}: Delete ${model}
+    await deleteOneById('${table}', ${modelVar}.id);`;
+    },
+  },
+
+  // --- Transition to lifecycle state ---
+  {
+    name: 'transition',
+    pattern: /^transition\s+(\w+)\s+to\s+(\w+)/i,
+    generateCall: (m, ctx) => {
+      const model = m[1]!;
+      const state = m[2]!;
+      const modelVar = toVar(model);
+      if (!ctx.declaredVars?.has(modelVar)) return '';
+      const table = toTable(model);
+      return `    // Step ${ctx.stepNum}: Transition ${model} to ${state}
+    if ((${modelVar} as any).status === '${state}') throw new Error('${model} is already ${state}');
+    await updateOneById('${table}', ${modelVar}.id, { status: '${state}' });`;
+    },
+  },
+
+  // --- Set field to value (on the controller's primary model) ---
+  {
+    name: 'set',
+    pattern: /^set\s+(\w+)\s+to\s+(.+)/i,
+    generateCall: (m, ctx) => {
+      const field = m[1]!;
+      const rawValue = m[2]!;
+      const modelVar = toVar(ctx.modelName);
+      const val = resolveValue(rawValue, ctx);
+      return `    // Step ${ctx.stepNum}: Set ${field} to ${rawValue.trim()}
+    await updateOneById(TABLE_NAME, (${modelVar} as any).id, { ${field}: ${val} });`;
+    },
+  },
+
+  // --- Increment / Decrement ---
+  // Postgres has no Mongo $inc; emit a column = column +/- N update. The
+  // helper updateOneById doesn't support raw SQL fragments, so emit query
+  // directly here.
+  {
+    name: 'increment',
+    pattern: /^increment\s+(\w+)\s+by\s+(\w+)/i,
+    generateCall: (m, ctx) => {
+      const field = m[1]!;
+      const amount = m[2]!;
+      const modelVar = toVar(ctx.modelName);
+      return `    // Step ${ctx.stepNum}: Increment ${field} by ${amount}
+    await query(\`UPDATE "\${TABLE_NAME}" SET "${field}" = "${field}" + $1 WHERE "id" = $2\`, [${amount}, (${modelVar} as any).id]);`;
+    },
+  },
+  {
+    name: 'decrement',
+    pattern: /^decrement\s+(\w+)\s+by\s+(\w+)/i,
+    generateCall: (m, ctx) => {
+      const field = m[1]!;
+      const amount = m[2]!;
+      const modelVar = toVar(ctx.modelName);
+      return `    // Step ${ctx.stepNum}: Decrement ${field} by ${amount}
+    await query(\`UPDATE "\${TABLE_NAME}" SET "${field}" = "${field}" - $1 WHERE "id" = $2\`, [${amount}, (${modelVar} as any).id]);`;
+    },
+  },
+
+  // --- Persist / Save / Store ---
+  {
+    name: 'persist',
+    pattern: /^(?:persist|save|store)\s+(\w+(?:\s+\w+)?)(?:\s+(?:for|to|record).*)?$/i,
+    generateCall: (m, ctx) => {
+      const target = toVar(m[1]!.replace(/\s+(.)/g, (_, c) => c.toUpperCase()));
+      const table = toTable(target);
+      const recordSrc = mostRecentStepResult(ctx) ?? 'args';
+      // Ensure `record` is an object before insertOne — wrap primitives.
+      return `    // Step ${ctx.stepNum}: Persist ${m[1]}
+    {
+      const _rec: any = ${recordSrc} && typeof ${recordSrc} === 'object' && !Array.isArray(${recordSrc})
+        ? ${recordSrc} : { value: ${recordSrc} };
+      await insertOne('${table}', _rec);
+    }`;
+    },
+  },
+
+  // --- Conditional create ---
+  {
+    name: 'conditional-create',
+    pattern: /^if\s+(\w+)\s+does\s+not\s+exist,?\s+create\s+new\s+(\w+)(?:\s+with\s+.+)?$/i,
+    generateCall: (m, ctx) => {
+      const modelVar = toVar(m[1]!);
+      const Model = pascal(m[2]!);
+      const table = toTable(Model);
+      if (!ctx.declaredVars?.has(modelVar)) return '';
+      const defaults = deriveModelDefaults(Model, ctx);
+      const defaultsBlock = defaults.length > 0 ? defaults.join(', ') + ',' : '';
+      return `    // Step ${ctx.stepNum}: If ${modelVar} does not exist, create new ${Model}
+    if (!${modelVar}) {
+      const _newRecord = { ${defaultsBlock} ...args, createdAt: new Date().toISOString(), updatedAt: new Date().toISOString() };
+      ${modelVar} = await insertOne('${table}', _newRecord) as any;
+    }`;
+    },
+  },
+
+  // --- Conditional update ---
+  {
+    name: 'conditional-update',
+    pattern: /^if\s+(\w+)\s+exists,?\s+update\s+(\w+)(?:\s+(.+))?$/i,
+    generateCall: (m, ctx) => {
+      const modelVar = toVar(m[1]!);
+      const field = m[2]!;
+      const table = toTable(m[1]!);
+      if (!ctx.declaredVars?.has(modelVar)) return '';
+      return `    // Step ${ctx.stepNum}: If ${modelVar} exists, update ${field}
+    if (${modelVar}) {
+      await updateOneById('${table}', (${modelVar} as any).id, { ${field}: new Date().toISOString() });
+    }`;
+    },
+  },
+
+  // --- Auto-create / Bulk fan-out ---
+  {
+    name: 'auto-create-loop',
+    pattern: /^(?:auto-create|bulk\s+create)\s+(\w+)\s+(?:profile|record|entry|entries)?s?\s+for\s+(?:all\s+)?(?:available\s+)?(\w+)$/i,
+    generateCall: (m, ctx) => {
+      const Model = pascal(m[1]!);
+      const targetTable = toTable(Model);
+      const sourceTable = m[2]!.toLowerCase();
+      const declared = Array.from(ctx.declaredVars || []);
+      const ownerVar = declared.find((v) => !/^step\d+Result$/.test(v) && v !== 'args');
+      const sourceSingular = sourceTable.replace(/s$/, '');
+      const linkField = sourceSingular + 'Id';
+      const ownerLinkField = ownerVar ? ownerVar + 'Id' : 'ownerId';
+
+      const modelDefaults = deriveModelDefaults(Model, ctx);
+      const defaultsBlock = modelDefaults.length > 0
+        ? modelDefaults.filter((d) => !d.startsWith(`${ownerLinkField}:`) && !d.startsWith(`${linkField}:`)).join(', ')
+        : '';
+
+      return `    // Step ${ctx.stepNum}: Auto-create ${m[1]} ${m[2]} for all ${m[2]}
+    {
+      const _allItems = await findAll('${sourceTable}');
+      const _ownerId = ${ownerVar ? `(${ownerVar} as any)?.id` : 'null'};
+      const _records = _allItems.map((_item: any) => ({
+        ${defaultsBlock ? defaultsBlock + ',' : ''}
+        ${ownerLinkField}: _ownerId,
+        ${linkField}: (_item as any).${sourceSingular}Id ?? (_item as any).id,
+        createdAt: new Date().toISOString(),
+        updatedAt: new Date().toISOString(),
+      }));
+      if (_records.length > 0) {
+        await insertMany('${targetTable}', _records);
+      }
+    }`;
+    },
+  },
+
+  // --- "Otherwise create new X record" ---
+  {
+    name: 'otherwise-create',
+    pattern: /^otherwise\s+create\s+(?:new\s+)?(\w+)\s+record$/i,
+    generateCall: (m, ctx) => {
+      const Model = pascal(m[1]!);
+      const modelVar = toVar(Model);
+      const table = toTable(Model);
+      const wasDeclared = ctx.declaredVars?.has(modelVar);
+      const declared = Array.from(ctx.declaredVars || []);
+      const defaults = deriveModelDefaults(Model, ctx);
+      const supplied = new Set<string>();
+      for (const entry of defaults) {
+        const colonIdx = entry.indexOf(':');
+        if (colonIdx > 0) supplied.add(entry.slice(0, colonIdx).trim());
+      }
+      const fkAssignments = declared
+        .filter((v) => v !== modelVar && !/^step\d+Result$/.test(v) && v !== 'args')
+        .filter((v) => !supplied.has(v + 'Id'))
+        .map((v) => {
+          supplied.add(v + 'Id');
+          return `${v}Id: (${v} as any)?.id`;
+        })
+        .join(', ');
+      const defaultsBlock = defaults.length > 0 ? defaults.join(', ') + ',' : '';
+      ctx.declaredVars?.add(modelVar);
+      return `    // Step ${ctx.stepNum}: Otherwise create new ${Model} record
+    else {
+      const _newRecord = { ${defaultsBlock} ${fkAssignments ? fkAssignments + ',' : ''} ...args, createdAt: new Date().toISOString(), updatedAt: new Date().toISOString() };
+      ${wasDeclared ? `${modelVar} = await insertOne('${table}', _newRecord) as any;` : `const ${modelVar} = await insertOne('${table}', _newRecord) as any;
+      void ${modelVar};`}
+    }`;
+    },
+  },
+
+  // --- Send/Emit/Publish event ---
+  {
+    name: 'send-event',
+    pattern: /^(?:send|emit|publish)\s+(\w+)\s+event/i,
+    generateCall: (m, ctx) => {
+      const event = m[1]!;
+      const modelVar = toVar(ctx.modelName);
+      const hasModelVar = ctx.declaredVars?.has(modelVar);
+      const payload = hasModelVar
+        ? `{ ${modelVar}Id: (${modelVar} as any)?.id, operation: '${ctx.operationName}', timestamp: new Date().toISOString() }`
+        : `{ operation: '${ctx.operationName}', timestamp: new Date().toISOString() }`;
+      return `    // Step ${ctx.stepNum}: Emit ${event} event
+    await eventBus.publish('${event}', ${payload} as any);`;
+    },
+  },
+
+  // --- Call service ---
+  {
+    name: 'call-service',
+    pattern: /^call\s+(\w+)\.(\w+)/i,
+    generateCall: (m, ctx) => {
+      const service = m[1]!;
+      const method = m[2]!;
+      const args = (ctx.parameterNames || []).join(', ');
+      return `    // Step ${ctx.stepNum}: Call ${service}.${method}
+    await (${toVar(service)} as any).${method}({ ${args} });`;
+    },
+  },
+
+  // --- Return ---
+  {
+    name: 'return-model',
+    pattern: /^return\s+(?:the\s+|updated\s+|created\s+)?(\w+)\s*$/i,
+    generateCall: (m, ctx) => {
+      const valueRaw = m[1]!.trim();
+      const declared = ctx.declaredVars || new Set();
+      const params = ctx.parameterNames || [];
+      const modelVar = toVar(ctx.modelName);
+      if (valueRaw.toLowerCase() === ctx.modelName.toLowerCase() || valueRaw === modelVar) {
+        return `    // Step ${ctx.stepNum}: Return ${ctx.modelName}
+    return ${modelVar};`;
+      }
+      if (declared.has(valueRaw)) {
+        return `    // Step ${ctx.stepNum}: Return ${valueRaw}
+    return ${valueRaw};`;
+      }
+      if (params.includes(valueRaw)) {
+        return `    // Step ${ctx.stepNum}: Return ${valueRaw}
+    return args.${valueRaw};`;
+      }
+      return `    // Step ${ctx.stepNum}: Return ${valueRaw} — TODO: resolve binding
+    return null;`;
+    },
+  },
+];
+
+/**
+ * Match a step against the postgres-native conventions. Same shape as
+ * `matchStep` (prisma) and `matchMongoStep` — the orm-agnostic
+ * AI-behaviors-generator can drive any of the three.
+ */
+export function matchPgStep(
+  step: string,
+  ctx: PgStepContext,
+): {
+  matched: boolean;
+  call: string;
+  helperMethod?: string;
+  functionName?: string;
+  inputs?: string[];
+  resultVar?: string;
+} {
+  // Same `args.X` plumbing as mongo-native — the controller wraps the
+  // action body with a single `args` parameter rather than destructuring.
+  const aiArgsExpr = (inputs: string[], paramNames: string[]) =>
+    inputs.length > 0
+      ? `{ ${inputs.map((n) => paramNames.includes(n) ? `${n}: args.${n}` : n).join(', ')} }`
+      : '{}';
+  return matchAgainstConventions(step, ctx, PG_STEP_CONVENTIONS, aiArgsExpr);
+}

package/libs/instance-factories/services/templates/prisma/ai-behaviors-generator.ts

@@ -135,7 +135,7 @@ async function validateTypeScriptTypes(code: string): Promise<string | null> {
  * produces a new hash. The prompt version is part of the hash so
  * prompt upgrades also invalidate.
  */
-const PROMPT_VERSION = '9.2.0';
+const PROMPT_VERSION = '9.7.0';
 
 function cacheKey(step: string, modelName: string, operationName: string, functionName: string, inputs: string[]): string {
   const payload = JSON.stringify({ step, modelName, operationName, functionName, inputs: [...inputs].sort(), v: PROMPT_VERSION });
@@ -173,10 +173,33 @@ function cacheWrite(key: string, body: string): void {
  *
  * This is async — it calls the AI engine to generate function bodies.
  */
-export default async function generateAiBehaviors(context: TemplateContext): Promise<string> {
+/** Optional matcher for orm-agnostic step matching. When present, this is
+ * used in place of the prisma library's `matchStep` so that the unmatched
+ * functions' names + inputs are computed by the same conventions library
+ * the consuming controller is using. Without this, mongo-native (or any
+ * non-prisma) controllers would diverge: they'd emit `aiBehaviors.X` calls
+ * with one inputs set while this generator would emit a function with a
+ * different signature. */
+type StepMatcher = (step: string, ctx: any) => {
+  matched: boolean;
+  call?: string;
+  helperMethod?: string;
+  functionName?: string;
+  inputs?: string[];
+  resultVar?: string;
+};
+
+export default async function generateAiBehaviors(
+  context: TemplateContext,
+  matcher?: StepMatcher,
+): Promise<string> {
   const { controller, model } = context;
   if (!controller?.actions) return '';
 
+  // Default matcher = prisma's matchStep for back-compat. Mongo-native
+  // (and any future ORM) passes its own matcher.
+  const stepMatcher: StepMatcher = matcher || (matchStep as any);
+
   const modelName = model?.name || controller.model || 'Model';
   const modelVar = modelName.charAt(0).toLowerCase() + modelName.slice(1);
   // List of all entity-model names available in the realized output. Forwarded
@@ -234,7 +257,7 @@ export default async function generateAiBehaviors(context: TemplateContext): Pro
         resultName: stepAs,
       };
 
-      const result = matchStep(stepText, ctx);
+      const result = stepMatcher(stepText, ctx);
       if (!result.matched && result.functionName) {
         // Avoid duplicate function definitions
         const existing = unmatchedFunctions.find(f => f.functionName === result.functionName);
@@ -322,13 +345,44 @@ export async function generateAiBehaviorsFile(opts: {
     /** Strip string/template/comment content so identifier-reference checks
      * don't match `reward` inside a kebab-case string like `'reward-not-granted'`
      * or `// reward stays as text`. We replace the literal contents with
-     * spaces so positions stay roughly aligned in case of debugging. */
+     * spaces so positions stay roughly aligned in case of debugging.
+     *
+     * Critical: template-string `${...}` interpolations are EXPRESSIONS, not
+     * literal text — they really do reference `input.X` etc. Strip only the
+     * non-interpolation segments of the template so the expression content
+     * survives the regex check. */
     const stripLiteralsAndComments = (src: string): string => {
-      return src
+      let out = src
         .replace(/\/\*[\s\S]*?\*\//g, (m) => ' '.repeat(m.length))
         .replace(/\/\/[^\n]*/g, (m) => ' '.repeat(m.length))
-        .replace(/(['"])(?:\\.|(?!\1).)*\1/g, (m) => m[0] + ' '.repeat(m.length - 2) + m[0])
-        .replace(/`(?:\\.|\$\{[^}]*\}|(?!`).)*`/g, (m) => '`' + ' '.repeat(m.length - 2) + '`');
+        .replace(/(['"])(?:\\.|(?!\1).)*\1/g, (m) => m[0] + ' '.repeat(m.length - 2) + m[0]);
+      // Strip text INSIDE backticks but preserve `${...}` expressions.
+      // We rebuild template contents by replacing literal-text spans
+      // (everything between `\`` and `${`, between `}` and `${`, or
+      // between `}` and the closing `\``) with spaces.
+      out = out.replace(/`((?:\\.|\$\{[^}]*\}|(?!`).)*)`/g, (full, content) => {
+        // Walk the content, copy `${...}` segments verbatim, replace other
+        // characters with spaces.
+        let result = '`';
+        let i = 0;
+        while (i < content.length) {
+          if (content[i] === '\\' && i + 1 < content.length) {
+            result += '  '; i += 2; continue;
+          }
+          if (content[i] === '$' && content[i + 1] === '{') {
+            // copy interpolation verbatim including the `${...}` markers
+            const end = content.indexOf('}', i);
+            const slice = end >= 0 ? content.slice(i, end + 1) : content.slice(i);
+            result += slice;
+            i += slice.length;
+            continue;
+          }
+          result += ' ';
+          i++;
+        }
+        return result + '`';
+      });
+      return out;
     };
 
     /** Detect whether the body manages its own input access (so adding our