[CI/Build] [1/3] Reorganize entrypoints tests (#5526)

2024-06-27 20:43:17 +08:00
parent 2061f0b8a7
commit e9d32d077d
3 changed files with 895 additions and 773 deletions
--- a/tests/entrypoints/test_openai_vision.py
+++ b/tests/entrypoints/test_openai_vision.py
@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Dict
+from typing import Dict, List

 import openai
 import pytest
@@ -216,7 +216,7 @@ async def test_chat_streaming_image(client: openai.AsyncOpenAI,
        temperature=0.0,
        stream=True,
    )
-    chunks = []
+    chunks: List[str] = []
    finish_reason_count = 0
    async for chunk in stream:
        delta = chunk.choices[0].delta