wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

transport(websocket): improve exception logging

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-02-04 23:50:45 -08:00
parent 0a333c26da
commit 49426aa9a1
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/pipecat/transports/network/fastapi_websocket.py
View File

@@ -102,7 +102,7 @@ class FastAPIWebsocketInputTransport(BaseInputTransport):
else: else:
await self.push_frame(frame) await self.push_frame(frame)
except Exception as e: except Exception as e:
logger.error(f"{self} exception receiving data (class: {e.__class__.__name__})") logger.error(f"{self} exception receiving data: {e.__class__.__name__} ({e})")
await self._callbacks.on_client_disconnected(self._websocket) await self._callbacks.on_client_disconnected(self._websocket)
@@ -178,7 +178,7 @@ class FastAPIWebsocketOutputTransport(BaseOutputTransport):
if payload and self._websocket.client_state == WebSocketState.CONNECTED: if payload and self._websocket.client_state == WebSocketState.CONNECTED:
await self._send_data(payload) await self._send_data(payload)
except Exception as e: except Exception as e:
logger.error(f"{self} exception sending data (class: {e.__class__.__name__})") logger.error(f"{self} exception sending data: {e.__class__.__name__} ({e})")
def _send_data(self, data: str | bytes): def _send_data(self, data: str | bytes):
if self._params.serializer.type == FrameSerializerType.BINARY: if self._params.serializer.type == FrameSerializerType.BINARY:

4
src/pipecat/transports/network/websocket_client.py
View File

@@ -101,7 +101,7 @@ class WebsocketClientSession:
if self._websocket: if self._websocket:
await self._websocket.send(message) await self._websocket.send(message)
except Exception as e: except Exception as e:
logger.error(f"{self} exception sending data (class: {e.__class__.__name__})") logger.error(f"{self} exception sending data: {e.__class__.__name__} ({e})")
async def _client_task_handler(self): async def _client_task_handler(self):
try: try:
@@ -109,7 +109,7 @@ class WebsocketClientSession:
async for message in self._websocket: async for message in self._websocket:
await self._callbacks.on_message(self._websocket, message) await self._callbacks.on_message(self._websocket, message)
except Exception as e: except Exception as e:
logger.error(f"{self} exception receiving data (class: {e.__class__.__name__})") logger.error(f"{self} exception receiving data: {e.__class__.__name__} ({e})")
await self._callbacks.on_disconnected(self._websocket) await self._callbacks.on_disconnected(self._websocket)

4
src/pipecat/transports/network/websocket_server.py
View File

@@ -128,7 +128,7 @@ class WebsocketServerInputTransport(BaseInputTransport):
else: else:
await self.push_frame(frame) await self.push_frame(frame)
except Exception as e: except Exception as e:
logger.error(f"{self} exception receiving data (class: {e.__class__.__name__})") logger.error(f"{self} exception receiving data: {e.__class__.__name__} ({e})")
# Notify disconnection # Notify disconnection
await self._callbacks.on_client_disconnected(websocket) await self._callbacks.on_client_disconnected(websocket)
@@ -223,7 +223,7 @@ class WebsocketServerOutputTransport(BaseOutputTransport):
if payload and self._websocket: if payload and self._websocket:
await self._websocket.send(payload) await self._websocket.send(payload)
except Exception as e: except Exception as e:
logger.error(f"{self} exception sending data (class: {e.__class__.__name__})") logger.error(f"{self} exception sending data: {e.__class__.__name__} ({e})")
async def _write_audio_sleep(self): async def _write_audio_sleep(self):
# Simulate a clock. # Simulate a clock.