Trouble with basic setup code for chromadb - generates numpy type error

[robot1125]

Hey all, I'm trying to run the basic setup code from the docs:

from vanna.ollama import Ollama
from vanna.chromadb import ChromaDB_VectorStore

class MyVanna(ChromaDB_VectorStore, Ollama):
    def __init__(self, config=None):
        ChromaDB_VectorStore.__init__(self, config=config)
        Ollama.__init__(self, config=config)

vn = MyVanna(config={"model": "llama3:22B"})

I'm using python 3.11.6, vanna 0.6.0, chromadb 0.5.0, and numpy 2.0.0.

Running the code above generates this error on the vanna.chromadb import line:

AttributeError: `np.float_` was removed in the NumPy 2.0 release. Use `np.float64` instead.. Did you mean: 'float16'?

Can anyone point me at a solution?

[zainhoda]

It looks like an underlying issue with Chroma:

chroma-core/chroma#2358

For now, you can try this:

pip uninstall numpy
pip install 'numpy<2.0.0'

Either chroma has to become compatible with numpy 2.0 or they need to add the specific version to their requirements.

[robot1125]

Thank you, that's the answer.