From 5deffbd769c6e37f455639a4cb46433220a517db Mon Sep 17 00:00:00 2001
From: Your Name <jonathan.f.hill@gmail.com>
Date: Sun, 17 May 2026 19:10:01 -0500
Subject: [PATCH] fix: guard against empty choices and message=None in LLM
 response

response.choices[0].message.content raises IndexError (empty choices) or
AttributeError (message=None on filtered content) during dataset generation.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 dataset/qa_dataset.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/dataset/qa_dataset.py b/dataset/qa_dataset.py
index 45f8265bb..df4ee5229 100644
--- a/dataset/qa_dataset.py
+++ b/dataset/qa_dataset.py
@@ -177,6 +177,8 @@ Answer:"""
             messages=[{"role": "user", "content": prompt}],
             temperature=0.3,
         )
+        if not response.choices or response.choices[0].message is None:
+            raise ValueError("LLM returned empty or filtered response")
         return response.choices[0].message.content