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