Mlflow débug

AgentReact/start.py

@@ -2,9 +2,14 @@ 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()

requirements.txt

@@ -16,3 +16,4 @@
     ipywidgets
     peft
     python-dotenv
+    mlflow[genai]