pkg_resources missing on LangGraph Cloud – pymilvus crashes due to stripped setuptools

[rayvishal]

Hi team 👋

I'm deploying a graph on LangGraph Cloud that uses pymilvus, but I'm encountering this runtime error:

ModuleNotFoundError: No module named 'pkg_resources'

This is coming from pymilvus, which relies on pkg_resources via:

from pkg_resources import get_distribution

Based on the logs, it looks like pkg_resources is missing because setuptools is stripped out as part of the final Docker image.

🔥 Traceback (excerpt):

File "/usr/local/lib/python3.11/site-packages/pymilvus/client/__init__.py", line 6, in <module>
    from pkg_resources import DistributionNotFound, get_distribution
ModuleNotFoundError: No module named 'pkg_resources'

🐳 Dockerfile command that removes pip and wheel:
This is the relevant part of the build output showing the removal step:

#7 [langgraph-api  3/12] RUN pip uninstall -y pip wheel && \
     rm -rf /usr/local/lib/python*/site-packages/pip* \
            /usr/local/lib/python*/site-packages/wheel* && \
     find /usr/local/bin -name "pip*" -delete

It seems setuptools is also not present at runtime (possibly removed earlier), which causes this error for packages that rely on pkg_resources.

❓Question:
Since I don’t have control over the final Dockerfile on LangGraph Cloud, is there any recommended way to:

Retain setuptools (or just pkg_resources) at runtime?

Work around this for libraries like pymilvus that depend on it?

If not, could you consider supporting a configuration option to preserve setuptools for cases like this?

Thanks for the awesome work on LangGraph — looking forward to any guidance you can provide! 🙏

[hinthornw]

I believe this constraint is removed from later versions of pymilvus!

[rayvishal]

okay.