From c949cd18d55ee3684f8bf5257e03d1e1d4c46aef Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Tue, 21 Oct 2025 02:11:07 -0400
Subject: [PATCH] more fields

---
 .../messages/websocket_server_message.rs      | 34 ++++++++++++++++---
 1 file changed, 30 insertions(+), 4 deletions(-)

diff --git a/crates/api_clients/grok_client/src/requests/listen_websocket/messages/websocket_server_message.rs b/crates/api_clients/grok_client/src/requests/listen_websocket/messages/websocket_server_message.rs
index a0d92a95dc..02f72a95c8 100644
--- a/crates/api_clients/grok_client/src/requests/listen_websocket/messages/websocket_server_message.rs
+++ b/crates/api_clients/grok_client/src/requests/listen_websocket/messages/websocket_server_message.rs
@@ -21,12 +21,20 @@ pub enum WebsocketServerMessage {
 /// We may receive several of these for a single prompt.
 #[derive(Deserialize, Clone)]
 pub struct ImageDataMessage {
+  /// UUID.
   pub id: Option<String>,
+
+  /// UUID.
   pub job_id: Option<String>,
+
+  /// UUID.
   pub request_id: Option<String>,
-  
-  /// The original text prompt used
+
+  /// Eg. "0", "50", "100"
+  pub percentage_complete: Option<i32>,
+
   pub prompt: Option<String>,
+  pub full_prompt: Option<String>,
 
   /// Base64 encoded image blob.
   pub blob: Option<String>,
@@ -40,10 +48,28 @@ pub struct ImageDataMessage {
   /// Name of the model used to generate the image
   /// eg. "imagine_h_1"
   pub model_name: Option<String>,
+}
 
-  /// Not sure if we're getting incomplete images.
-  /// I've seen percentages of "50", so I imagine so.
+#[derive(Deserialize, Clone)]
+pub struct JsonDataMessage {
+  /// UUID.
+  pub job_id: Option<String>,
+  
+  /// UUID.
+  pub request_id: Option<String>,
+
+  /// Eg. "0", "50", "100"
   pub percentage_complete: Option<i32>,
+
+  pub prompt: Option<String>,
+  pub full_prompt: Option<String>,
+
+  /// NSFW flag
+  pub r_rated: Option<bool>,
+
+  /// UUID.
+  /// NB: This is observed to be nullable.
+  pub image_id: Option<String>,
 }