Mlflow débug

This commit is contained in:
2026-02-03 15:55:53 +01:00
parent b2e84cdc9d
commit f01c060bce
3 changed files with 7 additions and 1 deletions

View File

@@ -2,9 +2,14 @@ from dotenv import load_dotenv
load_dotenv() load_dotenv()
from langchain.messages import HumanMessage, SystemMessage, AIMessage, ToolMessage from langchain.messages import HumanMessage, SystemMessage, AIMessage, ToolMessage
import mlflow
from agent import getGraph 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 ?")]}) out_state = getGraph().invoke({'messages':[HumanMessage("What's the price for bitcoin ?")]})
for message in out_state['messages']: for message in out_state['messages']:
message.pretty_print() message.pretty_print()

BIN
mlflow.db Normal file

Binary file not shown.

View File

@@ -15,4 +15,5 @@
unstructured[all-docs]>=0.18.21 unstructured[all-docs]>=0.18.21
ipywidgets ipywidgets
peft peft
python-dotenv python-dotenv
mlflow[genai]