File size: 1,738 Bytes
be9fd4a
7fa723a
 
 
 
 
 
 
 
 
 
 
 
 
be9fd4a
7fa723a
 
 
 
 
 
be9fd4a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7fa723a
be9fd4a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
"""Production / HF Spaces AI profile disables ingest LLM sanitisation only."""

from __future__ import annotations

import pytest

from app.config import Settings


def test_space_id_disables_rag_llm_sanitisation(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("SPACE_ID", "StormShadow308/RICS")
    monkeypatch.setenv("ENABLE_RAG_UPLOAD_SANITISATION", "true")
    monkeypatch.setenv("RAG_SANITISATION_USE_LLM", "true")
    s = Settings()
    assert s.enable_rag_upload_sanitisation is True
    assert s.rag_sanitisation_use_llm is False


def test_production_ai_profile_flag(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.delenv("SPACE_ID", raising=False)
    monkeypatch.setenv("PRODUCTION_AI_PROFILE", "true")
    monkeypatch.setenv("ENABLE_RAG_UPLOAD_SANITISATION", "false")
    monkeypatch.setenv("RAG_SANITISATION_USE_LLM", "true")
    s = Settings()
    assert s.enable_rag_upload_sanitisation is True
    assert s.rag_sanitisation_use_llm is False


def test_space_id_enables_async_parallel_for_sla(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.setenv("SPACE_ID", "StormShadow308/RICS")
    monkeypatch.setenv("ENABLE_ASYNC_PIPELINE", "false")
    s = Settings()
    assert s.enable_async_pipeline is True
    assert s.allow_sqlite_parallel_sections is True
    assert s.generation_sla_seconds == 600
    assert s.generation_timeout_seconds >= 660


def test_personalised_rag_forces_sanitisation(monkeypatch: pytest.MonkeyPatch) -> None:
    monkeypatch.delenv("SPACE_ID", raising=False)
    monkeypatch.setenv("PERSONALISED_STYLE_RAG_ENABLED", "true")
    monkeypatch.setenv("ENABLE_RAG_UPLOAD_SANITISATION", "false")
    s = Settings()
    assert s.enable_rag_upload_sanitisation is True