from dotenv import load_dotenv
load_dotenv()

from langchain.messages import HumanMessage, SystemMessage, AIMessage, ToolMessage
import mlflow

from agent import getGraph

# MLFLOW
mlflow.set_experiment("TEST PROJET") # VOIR AVEC LA COMMANDE "MLFLOW SERVER"
mlflow.langchain.autolog()

out_state = getGraph().invoke({'messages':[HumanMessage("What's the price for bitcoin ?")]})
for message in out_state['messages']:
    message.pretty_print()