Skip to content

Commit e01c54e

Browse files
committed
Test failed tests
Signed-off-by: Kaiyu Xie <[email protected]>
1 parent cd71e6b commit e01c54e

File tree

1 file changed

+0
-51
lines changed

1 file changed

+0
-51
lines changed

tests/llmapi/apps/openai_server.py

-51
Original file line numberDiff line numberDiff line change
@@ -35,54 +35,3 @@ def __init__(self,
3535
stderr=sys.stderr)
3636
self._wait_for_server(url=self.url_for("health"),
3737
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

Comments
 (0)