diff --git a/ufo/llm/openai.py b/ufo/llm/openai.py index 613ee7b4..982d86d5 100644 --- a/ufo/llm/openai.py +++ b/ufo/llm/openai.py @@ -204,6 +204,8 @@ def _chat_completion( completion_tokens, ) + 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], cost except openai.APITimeoutError as e: