+
diff --git a/packages/app/src/lib/url-state.test.ts b/packages/app/src/lib/url-state.test.ts
index b7ed6d5e..5a229e95 100644
--- a/packages/app/src/lib/url-state.test.ts
+++ b/packages/app/src/lib/url-state.test.ts
@@ -105,6 +105,42 @@ describe('readUrlParams', () => {
expect(params.g_model).toBe('test');
expect(params).not.toHaveProperty('unknown_key');
});
+
+ it('maps embed scatter URL params to inference state keys', async () => {
+ setupWindow(
+ '?model=dsr1&gpus=h200,b200,mi325x&isl=1024&osl=1024&y=tpPerGpu&precision=fp8',
+ '/embed/scatter',
+ );
+ const { readUrlParams } = await import('@/lib/url-state');
+ const params = readUrlParams();
+ expect(params.g_model).toBe('DeepSeek-R1-0528');
+ expect(params.i_gpus).toBe('h200-sxm,b200-sxm,mi325x');
+ expect(params.i_seq).toBe('1024/1024');
+ expect(params.i_metric).toBe('y_tpPerGpu');
+ expect(params.i_prec).toBe('fp8');
+ });
+
+ it('allows explicit seq on embed URLs over isl/osl', async () => {
+ setupWindow('?seq=8k/1k&isl=512&osl=512', '/embed/scatter');
+ const { readUrlParams } = await import('@/lib/url-state');
+ const params = readUrlParams();
+ expect(params.i_seq).toBe('8k/1k');
+ });
+
+ it('keeps canonical embed params when both canonical and aliases are present', async () => {
+ setupWindow(
+ '?g_model=Kimi-K2.5&i_seq=8k/1k&i_prec=fp4&i_gpus=b300-sxm&i_metric=y_costh&model=dsr1&seq=1024/1024&precision=fp8&gpus=h200,b200&y=tpPerGpu',
+ '/embed/scatter',
+ );
+ const { readUrlParams } = await import('@/lib/url-state');
+ const params = readUrlParams();
+
+ expect(params.g_model).toBe('Kimi-K2.5');
+ expect(params.i_seq).toBe('8k/1k');
+ expect(params.i_prec).toBe('fp4');
+ expect(params.i_gpus).toBe('b300-sxm');
+ expect(params.i_metric).toBe('y_costh');
+ });
});
describe('hasAnyUrlParams', () => {
diff --git a/packages/app/src/lib/url-state.ts b/packages/app/src/lib/url-state.ts
index 58682ec1..585b6025 100644
--- a/packages/app/src/lib/url-state.ts
+++ b/packages/app/src/lib/url-state.ts
@@ -107,11 +107,89 @@ const TAB_PARAM_PREFIXES: Record
= {
/** In-memory store of current param values (kept in sync via writeUrlParams). */
const currentState: Record = {};
+const EMBED_MODEL_ALIASES: Record = {
+ dsr1: 'DeepSeek-R1-0528',
+ 'deepseek-r1': 'DeepSeek-R1-0528',
+ kimi: 'Kimi-K2.5',
+ kimik25: 'Kimi-K2.5',
+};
+
+const EMBED_GPU_ALIASES: Record = {
+ h100: 'h100-sxm',
+ h200: 'h200-sxm',
+ b200: 'b200-sxm',
+ b300: 'b300-sxm',
+ gb200: 'gb200-nvl72',
+ gb300: 'gb300-nvl72',
+ mi300x: 'mi300x',
+ mi325x: 'mi325x',
+ mi355x: 'mi355x',
+};
+
+const EMBED_Y_AXIS_ALIASES: Record = {
+ tppergpu: 'y_tpPerGpu',
+ tppermw: 'y_tpPerMw',
+ inputtppergpu: 'y_inputTputPerGpu',
+ outputtppergpu: 'y_outputTputPerGpu',
+ costh: 'y_costh',
+ costn: 'y_costn',
+ costr: 'y_costr',
+};
+
+function normalizeEmbedScatterParams(searchParams: URLSearchParams): UrlStateParams {
+ const normalized: UrlStateParams = {};
+
+ const hasCanonicalModel = Boolean(searchParams.get('g_model'));
+ const hasCanonicalSeq = Boolean(searchParams.get('i_seq'));
+ const hasCanonicalPrec = Boolean(searchParams.get('i_prec'));
+ const hasCanonicalGpus = Boolean(searchParams.get('i_gpus'));
+ const hasCanonicalMetric = Boolean(searchParams.get('i_metric'));
+
+ const modelRaw = searchParams.get('model');
+ if (modelRaw && !hasCanonicalModel) {
+ const modelAlias = EMBED_MODEL_ALIASES[modelRaw.toLowerCase()];
+ normalized.g_model = modelAlias ?? modelRaw;
+ }
+
+ const seqRaw = searchParams.get('seq');
+ const islRaw = searchParams.get('isl');
+ const oslRaw = searchParams.get('osl');
+ if (seqRaw && !hasCanonicalSeq) {
+ normalized.i_seq = seqRaw;
+ } else if (islRaw && oslRaw && !hasCanonicalSeq) {
+ normalized.i_seq = `${islRaw}/${oslRaw}`;
+ }
+
+ const precRaw = searchParams.get('prec') || searchParams.get('precision');
+ if (precRaw && !hasCanonicalPrec) normalized.i_prec = precRaw;
+
+ const gpusRaw = searchParams.get('gpus');
+ if (gpusRaw && !hasCanonicalGpus) {
+ normalized.i_gpus = gpusRaw
+ .split(',')
+ .map((gpu) => {
+ const token = gpu.trim();
+ return EMBED_GPU_ALIASES[token.toLowerCase()] ?? token;
+ })
+ .filter(Boolean)
+ .join(',');
+ }
+
+ const yRaw = searchParams.get('y');
+ if (yRaw && !hasCanonicalMetric) {
+ const yAlias = EMBED_Y_AXIS_ALIASES[yRaw.toLowerCase()];
+ normalized.i_metric = yAlias ?? yRaw;
+ }
+
+ return normalized;
+}
+
// On module load: snapshot share-link params from the URL.
// Cleanup is deferred so it runs after Next.js hydration finishes.
const _initialParams: UrlStateParams = {};
if (typeof window !== 'undefined') {
const searchParams = new URLSearchParams(window.location.search);
+ const isEmbedScatterRoute = window.location.pathname === '/embed/scatter';
for (const key of URL_STATE_KEYS) {
const value = searchParams.get(key);
if (value !== null) {
@@ -119,6 +197,15 @@ if (typeof window !== 'undefined') {
currentState[key] = value;
}
}
+ if (isEmbedScatterRoute) {
+ const embedParams = normalizeEmbedScatterParams(searchParams);
+ for (const [key, value] of Object.entries(embedParams)) {
+ if (!value) continue;
+ const typedKey = key as UrlStateKey;
+ _initialParams[typedKey] = value;
+ currentState[typedKey] = value;
+ }
+ }
// Defer cleanup so the Next.js router doesn't overwrite it during hydration
setTimeout(() => {
const sp = new URLSearchParams(window.location.search);