Add checks

Author: jochenklar

rdmo_llm_views/apps.py

@@ -0,0 +1,8 @@
+from django.apps import AppConfig
+
+
+class LlmViewsConfig(AppConfig):
+    name = 'rdmo_llm_views'
+
+    def ready(self):
+        from . import checks  # noqa: F401

rdmo_llm_views/checks.py

@@ -0,0 +1,28 @@
+from django.conf import settings
+from django.core.checks import Error, register
+
+REQUIRED_SETTINGS = [
+    ('rdmo_llm_views.E001', 'LLM_VIEWS_ADAPTER', str),
+    ('rdmo_llm_views.E002', 'LLM_VIEWS_LLM_ARGS', dict),
+    ('rdmo_llm_views.E003', 'LLM_VIEWS_SELECT_MODEL', bool),
+    ('rdmo_llm_views.E004', 'LLM_VIEWS_TIMEOUT', int),
+    ('rdmo_llm_views.E005', 'Q_CLUSTER', dict),
+]
+
+
+@register()
+def check_required_settings(app_configs, **kwargs):
+    errors = []
+    sentinel = object()  # since None could be a valid value
+
+    for check_id, name, expected_type in REQUIRED_SETTINGS:
+        value = getattr(settings, name, sentinel)
+
+        if value is sentinel:
+            errors.append(Error(f'settings.{name} is not configured', id=check_id))
+            continue
+
+        if not isinstance(value, expected_type):
+            errors.append(Error(f'settings.{name} has wrong type: {type(value).__name__}', id=check_id))
+
+    return errors

rdmo_llm_views/utils.py

@@ -8,7 +8,9 @@
 
 
 def get_adapter():
-    return import_string(settings.LLM_VIEWS_ADAPTER)()
+    adapter_class = getattr(settings, 'LLM_VIEWS_ADAPTER', None)
+    if adapter_class:
+        return import_string(adapter_class)()
 
 
 def get_group(*args, **kwargs):