BagelVectorStore#

class llama_index.vector_stores.BagelVectorStore(collection: Any, **kwargs: Any)#

Bases: VectorStore

Vector store for Bagel.

Attributes Summary

client

Get the Bagel cluster.

flat_metadata

stores_text

Methods Summary

add(nodes, **add_kwargs)

Add a list of nodes with embeddings to the vector store.

delete(ref_doc_id, **kwargs)

Delete a document from the vector store.

query(query, **kwargs)

Query the vector store.

Attributes Documentation

client#

Get the Bagel cluster.

flat_metadata: bool = True#
stores_text: bool = True#

Methods Documentation

add(nodes: List[BaseNode], **add_kwargs: Any) List[str]#

Add a list of nodes with embeddings to the vector store.

Parameters
  • nodes – List of nodes with embeddings.

  • kwargs – Additional arguments.

Returns

List of document ids.

delete(ref_doc_id: str, **kwargs: Any) None#

Delete a document from the vector store.

Parameters
  • ref_doc_id – Reference document id.

  • kwargs – Additional arguments.

query(query: VectorStoreQuery, **kwargs: Any) VectorStoreQueryResult#

Query the vector store.

Parameters
  • query – Query to run.

  • kwargs – Additional arguments.

Returns

Query result.