diff --git a/routstr/proxy.py b/routstr/proxy.py index 7c63de49..8eece67b 100644 --- a/routstr/proxy.py +++ b/routstr/proxy.py @@ -83,9 +83,7 @@ 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(col(ModelRow.enabled).is_(True)) - ) + result = await session.exec(select(ModelRow).where(ModelRow.enabled == True)) override_rows = result.all() provider_result = await session.exec(select(UpstreamProviderRow)) @@ -103,7 +101,7 @@ async def refresh_model_maps() -> None: } disabled_result = await session.exec( - select(ModelRow.id).where(col(ModelRow.enabled).is_(False)) + select(ModelRow.id).where(ModelRow.enabled == False) ) disabled_model_ids = {row for row in disabled_result.all()}