Skip to content

Commit 9aae7d1

Browse files
committed
add top_logprobs to chatcompletions and litellm
1 parent 800baee commit 9aae7d1

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

src/agents/extensions/models/litellm_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -313,6 +313,7 @@ async def _fetch_response(
313313
stream=stream,
314314
stream_options=stream_options,
315315
reasoning_effort=reasoning_effort,
316+
top_logprobs=model_settings.top_logprobs,
316317
extra_headers={**HEADERS, **(model_settings.extra_headers or {})},
317318
api_key=self.api_key,
318319
base_url=self.base_url,

src/agents/models/openai_chatcompletions.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -286,6 +286,7 @@ async def _fetch_response(
286286
stream_options=self._non_null_or_not_given(stream_options),
287287
store=self._non_null_or_not_given(store),
288288
reasoning_effort=self._non_null_or_not_given(reasoning_effort),
289+
top_logprobs=self._non_null_or_not_given(model_settings.top_logprobs),
289290
extra_headers={**HEADERS, **(model_settings.extra_headers or {})},
290291
extra_query=model_settings.extra_query,
291292
extra_body=model_settings.extra_body,

0 commit comments

Comments
 (0)