mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-05 08:07:05 +00:00
fix(e2e): satisfy basedpyright on session-anomaly retried tests
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
7dc7cde5a6
commit
9dfd7d7f1b
1 changed files with 2 additions and 2 deletions
|
|
@ -62,7 +62,7 @@ class TestSummarizePlannedTurns:
|
|||
|
||||
class TestRetried:
|
||||
def test_transient_failures_then_success_returns_the_success(self) -> None:
|
||||
outcome = Success(data=SessionMessagesResponse())
|
||||
outcome = Success[SessionMessagesResponse](status_code=200, data=SessionMessagesResponse())
|
||||
calls = iter(
|
||||
(NetworkError(message="overloaded"), NetworkError(message="overloaded"), outcome)
|
||||
)
|
||||
|
|
@ -88,7 +88,7 @@ class TestRetried:
|
|||
raise AssertionError("slept after a successful attempt")
|
||||
|
||||
result = retried(
|
||||
lambda: Success(data=SessionMessagesResponse()),
|
||||
lambda: Success[SessionMessagesResponse](status_code=200, data=SessionMessagesResponse()),
|
||||
attempts=3,
|
||||
sleep=sleep_means_retry,
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue