Update to throw IndexError if model name not registered

lastmile-ai · Mar 22, 2024 · fd4c2e7 · fd4c2e7
1 parent fc9e3ae
commit fd4c2e7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/src/aiconfig/registry.py b/python/src/aiconfig/registry.py
@@ -61,6 +61,10 @@ def get_model_parser(model_id: str) -> ModelParser:
         Returns:
             ModelParser: The retrieved model parser
         """
+        if model_id not in ModelParserRegistry.parser_ids():
+            raise IndexError(
+                f"Model parser '{model_id}' not found in registry, available model parsers are:\n {ModelParserRegistry.parser_ids()}"
+            )
         return ModelParserRegistry._parsers[model_id]
 
     @staticmethod