Global variables implementation

src/backend/langflow/__main__.py

@@ -6,6 +6,11 @@
 
 import typer
 from dotenv import load_dotenv
+from langflow.main import setup_app
+from langflow.services.database.utils import session_getter
+from langflow.services.deps import get_db_service, get_settings_service
+from langflow.services.utils import initialize_services, initialize_settings_service
+from langflow.utils.logger import configure, logger
 from multiprocess import cpu_count # type: ignore
 from rich import box
 from rich import print as rprint
@@ -14,12 +19,6 @@
 from rich.table import Table
 from sqlmodel import select
 
-from langflow.main import setup_app
-from langflow.services.database.utils import session_getter
-from langflow.services.deps import get_db_service, get_settings_service
-from langflow.services.utils import initialize_services, initialize_settings_service
-from langflow.utils.logger import configure, logger
-
 console = Console()
 
 app = typer.Typer(no_args_is_help=True)
@@ -304,7 +303,7 @@ def run_langflow(host, port, log_level, options, app):
  app,
  host=host,
  port=port,
- log_level=log_level,
+ log_level=log_level.lower(),
  )
  else:
  from langflow.server import LangflowApplication

src/backend/langflow/api/v1/credential.py

@@ -6,7 +6,12 @@
 
 from langflow.services.auth import utils as auth_utils
 from langflow.services.auth.utils import get_current_active_user
-from langflow.services.database.models.credential import Credential, CredentialCreate, CredentialRead, CredentialUpdate
+from langflow.services.database.models.credential import (
+ Credential,
+ CredentialCreate,
+ CredentialRead,
+ CredentialUpdate,
+)
 from langflow.services.database.models.user.model import User
 from langflow.services.deps import get_session, get_settings_service
 
