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

decorate functions with decorator

Browse source
This commit is contained in:
italojohnny 2024-06-20 11:12:41 -03:00 • committed by Gabriel Luiz Freitas Almeida
commit 95e7f3882c
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/api/v1/chat.py
View file

@ -15,6 +15,7 @@ from langflow.api.utils import (
format_exception_message, format_exception_message,
get_top_level_vertices, get_top_level_vertices,
parse_exception, parse_exception,
cancel_on_disconnect
) )
from langflow.api.v1.schemas import ( from langflow.api.v1.schemas import (
FlowDataRequest, FlowDataRequest,
@ -56,6 +57,7 @@ async def try_running_celery_task(vertex, user_id):
@router.post("/build/{flow_id}/vertices", response_model=VerticesOrderResponse) @router.post("/build/{flow_id}/vertices", response_model=VerticesOrderResponse)
@cancel_on_disconnect
async def retrieve_vertices_order( async def retrieve_vertices_order(
flow_id: uuid.UUID, flow_id: uuid.UUID,
data: Optional[Annotated[Optional[FlowDataRequest], Body(embed=True)]] = None, data: Optional[Annotated[Optional[FlowDataRequest], Body(embed=True)]] = None,
@ -122,6 +124,7 @@ async def retrieve_vertices_order(
@router.post("/build/{flow_id}/vertices/{vertex_id}") @router.post("/build/{flow_id}/vertices/{vertex_id}")
@cancel_on_disconnect
async def build_vertex( async def build_vertex(
flow_id: uuid.UUID, flow_id: uuid.UUID,
vertex_id: str, vertex_id: str,
@ -262,6 +265,7 @@ async def build_vertex(
@router.get("/build/{flow_id}/{vertex_id}/stream", response_class=StreamingResponse) @router.get("/build/{flow_id}/{vertex_id}/stream", response_class=StreamingResponse)
@cancel_on_disconnect
async def build_vertex_stream( async def build_vertex_stream(
flow_id: uuid.UUID, flow_id: uuid.UUID,
vertex_id: str, vertex_id: str,