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

Update vertex class in socket utils.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-01 23:57:40 -03:00
commit 9dac7e7635
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/services/socket/utils.py
View file

@ -7,7 +7,7 @@ from sqlmodel import select
from langflow.api.utils import format_elapsed_time from langflow.api.utils import format_elapsed_time
from langflow.api.v1.schemas import ResultDataResponse, VertexBuildResponse from langflow.api.v1.schemas import ResultDataResponse, VertexBuildResponse
from langflow.graph.graph.base import Graph from langflow.graph.graph.base import Graph
from langflow.graph.vertex.base import StatelessVertex from langflow.graph.vertex.base import Vertex
from langflow.services.database.models.flow.model import Flow from langflow.services.database.models.flow.model import Flow
from langflow.services.deps import get_session from langflow.services.deps import get_session
from langflow.services.monitor.utils import log_vertex_build from langflow.services.monitor.utils import log_vertex_build
@ -63,7 +63,7 @@ async def build_vertex(
return return
start_time = time.perf_counter() start_time = time.perf_counter()
try: try:
if isinstance(vertex, StatelessVertex) or not vertex._built: if isinstance(vertex, Vertex) or not vertex._built:
await vertex.build(user_id=None, session_id=sid) await vertex.build(user_id=None, session_id=sid)
params = vertex._built_object_repr() params = vertex._built_object_repr()
valid = True valid = True
@ -96,7 +96,9 @@ async def build_vertex(
) )
# Emit the vertex build response # Emit the vertex build response
response = VertexBuildResponse(valid=valid, params=params, id=vertex.id, data=result_dict) response = VertexBuildResponse(
valid=valid, params=params, id=vertex.id, data=result_dict
)
await sio.emit("vertex_build", data=response.model_dump(), to=sid) await sio.emit("vertex_build", data=response.model_dump(), to=sid)
except Exception as exc: except Exception as exc: