From a6322461c047ad0f1bbeff07aace9554d5346de7 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Sep 2023 15:34:08 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(api=5Fkey/crud.py):=20fix=20?= =?UTF-8?q?variable=20name=20conflict=20in=20update=5Ftotal=5Fuses=20funct?= =?UTF-8?q?ion=20to=20prevent=20overwriting=20the=20api=5Fkey=20parameter?= =?UTF-8?q?=20=F0=9F=94=92=20chore(api=5Fkey/crud.py):=20add=20error=20han?= =?UTF-8?q?dling=20to=20update=5Ftotal=5Fuses=20function=20to=20raise=20Va?= =?UTF-8?q?lueError=20if=20API=20Key=20is=20not=20found?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../services/database/models/api_key/crud.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/services/database/models/api_key/crud.py b/src/backend/langflow/services/database/models/api_key/crud.py index 756236018..c1c6f786e 100644 --- a/src/backend/langflow/services/database/models/api_key/crud.py +++ b/src/backend/langflow/services/database/models/api_key/crud.py @@ -67,9 +67,11 @@ def update_total_uses(session, api_key: ApiKey): # but session is not thread safe so we need to create a new session with Session(session.get_bind()) as new_session: - api_key = new_session.get(ApiKey, api_key.id) - api_key.total_uses += 1 - api_key.last_used_at = datetime.datetime.now(datetime.timezone.utc) - new_session.add(api_key) + new_api_key = new_session.get(ApiKey, api_key.id) + if new_api_key is None: + raise ValueError("API Key not found") + new_api_key.total_uses += 1 + new_api_key.last_used_at = datetime.datetime.now(datetime.timezone.utc) + new_session.add(new_api_key) new_session.commit() - return api_key + return new_api_key