Skip to content

Commit

Permalink
Fix Embedding calls in Prem AI Embeddings (#14384)
Browse files Browse the repository at this point in the history
* fix: remove unnecessary non-keyword argument self inside embeddings call

* bump patch version of premai embeddings
  • Loading branch information
Anindyadeep authored Jun 25, 2024
1 parent 5a31c8c commit 2c60950
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,6 @@ def _get_text_embedding(self, text: str) -> List[float]:
def _get_text_embeddings(self, texts: List[str]) -> List[List[float]]:
"""Get text embeddings."""
embeddings = self._premai_client.embeddings.create(
self, model=self.model_name, project_id=self.project_id, input=texts
model=self.model_name, project_id=self.project_id, input=texts
).data
return [embedding.embedding for embedding in embeddings]
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
license = "MIT"
name = "llama-index-embeddings-premai"
readme = "README.md"
version = "0.1.3"
version = "0.1.4"

[tool.poetry.dependencies]
python = ">=3.8.1,<4.0"
Expand Down

0 comments on commit 2c60950

Please sign in to comment.