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>
This commit is contained in:
Your Name
2026-05-17 19:10:01 -05:00
committed by lyingbug
parent b3004f73d4
commit 5deffbd769

View File

@@ -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