yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-16 17:48:46 +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) self._client.complete(messages=azure_messages, tools=converted_tools, stream=True, **create_args)
) )
else: else:
task = asyncio.create_task( task = asyncio.create_task(self._client.complete(messages=azure_messages, stream=True, **create_args))
self._client.complete(messages=azure_messages, max_tokens=20, stream=True, **create_args)
)
if cancellation_token is not None: if cancellation_token is not None:
cancellation_token.link_future(task) cancellation_token.link_future(task)