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
    
def set_rag(vectorstore, chain):
    st.session_state['vectorstore'] = vectorstore
    st.session_state['chain'] = chain