yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-16 09:39:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove max_tokens from az ai client create call when stream=True (#5860)

Browse Source
This commit is contained in:
Eric Zhu 2025-03-06 17:18:37 -08:00 committed by GitHub
parent 05b14f197a
commit ea89a84c30
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/packages/autogen-ext/src/autogen_ext/models/azure/_azure_ai_client.py
View File

@ -404,9 +404,7 @@ class AzureAIChatCompletionClient(ChatCompletionClient):
self._client.complete(messages=azure_messages, tools=converted_tools, stream=True, **create_args)
)
else:
task = asyncio.create_task(
self._client.complete(messages=azure_messages, max_tokens=20, stream=True, **create_args)
)
task = asyncio.create_task(self._client.complete(messages=azure_messages, stream=True, **create_args))
if cancellation_token is not None:
cancellation_token.link_future(task)