feat: bedrock refactoring

deepset-ai · Nov 24, 2023 · 6d11af3 · 6d11af3
1 parent 98f997a
commit 6d11af3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/haystack/nodes/retriever/dense.py b/haystack/nodes/retriever/dense.py
@@ -1893,12 +1893,12 @@ def _infer_model_format(model_name_or_path: str, use_auth_token: Optional[Union[
         )
         if valid_openai_model_name:
             return "openai"
-        elif model_name_or_path in COHERE_EMBEDDING_MODELS:
+        if model_name_or_path in COHERE_EMBEDDING_MODELS:
             return "cohere"
-        elif model_name_or_path == "bedrock":
+        if model_name_or_path == "bedrock":
             return "bedrock"
         # Check if model name is a local directory with sentence transformers config file in it
-        elif Path(model_name_or_path).exists():
+        if Path(model_name_or_path).exists():
             if Path(f"{model_name_or_path}/config_sentence_transformers.json").exists():
                 return "sentence_transformers"
         # Check if sentence transformers config file in model hub