Websocket connection closes when gemini executes code using Matplotlib

[rkdlight]

I was working through Gemini live api and asking LLM to draw a graph based on some data. The model performed everything correctly and built a graph. But when I sent her the next request, the connection was lost. This is what is output in the logs:

2025-01-24 00:26:52 - DEBUG - websockets.client - < PONG 4b f2 5e d3 [binary, 4 bytes]
2025-01-24 00:26:52 - DEBUG - websockets.client - % received keepalive pong
2025-01-24 00:26:52 - DEBUG - websockets.client - < CLOSE 1008 (policy violation) Request trace id: 5abe86...: application/json isn't [125 bytes]
2025-01-24 00:26:52 - DEBUG - websockets.client - > CLOSE 1008 (policy violation) Request trace id: 5abe86...: application/json isn't [125 bytes]
2025-01-24 00:26:52 - DEBUG - websockets.client - = connection is CLOSING
2025-01-24 00:26:52 - DEBUG - websockets.client - < EOF
2025-01-24 00:26:52 - DEBUG - websockets.client - > EOF
2025-01-24 00:26:52 - DEBUG - websockets.client - = connection is CLOSED
2025-01-24 00:26:52 - DEBUG - websockets.client - x closing TCP connection

Then I decided to test the same thing through Google aistudio. Got the same result. The connection is terminated.
Moreover, when the model executes any other code, everything seems to be fine.

Steps to reproduce

1. Ask the live model to draw any graph using random values