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

🔧 chore(manager.py): refactor load_session method to use a more descriptive variable name 'key' instead of 'session_id' for better readability

Browse source 
🔧 chore(manager.py): refactor generate_key method to use a separate build_key method for better separation of concerns
🔧 chore(manager.py): refactor update_session method to use the build_key method for better separation of concerns
🔧 chore(manager.py): refactor clear_session method to use the build_key method for better separation of concerns
🔧 chore(manager.py): add type hinting for cache_manager parameter in the __init__ method for better code clarity
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-05 12:24:07 -03:00
commit 5b6c732213
1 changed files with 13 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/backend/langflow/services/session/manager.py
View file

@ -1,19 +1,21 @@
from typing import TYPE_CHECKING
from langflow.interface.run import build_sorted_vertices from langflow.interface.run import build_sorted_vertices
from langflow.services.base import Service from langflow.services.base import Service
from langflow.services.cache.utils import compute_dict_hash from langflow.services.cache.utils import compute_dict_hash
from langflow.services.session.utils import session_id_generator from langflow.services.session.utils import session_id_generator
if TYPE_CHECKING:
from langflow.services.cache.base import BaseCacheManager
class SessionManager(Service): class SessionManager(Service):
name = "session_manager" name = "session_manager"
def __init__(self, cache_manager): def __init__(self, cache_manager):
self.cache_manager = cache_manager self.cache_manager: "BaseCacheManager" = cache_manager
def load_session(self, session_id, data_graph):
key = self.generate_key(session_id, data_graph)
def load_session(self, key, data_graph):
# Check if the data is cached # Check if the data is cached
if key in self.cache_manager: if key in self.cache_manager:
return self.cache_manager.get(key) return self.cache_manager.get(key)
@ -24,18 +26,21 @@ class SessionManager(Service):
return graph, artifacts return graph, artifacts
def build_key(self, session_id, data_graph):
json_hash = compute_dict_hash(data_graph)
return f"{session_id}{':' if session_id else ''}{json_hash}"
def generate_key(self, session_id, data_graph): def generate_key(self, session_id, data_graph):
# Hash the JSON and combine it with the session_id to create a unique key # Hash the JSON and combine it with the session_id to create a unique key
json_hash = compute_dict_hash(data_graph)
if session_id is None: if session_id is None:
# generate a 5 char session_id to concatenate with the json_hash # generate a 5 char session_id to concatenate with the json_hash
session_id = session_id_generator() session_id = session_id_generator()
return f"{session_id}{':' if session_id else ''}{json_hash}" return self.build_key(session_id, data_graph=data_graph)
def update_session(self, session_id, data_graph, value): def update_session(self, session_id, data_graph, value):
key = self.generate_key(session_id, data_graph) key = self.build_key(session_id, data_graph)
self.cache_manager.set(key, value) self.cache_manager.set(key, value)
def clear_session(self, session_id, data_graph): def clear_session(self, session_id, data_graph):
key = self.generate_key(session_id, data_graph) key = self.build_key(session_id, data_graph)
self.cache_manager.delete(key) self.cache_manager.delete(key)