From 694a4ad0e5a9e4c7eb6d11fff5ae414292ef8169 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 23 Apr 2024 00:51:15 +0200
Subject: Prepend query if encoder is mxbai

---
 rag/retriever/encoder.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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)
-- 
cgit v1.2.3-70-g09d2