Skip to content

Commit 813522b

Browse files
committed
Fix #604 Chat Completion model raises runtime error when response.choices is empty
1 parent c2005f8 commit 813522b

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/agents/models/openai_chatcompletions.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@
77

88
from openai import NOT_GIVEN, AsyncOpenAI, AsyncStream
99
from openai.types import ChatModel
10-
from openai.types.chat import ChatCompletion, ChatCompletionChunk
10+
from openai.types.chat import ChatCompletion, ChatCompletionChunk, ChatCompletionMessage
11+
from openai.types.chat.chat_completion import Choice
1112
from openai.types.responses import Response
1213
from openai.types.responses.response_prompt_param import ResponsePromptParam
1314
from openai.types.responses.response_usage import InputTokensDetails, OutputTokensDetails
@@ -74,8 +75,11 @@ async def get_response(
7475
prompt=prompt,
7576
)
7677

77-
first_choice = response.choices[0]
78-
message = first_choice.message
78+
message: ChatCompletionMessage | None = None
79+
first_choice: Choice | None = None
80+
if response.choices and len(response.choices) > 0:
81+
first_choice = response.choices[0]
82+
message = first_choice.message
7983

8084
if _debug.DONT_LOG_MODEL_DATA:
8185
logger.debug("Received model response")

0 commit comments

Comments
 (0)