Function _make_client

tests/test_compact_plugin.py:15–23

(summary_text="<summary>Test summary with Key decisions: decided X.</summary>")

def _make_client(summary_text="<summary>Test summary with Key decisions: decided X.</summary>"):
    mock_response = MagicMock()
    mock_response.choices = [MagicMock()]
    mock_response.choices[0].message.content = summary_text
    mock_response.usage.completion_tokens = 80
    client = MagicMock()
    client.chat.completions.create.return_value = mock_response
    client.models.retrieve.return_value = MagicMock(spec=[])
    return client


class TestEstimateTokens: