Another round of 1DP Python Projects SDK updates (#40444)

Author: dargilco

sdk/ai/azure-ai-projects-onedp/apiview-properties.json

@@ -1,40 +1,44 @@
 {
     "CrossLanguagePackageId": "Azure.AI.Projects",
     "CrossLanguageDefinitionId": {
+        "azure.ai.projects.onedp.models.BaseCredentials": "Azure.AI.Projects.BaseCredentials",
+        "azure.ai.projects.onedp.models.ApiKeyCredentials": "Azure.AI.Projects.ApiKeyCredentials",
         "azure.ai.projects.onedp.models.Index": "Azure.AI.Projects.Index",
         "azure.ai.projects.onedp.models.AzureAISearchIndex": "Azure.AI.Projects.AzureAISearchIndex",
         "azure.ai.projects.onedp.models.BlobReferenceForConsumption": "Azure.AI.Projects.BlobReferenceForConsumption",
         "azure.ai.projects.onedp.models.Connection": "Azure.AI.Projects.Connection",
         "azure.ai.projects.onedp.models.CosmosDBIndex": "Azure.AI.Projects.CosmosDBIndex",
+        "azure.ai.projects.onedp.models.CustomCredential": "Azure.AI.Projects.CustomCredential",
         "azure.ai.projects.onedp.models.DatasetVersion": "Azure.AI.Projects.DatasetVersion",
         "azure.ai.projects.onedp.models.Deployment": "Azure.AI.Projects.Deployment",
         "azure.ai.projects.onedp.models.EmbeddingConfiguration": "Azure.AI.Projects.EmbeddingConfiguration",
+        "azure.ai.projects.onedp.models.EntraIDCredentials": "Azure.AI.Projects.EntraIDCredentials",
         "azure.ai.projects.onedp.models.Evaluation": "Azure.AI.Projects.Evaluation",
-        "azure.ai.projects.onedp.models.EvaluationResult": "Azure.AI.Projects.EvaluationResult",
         "azure.ai.projects.onedp.models.EvaluatorConfiguration": "Azure.AI.Projects.EvaluatorConfiguration",
         "azure.ai.projects.onedp.models.FileDatasetVersion": "Azure.AI.Projects.FileDatasetVersion",
         "azure.ai.projects.onedp.models.FolderDatasetVersion": "Azure.AI.Projects.FolderDatasetVersion",
         "azure.ai.projects.onedp.models.InputData": "Azure.AI.Projects.InputData",
         "azure.ai.projects.onedp.models.InputDataset": "Azure.AI.Projects.InputDataset",
         "azure.ai.projects.onedp.models.ManagedAzureAISearchIndex": "Azure.AI.Projects.ManagedAzureAISearchIndex",
         "azure.ai.projects.onedp.models.ModelDeployment": "Azure.AI.Projects.ModelDeployment",
+        "azure.ai.projects.onedp.models.NoAuthenticationCredentials": "Azure.AI.Projects.NoAuthenticationCredentials",
         "azure.ai.projects.onedp.models.PendingUploadRequest": "Azure.AI.Projects.PendingUploadRequest",
         "azure.ai.projects.onedp.models.PendingUploadResponse": "Azure.AI.Projects.PendingUploadResponse",
         "azure.ai.projects.onedp.models.RedTeam": "Azure.AI.Projects.RedTeam",
         "azure.ai.projects.onedp.models.SasCredential": "Azure.AI.Projects.SasCredential",
+        "azure.ai.projects.onedp.models.SASCredentials": "Azure.AI.Projects.SASCredentials",
         "azure.ai.projects.onedp.models.Sku": "Azure.AI.Projects.Sku",
         "azure.ai.projects.onedp.models.ConnectionType": "Azure.AI.Projects.ConnectionType",
-        "azure.ai.projects.onedp.models.AuthenticationType": "Azure.AI.Projects.AuthenticationType",
+        "azure.ai.projects.onedp.models.CredentialType": "Azure.AI.Projects.CredentialType",
         "azure.ai.projects.onedp.models.DatasetType": "Azure.AI.Projects.DatasetType",
         "azure.ai.projects.onedp.models.ListViewType": "Azure.AI.Projects.ListViewType",
-        "azure.ai.projects.onedp.models.RepeatabilityResult": "Azure.Core.RepeatabilityResult",
         "azure.ai.projects.onedp.models.PendingUploadType": "Azure.AI.Projects.PendingUploadType",
         "azure.ai.projects.onedp.models.IndexType": "Azure.AI.Projects.IndexType",
         "azure.ai.projects.onedp.models.DeploymentType": "Azure.AI.Projects.DeploymentType",
-        "azure.ai.projects.onedp.models.ResultType": "Azure.AI.Projects.ResultType",
         "azure.ai.projects.onedp.models.AttackStrategy": "Azure.AI.Projects.AttackStrategy",
         "azure.ai.projects.onedp.models.RiskCategory": "Azure.AI.Projects.RiskCategory",
         "azure.ai.projects.onedp.AIProjectClient.connections.get": "Azure.AI.Projects.Connections.get",
+        "azure.ai.projects.onedp.AIProjectClient.connections.get_with_credentials": "Azure.AI.Projects.Connections.getWithCredentials",
         "azure.ai.projects.onedp.AIProjectClient.connections.list": "Azure.AI.Projects.Connections.list",
         "azure.ai.projects.onedp.AIProjectClient.evaluations.get": "Azure.AI.Projects.Evaluations.get",
         "azure.ai.projects.onedp.AIProjectClient.evaluations.list": "Azure.AI.Projects.Evaluations.list",
@@ -43,25 +47,15 @@
         "azure.ai.projects.onedp.AIProjectClient.datasets.list_latest": "Azure.AI.Projects.ServicePatterns.Datasets.listLatest",
         "azure.ai.projects.onedp.AIProjectClient.datasets.get_version": "Azure.AI.Projects.ServicePatterns.Datasets.getVersion",
         "azure.ai.projects.onedp.AIProjectClient.datasets.delete_version": "Azure.AI.Projects.ServicePatterns.Datasets.deleteVersion",
-        "azure.ai.projects.onedp.AIProjectClient.datasets.create": "Azure.AI.Projects.ServicePatterns.Datasets.create",
         "azure.ai.projects.onedp.AIProjectClient.datasets.create_version": "Azure.AI.Projects.ServicePatterns.Datasets.createVersion",
         "azure.ai.projects.onedp.AIProjectClient.datasets.start_pending_upload_version": "Azure.AI.Projects.Datasets.startPendingUploadVersion",
-        "azure.ai.projects.onedp.AIProjectClient.datasets.start_pending_upload": "Azure.AI.Projects.Datasets.startPendingUpload",
         "azure.ai.projects.onedp.AIProjectClient.indexes.list_versions": "Azure.AI.Projects.ServicePatterns.Indexes.listVersions",
         "azure.ai.projects.onedp.AIProjectClient.indexes.list_latest": "Azure.AI.Projects.ServicePatterns.Indexes.listLatest",
         "azure.ai.projects.onedp.AIProjectClient.indexes.get_version": "Azure.AI.Projects.ServicePatterns.Indexes.getVersion",
         "azure.ai.projects.onedp.AIProjectClient.indexes.delete_version": "Azure.AI.Projects.ServicePatterns.Indexes.deleteVersion",
-        "azure.ai.projects.onedp.AIProjectClient.indexes.create": "Azure.AI.Projects.ServicePatterns.Indexes.create",
         "azure.ai.projects.onedp.AIProjectClient.indexes.create_version": "Azure.AI.Projects.ServicePatterns.Indexes.createVersion",
         "azure.ai.projects.onedp.AIProjectClient.deployments.get": "Azure.AI.Projects.Deployments.get",
         "azure.ai.projects.onedp.AIProjectClient.deployments.list": "Azure.AI.Projects.Deployments.list",
-        "azure.ai.projects.onedp.AIProjectClient.evaluation_results.list_versions": "Azure.AI.Projects.ServicePatterns.EvaluationResults.listVersions",
-        "azure.ai.projects.onedp.AIProjectClient.evaluation_results.list_latest": "Azure.AI.Projects.ServicePatterns.EvaluationResults.listLatest",
-        "azure.ai.projects.onedp.AIProjectClient.evaluation_results.get_version": "Azure.AI.Projects.ServicePatterns.EvaluationResults.getVersion",
-        "azure.ai.projects.onedp.AIProjectClient.evaluation_results.delete_version": "Azure.AI.Projects.ServicePatterns.EvaluationResults.deleteVersion",
-        "azure.ai.projects.onedp.AIProjectClient.evaluation_results.create": "Azure.AI.Projects.ServicePatterns.EvaluationResults.create",
-        "azure.ai.projects.onedp.AIProjectClient.evaluation_results.create_version": "Azure.AI.Projects.ServicePatterns.EvaluationResults.createVersion",
-        "azure.ai.projects.onedp.AIProjectClient.evaluation_results.start_pending_upload": "Azure.AI.Projects.EvaluationResults.startPendingUpload",
         "azure.ai.projects.onedp.AIProjectClient.red_teams.get": "Azure.AI.Projects.RedTeams.get",
         "azure.ai.projects.onedp.AIProjectClient.red_teams.list": "Azure.AI.Projects.RedTeams.list",
         "azure.ai.projects.onedp.AIProjectClient.red_teams.create_run": "Azure.AI.Projects.RedTeams.createRun"

sdk/ai/azure-ai-projects-onedp/azure/ai/projects/onedp/_client.py

@@ -21,7 +21,6 @@
     ConnectionsOperations,
     DatasetsOperations,
     DeploymentsOperations,
-    EvaluationResultsOperations,
     EvaluationsOperations,
     IndexesOperations,
     RedTeamsOperations,
@@ -47,8 +46,6 @@ class AIProjectClient:  # pylint: disable=too-many-instance-attributes
     :vartype indexes: azure.ai.projects.onedp.operations.IndexesOperations
     :ivar deployments: DeploymentsOperations operations
     :vartype deployments: azure.ai.projects.onedp.operations.DeploymentsOperations
-    :ivar evaluation_results: EvaluationResultsOperations operations
-    :vartype evaluation_results: azure.ai.projects.onedp.operations.EvaluationResultsOperations
     :ivar red_teams: RedTeamsOperations operations
     :vartype red_teams: azure.ai.projects.onedp.operations.RedTeamsOperations
     :param endpoint: Project endpoint in the form of:
@@ -97,9 +94,6 @@ def __init__(self, endpoint: str, credential: Union[AzureKeyCredential, "TokenCr
         self.datasets = DatasetsOperations(self._client, self._config, self._serialize, self._deserialize)
         self.indexes = IndexesOperations(self._client, self._config, self._serialize, self._deserialize)
         self.deployments = DeploymentsOperations(self._client, self._config, self._serialize, self._deserialize)
-        self.evaluation_results = EvaluationResultsOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
         self.red_teams = RedTeamsOperations(self._client, self._config, self._serialize, self._deserialize)
 
     def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse:

sdk/ai/azure-ai-projects-onedp/azure/ai/projects/onedp/_patch.py

@@ -9,7 +9,8 @@
 from typing import List, Any, Union, Optional
 from azure.core.credentials import AzureKeyCredential
 from ._client import AIProjectClient as AIProjectClientGenerated
-from .operations import TelemetryOperations, InferenceOperations
+from .operations import TelemetryOperations, InferenceOperations, AssistantsOperations
+
 
 class AIProjectClient(AIProjectClientGenerated):  # pylint: disable=too-many-instance-attributes
     """AIProjectClient.
@@ -48,11 +49,10 @@ def __init__(self, endpoint: str, credential: Union[AzureKeyCredential, "TokenCr
         super().__init__(endpoint=endpoint, credential=credential, **kwargs)
         self.telemetry = TelemetryOperations(self)
         self.inference = InferenceOperations(self)
+        self.assistants = AssistantsOperations(self)
 
 
-__all__: List[str] = [
-    "AIProjectClient"
-]  # Add all objects you want publicly available to users at this package level
+__all__: List[str] = ["AIProjectClient"]  # Add all objects you want publicly available to users at this package level
 
 
 def patch_sdk():

sdk/ai/azure-ai-projects-onedp/azure/ai/projects/onedp/aio/_client.py

@@ -21,7 +21,6 @@
     ConnectionsOperations,
     DatasetsOperations,
     DeploymentsOperations,
-    EvaluationResultsOperations,
     EvaluationsOperations,
     IndexesOperations,
     RedTeamsOperations,
@@ -47,8 +46,6 @@ class AIProjectClient:  # pylint: disable=too-many-instance-attributes
     :vartype indexes: azure.ai.projects.onedp.aio.operations.IndexesOperations
     :ivar deployments: DeploymentsOperations operations
     :vartype deployments: azure.ai.projects.onedp.aio.operations.DeploymentsOperations
-    :ivar evaluation_results: EvaluationResultsOperations operations
-    :vartype evaluation_results: azure.ai.projects.onedp.aio.operations.EvaluationResultsOperations
     :ivar red_teams: RedTeamsOperations operations
     :vartype red_teams: azure.ai.projects.onedp.aio.operations.RedTeamsOperations
     :param endpoint: Project endpoint in the form of:
@@ -99,9 +96,6 @@ def __init__(
         self.datasets = DatasetsOperations(self._client, self._config, self._serialize, self._deserialize)
         self.indexes = IndexesOperations(self._client, self._config, self._serialize, self._deserialize)
         self.deployments = DeploymentsOperations(self._client, self._config, self._serialize, self._deserialize)
-        self.evaluation_results = EvaluationResultsOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
         self.red_teams = RedTeamsOperations(self._client, self._config, self._serialize, self._deserialize)
 
     def send_request(

sdk/ai/azure-ai-projects-onedp/azure/ai/projects/onedp/aio/_patch.py

@@ -9,7 +9,8 @@
 from typing import List, Optional, Union, Any
 from azure.core.credentials import AzureKeyCredential
 from ._client import AIProjectClient as AIProjectClientGenerated
-from .operations import InferenceOperations
+from .operations import InferenceOperations, AssistantsOperations, TelemetryOperations
+
 
 class AIProjectClient(AIProjectClientGenerated):  # pylint: disable=too-many-instance-attributes
     """AIProjectClient.
@@ -48,12 +49,12 @@ def __init__(
     ) -> None:
         self._user_agent: Optional[str] = kwargs.get("user_agent", None)
         super().__init__(endpoint=endpoint, credential=credential, **kwargs)
+        self.telemetry = TelemetryOperations(self)
         self.inference = InferenceOperations(self)
+        self.assistants = AssistantsOperations(self)
 
 
-__all__: List[str] = [
-    "AIProjectClient"
-]  # Add all objects you want publicly available to users at this package level
+__all__: List[str] = ["AIProjectClient"]  # Add all objects you want publicly available to users at this package level
 
 
 def patch_sdk():

sdk/ai/azure-ai-projects-onedp/azure/ai/projects/onedp/aio/operations/__init__.py

@@ -18,7 +18,6 @@
 from ._operations import DatasetsOperations  # type: ignore
 from ._operations import IndexesOperations  # type: ignore
 from ._operations import DeploymentsOperations  # type: ignore
-from ._operations import EvaluationResultsOperations  # type: ignore
 from ._operations import RedTeamsOperations  # type: ignore
 
 from ._patch import __all__ as _patch_all
@@ -32,7 +31,6 @@
     "DatasetsOperations",
     "IndexesOperations",
     "DeploymentsOperations",
-    "EvaluationResultsOperations",
     "RedTeamsOperations",
 ]
 __all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore