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

Add STARTER_FOLDER_NAME filter to read_flows function

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-06 11:37:19 -03:00
commit 308c27c381
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -10,6 +10,7 @@ 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.initial_setup.setup import STARTER_FOLDER_NAME
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 ( from langflow.services.database.models.flow import (
Flow, Flow,
@ -57,7 +58,9 @@ def read_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( example_flows = session.exec(
select(Flow).where(Flow.user_id == None) select(Flow).where(
Flow.user_id == None, Flow.folder == STARTER_FOLDER_NAME
)
).all() # noqa ).all() # noqa
flows.extend(example_flows) flows.extend(example_flows)
except Exception as e: except Exception as e: