diff --git a/src/alpine_bits_python/db.py b/src/alpine_bits_python/db.py index b9767fa..dfaa7ad 100644 --- a/src/alpine_bits_python/db.py +++ b/src/alpine_bits_python/db.py @@ -4,7 +4,7 @@ import os from typing import Any, AsyncGenerator, Callable, TypeVar from sqlalchemy import Boolean, Column, Date, DateTime, ForeignKey, Integer, String -from sqlalchemy.exc import DBAPIError, InternalServerError +from sqlalchemy.exc import DBAPIError from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, create_async_engine, async_sessionmaker from sqlalchemy.orm import declarative_base, relationship @@ -146,7 +146,7 @@ class ResilientAsyncSession: try: async with self.async_sessionmaker() as session: return await func(session, *args, **kwargs) - except (InternalServerError, DBAPIError) as e: + except DBAPIError as e: last_error = e error_msg = str(e).lower()