search

vanna-ai / vanna

🤖 Chat with your SQL database 📊. Accurate Text-to-SQL Generation via LLMs using RAG 🔄.
https://vanna.ai/docs/
MIT License
11.04k stars 862 forks source link

[Scalabrese] Added collection_metadata parameter for ChromaDB #349

Closed scalabreseGD closed 5 months ago

scalabreseGD commented 6 months ago

Added parameter for ChromaDB collection_metadata. This will allow the user to change metadatasa during the creation (e.g. distance function)

andreped commented 6 months ago

@scalabreseGD This solution is quite elegant. I like it. Surprised we didnt do this from the get-go :P Great work!

@zainhoda What do you think?

zainhoda commented 5 months ago

@scalabreseGD This solution is quite elegant. I like it. Surprised we didnt do this from the get-go :P Great work!

@zainhoda What do you think?

Yeah, this is much cleaner code lol. Great job! I think we were just bolting on more and more options as the requests came in.

For the Discord bot:

This is the relevant chroma documentation on what the collection_metadata parameter should contain: https://docs.trychroma.com/usage-guide#changing-the-distance-function