Agentic_RAG / session_functions.py
fansa34's picture
first version without fastapi
1739591
raw
history blame contribute delete
999 Bytes
from global_settings import SESSION_FILE, STORAGE_PATH
import yaml
import os
def save_session(state):
state_to_save = {key: value for key, value in state.items()}
with open(SESSION_FILE, 'w') as file:
yaml.dump(state_to_save, file)
def load_session(state):
if os.path.exists(SESSION_FILE):
with open(SESSION_FILE, 'r') as file:
try:
loaded_state = yaml.safe_load(file) or {}
for key, value in loaded_state.items():
state[key] = value
return True
except yaml.YAMLError as e:
return False
return False
def delete_session(state):
if os.path.exists(SESSION_FILE):
os.remove(SESSION_FILE)
for filename in os.listdir(STORAGE_PATH):
file_path = os.path.join(STORAGE_PATH, filename)
if os.path.isfile(file_path) or os.path.islink(file_path):
os.remove(file_path)
for key in list(state.keys()):
del state[key]