15 lines
443 B
Python
15 lines
443 B
Python
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() |