fix llama index validate environment error

This commit is contained in:
Krrish Dholakia 2023-09-26 17:53:22 -07:00
parent 9ee51872e6
commit 076affbff2
4 changed files with 15 additions and 3 deletions

View file

@ -1325,9 +1325,12 @@ def load_test_model(
"exception": e,
}
def validate_environment(model: str) -> dict:
def validate_environment(model: Optional[str]=None) -> dict:
keys_in_environment = False
missing_keys = []
if model is None:
return {"keys_in_environment": keys_in_environment, "missing_keys": missing_keys}
## EXTRACT LLM PROVIDER - if model name provided
custom_llm_provider = None
# check if llm provider part of model name