Skip to content

Commit 897eb61

Browse files
committed
refactor: simplify embedding method to use client embed call
1 parent 88f0310 commit 897eb61

File tree

1 file changed

+3
-3
lines changed
  • apps/models_provider/impl/ollama_model_provider/model

1 file changed

+3
-3
lines changed

apps/models_provider/impl/ollama_model_provider/model/embedding.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,9 @@ def embed_documents(self, texts: List[str]) -> List[List[float]]:
3131
Returns:
3232
List of embeddings, one for each text.
3333
"""
34-
instruction_pairs = [f"{text}" for text in texts]
35-
embeddings = self._embed(instruction_pairs)
36-
return embeddings
34+
return self._client.embed(
35+
self.model, texts, options=self._default_params, keep_alive=self.keep_alive
36+
)["embeddings"]
3737

3838
def embed_query(self, text: str) -> List[float]:
3939
"""Embed a query using a Ollama deployed embedding model.

0 commit comments

Comments
 (0)