fastembed_text_embedding_quickstart_with_document_list.py

1from fastembed import TextEmbedding
2from typing import List
3
4# Example list of documents
5documents: List[str] = [
6    "FastEmbed is supported by and maintained by Qdrant.",
7    "This is a quickstart guide for FastEmbed.",
8    "FastEmbed is a lightweight Python library for generating text embeddings."
9]
10
11# Initialize the TextEmbedding model
12# This will download the default model (BAAI/bge-small-en-v1.5) automatically
13model = TextEmbedding()
14
15# Generate embeddings
16# The embed method returns a generator for performance efficiency
17embeddings_generator = model.embed(documents)
18
19# Convert generator to a list to see the results
20embeddings = list(embeddings_generator)
21
22print(f"Generated {len(embeddings)} embeddings.")
23print(f"Dimension of each embedding: {len(embeddings[0])}")