From 966d0ec734df0784467be9ce53495f5b691abbbb Mon Sep 17 00:00:00 2001 From: Siddharth Balyan Date: Thu, 2 May 2024 18:44:10 +0530 Subject: [PATCH] use VALID_MODELS to support JULEP_MODELS --- agents-api/agents_api/model_registry.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agents-api/agents_api/model_registry.py b/agents-api/agents_api/model_registry.py index e5a72c7fc..350b5559b 100644 --- a/agents-api/agents_api/model_registry.py +++ b/agents-api/agents_api/model_registry.py @@ -111,7 +111,7 @@ def validate_configuration(model: str): """ if model not in ALL_AVAILABLE_MODELS: raise AgentModelNotValid(model, ALL_AVAILABLE_MODELS) - elif model not in get_valid_models(): + elif model not in VALID_MODELS: raise MissingAgentModelAPIKeyError(model)