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

fix(service.py): improve handling of environment variables in DatabaseVariableService to update or create variables based on changes in secret_key (#2481)

Browse source 
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-07-02 16:19:47 -03:00 • committed by GitHub
commit e424076677
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/base/langflow/services/variable/service.py
View file

@ -31,9 +31,22 @@ class DatabaseVariableService(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.")
if not session.exec(
if found_variable := session.exec(
select(Variable).where(Variable.user_id == user_id, Variable.name == var)
).first():
# Update it
value = os.environ[var]
if isinstance(value, str):
value = value.strip()
# If the secret_key changes the stored value could be invalid
# so we need to re-encrypt it
encrypted = auth_utils.encrypt_api_key(value, settings_service=self.settings_service)
found_variable.value = encrypted
session.add(found_variable)
session.commit()
else:
# Create it
try:
value = os.environ[var]
if isinstance(value, str):