UB-AICLUB / RagBot

A discord bot using RAG methodology for the club
MIT License
5 stars 18 forks source link

ChromaDB #5

Open UB-AICLUB opened 1 month ago

UB-AICLUB commented 1 month ago

handle all chromadb operations

There should be a file in task_queue folder called db.py

3 functions in total

get_connection() return a connection object to our chromadb instance hard code connection parameters will set them to correct values later

write(embedding, text, metadata) - will write the embeddings, text and metadata to chromadb add any parameters required but mention in the comment

retrieve(embedding, filter=None) - retrieve to n closest embeddings, along with their metadata mention the format in the comments

aayushsss1 commented 1 month ago

Hey @UB-AICLUB I can take this up!