incorrect calling of mlflow.langchain.autolog

[marco-t-314]

Hi, thanks for putting this demo together. I'm following the instructions from the readme. Dependencies are correctly installed with ./setup.sh, when starting watch.sh I get the following error TypeError: autolog() got an unexpected keyword argument 'log_inputs_outputs'.

The error originates in tracing.py when calling the autolog function

mlflow.langchain.autolog(
    log_input_examples=True,
    log_model_signatures=True,
    log_models=False,  # We don't need to log models, just traces
    log_datasets=False,
    log_inputs_outputs=True,
    disable=False,
    exclusive=False,
    disable_for_unsupported_versions=False,
    silent=False
  )

According to the docs the function signature is

mlflow.langchain.autolog(disable=False, exclusive=False, disable_for_unsupported_versions=False, silent=False, log_traces=True)

When I change the function call the demo starts correctly.

The full trace is here..

Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/server/make_openapi.py", line 11, in <module>
    from .app import app
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/server/app.py", line 102, in <module>
    setup_mlflow_tracing()
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/server/tracing.py", line 33, in setup_mlflow_tracing
    mlflow.langchain.autolog(
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/mlflow/utils/autologging_utils/__init__.py", line 96, in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/mlflow/utils/autologging_utils/__init__.py", line 478, in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: autolog() got an unexpected keyword argument 'log_inputs_outputs'
Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/scripts/make_fastapi_client.py", line 49, in <module>
    main()
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/click/core.py", line 1442, in __call__
    return self.main(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/click/core.py", line 1363, in main
    rv = self.invoke(ctx)
         ^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/click/core.py", line 1226, in invoke
    return ctx.invoke(self.callback, **ctx.params)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/click/core.py", line 794, in invoke
    return callback(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/scripts/make_fastapi_client.py", line 29, in main
    run(f'uv run python -m server.make_openapi --output={openapi_input}')
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/scripts/make_fastapi_client.py", line 45, in run
    return subprocess.run(cmd, shell=True, check=True)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/.local/share/uv/python/cpython-3.12.0-macos-aarch64-none/lib/python3.12/subprocess.py", line 571, in run
    raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command 'uv run python -m server.make_openapi --output=/tmp/openapi.json' returned non-zero exit status 1.
❯ ./watch.sh
Loading .env.local
Profile adb-1657942930699789 was successfully saved
here
Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/server/make_openapi.py", line 11, in <module>
    from .app import app
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/server/app.py", line 102, in <module>
    setup_mlflow_tracing()
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/server/tracing.py", line 33, in setup_mlflow_tracing
    mlflow.langchain.autolog(
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/mlflow/utils/autologging_utils/__init__.py", line 96, in wrapper
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/mlflow/utils/autologging_utils/__init__.py", line 478, in autolog
    return _autolog(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: autolog() got an unexpected keyword argument 'log_input_examples'
Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/scripts/make_fastapi_client.py", line 49, in <module>
    main()
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/click/core.py", line 1442, in __call__
    return self.main(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/click/core.py", line 1363, in main
    rv = self.invoke(ctx)
         ^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/click/core.py", line 1226, in invoke
    return ctx.invoke(self.callback, **ctx.params)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/.venv/lib/python3.12/site-packages/click/core.py", line 794, in invoke
    return callback(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/scripts/make_fastapi_client.py", line 29, in main
    run(f'uv run python -m server.make_openapi --output={openapi_input}')
  File "/Users/jd/Documents/projects/azure_databricks/app/agent_demo/agent-monitoring-demo-app/scripts/make_fastapi_client.py", line 45, in run
    return subprocess.run(cmd, shell=True, check=True)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/jd/.local/share/uv/python/cpython-3.12.0-macos-aarch64-none/lib/python3.12/subprocess.py", line 571, in run
    raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command 'uv run python -m server.make_openapi --output=/tmp/openapi.json' returned non-zero exit status 1.