fix hanging issue: missing entries

Author: DNXie

apps/grpo/main.py

@@ -47,6 +47,8 @@ class Episode:
     request_len: int
     response_len: int
     target: Any | None = None
+    request: str | None = None
+    response: str | None = None
     # Processed data
     completion: Completion | None = None
     ref_logprobs: torch.Tensor | None = None
@@ -391,6 +393,8 @@ async def continuous_rollouts():
                     request_len=max_req_tokens,
                     response_len=max_res_tokens,
                     target=target,
+                    request=prompt,
+                    response=response.text,
                     completion=response,
                 )
                 episode.reward_breakdown = await reward_actor.evaluate_response.route(

src/forge/observability/metrics.py

@@ -231,6 +231,8 @@ def record_episode_sample(table_name: str, episode):
     print(
         "[DEBUG] Adding sample to table via record_metric, episode_id: ",
         episode.episode_id,
+        # "episode: ",
+        # episode,
     )
     record_metric(table_name, sample, Reduce.SAMPLE)
     print(