### What problem does this PR solve? fix "TypeError: only length-1 arrays can be converted to Python scalars" while using cohere embedding model. ### Type of change - [ ] Bug Fix (non-breaking change which fixes an issue) tags/v0.14.0
| @@ -558,7 +558,7 @@ class CoHereEmbed(Base): | |||
| input_type="search_query", | |||
| embedding_types=["float"], | |||
| ) | |||
| return np.array([d for d in res.embeddings.float]), int( | |||
| return np.array(res.embeddings.float[0]), int( | |||
| res.meta.billed_units.input_tokens | |||
| ) | |||