utils/run_agent.py

from helper.graphs import local_agent

import sys
sys.dont_write_bytecode = True

def execute_agent(query):
    output = local_agent.invoke(
        {
            "question": query
        }
    )

    return output['generation']

print(execute_agent("What is AI"))