Function send_websocket_message

examples/chatbot/frontend/chatbot.py:92–105

Connect, send a message, yield responses, and disconnect.

(message)

def send_websocket_message(message):
    """Connect, send a message, yield responses, and disconnect."""
    ws = connect_websocket()
    if ws:
        try:
            ws.send(json.dumps({"message": message, "session_id": str(uuid.uuid4())}))
            while True:
                try:
                    response = ws.recv()
                    yield response
                except websocket.WebSocketConnectionClosedException:
                    break
        finally:
            disconnect_websocket()


# Accept user input