✨ Feat(settings): migrate FILE_SIZE_LIMIT

helsingborg-stad · Dec 2, 2024 · 7394f9c · 7394f9c
1 parent 99531c4
commit 7394f9c
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/fai-rag-app/fai-backend/fai_backend/documents/routes.py b/fai-rag-app/fai-backend/fai_backend/documents/routes.py
@@ -79,7 +79,8 @@ async def list_view(
 
 
 @router.get('/documents/upload', response_model=list, response_model_exclude_none=True)
-def upload_view(view=Depends(get_page_template_for_logged_in_users)) -> list:
+async def upload_view(view=Depends(get_page_template_for_logged_in_users)) -> list:
+    settings_service = SettingsServiceFactory().get_service()
     return view(
         [c.Div(components=[
             c.Div(components=[
@@ -101,7 +102,7 @@ def upload_view(view=Depends(get_page_template_for_logged_in_users)) -> list:
                             title=_('file', 'File'),
                             required=True,
                             multiple=True,
-                            file_size_limit=settings.FILE_SIZE_LIMIT,
+                            file_size_limit=await settings_service.get_value(SettingKey.FILE_SIZE_LIMIT),
                         ),
                         c.Button(
                             html_type='submit',

diff --git a/fai-rag-app/fai-backend/fai_backend/settings/routes.py b/fai-rag-app/fai-backend/fai_backend/settings/routes.py
@@ -107,6 +107,16 @@ async def edit_settings(
                             size='sm',
                             value=settings.FIXED_PIN,
                         ),
+                        c.Range(
+                            name='settings.FILE_SIZE_LIMIT',
+                            label=_('Document upload size limit (MB)'),
+                            required=True,
+                            size='sm',
+                            min=0,
+                            max=2000,
+                            step=10,
+                            value=settings.FILE_SIZE_LIMIT,
+                        ),
 
                         c.Heading(text='AI Settings', class_name='font-bold'),
                         c.InputField(

diff --git a/fai-rag-app/fai-backend/fai_backend/settings/service.py b/fai-rag-app/fai-backend/fai_backend/settings/service.py
@@ -13,8 +13,8 @@ class SettingKey(Enum):
     FIXED_PIN = 'FIXED_PIN'
     OPENAI_API_KEY = 'OPENAI_API_KEY'
     VLLM_CONFIG = 'VLLM_CONFIG'
-    OPENAI_API_KEY = 'OPENAI_API_KEY'
     APP_VECTOR_DB_EMBEDDING_MODEL = 'APP_VECTOR_DB_EMBEDDING_MODEL'
+    FILE_SIZE_LIMIT = 'FILE_SIZE_LIMIT'
 
     # Email
     BREVO_API_URL = 'BREVO_API_URL'