From b0b2710d854468d309c5b92b9cb778584fd6640f Mon Sep 17 00:00:00 2001 From: Charles Packer Date: Sun, 29 Oct 2023 21:04:49 -0700 Subject: [PATCH] azure typo patch (#192) * azure typo patch * Make azure embeddings var consistent --------- Co-authored-by: Vivian Fang --- memgpt/openai_tools.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/memgpt/openai_tools.py b/memgpt/openai_tools.py index 23cd9d0551..7a5f1f316e 100644 --- a/memgpt/openai_tools.py +++ b/memgpt/openai_tools.py @@ -179,8 +179,8 @@ def get_set_azure_env_vars(): ("AZURE_OPENAI_VERSION", os.getenv("AZURE_OPENAI_VERSION")), ("AZURE_OPENAI_DEPLOYMENT", os.getenv("AZURE_OPENAI_DEPLOYMENT")), ( - "AZURE_OPENAI_EMBEDDING_DEPLOYMENT", - os.getenv("AZURE_OPENAI_EMBEDDING_DEPLOYMENT"), + "AZURE_OPENAI_EMBEDDINGS_DEPLOYMENT", + os.getenv("AZURE_OPENAI_EMBEDDINGS_DEPLOYMENT"), ), ] return [x for x in azure_env_variables if x[1] is not None] @@ -211,7 +211,7 @@ def configure_azure_support(): def check_azure_embeddings(): azure_openai_deployment = os.getenv("AZURE_OPENAI_DEPLOYMENT") - azure_openai_embedding_deployment = os.getenv("AZURE_OPENAI_EMBEDDING_DEPLOYMENT") + azure_openai_embedding_deployment = os.getenv("AZURE_OPENAI_EMBEDDINGS_DEPLOYMENT") if azure_openai_deployment is not None and azure_openai_embedding_deployment is None: raise ValueError( f"Error: It looks like you are using Azure deployment ids and computing embeddings, make sure you are setting one for embeddings as well. Please see README section on Azure"