diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-23 00:51:15 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-23 00:51:15 +0200 |
| commit | 694a4ad0e5a9e4c7eb6d11fff5ae414292ef8169 (patch) | |
| tree | 9d6786c180e02dafd7c6c305d5e01808d571f22b | |
| parent | d66e43b5c8678fc3e8d9f459f861fdeb9652f7fb (diff) | |
Prepend query if encoder is mxbai
| -rw-r--r-- | rag/retriever/encoder.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/rag/retriever/encoder.py b/rag/retriever/encoder.py index db69ee5..68d0e26 100644 --- a/rag/retriever/encoder.py +++ b/rag/retriever/encoder.py @@ -39,5 +39,6 @@ class Encoder: def encode_query(self, query: str) -> List[StrictFloat]: log.debug(f"Encoding query: {query}") - query = self.query_prompt + query + if self.model == "mxbai-embed-large": + query = self.query_prompt + query return self.__encode(query) |