def _coerce_embedding(x: NDArray[np.floating]) -> NDArray[np.float32]:
    # pgvector elements are single-precision floats; enforce dtype + 1D
    arr = np.asarray(x, dtype=np.float32)
    if arr.ndim != 1:
        raise ValueError(f"embedding must be 1-D, got shape {arr.shape}")
    return arr

from sqlalchemy import event
from pgvector.psycopg import register_vector

@event.listens_for(engine, "connect")
def _pgvector_connect(dbapi_connection, _):
    register_vector(dbapi_connection)