Skip to content

Commit e625cb4

Browse files
authored
Only include stream_options when streaming (#519)
Closes #518
1 parent f329eef commit e625cb4

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/agents/models/openai_chatcompletions.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -528,7 +528,9 @@ async def _fetch_response(
528528
reasoning_effort = model_settings.reasoning.effort if model_settings.reasoning else None
529529
store = _Converter.get_store_param(self._get_client(), model_settings)
530530

531-
stream_options = _Converter.get_stream_options_param(self._get_client(), model_settings)
531+
stream_options = _Converter.get_stream_options_param(
532+
self._get_client(), model_settings, stream=stream
533+
)
532534

533535
ret = await self._get_client().chat.completions.create(
534536
model=self.model,
@@ -591,8 +593,11 @@ def get_store_param(cls, client: AsyncOpenAI, model_settings: ModelSettings) ->
591593

592594
@classmethod
593595
def get_stream_options_param(
594-
cls, client: AsyncOpenAI, model_settings: ModelSettings
596+
cls, client: AsyncOpenAI, model_settings: ModelSettings, stream: bool
595597
) -> dict[str, bool] | None:
598+
if not stream:
599+
return None
600+
596601
default_include_usage = True if cls.is_openai(client) else None
597602
include_usage = (
598603
model_settings.include_usage

0 commit comments

Comments
 (0)