Frequently missing packages in the Edit with AI feature

[teristam]

Describe the bug

I am trying to use the "Edit with AI" feature in the in-browser notebook. There are so many package incompatibility issues. e.g.

No module named 'opentelemetry._events'

I have a complex project with lots of dependencies. uv usually can resolve them without many issues. But I guess the dependencies packages of marimo is not being pinned down precisely enough? It is quite frustrating to keep running into error when using the AI feature in-browser.

Thank you!

Will you submit a PR?

• Yes

Environment

Environment

{
  "marimo": "0.24.0",
  "editable": false,
  "location": "~/code/trialexp_notebook_learning/.venv/lib/python3.12/site-packages/marimo",
  "OS": "Linux",
  "OS Version": "6.14.0-33-generic",
  "Processor": "x86_64",
  "Python Version": "3.12.3",
  "Locale": "en_US",
  "Binaries": {
    "Browser": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/151.0.0.0 Safari/537.36 Edg/151.0.0.0",
    "Node": "v24.8.0",
    "uv": "0.5.16"
  },
  "Dependencies": {
    "click": "8.1.7",
    "docutils": "0.21.2",
    "itsdangerous": "2.2.0",
    "jedi": "0.19.1",
    "markdown": "3.6",
    "narwhals": "2.25.0",
    "packaging": "24.1",
    "psutil": "6.0.0",
    "pygments": "2.21.0",
    "pymdown-extensions": "11.0.2",
    "pyyaml": "6.0.1",
    "starlette": "0.41.3",
    "tomlkit": "0.13.2",
    "typing-extensions": "4.12.2",
    "uvicorn": "0.30.1",
    "websockets": "17.0.1"
  },
  "Optional Dependencies": {
    "loro": "1.13.2",
    "mcp": "1.29.1",
    "nbformat": "5.10.4",
    "openai": "1.109.1",
    "pandas": "2.1.4",
    "pyarrow": "17.0.0"
  },
  "Experimental Flags": {}
}

Code to reproduce

No response

[VishakBaddur]

Thanks for the report @teristam . I tried to reproduce this with clean installs across a few scenarios: marimo[recommended] + marimo[otel] + mcp==1.29.1, using both current uv (0.10.7) and the uv version from your environment (0.5.16). Every combination resolved cleanly to opentelemetry-api==1.28.2 / opentelemetry-sdk==1.28.2 with no conflicts.

Since you mentioned this is a complex project with a lot of dependencies, my best guess is that something else in your dependency tree is independently requiring a newer opentelemetry-api, which can end up conflicting with the older opentelemetry-sdk that marimo's otel extra pins.

Could you share:

• uv pip list | grep -i opentelemetry
• uv pip check
• your pyproject.toml or uv.lock, if you're able to share them

That should tell whether we're looking at a marimo pinning gap or a conflict somewhere else in the dependency tree. Thanks!

[Light2Dark]

Hi @teristam , sorry for the issue. May I know how are you running marimo (eg. marimo edit ...)? In sandbox mode? Did you add otel? Would you be able to share a minimal repro / the notebook?

[teristam]

Hi,

That's my uv pip list | grep -i opentelemetry

opentelemetry-api                  1.44.0
opentelemetry-proto                1.44.0
opentelemetry-sdk                  1.44.0
opentelemetry-semantic-conventions 0.65b0

uv pip check reveals nothing related to opentelemetry or marimo

My pyproject.toml:

[project]
name = "trialexp-notebook-learning"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.11"
dependencies = [
    "fast-fmm-rpy2>=1.0.2",
    "ipykernel>=6.29.4",
    "jupyter-ai[all]>=2.31.5",
    "jupyterlab>=4.2.2",
    "marimo>=0.24",
    "openai>=1.109.1",
    "papermill>=2.6.0",
    "patchworklib>=0.6.5",
    "plotnine>=0.13.6",
    "pydantic-ai-slim[openai]>=1.30.1",
    "pyreadr>=0.5.3",
    "rust-just>=1.40.0",
    "scikit-fda>=0.10.1",
    "trialexp",
    "websockets>=14.0",
]

[tool.uv.sources]
trialexp = { path = "../trialexp", editable = true }

trialexp is my own project with complex dependencies. uv sync is able to resolve all the dependencies without problem.

The latest error I have is No module named 'opentelemetry._events'

And I am just using marimo edit to start the notebook server. I did not use any sandbox

Thank you!

[Light2Dark]

Are you able to upgrade your pydantic-ai package? It's running an older version, we've tested marimo against 1.107 and above.

[teristam]

Thanks! I am able to get the AI features working by installing marimo[recommended].

Personally, I think since the AI feature is an important part of marimo now, and there are so many UI elements implemented for it, maybe it is better to move the AI feature dependencies to the base package instead of keeping them in extra? Or least hide the UI elements related to it when the extra dependencies are not installed to avoid confusing the users.

Thank you for making this amazing software!

[Light2Dark]

Nice! I don't think we can include in base package as some folks don't like it. But might be good if

1. Install package UI to enforce the constraints we set.
2. On all the UI elements which needs AI, it should prompt immediately when clicked. Instead of typing something and then you hit the issue.

[teristam]

I also notice that when the AI feature failed (e.g. when missing a package). the "generate" blue pill will remain in the cell unless refreshing the whole page, which confuses the user even further.

[teristam]

Nice! I don't think we can include in base package as some folks don't like it. But might be good if

1. Install package UI to enforce the constraints we set.
2. On all the UI elements which needs AI, it should prompt immediately when clicked. Instead of typing something and then you hit the issue.

It currently does prompt for missing individual package, but there are so many package dependencies for the AI feature that installing one by one when prompted is really annoying.