From 4998022b07aadc8b6117827938251a519615a783 Mon Sep 17 00:00:00 2001 From: Vivian Fang Date: Thu, 26 Oct 2023 14:45:25 -0700 Subject: [PATCH] cleanup --- memgpt/openai_tools.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/memgpt/openai_tools.py b/memgpt/openai_tools.py index 0c6dc01e86..20ad5f9a46 100644 --- a/memgpt/openai_tools.py +++ b/memgpt/openai_tools.py @@ -134,7 +134,7 @@ async def acreate_embedding_with_backoff(**kwargs): if azure_openai_deployment is not None: kwargs["deployment_id"] = azure_openai_deployment else: - kwargs["engine"] = "text-embedding-ada-002" + kwargs["engine"] = kwargs["model"] kwargs.pop("model") return await openai.Embedding.acreate(**kwargs) @@ -183,7 +183,6 @@ def configure_azure_support(): azure_openai_key, azure_openai_endpoint, azure_openai_version, - # azure_openai_deployment, ]: print( f"Error: missing Azure OpenAI environment variables. Please see README section on Azure."