AIX: OpenRouter: provider infra label

Author: enricoros

src/common/stores/chat/chat.message.ts

@@ -141,6 +141,7 @@ export type DMessageGenerator = ({
   },
 }) & {
   metrics?: DMetricsChatGenerate_Md;   // medium-sized metrics stored in the message
+  providerInfraLabel?: string;         // upstream provider that served the request (e.g., OpenRouter provider routing)
   upstreamHandle?: {
     uht: 'vnd.oai.responses',
     responseId: string,
@@ -244,6 +245,7 @@ export function duplicateDMessageGenerator(generator: Readonly<DMessageGenerator
         name: generator.name,
         // ...(generator.xeOpCode ? { xeOpCode: generator.xeOpCode } : {}),
         ...(generator.metrics ? { metrics: { ...generator.metrics } } : {}),
+        ...(generator.providerInfraLabel ? { providerInfraLabel: generator.providerInfraLabel } : {}),
         ...(generator.upstreamHandle ? { upstreamHandle: { ...generator.upstreamHandle } } : {}),
         ...(generator.tokenStopReason ? { tokenStopReason: generator.tokenStopReason } : {}),
       };
@@ -253,6 +255,7 @@ export function duplicateDMessageGenerator(generator: Readonly<DMessageGenerator
         name: generator.name,
         aix: { ...generator.aix },
         ...(generator.metrics ? { metrics: { ...generator.metrics } } : {}),
+        ...(generator.providerInfraLabel ? { providerInfraLabel: generator.providerInfraLabel } : {}),
         ...(generator.upstreamHandle ? { upstreamHandle: { ...generator.upstreamHandle } } : {}),
         ...(generator.tokenStopReason ? { tokenStopReason: generator.tokenStopReason } : {}),
       };

src/common/util/dMessageUtils.tsx

@@ -258,6 +258,7 @@ export function useMessageAvatarLabel(
       tooltip: complexity === 'minimal' ? null : (
         <Box sx={tooltipSx}>
           {VendorIcon ? <Box sx={tooltipIconContainerSx}><VendorIcon />{generator.name}</Box> : <div>{generator.name}</div>}
+          {generator.providerInfraLabel && <div>{vendorId} -&gt; via &lsquo;{generator.providerInfraLabel}&rsquo;</div>}
           {(modelId && complexity === 'extra') && <div>{modelId}</div>}
           {metrics && <div>{metrics}</div>}
           {stopReason && <div>{stopReason}</div>}

src/modules/aix/client/ContentReassembler.ts

@@ -302,6 +302,9 @@ export class ContentReassembler {
           case 'set-model':
             this.onModelName(op);
             break;
+          case 'set-provider-infra':
+            this.onProviderInfra(op);
+            break;
           case 'set-upstream-handle':
             this.onResponseHandle(op);
             break;
@@ -791,6 +794,10 @@ export class ContentReassembler {
     this.accumulator.genModelName = name;
   }
 
+  private onProviderInfra({ label }: Extract<AixWire_Particles.ChatGenerateOp, { cg: 'set-provider-infra' }>): void {
+    this.accumulator.genProviderInfraLabel = label;
+  }
+
   private onResponseHandle({ handle }: Extract<AixWire_Particles.ChatGenerateOp, { cg: 'set-upstream-handle' }>): void {
     // validate the handle
     if (handle?.uht !== 'vnd.oai.responses' || !handle?.responseId || handle?.expiresAt === undefined) {

src/modules/aix/client/aix.client.ts

@@ -537,6 +537,8 @@ function _llToDMessageGuts(src: AixChatGenerateContent_LL, dest: AixChatGenerate
     dest.generator.metrics = metricsChatGenerateLgToMd(src.genMetricsLg); // reduce the size to store in DMessage
   if (src.genModelName)
     dest.generator.name = src.genModelName;
+  if (src.genProviderInfraLabel)
+    dest.generator.providerInfraLabel = src.genProviderInfraLabel;
   if (src.genUpstreamHandle)
     dest.generator.upstreamHandle = src.genUpstreamHandle;
   if (src.legacyGenTokenStopReason)
@@ -577,6 +579,7 @@ export interface AixChatGenerateContent_LL {
   // pieces of generator
   genMetricsLg?: DMetricsChatGenerate_Lg;
   genModelName?: string;
+  genProviderInfraLabel?: string;
   genUpstreamHandle?: DMessageGenerator['upstreamHandle'];
   legacyGenTokenStopReason?: DMessageGenerator['tokenStopReason'];
 }

src/modules/aix/server/api/aix.wiretypes.ts

@@ -644,6 +644,7 @@ export namespace AixWire_Particles {
     | { cg: 'retry-reset', rScope: 'srv-dispatch' | 'srv-op' | 'cli-ll', rShallClear: boolean, reason: string, attempt: number, maxAttempts: number, delayMs: number, causeHttp?: number, causeConn?: string }
     | { cg: 'set-metrics', metrics: CGSelectMetrics }
     | { cg: 'set-model', name: string }
+    | { cg: 'set-provider-infra', label: string }
     | { cg: 'set-upstream-handle', handle: { uht: 'vnd.oai.responses', responseId: string, expiresAt: number | null } }
     | { cg: '_debugDispatchRequest', security: 'dev-env', dispatchRequest: { url: string, headers: string, body: string, bodySize: number } } // may generalize this in the future
     | { cg: '_debugProfiler', measurements: Record<string, number | string>[] };

src/modules/aix/server/dispatch/chatGenerate/ChatGenerateTransmitter.ts

@@ -494,6 +494,14 @@ export class ChatGenerateTransmitter implements IParticleTransmitter {
       this._queueParticleS();
   }
 
+  /** Communicates the provider name to the client (e.g., OpenRouter provider routing) */
+  setProviderInfraLabel(label: string) {
+    this.transmissionQueue.push({
+      cg: 'set-provider-infra',
+      label: label,
+    });
+  }
+
   /** Communicates the upstream response handle, for remote control/resumability */
   setUpstreamHandle(handle: string, _type: 'oai-responses' /* the only one for now, used for type safety */) {
     if (SERVER_DEBUG_WIRE)

src/modules/aix/server/dispatch/chatGenerate/parsers/IParticleTransmitter.ts

@@ -85,6 +85,9 @@ export interface IParticleTransmitter {
   /** Communicates the model name to the client */
   setModelName(modelName: string): void;
 
+  /** Communicates the provider name to the client (e.g., OpenRouter provider routing) */
+  setProviderInfraLabel(label: string): void;
+
   /** Communicates the upstream response handle, for remote control/resumability */
   setUpstreamHandle(handle: string, type: 'oai-responses'): void;
 

src/modules/aix/server/dispatch/chatGenerate/parsers/openai.parser.ts

@@ -41,6 +41,9 @@ export function createOpenAIChatCompletionsChunkParser(): ChatGenerateParseFunct
   let processedSearchResultUrls = new Set<string>();
   // NOTE: could compute rate (tok/s) from the first textful event to the last (to ignore the prefill time)
 
+  // [OpenRouter] Provider routing info - extracted from raw JSON before Zod strips it
+  let openRouterProviderInfraSent = false;
+
   // Supporting structure to accumulate the assistant message
   const accumulator: {
     content: string | null;
@@ -89,6 +92,12 @@ export function createOpenAIChatCompletionsChunkParser(): ChatGenerateParseFunct
       return;
     }
 
+    // [OpenRouter] Extract provider routing info (before Zod parsing strips unknown fields)
+    if (!openRouterProviderInfraSent && typeof chunkData?.provider === 'string' && chunkData.provider) {
+      openRouterProviderInfraSent = true;
+      pt.setProviderInfraLabel(chunkData.provider);
+    }
+
     const json = OpenAIWire_API_Chat_Completions.ChunkResponse_schema.parse(chunkData);
 
     // -> Model
@@ -409,6 +418,10 @@ export function createOpenAIChatCompletionsParserNS(): ChatGenerateParseFunction
     if (completeData.warning)
       console.log('AIX: OpenAI-dispatch-NS warning:', completeData.warning);
 
+    // [OpenRouter] Extract provider routing info (before Zod parsing strips unknown fields)
+    if (typeof completeData?.provider === 'string' && completeData.provider)
+      pt.setProviderInfraLabel(completeData.provider);
+
     // Parse the complete response
 
     // [Fixup, 2025-11-11] Some OpenAI-compatible APIs omit the 'object' field - inject it if needed