Qdrant not working
icfai-aniketsingh opened this issue · comments
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
Was going through the code and found below points:
- Method is not defined in the qdrant.py with documentation_collection
- 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.
Thanks for reporting!
I see the docs have been updated with the proper imports.
@icfai-aniketsingh, could you please confirm if they work for you now?