Updating the azure-ai-agents package to the latest spec

.vscode/cspell.json

@@ -359,6 +359,7 @@
     "Mordor",
     "mosca",
     "mpga",
+    "mpkjc",
     "msal",
     "msix",
     "MSRC",

sdk/ai/STAINLESS_DUPLICATES.md

@@ -0,0 +1,27 @@
+# OpenAI Java duplicate model check (full repo)
+
+## Scope
+- Repository: `azure-sdk-for-java` (full scan)
+- `com.openai:openai-java` dependency found in:
+  - `sdk/ai/azure-ai-agents`
+  - `sdk/ai/azure-ai-projects`
+  - `sdk/openai/azure-ai-openai-stainless`
+- OpenAI model shapes sourced from **openai-java-core 4.14.0** (the runtime classes behind the `openai-java` BOM).
+- Model name overlap scan performed across `src/main/java/**/models/**` in each module, followed by field-by-field checks for overlaps.
+
+## Likely duplicates (shape match)
+> Found only in `sdk/ai/azure-ai-agents` based on full-repo scan.
+
+| Azure model | OpenAI model | Matching fields / notes |
+| --- | --- | --- |
+| `com.azure.ai.agents.models.ComparisonFilter` | `com.openai.models.ComparisonFilter` | Fields: `type`, `key`, `value` |
+| `com.azure.ai.agents.models.CompoundFilter` | `com.openai.models.CompoundFilter` | Fields: `type`, `filters` |
+| `com.azure.ai.agents.models.FunctionTool` | `com.openai.models.responses.FunctionTool` | Fields: `name`, `parameters`, `strict`, `type`, `description` |
+| `com.azure.ai.agents.models.FileSearchTool` | `com.openai.models.responses.FileSearchTool` | Fields: `type`, `vectorStoreIds`, `filters`, `maxNumResults` (Azure: `maxResults`), `rankingOptions` |
+| `com.azure.ai.agents.models.Reasoning` | `com.openai.models.Reasoning` | Fields: `effort`, `summary`, `generateSummary` |
+| `com.azure.ai.agents.models.ReasoningEffort` | `com.openai.models.ReasoningEffort` | Enum values: `none`, `minimal`, `low`, `medium`, `high`, `xhigh` |
+| `com.azure.ai.agents.models.WebSearchPreviewTool` | `com.openai.models.responses.WebSearchPreviewTool` | Fields: `type`, `userLocation`, `searchContextSize` |
+
+## Notes on other modules
+- **`sdk/ai/azure-ai-projects`**: no model name overlaps with OpenAI Java models found in the scan; no likely duplicates identified.
+- **`sdk/openai/azure-ai-openai-stainless`**: no `models` Java classes under `src/main/java`, so no duplicates identified.

sdk/ai/azure-ai-agents/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "java",
   "TagPrefix": "java/ai/azure-ai-agents",
-  "Tag": "java/ai/azure-ai-agents_3f32cd8dff"
+  "Tag": "java/ai/azure-ai-agents_ab6a883fcd"
 }

sdk/ai/azure-ai-agents/checkstyle-suppressions.xml

@@ -19,4 +19,6 @@
   <suppress files="com.azure.ai.agents.ResponsesClient.java" checks="io.clientcore.linting.extensions.checkstyle.checks.ExternalDependencyExposedCheck" />
   <suppress files="com.azure.ai.agents.ConversationsAsyncClient.java" checks="io.clientcore.linting.extensions.checkstyle.checks.ExternalDependencyExposedCheck" />
   <suppress files="com.azure.ai.agents.ConversationsClient.java" checks="io.clientcore.linting.extensions.checkstyle.checks.ExternalDependencyExposedCheck" />
+  <suppress files="com.azure.ai.agents.MemoryStoresAsyncClient.java" checks="io.clientcore.linting.extensions.checkstyle.checks.ExternalDependencyExposedCheck" />
+  <suppress files="com.azure.ai.agents.MemoryStoresClient.java" checks="io.clientcore.linting.extensions.checkstyle.checks.ExternalDependencyExposedCheck" />
 </suppressions>

