add a check for whether there are any choices in litellm response

handrew · handrew · commit faeea9f9d00d · 2025-08-22T13:34:45.000-07:00
diff --git a/src/agents/extensions/models/litellm_model.py b/src/agents/extensions/models/litellm_model.py
@@ -104,18 +104,19 @@ async def get_response(
                 prompt=prompt,
             )
 
-            assert isinstance(response.choices[0], litellm.types.utils.Choices)
-
-            if _debug.DONT_LOG_MODEL_DATA:
-                logger.debug("Received model response")
-            else:
-                logger.debug(
-                    f"""LLM resp:\n{
-                        json.dumps(
-                            response.choices[0].message.model_dump(), indent=2, ensure_ascii=False
-                        )
-                    }\n"""
-                )
+            if len(response.choices):
+                assert isinstance(response.choices[0], litellm.types.utils.Choices)
+
+                if _debug.DONT_LOG_MODEL_DATA:
+                    logger.debug("Received model response")
+                else:
+                    logger.debug(
+                        f"""LLM resp:\n{
+                            json.dumps(
+                                response.choices[0].message.model_dump(), indent=2, ensure_ascii=False
+                            )
+                        }\n"""
+                    )
 
             if hasattr(response, "usage"):
                 response_usage = response.usage
