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

🐛 fix(tabsContext.tsx): add null check for flow.data to prevent errors

Browse source 
The code now checks if flow.data is null before iterating over its nodes. This prevents errors that would occur when flow.data is null.
This commit is contained in:
Gabriel Almeida 2023-05-22 08:21:52 -03:00
commit 5635a86f04
1 changed files with 30 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

53
src/frontend/src/contexts/tabsContext.tsx
View file

@ -64,29 +64,36 @@ export function TabsProvider({ children }: { children: ReactNode }) {
if (cookie && Object.keys(templates).length > 0) {
let cookieObject: LangFlowState = JSON.parse(cookie);
cookieObject.flows.forEach((flow) => {
flow.data.nodes.forEach((node) => {
if (Object.keys(templates[node.data.type]["template"]).length > 0) {
node.data.node.base_classes =
templates[node.data.type]["base_classes"];
flow.data.edges.forEach((edge) => {
if (edge.source === node.id) {
edge.sourceHandle = edge.sourceHandle
.split("|")
.slice(0, 2)
.concat(templates[node.data.type]["base_classes"])
.join("|");
}
});
node.data.node.description =
templates[node.data.type]["description"];
node.data.node.template = updateTemplate(
templates[node.data.type][
"template"
] as unknown as APITemplateType,
node.data.node.template as APITemplateType
);
}
});
// check if flow.data is null
if (flow.data) {
flow.data.nodes.forEach((node) => {
// check if node.data.type is in templates
if (
Object.keys(templates).includes(node.data.type) &&
Object.keys(templates[node.data.type]["template"]).length > 0
) {
node.data.node.base_classes =
templates[node.data.type]["base_classes"];
flow.data.edges.forEach((edge) => {
if (edge.source === node.id) {
edge.sourceHandle = edge.sourceHandle
.split("|")
.slice(0, 2)
.concat(templates[node.data.type]["base_classes"])
.join("|");
}
});
node.data.node.description =
templates[node.data.type]["description"];
node.data.node.template = updateTemplate(
templates[node.data.type][
"template"
] as unknown as APITemplateType,
node.data.node.template as APITemplateType
);
}
});
}
});
setTabIndex(cookieObject.tabIndex);
setFlows(cookieObject.flows);