21 lines
582 B
Python
21 lines
582 B
Python
# 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?")
|