Spaces:
Sleeping
Sleeping
File size: 768 Bytes
91c6bea 40ca01e ba900f0 40ca01e ba900f0 40ca01e ba900f0 82b6d3b 40ca01e 91c6bea 40ca01e 91c6bea ba900f0 40ca01e ba900f0 40ca01e ba900f0 91c6bea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from langchain_community.vectorstores import SKLearnVectorStore
from langchain_community.embeddings.fastembed import FastEmbedEmbeddings
class VectorStoreManager:
def __init__(self, embedding_model="BAAI/bge-base-en-v1.5"):
self.embeddings = FastEmbedEmbeddings(model_name=embedding_model)
def create_vector_store(self):
"""Create a new vector store"""
vector_store = SKLearnVectorStore(
metric="cosine",
embedding=self.embeddings,
)
return vector_store
def index_documents(self, documents):
"""Index documents into vector store"""
vector_store = self.create_vector_store()
vector_store.add_documents(documents=documents)
return vector_store
|