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

chore(codeAreaModal): import cleanEdges function from reactflowUtils to improve code readability and maintainability

Browse source 
feat(codeAreaModal): add typesContext to access reactFlowInstance for cleaning edges before processing code
This commit is contained in:
anovazzi1 2023-07-19 18:14:22 -03:00
commit 1a5f33ddd4
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/frontend/src/modals/codeAreaModal/index.tsx
View file

@ -11,8 +11,10 @@ import { Button } from "../../components/ui/button";
import { CODE_PROMPT_DIALOG_SUBTITLE } from "../../constants"; import { CODE_PROMPT_DIALOG_SUBTITLE } from "../../constants";
import { alertContext } from "../../contexts/alertContext"; import { alertContext } from "../../contexts/alertContext";
import { darkContext } from "../../contexts/darkContext"; import { darkContext } from "../../contexts/darkContext";
import { typesContext } from "../../contexts/typesContext";
import { postCustomComponent, postValidateCode } from "../../controllers/API"; import { postCustomComponent, postValidateCode } from "../../controllers/API";
import { APIClassType } from "../../types/api"; import { APIClassType } from "../../types/api";
import { cleanEdges } from "../../util/reactflowUtils";
import BaseModal from "../baseModal"; import BaseModal from "../baseModal";
export default function CodeAreaModal({ export default function CodeAreaModal({
@ -32,6 +34,7 @@ export default function CodeAreaModal({
}) { }) {
const [code, setCode] = useState(value); const [code, setCode] = useState(value);
const { dark } = useContext(darkContext); const { dark } = useContext(darkContext);
const { reactFlowInstance } = useContext(typesContext);
const [height, setHeight] = useState(null); const [height, setHeight] = useState(null);
const { setErrorData, setSuccessData } = useContext(alertContext); const { setErrorData, setSuccessData } = useContext(alertContext);
const [error, setError] = useState<{ const [error, setError] = useState<{
@ -111,6 +114,13 @@ export default function CodeAreaModal({
} }
function handleClick() { function handleClick() {
cleanEdges({
flow: {
nodes: reactFlowInstance.getNodes(),
edges: reactFlowInstance.getEdges(),
},
updateEdge: reactFlowInstance.setEdges,
});
processCode(); processCode();
} }