@@ -25,7 +30,10 @@ def create_credential(
  try:
  # check if credential name already exists
  credential_exists = session.exec(
- select(Credential).where(Credential.name == credential.name, Credential.user_id == current_user.id)
+ select(Credential).where(
+ Credential.name == credential.name,
+ Credential.user_id == current_user.id,
+ )
  ).first()
  if credential_exists:
  raise HTTPException(status_code=400, detail="Credential name already exists")
@@ -90,7 +98,7 @@ def update_credential(
  raise HTTPException(status_code=500, detail=str(e)) from e
 
 
-@router.delete("/{credential_id}", response_model=CredentialRead, status_code=200)
+@router.delete("/{credential_id}", status_code=204)
 def delete_credential(
  *,
  session: Session = Depends(get_session),
@@ -106,6 +114,5 @@ def delete_credential(
  raise HTTPException(status_code=404, detail="Credential not found")
  session.delete(db_credential)
  session.commit()
- return db_credential
  except Exception as e:
  raise HTTPException(status_code=500, detail=str(e)) from e

src/backend/langflow/graph/vertex/base.py

@@ -14,8 +14,6 @@
  Optional,
 )
 
-from loguru import logger
-
 from langflow.graph.schema import (
  INPUT_COMPONENTS,
  INPUT_FIELD_NAME,
@@ -31,6 +29,7 @@
 from langflow.utils.constants import DIRECT_TYPES
 from langflow.utils.schemas import ChatOutputResponse
 from langflow.utils.util import sync_to_async, unescape_string
+from loguru import logger
 
 if TYPE_CHECKING:
  from langflow.graph.edge.base import ContractEdge
@@ -81,6 +80,7 @@ def __init__(
  self.is_task = is_task
  self.params = params or {}
  self.parent_node_id: Optional[str] = self._data.get("parent_node_id")
+ self.load_from_db_fields: List[str] = []
  self.parent_is_top_level = False
  self.layer = None
  self.should_run = True
@@ -167,6 +167,7 @@ def __getstate__(self):
  "_built": False,
  "parent_node_id": self.parent_node_id,
  "parent_is_top_level": self.parent_is_top_level,
+ "load_from_db_fields": self.load_from_db_fields,
  "is_input": self.is_input,
  "is_output": self.is_output,
  }
@@ -196,6 +197,7 @@ def __setstate__(self, state):
  self.task_id: Optional[str] = None
  self.parent_node_id = state["parent_node_id"]
  self.parent_is_top_level = state["parent_is_top_level"]
+ self.load_from_db_fields = state["load_from_db_fields"]
  self.layer = state.get("layer")
  self.steps = state.get("steps", [self._build])
 
@@ -294,68 +296,76 @@ def _build_params(self):
  else:
  params[param_key] = self.graph.get_vertex(edge.source_id)
 
- for key, value in template_dict.items():
- if key in params:
+ load_from_db_fields = []
+ for field_name, field in template_dict.items():
+ if field_name in params:
  continue
  # Skip _type and any value that has show == False and is not code
  # If we don't want to show code but we want to use it
- if key == "_type" or (not value.get("show") and key != "code"):
+ if field_name == "_type" or (not field.get("show") and field_name != "code"):
  continue
  # If the type is not transformable to a python base class
  # then we need to get the edge that connects to this node
- if value.get("type") == "file":
+ if field.get("type") == "file":
  # Load the type in value.get('fileTypes') using
  # what is inside value.get('content')
  # value.get('value') is the file name
- if file_path := value.get("file_path"):
+ if file_path := field.get("file_path"):
  storage_service = get_storage_service()
  flow_id, file_name = file_path.split("/")
  full_path = storage_service.build_full_path(flow_id, file_name)
- params[key] = full_path
+ params[field_name] = full_path
  else:
  raise ValueError(f"File path not found for {self.display_name}")
- elif value.get("type") in DIRECT_TYPES and params.get(key) is None:
- val = value.get("value")
- if value.get("type") == "code":
+ elif field.get("type") in DIRECT_TYPES and params.get(field_name) is None:
+ val = field.get("value")
+ if field.get("type") == "code":
  try:
- params[key] = ast.literal_eval(val) if val else None
+ params[field_name] = ast.literal_eval(val) if val else None
  except Exception:
- params[key] = val
- elif value.get("type") in ["dict", "NestedDict"]:
+ params[field_name] = val
+ elif field.get("type") in ["dict", "NestedDict"]:
  # When dict comes from the frontend it comes as a
  # list of dicts, so we need to convert it to a dict
  # before passing it to the build method
  if isinstance(val, list):
- params[key] = {k: v for item in value.get("value", []) for k, v in item.items()}
+ params[field_name] = {k: v for item in field.get("value", []) for k, v in item.items()}
  elif isinstance(val, dict):
- params[key] = val
- elif value.get("type") == "int" and val is not None:
+ params[field_name] = val
+ elif field.get("type") == "int" and val is not None:
  try:
- params[key] = int(val)
+ params[field_name] = int(val)
  except ValueError:
- params[key] = val
- elif value.get("type") == "float" and val is not None:
+ params[field_name] = val
+ elif field.get("type") == "float" and val is not None:
  try:
- params[key] = float(val)
+ params[field_name] = float(val)
  except ValueError:
- params[key] = val
- elif value.get("type") == "str" and val is not None:
+ params[field_name] = val
+ params[field_name] = val
+ elif field.get("type") == "str" and val is not None:
  # val may contain escaped \n, \t, etc.
  # so we need to unescape it
  if isinstance(val, list):
- params[key] = [unescape_string(v) for v in val]
+ params[field_name] = [unescape_string(v) for v in val]
  elif isinstance(val, str):
- params[key] = unescape_string(val)
+ params[field_name] = unescape_string(val)
+ elif val is not None and val != "":
+ params[field_name] = val
+
  elif val is not None and val != "":
- params[key] = val
+ params[field_name] = val
+ if field.get("load_from_db"):
+ load_from_db_fields.append(field_name)
 
- if not value.get("required") and params.get(key) is None:
- if value.get("default"):
- params[key] = value.get("default")
+ if not field.get("required") and params.get(field_name) is None:
+ if field.get("default"):
+ params[field_name] = field.get("default")
  else:
- params.pop(key, None)
+ params.pop(field_name, None)
  # Add _type to params
  self.params = params
+ self.load_from_db_fields = load_from_db_fields
  self._raw_params = params.copy()
 
  def update_raw_params(self, new_params: Dict[str, str], overwrite: bool = False):

src/backend/langflow/interface/custom/custom_component/component.py

@@ -39,8 +39,7 @@ def __init__(self, **data):
  def __setattr__(self, key, value):
  if key == "_user_id" and hasattr(self, "_user_id"):
  warnings.warn("user_id is immutable and cannot be changed.")
- else:
- super().__setattr__(key, value)
+ super().__setattr__(key, value)
 
  @cachedmethod(cache=operator.attrgetter("cache"))
  def get_code_tree(self, code: str):
@@ -66,6 +65,12 @@ def get_function(self):
  return validate.create_function(self.code, self._function_entrypoint_name)
 
  def build_template_config(self) -> dict:
+ """
+ Builds the template configuration for the custom component.
+
+ Returns:
+ A dictionary representing the template configuration.
+ """
  if not self.code:
  return {}