jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

update cleanEdges and fix updateNewOutput

Browse source
This commit is contained in:
anovazzi1 2024-05-28 18:29:07 -03:00
commit 02a1624bf4
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/frontend/src/utils/reactflowUtils.ts
View file

@ -46,6 +46,7 @@ export function checkChatInput(nodes: Node[]) {
} }
export function cleanEdges(nodes: Node[], edges: Edge[]) { export function cleanEdges(nodes: Node[], edges: Edge[]) {
console.log("cleanEdges");
let newEdges = cloneDeep(edges); let newEdges = cloneDeep(edges);
edges.forEach((edge) => { edges.forEach((edge) => {
// check if the source and target node still exists // check if the source and target node still exists
@ -75,12 +76,12 @@ export function cleanEdges(nodes: Node[], edges: Edge[]) {
} }
} }
if (sourceHandle) { if (sourceHandle) {
const index = scapeJSONParse(sourceHandle).idx ?? 0;
const id: sourceHandleType = { const id: sourceHandleType = {
id: sourceNode.data.id, id: sourceNode.data.id,
baseClasses: sourceNode.data.node!.base_classes, baseClasses: [sourceNode.data.node.outputs[index].selected],
dataType: sourceNode.data.type, dataType: sourceNode.data.type,
idx: idx: index,
sourceNode.data.node!.outputs[scapeJSONParse(sourceHandle).idx] ?? 0,
}; };
if (scapedJSONStringfy(id) !== sourceHandle) { if (scapedJSONStringfy(id) !== sourceHandle) {
newEdges = newEdges.filter((e) => e.id !== edge.id); newEdges = newEdges.filter((e) => e.id !== edge.id);
@ -414,6 +415,7 @@ export function updateEdgesHandleIds({
} }
export function updateNewOutput({ nodes, edges }: updateEdgesHandleIdsType) { export function updateNewOutput({ nodes, edges }: updateEdgesHandleIdsType) {
console.log("updateNewOutput");
let newEdges = cloneDeep(edges); let newEdges = cloneDeep(edges);
let newNodes = cloneDeep(nodes); let newNodes = cloneDeep(nodes);
newEdges.forEach((edge) => { newEdges.forEach((edge) => {
@ -434,6 +436,7 @@ export function updateNewOutput({ nodes, edges }: updateEdgesHandleIdsType) {
const selected = intersection[0]; const selected = intersection[0];
newSourceHandle.baseClasses = [selected]; newSourceHandle.baseClasses = [selected];
const id = newSourceHandle.id; const id = newSourceHandle.id;
newSourceHandle.idx = 0;
const sourceNodeIndex = newNodes.findIndex((node) => node.id === id); const sourceNodeIndex = newNodes.findIndex((node) => node.id === id);
if (sourceNodeIndex > -1) { if (sourceNodeIndex > -1) {
const sourceNode = newNodes[sourceNodeIndex]; const sourceNode = newNodes[sourceNodeIndex];