truss 0.11.1rc3__tar.gz → 0.11.1rc4__tar.gz

{truss-0.11.1rc3 → truss-0.11.1rc4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: truss
-Version: 0.11.1rc3
+Version: 0.11.1rc4
 Summary: A seamless bridge from model development to model delivery
 Project-URL: Repository, https://github.com/basetenlabs/truss
 Project-URL: Homepage, https://truss.baseten.co

{truss-0.11.1rc3 → truss-0.11.1rc4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "truss"
-version = "0.11.1rc3"
+version = "0.11.1rc4"
 description = "A seamless bridge from model development to model delivery"
 authors = [
     { name = "Pankaj Gupta", email = "no-reply@baseten.co" },

{truss-0.11.1rc3 → truss-0.11.1rc4}/truss-chains/truss_chains/remote_chainlet/utils.py

@@ -589,7 +589,12 @@ class WebsocketWrapperFastAPI:
 
     async def receive(self) -> Union[str, bytes]:
         message = await self._websocket.receive()
-        if message.get("text"):
+
+        if message.get("type") == "websocket.disconnect":
+            # NB(nikhil): Mimics FastAPI `_raise_on_disconnect`, since otherwise the user has no
+            # way of detecting that the client disconnected.
+            raise fastapi.WebSocketDisconnect(message["code"], message.get("reason"))
+        elif message.get("text"):
             return typing.cast(str, message["text"])
         else:
             return typing.cast(bytes, message["bytes"])