@@ -35,54 +35,3 @@ def __init__(self,
35
35
stderr = sys .stderr )
36
36
self ._wait_for_server (url = self .url_for ("health" ),
37
37
timeout = self .MAX_SERVER_START_WAIT_S )
38
-
39
- def __enter__ (self ):
40
- return self
41
-
42
- def __exit__ (self , exc_type , exc_value , traceback ):
43
- self .proc .terminate ()
44
- try :
45
- self .proc .wait (timeout = 30 )
46
- except subprocess .TimeoutExpired as e :
47
- self .proc .kill ()
48
- self .proc .wait (timeout = 30 )
49
-
50
- def _wait_for_server (self , * , url : str , timeout : float ):
51
- # run health check on the first rank only.
52
- start = time .time ()
53
- while True :
54
- try :
55
- if self .rank == 0 :
56
- if requests .get (url ).status_code == 200 :
57
- break
58
- else :
59
- time .sleep (timeout )
60
- break
61
- except Exception as err :
62
- result = self .proc .poll ()
63
- if result is not None and result != 0 :
64
- raise RuntimeError ("Server exited unexpectedly." ) from err
65
-
66
- time .sleep (0.5 )
67
- if time .time () - start > timeout :
68
- raise RuntimeError (
69
- "Server failed to start in time." ) from err
70
-
71
- @property
72
- def url_root (self ) -> str :
73
- return f"http://{ self .host } :{ self .port } "
74
-
75
- def url_for (self , * parts : str ) -> str :
76
- return self .url_root + "/" + "/" .join (parts )
77
-
78
- def get_client (self ):
79
- return openai .OpenAI (
80
- base_url = self .url_for ("v1" ),
81
- api_key = self .DUMMY_API_KEY ,
82
- )
83
-
84
- def get_async_client (self ):
85
- return openai .AsyncOpenAI (
86
- base_url = self .url_for ("v1" ),
87
- api_key = self .DUMMY_API_KEY ,
88
- )
0 commit comments