From f5d02f34dffe90bd4d4e0119227585d6d20dd682 Mon Sep 17 00:00:00 2001 From: Faizan Azim Date: Sat, 4 Oct 2025 21:48:28 +0000 Subject: [PATCH 1/2] fix(cookie fix): :bug: tenant-id cookie fix --- sql_db_utils/__version__.py | 2 +- sql_db_utils/asyncio/session_management.py | 2 +- sql_db_utils/session_management.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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..8f6ab27 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[str, 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 From 31ce2a5c637878cd2d057f6e5584d035d018e580 Mon Sep 17 00:00:00 2001 From: Faizan Azim Date: Sun, 5 Oct 2025 03:21:09 +0530 Subject: [PATCH 2/2] Update sql_db_utils/asyncio/session_management.py Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> --- sql_db_utils/asyncio/session_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sql_db_utils/asyncio/session_management.py b/sql_db_utils/asyncio/session_management.py index 8f6ab27..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()] = None) -> AsyncGenerator[AsyncSession, Any]: + 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