tiennm99/DocsGPT
1
0 Fork 0
mirror of https://github.com/tiennm99/DocsGPT.git synced 2026-10-18 04:14:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update EMBEDDINGS_NAME in settings.py and test_vector_store.py

Browse Source
This commit is contained in:
Alex committed 2024-01-09 00:34:04 +00:00
1 parent b3149def82
commit b45fd58944
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

application/core/settings.py
+1 -1
View File
@@ -8,7 +8,7 @@ current_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__
class Settings(BaseSettings): class Settings(BaseSettings):
LLM_NAME: str = "docsgpt" 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_BROKER_URL: str = "redis://localhost:6379/0"
CELERY_RESULT_BACKEND: str = "redis://localhost:6379/1" CELERY_RESULT_BACKEND: str = "redis://localhost:6379/1"
MONGO_URI: str = "mongodb://localhost:27017/docsgpt" MONGO_URI: str = "mongodb://localhost:27017/docsgpt"
tests/test_vector_store.py
+1 -1
View File
@@ -14,6 +14,6 @@ def test_init_local_faiss_store_huggingface():
index.faiss file in the application/ folder results in a index.faiss file in the application/ folder results in a
dimension mismatch error. 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): with pytest.raises(ValueError):
FaissStore("application/", "", None) FaissStore("application/", "", None)