From 1a4fd7c33f90889319e5e8ef7c13ebaaef0c258c Mon Sep 17 00:00:00 2001 From: trducng Date: Mon, 5 Feb 2024 18:04:36 +0700 Subject: [PATCH] Update default settings to conform Langchain's Azure implementation --- libs/ktem/flowsettings.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/libs/ktem/flowsettings.py b/libs/ktem/flowsettings.py index 8281dc1..ddf4425 100644 --- a/libs/ktem/flowsettings.py +++ b/libs/ktem/flowsettings.py @@ -28,8 +28,8 @@ KH_LLMS = { "def": { "__type__": "kotaemon.llms.AzureChatOpenAI", "temperature": 0, - "azure_endpoint": config("OPENAI_API_BASE", default=""), - "openai_api_key": config("OPENAI_API_KEY", default=""), + "azure_endpoint": config("AZURE_OPENAI_ENDPOINT", default=""), + "openai_api_key": config("AZURE_OPENAI_API_KEY", default=""), "openai_api_version": config("OPENAI_API_VERSION", default=""), "deployment_name": "dummy-q2", "stream": True, @@ -42,8 +42,8 @@ KH_LLMS = { "def": { "__type__": "kotaemon.llms.AzureChatOpenAI", "temperature": 0, - "azure_endpoint": config("OPENAI_API_BASE", default=""), - "openai_api_key": config("OPENAI_API_KEY", default=""), + "azure_endpoint": config("AZURE_OPENAI_ENDPOINT", default=""), + "openai_api_key": config("AZURE_OPENAI_API_KEY", default=""), "openai_api_version": config("OPENAI_API_VERSION", default=""), "deployment_name": "dummy-q2", "request_timeout": 10, @@ -59,8 +59,8 @@ KH_EMBEDDINGS = { "def": { "__type__": "kotaemon.embeddings.AzureOpenAIEmbeddings", "model": "text-embedding-ada-002", - "azure_endpoint": config("OPENAI_API_BASE", default=""), - "openai_api_key": config("OPENAI_API_KEY", default=""), + "azure_endpoint": config("AZURE_OPENAI_ENDPOINT", default=""), + "openai_api_key": config("AZURE_OPENAI_API_KEY", default=""), "deployment": "dummy-q2-text-embedding", "chunk_size": 16, },