Skip to content

Commit

Permalink
Merge pull request #838 from sallyom/update-rag-manage-vector-db
Browse files Browse the repository at this point in the history
update rag/app/manage_vectordb.py imports
  • Loading branch information
rhatdan authored Jan 8, 2025
2 parents 52f7c72 + 4088d68 commit 146f923
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,12 @@
from chromadb import HttpClient
from chromadb.config import Settings
import chromadb.utils.embedding_functions as embedding_functions
from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
from langchain_community.embeddings import SentenceTransformerEmbeddings
from langchain_community.vectorstores import Milvus
from pymilvus import MilvusClient
from pymilvus import connections, utility


class VectorDB:
def __init__(self, vector_vendor, host, port, collection_name, embedding_model):
self.vector_vendor = vector_vendor
Expand Down

0 comments on commit 146f923

Please sign in to comment.