Skip to content

.NET: Support primitives and arrays for SO#3696

Merged
SergeyMenshykh merged 4 commits intomicrosoft:feature-sofrom
SergeyMenshykh:so-wrap-primitives-and-arrays
Feb 6, 2026
Merged

.NET: Support primitives and arrays for SO#3696
SergeyMenshykh merged 4 commits intomicrosoft:feature-sofrom
SergeyMenshykh:so-wrap-primitives-and-arrays

Conversation

@SergeyMenshykh
Copy link
Member

Motivation and Context

The new solution for structured output is not handling primitives and arrays, which means that LLMs will reject requests where the SO schema does not have an object as the root.

Description

  • Wrap primitives, arrays, and enums provided as SO types into a JSON object schema.
  • Unwrap the response from the agent if it was wrapped.

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR adds support for primitives, arrays, and enums in structured output by automatically wrapping non-object JSON schemas. Previously, the structured output feature only supported object types at the root level, which caused LLMs to reject requests for primitive or array types.

Changes:

  • Wraps primitives, arrays, and enums in a JSON object schema with a "data" property when used as structured output types
  • Unwraps the response from the agent when deserializing wrapped types
  • Adds comprehensive unit tests and an integration test for primitive and array type handling

Reviewed changes

Copilot reviewed 4 out of 4 changed files in this pull request and generated 3 comments.

File Description
dotnet/src/Microsoft.Agents.AI.Abstractions/AIAgentStructuredOutput.cs Implements schema wrapping logic with EnsureObjectSchema, SchemaRepresentsObject, and helper methods to wrap non-object schemas
dotnet/src/Microsoft.Agents.AI.Abstractions/AgentResponse{T}.cs Adds IsWrappedInObject property and UnwrapDataProperty method to unwrap responses from wrapped schemas
dotnet/tests/Microsoft.Agents.AI.Abstractions.UnitTests/AIAgentStructuredOutputTests.cs Adds unit tests for schema wrapping, response unwrapping, and end-to-end scenarios
dotnet/tests/AgentConformance.IntegrationTests/StructuredOutputRunTests.cs Adds integration test for primitive type structured output

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Copilot reviewed 5 out of 5 changed files in this pull request and generated 2 comments.

@SergeyMenshykh SergeyMenshykh merged commit c26a773 into microsoft:feature-so Feb 6, 2026
10 checks passed
@github-project-automation github-project-automation bot moved this from In Review to Done in Agent Framework Feb 6, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants