diff --git a/application/core/settings.py b/application/core/settings.py index 2d4169e..42dea0f 100644 --- a/application/core/settings.py +++ b/application/core/settings.py @@ -8,7 +8,7 @@ current_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__ class Settings(BaseSettings): LLM_NAME: str = "docsgpt" - EMBEDDINGS_NAME: str = "openai_text-embedding-ada-002" + EMBEDDINGS_NAME: str = "huggingface_sentence-transformers/all-mpnet-base-v2" CELERY_BROKER_URL: str = "redis://localhost:6379/0" CELERY_RESULT_BACKEND: str = "redis://localhost:6379/1" MONGO_URI: str = "mongodb://localhost:27017/docsgpt" diff --git a/tests/test_vector_store.py b/tests/test_vector_store.py index 297a225..6b700dc 100644 --- a/tests/test_vector_store.py +++ b/tests/test_vector_store.py @@ -14,6 +14,6 @@ def test_init_local_faiss_store_huggingface(): index.faiss file in the application/ folder results in a dimension mismatch error. """ - settings.EMBEDDINGS_NAME = "huggingface_sentence-transformers/all-mpnet-base-v2" + settings.EMBEDDINGS_NAME = "openai_text-embedding-ada-002" with pytest.raises(ValueError): FaissStore("application/", "", None)