session.py

import streamlit as st

def get_rag():
    if 'vectorstore' in st.session_state and 'chain' in st.session_state:
        vectorstore = st.session_state['vectorstore']
        chain = st.session_state['chain']
        
        return vectorstore, chain
    else:
        return None, None