fix: tuple schema and websocket connection (#4709)

Signed-off-by: Frost Ming <me@frostming.com>
bentoml · May 6, 2024 · e2c21b8 · e2c21b8
1 parent 75acab0
commit e2c21b8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/_bentoml_impl/server/app.py b/src/_bentoml_impl/server/app.py
@@ -51,7 +51,7 @@ async def __call__(
     ) -> None:
         from starlette.requests import Request
 
-        if scope["type"] not in ("http", "websocket"):
+        if scope["type"] not in ("http",):
             return await self.app(scope, receive, send)
 
         req = Request(scope, receive, send)

diff --git a/src/bentoml/_internal/service/openapi/specification.py b/src/bentoml/_internal/service/openapi/specification.py
@@ -96,6 +96,7 @@ class Schema:
     maxItems: t.Optional[int] = None
     minItems: t.Optional[int] = None
     uniqueItems: t.Optional[bool] = None
+    prefixItems: t.Optional[t.List[Schema]] = None
     maxProperties: t.Optional[int] = None
     minProperties: t.Optional[int] = None
     required: t.Optional[t.List[str]] = None