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

Fix vertex reset issue when pinned

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 16:17:24 -03:00
commit e104d0381e
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/langflow/graph/graph/base.py
View file

@ -2,6 +2,8 @@ from collections import defaultdict, deque
from typing import Dict, Generator, List, Optional, Type, Union from typing import Dict, Generator, List, Optional, Type, Union
from langchain.chains.base import Chain from langchain.chains.base import Chain
from loguru import logger
from langflow.graph.edge.base import ContractEdge from langflow.graph.edge.base import ContractEdge
from langflow.graph.graph.constants import lazy_load_vertex_dict from langflow.graph.graph.constants import lazy_load_vertex_dict
from langflow.graph.graph.utils import process_flow from langflow.graph.graph.utils import process_flow
@ -15,7 +17,6 @@ from langflow.graph.vertex.types import (
) )
from langflow.interface.tools.constants import FILE_TOOLS from langflow.interface.tools.constants import FILE_TOOLS
from langflow.utils import payload from langflow.utils import payload
from loguru import logger
class Graph: class Graph:
@ -140,7 +141,13 @@ class Graph:
self_vertex.params = {} self_vertex.params = {}
self_vertex._build_params() self_vertex._build_params()
self_vertex.graph = self self_vertex.graph = self
# If the vertex is pinned, we don't want
# to reset the results nor the _built attribute
if not self_vertex.pinned:
self_vertex._built = False self_vertex._built = False
self_vertex.result = None
self_vertex.artifacts = None
self_vertex.set_top_level(self.top_level_vertices)
self.reset_all_edges_of_vertex(self_vertex) self.reset_all_edges_of_vertex(self_vertex)
# Remove vertices # Remove vertices