We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6c9eb6e commit de7376dCopy full SHA for de7376d
1 file changed
bertopic/backend/_cohere.py
@@ -76,7 +76,7 @@ def embed(self,
76
if self.batch_size is not None:
77
embeddings = []
78
for batch in tqdm(self._chunks(documents), disable=not verbose):
79
- response = self.client.embed(batch, **self.embed_kwargs)
+ response = self.client.embed(texts=batch, **self.embed_kwargs)
80
embeddings.extend(response.embeddings)
81
82
# Delay subsequent calls
@@ -85,7 +85,7 @@ def embed(self,
85
86
# Extract embeddings all at once
87
else:
88
- response = self.client.embed(documents, **self.embed_kwargs)
+ response = self.client.embed(texts=documents, **self.embed_kwargs)
89
embeddings = response.embeddings
90
return np.array(embeddings)
91
0 commit comments