diff --git a/defog_utils/utils_llm.py b/defog_utils/utils_llm.py index de3521d..994163a 100644 --- a/defog_utils/utils_llm.py +++ b/defog_utils/utils_llm.py @@ -52,7 +52,7 @@ def __post_init__(self): if len(potential_model_names) == 1: model_name = potential_model_names[0] - else: + elif len(potential_model_names) > 1: # if there are multiple potential matches, then find the one with the longest prefix model_name = max(potential_model_names, key=len)