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

fix: correct vertex build delete function (#3289)

Browse source 
* refactor: Rearrange imports for clarity and consistency.

* fix: delete now executes the sql statement correctly
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-08-12 16:21:27 -03:00 • committed by GitHub
commit 775d659380
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/services/database/models/vertex_builds/crud.py
View file

@ -2,8 +2,7 @@ from typing import Optional
from uuid import UUID from uuid import UUID
from sqlalchemy.exc import IntegrityError from sqlalchemy.exc import IntegrityError
from sqlmodel import Session, select, col from sqlmodel import Session, col, delete, select
from sqlalchemy import delete
from langflow.services.database.models.vertex_builds.model import VertexBuildBase, VertexBuildTable from langflow.services.database.models.vertex_builds.model import VertexBuildBase, VertexBuildTable
@ -32,5 +31,5 @@ def log_vertex_build(db: Session, vertex_build: VertexBuildBase) -> VertexBuildT
def delete_vertex_builds_by_flow_id(db: Session, flow_id: UUID) -> None: def delete_vertex_builds_by_flow_id(db: Session, flow_id: UUID) -> None:
delete(VertexBuildTable).where(VertexBuildTable.flow.has(id=flow_id)) # type: ignore db.exec(delete(VertexBuildTable).where(VertexBuildTable.flow_id == flow_id))
db.commit() db.commit()