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

Refactor load_session method to handle coroutine result

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-01 09:17:20 -03:00
commit 0966a30043
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/services/session/service.py
View file

@ -15,7 +15,10 @@ class SessionService(Service):
async def load_session(self, key, flow_id: str, data_graph: Optional[dict] = None): async def load_session(self, key, flow_id: str, data_graph: Optional[dict] = None):
# Check if the data is cached # Check if the data is cached
if key in self.cache_service: 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: if key is None:
key = self.generate_key(session_id=None, data_graph=data_graph) key = self.generate_key(session_id=None, data_graph=data_graph)