Enhancement (beta): add direct LLM language setting (#12906)

This commit is contained in:
shamoon
2026-06-03 08:53:22 -07:00
committed by GitHub
parent 47a6fcfc39
commit 1663ed170c
10 changed files with 100 additions and 4 deletions
+39
View File
@@ -408,6 +408,45 @@ class TestAISuggestions(DirectoriesMixin, TestCase):
"KI Title",
)
@patch("documents.views.get_ai_document_classification")
@override_settings(
AI_ENABLED=True,
LLM_BACKEND="mock_backend",
LLM_OUTPUT_LANGUAGE="fr-fr",
)
def test_ai_suggestions_configured_language_takes_precedence(
self,
mock_get_ai_classification,
) -> None:
UiSettings.objects.create(user=self.user, settings={"language": "de-de"})
mock_get_ai_classification.return_value = {
"title": "Titre IA",
"tags": [],
"correspondents": [],
"document_types": [],
"storage_paths": [],
"dates": [],
}
self.client.force_login(user=self.user)
response = self.client.get(
f"/api/documents/{self.document.pk}/ai_suggestions/",
)
self.assertEqual(response.status_code, status.HTTP_200_OK)
mock_get_ai_classification.assert_called_once_with(
self.document,
self.user,
"fr-fr",
)
self.assertEqual(
get_llm_suggestion_cache(
self.document.pk,
backend="mock_backend:fr-fr",
).suggestions["title"],
"Titre IA",
)
@patch("documents.views.get_ai_document_classification")
@override_settings(
AI_ENABLED=True,