diff options
Diffstat (limited to 'rag/retriever')
| -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) |