fix syntax

This commit is contained in:
kandrusyak
2026-07-25 15:47:36 +03:00
parent 3f63d0305c
commit c04fb9480b
8 changed files with 56 additions and 32 deletions

View File

@@ -38,6 +38,15 @@ class FakeChatCompletions:
return self.completion
class FakeHTTPError(RuntimeError):
def __init__(self) -> None:
super().__init__("forbidden")
self.response = SimpleNamespace(
status_code=403,
text='{"error":{"message":"folder mismatch"}}',
)
class YandexAIClientTests(unittest.IsolatedAsyncioTestCase):
def test_multiple_system_messages_are_merged_at_the_beginning(self) -> None:
self.assertEqual(
@@ -62,11 +71,7 @@ class YandexAIClientTests(unittest.IsolatedAsyncioTestCase):
)
def test_http_error_includes_safe_response_body(self) -> None:
error = RuntimeError("forbidden")
error.response = SimpleNamespace(
status_code=403,
text='{"error":{"message":"folder mismatch"}}',
)
error = FakeHTTPError()
self.assertEqual(
describe_yandex_error(error),