diff --git a/sql_db_utils/__version__.py b/sql_db_utils/__version__.py index 67bc602..9c73af2 100644 --- a/sql_db_utils/__version__.py +++ b/sql_db_utils/__version__.py @@ -1 +1 @@ -__version__ = "1.3.0" +__version__ = "1.3.1" diff --git a/sql_db_utils/asyncio/session_management.py b/sql_db_utils/asyncio/session_management.py index 7bcc9b0..1b7ef70 100644 --- a/sql_db_utils/asyncio/session_management.py +++ b/sql_db_utils/asyncio/session_management.py @@ -124,7 +124,7 @@ async def get_engine_obj( def get_db_factory(self, database: str, retrying: bool = False) -> AsyncGenerator[AsyncSession, Any]: from fastapi import Cookie - async def get_db(tenant_id: Annotated[str, Cookie]): + async def get_db(tenant_id: Annotated[Union[str, None], Cookie()] = None) -> AsyncGenerator[AsyncSession, Any]: yield await self.get_session(database=database, tenant_id=tenant_id, retrying=retrying) return get_db diff --git a/sql_db_utils/session_management.py b/sql_db_utils/session_management.py index be37210..83cf72a 100644 --- a/sql_db_utils/session_management.py +++ b/sql_db_utils/session_management.py @@ -124,7 +124,7 @@ def get_engine_obj( def get_db_factory(self, database: str, retrying: bool = False) -> Callable: from fastapi import Cookie - def get_db(tenant_id: Annotated[str, Cookie]): + def get_db(tenant_id: Annotated[str, Cookie()] = None) -> Session: yield self.get_session(database=database, tenant_id=tenant_id, retrying=retrying) return get_db