[Bug fix] Fix Azure OpenAI related issue (#973)

2023-11-23 01:40:54 -08:00
parent f6b80e01a1
commit 33409140b4
4 changed files with 4 additions and 5 deletions
--- a/tests/apps/test_apps.py
+++ b/tests/apps/test_apps.py
@@ -119,7 +119,7 @@ class TestAppFromConfig:
        # Validate the Embedder config values
        embedder_config = config_data["embedder"]["config"]
        assert app.embedder.config.model == embedder_config["model"]
-        assert app.embedder.config.deployment_name == embedder_config["deployment_name"]
+        assert app.embedder.config.deployment_name == embedder_config.get("deployment_name")

    def test_from_opensource_config(self, mocker):
        mocker.patch("embedchain.vectordb.chroma.chromadb.Client")