components/init_state.py

def init_state(session_state, config):
    initial_prompt_engineering_dict = config["PROMPT_ENGINEERING_DICT"]
    if "messages" not in session_state:
        session_state.messages = []

    if "tokens_used" not in session_state:
        session_state.tokens_used = 0

    if "tps" not in session_state:
        session_state.tps = 0

    if "temp" not in session_state:
        session_state.temp = 0.8

    if "history" not in session_state:
        session_state.history = [
            [
                initial_prompt_engineering_dict["SYSTEM_INSTRUCTION"],
                initial_prompt_engineering_dict["SYSTEM_RESPONSE"],
            ]
        ]

    if "n_crawl" not in session_state:
        session_state.n_crawl = 5

    if "repetion_penalty" not in session_state:
        session_state.repetion_penalty = 1

    if "rag_enabled" not in session_state:
        session_state.rag_enabled = True

    if "chat_bot" not in session_state:
        session_state.chat_bot = "Mixtral 8x7B v0.1"

    if "search_vendor" not in session_state:
        session_state.search_vendor = "Bing"

    if "system_instruction" not in session_state:
        session_state.system_instruction = initial_prompt_engineering_dict[
            "SYSTEM_INSTRUCTION"
        ]

    if "system_response" not in session_state:
        session_state.system_instruction = initial_prompt_engineering_dict[
            "SYSTEM_RESPONSE"
        ]

    if "pre_context" not in session_state:
        session_state.pre_context = initial_prompt_engineering_dict["PRE_CONTEXT"]

    if "post_context" not in session_state:
        session_state.post_context = initial_prompt_engineering_dict["POST_CONTEXT"]

    if "pre_prompt" not in session_state:
        session_state.pre_prompt = initial_prompt_engineering_dict["PRE_PROMPT"]

    if "post_prompt" not in session_state:
        session_state.post_prompt = initial_prompt_engineering_dict["POST_PROMPT"]

    if "pass_prev" not in session_state:
        session_state.pass_prev = False

    if "chunk_size" not in session_state:
        session_state.chunk_size = 512