Skip to content

Commit de7376d

Browse files
authored
Fix Cohere client.embed TypeError (#1904)
1 parent 6c9eb6e commit de7376d

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

bertopic/backend/_cohere.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def embed(self,
7676
if self.batch_size is not None:
7777
embeddings = []
7878
for batch in tqdm(self._chunks(documents), disable=not verbose):
79-
response = self.client.embed(batch, **self.embed_kwargs)
79+
response = self.client.embed(texts=batch, **self.embed_kwargs)
8080
embeddings.extend(response.embeddings)
8181

8282
# Delay subsequent calls
@@ -85,7 +85,7 @@ def embed(self,
8585

8686
# Extract embeddings all at once
8787
else:
88-
response = self.client.embed(documents, **self.embed_kwargs)
88+
response = self.client.embed(texts=documents, **self.embed_kwargs)
8989
embeddings = response.embeddings
9090
return np.array(embeddings)
9191

0 commit comments

Comments
 (0)