From 0966a3004387de7470955aafc3f5787920cefea0 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 1 Apr 2024 09:17:20 -0300 Subject: [PATCH] Refactor load_session method to handle coroutine result --- src/backend/base/langflow/services/session/service.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/services/session/service.py b/src/backend/base/langflow/services/session/service.py index 0bf4412e5..0c925db19 100644 --- a/src/backend/base/langflow/services/session/service.py +++ b/src/backend/base/langflow/services/session/service.py @@ -15,7 +15,10 @@ class SessionService(Service): async def load_session(self, key, flow_id: str, data_graph: Optional[dict] = None): # Check if the data is cached if key in self.cache_service: - return self.cache_service.get(key) + result = self.cache_service.get(key) + if isinstance(result, Coroutine): + result = await result + return result if key is None: key = self.generate_key(session_id=None, data_graph=data_graph)