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

ruff linting

Browse Source
This commit is contained in:
Alex committed 2023-09-27 18:01:40 +01:00
1 parent b8acb860aa
commit 852de8bdfc
4 files changed
+23 -16
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

application/vectorstore/base.py
+9 -3
View File
@@ -33,11 +33,17 @@ class BaseVectorStore(ABC):
if embeddings_name == "openai_text-embedding-ada-002":
if self.is_azure_configured():
os.environ["OPENAI_API_TYPE"] = "azure"
embedding_instance = embeddings_factory[embeddings_name](model=settings.AZURE_EMBEDDINGS_DEPLOYMENT_NAME)
embedding_instance = embeddings_factory[embeddings_name](
model=settings.AZURE_EMBEDDINGS_DEPLOYMENT_NAME
)
else:
embedding_instance = embeddings_factory[embeddings_name](openai_api_key=embeddings_key)
embedding_instance = embeddings_factory[embeddings_name](
openai_api_key=embeddings_key
)
elif embeddings_name == "cohere_medium":
embedding_instance = embeddings_factory[embeddings_name](cohere_api_key=embeddings_key)
embedding_instance = embeddings_factory[embeddings_name](
cohere_api_key=embeddings_key
)
else:
embedding_instance = embeddings_factory[embeddings_name]()