Update EMBEDDINGS_NAME in settings.py and test_vector_store.py

pull/825/head
Alex 4 months ago
parent b3149def82
commit b45fd58944

@ -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"

@ -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)

Loading…
Cancel
Save