From 8b598ac10372093a817dc95652b329f3a3fe228d Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Mon, 17 Jun 2024 18:49:18 -0300 Subject: [PATCH] refactor: Fix variable type check in VariableService --- src/backend/base/langflow/services/variable/service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/base/langflow/services/variable/service.py b/src/backend/base/langflow/services/variable/service.py index b2389e890..b3bde028f 100644 --- a/src/backend/base/langflow/services/variable/service.py +++ b/src/backend/base/langflow/services/variable/service.py @@ -61,7 +61,7 @@ class VariableService(Service): # credential = session.query(Variable).filter(Variable.user_id == user_id, Variable.name == name).first() variable = session.exec(select(Variable).where(Variable.user_id == user_id, Variable.name == name)).first() - if variable.type == "Credential" and field == "session_id": + if variable and variable.type == "Credential" and field == "session_id": raise TypeError( f"variable {name} of type 'Credential' cannot be used in a Session ID field " "because its purpose is to prevent the exposure of values."