Add query prefix for Stella base model (#1023)

marqo-ai · Oct 28, 2024 · d55d6da · d55d6da
1 parent 9ac15c5
commit d55d6da
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/marqo/marqo_docs.py b/src/marqo/marqo_docs.py
@@ -78,5 +78,4 @@ def search_api_score_modifiers_parameter():
 
 
 def hugging_face_trust_remote_code():
-    # TODO - Update to valid docs URL
-    return _build_url('reference/api/models/hugging-face-models/#trust-remote-code')
+    return _build_url('models/marqo/bring-your-own-model/#bring-your-own-hugging-face-sentence-transformers-models')
diff --git a/src/marqo/s2_inference/model_registry.py b/src/marqo/s2_inference/model_registry.py
@@ -900,7 +900,8 @@ def _get_hf_properties() -> Dict:
                 "dimensions": 1024,
                 "tokens": 512,
                 "type": "hf_stella",
-                "trustRemoteCode": True
+                "trustRemoteCode": True,
+                "text_query_prefix": "Instruct: Given a web search query, retrieve relevant passages that answer the query.\nQuery: "
             },
     }
     return HF_MODEL_PROPERTIES