diff --git a/dotnet/Directory.Packages.props b/dotnet/Directory.Packages.props index 396a316575..a38bb7d1f5 100644 --- a/dotnet/Directory.Packages.props +++ b/dotnet/Directory.Packages.props @@ -19,8 +19,8 @@ - - + + @@ -35,7 +35,7 @@ - + diff --git a/dotnet/samples/02-agents/FoundryAgents/FoundryAgents_Evaluations_Step01_RedTeaming/Program.cs b/dotnet/samples/02-agents/FoundryAgents/FoundryAgents_Evaluations_Step01_RedTeaming/Program.cs index 60a859c28f..1e1e48d54b 100644 --- a/dotnet/samples/02-agents/FoundryAgents/FoundryAgents_Evaluations_Step01_RedTeaming/Program.cs +++ b/dotnet/samples/02-agents/FoundryAgents/FoundryAgents_Evaluations_Step01_RedTeaming/Program.cs @@ -60,7 +60,7 @@ // Submit the red team run to the service Console.WriteLine("Submitting red team run..."); -RedTeam redTeamRun = await aiProjectClient.RedTeams.CreateAsync(redTeamConfig); +RedTeam redTeamRun = await aiProjectClient.RedTeams.CreateAsync(redTeamConfig, options: null); Console.WriteLine($"Red team run created: {redTeamRun.Name}"); Console.WriteLine($"Status: {redTeamRun.Status}"); diff --git a/dotnet/samples/02-agents/FoundryAgents/FoundryAgents_Step22_MemorySearch/Program.cs b/dotnet/samples/02-agents/FoundryAgents/FoundryAgents_Step22_MemorySearch/Program.cs index 97eed4e838..836bf1b684 100644 --- a/dotnet/samples/02-agents/FoundryAgents/FoundryAgents_Step22_MemorySearch/Program.cs +++ b/dotnet/samples/02-agents/FoundryAgents/FoundryAgents_Step22_MemorySearch/Program.cs @@ -35,7 +35,7 @@ Use the memory search tool to recall relevant information from previous interact AIProjectClient aiProjectClient = new(new Uri(endpoint), new AzureCliCredential()); // Create the Memory Search tool configuration -MemorySearchTool memorySearchTool = new(memoryStoreName, userScope) +MemorySearchPreviewTool memorySearchTool = new(memoryStoreName, userScope) { // Optional: Configure how quickly new memories are indexed (in seconds) UpdateDelay = 1, diff --git a/dotnet/samples/03-workflows/Declarative/HostedWorkflow/Program.cs b/dotnet/samples/03-workflows/Declarative/HostedWorkflow/Program.cs index 272e83f983..81e2abbafe 100644 --- a/dotnet/samples/03-workflows/Declarative/HostedWorkflow/Program.cs +++ b/dotnet/samples/03-workflows/Declarative/HostedWorkflow/Program.cs @@ -88,7 +88,9 @@ private static async Task CreateWorkflowAsync(AIProjectClient agen { string workflowYaml = File.ReadAllText("MathChat.yaml"); +#pragma warning disable AAIP001 // WorkflowAgentDefinition is experimental WorkflowAgentDefinition workflowAgentDefinition = WorkflowAgentDefinition.FromYaml(workflowYaml); +#pragma warning restore AAIP001 return await agentClient.CreateAgentAsync( diff --git a/dotnet/src/Microsoft.Agents.AI.AzureAI/AzureAIProjectChatClientExtensions.cs b/dotnet/src/Microsoft.Agents.AI.AzureAI/AzureAIProjectChatClientExtensions.cs index 027eea1bca..126e629af2 100644 --- a/dotnet/src/Microsoft.Agents.AI.AzureAI/AzureAIProjectChatClientExtensions.cs +++ b/dotnet/src/Microsoft.Agents.AI.AzureAI/AzureAIProjectChatClientExtensions.cs @@ -39,7 +39,7 @@ public static partial class AzureAIProjectChatClientExtensions /// The agent with the specified name was not found. /// /// When instantiating a by using an , minimal information will be available about the agent in the instance level, and any logic that relies - /// on to retrieve information about the agent like will receive as the result. + /// on to retrieve information about the agent like will receive as the result. /// public static ChatClientAgent AsAIAgent( this AIProjectClient aiProjectClient, @@ -355,28 +355,27 @@ public static Task CreateAIAgentAsync( private static readonly ModelReaderWriterOptions s_modelWriterOptionsWire = new("W"); /// - /// Asynchronously retrieves an agent record by name using the Protocol method with user-agent header. + /// Asynchronously retrieves an agent record by name using the protocol method to inject user-agent headers. /// private static async Task GetAgentRecordByNameAsync(AIProjectClient aiProjectClient, string agentName, CancellationToken cancellationToken) { ClientResult protocolResponse = await aiProjectClient.Agents.GetAgentAsync(agentName, cancellationToken.ToRequestOptions(false)).ConfigureAwait(false); var rawResponse = protocolResponse.GetRawResponse(); AgentRecord? result = ModelReaderWriter.Read(rawResponse.Content, s_modelWriterOptionsWire, AzureAIProjectsOpenAIContext.Default); - return ClientResult.FromOptionalValue(result, rawResponse).Value! - ?? throw new InvalidOperationException($"Agent with name '{agentName}' not found."); + return result ?? throw new InvalidOperationException($"Agent with name '{agentName}' not found."); } /// - /// Asynchronously creates an agent version using the Protocol method with user-agent header. + /// Asynchronously creates an agent version using the protocol method to inject user-agent headers. /// private static async Task CreateAgentVersionWithProtocolAsync(AIProjectClient aiProjectClient, string agentName, AgentVersionCreationOptions creationOptions, CancellationToken cancellationToken) { - using BinaryContent protocolRequest = BinaryContent.Create(ModelReaderWriter.Write(creationOptions, ModelReaderWriterOptions.Json, AzureAIProjectsContext.Default)); - ClientResult protocolResponse = await aiProjectClient.Agents.CreateAgentVersionAsync(agentName, protocolRequest, cancellationToken.ToRequestOptions(false)).ConfigureAwait(false); - + BinaryData serializedOptions = ModelReaderWriter.Write(creationOptions, s_modelWriterOptionsWire, AzureAIProjectsContext.Default); + BinaryContent content = BinaryContent.Create(serializedOptions); + ClientResult protocolResponse = await aiProjectClient.Agents.CreateAgentVersionAsync(agentName, content, foundryFeatures: null, cancellationToken.ToRequestOptions(false)).ConfigureAwait(false); var rawResponse = protocolResponse.GetRawResponse(); AgentVersion? result = ModelReaderWriter.Read(rawResponse.Content, s_modelWriterOptionsWire, AzureAIProjectsOpenAIContext.Default); - return ClientResult.FromValue(result, rawResponse).Value!; + return result ?? throw new InvalidOperationException($"Failed to create agent version for agent '{agentName}'."); } private static async Task CreateAIAgentAsync( diff --git a/dotnet/tests/Microsoft.Agents.AI.AzureAI.UnitTests/AzureAIProjectChatClientExtensionsTests.cs b/dotnet/tests/Microsoft.Agents.AI.AzureAI.UnitTests/AzureAIProjectChatClientExtensionsTests.cs index 2f2e276ae9..da81471895 100644 --- a/dotnet/tests/Microsoft.Agents.AI.AzureAI.UnitTests/AzureAIProjectChatClientExtensionsTests.cs +++ b/dotnet/tests/Microsoft.Agents.AI.AzureAI.UnitTests/AzureAIProjectChatClientExtensionsTests.cs @@ -467,7 +467,7 @@ public async Task GetAIAgentAsync_WithNameAndTools_CreatesAgentAsync() public async Task CreateAIAgentAsync_WithModelAndOptions_CreatesValidAgentAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", instructions: "Test instructions"); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", instructions: "Test instructions"); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -475,7 +475,7 @@ public async Task CreateAIAgentAsync_WithModelAndOptions_CreatesValidAgentAsync( }; // Act - var agent = await client.CreateAIAgentAsync("test-model", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -490,7 +490,7 @@ public async Task CreateAIAgentAsync_WithModelAndOptions_CreatesValidAgentAsync( public async Task CreateAIAgentAsync_WithModelAndOptions_WithClientFactory_AppliesFactoryCorrectlyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", instructions: "Test instructions"); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", instructions: "Test instructions"); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -499,7 +499,7 @@ public async Task CreateAIAgentAsync_WithModelAndOptions_WithClientFactory_Appli TestChatClient? testChatClient = null; // Act - var agent = await client.CreateAIAgentAsync( + var agent = await testClient.Client.CreateAIAgentAsync( "test-model", options, clientFactory: (innerClient) => testChatClient = new TestChatClient(innerClient)); @@ -560,12 +560,12 @@ public async Task CreateAIAgentAsync_WithAgentDefinition_WithNullDefinition_Thro public async Task CreateAIAgentAsync_WithDefinition_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var definition = new PromptAgentDefinition("test-model") { Instructions = "Test" }; var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -582,12 +582,12 @@ public async Task CreateAIAgentAsync_WithoutToolsParameter_CreatesAgentSuccessfu var definition = new PromptAgentDefinition("test-model") { Instructions = "Test" }; var definitionResponse = GeneratePromptDefinitionResponse(definition, null); - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: definitionResponse); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -602,12 +602,12 @@ public async Task CreateAIAgentAsync_WithoutToolsInDefinition_CreatesAgentSucces { // Arrange var definition = new PromptAgentDefinition("test-model") { Instructions = "Test" }; - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: definition); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: definition); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -628,12 +628,12 @@ public async Task CreateAIAgentAsync_WithDefinitionTools_UsesDefinitionToolsAsyn // Create a response definition with the same tool var definitionResponse = GeneratePromptDefinitionResponse(definition, definition.Tools.Select(t => t.AsAITool()).ToList()); - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: definitionResponse); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -667,12 +667,12 @@ public async Task CreateAIAgentAsync_WithMixedToolsInDefinition_CreatesAgentSucc definitionResponse.Tools.Add(tool); } - AIProjectClient client = this.CreateTestAgentClient(agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentDefinitionResponse: definitionResponse); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -803,10 +803,10 @@ public async Task CreateAIAgentAsync_WithStringParamsAndTools_CreatesAgentAsync( var definitionResponse = GeneratePromptDefinitionResponse(new PromptAgentDefinition("test-model") { Instructions = "Test instructions" }, tools); - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: definitionResponse); // Act - var agent = await client.CreateAIAgentAsync( + var agent = await testClient.Client.CreateAIAgentAsync( "test-agent", "test-model", "Test instructions", @@ -831,14 +831,14 @@ public async Task CreateAIAgentAsync_WithStringParamsAndTools_CreatesAgentAsync( public async Task CreateAIAgentAsync_WithDefinitionTools_CreatesAgentAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var definition = new PromptAgentDefinition("test-model") { Instructions = "Test instructions" }; definition.Tools.Add(ResponseTool.CreateFunctionTool("async_tool", BinaryData.FromString("{}"), strictModeEnabled: false)); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -885,7 +885,7 @@ public async Task CreateAIAgentAsync_WithResponseToolsInDefinition_CreatesAgentS var sharepointOptions = new SharePointGroundingToolOptions(); sharepointOptions.ProjectConnections.Add(new ToolProjectConnection("connection-id")); - var structuredOutputs = new StructuredOutputDefinition("name", "description", BinaryData.FromString(AIJsonUtilities.CreateJsonSchema(new { id = "test" }.GetType()).ToString()), false); + var structuredOutputs = new StructuredOutputDefinition("name", "description", new Dictionary { ["schema"] = BinaryData.FromString(AIJsonUtilities.CreateJsonSchema(new { id = "test" }.GetType()).ToString()) }, false); // Add tools to the definition definition.Tools.Add(ResponseTool.CreateFunctionTool("create_tool", BinaryData.FromString("{}"), strictModeEnabled: false)); @@ -902,12 +902,12 @@ public async Task CreateAIAgentAsync_WithResponseToolsInDefinition_CreatesAgentS // Generate agent definition response with the tools var definitionResponse = GeneratePromptDefinitionResponse(definition, definition.Tools.Select(t => t.AsAITool()).ToList()); - AIProjectClient client = this.CreateTestAgentClient(agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentDefinitionResponse: definitionResponse); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -942,12 +942,12 @@ public async Task CreateAIAgentAsync_WithFunctionToolsInDefinition_AcceptsDeclar var definitionResponse = new PromptAgentDefinition("test-model") { Instructions = "Test" }; definitionResponse.Tools.Add(functionTool); - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: definitionResponse); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -961,7 +961,7 @@ public async Task CreateAIAgentAsync_WithFunctionToolsInDefinition_AcceptsDeclar public async Task CreateAIAgentAsync_WithDeclarativeFunctionFromDefinition_AcceptsDeclarativeFunctionAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var definition = new PromptAgentDefinition("test-model") { Instructions = "Test" }; // Create a declarative function (not invocable) using AIFunctionFactory.CreateDeclaration @@ -974,7 +974,7 @@ public async Task CreateAIAgentAsync_WithDeclarativeFunctionFromDefinition_Accep var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -1001,12 +1001,12 @@ public async Task CreateAIAgentAsync_WithDeclarativeFunctionInDefinition_Accepts var definitionResponse = new PromptAgentDefinition("test-model") { Instructions = "Test" }; definitionResponse.Tools.Add(declarativeFunction.AsOpenAIResponseTool() ?? throw new InvalidOperationException()); - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: definitionResponse); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -1027,12 +1027,12 @@ public async Task CreateAIAgentAsync_GeneratesCorrectChatClientAgentOptionsAsync var definition = new PromptAgentDefinition("test-model") { Instructions = "Test instructions" }; var definitionResponse = GeneratePromptDefinitionResponse(definition, null); - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: definitionResponse); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync("test-agent", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-agent", options); // Assert Assert.NotNull(agent); @@ -1083,7 +1083,7 @@ public async Task CreateAIAgentAsync_WithOptionsAndTools_GeneratesCorrectOptions new PromptAgentDefinition("test-model") { Instructions = "Test" }, tools); - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: definitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: definitionResponse); var options = new ChatClientAgentOptions { @@ -1092,7 +1092,7 @@ public async Task CreateAIAgentAsync_WithOptionsAndTools_GeneratesCorrectOptions }; // Act - var agent = await client.CreateAIAgentAsync("test-model", options); + var agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -1278,14 +1278,14 @@ public void AsAIAgent_WithClientFactory_WrapsUnderlyingChatClient() public async Task CreateAIAgentAsync_WithClientFactory_ReceivesCorrectUnderlyingClientAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var definition = new PromptAgentDefinition("test-model") { Instructions = "Test" }; IChatClient? receivedClient = null; var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync( + var agent = await testClient.Client.CreateAIAgentAsync( "test-agent", options, clientFactory: (innerClient) => @@ -1340,10 +1340,10 @@ public async Task CreateAIAgentAsync_WithClientFactory_PreservesAgentPropertiesA const string AgentName = "test-agent"; const string Model = "test-model"; const string Instructions = "Test instructions"; - AIProjectClient client = this.CreateTestAgentClient(AgentName, Instructions); + using var testClient = CreateTestAgentClientWithHandler(AgentName, Instructions); // Act - var agent = await client.CreateAIAgentAsync( + var agent = await testClient.Client.CreateAIAgentAsync( AgentName, Model, Instructions, @@ -1367,12 +1367,12 @@ public async Task CreateAIAgentAsync_WithClientFactory_CreatesAgentSuccessfullyA var definition = new PromptAgentDefinition("test-model") { Instructions = "Test" }; var agentDefinitionResponse = GeneratePromptDefinitionResponse(definition, null); - AIProjectClient client = this.CreateTestAgentClient(agentName: "test-agent", agentDefinitionResponse: agentDefinitionResponse); + using var testClient = CreateTestAgentClientWithHandler(agentName: "test-agent", agentDefinitionResponse: agentDefinitionResponse); var options = new AgentVersionCreationOptions(definition); // Act - var agent = await client.CreateAIAgentAsync( + var agent = await testClient.Client.CreateAIAgentAsync( "test-agent", options, clientFactory: (innerClient) => new TestChatClient(innerClient)); @@ -1390,7 +1390,8 @@ public async Task CreateAIAgentAsync_WithClientFactory_CreatesAgentSuccessfullyA #region User-Agent Header Tests /// - /// Verifies that the user-agent header is added to both synchronous and asynchronous requests made by agent creation methods. + /// Verifies that the MEAI user-agent header is added to CreateAIAgentAsync POST requests + /// via the protocol method's RequestOptions pipeline policy. /// [Fact] public async Task CreateAIAgentAsync_UserAgentHeaderAddedToRequestsAsync() @@ -1398,9 +1399,12 @@ public async Task CreateAIAgentAsync_UserAgentHeaderAddedToRequestsAsync() using var httpHandler = new HttpHandlerAssert(request => { Assert.Equal("POST", request.Method.Method); - Assert.Contains("MEAI", request.Headers.UserAgent.ToString()); - return new HttpResponseMessage(HttpStatusCode.OK) { Content = new StringContent(TestDataUtil.GetAgentResponseJson(), Encoding.UTF8, "application/json") }; + // Verify MEAI user-agent header is present on CreateAgentVersion POST request + Assert.True(request.Headers.TryGetValues("User-Agent", out var userAgentValues)); + Assert.Contains(userAgentValues, v => v.Contains("MEAI")); + + return new HttpResponseMessage(HttpStatusCode.OK) { Content = new StringContent(TestDataUtil.GetAgentVersionResponseJson(), Encoding.UTF8, "application/json") }; }); #pragma warning disable CA5399 @@ -1940,7 +1944,7 @@ public async Task CreateAIAgentAsync_WithModelAndOptions_WithWhitespaceName_Thro public async Task CreateAIAgentAsync_WithTextResponseFormat_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -1952,7 +1956,7 @@ public async Task CreateAIAgentAsync_WithTextResponseFormat_CreatesAgentSuccessf }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -1966,7 +1970,7 @@ public async Task CreateAIAgentAsync_WithTextResponseFormat_CreatesAgentSuccessf public async Task CreateAIAgentAsync_WithJsonResponseFormatWithoutSchema_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -1978,7 +1982,7 @@ public async Task CreateAIAgentAsync_WithJsonResponseFormatWithoutSchema_Creates }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -1992,7 +1996,7 @@ public async Task CreateAIAgentAsync_WithJsonResponseFormatWithoutSchema_Creates public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchema_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); JsonElement schemaElement = AIJsonUtilities.CreateJsonSchema(typeof(TestSchema)); var jsonFormat = ChatResponseFormat.ForJsonSchema(schemaElement, "test_schema", "A test schema"); var options = new ChatClientAgentOptions @@ -2006,7 +2010,7 @@ public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchema_CreatesAge }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2020,7 +2024,7 @@ public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchema_CreatesAge public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchemaAndStrictMode_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); JsonElement schemaElement = AIJsonUtilities.CreateJsonSchema(typeof(TestSchema)); var jsonFormat = ChatResponseFormat.ForJsonSchema(schemaElement, "test_schema", "A test schema"); var additionalProps = new AdditionalPropertiesDictionary @@ -2039,7 +2043,7 @@ public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchemaAndStrictMo }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2053,7 +2057,7 @@ public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchemaAndStrictMo public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchemaAndStrictModeFalse_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); JsonElement schemaElement = AIJsonUtilities.CreateJsonSchema(typeof(TestSchema)); var jsonFormat = ChatResponseFormat.ForJsonSchema(schemaElement, "test_schema", "A test schema"); var additionalProps = new AdditionalPropertiesDictionary @@ -2072,7 +2076,7 @@ public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchemaAndStrictMo }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2090,7 +2094,7 @@ public async Task CreateAIAgentAsync_WithJsonResponseFormatWithSchemaAndStrictMo public async Task CreateAIAgentAsync_WithRawRepresentationFactory_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -2102,7 +2106,7 @@ public async Task CreateAIAgentAsync_WithRawRepresentationFactory_CreatesAgentSu }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2116,7 +2120,7 @@ public async Task CreateAIAgentAsync_WithRawRepresentationFactory_CreatesAgentSu public async Task CreateAIAgentAsync_WithRawRepresentationFactoryReturningNull_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -2128,7 +2132,7 @@ public async Task CreateAIAgentAsync_WithRawRepresentationFactoryReturningNull_C }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2142,7 +2146,7 @@ public async Task CreateAIAgentAsync_WithRawRepresentationFactoryReturningNull_C public async Task CreateAIAgentAsync_WithRawRepresentationFactoryReturningNonCreateResponseOptions_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -2154,7 +2158,7 @@ public async Task CreateAIAgentAsync_WithRawRepresentationFactoryReturningNonCre }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2172,7 +2176,7 @@ public async Task CreateAIAgentAsync_WithRawRepresentationFactoryReturningNonCre public async Task CreateAIAgentAsync_WithDescription_SetsDescriptionAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(description: "Test description"); + using var testClient = CreateTestAgentClientWithHandler(description: "Test description"); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -2181,7 +2185,7 @@ public async Task CreateAIAgentAsync_WithDescription_SetsDescriptionAsync() }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2195,7 +2199,7 @@ public async Task CreateAIAgentAsync_WithDescription_SetsDescriptionAsync() public async Task CreateAIAgentAsync_WithoutDescription_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var options = new ChatClientAgentOptions { Name = "test-agent", @@ -2203,7 +2207,7 @@ public async Task CreateAIAgentAsync_WithoutDescription_CreatesAgentSuccessfully }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2624,7 +2628,7 @@ public async Task CreateAIAgentAsync_WithResponseToolAsAITool_CreatesAgentSucces public async Task CreateAIAgentAsync_WithHostedToolTypes_CreatesAgentSuccessfullyAsync() { // Arrange - AIProjectClient client = this.CreateTestAgentClient(); + using var testClient = CreateTestAgentClientWithHandler(); var webSearchTool = new HostedWebSearchTool(); var options = new ChatClientAgentOptions @@ -2638,7 +2642,7 @@ public async Task CreateAIAgentAsync_WithHostedToolTypes_CreatesAgentSuccessfull }; // Act - ChatClientAgent agent = await client.CreateAIAgentAsync("test-model", options); + ChatClientAgent agent = await testClient.Client.CreateAIAgentAsync("test-model", options); // Assert Assert.NotNull(agent); @@ -2791,6 +2795,54 @@ private FakeAgentClient CreateTestAgentClient(string? agentName = null, string? return new FakeAgentClient(agentName, instructions, description, agentDefinitionResponse); } + /// + /// Creates a test AIProjectClient backed by an HTTP handler that returns canned responses. + /// Used for tests that exercise the protocol-method code path (CreateAgentVersion). + /// The returned client must be disposed to clean up the underlying HttpClient/handler. + /// + private static DisposableTestClient CreateTestAgentClientWithHandler(string? agentName = null, string? instructions = null, string? description = null, AgentDefinition? agentDefinitionResponse = null) + { + var responseJson = TestDataUtil.GetAgentVersionResponseJson(agentName, agentDefinitionResponse, instructions, description); + + var httpHandler = new HttpHandlerAssert(_ => + new HttpResponseMessage(HttpStatusCode.OK) { Content = new StringContent(responseJson, Encoding.UTF8, "application/json") }); + +#pragma warning disable CA5399 + var httpClient = new HttpClient(httpHandler); +#pragma warning restore CA5399 + + var client = new AIProjectClient( + new Uri("https://test.openai.azure.com/"), + new FakeAuthenticationTokenProvider(), + new() { Transport = new HttpClientPipelineTransport(httpClient) }); + + return new DisposableTestClient(client, httpClient, httpHandler); + } + + /// + /// Wraps an AIProjectClient and its disposable dependencies for deterministic cleanup. + /// + private sealed class DisposableTestClient : IDisposable + { + private readonly HttpClient _httpClient; + private readonly HttpHandlerAssert _httpHandler; + + public DisposableTestClient(AIProjectClient client, HttpClient httpClient, HttpHandlerAssert httpHandler) + { + this.Client = client; + this._httpClient = httpClient; + this._httpHandler = httpHandler; + } + + public AIProjectClient Client { get; } + + public void Dispose() + { + this._httpClient.Dispose(); + this._httpHandler.Dispose(); + } + } + /// /// Creates a test AgentRecord for testing. /// @@ -2975,25 +3027,13 @@ public override Task> GetAgentAsync(string agentName, return Task.FromResult(ClientResult.FromValue(ModelReaderWriter.Read(BinaryData.FromString(responseJson))!, new MockPipelineResponse(200))); } - public override ClientResult CreateAgentVersion(string agentName, BinaryContent content, RequestOptions? options = null) - { - var responseJson = this.GetAgentVersionResponseJson(); - return ClientResult.FromValue(ModelReaderWriter.Read(BinaryData.FromString(responseJson))!, new MockPipelineResponse(200, BinaryData.FromString(responseJson))); - } - - public override ClientResult CreateAgentVersion(string agentName, AgentVersionCreationOptions? options = null, CancellationToken cancellationToken = default) + public override ClientResult CreateAgentVersion(string agentName, AgentVersionCreationOptions? options = null, string? foundryFeatures = null, CancellationToken cancellationToken = default) { var responseJson = this.GetAgentVersionResponseJson(); return ClientResult.FromValue(ModelReaderWriter.Read(BinaryData.FromString(responseJson))!, new MockPipelineResponse(200)); } - public override Task CreateAgentVersionAsync(string agentName, BinaryContent content, RequestOptions? options = null) - { - var responseJson = this.GetAgentVersionResponseJson(); - return Task.FromResult(ClientResult.FromValue(ModelReaderWriter.Read(BinaryData.FromString(responseJson))!, new MockPipelineResponse(200, BinaryData.FromString(responseJson)))); - } - - public override Task> CreateAgentVersionAsync(string agentName, AgentVersionCreationOptions? options = null, CancellationToken cancellationToken = default) + public override Task> CreateAgentVersionAsync(string agentName, AgentVersionCreationOptions? options = null, string? foundryFeatures = null, CancellationToken cancellationToken = default) { var responseJson = this.GetAgentVersionResponseJson(); return Task.FromResult(ClientResult.FromValue(ModelReaderWriter.Read(BinaryData.FromString(responseJson))!, new MockPipelineResponse(200))); diff --git a/dotnet/tests/Microsoft.Agents.AI.AzureAI.UnitTests/AzureAIProjectChatClientTests.cs b/dotnet/tests/Microsoft.Agents.AI.AzureAI.UnitTests/AzureAIProjectChatClientTests.cs index 9cc340ef5e..5c61e0b457 100644 --- a/dotnet/tests/Microsoft.Agents.AI.AzureAI.UnitTests/AzureAIProjectChatClientTests.cs +++ b/dotnet/tests/Microsoft.Agents.AI.AzureAI.UnitTests/AzureAIProjectChatClientTests.cs @@ -22,7 +22,7 @@ public async Task ChatClient_UsesDefaultConversationIdAsync() var requestTriggered = false; using var httpHandler = new HttpHandlerAssert(async (request) => { - if (request.RequestUri!.PathAndQuery.Contains("openai/responses")) + if (request.Method == HttpMethod.Post && request.RequestUri!.PathAndQuery.Contains("/responses")) { requestTriggered = true; @@ -71,7 +71,7 @@ public async Task ChatClient_UsesPerRequestConversationId_WhenNoDefaultConversat var requestTriggered = false; using var httpHandler = new HttpHandlerAssert(async (request) => { - if (request.RequestUri!.PathAndQuery.Contains("openai/responses")) + if (request.Method == HttpMethod.Post && request.RequestUri!.PathAndQuery.Contains("/responses")) { requestTriggered = true; @@ -120,7 +120,7 @@ public async Task ChatClient_UsesPerRequestConversationId_EvenWhenDefaultConvers var requestTriggered = false; using var httpHandler = new HttpHandlerAssert(async (request) => { - if (request.RequestUri!.PathAndQuery.Contains("openai/responses")) + if (request.Method == HttpMethod.Post && request.RequestUri!.PathAndQuery.Contains("/responses")) { requestTriggered = true; @@ -169,7 +169,7 @@ public async Task ChatClient_UsesPreviousResponseId_WhenConversationIsNotPrefixe var requestTriggered = false; using var httpHandler = new HttpHandlerAssert(async (request) => { - if (request.RequestUri!.PathAndQuery.Contains("openai/responses")) + if (request.Method == HttpMethod.Post && request.RequestUri!.PathAndQuery.Contains("/responses")) { requestTriggered = true;