fix: prefer config resources for canvas mentions

This commit is contained in:
stupid-h4er
2026-06-03 12:49:29 +08:00
parent c70e7af14d
commit 76a4e7c4a2
@@ -54,18 +54,20 @@ export function buildInputMentionReferences(inputs: MentionInput[]) {
} }
export function getMentionResourceNodes(nodeId: string, nodes: CanvasNodeData[], connections: CanvasConnection[]) { export function getMentionResourceNodes(nodeId: string, nodes: CanvasNodeData[], connections: CanvasConnection[]) {
const ownInputs = getContextResourceNodes(nodeId, nodes, connections);
if (ownInputs.length) return ownInputs;
const configInputs = getConnectedConfigResourceNodes(nodeId, nodes, connections); const configInputs = getConnectedConfigResourceNodes(nodeId, nodes, connections);
if (configInputs.length) return configInputs; if (configInputs.length) return configInputs;
const ownInputs = getContextResourceNodes(nodeId, nodes, connections);
if (ownInputs.length) return ownInputs;
const node = nodes.find((item) => item.id === nodeId); const node = nodes.find((item) => item.id === nodeId);
return node && isResourceNode(node) ? [node] : []; return node && isResourceNode(node) ? [node] : [];
} }
export function getGenerationResourceNodes(nodeId: string, nodes: CanvasNodeData[], connections: CanvasConnection[]) { export function getGenerationResourceNodes(nodeId: string, nodes: CanvasNodeData[], connections: CanvasConnection[]) {
const configInputs = getConnectedConfigResourceNodes(nodeId, nodes, connections);
if (configInputs.length) return configInputs;
const ownInputs = getContextResourceNodes(nodeId, nodes, connections); const ownInputs = getContextResourceNodes(nodeId, nodes, connections);
if (ownInputs.length) return ownInputs; if (ownInputs.length) return ownInputs;
return getConnectedConfigResourceNodes(nodeId, nodes, connections); return [];
} }
function getContextResourceNodes(nodeId: string, nodes: CanvasNodeData[], connections: CanvasConnection[]) { function getContextResourceNodes(nodeId: string, nodes: CanvasNodeData[], connections: CanvasConnection[]) {