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

Format flows.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-06 11:22:59 -03:00
commit 91a98ae877
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/backend/langflow/api/v1/flows.py
View file

@ -11,8 +11,12 @@ from sqlmodel import Session, select
from langflow.api.utils import remove_api_keys, validate_is_component from langflow.api.utils import remove_api_keys, validate_is_component
from langflow.api.v1.schemas import FlowListCreate, FlowListRead from langflow.api.v1.schemas import FlowListCreate, FlowListRead
from langflow.services.auth.utils import get_current_active_user from langflow.services.auth.utils import get_current_active_user
from langflow.services.database.models.flow import (Flow, FlowCreate, FlowRead, from langflow.services.database.models.flow import (
FlowUpdate) Flow,
FlowCreate,
FlowRead,
FlowUpdate,
)
from langflow.services.database.models.user.model import User from langflow.services.database.models.user.model import User
from langflow.services.deps import get_session, get_settings_service from langflow.services.deps import get_session, get_settings_service
@ -52,7 +56,9 @@ def read_flows(
flows = validate_is_component(flows) flows = validate_is_component(flows)
# with the session get the flows that DO NOT have a user_id # with the session get the flows that DO NOT have a user_id
try: try:
example_flows = session.exec(select(Flow).where(Flow.user_id == None)).all() example_flows = session.exec(
select(Flow).where(Flow.user_id == None)
).all() # noqa
flows.extend(example_flows) flows.extend(example_flows)
except Exception as e: except Exception as e:
logger.error(e) logger.error(e)