Python: Properly handle AzureAI Search results for AzureAIAgent #12654
+53
−36
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Motivation and Context
After the Azure AI Agent service went to GA, there were some updates in the underlying SDK. When there are azure_ai_search results while running the agent, they no longer show up as attributes related to the tool call. They are present in the dictionary, so our current code is failing when trying to access the attribute. This PR updates how we form the FunctionCallContent or FunctionResultContent for both azure_ai_search streaming and non-streaming.
Description
Update how we form the FunctionCallContent or FunctionResultContent for both azure_ai_search streaming and non-streaming for the AzureAIAgent.
Contribution Checklist