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): ...@@ -157,6 +157,7 @@ class MultiModalRequest(BaseModel):
max_tokens: Optional[int] = None max_tokens: Optional[int] = None
temperature: Optional[float] = None temperature: Optional[float] = None
stream: Optional[bool] = True stream: Optional[bool] = True
stream_options: Optional[dict] = None
class MultiModalInput(BaseModel): class MultiModalInput(BaseModel):
......
...@@ -221,6 +221,7 @@ class Processor(ProcessMixIn): ...@@ -221,6 +221,7 @@ class Processor(ProcessMixIn):
model=raw_request.model, model=raw_request.model,
messages=[msg], messages=[msg],
stream=True, stream=True,
stream_options=raw_request.stream_options,
max_tokens=raw_request.max_tokens, max_tokens=raw_request.max_tokens,
temperature=raw_request.temperature, temperature=raw_request.temperature,
request_id=str(uuid.uuid4()), request_id=str(uuid.uuid4()),
......
...@@ -146,6 +146,7 @@ class MultiModalRequest(BaseModel): ...@@ -146,6 +146,7 @@ class MultiModalRequest(BaseModel):
max_tokens: Optional[int] = None max_tokens: Optional[int] = None
temperature: Optional[float] = None temperature: Optional[float] = None
stream: Optional[bool] = True stream: Optional[bool] = True
stream_options: Optional[dict] = None
class MultiModalInput(BaseModel): 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