Update embeddings.py

embeddings.py

@@ -9,4 +9,22 @@ def get_hf_embeddings(model_name=None):
 
     embeddings = HuggingFaceEmbeddings(model_name=model_name)
 
+    return embeddings
+
+def get_SFR_Code_embedding_model(
+    model_name="Salesforce/SFR-Embedding-Code-400M_R", device="auto"
+):
+    """Get jinaai embedding."""
+
+    # device: cpu or cuda
+    if device == "auto":
+        device = "cuda" if torch.cuda.is_available() else "cpu"
+
+    model_name = model_name
+    model_kwargs = {"device": device, "trust_remote_code": True}
+    embeddings = HuggingFaceEmbeddings(
+        model_name=model_name,
+        model_kwargs=model_kwargs,
+    )
+
     return embeddings