INNER CODE UNIT · Python
main
wzdavid/ThinkRAG · frontend/Document_QA.py:103
def main():
st.header("Query")
if st.session_state.llm is not None:
current_llm_info = CONFIG_STORE.get(key="current_llm_info")
current_llm_settings = CONFIG_STORE.get(key="current_llm_settings")
st.caption("LLM `" + current_llm_info["service_provider"] + "` `" + current_llm_info["model"] +
"` Response mode `" + current_llm_settings["response_mode"] +
"` Top K `" + str(current_llm_settings["top_k"]) +
"` Temperature `" + str(current_llm_settings["temperature"]) +
"` Reranking `" + str(current_llm_settings["use_reranker"]) +
"` Top N `" + str(current_llm_settings["top_n"]) +
"` Reranker `" + current_llm_settings["reranker_model"] + "`"
)
if st.session_state.index_manager is not None:
if st.session_state.index_manager.check_index_exists():
st.session_state.index_manager.load_index()
st.session_state.query_engine = create_query_engine(
index=st.session_state.index_manager.index,