diff --git a/sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/ResponsesAsyncClient.java b/sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/ResponsesAsyncClient.java index 52c82be1ff2b..7d7bed43dc0f 100644 --- a/sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/ResponsesAsyncClient.java +++ b/sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/ResponsesAsyncClient.java @@ -7,7 +7,9 @@ import com.azure.ai.agents.implementation.OpenAIJsonHelper; import com.azure.ai.agents.implementation.StreamingUtils; import com.azure.ai.agents.models.AzureCreateResponseOptions; +import com.azure.core.annotation.ReturnType; import com.azure.core.annotation.ServiceClient; +import com.azure.core.annotation.ServiceMethod; import com.openai.client.OpenAIClientAsync; import com.openai.core.JsonValue; import com.openai.models.responses.Response; @@ -53,6 +55,7 @@ public ResponseServiceAsync getResponseServiceAsync() { * @param params The parameters to create the response. * @return The created Response. */ + @ServiceMethod(returns = ReturnType.SINGLE) public Mono createAzureResponse(AzureCreateResponseOptions createResponse, ResponseCreateParams.Builder params) { Objects.requireNonNull(createResponse, "createResponse cannot be null"); @@ -71,6 +74,7 @@ public Mono createAzureResponse(AzureCreateResponseOptions createRespo * @param params The parameters to create the response. * @return A {@link Flux} stream of {@link ResponseStreamEvent} items. */ + @ServiceMethod(returns = ReturnType.SINGLE) public Flux createStreamingAzureResponse(AzureCreateResponseOptions createResponse, ResponseCreateParams.Builder params) { Objects.requireNonNull(createResponse, "createResponse cannot be null"); diff --git a/sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/ResponsesClient.java b/sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/ResponsesClient.java index d4fa435cc810..73bb2774919a 100644 --- a/sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/ResponsesClient.java +++ b/sdk/ai/azure-ai-agents/src/main/java/com/azure/ai/agents/ResponsesClient.java @@ -8,6 +8,8 @@ import com.azure.ai.agents.implementation.StreamingUtils; import com.azure.ai.agents.models.AzureCreateResponseOptions; import com.azure.core.annotation.ServiceClient; +import com.azure.core.annotation.ServiceMethod; +import com.azure.core.annotation.ReturnType; import com.azure.core.util.IterableStream; import com.openai.client.OpenAIClient; import com.openai.core.JsonValue; @@ -52,6 +54,7 @@ public ResponseService getResponseService() { * @param params The parameters to create the response. * @return The created Response. */ + @ServiceMethod(returns = ReturnType.SINGLE) public Response createAzureResponse(AzureCreateResponseOptions createResponse, ResponseCreateParams.Builder params) { Objects.requireNonNull(createResponse, "createResponse cannot be null"); @@ -70,6 +73,7 @@ public Response createAzureResponse(AzureCreateResponseOptions createResponse, * @param params The parameters to create the response. * @return An IterableStream of ResponseStreamEvent. */ + @ServiceMethod(returns = ReturnType.SINGLE) public IterableStream createStreamingAzureResponse(AzureCreateResponseOptions createResponse, ResponseCreateParams.Builder params) { Objects.requireNonNull(createResponse, "createResponse cannot be null");