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

Fixed name, description and showNode to use the interceptor function

Browse source
This commit is contained in:
Lucas Oliveira 2023-12-31 11:53:12 -03:00
commit 85979b8d37
1 changed files with 35 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

43
src/frontend/src/CustomNodes/GenericNode/index.tsx
View file

@ -7,7 +7,6 @@ import InputComponent from "../../components/inputComponent";
import { Textarea } from "../../components/ui/textarea"; import { Textarea } from "../../components/ui/textarea";
import { priorityFields } from "../../constants/constants"; import { priorityFields } from "../../constants/constants";
import { useSSE } from "../../contexts/SSEContext"; import { useSSE } from "../../contexts/SSEContext";
import { alertContext } from "../../contexts/alertContext";
import { FlowsContext } from "../../contexts/flowsContext"; import { FlowsContext } from "../../contexts/flowsContext";
import { typesContext } from "../../contexts/typesContext"; import { typesContext } from "../../contexts/typesContext";
import { undoRedoContext } from "../../contexts/undoRedoContext"; import { undoRedoContext } from "../../contexts/undoRedoContext";
@ -32,7 +31,7 @@ export default function GenericNode({
}): JSX.Element { }): JSX.Element {
const updateNodeInternals = useUpdateNodeInternals(); const updateNodeInternals = useUpdateNodeInternals();
const { types } = useContext(typesContext); const { types } = useContext(typesContext);
const { deleteNode } = useContext(FlowsContext); const { deleteNode, setNode } = useContext(FlowsContext);
const name = nodeIconsLucide[data.type] ? data.type : types[data.type]; const name = nodeIconsLucide[data.type] ? data.type : types[data.type];
const [inputName, setInputName] = useState(false); const [inputName, setInputName] = useState(false);
const [nodeName, setNodeName] = useState(data.node!.display_name); const [nodeName, setNodeName] = useState(data.node!.display_name);
@ -117,7 +116,10 @@ export default function GenericNode({
deleteNode(id); deleteNode(id);
}} }}
setShowNode={(show: boolean) => { setShowNode={(show: boolean) => {
data.showNode = show; setNode(data.id, (old) => ({
...old,
data: { ...old.data, showNode: show },
}));
}} }}
numberOfHandles={handles} numberOfHandles={handles}
showNode={showNode} showNode={showNode}
@ -169,8 +171,16 @@ export default function GenericNode({
setInputName(false); setInputName(false);
if (nodeName.trim() !== "") { if (nodeName.trim() !== "") {
setNodeName(nodeName); setNodeName(nodeName);
data.node!.display_name = nodeName; setNode(data.id, (old) => ({
updateNodeInternals(data.id); ...old,
data: {
...old.data,
node: {
...old.data.node,
display_name: nodeName,
},
},
}));
} else { } else {
setNodeName(data.node!.display_name); setNodeName(data.node!.display_name);
} }
@ -376,8 +386,16 @@ export default function GenericNode({
onBlur={() => { onBlur={() => {
setInputDescription(false); setInputDescription(false);
setNodeDescription(nodeDescription); setNodeDescription(nodeDescription);
data.node!.description = nodeDescription; setNode(data.id, (old) => ({
updateNodeInternals(data.id); ...old,
data: {
...old.data,
node: {
...old.data.node,
description: nodeDescription,
},
},
}));
}} }}
value={nodeDescription} value={nodeDescription}
onChange={(e) => setNodeDescription(e.target.value)} onChange={(e) => setNodeDescription(e.target.value)}
@ -391,7 +409,16 @@ export default function GenericNode({
) { ) {
setInputDescription(false); setInputDescription(false);
setNodeDescription(nodeDescription); setNodeDescription(nodeDescription);
data.node!.description = nodeDescription; setNode(data.id, (old) => ({
...old,
data: {
...old.data,
node: {
...old.data.node,
description: nodeDescription,
},
},
}));
updateNodeInternals(data.id); updateNodeInternals(data.id);
} }
}} }}