INNER CODE UNIT · Python
perform_query
wzdavid/ThinkRAG · frontend/Document_QA.py:11
def perform_query(prompt):
if not st.session_state.query_engine:
print("Index is not initialized yet")
if (not prompt) or prompt.strip() == "":
print("Query text is required")
try:
query_response = st.session_state.query_engine.query(prompt)
return query_response
except Exception as e:
# print(f"An error occurred while processing the query: {e}")
print(f"An error occurred while processing the query: {type(e).__name__}: {e}")
# https://github.com/halilergul1/QA-app
def simple_format_response_and_sources(response):
primary_response = getattr(response, 'response', '')
output = {"response": primary_response}
sources = []
if hasattr(response, 'source_nodes'):