fix oai completion api error
Signed-off-by: pandyamarut <pandyamarut@gmail.com>
This commit is contained in:
+2
-1
@@ -176,7 +176,8 @@ class OpenAIvLLMEngine(vLLMEngine):
|
||||
yield create_error_response(str(e)).model_dump()
|
||||
return
|
||||
|
||||
response_generator = await generator_function(request, raw_request=None)
|
||||
dummy_request = DummyRequest()
|
||||
response_generator = await generator_function(request, raw_request=dummy_request)
|
||||
|
||||
if not openai_request.openai_input.get("stream") or isinstance(response_generator, ErrorResponse):
|
||||
yield response_generator.model_dump()
|
||||
|
||||
+3
-1
@@ -49,9 +49,11 @@ class JobInput:
|
||||
self.openai_input = job.get("openai_input")
|
||||
|
||||
class DummyRequest:
|
||||
def __init__(self):
|
||||
self.headers = {}
|
||||
async def is_disconnected(self):
|
||||
return False
|
||||
|
||||
|
||||
class BatchSize:
|
||||
def __init__(self, max_batch_size, min_batch_size, batch_size_growth_factor):
|
||||
self.max_batch_size = max_batch_size
|
||||
|
||||
Reference in New Issue
Block a user