Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update default settings to conform Langchain's Azure implementation

Browse Source
This commit is contained in:
trducng 2024-02-05 18:04:36 +07:00
parent 771f074c0e
commit 1a4fd7c33f
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
libs/ktem/flowsettings.py
View File

@ -28,8 +28,8 @@ KH_LLMS = {
"def": { "def": {
"__type__": "kotaemon.llms.AzureChatOpenAI", "__type__": "kotaemon.llms.AzureChatOpenAI",
"temperature": 0, "temperature": 0,
"azure_endpoint": config("OPENAI_API_BASE", default=""), "azure_endpoint": config("AZURE_OPENAI_ENDPOINT", default=""),
"openai_api_key": config("OPENAI_API_KEY", default=""), "openai_api_key": config("AZURE_OPENAI_API_KEY", default=""),
"openai_api_version": config("OPENAI_API_VERSION", default=""), "openai_api_version": config("OPENAI_API_VERSION", default=""),
"deployment_name": "dummy-q2", "deployment_name": "dummy-q2",
"stream": True, "stream": True,
@ -42,8 +42,8 @@ KH_LLMS = {
"def": { "def": {
"__type__": "kotaemon.llms.AzureChatOpenAI", "__type__": "kotaemon.llms.AzureChatOpenAI",
"temperature": 0, "temperature": 0,
"azure_endpoint": config("OPENAI_API_BASE", default=""), "azure_endpoint": config("AZURE_OPENAI_ENDPOINT", default=""),
"openai_api_key": config("OPENAI_API_KEY", default=""), "openai_api_key": config("AZURE_OPENAI_API_KEY", default=""),
"openai_api_version": config("OPENAI_API_VERSION", default=""), "openai_api_version": config("OPENAI_API_VERSION", default=""),
"deployment_name": "dummy-q2", "deployment_name": "dummy-q2",
"request_timeout": 10, "request_timeout": 10,
@ -59,8 +59,8 @@ KH_EMBEDDINGS = {
"def": { "def": {
"__type__": "kotaemon.embeddings.AzureOpenAIEmbeddings", "__type__": "kotaemon.embeddings.AzureOpenAIEmbeddings",
"model": "text-embedding-ada-002", "model": "text-embedding-ada-002",
"azure_endpoint": config("OPENAI_API_BASE", default=""), "azure_endpoint": config("AZURE_OPENAI_ENDPOINT", default=""),
"openai_api_key": config("OPENAI_API_KEY", default=""), "openai_api_key": config("AZURE_OPENAI_API_KEY", default=""),
"deployment": "dummy-q2-text-embedding", "deployment": "dummy-q2-text-embedding",
"chunk_size": 16, "chunk_size": 16,
}, },