Qdrant not working

[icfai-aniketsingh]

Describe the bug With the use of qdrant vector database we are getting the attributes error as below"

To Reproduce Steps to reproduce the behavior: Pasting jupyter code :

from vanna.ollama import Ollama
from vanna.qdrant import Qdrant_VectorStore
from qdrant_client import QdrantClient

class MyVanna(Qdrant_VectorStore, Ollama):
    def __init__(self, config=None):
        Qdrant_VectorStore.__init__(self, config=config)
        Ollama.__init__(self, config={'model': 'mistral'})

vn = MyVanna(config={'client': QdrantClient(url="http://localhost:6333/")})

vn.connect_to_mysql(host='192.168.96.1', dbname='rag', user='wsl_root', password='password', port=3306)

# The information schema query may need some tweaking depending on your database. This is a good starting point.
df_information_schema = vn.run_sql("SELECT * FROM INFORMATION_SCHEMA.COLUMNS")

# This will break up the information schema into bite-sized chunks that can be referenced by the LLM
plan = vn.get_training_plan_generic(df_information_schema)
plan

# If you like the plan, then uncomment this and run it to train
vn.train(plan=plan)

from vanna.flask import VannaFlaskApp
app = VannaFlaskApp(vn)
app.allow_llm_to_see_data = True
app.run()

Expected behavior Training data should show all the trained data

Error logs/Screenshots AttributeError AttributeError: 'MyVanna' object has no attribute 'documentation_collection'

Desktop (please complete the following information where):

• OS: Ubuntu
• Version: 22.04]
• Python: 3.10.12
• Vanna: 0.4.2

[icfai-aniketsingh]

Was going through the code and found below points:

1. Method is not defined in the qdrant.py with documentation_collection
2. Import is not correct, examples is having from vanna.qdrant import Qdrant but code is working with from vanna.qdrant import Qdrant_VectorStore as method defined with the same.

[zainhoda]

Thanks for reporting!

[Anush008]

I see the docs have been updated with the proper imports. @icfai-aniketsingh, could you please confirm if they work for you now?