vanna-ai / vanna

🤖 Chat with your SQL database 📊. Accurate Text-to-SQL Generation via LLMs using RAG 🔄.

Home Page:https://vanna.ai/docs/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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:

  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.

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?