main.py

from utils import create_index, get_agent_chain, get_prompt_and_tools, get_search_index


def index():
    create_index()
    return True

def run(question):
    index = get_search_index()

    prompt, tools = get_prompt_and_tools()

    agent_chain = get_agent_chain(prompt, tools)

    result = None

    try:
        result = agent_chain.run(question)
    except ValueError as ve:
        if "Could not parse LLM output:" in ve.args[0] and question.lower().startswith(tuple(question_starters)):
            question = question + '?'
            result = agent_chain.run(question)

    return result