mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-08-11 23:29:01 +08:00
fix: TypeError: only length-1 arrays can be converted to Python scalars (#3211)
### 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) 
This commit is contained in:
parent
601a128cd3
commit
0dff64f6ad
@ -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
|
||||
)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user