Skip to content

Commit 4244093

Browse files
author
openingnow
committed
Add tests to check /v1/responses
1 parent 3b16dac commit 4244093

File tree

2 files changed

+50
-1
lines changed

2 files changed

+50
-1
lines changed

tools/server/tests/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ aiohttp~=3.9.3
22
pytest~=8.3.3
33
huggingface_hub>=0.34.0,<1.0
44
numpy~=1.26.4
5-
openai~=1.55.3
5+
openai~=2.14.0
66
prometheus-client~=0.20.0
77
requests~=2.32.3
88
wget~=3.2
Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,49 @@
1+
import pytest
2+
from openai import OpenAI
3+
import openai
4+
from utils import *
5+
6+
server: ServerProcess
7+
8+
@pytest.fixture(autouse=True)
9+
def create_server():
10+
global server
11+
server = ServerPreset.tinyllama2()
12+
13+
def test_responses_with_openai_library():
14+
global server
15+
server.start()
16+
client = OpenAI(api_key="dummy", base_url=f"http://{server.server_host}:{server.server_port}/v1")
17+
res = client.responses.create(
18+
model="gpt-4.1",
19+
input=[
20+
{"role": "system", "content": "Book"},
21+
{"role": "user", "content": "What is the best book"},
22+
],
23+
max_output_tokens=8,
24+
temperature=0.8,
25+
)
26+
assert match_regex("(Suddenly)+", res.output_text)
27+
28+
def test_responses_stream_with_openai_library():
29+
global server
30+
server.start()
31+
client = OpenAI(api_key="dummy", base_url=f"http://{server.server_host}:{server.server_port}/v1")
32+
stream = client.responses.create(
33+
model="gpt-4.1",
34+
input=[
35+
{"role": "system", "content": "Book"},
36+
{"role": "user", "content": "What is the best book"},
37+
],
38+
max_output_tokens=8,
39+
temperature=0.8,
40+
stream=True,
41+
)
42+
43+
gathered_text = ''
44+
for r in stream:
45+
if type(r) == openai.types.responses.response_text_delta_event.ResponseTextDeltaEvent:
46+
gathered_text += r.delta
47+
if type(r) == openai.types.responses.response_completed_event.ResponseCompletedEvent:
48+
assert gathered_text == r.response.output_text
49+
assert match_regex("(Suddenly)+", r.response.output_text)

0 commit comments

Comments
 (0)