sdk/ai/azure-ai-agents/com/azure/core/test/TestProxyTestBase.java

@@ -0,0 +1,35 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+package com.azure.core.test;
+
+import com.azure.core.test.utils.TestProxyManager;
+import org.junit.jupiter.api.BeforeAll;
+
+/**
+ * Base class for running live and playback tests using test-proxy
+ */
+public abstract class TestProxyTestBase extends TestBase {
+    static {
+        setTestProxyEnabled();
+    }
+
+    /**
+     * Creates an instance of {@link TestProxyTestBase}.
+     */
+    public TestProxyTestBase() {
+        super();
+    }
+
+    /**
+     * Before tests are executed, determines the test mode by reading the {@code AZURE_TEST_MODE} environment variable.
+     * If it is not set, {@link TestMode#PLAYBACK}
+     */
+    @BeforeAll
+    public static void setupTestProxy() {
+        testMode = initializeTestMode();
+        if (isTestProxyEnabled() && (testMode == TestMode.PLAYBACK || testMode == TestMode.RECORD)) {
+            TestProxyManager.startProxy();
+        }
+    }
+}

sdk/ai/azure-ai-agents/customizations/src/main/java/AgentsCustomizations.java

@@ -11,44 +11,28 @@ public class AgentsCustomizations extends Customization {
 
     @Override
     public void customize(LibraryCustomization libraryCustomization, Logger logger) {
-        renameImageGenSizeEnums(libraryCustomization, logger);
+        renameContainerMemoryLimitVariants(libraryCustomization, logger);
     }
 
-    /**
-     * Customization for enum values that are originally numbers and get transliterated by the emitter.
-     * @param customization
-     * @param logger
-     */
-    private void renameImageGenSizeEnums(LibraryCustomization customization, Logger logger) {
-        logger.info("Renaming enum ImageGenToolSize variants");
-        customization.getClass("com.azure.ai.agents.models", "ImageGenToolSize").customizeAst(ast -> ast.getEnumByName("ImageGenToolSize")
-                .ifPresent(clazz -> clazz.getEntries().stream()
-                        .filter(entry -> "ONE_ZERO_TWO_FOURX_ONE_ZERO_TWO_FOUR".equals(entry.getName().getIdentifier()))
-                        .forEach(entry -> entry.setName("SIZE_1024X1024"))));
-
-        customization.getClass("com.azure.ai.agents.models", "ImageGenToolSize").customizeAst(ast -> ast.getEnumByName("ImageGenToolSize")
-                .ifPresent(clazz -> clazz.getEntries().stream()
-                        .filter(entry -> "ONE_ZERO_TWO_FOURX_ONE_FIVE_THREE_SIX".equals(entry.getName().getIdentifier()))
-                        .forEach(entry -> entry.setName("SIZE_1024X1536"))));
-
-        customization.getClass("com.azure.ai.agents.models", "ImageGenToolSize").customizeAst(ast -> ast.getEnumByName("ImageGenToolSize")
-                .ifPresent(clazz -> clazz.getEntries().stream()
-                        .filter(entry -> "ONE_FIVE_THREE_SIXX_ONE_ZERO_TWO_FOUR".equals(entry.getName().getIdentifier()))
-                        .forEach(entry -> entry.setName("SIZE_1536X1024"))));
-
-        customization.getClass("com.azure.ai.agents.models", "ComputerActionType").customizeAst(ast -> ast.getEnumByName("ComputerActionType")
-                .ifPresent(clazz -> clazz.getEntries().stream()
-                        .filter(entry -> "KEYPRESS".equals(entry.getName().getIdentifier()))
-                        .forEach(entry -> entry.setName("KEY_PRESS"))));
-
-        customization.getClass("com.azure.ai.agents.models", "ComputerActionKeyPress")
-                .customizeAst(ast -> ast.getClassByName("ComputerActionKeyPress")
-                        .flatMap(clazz -> clazz.getFieldByName("type"))
-                        .ifPresent(barField ->
-                                barField.getVariables().forEach(var -> {
-                                    if (var.getNameAsString().equals("type")) {
-                                        var.setInitializer("ComputerActionType.KEY_PRESS");
-                                    }
-                                })));
+    private void renameContainerMemoryLimitVariants(LibraryCustomization customization, Logger logger) {
+        customization.getClass("com.azure.ai.agents.models", "ContainerMemoryLimit").customizeAst(ast -> ast.getEnumByName("ContainerMemoryLimit")
+            .ifPresent(clazz -> clazz.getEntries().stream()
+                .filter(entry -> "ONEG".equals(entry.getName().getIdentifier()))
+                .forEach(entry -> entry.setName("ONE_G"))));
+
+        customization.getClass("com.azure.ai.agents.models", "ContainerMemoryLimit").customizeAst(ast -> ast.getEnumByName("ContainerMemoryLimit")
+            .ifPresent(clazz -> clazz.getEntries().stream()
+                .filter(entry -> "FOURG".equals(entry.getName().getIdentifier()))
+                .forEach(entry -> entry.setName("FOUR_G"))));
+
+        customization.getClass("com.azure.ai.agents.models", "ContainerMemoryLimit").customizeAst(ast -> ast.getEnumByName("ContainerMemoryLimit")
+            .ifPresent(clazz -> clazz.getEntries().stream()
+                .filter(entry -> "ONE_SIXG".equals(entry.getName().getIdentifier()))
+                .forEach(entry -> entry.setName("SIXTEEN_G"))));
+
+        customization.getClass("com.azure.ai.agents.models", "ContainerMemoryLimit").customizeAst(ast -> ast.getEnumByName("ContainerMemoryLimit")
+            .ifPresent(clazz -> clazz.getEntries().stream()
+                .filter(entry -> "SIX_FOURG".equals(entry.getName().getIdentifier()))
+                .forEach(entry -> entry.setName("SIXTY_FOUR_G"))));
     }
-}
+}

