Skip to content

Commit 3db8055

Browse files
committed
Refactor text embedding query handling
1 parent 63aea62 commit 3db8055

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

fastembed/text/text_embedding_base.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,6 @@ def query_embed(self, query: Union[str, Iterable[str]], **kwargs) -> Iterable[np
5252

5353
# This is model-specific, so that different models can have specialized implementations
5454
if isinstance(query, str):
55-
yield from self.embed([f"Represent this sentence for searching relevant passages: {query}"], **kwargs)
55+
yield from self.embed([query], **kwargs)
5656
if isinstance(query, Iterable):
57-
yield from self.embed([f"Represent this sentence for searching relevant passages:: {text}" for text in query], **kwargs)
57+
yield from self.embed(query, **kwargs)

0 commit comments

Comments
 (0)