KeyError in gradio_client stream_messages() pending_event_ids

[linzack]

Describe the bug

Traceback (most recent call last):
  File "/usr/local/lib/python3.12/site-packages/gradio_client/client.py", line 277, in stream_messages
    self.pending_event_ids.remove(event_id)
KeyError: 'a4306ceb969b452f8b4e2d6de58cd88f'

Hi, a httpx.ReadTimeout during client.predict() might prevent self.client.pending_event_ids.add(event_id) from executing in make_predict().
This results in a missing event_id in pending_event_ids, causing the KeyError when stream_messages() attempts removal.

Have you searched existing issues? 🔎

• I have searched and found no existing issues

Reproduction

from time import sleep
import httpx
from gradio_client import Client

client = Client(SPACE_NAME, HF_TOKEN)

while True:
    try:
        result = client.predict(
            state="0",
            api_name="/check_status"
        )
        if result:
            break
        else:
            sleep(60)
            continue
    except (httpx.ReadTimeout, httpx.ConnectError) as exc:
        print('this line executed when httpx.ReadTimeout')
        sleep(60)
        continue
    except Exception as exc:
        print('this line executed when pending_event_ids KeyError')
        break

Screenshot

No response

Logs

No response

System Info

print(gradio_client.__version__)
1.5.2

Severity

I can work around it