mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-12 08:42:18 +00:00
test(proxy): type the first-chunk poll test parameters
This commit is contained in:
parent
f343c2421a
commit
e6290016c1
1 changed files with 4 additions and 1 deletions
|
|
@ -16,6 +16,7 @@ following the OpenAI Response API format.
|
|||
|
||||
import asyncio
|
||||
import json
|
||||
from collections.abc import Mapping
|
||||
from datetime import datetime, timezone
|
||||
from typing import Any, Dict, Final, Optional
|
||||
from unittest.mock import AsyncMock, Mock, patch
|
||||
|
|
@ -1597,7 +1598,9 @@ class TestBackgroundStreamingTerminalEvents:
|
|||
(429, "Too Many Requests", {"code": "rate_limit_exceeded", "message": None}),
|
||||
],
|
||||
)
|
||||
async def test_a_first_chunk_error_polls_failed_like_a_late_one(self, status_code, body, expected_error):
|
||||
async def test_a_first_chunk_error_polls_failed_like_a_late_one(
|
||||
self, status_code: int, body: str, expected_error: Mapping[str, str | None]
|
||||
):
|
||||
from fastapi import Response
|
||||
|
||||
from litellm.proxy.response_polling.background_streaming import (
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue