Skip to content

azure-ai-projects test re-recordings and alignment with Python on code customizations#48504

Open
jpalvarezl wants to merge 8 commits intomainfrom
jpalvarezl/feature/test_recordings_projects
Open

azure-ai-projects test re-recordings and alignment with Python on code customizations#48504
jpalvarezl wants to merge 8 commits intomainfrom
jpalvarezl/feature/test_recordings_projects

Conversation

@jpalvarezl
Copy link
Member

Human summary

  • Gap analysis against the python SDK
  • Re-recorded tests
  • Adjusted old customizations.

AI Summary

This pull request introduces several enhancements and API improvements to the Java SDK for Azure AI Projects, particularly around dataset creation and connection management. The main focus is on adding support for specifying a storage connection when uploading datasets (both files and folders), aligning the Java SDK more closely with the Python SDK, and introducing a new method for retrieving the default connection for a given type. Additionally, supporting documentation and metadata have been updated.

Key changes:

Dataset Upload Enhancements

  • Added connectionName parameter to createDatasetWithFile and createDatasetWithFolder methods in both DatasetsClient and DatasetsAsyncClient, allowing users to specify which Azure Storage Account connection to use for uploads. This brings the Java SDK in line with the Python SDK's capabilities. [1] [2] [3] [4]
  • Refactored file and folder upload logic to use the correct blob/file names and improved handling of blob container clients for uploading multiple files. [1] [2] [3] [4]

Connection Management Improvements

  • Introduced getDefaultConnection(ConnectionType connectionType, boolean includeCredentials) in both ConnectionsClient and ConnectionsAsyncClient. This method retrieves the default connection for a given type, mirroring the Python SDK's functionality. [1] [2]

Documentation and Metadata

  • Added a new markdown file, PYTHON_GAPS.md, providing a detailed comparison of features and gaps between the Python and Java SDKs for Azure AI Projects.
  • Updated the assets.json tag to reference the latest asset version.

Dependency Updates

  • Added missing imports for BlobContainerClient and BlobContainerAsyncClient where required for the new upload logic. [1] [2]

Copilot AI review requested due to automatic review settings March 21, 2026 14:53
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 updates azure-ai-projects to align Java customizations and test coverage with the Python SDK, primarily around dataset upload behavior and connection management, while also re-enabling/reworking previously disabled tests and refreshing assets/metadata.

Changes:

  • Added connectionName support to dataset upload helpers (file/folder) in sync and async clients.
  • Added getDefaultConnection(connectionType, includeCredentials) to sync and async connections clients.
  • Reworked tests (including re-enabling) and updated assets + added a Python-vs-Java gap analysis doc.

Reviewed changes

Copilot reviewed 22 out of 22 changed files in this pull request and generated 8 comments.

Show a summary per file
File Description
sdk/ai/azure-ai-projects/src/main/java/com/azure/ai/projects/DatasetsClient.java Adds connectionName overloads and refactors upload logic.
sdk/ai/azure-ai-projects/src/main/java/com/azure/ai/projects/DatasetsAsyncClient.java Adds connectionName overloads and refactors upload logic (reactive).
sdk/ai/azure-ai-projects/src/main/java/com/azure/ai/projects/ConnectionsClient.java Adds getDefaultConnection convenience API.
sdk/ai/azure-ai-projects/src/main/java/com/azure/ai/projects/ConnectionsAsyncClient.java Adds async getDefaultConnection convenience API.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/ClientTestBase.java Updates test proxy sanitization and logging options for tests.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/DatasetsClientTest.java Reworks dataset tests; marks create/get/delete flows as LiveOnly and adds cleanup.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/DatasetsAsyncClientTest.java Reworks async dataset tests; marks create/get/delete flows as LiveOnly and adds cleanup.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/ConnectionsClientTest.java Reworks connection tests and adds coverage for default connection behavior.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/ConnectionsAsyncClientTest.java Reworks async connection tests and adds coverage for default connection behavior.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/IndexesClientTest.java Re-enables and refactors index tests to discover existing resources dynamically.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/IndexesAsyncClientTest.java Re-enables and refactors async index tests; adds create/update + cleanup.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/DeploymentsClientTest.java Re-enables deployments test class (removes Disabled).
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/DeploymentsAsyncClientTest.java Re-enables async deployments test class (removes Disabled).
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/EvalsServiceTest.java Removes flaky/disabled eval service test class.
sdk/ai/azure-ai-projects/src/test/java/com/azure/ai/projects/EvalServiceAsyncTest.java Removes flaky/disabled async eval service test class.
sdk/ai/azure-ai-projects/src/samples/java/com/azure/ai/projects/TelemetrySample.java Removes commented-out sample stub.
sdk/ai/azure-ai-projects/src/samples/java/com/azure/ai/projects/TelemetryAsyncSample.java Removes commented-out sample stub.
sdk/ai/azure-ai-projects/src/samples/java/com/azure/ai/projects/InferenceOpenAISample.java Removes commented-out sample stub.
sdk/ai/azure-ai-projects/src/samples/java/com/azure/ai/projects/EvaluationsSample.java Removes commented-out sample stub.
sdk/ai/azure-ai-projects/src/samples/java/com/azure/ai/projects/EvaluationsAsyncSample.java Removes commented-out sample stub.
sdk/ai/azure-ai-projects/assets.json Updates asset tag reference.
sdk/ai/azure-ai-projects/PYTHON_GAPS.md Adds documented gap analysis vs Python SDK custom code.

@github-actions
Copy link
Contributor

github-actions bot commented Mar 21, 2026

API Change Check

APIView identified API level changes in this PR and created the following API reviews

com.azure:azure-ai-projects

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants