# Displaying final output format
# from IPython.display import display, Markdown, Latex

from agent_util import config_agent, configure_llm
local_llm = 'llama3.2'
llama3, llama3_json = configure_llm(local_llm, 0)

local_agent = config_agent(llama3, llama3_json)


def run_agent(query):
    output = local_agent.invoke({"question": [query]})
    print("=======")
    print(output["generation"])
    # display(Markdown(output["generation"]))


# run_agent("I want to buy 100 CEC")
run_agent("I'm Cz, I want to buy 100 CEC, how about the price?")
# run_agent("I'm Cz, How about CEC?")