From e1800ff682778eeb1ada5a60ef8378b9efdcf5f3 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Sep 2023 11:54:53 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(api=5Fkey/crud.py):=20remo?= =?UTF-8?q?ve=20unused=20imports=20and=20variables=20for=20cleaner=20code?= =?UTF-8?q?=20=F0=9F=90=9B=20fix(api=5Fkey/crud.py):=20update=5Ftotal=5Fus?= =?UTF-8?q?es=20function=20now=20uses=20the=20same=20session=20as=20the=20?= =?UTF-8?q?caller=20to=20avoid=20thread=20safety=20issues?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/services/database/models/api_key/crud.py | 6 ++---- 1 file changed, 2 insertions(+), 4 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 d78da5a05..756236018 100644 --- a/src/backend/langflow/services/database/models/api_key/crud.py +++ b/src/backend/langflow/services/database/models/api_key/crud.py @@ -3,8 +3,6 @@ import secrets import threading from uuid import UUID from typing import List, Optional -from langflow.services.database.utils import session_getter -from langflow.services.utils import get_db_manager from sqlmodel import Session, select from langflow.services.database.models.api_key import ( ApiKey, @@ -67,8 +65,8 @@ def update_total_uses(session, api_key: ApiKey): """Update the total uses and last used at.""" # This is running in a separate thread to avoid slowing down the request # but session is not thread safe so we need to create a new session - db_manager = get_db_manager() - with session_getter(db_manager) as 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)