fix: align playground default model id with org namespace

Opening an org-scoped deployment in the Playground pre-selected the bare
model name (e.g. `qwen3-0.6b`), which never matched the org-namespaced
option the `/v1/models` dropdown actually lists (`org1/qwen3-0.6b`), so
the selection showed an unmatched value.

Resolve the owning org from the route row's `owner_principal_id` (falling
back to the org the caller is currently acting under for the admin "All"
view) and reconstruct the same `{org}/{name}` id the server reports. The
platform org carries no prefix, matching the server's behaviour.

- user.ts: add `getOrgById`/`getCurrentOrg` that scan both org caches
  (`organizationList` + admin-only `allOrganizations`) with string-
  normalised id comparison; `getOrgNameById` is now a thin wrapper.
  Helpers accept `undefined` so optional row fields type-check.
- use-open-playground: build the prefix from the resolved org record,
  keying the skip-prefix decision off `is_platform`.
- RouteItem: declare the `owner_principal_id` the list API returns.
- RouteTargetFormItem: drop the duplicate `overridden_model_name`
  declaration that TS flagged as a duplicate identifier.
This commit is contained in:
gitlawr
2026-06-03 10:59:43 +08:00
committed by jialin
parent fadbcc3e44
commit 72e794c281
3 changed files with 51 additions and 26 deletions
+38 -19
View File
@@ -90,20 +90,33 @@ const getStoredCurrentOrgId = (): number | null => {
// cluster-owner fallback.
const ORG_CACHE_KEYS = ['organizationList', 'allOrganizations'] as const;
const lookupOrgNamespace = (id: number | null): string | null => {
export interface CachedOrg {
id: number;
name?: string;
// The platform Org (single global tenant). Its models are NOT
// namespaced in ``/v1/models`` — they appear under their bare name.
is_platform?: boolean;
}
// Resolve the cached Org record for an owner/principal id by scanning both
// org caches. ``organizationList`` (the caller's member orgs) is checked
// alongside the admin-only ``allOrganizations`` so member sessions resolve
// too. Id types vary between localStorage payloads (some writers stringify,
// others persist as a JSON number), so compare as strings — strict equality
// would silently miss those cases.
export const getOrgById = (
id: number | string | null | undefined
): CachedOrg | null => {
if (id == null) return null;
// Normalise both sides to strings — the stored id type varies between
// localStorage payloads (some writers stringify, others persist as a
// JSON number); strict equality would silently miss those cases.
const target = String(id);
for (const key of ORG_CACHE_KEYS) {
try {
const raw = localStorage.getItem(key);
if (!raw) continue;
const list = JSON.parse(raw) as Array<{ id: number; name?: string }>;
const list = JSON.parse(raw) as CachedOrg[];
if (!Array.isArray(list)) continue;
const match = list.find((item) => String(item?.id) === target);
if (match?.name) return `gpustack-${match.name}`;
if (match) return match;
} catch {
// ignore malformed cache; continue checking other keys
}
@@ -111,17 +124,23 @@ const lookupOrgNamespace = (id: number | null): string | null => {
return null;
};
export const getAllOrganizations = (): Array<{ id: number; name?: string }> => {
const allOrganizations = localStorage.getItem('allOrganizations');
if (!allOrganizations) return [];
try {
const list = JSON.parse(allOrganizations) as Array<{
id: number;
name?: string;
}>;
if (!Array.isArray(list)) return [];
return list;
} catch {
return [];
}
// Bare Org *name* (e.g. ``org1``) for an owner/principal id, or null.
export const getOrgNameById = (
id: number | string | null | undefined
): string | null => {
return getOrgById(id)?.name ?? null;
};
const lookupOrgNamespace = (id: number | null): string | null => {
const name = getOrgNameById(id);
return name ? `gpustack-${name}` : null;
};
// The Org the caller is currently acting under, or null in the admin-"All"
// context. The org-switcher reloads the page on switch, so the list pages
// always show this org's resources — which is how ``/v1/models`` namespaces
// their model ids (``{org}/{name}``). Callers reconstructing that id use this
// as the fallback owner when a row carries no explicit ``owner_principal_id``.
export const getCurrentOrg = (): CachedOrg | null => {
return getOrgById(getStoredCurrentOrgId());
};
+3 -1
View File
@@ -4,7 +4,6 @@ export interface RouteTargetFormItem {
weight?: number | null;
model_id?: number;
provider_id?: number;
overridden_model_name?: string;
fallback_status_codes?: string[];
parentId?: string | number;
}
@@ -32,6 +31,9 @@ export interface RouteItem {
targets: number;
ready_targets: number;
access_policy: string;
// Org principal that owns this route. Drives the ``{org}/{name}`` model-id
// prefix when opening the route in the Playground.
owner_principal_id?: number;
}
export interface RouteTarget extends RouteTargetFormItem {
@@ -1,4 +1,4 @@
import { getAllOrganizations } from '@/atoms/user';
import { getCurrentOrg, getOrgById } from '@/atoms/user';
import { useNavigate } from '@umijs/max';
import { modelCategoriesMap } from '../../llmodels/config';
import { categoryToPathMap } from '../../llmodels/config/button-actions';
@@ -7,11 +7,15 @@ const useOpenPlayground = () => {
const navigate = useNavigate();
const handleOpenPlayGround = (row: any) => {
const allOrganizations = getAllOrganizations();
const orgName = allOrganizations.find(
(org) => org.id === row.owner_principal_id
)?.name;
const rawModel = orgName ? `${orgName}/${row.name}` : row.name;
// Match the id format the OpenAI ``/v1/models`` endpoint reports: an
// org's models are namespaced as ``{org}/{name}``, while the platform
// org's carry no prefix (the server strips it). Prefer the row's own
// ``owner_principal_id`` (an org principal id); fall back to the Org the
// caller is currently acting under for the admin "All" view, where the
// row's owner still resolves via the platform-wide org cache.
const org = getOrgById(row.owner_principal_id) ?? getCurrentOrg();
const rawModel =
org?.name && !org.is_platform ? `${org.name}/${row.name}` : row.name;
const modelName = encodeURIComponent(rawModel);
for (const [category, path] of Object.entries(categoryToPathMap)) {