summaryrefslogtreecommitdiff
path: root/rag
diff options
context:
space:
mode:
Diffstat (limited to 'rag')
-rw-r--r--rag/retriever/encoder.py3
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)