Fixes error when allowed or blocked model list is passed in config (#855

) * Fixes error when allowed or blocked model list is passed in config. * Simplified null check.
jupyterlab · Jun 26, 2024 · cee178b · cee178b
1 parent 2d392b7
commit cee178b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/packages/jupyter-ai/jupyter_ai/handlers.py b/packages/jupyter-ai/jupyter_ai/handlers.py
@@ -291,10 +291,12 @@ def filter_predicate(local_model_id: str):
 
         # filter out every model w/ model ID according to allow/blocklist
         for provider in providers:
-            provider.models = list(filter(filter_predicate, provider.models))
-            provider.chat_models = list(filter(filter_predicate, provider.chat_models))
+            provider.models = list(filter(filter_predicate, provider.models or []))
+            provider.chat_models = list(
+                filter(filter_predicate, provider.chat_models or [])
+            )
             provider.completion_models = list(
-                filter(filter_predicate, provider.completion_models)
+                filter(filter_predicate, provider.completion_models or [])
             )
 
         # filter out every provider with no models which satisfy the allow/blocklist, then return