Handle slow AI backend timeouts

This commit is contained in:
Hermes Agent
2026-09-11 13:42:42 +02:00
parent 9320217e0f
commit b2d2ba58fc
6 changed files with 56 additions and 5 deletions
+17 -1
View File
@@ -1,4 +1,7 @@
from ai_typewriter.ai_client import ask_ai
import pytest
import requests
from ai_typewriter.ai_client import AIClientError, ask_ai
from ai_typewriter.config import AppConfig
@@ -27,6 +30,19 @@ def test_ollama_payload_contains_strict_system_prompt(monkeypatch):
assert "Uniquement la réponse brute" in seen["json"]["messages"][0]["content"]
def test_ollama_timeout_message_suggests_config_change(monkeypatch):
def fake_post(url, json, timeout, **kwargs):
raise requests.Timeout("too slow")
monkeypatch.setattr("ai_typewriter.ai_client.requests.post", fake_post)
with pytest.raises(AIClientError) as exc:
ask_ai("texte", AppConfig(provider="ollama", model="m", request_timeout_seconds=300))
assert "300 s" in str(exc.value)
assert "request_timeout_seconds" in str(exc.value)
assert "0 pour désactiver" in str(exc.value)
def test_gemini_payload(monkeypatch):
seen = {}
def fake_post(url, params, json, timeout):
+9
View File
@@ -16,6 +16,15 @@ def test_load_default_config(tmp_path):
assert "Réponds directement" in cfg.system_prompt
def test_timeout_zero_disables_timeout(tmp_path):
path = tmp_path / "config.json"
path.write_text(json.dumps({"provider": "ollama", "request_timeout_seconds": 0}), encoding="utf-8")
cfg = load_config(path)
assert cfg.request_timeout_seconds is None
def test_reject_invalid_provider(tmp_path):
path = tmp_path / "config.json"
path.write_text(json.dumps({"provider": "bad"}), encoding="utf-8")