Mlflow débug
This commit is contained in:
@@ -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()
|
||||||
@@ -16,3 +16,4 @@
|
|||||||
ipywidgets
|
ipywidgets
|
||||||
peft
|
peft
|
||||||
python-dotenv
|
python-dotenv
|
||||||
|
mlflow[genai]
|
||||||
Reference in New Issue
Block a user