diff --git a/AgentReact/start.py b/AgentReact/start.py index 66d1807..6ad670a 100644 --- a/AgentReact/start.py +++ b/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() \ No newline at end of file diff --git a/mlflow.db b/mlflow.db new file mode 100644 index 0000000..f939e67 Binary files /dev/null and b/mlflow.db differ diff --git a/requirements.txt b/requirements.txt index 78de078..a96d410 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,4 +15,5 @@ unstructured[all-docs]>=0.18.21 ipywidgets peft - python-dotenv \ No newline at end of file + python-dotenv + mlflow[genai] \ No newline at end of file