simplify code and remove redundancy

defog-ai · Dec 15, 2024 · 60e00b0 · 60e00b0
1 parent 5d98561
commit 60e00b0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/defog_utils/utils_llm.py b/defog_utils/utils_llm.py
@@ -50,9 +50,7 @@ def __post_init__(self):
                 if mname in self.model:
                     potential_model_names.append(mname)
 
-            if len(potential_model_names) == 1:
-                model_name = potential_model_names[0]
-            elif len(potential_model_names) > 1:
+            if len(potential_model_names) > 0:
                 # if there are multiple potential matches, then find the one with the longest prefix
                 model_name = max(potential_model_names, key=len)