fix: lora tag

This commit is contained in:
jialin
2026-05-27 15:45:28 +08:00
committed by jialin
parent e40b0c2e07
commit 34be4800d7
4 changed files with 10 additions and 9 deletions
@@ -98,7 +98,7 @@ const RouteItem: React.FC<TargetItemProps> = ({
}} }}
> >
<AutoTooltip ghost>{data.name}</AutoTooltip> <AutoTooltip ghost>{data.name}</AutoTooltip>
{!!data.lora_module_name && ( {!!data.overridden_model_name && !!data.model_id && (
<FilesTag color="purple" variant="outlined"> <FilesTag color="purple" variant="outlined">
<span style={{ opacity: 1 }}>LoRA</span> <span style={{ opacity: 1 }}>LoRA</span>
</FilesTag> </FilesTag>
+1 -1
View File
@@ -4,7 +4,7 @@ export interface RouteTargetFormItem {
weight?: number | null; weight?: number | null;
model_id?: number; model_id?: number;
provider_id?: number; provider_id?: number;
lora_module_name?: string; overridden_model_name?: string;
fallback_status_codes?: string[]; fallback_status_codes?: string[];
parentId?: string | number; parentId?: string | number;
} }
+7 -6
View File
@@ -31,7 +31,7 @@ const isSameTarget = (
return false; return false;
} }
if (left.model_id && left.model_id === right.model_id) { if (left.model_id && left.model_id === right.model_id) {
return left.lora_module_name === right.lora_module_name; return left.overridden_model_name === right.overridden_model_name;
} }
if (left.provider_id && left.provider_id === right.provider_id) { if (left.provider_id && left.provider_id === right.provider_id) {
return left.overridden_model_name === right.overridden_model_name; return left.overridden_model_name === right.overridden_model_name;
@@ -48,10 +48,11 @@ const normalizeTarget = (
model_id: target?.model_id, model_id: target?.model_id,
provider_id: target?.provider_id, provider_id: target?.provider_id,
overridden_model_name: target?.overridden_model_name, overridden_model_name: target?.overridden_model_name,
lora_module_name: target?.lora_module_name,
fallback_status_codes: target?.fallback_status_codes fallback_status_codes: target?.fallback_status_codes
}; };
console.log('normalizeTarget', target, normalizedTarget);
return _.omitBy(normalizedTarget, _.isUndefined) as RouteTargetFormItem; return _.omitBy(normalizedTarget, _.isUndefined) as RouteTargetFormItem;
}; };
@@ -174,8 +175,8 @@ const AccessForm: React.FC<ProviderFormProps> = forwardRef((props, ref) => {
value: ep.model_id value: ep.model_id
? [ ? [
'deployments', 'deployments',
ep.lora_module_name ep.overridden_model_name
? `${ep.model_id}_lora_${ep.lora_module_name}` ? `${ep.model_id}_lora_${ep.overridden_model_name}`
: ep.model_id : ep.model_id
] ]
: [ep.provider_id, ep.overridden_model_name] : [ep.provider_id, ep.overridden_model_name]
@@ -202,8 +203,8 @@ const AccessForm: React.FC<ProviderFormProps> = forwardRef((props, ref) => {
value: fallbackTarget.model_id value: fallbackTarget.model_id
? [ ? [
'deployments', 'deployments',
fallbackTarget.lora_module_name fallbackTarget.overridden_model_name
? `${fallbackTarget.model_id}_lora_${fallbackTarget.lora_module_name}` ? `${fallbackTarget.model_id}_lora_${fallbackTarget.overridden_model_name}`
: fallbackTarget.model_id : fallbackTarget.model_id
] ]
: [fallbackTarget.provider_id, fallbackTarget.overridden_model_name] : [fallbackTarget.provider_id, fallbackTarget.overridden_model_name]
@@ -71,7 +71,7 @@ const useTargetSourceModels = () => {
data: { data: {
model_id: model.id, model_id: model.id,
parentId: 'deployments', parentId: 'deployments',
lora_module_name: loraName overridden_model_name: loraName
}, },
source: 'deployment', source: 'deployment',
isLora: true isLora: true