NandiniN
Databricks Employee
Databricks Employee

Hi @MisterT , 

In our docs, it is mentioned we use MLflow 3(major upgrade) with GenAI monitoring enabled. Each agent endpoint is assigned an MLflow experiment, and log agent traces from the endpoint to that experiment in real-time.

Internally an  MLFLOW_EXPERIMENT_ID environment variable is set to the endpoint’s experiment. Traces are automatically configured to be written to the Databricks MLflow tracking server. But it is failing because the agent serving endpoint itself is not configured to log metrics/params to Databricks. To resolve this issue, you can try any of the following options,

1) While calling agents.deploy(<existing_params>, environment_vars={"MLFLOW_TRACKING_URI":"databricks"}) to configure the tracking URI to databricks and resolve this, or

2) Remove the log_param call from your agent code.

Also referring https://docs.databricks.com/aws/en/mlflow3/genai/getting-started/connect-environment

Thanks!