File tree Expand file tree Collapse file tree 2 files changed +50
-1
lines changed
Expand file tree Collapse file tree 2 files changed +50
-1
lines changed Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ aiohttp~=3.9.3
22pytest ~= 8.3.3
33huggingface_hub >= 0.34.0 ,< 1.0
44numpy ~= 1.26.4
5- openai ~= 1.55.3
5+ openai ~= 2.14.0
66prometheus-client ~= 0.20.0
77requests ~= 2.32.3
88wget ~= 3.2
Original file line number Diff line number Diff line change 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 )
You can’t perform that action at this time.
0 commit comments