@eeacms/volto-eea-chatbot 4.0.0 → 4.1.0

package/CHANGELOG.md

@@ -4,7 +4,34 @@ All notable changes to this project will be documented in this file. Dates are d
 
 Generated by [`auto-changelog`](https://github.com/CookPete/auto-changelog).
 
-### [4.0.0](https://github.com/eea/volto-eea-chatbot/compare/3.0.1...4.0.0) - 13 August 2026
+### [4.1.0](https://github.com/eea/volto-eea-chatbot/compare/4.0.0...4.1.0) - 31 August 2026
+
+#### :rocket: New Features
+
+- feat: expose message context + thread extra markdown into classic message [Tiberiu Ichim - [`b1600ce`](https://github.com/eea/volto-eea-chatbot/commit/b1600cede2221b846d36be42247f9068f22f06c4)]
+- feat: allow presentations to plug custom components into message rendering [Tiberiu Ichim - [`3f7d82b`](https://github.com/eea/volto-eea-chatbot/commit/3f7d82b9231fc2a8a92c72471068998988dadde5)]
+- feat: block presentation variations registry [Tiberiu Ichim - [`bc92ba7`](https://github.com/eea/volto-eea-chatbot/commit/bc92ba7a98b832cf34f644b8eb7294ae4542f04d)]
+- feat: add configurable batch size for fact-checking [Tiberiu Ichim - [`4e81cb9`](https://github.com/eea/volto-eea-chatbot/commit/4e81cb94075f8f31a70affed6f02ac8495863a57)]
+
+#### :bug: Bug Fixes
+
+- fix: stop hand-declaring the 'variation' schema field [Tiberiu Ichim - [`f0ada7e`](https://github.com/eea/volto-eea-chatbot/commit/f0ada7e7fb1843c562b0af44235f3a07b5253e7d)]
+- fix: skip citations with empty document_id in CITATION_DELTA [Tiberiu Ichim - [`81c9441`](https://github.com/eea/volto-eea-chatbot/commit/81c9441fa041b966f0e05b13fe271d44d9ee1421)]
+
+#### :house: Internal changes
+
+- chore: bump version to 4.1.0 (block presentation variations) [Tiberiu Ichim - [`45d7d3d`](https://github.com/eea/volto-eea-chatbot/commit/45d7d3db43158664c960d47585ffe749c490782c)]
+- chore: change default batchSize from 30 to 20 [Tiberiu Ichim - [`c5c5f4c`](https://github.com/eea/volto-eea-chatbot/commit/c5c5f4cb90fd74accd534ccd266f79592e38ec3a)]
+
+#### :house: Documentation changes
+
+- docs: document block presentation variations API [Tiberiu Ichim - [`8e98061`](https://github.com/eea/volto-eea-chatbot/commit/8e98061c3ce53eb701c78d3cb1eb783c8cb47234)]
+
+#### :hammer_and_wrench: Others
+
+- test: update citation delta test to match restored behavior [Tiberiu Ichim - [`de9f40e`](https://github.com/eea/volto-eea-chatbot/commit/de9f40e4dfd4a69f800aaf9f1b83fdf7e197b9ab)]
+- revert: restore original CITATION_DELTA handling in messageProcessor [Tiberiu Ichim - [`1d19fd3`](https://github.com/eea/volto-eea-chatbot/commit/1d19fd382bda782d7165d34d7edba4310ea65a25)]
+## [4.0.0](https://github.com/eea/volto-eea-chatbot/compare/3.0.1...4.0.0) - 13 August 2026
 
 #### :bug: Bug Fixes
 

package/README.md

@@ -116,6 +116,78 @@ Go to http://localhost:3000
 
    For a legacy Volto 17 project, install the package with `yarn` and restart the frontend as usual.
 
+## Block presentation variations
+
+Since 4.1.0 the chatbot block's presentation is pluggable. The block ships with the default `classic` presentation (the standard chat window), and other add-ons can register additional presentations ("variations") — e.g. to re-style the chat, replace the sources UI, or render custom elements inline in the assistant's answer. When two or more variations are registered, Volto core automatically adds a **Presentation** choice field to the block's edit sidebar, so editors can pick the variation per block. Old block content without a `variation` value keeps resolving to the default.
+
+### Registering a variation
+
+Push your variation onto `blocksConfig.eeaChatbot.variations` from your add-on's `applyConfig` (same cross-add-on pattern as `volto-tabs-block`):
+
+```js
+export default function applyConfig(config) {
+  const block = config.blocks.blocksConfig.eeaChatbot;
+  if (block) {
+    block.variations = block.variations || [];
+    if (!block.variations.find((v) => v.id === 'catalogue')) {
+      block.variations.push({
+        id: 'catalogue',
+        title: 'Catalogue',
+        isDefault: false,
+        view: CatalogueChatView,
+      });
+    }
+  }
+  return config;
+}
+```
+
+A variation is `{ id, title, isDefault, view }` (optionally `edit` / `schemaEnhancer`); `view` is the React component rendered for the block.
+
+### Variation view contract
+
+The variation `view` receives the block's fields as **top-level props** (there is no `data` prop), plus presentation props:
+
+```
+<View
+  persona={assistantData}   // the selected assistant
+  block_id                  // the block's internal id
+  isEditMode
+  isPlaywrightTest          // ?playwright=yes query flag
+  initialQuery              // ?query=… pre-filled question
+  initialDeepResearch       // ?deepResearch=… flag
+  {...blockFields}          // assistant, onyxVersion, height, …
+/>
+```
+
+If the registry is empty the block falls back to the classic `ChatWindow`, so registering a variation can never break the block.
+
+### Reusing the classic chat window
+
+A variation does not have to build a presentation from scratch — it can wrap the classic `ChatWindow` and only change what it needs to:
+
+```jsx
+import { ChatWindow } from '@eeacms/volto-eea-chatbot/ChatBlock/chat';
+
+export default function CatalogueChatView(props) {
+  return (
+    <ChatWindow
+      {...props}
+      hideSourcesTab
+      extraRemarkPlugins={[myRemarkPlugin]}
+      extraMarkdownComponents={{ myElement: MyElementComponent }}
+    />
+  );
+}
+```
+
+- `hideSourcesTab` — suppresses the classic Sources tab, sidebar and inline citation list only (the answer text and the quality-check logic are unaffected).
+- `extraRemarkPlugins` / `extraRehypePlugins` / `extraMarkdownComponents` — additional remark/rehype plugins and react-markdown component overrides, merged with the built-ins (the same mechanism the quality markers use). This lets a variation render custom inline elements inside the streamed answer.
+
+Custom markdown components can read the message that owns them through `ChatMessageContext` (also exported from `@eeacms/volto-eea-chatbot/ChatBlock/chat`): `ChatMessage` wraps every message in the context provider, so a component rendered from the answer text can e.g. match a marker against `message.documents`.
+
+> **Note:** the variations registry, `hideSourcesTab`, the extra-markdown pass-through props and `ChatMessageContext` are new in 4.1.0. If your project resolves an older published version, these seams don't exist — import the module namespace (e.g. `import * as chat from '…/ChatBlock/chat'`) and guard against `undefined` instead of using named imports.
+
 ## Quality Checks (Fact-Checking)
 
 When `qualityCheck` is enabled, the chatbot sends AI answers and their source

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eeacms/volto-eea-chatbot",
-  "version": "4.0.0",
+  "version": "4.1.0",
   "description": "@eeacms/volto-eea-chatbot: Volto add-on",
   "main": "src/index.js",
   "author": "European Environment Agency: IDM2 A-Team",

package/src/ChatBlock/ChatBlockView.jsx

@@ -3,6 +3,7 @@ import superagent from 'superagent';
 import { parse } from 'qs';
 import withOnyxData from './hocs/withOnyxData';
 import { ChatWindow } from './chat';
+import { getChatVariation } from './variations';
 
 function ChatBlockView(props) {
   const { id, assistantData, data, isEditMode, location } = props;
@@ -25,8 +26,13 @@ function ChatBlockView(props) {
     }
   }, [id, isPlaywrightTest, data]);
 
+  // Resolve the presentation variation for this block. Falls back to the
+  // classic ChatWindow so the block keeps working even if the registry is
+  // somehow empty.
+  const Presentation = getChatVariation(data)?.view || ChatWindow;
+
   return assistantData ? (
-    <ChatWindow
+    <Presentation
       persona={assistantData}
       isEditMode={isEditMode}
       isPlaywrightTest={isPlaywrightTest}

package/src/ChatBlock/chat/AIMessage.tsx

@@ -205,10 +205,15 @@ export function AIMessage({
   enableMatomoTracking,
   persona,
   maxContextSegments,
+  batchSize,
   isLastMessage,
   className = '',
   chatWindowEndRef,
   showTools,
+  hideSourcesTab,
+  extraRemarkPlugins,
+  extraMarkdownComponents,
+  extraRehypePlugins,
 }: ChatMessageProps) {
   const [allToolsDisplayed, setAllToolsDisplayed] = useState(false);
   const [messageDisplayed, setMessageDisplayed] = useState(false);
@@ -277,6 +282,12 @@ export function AIMessage({
 
   const showSources = messageDisplayed && sources.length > 0;
 
+  // Whether to actually render the classic Sources UI (tab, sidebar, inline
+  // list). `hideSourcesTab` lets a variation present the cited documents a
+  // different way (e.g. inline catalogue cards) while keeping quality-check
+  // logic intact.
+  const showSourcesTab = showSources && !hideSourcesTab;
+
   const contextSources = getContextSources(
     message,
     sources,
@@ -319,6 +330,7 @@ export function AIMessage({
     addCitations(message.message, message),
     stableContextSources,
     maxContextSegments,
+    batchSize,
   );
 
   const claims = markers?.claims || [];
@@ -384,7 +396,7 @@ export function AIMessage({
   const answerTab = (
     <div className="answer-tab">
       {/* Show first 3 sources inline */}
-      {showSources && (
+      {showSourcesTab && (
         <div className="sources">
           {sources.slice(0, 3).map((source: any, i: number) => (
             <SourceDetails source={source} key={i} index={source.index} />
@@ -448,6 +460,9 @@ export function AIMessage({
                 markers={markers}
                 stableContextSources={stableContextSources}
                 addQualityMarkersPlugin={addQualityMarkersPlugin}
+                extraRemarkPlugins={extraRemarkPlugins}
+                extraMarkdownComponents={extraMarkdownComponents}
+                extraRehypePlugins={extraRehypePlugins}
               >
                 {({ content }) => (
                   <div className="message-text-wrapper">{content}</div>
@@ -524,7 +539,7 @@ export function AIMessage({
       menuItem: { key: 'answer', content: 'Answer', className: 'answer-tab' },
       pane: <Tab.Pane key="answer">{answerTab}</Tab.Pane>,
     },
-    ...(showSources && !error
+    ...(showSourcesTab && !error
       ? [
           {
             menuItem: {
@@ -573,14 +588,14 @@ export function AIMessage({
               secondary: true,
               pointing: true,
               fluid: true,
-              className: cx({ 'without-sources': !showSources }),
+              className: cx({ 'without-sources': !showSourcesTab }),
             }}
             panes={panes}
             renderActiveOnly={false}
           />
 
           {/* Sources sidebar */}
-          {showSources && !error && (
+          {showSourcesTab && !error && (
             <Sidebar
               visible={showSourcesSidebar}
               animation="overlay"

package/src/ChatBlock/chat/ChatMessage.tsx

@@ -1,12 +1,15 @@
 import { Message as SemanticMessage } from 'semantic-ui-react';
 import type { ChatMessageProps } from '@eeacms/volto-eea-chatbot/ChatBlock/types/interfaces';
 import { UserMessage, AIMessage } from '.';
+import { ChatMessageContext } from './MessageContext';
 
 export function ChatMessage(props: ChatMessageProps) {
   const { message, libs, className = '' } = props;
 
+  let content: React.ReactNode = null;
+
   if (message.type === 'user') {
-    return (
+    content = (
       <UserMessage
         message={message}
         libs={libs}
@@ -14,14 +17,10 @@ export function ChatMessage(props: ChatMessageProps) {
         isLoading={props.isLoading}
       />
     );
-  }
-
-  if (message.type === 'assistant') {
-    return <AIMessage {...props} />;
-  }
-
-  if (message.type === 'error') {
-    return (
+  } else if (message.type === 'assistant') {
+    content = <AIMessage {...props} />;
+  } else if (message.type === 'error') {
+    content = (
       <div className="message-error">
         <SemanticMessage color="red" className="error-message">
           <div className="error-title">Error</div>
@@ -31,5 +30,13 @@ export function ChatMessage(props: ChatMessageProps) {
     );
   }
 
-  return null;
+  if (content === null) return null;
+
+  // Expose the owning message to deeply-nested descendants (e.g. custom
+  // markdown components injected by addons via `extraMarkdownComponents`).
+  return (
+    <ChatMessageContext.Provider value={message}>
+      {content}
+    </ChatMessageContext.Provider>
+  );
 }

package/src/ChatBlock/chat/ChatWindow.tsx

@@ -49,6 +49,7 @@ interface ChatWindowProps {
   enableMatomoTracking?: boolean;
   onDemandInputToggle?: boolean;
   maxContextSegments?: number;
+  batchSize?: number;
   onyxVersion?: '2' | '3';
   isPlaywrightTest?: boolean;
   initialQuery?: string | null;
@@ -89,7 +90,12 @@ function ChatWindow({
     enableMatomoTracking = true,
     onDemandInputToggle = true,
     maxContextSegments = 0,
+    batchSize,
     onyxVersion = '2',
+    hideSourcesTab,
+    extraRemarkPlugins,
+    extraMarkdownComponents,
+    extraRehypePlugins,
   } = data;
   const [qualityCheckEnabled, setQualityCheckEnabled] = useState(
     onDemandInputToggle ?? true,
@@ -231,6 +237,7 @@ function ChatWindow({
                     enableMatomoTracking={enableMatomoTracking}
                     persona={persona.id}
                     maxContextSegments={maxContextSegments}
+                    batchSize={batchSize}
                     isLastMessage={index === messages.length - 1}
                     className={
                       index === messages.length - 1 ? 'most-recent' : ''
@@ -238,6 +245,10 @@ function ChatWindow({
                     chatWindowRef={chatWindowRef}
                     chatWindowEndRef={chatWindowEndRef}
                     showTools={showTools}
+                    hideSourcesTab={hideSourcesTab}
+                    extraRemarkPlugins={extraRemarkPlugins}
+                    extraMarkdownComponents={extraMarkdownComponents}
+                    extraRehypePlugins={extraRehypePlugins}
                   />
                 </React.Fragment>
               ))}

package/src/ChatBlock/chat/MessageContext.ts

@@ -0,0 +1,12 @@
+import { createContext } from 'react';
+import type { Message } from '@eeacms/volto-eea-chatbot/ChatBlock/types/interfaces';
+
+/**
+ * Provides the currently-rendered chat message to deeply-nested descendants
+ * (e.g. custom react-markdown components injected by an addon).
+ *
+ * Set by `ChatMessage`, so any component rendered inside a given message
+ * (including custom markdown elements) can read the message that owns it —
+ * e.g. to match a `![[doc: Title]]` marker against `message.documents`.
+ */
+export const ChatMessageContext = createContext<Message | null>(null);

package/src/ChatBlock/chat/index.ts

@@ -1,4 +1,5 @@
 export { default as ChatWindow } from './ChatWindow';
 export { ChatMessage } from './ChatMessage';
+export { ChatMessageContext } from './MessageContext';
 export { AIMessage } from './AIMessage';
 export { UserMessage } from './UserMessage';

package/src/ChatBlock/hooks/useQualityMarkers.js

@@ -5,13 +5,18 @@ const Sentry = loadable.lib(
   () => import(/* webpackChunkName: "s_entry-browser" */ '@sentry/browser'), // chunk name avoids ad blockers
 );
 
-async function fetchHalloumi(answer, sources, maxContextSegments) {
+async function fetchHalloumi(answer, sources, maxContextSegments, batchSize) {
   const halloumiResponse = await fetch('/_ha/generate', {
     method: 'POST',
     headers: {
       'Content-Type': 'application/json',
     },
-    body: JSON.stringify({ answer, sources, maxContextSegments }),
+    body: JSON.stringify({
+      answer,
+      sources,
+      maxContextSegments,
+      ...(batchSize != null && { batch_size: batchSize }),
+    }),
   });
   return halloumiResponse;
 }
@@ -38,6 +43,7 @@ export function useQualityMarkers(
   message,
   sources,
   maxContextSegments = 0,
+  batchSize,
 ) {
   const [halloumiResponse, setHalloumiResponse] = React.useState(null);
   const [isLoading, setIsLoading] = React.useState(false);
@@ -70,6 +76,7 @@ export function useQualityMarkers(
           message,
           halloumiSources,
           maxContextSegments,
+          batchSize,
         );
         const body = await feedback.json();
         // console.log({ message, sources, body });
@@ -97,6 +104,7 @@ export function useQualityMarkers(
     message,
     sources,
     maxContextSegments,
+    batchSize,
   ]);
 
   if (halloumiResponse !== null) {

package/src/ChatBlock/index.js

@@ -1,6 +1,7 @@
 import codeSVG from '@plone/volto/icons/code.svg';
 import ChatBlockView from './ChatBlockView';
 import ChatBlockEdit from './ChatBlockEdit';
+import ChatWindow from './chat/ChatWindow';
 import { ChatBlockSchema } from './schema';
 
 export default function installChatBlock(config) {
@@ -26,7 +27,14 @@ export default function installChatBlock(config) {
       addPermission: [],
       view: [],
     },
-    variations: [],
+    variations: [
+      {
+        id: 'classic',
+        title: 'Classic',
+        isDefault: true,
+        view: ChatWindow,
+      },
+    ],
   };
 
   if (!config.blocks.blocksConfig.danswerChat) {

package/src/ChatBlock/packets/RendererComponent.tsx

@@ -87,6 +87,9 @@ export function RendererComponent({
   markers,
   stableContextSources,
   addQualityMarkersPlugin,
+  extraMarkdownComponents,
+  extraRemarkPlugins,
+  extraRehypePlugins,
 }: {
   packets: Packet[];
   onComplete: () => void;
@@ -98,6 +101,9 @@ export function RendererComponent({
   markers?: any;
   stableContextSources?: any;
   addQualityMarkersPlugin?: any;
+  extraMarkdownComponents?: Record<string, React.ComponentType<any>>;
+  extraRemarkPlugins?: any[];
+  extraRehypePlugins?: any[];
 }) {
   const RendererFn = findRenderer({ packets });
 
@@ -116,6 +122,9 @@ export function RendererComponent({
       markers={markers}
       stableContextSources={stableContextSources}
       addQualityMarkersPlugin={addQualityMarkersPlugin}
+      extraMarkdownComponents={extraMarkdownComponents}
+      extraRemarkPlugins={extraRemarkPlugins}
+      extraRehypePlugins={extraRehypePlugins}
     >
       {children}
     </RendererFn>

package/src/ChatBlock/packets/renderers/MessageTextRenderer.tsx

@@ -30,6 +30,9 @@ export const MessageTextRenderer: MessageRenderer<ChatPacket> = ({
   markers,
   stableContextSources,
   addQualityMarkersPlugin,
+  extraMarkdownComponents,
+  extraRemarkPlugins = [],
+  extraRehypePlugins = [],
 }) => {
   const { remarkGfm } = libs;
 
@@ -163,9 +166,18 @@ export const MessageTextRenderer: MessageRenderer<ChatPacket> = ({
   const renderedContent = (
     <div className="message-text-content">
       <Markdown
-        components={components(message, markers, stableContextSources)}
-        remarkPlugins={remarkGfm ? [remarkGfm.default] : []}
-        rehypePlugins={addQualityMarkersPlugin ? [addQualityMarkersPlugin] : []}
+        components={{
+          ...components(message, markers, stableContextSources),
+          ...extraMarkdownComponents,
+        }}
+        remarkPlugins={[
+          ...(remarkGfm ? [remarkGfm.default] : []),
+          ...extraRemarkPlugins,
+        ]}
+        rehypePlugins={[
+          ...(addQualityMarkersPlugin ? [addQualityMarkersPlugin] : []),
+          ...extraRehypePlugins,
+        ]}
       >
         {displayContent}
       </Markdown>

package/src/ChatBlock/schema.jsx

@@ -92,6 +92,7 @@ export function ChatBlockSchema({ assistants, data }) {
           ...(data.qualityCheck && data.qualityCheck !== 'disabled'
             ? [
                 'maxContextSegments',
+                'batchSize',
                 'noSupportDocumentsMessage',
                 'qualityCheckContext',
                 'qualityCheckStages',
@@ -109,6 +110,11 @@ export function ChatBlockSchema({ assistants, data }) {
       },
     ],
     properties: {
+      // NOTE: the `variation` field is NOT declared here on purpose. Volto
+      // core injects it automatically in the block sidebar
+      // (withVariationSchemaEnhancer, wired into BlockDataForm) from
+      // blocksConfig.eeaChatbot.variations whenever 2+ variations are
+      // registered. See @plone/volto/helpers/Extensions.
       enableShowTotalFailMessage: {
         title: 'Show total failure message',
         type: 'boolean',
@@ -252,6 +258,13 @@ range is from 0 to 100`,
         description:
           'Optimize quality check performance by grouping sentences into a max number of segments. Set to 0 to disable grouping and instead use one sentence per segment.',
       },
+      batchSize: {
+        title: 'Batch size',
+        type: 'number',
+        default: 20,
+        description:
+          'Number of claims to verify per LLM call. Higher values mean fewer LLM calls but larger prompts. Set to 1 for sequential per-claim verification.',
+      },
       feedbackReasons: {
         title: 'Feedback reasons',
         description: 'Select the reasons for negative feedback.',

package/src/ChatBlock/services/messageProcessor.ts

@@ -258,15 +258,8 @@ export class MessageProcessor {
       return;
     }
     const citationDelta = packet.obj as CitationDelta;
-    const explicitCitations = (citationDelta.citations || []).filter(
-      (citation: StreamingCitation) =>
-        citation.citation_num != null && Boolean(citation.document_id),
-    );
-    if (explicitCitations.length > 0) {
-      this.clearFallbackCitations();
-    }
-    explicitCitations.forEach((citation: StreamingCitation) => {
-      if (!this._citations.has(citation.citation_num)) {
+    citationDelta.citations?.forEach((citation: StreamingCitation) => {
+      if (citation.document_id && !this._citations.has(citation.citation_num)) {
         this._citations.set(citation.citation_num, citation.document_id);
       }
     });

package/src/ChatBlock/types/interfaces.ts

@@ -82,6 +82,17 @@ export interface MessageRendererProps<T extends Packet = Packet> {
   markers?: any;
   stableContextSources?: any;
   addQualityMarkersPlugin?: any;
+  /**
+   * Extra react-markdown component overrides (e.g. for custom elements
+   * produced by remark/rehype plugins), merged on top of the built-in
+   * mapping. Allows presentations to plug custom components inline in the
+   * message text.
+   */
+  extraMarkdownComponents?: Record<string, React.ComponentType<any>>;
+  /** Extra remark plugins, appended after the built-in ones. */
+  extraRemarkPlugins?: any[];
+  /** Extra rehype plugins, appended after the built-in ones. */
+  extraRehypePlugins?: any[];
   onComplete: () => void;
   animate: boolean;
   stopPacketSeen: boolean;
@@ -141,11 +152,25 @@ export interface ChatMessageProps {
   enableMatomoTracking?: boolean;
   persona?: number;
   maxContextSegments?: number;
+  batchSize?: number;
   isLastMessage?: boolean;
   className?: string;
   chatWindowRef?: React.RefObject<HTMLDivElement>;
   chatWindowEndRef?: React.RefObject<HTMLDivElement>;
   showTools?: PacketType[];
+  /**
+   * Hide the classic "Sources" tab, sidebar and inline source list for this
+   * presentation (the answer text still renders, and any injected markdown
+   * components still work). Used by variations that present the cited
+   * documents differently (e.g. inline catalogue cards).
+   */
+  hideSourcesTab?: boolean;
+  /** Extra remark plugins merged after the built-in ones. */
+  extraRemarkPlugins?: any[];
+  /** Extra react-markdown components merged after the built-in ones. */
+  extraMarkdownComponents?: any;
+  /** Extra rehype plugins merged after the built-in ones. */
+  extraRehypePlugins?: any[];
 }
 
 export interface Persona {

package/src/ChatBlock/variations.ts

@@ -0,0 +1,44 @@
+import type { ComponentType } from 'react';
+import config from '@plone/volto/registry';
+
+/**
+ * A chatbot block presentation variation.
+ *
+ * The `view` component is a complete presentation of the chatbot block.
+ * It receives the same props the classic presentation (ChatWindow) receives
+ * today: `{ persona, data, block_id, isEditMode, isPlaywrightTest,
+ * initialQuery, initialDeepResearch, ... }` plus whatever the HOCs inject
+ * (e.g. `rehypePrism`, `remarkGfm` when the view wraps itself with
+ * `injectLazyLibs`).
+ */
+export interface ChatBlockVariation {
+  id: string;
+  title: string;
+  isDefault?: boolean;
+  view: ComponentType<any>;
+  edit?: ComponentType<any>;
+  schemaEnhancer?: (args: { schema: any; formData: any }) => any;
+}
+
+/**
+ * Resolve the active variation from the block data.
+ *
+ * Semantics follow the established `variations` pattern of the EEA Volto
+ * addons (volto-tabs-block, volto-block-data-table, listing block):
+ * explicit `data.variation` → `isDefault` → first registered → undefined.
+ * Old block content without a `variation` field resolves to the default.
+ */
+export function getChatVariation(
+  data: Record<string, any>,
+): ChatBlockVariation | undefined {
+  const variations: ChatBlockVariation[] =
+    config.blocks?.blocksConfig?.eeaChatbot?.variations ?? [];
+  if (variations.length === 0) {
+    return undefined;
+  }
+  return (
+    variations.find((v) => v.id === data?.variation) ??
+    variations.find((v) => v.isDefault) ??
+    variations[0]
+  );
+}

package/src/halloumi/middleware.js

@@ -33,7 +33,7 @@ export default async function middleware(req, res, next) {
   }
 
   const body = req.body;
-  const { sources, answer } = body;
+  const { sources, answer, batch_size } = body;
 
   log('Halloumi request body', {
     answer,
@@ -52,6 +52,7 @@ export default async function middleware(req, res, next) {
         answer,
         sources,
         max_context_segments: 0,
+        ...(batch_size != null && { batch_size }),
       }),
     });