fix(playground): inject X-Organization-Id on non-umi fetch paths

Playground completions (chat streaming via ``fetchChunkedData``)
and the raw ``fetch()`` calls for image / audio generation skip
umi's ``request`` pipeline and so miss the request interceptor in
``request.extensions.ts`` that pins tenant context. Higress
receives those calls without ``X-Organization-Id`` and the auth
callback falls back to the user's USER-principal id rather than
their active Org, which makes Playground usage invisible from
Org-scoped Usage views even after the backend learned to backfill
``consumer_principal_id``.

Centralise the lookup in ``tenantHeaders()`` next to
``fetchChunkedData`` so both streaming helpers and the four
``fetch()``-direct entry points in ``playground/apis/index.ts``
(image / TTS) attach the header with the same precedence as the
umi interceptor (``createScopeOrgOverride`` first,
``currentOrganizationId`` second). Returns an empty object when
no active org context is set, leaving requests unchanged.
This commit is contained in:
gitlawr
2026-05-25 18:02:31 +08:00
committed by jialin
parent 7944b3f579
commit 7bee337554
2 changed files with 64 additions and 7 deletions
+49 -2
View File
@@ -3,6 +3,48 @@ import qs from 'query-string';
const extractStreamRegx = /(data|error):\s*({.*?})(?=\n|$)/g;
const readJsonNumber = (
storage: Storage | null,
key: string
): number | null => {
if (storage == null) {
return null;
}
try {
const raw = storage.getItem(key);
if (raw == null) {
return null;
}
const parsed = JSON.parse(raw);
return typeof parsed === 'number' ? parsed : null;
} catch {
return null;
}
};
/**
* Read the active Org id the same way the umi request interceptor in
* ``request.extensions.ts`` does, then translate it to the header the
* backend's tenant resolver expects. Lets non-umi ``fetch()`` paths —
* streaming Playground completions, raw image / TTS POSTs — pin
* tenant context with the same precedence rules as everywhere else
* (createScope override first, current org second). Returns an empty
* object when no active org context is set.
*/
export const tenantHeaders = (): Record<string, string> => {
if (typeof window === 'undefined') {
return {};
}
try {
const orgId =
readJsonNumber(window.sessionStorage, 'createScopeOrgOverride') ??
readJsonNumber(window.localStorage, 'currentOrganizationId');
return orgId == null ? {} : { 'X-Organization-Id': String(orgId) };
} catch {
return {};
}
};
const extractJSON = (
dataStr: string
): { results: any[]; remaining: string } => {
@@ -71,7 +113,8 @@ export const fetchChunkedData = async (params: {
signal: params.signal,
headers: {
'Content-Type': 'application/json',
...params.headers
...tenantHeaders(),
...(params.headers || {})
}
});
@@ -122,7 +165,11 @@ export const fetchChunkedDataPostFormData = async (params: {
const response = await fetch(url, {
method: 'POST',
body: createFormData(params.data),
signal: params.signal
signal: params.signal,
headers: {
...tenantHeaders(),
...(params.headers || {})
}
});
if (!response.ok) {
return await errorHandler(response);