pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-24 07:02:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Improve variable creation logic in VariableService (#1875)

Browse Source 
* refactor: Improve variable creation logic in VariableService
This commit is contained in:
Gabriel Luiz Freitas Almeida
2024-05-10 12:34:24 -03:00
committed by GitHub
parent e36901bb34
commit da256785d0
1 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/backend/base/langflow/services/variable/service.py
View File

@ -29,17 +29,20 @@ class VariableService(Service):
for var in self.settings_service.settings.variables_to_get_from_environment:
if var in os.environ:
logger.debug(f"Creating {var} variable from environment.")
try:
self.create_variable(
user_id=user_id,
name=var,
value=os.environ[var],
default_fields=[],
_type="Credential",
session=session,
)
except Exception as e:
logger.error(f"Error creating {var} variable: {e}")
if not session.exec(
select(Variable).where(Variable.user_id == user_id, Variable.name == var)
).first():
try:
self.create_variable(
user_id=user_id,
name=var,
value=os.environ[var],
default_fields=[],
_type="Credential",
session=session,
)
except Exception as e:
logger.error(f"Error creating {var} variable: {e}")
else:
logger.info("Skipping environment variable storage.")