Unverified Commit f6d4351f authored by Blake Ledden's avatar Blake Ledden Committed by GitHub
Browse files

fix: Forward stream_options through multimodal request pipeline (#6474)


Signed-off-by: default avatarBlake Ledden <blake.ledden@gmail.com>
parent d100c6e2
......@@ -157,6 +157,7 @@ class MultiModalRequest(BaseModel):
max_tokens: Optional[int] = None
temperature: Optional[float] = None
stream: Optional[bool] = True
stream_options: Optional[dict] = None
class MultiModalInput(BaseModel):
......
......@@ -221,6 +221,7 @@ class Processor(ProcessMixIn):
model=raw_request.model,
messages=[msg],
stream=True,
stream_options=raw_request.stream_options,
max_tokens=raw_request.max_tokens,
temperature=raw_request.temperature,
request_id=str(uuid.uuid4()),
......
......@@ -146,6 +146,7 @@ class MultiModalRequest(BaseModel):
max_tokens: Optional[int] = None
temperature: Optional[float] = None
stream: Optional[bool] = True
stream_options: Optional[dict] = None
class MultiModalInput(BaseModel):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment