LJ5O
633726b2a0
C'est moche, bancal, et mal foutu, mais ça compile et ça crache un rapport de stage dans un fichier
15 lines
532 B
Python
15 lines
532 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("Observe la base de documents, et génère un rapport de stage à partir de celle-ci. Ecris le dans un fichier markdown.")]})
|
|
for message in out_state['messages']:
|
|
message.pretty_print() |