embeddings_df.py

def embeddings_df(urls: pd.DataFrame, embedder: URLEmbedder) -> pd.DataFrame:
    """Adds embeddings to the given URL DataFrame."""
    urls["embeddings"] = [embedder.embed(url, normalize_embeddings=True) for url in urls["url"]]
    return urls