Spaces:
Runtime error
Runtime error
File size: 1,912 Bytes
aad7814 | 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 46 | from __future__ import annotations
from unittest.mock import MagicMock, patch
from backend.config import settings
from backend.llm import openai_client
def test_chat_text_uses_pipeline_timeout(monkeypatch):
monkeypatch.setattr(settings, "openai_pipeline_timeout_seconds", 20.0)
openai_client.reset_client()
mock_client = MagicMock()
mock_client.chat.completions.create.return_value = MagicMock(
choices=[MagicMock(message=MagicMock(content="ok"))]
)
with patch.object(openai_client, "_get_client", return_value=mock_client):
openai_client.chat_text([{"role": "user", "content": "hi"}])
_, kwargs = mock_client.chat.completions.create.call_args
assert kwargs["timeout"] == 20.0
def test_chat_json_uses_pipeline_timeout(monkeypatch):
monkeypatch.setattr(settings, "openai_pipeline_timeout_seconds", 20.0)
openai_client.reset_client()
mock_client = MagicMock()
mock_client.chat.completions.create.return_value = MagicMock(
choices=[MagicMock(message=MagicMock(content='{"passed": true}'))]
)
with patch.object(openai_client, "_get_client", return_value=mock_client):
openai_client.chat_json([{"role": "user", "content": "audit"}])
_, kwargs = mock_client.chat.completions.create.call_args
assert kwargs["timeout"] == 20.0
def test_chat_text_explicit_timeout_override(monkeypatch):
monkeypatch.setattr(settings, "openai_pipeline_timeout_seconds", 20.0)
openai_client.reset_client()
mock_client = MagicMock()
mock_client.chat.completions.create.return_value = MagicMock(
choices=[MagicMock(message=MagicMock(content="ok"))]
)
with patch.object(openai_client, "_get_client", return_value=mock_client):
openai_client.chat_text([{"role": "user", "content": "hi"}], timeout=8.0)
_, kwargs = mock_client.chat.completions.create.call_args
assert kwargs["timeout"] == 8.0
|