Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: fix azure credential configuration (#1152)

Browse Source
This commit is contained in:
Peiran Yao
2024-03-17 18:04:04 -06:00
committed by GitHub
parent b6864e0e2b
commit 7ea83fa7f0
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
memgpt/cli/cli_config.py
View File

@@ -118,6 +118,10 @@ def configure_llm_endpoint(config: MemGPTConfig, credentials: MemGPTCredentials)
)
else:
credentials.azure_key = azure_creds["azure_key"]
credentials.azure_version = azure_creds["azure_version"]
credentials.azure_endpoint = azure_creds["azure_endpoint"]
if "azure_deployment" in azure_creds:
credentials.azure_deployment = azure_creds["azure_deployment"]
credentials.azure_embedding_version = azure_creds["azure_embedding_version"]
credentials.azure_embedding_endpoint = azure_creds["azure_embedding_endpoint"]
if "azure_embedding_deployment" in azure_creds:
@@ -203,7 +207,7 @@ def get_model_options(
model_options = [obj["id"] for obj in fetched_model_options_response["data"]]
elif model_endpoint_type == "azure":
if credentials.azure_version is None:
if credentials.azure_key is None:
raise ValueError("Missing Azure key")
if credentials.azure_version is None:
raise ValueError("Missing Azure version")
@@ -469,8 +473,6 @@ def configure_embedding_endpoint(config: MemGPTConfig, credentials: MemGPTCreden
credentials.azure_key = azure_creds["azure_key"]
credentials.azure_version = azure_creds["azure_version"]
credentials.azure_embedding_endpoint = azure_creds["azure_embedding_endpoint"]
if "azure_deployment" in azure_creds:
credentials.azure_deployment = azure_creds["azure_deployment"]
credentials.save()
embedding_endpoint_type = "azure"
@@ -627,7 +629,6 @@ def configure():
# check credentials
credentials = MemGPTCredentials.load()
openai_key = get_openai_credentials()
get_azure_credentials()
MemGPTConfig.create_config_dir()