fn cosine_similarity(a: &[f32], b: &[f32]) -> f32
Calculate cosine similarity between two embeddings.