Merge pull request #5 from TechPrismatica/fix/tenant_id-cookie-fix

fix(cookie fix): 🐛 tenant-id cookie fix

Author: faizanazim11

sql_db_utils/__version__.py

@@ -1 +1 @@
-__version__ = "1.3.0"
+__version__ = "1.3.1"

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

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