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

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
commit da256785d0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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.")