Skip to content

Commit 88f0310

Browse files
committed
fix: update embedding method to use embed_documents for improved performance
1 parent 15b1466 commit 88f0310

File tree

1 file changed

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

1 file changed

+1
-3
lines changed

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

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,4 @@ def embed_query(self, text: str) -> List[float]:
4444
Returns:
4545
Embeddings for the text.
4646
"""
47-
instruction_pair = f"{text}"
48-
embedding = self._embed([instruction_pair])[0]
49-
return embedding
47+
return self.embed_documents([text])[0]

0 commit comments

Comments
 (0)