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

Nodes connection after deleting a node with edge connected bug fixed

Browse source
This commit is contained in:
Lucas Oliveira 2023-03-08 19:40:40 -03:00
commit 0e4bb1bc1d
3 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
langflow/frontend/src/contexts/typesContext.tsx
View file

@ -21,6 +21,7 @@ export function TypesProvider({ children }:{children:ReactNode}) {
reactFlowInstance.setNodes( reactFlowInstance.setNodes(
reactFlowInstance.getNodes().filter((n:Node) => n.id !== idx) reactFlowInstance.getNodes().filter((n:Node) => n.id !== idx)
); );
reactFlowInstance.setEdges(reactFlowInstance.getEdges().filter((ns) => ns.source !== idx && ns.target !== idx));
} }
return ( return (
<typesContext.Provider <typesContext.Provider

5
langflow/frontend/src/pages/FlowPage/index.tsx
View file

@ -155,6 +155,10 @@ export default function FlowPage({ flow }:{flow:FlowType}) {
); );
const onDelete = (mynodes) => {
setEdges(edges.filter((ns) => !nodes.some((n) => ns.source === n.id || ns.target === n.id)));
}
return ( return (
<div className="w-full h-full" ref={reactFlowWrapper}> <div className="w-full h-full" ref={reactFlowWrapper}>
{Object.keys(types).length > 0 ? ( {Object.keys(types).length > 0 ? (
@ -174,6 +178,7 @@ export default function FlowPage({ flow }:{flow:FlowType}) {
connectionLineComponent={ConnectionLineComponent} connectionLineComponent={ConnectionLineComponent}
onDragOver={onDragOver} onDragOver={onDragOver}
onDrop={onDrop} onDrop={onDrop}
onNodesDelete={onDelete}
> >
<Background className="dark:bg-gray-900"/> <Background className="dark:bg-gray-900"/>
<Controls className="[&>button]:text-black [&>button]:dark:bg-gray-800 hover:[&>button]:dark:bg-gray-700 [&>button]:dark:text-gray-400 [&>button]:dark:fill-gray-400 [&>button]:dark:border-gray-600"> <Controls className="[&>button]:text-black [&>button]:dark:bg-gray-800 hover:[&>button]:dark:bg-gray-700 [&>button]:dark:text-gray-400 [&>button]:dark:fill-gray-400 [&>button]:dark:border-gray-600">

1
langflow/frontend/src/utils.ts
View file

@ -315,7 +315,6 @@ export function isValidConnection(
{ source, target, sourceHandle, targetHandle }:Connection, { source, target, sourceHandle, targetHandle }:Connection,
reactFlowInstance:ReactFlowInstance reactFlowInstance:ReactFlowInstance
) { ) {
console.log(target)
if ( if (
sourceHandle.split('|')[0] === targetHandle.split("|")[0] || sourceHandle.split('|')[0] === targetHandle.split("|")[0] ||
sourceHandle.split('|').slice(2).some((t) => t === targetHandle.split("|")[0]) || sourceHandle.split('|').slice(2).some((t) => t === targetHandle.split("|")[0]) ||