Update src/rag_engine.py

added custom embedding model support

src/rag_engine.py

@@ -30,6 +30,10 @@ def get_embedding_func(model_name: str = "sentence-transformers/all-MiniLM-L6-v2
         if "openai" in model_name.lower():
             if not os.getenv("OPENAI_API_KEY"): raise ValueError("OpenAI API Key not found.")
             return OpenAIEmbeddings(model=model_name)
+
+        elif "navy-custom-models" in model_name:
+            return HuggingFaceEmbeddings(model_name=model_name)
+        
         else:
             return HuggingFaceEmbeddings(model_name=model_name)
     except Exception as e: