mirror of
https://github.com/Stirling-Tools/Stirling-PDF.git
synced 2026-09-03 05:10:16 +03:00
Anthony Stirling
8de94ff152
# Description of Changes AI settings customisation in settings menu, as part of this also tested and fixed ollama and other 3rd party AI integrations - Adds an admin AI settings UI for customizing AI behaviour, including per-provider model and API-key configuration - Backend pushes AI config changes to the Python engine at runtime via a config-push bridge, so changes apply without a restart - Config-push is gated off in SaaS; engine now drains background tasks on shutdown instead of cancelling them --- ## Checklist ### General - [ ] I have read the [Contribution Guidelines](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/CONTRIBUTING.md) - [ ] I have read the [Stirling-PDF Developer Guide](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/DeveloperGuide.md) (if applicable) - [ ] I have read the [How to add new languages to Stirling-PDF](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/devGuide/HowToAddNewLanguage.md) (if applicable) - [ ] I have performed a self-review of my own code - [ ] My changes generate no new warnings ### Documentation - [ ] I have updated relevant docs on [Stirling-PDF's doc repo](https://github.com/Stirling-Tools/Stirling-Tools.github.io/blob/main/docs/) (if functionality has heavily changed) - [ ] I have read the section [Add New Translation Tags](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/devGuide/HowToAddNewLanguage.md#add-new-translation-tags) (for new translation tags only) ### Translations (if applicable) - [ ] I ran [`scripts/counter_translation.py`](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/docs/counter_translation.md) ### UI Changes (if applicable) - [ ] Screenshots or videos demonstrating the UI changes are attached (e.g., as comments or direct attachments in the PR) ### Testing (if applicable) - [ ] I have run `task check` to verify linters, typechecks, and tests pass - [ ] I have tested my changes locally. Refer to the [Testing Guide](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/DeveloperGuide.md#7-testing) for more details.
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
"""Wire-contract test pinning the camelCase processor -> engine config push; keep in sync with the Java test."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from stirling.contracts import ConfigPushRequest
|
|
|
|
FIXTURE = Path(__file__).parent / "fixtures" / "processor_config_push.json"
|
|
|
|
|
|
def _load() -> dict[str, Any]:
|
|
return json.loads(FIXTURE.read_text(encoding="utf-8"))
|
|
|
|
|
|
def test_processor_contract_round_trips_every_field() -> None:
|
|
payload = _load()
|
|
req = ConfigPushRequest.model_validate(payload)
|
|
|
|
m = payload["models"]
|
|
assert req.models.provider == m["provider"]
|
|
assert req.models.smart_model == m["smartModel"]
|
|
assert req.models.fast_model == m["fastModel"]
|
|
assert req.models.smart_max_tokens == m["smartMaxTokens"]
|
|
assert req.models.fast_max_tokens == m["fastMaxTokens"]
|
|
assert req.models.api_key == m["apiKey"]
|
|
assert req.models.base_url == m["baseUrl"]
|
|
|
|
r = payload["rag"]
|
|
assert req.rag.embedding_provider == r["embeddingProvider"]
|
|
assert req.rag.embedding_model == r["embeddingModel"]
|
|
assert req.rag.embedding_api_key == r["embeddingApiKey"]
|
|
assert req.rag.embedding_base_url == r["embeddingBaseUrl"]
|
|
assert req.rag.top_k == r["topK"]
|
|
assert req.rag.max_searches == r["maxSearches"]
|
|
|
|
limits = payload["limits"]
|
|
assert req.limits.max_pages == limits["maxPages"]
|
|
assert req.limits.max_characters == limits["maxCharacters"]
|
|
assert req.limits.model_max_concurrency == limits["modelMaxConcurrency"]
|
|
|
|
|
|
def test_processor_contract_has_no_unmapped_keys() -> None:
|
|
"""Guard the fixture itself: every wire key must map to a model field, none silently absorbed by extra="ignore"."""
|
|
payload = _load()
|
|
expected = {
|
|
"models": {
|
|
"provider",
|
|
"smartModel",
|
|
"fastModel",
|
|
"smartMaxTokens",
|
|
"fastMaxTokens",
|
|
"apiKey",
|
|
"baseUrl",
|
|
},
|
|
"rag": {
|
|
"embeddingProvider",
|
|
"embeddingModel",
|
|
"embeddingApiKey",
|
|
"embeddingBaseUrl",
|
|
"topK",
|
|
"maxSearches",
|
|
},
|
|
"limits": {"maxPages", "maxCharacters", "modelMaxConcurrency"},
|
|
}
|
|
assert set(payload) == set(expected)
|
|
for section, keys in expected.items():
|
|
assert set(payload[section]) == keys
|