ModuleNotFoundError: No module named 'vanna.ollama'

[elvisbegovic]

Hi, an idea why :

1. following https://vanna.ai/docs/mssql-ollama-chromadb/
2. i run pip install 'vanna[chromadb]'
3. add script with

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)

4. run script then result :

Traceback (most recent call last):
  File "test.py", line 1, in <module>
    from vanna.ollama import Ollama
ModuleNotFoundError: No module named 'vanna.ollama'

[RodolfoMontenegro]

Try:
from vanna.ollama.ollama import Ollama

[zainhoda]

@elvisbegovic is your filename called vanna.py by any chance? If so, rename it to anything else and it should work.

[Herz3h]

lmao thanks

[elvisbegovic]

my mistake: installed under WSL while VS conding under Windows.🤦‍♂️