jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔀 chore(utils.py): add type hint for get_session() function to indicate it returns a generator instead of a Session object

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-04 07:01:05 -03:00
commit 833fb4ad8f
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/services/utils.py
View file

@ -1,5 +1,5 @@
from langflow.services import ServiceType, service_manager
from typing import TYPE_CHECKING
from typing import TYPE_CHECKING, Generator
if TYPE_CHECKING:
@ -17,7 +17,7 @@ def get_db_manager() -> "DatabaseManager":
return service_manager.get(ServiceType.DATABASE_MANAGER)
def get_session() -> "Session":
def get_session() -> Generator["Session", None, None]:
db_manager = service_manager.get(ServiceType.DATABASE_MANAGER)
yield from db_manager.get_session()