sdk/ai/azure-ai-agents/pom.xml

@@ -46,6 +46,10 @@
     <!-- TODO: remove on Public Preview release -->
     <jacoco.min.linecoverage>0.0</jacoco.min.linecoverage>
     <jacoco.min.branchcoverage>0.0</jacoco.min.branchcoverage>
+    <javaModulesSurefireArgLine>
+      --add-exports com.azure.core/com.azure.core.implementation.util=ALL-UNNAMED
+      --add-opens com.azure.core/com.azure.core.implementation.util=ALL-UNNAMED
+    </javaModulesSurefireArgLine>
   </properties>
   <dependencies>
     <!-- compile scope -->
@@ -78,12 +82,6 @@
       <version>1.27.0-beta.14</version> <!-- {x-version-update;com.azure:azure-core-test;dependency} -->
       <scope>test</scope>
     </dependency>
-    <dependency>
-      <groupId>com.azure</groupId>
-      <artifactId>azure-core-http-okhttp</artifactId>
-      <version>1.13.3</version> <!-- {x-version-update;com.azure:azure-core-http-okhttp;dependency} -->
-      <scope>test</scope>
-    </dependency>
 
   </dependencies>
 
@@ -110,4 +108,17 @@
       </plugin>
     </plugins>
   </build>
+
+  <repositories>
+    <repository>
+      <id>azure-sdk-for-java</id>
+      <url>https://pkgs.dev.azure.com/azure-sdk/public/_packaging/azure-sdk-for-java/maven/v1</url>
+      <releases>
+        <enabled>true</enabled>
+      </releases>
+      <snapshots>
+        <enabled>true</enabled>
+      </snapshots>
+    </repository>
+  </repositories>
 </project>