From 44286067aecffbba28b0ec42fe5cef3626b808d8 Mon Sep 17 00:00:00 2001 From: Shroominic Date: Mon, 3 Nov 2025 22:24:46 +0800 Subject: [PATCH] fix typing --- routstr/proxy.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/routstr/proxy.py b/routstr/proxy.py index 8eece67b..7c63de49 100644 --- a/routstr/proxy.py +++ b/routstr/proxy.py @@ -83,7 +83,9 @@ async def refresh_model_maps() -> None: # Gather database overrides and disabled models async with create_session() as session: - result = await session.exec(select(ModelRow).where(ModelRow.enabled == True)) + result = await session.exec( + select(ModelRow).where(col(ModelRow.enabled).is_(True)) + ) override_rows = result.all() provider_result = await session.exec(select(UpstreamProviderRow)) @@ -101,7 +103,7 @@ async def refresh_model_maps() -> None: } disabled_result = await session.exec( - select(ModelRow.id).where(ModelRow.enabled == False) + select(ModelRow.id).where(col(ModelRow.enabled).is_(False)) ) disabled_model_ids = {row for row in disabled_result.all()}