Move openai api server into a separate file (#429)

This commit is contained in:
Lianmin Zheng
2024-05-12 06:41:32 -07:00
committed by GitHub
parent abc548c707
commit 3fc97f6709
5 changed files with 423 additions and 380 deletions

View File

@@ -1,6 +1,7 @@
import asyncio
import concurrent.futures
import dataclasses
import logging
import multiprocessing as mp
import os
from typing import List
@@ -31,6 +32,8 @@ from sglang.srt.utils import get_exception_traceback, is_multimodal_model, load_
asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
logger = logging.getLogger(__name__)
@dataclasses.dataclass
class ReqState:
@@ -185,10 +188,15 @@ class TokenizerManager:
while True:
await event.wait()
yield self.convert_logprob_style(state.out_list[-1],
out = self.convert_logprob_style(state.out_list[-1],
obj.return_logprob,
obj.top_logprobs_num,
obj.return_text_in_logprobs)
if self.server_args.log_requests and state.finished:
logger.info(f"in={obj.text}, out={out}")
yield out
state.out_list = []
if state.finished:
del self.rid_to_state[rid]