From 308c27c3813cc203cda21eadac9597eef1f5a356 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 6 Mar 2024 11:37:19 -0300 Subject: [PATCH] Add STARTER_FOLDER_NAME filter to read_flows function --- src/backend/langflow/api/v1/flows.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/v1/flows.py b/src/backend/langflow/api/v1/flows.py index 6af3286ae..3f5b902d2 100644 --- a/src/backend/langflow/api/v1/flows.py +++ b/src/backend/langflow/api/v1/flows.py @@ -10,6 +10,7 @@ from sqlmodel import Session, select from langflow.api.utils import remove_api_keys, validate_is_component 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.database.models.flow import ( Flow, @@ -57,7 +58,9 @@ def read_flows( # with the session get the flows that DO NOT have a user_id try: 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 flows.extend(example_flows) except Exception as e: