feat: support websocket ping

conn_test.go

@@ -22,6 +22,7 @@ type mockWebSocketConn struct {
 	writeMessageFunc func(ctx context.Context, messageType openairt.MessageType, data []byte) error
 	closeFunc        func() error
 	responseFunc     func() *http.Response
+	pingFunc         func(ctx context.Context) error
 }
 
 func (m *mockWebSocketConn) ReadMessage(ctx context.Context) (openairt.MessageType, []byte, error) {
@@ -40,6 +41,10 @@ func (m *mockWebSocketConn) Response() *http.Response {
 	return m.responseFunc()
 }
 
+func (m *mockWebSocketConn) Ping(ctx context.Context) error {
+	return m.pingFunc(ctx)
+}
+
 func TestConnect(t *testing.T) {
 	token := "mock-token"
 	model := "test-model"

ws.go

@@ -54,6 +54,9 @@ type WebSocketConn interface {
 	// Response returns the *http.Response of the WebSocket connection.
 	// Commonly used to get response headers.
 	Response() *http.Response
+
+	// Ping sends a ping message to the WebSocket connection.
+	Ping(ctx context.Context) error
 }
 
 // WebSocketDialer is a WebSocket connection dialer abstraction.

ws_coder.go

@@ -128,3 +128,8 @@ func (c *CoderWebSocketConn) Close() error {
 func (c *CoderWebSocketConn) Response() *http.Response {
 	return c.resp
 }
+
+// Ping sends a ping message to the WebSocket connection.
+func (c *CoderWebSocketConn) Ping(ctx context.Context) error {
+	return c.conn.Ping(ctx)
+}