From ebfce9d8ea70d538c028dcfdbe77dd60b491d53a Mon Sep 17 00:00:00 2001 From: Jordan Frazier <122494242+jordanrfrazier@users.noreply.github.com> Date: Wed, 11 Dec 2024 04:24:39 -0800 Subject: [PATCH] feat: add optional env var to skip starter project updates (#5193) * Add optional env var to skip starter project updates * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> --- .../base/langflow/initial_setup/setup.py | 20 ++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/src/backend/base/langflow/initial_setup/setup.py b/src/backend/base/langflow/initial_setup/setup.py index 7a356cd5ce..117581c410 100644 --- a/src/backend/base/langflow/initial_setup/setup.py +++ b/src/backend/base/langflow/initial_setup/setup.py @@ -1,6 +1,7 @@ import asyncio import copy import json +import os import shutil from collections import defaultdict from copy import deepcopy @@ -628,15 +629,16 @@ async def create_or_update_starter_projects(all_types_dict: dict) -> None: project_gradient, project_tags, ) = get_project_data(project) - updated_project_data = update_projects_components_with_latest_component_versions( - project_data.copy(), all_types_dict - ) - updated_project_data = update_edges_with_latest_component_versions(updated_project_data) - if updated_project_data != project_data: - project_data = updated_project_data - # We also need to update the project data in the file - - await update_project_file(project_path, project, updated_project_data) + do_update_starter_projects = os.environ.get("LANGFLOW_UPDATE_STARTER_PROJECTS", "true").lower() == "true" + if do_update_starter_projects: + updated_project_data = update_projects_components_with_latest_component_versions( + project_data.copy(), all_types_dict + ) + updated_project_data = update_edges_with_latest_component_versions(updated_project_data) + if updated_project_data != project_data: + project_data = updated_project_data + # We also need to update the project data in the file + await update_project_file(project_path, project, updated_project_data) if project_name and project_data: for existing_project in await get_all_flows_similar_to_project(session, new_folder.id): await session.delete(existing_project)