From dc56daebc8240df638ab9c43003738c0c99d5322 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 25 Jan 2024 13:05:48 -0300 Subject: [PATCH] Delete Flow model and related classes --- .../services/database/models/flow/flow.py | 54 ------------------- 1 file changed, 54 deletions(-) delete mode 100644 src/backend/langflow/services/database/models/flow/flow.py diff --git a/src/backend/langflow/services/database/models/flow/flow.py b/src/backend/langflow/services/database/models/flow/flow.py deleted file mode 100644 index e578f37c4..000000000 --- a/src/backend/langflow/services/database/models/flow/flow.py +++ /dev/null @@ -1,54 +0,0 @@ -# Path: src/backend/langflow/database/models/flow.py - -from langflow.services.database.models.base import SQLModelSerializable -from pydantic import validator - -from sqlmodel import Field, JSON, Column, Relationship -from uuid import UUID, uuid4 -from typing import Dict, Optional, TYPE_CHECKING - -if TYPE_CHECKING: - from langflow.services.database.models.user import User - - -class FlowBase(SQLModelSerializable): - name: str = Field(index=True) - description: Optional[str] = Field(index=True) - data: Optional[Dict] = Field(default=None, nullable=True) - - @validator("data") - def validate_json(v): - if not v: - return v - if not isinstance(v, dict): - raise ValueError("Flow must be a valid JSON") - - # data must contain nodes and edges - if "nodes" not in v.keys(): - raise ValueError("Flow must have nodes") - if "edges" not in v.keys(): - raise ValueError("Flow must have edges") - - return v - - -class Flow(FlowBase, table=True): - id: UUID = Field(default_factory=uuid4, primary_key=True, unique=True) - data: Optional[Dict] = Field(default=None, sa_column=Column(JSON)) - user_id: UUID = Field(index=True, foreign_key="user.id") - user: "User" = Relationship(back_populates="flows") - - -class FlowCreate(FlowBase): - user_id: Optional[UUID] = None - - -class FlowRead(FlowBase): - id: UUID - user_id: UUID = Field() - - -class FlowUpdate(SQLModelSerializable): - name: Optional[str] = None - description: Optional[str] = None - data: Optional[Dict] = None