azure-ai-evaluation 1.5.0__py3-none-any.whl → 1.6.0__py3-none-any.whl

azure/ai/evaluation/_common/onedp/models/_models.py

@@ -0,0 +1,2228 @@
+# pylint: disable=line-too-long,useless-suppression,too-many-lines
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) Python Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+# pylint: disable=useless-super-delegation
+
+import datetime
+from typing import Any, Dict, List, Literal, Mapping, Optional, TYPE_CHECKING, Union, overload
+
+from .. import _model_base
+from .._model_base import rest_discriminator, rest_field
+from ._enums import CredentialType, DatasetType, DeploymentType, IndexType, PendingUploadType
+
+if TYPE_CHECKING:
+    from .. import models as _models
+
+
+class AnnotationDTO(_model_base.Model):
+    """Represents the data transfer object for an annotation.
+
+    :ivar annotation_task: The task associated with the annotation. Required.
+    :vartype annotation_task: str
+    :ivar content_type: The type of content being annotated. Required.
+    :vartype content_type: str
+    :ivar user_text_list: A list of user-provided text inputs. Required.
+    :vartype user_text_list: list[str]
+    :ivar contents: A collection of content objects related to the annotation. Required.
+    :vartype contents: list[~azure.ai.projects.onedp.models.Content]
+    :ivar metric_list: A list of metrics associated with the annotation. Required.
+    :vartype metric_list: list[str]
+    :ivar prompt_version: The version of the prompt used for the annotation. Required.
+    :vartype prompt_version: str
+    :ivar user_agent: The user agent information. Required.
+    :vartype user_agent: str
+    :ivar partner_id: The partner identifier. Required.
+    :vartype partner_id: str
+    :ivar model_id: The model identifier. Required.
+    :vartype model_id: str
+    :ivar inference_type: The type of inference performed. Required.
+    :vartype inference_type: str
+    :ivar client_request_id: The client request identifier. Required.
+    :vartype client_request_id: str
+    """
+
+    annotation_task: str = rest_field(name="AnnotationTask", visibility=["read", "create", "update", "delete", "query"])
+    """The task associated with the annotation. Required."""
+    content_type: str = rest_field(name="ContentType", visibility=["read", "create", "update", "delete", "query"])
+    """The type of content being annotated. Required."""
+    user_text_list: List[str] = rest_field(
+        name="UserTextList", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """A list of user-provided text inputs. Required."""
+    contents: List["_models.Content"] = rest_field(
+        name="Contents", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """A collection of content objects related to the annotation. Required."""
+    metric_list: List[str] = rest_field(name="MetricList", visibility=["read", "create", "update", "delete", "query"])
+    """A list of metrics associated with the annotation. Required."""
+    prompt_version: str = rest_field(name="PromptVersion", visibility=["read", "create", "update", "delete", "query"])
+    """The version of the prompt used for the annotation. Required."""
+    user_agent: str = rest_field(name="UserAgent", visibility=["read", "create", "update", "delete", "query"])
+    """The user agent information. Required."""
+    partner_id: str = rest_field(name="PartnerId", visibility=["read", "create", "update", "delete", "query"])
+    """The partner identifier. Required."""
+    model_id: str = rest_field(name="ModelId", visibility=["read", "create", "update", "delete", "query"])
+    """The model identifier. Required."""
+    inference_type: str = rest_field(name="InferenceType", visibility=["read", "create", "update", "delete", "query"])
+    """The type of inference performed. Required."""
+    client_request_id: str = rest_field(
+        name="ClientRequestId", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """The client request identifier. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        annotation_task: str,
+        content_type: str,
+        user_text_list: List[str],
+        contents: List["_models.Content"],
+        metric_list: List[str],
+        prompt_version: str,
+        user_agent: str,
+        partner_id: str,
+        model_id: str,
+        inference_type: str,
+        client_request_id: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class TargetModelConfig(_model_base.Model):
+    """Abstract class for model configuration.
+
+    You probably want to use the sub-classes and not this class directly. Known sub-classes are:
+    AOAIModelConfig, MAASModelConfig
+
+    :ivar type: Type of the model configuration. Required. Default value is None.
+    :vartype type: str
+    """
+
+    __mapping__: Dict[str, _model_base.Model] = {}
+    type: str = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])
+    """Type of the model configuration. Required. Default value is None."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        type: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class AOAIModelConfig(TargetModelConfig, discriminator="AOAI"):
+    """Azure OpenAI model configuration. The API version would be selected by the service for querying
+    the model.
+
+    :ivar type: Required. Default value is "AOAI".
+    :vartype type: str
+    :ivar azure_endpoint: Endpoint targetURI for AOAI model. Required.
+    :vartype azure_endpoint: str
+    :ivar api_key: API Key for AOAI model. Required.
+    :vartype api_key: str
+    :ivar azure_deployment: Deployment name for AOAI model. Required.
+    :vartype azure_deployment: str
+    """
+
+    type: Literal["AOAI"] = rest_discriminator(name="type", visibility=["read"])  # type: ignore
+    """Required. Default value is \"AOAI\"."""
+    azure_endpoint: str = rest_field(name="azureEndpoint", visibility=["read", "create", "update", "delete", "query"])
+    """Endpoint targetURI for AOAI model. Required."""
+    api_key: str = rest_field(name="apiKey", visibility=["read", "create", "update", "delete", "query"])
+    """API Key for AOAI model. Required."""
+    azure_deployment: str = rest_field(
+        name="azureDeployment", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Deployment name for AOAI model. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        azure_endpoint: str,
+        api_key: str,
+        azure_deployment: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type="AOAI", **kwargs)
+
+
+class BaseCredentials(_model_base.Model):
+    """A base class for connection credentials.
+
+    You probably want to use the sub-classes and not this class directly. Known sub-classes are:
+    EntraIDCredentials, ApiKeyCredentials, CustomCredential, NoAuthenticationCredentials,
+    SASCredentials
+
+    :ivar auth_type: The type of credential used by the connection. Required. Known values are:
+     "ApiKey", "AAD", "SAS", "CustomKeys", and "None".
+    :vartype auth_type: str or ~azure.ai.projects.onedp.models.CredentialType
+    """
+
+    __mapping__: Dict[str, _model_base.Model] = {}
+    auth_type: str = rest_discriminator(name="authType", visibility=["read"])
+    """The type of credential used by the connection. Required. Known values are: \"ApiKey\", \"AAD\",
+     \"SAS\", \"CustomKeys\", and \"None\"."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        auth_type: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class ApiKeyCredentials(BaseCredentials, discriminator="ApiKey"):
+    """API Key Credential definition.
+
+    :ivar auth_type: The credentail type. Required. API Key credential
+    :vartype auth_type: str or ~azure.ai.projects.onedp.models.API_KEY
+    :ivar api_key: API Key.
+    :vartype api_key: str
+    """
+
+    auth_type: Literal[CredentialType.API_KEY] = rest_discriminator(name="authType", visibility=["read"])  # type: ignore
+    """The credentail type. Required. API Key credential"""
+    api_key: Optional[str] = rest_field(name="apiKey", visibility=["read"])
+    """API Key."""
+
+    @overload
+    def __init__(
+        self,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, auth_type=CredentialType.API_KEY, **kwargs)
+
+
+class AssetCredentialRequest(_model_base.Model):
+    """Asset Credential Request.
+
+    :ivar blob_uri: Blob URI. Required.
+    :vartype blob_uri: str
+    """
+
+    blob_uri: str = rest_field(name="BlobUri", visibility=["read", "create", "update", "delete", "query"])
+    """Blob URI. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        blob_uri: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class AssetCredentialResponse(_model_base.Model):
+    """Represents a reference to a blob for consumption.
+
+    :ivar blob_reference_for_consumption: Credential info to access the storage account. Required.
+    :vartype blob_reference_for_consumption:
+     ~azure.ai.projects.onedp.models.BlobReferenceForConsumption
+    """
+
+    blob_reference_for_consumption: "_models.BlobReferenceForConsumption" = rest_field(
+        name="blobReferenceForConsumption", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Credential info to access the storage account. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        blob_reference_for_consumption: "_models.BlobReferenceForConsumption",
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class AttackObjective(_model_base.Model):
+    """Attack Objective.
+
+    :ivar id: The unique identifier. Required.
+    :vartype id: str
+    :ivar metadata: The metadata.
+    :vartype metadata: ~azure.ai.projects.onedp.models.Metadata
+    :ivar source: List of sources. Required.
+    :vartype source: list[str]
+    :ivar modality: The modality. Required.
+    :vartype modality: str
+    :ivar messages: The messages. Required.
+    :vartype messages: list[~azure.ai.projects.onedp.models.Message]
+    """
+
+    id: str = rest_field(name="Id", visibility=["read", "create", "update", "delete", "query"])
+    """The unique identifier. Required."""
+    metadata: Optional["_models.Metadata"] = rest_field(
+        name="Metadata", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """The metadata."""
+    source: List[str] = rest_field(name="Source", visibility=["read", "create", "update", "delete", "query"])
+    """List of sources. Required."""
+    modality: str = rest_field(name="Modality", visibility=["read", "create", "update", "delete", "query"])
+    """The modality. Required."""
+    messages: List["_models.Message"] = rest_field(
+        name="Messages", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """The messages. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        id: str,  # pylint: disable=redefined-builtin
+        source: List[str],
+        modality: str,
+        messages: List["_models.Message"],
+        metadata: Optional["_models.Metadata"] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class Index(_model_base.Model):
+    """Index resource Definition.
+
+    You probably want to use the sub-classes and not this class directly. Known sub-classes are:
+    AzureAISearchIndex, CosmosDBIndex, ManagedAzureAISearchIndex
+
+    :ivar type: Type of index. Required. Known values are: "AzureSearch",
+     "CosmosDBNoSqlVectorStore", and "ManagedAzureSearch".
+    :vartype type: str or ~azure.ai.projects.onedp.models.IndexType
+    :ivar stage: Asset stage.
+    :vartype stage: str
+    :ivar id: A unique identifier for the asset, assetId probably?.
+    :vartype id: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar version: The version of the resource. Required.
+    :vartype version: str
+    :ivar description: The asset description text.
+    :vartype description: str
+    :ivar tags: Tag dictionary. Tags can be added, removed, and updated.
+    :vartype tags: dict[str, str]
+    """
+
+    __mapping__: Dict[str, _model_base.Model] = {}
+    type: str = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])
+    """Type of index. Required. Known values are: \"AzureSearch\", \"CosmosDBNoSqlVectorStore\", and
+     \"ManagedAzureSearch\"."""
+    stage: Optional[str] = rest_field(visibility=["read", "create", "update"])
+    """Asset stage."""
+    id: Optional[str] = rest_field(visibility=["read"])
+    """A unique identifier for the asset, assetId probably?."""
+    name: str = rest_field(visibility=["read"])
+    """The name of the resource. Required."""
+    version: str = rest_field(visibility=["read"])
+    """The version of the resource. Required."""
+    description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """The asset description text."""
+    tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Tag dictionary. Tags can be added, removed, and updated."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        type: str,
+        stage: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class AzureAISearchIndex(Index, discriminator="AzureSearch"):
+    """Azure AI Search Index Definition.
+
+    :ivar stage: Asset stage.
+    :vartype stage: str
+    :ivar id: A unique identifier for the asset, assetId probably?.
+    :vartype id: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar version: The version of the resource. Required.
+    :vartype version: str
+    :ivar description: The asset description text.
+    :vartype description: str
+    :ivar tags: Tag dictionary. Tags can be added, removed, and updated.
+    :vartype tags: dict[str, str]
+    :ivar type: Type of index. Required. Azure search
+    :vartype type: str or ~azure.ai.projects.onedp.models.AZURE_SEARCH
+    :ivar connection_name: Name of connection to Azure AI Search. Required.
+    :vartype connection_name: str
+    :ivar index_name: Name of index in Azure AI Search resource to attach. Required.
+    :vartype index_name: str
+    """
+
+    type: Literal[IndexType.AZURE_SEARCH] = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])  # type: ignore
+    """Type of index. Required. Azure search"""
+    connection_name: str = rest_field(name="connectionName", visibility=["read", "create", "update", "delete", "query"])
+    """Name of connection to Azure AI Search. Required."""
+    index_name: str = rest_field(name="indexName", visibility=["read", "create", "update", "delete", "query"])
+    """Name of index in Azure AI Search resource to attach. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        connection_name: str,
+        index_name: str,
+        stage: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type=IndexType.AZURE_SEARCH, **kwargs)
+
+
+class BlobReferenceForConsumption(_model_base.Model):
+    """Represents a reference to a blob for consumption.
+
+    :ivar blob_uri: Blob URI path for client to upload data. Example:
+     `https://blob.windows.core.net/Container/Path <https://blob.windows.core.net/Container/Path>`_.
+     Required.
+    :vartype blob_uri: str
+    :ivar storage_account_arm_id: ARM ID of the storage account to use. Required.
+    :vartype storage_account_arm_id: str
+    :ivar credential: Credential info to access the storage account. Required.
+    :vartype credential: ~azure.ai.projects.onedp.models.SasCredential
+    """
+
+    blob_uri: str = rest_field(name="blobUri", visibility=["read", "create", "update", "delete", "query"])
+    """Blob URI path for client to upload data. Example: `https://blob.windows.core.net/Container/Path
+     <https://blob.windows.core.net/Container/Path>`_. Required."""
+    storage_account_arm_id: str = rest_field(
+        name="storageAccountArmId", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """ARM ID of the storage account to use. Required."""
+    credential: "_models.SasCredential" = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Credential info to access the storage account. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        blob_uri: str,
+        storage_account_arm_id: str,
+        credential: "_models.SasCredential",
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class Connection(_model_base.Model):
+    """Response from the list and get connections operations.
+
+    :ivar auth_type: Discriminator property for Connection. Required.
+    :vartype auth_type: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar type: Category of the connection. Required. Known values are: "AzureOpenAI", "AzureBlob",
+     "AzureStorageAccount", "CognitiveSearch", "CosmosDB", "ApiKey", "AppConfig", "AppInsights", and
+     "CustomKeys".
+    :vartype type: str or ~azure.ai.projects.onedp.models.ConnectionType
+    :ivar target: The connection URL to be used for this service. Required.
+    :vartype target: str
+    :ivar is_default: Whether the connection is tagged as the default connection of its type.
+     Required.
+    :vartype is_default: bool
+    :ivar credentials: The credentials used by the connection. Required.
+    :vartype credentials: ~azure.ai.projects.onedp.models.BaseCredentials
+    :ivar metadata: Metadata of the connection. Required.
+    :vartype metadata: dict[str, str]
+    """
+
+    auth_type: str = rest_discriminator(name="authType")
+    """Discriminator property for Connection. Required."""
+    name: str = rest_field(visibility=["read"])
+    """The name of the resource. Required."""
+    type: Union[str, "_models.ConnectionType"] = rest_field(visibility=["read"])
+    """Category of the connection. Required. Known values are: \"AzureOpenAI\", \"AzureBlob\",
+     \"AzureStorageAccount\", \"CognitiveSearch\", \"CosmosDB\", \"ApiKey\", \"AppConfig\",
+     \"AppInsights\", and \"CustomKeys\"."""
+    target: str = rest_field(visibility=["read"])
+    """The connection URL to be used for this service. Required."""
+    is_default: bool = rest_field(name="isDefault", visibility=["read"])
+    """Whether the connection is tagged as the default connection of its type. Required."""
+    credentials: "_models.BaseCredentials" = rest_field(visibility=["read"])
+    """The credentials used by the connection. Required."""
+    metadata: Dict[str, str] = rest_field(visibility=["read"])
+    """Metadata of the connection. Required."""
+
+    @overload
+    def __init__(
+        self,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class Content(_model_base.Model):
+    """Message content.
+
+    :ivar messages: The type of content. Required.
+    :vartype messages: list[any]
+    """
+
+    messages: List[Any] = rest_field(name="Messages", visibility=["read", "create", "update", "delete", "query"])
+    """The type of content. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        messages: List[Any],
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class CosmosDBIndex(Index, discriminator="CosmosDBNoSqlVectorStore"):
+    """CosmosDB Vector Store Index Definition.
+
+    :ivar stage: Asset stage.
+    :vartype stage: str
+    :ivar id: A unique identifier for the asset, assetId probably?.
+    :vartype id: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar version: The version of the resource. Required.
+    :vartype version: str
+    :ivar description: The asset description text.
+    :vartype description: str
+    :ivar tags: Tag dictionary. Tags can be added, removed, and updated.
+    :vartype tags: dict[str, str]
+    :ivar type: Type of index. Required. CosmosDB
+    :vartype type: str or ~azure.ai.projects.onedp.models.COSMOS_DB
+    :ivar connection_name: Name of connection to CosmosDB. Required.
+    :vartype connection_name: str
+    :ivar database_name: Name of the CosmosDB Database. Required.
+    :vartype database_name: str
+    :ivar container_name: Name of CosmosDB Container. Required.
+    :vartype container_name: str
+    :ivar embedding_configuration: Embedding model configuration. Required.
+    :vartype embedding_configuration: ~azure.ai.projects.onedp.models.EmbeddingConfiguration
+    """
+
+    type: Literal[IndexType.COSMOS_DB] = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])  # type: ignore
+    """Type of index. Required. CosmosDB"""
+    connection_name: str = rest_field(name="connectionName", visibility=["read", "create", "update", "delete", "query"])
+    """Name of connection to CosmosDB. Required."""
+    database_name: str = rest_field(name="databaseName", visibility=["read", "create", "update", "delete", "query"])
+    """Name of the CosmosDB Database. Required."""
+    container_name: str = rest_field(name="containerName", visibility=["read", "create", "update", "delete", "query"])
+    """Name of CosmosDB Container. Required."""
+    embedding_configuration: "_models.EmbeddingConfiguration" = rest_field(
+        name="embeddingConfiguration", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Embedding model configuration. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        connection_name: str,
+        database_name: str,
+        container_name: str,
+        embedding_configuration: "_models.EmbeddingConfiguration",
+        stage: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type=IndexType.COSMOS_DB, **kwargs)
+
+
+class CustomCredential(BaseCredentials, discriminator="CustomKeys"):
+    """Custom credential defintion.
+
+    :ivar auth_type: The credential type. Required. Custom credential
+    :vartype auth_type: str or ~azure.ai.projects.onedp.models.CUSTOM
+    """
+
+    auth_type: Literal[CredentialType.CUSTOM] = rest_discriminator(name="authType", visibility=["read"])  # type: ignore
+    """The credential type. Required. Custom credential"""
+
+    @overload
+    def __init__(
+        self,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, auth_type=CredentialType.CUSTOM, **kwargs)
+
+
+class CustomizationParameters(_model_base.Model):
+    """Customization Parameters.
+
+    :ivar application_scenario: Application scenario.
+    :vartype application_scenario: str
+    :ivar harm_categories: List of harm categories. Required.
+    :vartype harm_categories: list[str]
+    """
+
+    application_scenario: Optional[str] = rest_field(
+        name="ApplicationScenario", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Application scenario."""
+    harm_categories: List[str] = rest_field(
+        name="HarmCategories", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """List of harm categories. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        harm_categories: List[str],
+        application_scenario: Optional[str] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class DatasetVersion(_model_base.Model):
+    """DatasetVersion Definition.
+
+    You probably want to use the sub-classes and not this class directly. Known sub-classes are:
+    FileDatasetVersion, FolderDatasetVersion
+
+    :ivar dataset_uri: [Required] Uri of the data. Example:
+     `https://go.microsoft.com/fwlink/?linkid=2202330
+     <https://go.microsoft.com/fwlink/?linkid=2202330>`_. Required.
+    :vartype dataset_uri: str
+    :ivar type: Dataset type. Required. Known values are: "uri_file" and "uri_folder".
+    :vartype type: str or ~azure.ai.projects.onedp.models.DatasetType
+    :ivar is_reference: Indicates if dataset is reference only or managed by dataset service. If
+     true, the underlying data will be deleted when the dataset version is deleted.
+    :vartype is_reference: bool
+    :ivar stage: Asset stage.
+    :vartype stage: str
+    :ivar id: A unique identifier for the asset, assetId probably?.
+    :vartype id: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar version: The version of the resource. Required.
+    :vartype version: str
+    :ivar description: The asset description text.
+    :vartype description: str
+    :ivar tags: Tag dictionary. Tags can be added, removed, and updated.
+    :vartype tags: dict[str, str]
+    """
+
+    __mapping__: Dict[str, _model_base.Model] = {}
+    dataset_uri: str = rest_field(name="datasetUri", visibility=["read", "create"])
+    """[Required] Uri of the data. Example: `https://go.microsoft.com/fwlink/?linkid=2202330
+     <https://go.microsoft.com/fwlink/?linkid=2202330>`_. Required."""
+    type: str = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])
+    """Dataset type. Required. Known values are: \"uri_file\" and \"uri_folder\"."""
+    is_reference: Optional[bool] = rest_field(name="isReference", visibility=["read"])
+    """Indicates if dataset is reference only or managed by dataset service. If true, the underlying
+     data will be deleted when the dataset version is deleted."""
+    stage: Optional[str] = rest_field(visibility=["read", "create", "update"])
+    """Asset stage."""
+    id: Optional[str] = rest_field(visibility=["read"])
+    """A unique identifier for the asset, assetId probably?."""
+    name: str = rest_field(visibility=["read"])
+    """The name of the resource. Required."""
+    version: str = rest_field(visibility=["read"])
+    """The version of the resource. Required."""
+    description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """The asset description text."""
+    tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Tag dictionary. Tags can be added, removed, and updated."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        dataset_uri: str,
+        type: str,
+        stage: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class Deployment(_model_base.Model):
+    """Model Deployment Definition.
+
+    You probably want to use the sub-classes and not this class directly. Known sub-classes are:
+    ModelDeployment
+
+    :ivar type: The type of the deployment. Required. "ModelDeployment"
+    :vartype type: str or ~azure.ai.projects.onedp.models.DeploymentType
+    :ivar name: Name of the deployment. Required.
+    :vartype name: str
+    """
+
+    __mapping__: Dict[str, _model_base.Model] = {}
+    type: str = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])
+    """The type of the deployment. Required. \"ModelDeployment\""""
+    name: str = rest_field(visibility=["read"])
+    """Name of the deployment. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        type: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class EmbeddingConfiguration(_model_base.Model):
+    """Embedding configuration class.
+
+    :ivar model_deployment_name: Deployment name of embedding model. It can point to a model
+     deployment either in the parent AIServices or a connection. Required.
+    :vartype model_deployment_name: str
+    :ivar embedding_field: Embedding field. Required.
+    :vartype embedding_field: str
+    """
+
+    model_deployment_name: str = rest_field(
+        name="modelDeploymentName", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Deployment name of embedding model. It can point to a model deployment either in the parent
+     AIServices or a connection. Required."""
+    embedding_field: str = rest_field(name="embeddingField", visibility=["read", "create", "update", "delete", "query"])
+    """Embedding field. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        model_deployment_name: str,
+        embedding_field: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class EntraIDCredentials(BaseCredentials, discriminator="AAD"):
+    """Entra ID credential definition.
+
+    :ivar auth_type: The credential type. Required. Entra ID credential (formerly known as AAD)
+    :vartype auth_type: str or ~azure.ai.projects.onedp.models.ENTRA_ID
+    """
+
+    auth_type: Literal[CredentialType.ENTRA_ID] = rest_discriminator(name="authType", visibility=["read"])  # type: ignore
+    """The credential type. Required. Entra ID credential (formerly known as AAD)"""
+
+    @overload
+    def __init__(
+        self,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, auth_type=CredentialType.ENTRA_ID, **kwargs)
+
+
+class Evaluation(_model_base.Model):
+    """Evaluation Definition.
+
+    :ivar id: Identifier of the evaluation. Required.
+    :vartype id: str
+    :ivar data: Data for evaluation. Required.
+    :vartype data: ~azure.ai.projects.onedp.models.InputData
+    :ivar target: Evaluation target specifying the model config and parameters.
+    :vartype target: ~azure.ai.projects.onedp.models.EvaluationTarget
+    :ivar display_name: Display Name for evaluation. It helps to find the evaluation easily in AI
+     Foundry. It does not need to be unique.
+    :vartype display_name: str
+    :ivar description: Description of the evaluation. It can be used to store additional
+     information about the evaluation and is mutable.
+    :vartype description: str
+    :ivar system_data: Metadata containing createdBy and modifiedBy information.
+    :vartype system_data: ~azure.ai.projects.onedp.models.SystemData
+    :ivar status: Status of the evaluation. It is set by service and is read-only.
+    :vartype status: str
+    :ivar tags: Evaluation's tags. Unlike properties, tags are fully mutable.
+    :vartype tags: dict[str, str]
+    :ivar properties: Evaluation's properties. Unlike tags, properties are add-only. Once added, a
+     property cannot be removed.
+    :vartype properties: dict[str, str]
+    :ivar evaluators: Evaluators to be used for the evaluation. Required.
+    :vartype evaluators: dict[str, ~azure.ai.projects.onedp.models.EvaluatorConfiguration]
+    :ivar outputs: Read-only result outputs. Evaluation Results will have logs and results under
+     the eval_results folder. Example: { 'evaluationResultId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/evaluationresults/{name}/versions/{version}',
+     'evaluationMetrics': '{serialized-json-metrics}' }. Required.
+    :vartype outputs: dict[str, str]
+    """
+
+    id: str = rest_field(visibility=["read"])
+    """Identifier of the evaluation. Required."""
+    data: "_models.InputData" = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Data for evaluation. Required."""
+    target: Optional["_models.EvaluationTarget"] = rest_field(visibility=["read", "create"])
+    """Evaluation target specifying the model config and parameters."""
+    display_name: Optional[str] = rest_field(
+        name="displayName", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Display Name for evaluation. It helps to find the evaluation easily in AI Foundry. It does not
+     need to be unique."""
+    description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Description of the evaluation. It can be used to store additional information about the
+     evaluation and is mutable."""
+    system_data: Optional["_models.SystemData"] = rest_field(name="systemData", visibility=["read"])
+    """Metadata containing createdBy and modifiedBy information."""
+    status: Optional[str] = rest_field(visibility=["read"])
+    """Status of the evaluation. It is set by service and is read-only."""
+    tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Evaluation's tags. Unlike properties, tags are fully mutable."""
+    properties: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Evaluation's properties. Unlike tags, properties are add-only. Once added, a property cannot be
+     removed."""
+    evaluators: Dict[str, "_models.EvaluatorConfiguration"] = rest_field(
+        visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Evaluators to be used for the evaluation. Required."""
+    outputs: Dict[str, str] = rest_field(visibility=["read"])
+    """Read-only result outputs. Evaluation Results will have logs and results under the eval_results
+     folder. Example: { 'evaluationResultId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/evaluationresults/{name}/versions/{version}',
+     'evaluationMetrics': '{serialized-json-metrics}' }. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        data: "_models.InputData",
+        evaluators: Dict[str, "_models.EvaluatorConfiguration"],
+        target: Optional["_models.EvaluationTarget"] = None,
+        display_name: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+        properties: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class EvaluationResult(_model_base.Model):
+    """Evaluation Result resource Definition.
+
+    :ivar result_type: Type of Evaluation result. Known values are: "Benchmark", "Evaluation",
+     "Redteam", and "Simulation".
+    :vartype result_type: str or ~azure.ai.projects.onedp.models.ResultType
+    :ivar metrics: Aggregated metrics.
+    :vartype metrics: dict[str, float]
+    :ivar blob_uri: Blob URI.
+    :vartype blob_uri: str
+    :ivar stage: Asset stage.
+    :vartype stage: str
+    :ivar id: A unique identifier for the asset, assetId probably?.
+    :vartype id: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar version: The version of the resource. Required.
+    :vartype version: str
+    :ivar description: The asset description text.
+    :vartype description: str
+    :ivar tags: Tag dictionary. Tags can be added, removed, and updated.
+    :vartype tags: dict[str, str]
+    """
+
+    result_type: Optional[Union[str, "_models.ResultType"]] = rest_field(
+        name="ResultType", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Type of Evaluation result. Known values are: \"Benchmark\", \"Evaluation\", \"Redteam\", and
+     \"Simulation\"."""
+    metrics: Optional[Dict[str, float]] = rest_field(
+        name="Metrics", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Aggregated metrics."""
+    blob_uri: Optional[str] = rest_field(name="BlobUri", visibility=["read", "create", "update", "delete", "query"])
+    """Blob URI."""
+    stage: Optional[str] = rest_field(visibility=["read", "create", "update"])
+    """Asset stage."""
+    id: Optional[str] = rest_field(visibility=["read"])
+    """A unique identifier for the asset, assetId probably?."""
+    name: str = rest_field(visibility=["read"])
+    """The name of the resource. Required."""
+    version: str = rest_field(visibility=["read"])
+    """The version of the resource. Required."""
+    description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """The asset description text."""
+    tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Tag dictionary. Tags can be added, removed, and updated."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        result_type: Optional[Union[str, "_models.ResultType"]] = None,
+        metrics: Optional[Dict[str, float]] = None,
+        blob_uri: Optional[str] = None,
+        stage: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class EvaluationTarget(_model_base.Model):
+    """Target for the evaluation process.
+
+    :ivar system_message: System message related to the evaluation target. Required.
+    :vartype system_message: str
+    :ivar model_config: Model configuration for the evaluation. Required.
+    :vartype model_config: ~azure.ai.projects.onedp.models.TargetModelConfig
+    :ivar model_params: A dictionary of parameters for the model.
+    :vartype model_params: dict[str, any]
+    """
+
+    system_message: str = rest_field(name="systemMessage", visibility=["read", "create", "update", "delete", "query"])
+    """System message related to the evaluation target. Required."""
+    model_config: "_models.TargetModelConfig" = rest_field(
+        name="modelConfig", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Model configuration for the evaluation. Required."""
+    model_params: Optional[Dict[str, Any]] = rest_field(
+        name="modelParams", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """A dictionary of parameters for the model."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        system_message: str,
+        model_config: "_models.TargetModelConfig",
+        model_params: Optional[Dict[str, Any]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class EvaluationUpload(_model_base.Model):
+    """Upload a local SDK evaluation run. Currently update supports status, outputs, properties, and
+    tags updates.
+
+    :ivar id: Identifier of the evaluation. Required.
+    :vartype id: str
+    :ivar data: Data for evaluation.
+    :vartype data: ~azure.ai.projects.onedp.models.InputData
+    :ivar target: Evaluation target specifying the model config and parameters.
+    :vartype target: ~azure.ai.projects.onedp.models.EvaluationTarget
+    :ivar display_name: Display Name for evaluation. It helps to find the evaluation easily in AI
+     Foundry. It does not need to be unique.
+    :vartype display_name: str
+    :ivar description: Description of the evaluation. It can be used to store additional
+     information about the evaluation and is mutable.
+    :vartype description: str
+    :ivar system_data: Metadata containing createdBy and modifiedBy information.
+    :vartype system_data: ~azure.ai.projects.onedp.models.SystemData
+    :ivar status: Status of the evaluation. For upload: Failed or Completed.
+    :vartype status: str
+    :ivar tags: Evaluation's tags. Unlike properties, tags are fully mutable.
+    :vartype tags: dict[str, str]
+    :ivar properties: Evaluation's properties. Unlike tags, properties are add-only. Once added, a
+     property cannot be removed.
+    :vartype properties: dict[str, str]
+    :ivar evaluators: Evaluators to be used for the evaluation.
+    :vartype evaluators: dict[str, ~azure.ai.projects.onedp.models.EvaluatorConfiguration]
+    :ivar outputs: Outputs of the evaluation as a dictionary of IDs. Example: {
+     'evaluationResultId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/evaluationresults/{name}/versions/{version}'}.
+    :vartype outputs: dict[str, str]
+    """
+
+    id: str = rest_field(visibility=["read"])
+    """Identifier of the evaluation. Required."""
+    data: Optional["_models.InputData"] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Data for evaluation."""
+    target: Optional["_models.EvaluationTarget"] = rest_field(visibility=["read", "create"])
+    """Evaluation target specifying the model config and parameters."""
+    display_name: Optional[str] = rest_field(
+        name="displayName", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Display Name for evaluation. It helps to find the evaluation easily in AI Foundry. It does not
+     need to be unique."""
+    description: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Description of the evaluation. It can be used to store additional information about the
+     evaluation and is mutable."""
+    system_data: Optional["_models.SystemData"] = rest_field(name="systemData", visibility=["read"])
+    """Metadata containing createdBy and modifiedBy information."""
+    status: Optional[str] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Status of the evaluation. For upload: Failed or Completed."""
+    tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Evaluation's tags. Unlike properties, tags are fully mutable."""
+    properties: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Evaluation's properties. Unlike tags, properties are add-only. Once added, a property cannot be
+     removed."""
+    evaluators: Optional[Dict[str, "_models.EvaluatorConfiguration"]] = rest_field(
+        visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Evaluators to be used for the evaluation."""
+    outputs: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Outputs of the evaluation as a dictionary of IDs. Example: { 'evaluationResultId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/evaluationresults/{name}/versions/{version}'}."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        data: Optional["_models.InputData"] = None,
+        target: Optional["_models.EvaluationTarget"] = None,
+        display_name: Optional[str] = None,
+        description: Optional[str] = None,
+        status: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+        properties: Optional[Dict[str, str]] = None,
+        evaluators: Optional[Dict[str, "_models.EvaluatorConfiguration"]] = None,
+        outputs: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class EvaluatorConfiguration(_model_base.Model):
+    """Evaluator Configuration.
+
+    :ivar id: Identifier of the evaluator. Required.
+    :vartype id: str
+    :ivar init_params: Initialization parameters of the evaluator.
+    :vartype init_params: dict[str, any]
+    :ivar data_mapping: Data parameters of the evaluator.
+    :vartype data_mapping: dict[str, str]
+    """
+
+    id: str = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Identifier of the evaluator. Required."""
+    init_params: Optional[Dict[str, Any]] = rest_field(
+        name="initParams", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Initialization parameters of the evaluator."""
+    data_mapping: Optional[Dict[str, str]] = rest_field(
+        name="dataMapping", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Data parameters of the evaluator."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        id: str,  # pylint: disable=redefined-builtin
+        init_params: Optional[Dict[str, Any]] = None,
+        data_mapping: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class FileDatasetVersion(DatasetVersion, discriminator="uri_file"):
+    """FileDatasetVersion Definition.
+
+    :ivar dataset_uri: [Required] Uri of the data. Example:
+     `https://go.microsoft.com/fwlink/?linkid=2202330
+     <https://go.microsoft.com/fwlink/?linkid=2202330>`_. Required.
+    :vartype dataset_uri: str
+    :ivar is_reference: Indicates if dataset is reference only or managed by dataset service. If
+     true, the underlying data will be deleted when the dataset version is deleted.
+    :vartype is_reference: bool
+    :ivar stage: Asset stage.
+    :vartype stage: str
+    :ivar id: A unique identifier for the asset, assetId probably?.
+    :vartype id: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar version: The version of the resource. Required.
+    :vartype version: str
+    :ivar description: The asset description text.
+    :vartype description: str
+    :ivar tags: Tag dictionary. Tags can be added, removed, and updated.
+    :vartype tags: dict[str, str]
+    :ivar type: Dataset type. Required. URI file.
+    :vartype type: str or ~azure.ai.projects.onedp.models.URI_FILE
+    :ivar open_ai_purpose: Indicates OpenAI Purpose. FileDatasets created with this field will be
+     compatible with OpenAI-specific features. Required.
+    :vartype open_ai_purpose: str
+    """
+
+    type: Literal[DatasetType.URI_FILE] = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])  # type: ignore
+    """Dataset type. Required. URI file."""
+    open_ai_purpose: str = rest_field(name="openAIPurpose", visibility=["read", "create", "update", "delete", "query"])
+    """Indicates OpenAI Purpose. FileDatasets created with this field will be compatible with
+     OpenAI-specific features. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        dataset_uri: str,
+        open_ai_purpose: str,
+        stage: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type=DatasetType.URI_FILE, **kwargs)
+
+
+class FolderDatasetVersion(DatasetVersion, discriminator="uri_folder"):
+    """FileDatasetVersion Definition.
+
+    :ivar dataset_uri: [Required] Uri of the data. Example:
+     `https://go.microsoft.com/fwlink/?linkid=2202330
+     <https://go.microsoft.com/fwlink/?linkid=2202330>`_. Required.
+    :vartype dataset_uri: str
+    :ivar is_reference: Indicates if dataset is reference only or managed by dataset service. If
+     true, the underlying data will be deleted when the dataset version is deleted.
+    :vartype is_reference: bool
+    :ivar stage: Asset stage.
+    :vartype stage: str
+    :ivar id: A unique identifier for the asset, assetId probably?.
+    :vartype id: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar version: The version of the resource. Required.
+    :vartype version: str
+    :ivar description: The asset description text.
+    :vartype description: str
+    :ivar tags: Tag dictionary. Tags can be added, removed, and updated.
+    :vartype tags: dict[str, str]
+    :ivar type: Dataset type. Required. URI folder.
+    :vartype type: str or ~azure.ai.projects.onedp.models.URI_FOLDER
+    """
+
+    type: Literal[DatasetType.URI_FOLDER] = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])  # type: ignore
+    """Dataset type. Required. URI folder."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        dataset_uri: str,
+        stage: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type=DatasetType.URI_FOLDER, **kwargs)
+
+
+class InputData(_model_base.Model):
+    """Abstract data class.
+
+    You probably want to use the sub-classes and not this class directly. Known sub-classes are:
+    InputDataset
+
+    :ivar type: Type of the data. Required. Default value is None.
+    :vartype type: str
+    """
+
+    __mapping__: Dict[str, _model_base.Model] = {}
+    type: str = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])
+    """Type of the data. Required. Default value is None."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        type: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class InputDataset(InputData, discriminator="dataset"):
+    """Dataset as source for evaluation.
+
+    :ivar type: Required. Default value is "dataset".
+    :vartype type: str
+    :ivar id: Evaluation input data. Required.
+    :vartype id: str
+    """
+
+    type: Literal["dataset"] = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])  # type: ignore
+    """Required. Default value is \"dataset\"."""
+    id: str = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Evaluation input data. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        id: str,  # pylint: disable=redefined-builtin
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type="dataset", **kwargs)
+
+
+class LongRunningResponse(_model_base.Model):
+    """Long Running Response.
+
+    :ivar location: The location. Required.
+    :vartype location: str
+    :ivar operation_result: The OperationResult. Required.
+    :vartype operation_result: any
+    """
+
+    location: str = rest_field(name="Location", visibility=["read", "create", "update", "delete", "query"])
+    """The location. Required."""
+    operation_result: Any = rest_field(
+        name="OperationResult", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """The OperationResult. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        location: str,
+        operation_result: Any,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class MAASModelConfig(TargetModelConfig, discriminator="MAAS"):
+    """MaaS model configuration. The API version would be selected by the service for querying the
+    model.
+
+    :ivar type: Required. Default value is "MAAS".
+    :vartype type: str
+    :ivar azure_endpoint: Endpoint targetURI for MAAS model. Required.
+    :vartype azure_endpoint: str
+    :ivar api_key: API Key for MAAS model. Required.
+    :vartype api_key: str
+    """
+
+    type: Literal["MAAS"] = rest_discriminator(name="type", visibility=["read"])  # type: ignore
+    """Required. Default value is \"MAAS\"."""
+    azure_endpoint: str = rest_field(name="azureEndpoint", visibility=["read", "create", "update", "delete", "query"])
+    """Endpoint targetURI for MAAS model. Required."""
+    api_key: str = rest_field(name="apiKey", visibility=["read", "create", "update", "delete", "query"])
+    """API Key for MAAS model. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        azure_endpoint: str,
+        api_key: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type="MAAS", **kwargs)
+
+
+class ManagedAzureAISearchIndex(Index, discriminator="ManagedAzureSearch"):
+    """Managed Azure AI Search Index Definition.
+
+    :ivar stage: Asset stage.
+    :vartype stage: str
+    :ivar id: A unique identifier for the asset, assetId probably?.
+    :vartype id: str
+    :ivar name: The name of the resource. Required.
+    :vartype name: str
+    :ivar version: The version of the resource. Required.
+    :vartype version: str
+    :ivar description: The asset description text.
+    :vartype description: str
+    :ivar tags: Tag dictionary. Tags can be added, removed, and updated.
+    :vartype tags: dict[str, str]
+    :ivar type: Type of index. Required. Managed Azure Search
+    :vartype type: str or ~azure.ai.projects.onedp.models.MANAGED_AZURE_SEARCH
+    :ivar vector_store_id: Vector store id of managed index. Required.
+    :vartype vector_store_id: str
+    """
+
+    type: Literal[IndexType.MANAGED_AZURE_SEARCH] = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])  # type: ignore
+    """Type of index. Required. Managed Azure Search"""
+    vector_store_id: str = rest_field(name="vectorStoreId", visibility=["read", "create", "update", "delete", "query"])
+    """Vector store id of managed index. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        vector_store_id: str,
+        stage: Optional[str] = None,
+        description: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type=IndexType.MANAGED_AZURE_SEARCH, **kwargs)
+
+
+class Message(_model_base.Model):
+    """Message.
+
+    :ivar role: The role.
+    :vartype role: str
+    :ivar content: The content.
+    :vartype content: str
+    """
+
+    role: Optional[str] = rest_field(name="Role", visibility=["read", "create", "update", "delete", "query"])
+    """The role."""
+    content: Optional[str] = rest_field(name="Content", visibility=["read", "create", "update", "delete", "query"])
+    """The content."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        role: Optional[str] = None,
+        content: Optional[str] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class Metadata(_model_base.Model):
+    """Metadata.
+
+    :ivar target_harms: List of target harms. Required.
+    :vartype target_harms: list[~azure.ai.projects.onedp.models.TargetHarm]
+    :ivar language: The language. Required.
+    :vartype language: str
+    """
+
+    target_harms: List["_models.TargetHarm"] = rest_field(
+        name="TargetHarms", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """List of target harms. Required."""
+    language: str = rest_field(name="Language", visibility=["read", "create", "update", "delete", "query"])
+    """The language. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        target_harms: List["_models.TargetHarm"],
+        language: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class ModelDeployment(Deployment, discriminator="ModelDeployment"):
+    """Model Deployment Definition.
+
+    :ivar name: Name of the deployment. Required.
+    :vartype name: str
+    :ivar type: The type of the deployment. Required. Model deployment
+    :vartype type: str or ~azure.ai.projects.onedp.models.MODEL_DEPLOYMENT
+    :ivar model_name: Publisher-specific name of the deployed model. Required.
+    :vartype model_name: str
+    :ivar model_version: Publisher-specific version of the deployed model. Required.
+    :vartype model_version: str
+    :ivar model_publisher: Name of the deployed model's publisher. Required.
+    :vartype model_publisher: str
+    :ivar capabilities: Capabilities of deployed model. Required.
+    :vartype capabilities: dict[str, str]
+    :ivar sku: Sku of the model deployment. Required.
+    :vartype sku: ~azure.ai.projects.onedp.models.Sku
+    :ivar connection_name: Name of the connection the deployment comes from.
+    :vartype connection_name: str
+    """
+
+    type: Literal[DeploymentType.MODEL_DEPLOYMENT] = rest_discriminator(name="type", visibility=["read", "create", "update", "delete", "query"])  # type: ignore
+    """The type of the deployment. Required. Model deployment"""
+    model_name: str = rest_field(name="modelName", visibility=["read"])
+    """Publisher-specific name of the deployed model. Required."""
+    model_version: str = rest_field(name="modelVersion", visibility=["read"])
+    """Publisher-specific version of the deployed model. Required."""
+    model_publisher: str = rest_field(name="modelPublisher", visibility=["read"])
+    """Name of the deployed model's publisher. Required."""
+    capabilities: Dict[str, str] = rest_field(visibility=["read"])
+    """Capabilities of deployed model. Required."""
+    sku: "_models.Sku" = rest_field(visibility=["read"])
+    """Sku of the model deployment. Required."""
+    connection_name: Optional[str] = rest_field(name="connectionName", visibility=["read"])
+    """Name of the connection the deployment comes from."""
+
+    @overload
+    def __init__(
+        self,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, type=DeploymentType.MODEL_DEPLOYMENT, **kwargs)
+
+
+class NoAuthenticationCredentials(BaseCredentials, discriminator="None"):
+    """Credentials that do not require authentication.
+
+    :ivar auth_type: The credential type. Required. No credential
+    :vartype auth_type: str or ~azure.ai.projects.onedp.models.NONE
+    """
+
+    auth_type: Literal[CredentialType.NONE] = rest_discriminator(name="authType", visibility=["read"])  # type: ignore
+    """The credential type. Required. No credential"""
+
+    @overload
+    def __init__(
+        self,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, auth_type=CredentialType.NONE, **kwargs)
+
+
+class PendingUploadRequest(_model_base.Model):
+    """Represents a request for a pending upload.
+
+    :ivar pending_upload_id: If PendingUploadId is not provided, a random GUID will be used.
+    :vartype pending_upload_id: str
+    :ivar connection_name: Name of Azure blob storage connection to use for generating temporary
+     SAS token.
+    :vartype connection_name: str
+    :ivar pending_upload_type: TemporaryBlobReference is the only supported type. Required.
+     Temporary Blob Reference is the only supported type.
+    :vartype pending_upload_type: str or ~azure.ai.projects.onedp.models.TEMPORARY_BLOB_REFERENCE
+    """
+
+    pending_upload_id: Optional[str] = rest_field(
+        name="pendingUploadId", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """If PendingUploadId is not provided, a random GUID will be used."""
+    connection_name: Optional[str] = rest_field(
+        name="connectionName", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Name of Azure blob storage connection to use for generating temporary SAS token."""
+    pending_upload_type: Literal[PendingUploadType.TEMPORARY_BLOB_REFERENCE] = rest_field(
+        name="pendingUploadType", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """TemporaryBlobReference is the only supported type. Required. Temporary Blob Reference is the
+     only supported type."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        pending_upload_type: Literal[PendingUploadType.TEMPORARY_BLOB_REFERENCE],
+        pending_upload_id: Optional[str] = None,
+        connection_name: Optional[str] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class PendingUploadResponse(_model_base.Model):
+    """Represents the response for a pending upload request.
+
+    :ivar blob_reference_for_consumption: Container-level read, write, list SAS. Required.
+    :vartype blob_reference_for_consumption:
+     ~azure.ai.projects.onedp.models.BlobReferenceForConsumption
+    :ivar pending_upload_id: ID for this upload request. Required.
+    :vartype pending_upload_id: str
+    :ivar dataset_version: Version of dataset to be created if user did not specify version when
+     initially creating upload.
+    :vartype dataset_version: str
+    :ivar pending_upload_type: TemporaryBlobReference is the only supported type. Required.
+     Temporary Blob Reference is the only supported type.
+    :vartype pending_upload_type: str or ~azure.ai.projects.onedp.models.TEMPORARY_BLOB_REFERENCE
+    """
+
+    blob_reference_for_consumption: "_models.BlobReferenceForConsumption" = rest_field(
+        name="blobReferenceForConsumption", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Container-level read, write, list SAS. Required."""
+    pending_upload_id: str = rest_field(
+        name="pendingUploadId", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """ID for this upload request. Required."""
+    dataset_version: Optional[str] = rest_field(
+        name="datasetVersion", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Version of dataset to be created if user did not specify version when initially creating
+     upload."""
+    pending_upload_type: Literal[PendingUploadType.TEMPORARY_BLOB_REFERENCE] = rest_field(
+        name="pendingUploadType", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """TemporaryBlobReference is the only supported type. Required. Temporary Blob Reference is the
+     only supported type."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        blob_reference_for_consumption: "_models.BlobReferenceForConsumption",
+        pending_upload_id: str,
+        pending_upload_type: Literal[PendingUploadType.TEMPORARY_BLOB_REFERENCE],
+        dataset_version: Optional[str] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class RedTeam(_model_base.Model):
+    """Red team details.
+
+    :ivar id: Identifier of the red team. Required.
+    :vartype id: str
+    :ivar scan_name: Name of the red-team scan. Required.
+    :vartype scan_name: str
+    :ivar num_turns: Number of simulation rounds. Required.
+    :vartype num_turns: int
+    :ivar attack_strategy: List of attack strategies or nested lists of attack strategies.
+     Required.
+    :vartype attack_strategy: list[str or ~azure.ai.projects.onedp.models.AttackStrategy]
+    :ivar simulation_only: Simulation-only or Simulation + Evaluation. Default false, if true the
+     scan outputs conversation not evaluation result. Required.
+    :vartype simulation_only: bool
+    :ivar risk_categories: List of risk categories to generate attack objectives for. Required.
+    :vartype risk_categories: list[str or ~azure.ai.projects.onedp.models.RiskCategory]
+    :ivar application_scenario: Application scenario for the red team operation, to generate
+     scenario specific attacks.
+    :vartype application_scenario: str
+    :ivar tags: Red team's tags. Unlike properties, tags are fully mutable.
+    :vartype tags: dict[str, str]
+    :ivar properties: Red team's properties. Unlike tags, properties are add-only. Once added, a
+     property cannot be removed.
+    :vartype properties: dict[str, str]
+    :ivar status: Status of the red-team. It is set by service and is read-only.
+    :vartype status: str
+    :ivar outputs: Read-only result outputs. Example: { 'evaluationResultId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/evaluationresults/{name}/versions/{version}',
+     'logId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/datasets/{dataset-name}/versions/{dataset-version}'
+     }. Required.
+    :vartype outputs: dict[str, str]
+    :ivar system_data: Metadata containing createdBy and modifiedBy information.
+    :vartype system_data: ~azure.ai.projects.onedp.models.SystemData
+    """
+
+    id: str = rest_field(visibility=["read"])
+    """Identifier of the red team. Required."""
+    scan_name: str = rest_field(name="scanName", visibility=["read", "create", "update", "delete", "query"])
+    """Name of the red-team scan. Required."""
+    num_turns: int = rest_field(name="numTurns", visibility=["read", "create", "update", "delete", "query"])
+    """Number of simulation rounds. Required."""
+    attack_strategy: List[Union[str, "_models.AttackStrategy"]] = rest_field(
+        name="attackStrategy", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """List of attack strategies or nested lists of attack strategies. Required."""
+    simulation_only: bool = rest_field(
+        name="simulationOnly", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Simulation-only or Simulation + Evaluation. Default false, if true the scan outputs
+     conversation not evaluation result. Required."""
+    risk_categories: List[Union[str, "_models.RiskCategory"]] = rest_field(
+        name="riskCategories", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """List of risk categories to generate attack objectives for. Required."""
+    application_scenario: Optional[str] = rest_field(
+        name="applicationScenario", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Application scenario for the red team operation, to generate scenario specific attacks."""
+    tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Red team's tags. Unlike properties, tags are fully mutable."""
+    properties: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Red team's properties. Unlike tags, properties are add-only. Once added, a property cannot be
+     removed."""
+    status: Optional[str] = rest_field(visibility=["read"])
+    """Status of the red-team. It is set by service and is read-only."""
+    outputs: Dict[str, str] = rest_field(visibility=["read"])
+    """Read-only result outputs. Example: { 'evaluationResultId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/evaluationresults/{name}/versions/{version}',
+     'logId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/datasets/{dataset-name}/versions/{dataset-version}'
+     }. Required."""
+    system_data: Optional["_models.SystemData"] = rest_field(name="systemData", visibility=["read"])
+    """Metadata containing createdBy and modifiedBy information."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        scan_name: str,
+        num_turns: int,
+        attack_strategy: List[Union[str, "_models.AttackStrategy"]],
+        simulation_only: bool,
+        risk_categories: List[Union[str, "_models.RiskCategory"]],
+        application_scenario: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+        properties: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class RedTeamUpload(_model_base.Model):
+    """Red team details.
+
+    :ivar id: Identifier of the red team. Required.
+    :vartype id: str
+    :ivar scan_name: Name of the red-team scan.
+    :vartype scan_name: str
+    :ivar num_turns: Number of simulation rounds.
+    :vartype num_turns: int
+    :ivar attack_strategy: List of attack strategies or nested lists of attack strategies.
+    :vartype attack_strategy: list[str or ~azure.ai.projects.onedp.models.AttackStrategy]
+    :ivar simulation_only: Simulation-only or Simulation + Evaluation. Default false, if true the
+     scan outputs conversation not evaluation result.
+    :vartype simulation_only: bool
+    :ivar risk_categories: List of risk categories to generate attack objectives for.
+    :vartype risk_categories: list[str or ~azure.ai.projects.onedp.models.RiskCategory]
+    :ivar application_scenario: Application scenario for the red team operation, to generate
+     scenario specific attacks.
+    :vartype application_scenario: str
+    :ivar tags: Red team's tags. Unlike properties, tags are fully mutable.
+    :vartype tags: dict[str, str]
+    :ivar properties: Red team's properties. Unlike tags, properties are add-only. Once added, a
+     property cannot be removed.
+    :vartype properties: dict[str, str]
+    :ivar status: Status of the red-team. It is set by service and is read-only.
+    :vartype status: str
+    :ivar outputs: Read-only result outputs. Example: { 'evaluationResultId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/evaluationresults/{name}/versions/{version}'
+     }.
+    :vartype outputs: dict[str, str]
+    :ivar system_data: Metadata containing createdBy and modifiedBy information.
+    :vartype system_data: ~azure.ai.projects.onedp.models.SystemData
+    """
+
+    id: str = rest_field(visibility=["read"])
+    """Identifier of the red team. Required."""
+    scan_name: Optional[str] = rest_field(name="scanName", visibility=["read", "create", "update", "delete", "query"])
+    """Name of the red-team scan."""
+    num_turns: Optional[int] = rest_field(name="numTurns", visibility=["read", "create", "update", "delete", "query"])
+    """Number of simulation rounds."""
+    attack_strategy: Optional[List[Union[str, "_models.AttackStrategy"]]] = rest_field(
+        name="attackStrategy", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """List of attack strategies or nested lists of attack strategies."""
+    simulation_only: Optional[bool] = rest_field(
+        name="simulationOnly", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Simulation-only or Simulation + Evaluation. Default false, if true the scan outputs
+     conversation not evaluation result."""
+    risk_categories: Optional[List[Union[str, "_models.RiskCategory"]]] = rest_field(
+        name="riskCategories", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """List of risk categories to generate attack objectives for."""
+    application_scenario: Optional[str] = rest_field(
+        name="applicationScenario", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Application scenario for the red team operation, to generate scenario specific attacks."""
+    tags: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Red team's tags. Unlike properties, tags are fully mutable."""
+    properties: Optional[Dict[str, str]] = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Red team's properties. Unlike tags, properties are add-only. Once added, a property cannot be
+     removed."""
+    status: Optional[str] = rest_field(visibility=["read"])
+    """Status of the red-team. It is set by service and is read-only."""
+    outputs: Optional[Dict[str, str]] = rest_field(visibility=["read"])
+    """Read-only result outputs. Example: { 'evaluationResultId':
+     'azureai://accounts/{AccountName}/projects/{myproject}/evaluationresults/{name}/versions/{version}'
+     }."""
+    system_data: Optional["_models.SystemData"] = rest_field(name="systemData", visibility=["read"])
+    """Metadata containing createdBy and modifiedBy information."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        scan_name: Optional[str] = None,
+        num_turns: Optional[int] = None,
+        attack_strategy: Optional[List[Union[str, "_models.AttackStrategy"]]] = None,
+        simulation_only: Optional[bool] = None,
+        risk_categories: Optional[List[Union[str, "_models.RiskCategory"]]] = None,
+        application_scenario: Optional[str] = None,
+        tags: Optional[Dict[str, str]] = None,
+        properties: Optional[Dict[str, str]] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class SasCredential(_model_base.Model):
+    """SAS Credential definition.
+
+    :ivar sas_uri: SAS uri. Required.
+    :vartype sas_uri: str
+    :ivar type: Type of credential. Required. Default value is "SAS".
+    :vartype type: str
+    """
+
+    sas_uri: str = rest_field(name="sasUri", visibility=["read"])
+    """SAS uri. Required."""
+    type: Literal["SAS"] = rest_field(visibility=["read"])
+    """Type of credential. Required. Default value is \"SAS\"."""
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+        self.type: Literal["SAS"] = "SAS"
+
+
+class SASCredentials(BaseCredentials, discriminator="SAS"):
+    """Shared Access Signature (SAS) credential definition.
+
+    :ivar auth_type: The credential type. Required. Shared Access Signature (SAS) credential
+    :vartype auth_type: str or ~azure.ai.projects.onedp.models.SAS
+    :ivar sas_token: SAS token.
+    :vartype sas_token: str
+    """
+
+    auth_type: Literal[CredentialType.SAS] = rest_discriminator(name="authType", visibility=["read"])  # type: ignore
+    """The credential type. Required. Shared Access Signature (SAS) credential"""
+    sas_token: Optional[str] = rest_field(name="sasToken", visibility=["read"])
+    """SAS token."""
+
+    @overload
+    def __init__(
+        self,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, auth_type=CredentialType.SAS, **kwargs)
+
+
+class SimulationDTO(_model_base.Model):
+    """Customization Parameters.
+
+    :ivar headers: Additional headers.
+    :vartype headers: dict[str, str]
+    :ivar params: Parameters.
+    :vartype params: dict[str, str]
+    :ivar template_parameters: Template parameters.
+    :vartype template_parameters: dict[str, str]
+    :ivar customization_parameters: Customization parameters.
+    :vartype customization_parameters: ~azure.ai.projects.onedp.models.CustomizationParameters
+    :ivar json: Json.
+    :vartype json: str
+    :ivar url: Url.
+    :vartype url: str
+    :ivar template_key: Template key.
+    :vartype template_key: str
+    :ivar simulation_type: Type of Simulation. Known values are: "Default", "CustomPersona", and
+     "HarmTurnGenerator".
+    :vartype simulation_type: str or ~azure.ai.projects.onedp.models.SimulationType
+    :ivar is_microsoft_tenant: 'True' if Microsoft internal tenant and 'False' otherwise.
+    :vartype is_microsoft_tenant: bool
+    :ivar subscription_id: Azure subscription id.
+    :vartype subscription_id: str
+    :ivar resource_group_name: Resource group name.
+    :vartype resource_group_name: str
+    :ivar workspace_name: Workspace name.
+    :vartype workspace_name: str
+    """
+
+    headers: Optional[Dict[str, str]] = rest_field(
+        name="Headers", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Additional headers."""
+    params: Optional[Dict[str, str]] = rest_field(
+        name="Params", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Parameters."""
+    template_parameters: Optional[Dict[str, str]] = rest_field(
+        name="TemplateParameters", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Template parameters."""
+    customization_parameters: Optional["_models.CustomizationParameters"] = rest_field(
+        name="CustomizationParameters", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Customization parameters."""
+    json: Optional[str] = rest_field(name="Json", visibility=["read", "create", "update", "delete", "query"])
+    """Json."""
+    url: Optional[str] = rest_field(name="Url", visibility=["read", "create", "update", "delete", "query"])
+    """Url."""
+    template_key: Optional[str] = rest_field(
+        name="TemplateKey", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Template key."""
+    simulation_type: Optional[Union[str, "_models.SimulationType"]] = rest_field(
+        name="SimulationType", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Type of Simulation. Known values are: \"Default\", \"CustomPersona\", and
+     \"HarmTurnGenerator\"."""
+    is_microsoft_tenant: Optional[bool] = rest_field(
+        name="IsMicrosoftTenant", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """'True' if Microsoft internal tenant and 'False' otherwise."""
+    subscription_id: Optional[str] = rest_field(
+        name="SubscriptionId", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Azure subscription id."""
+    resource_group_name: Optional[str] = rest_field(
+        name="ResourceGroupName", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Resource group name."""
+    workspace_name: Optional[str] = rest_field(
+        name="WorkspaceName", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """Workspace name."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        headers: Optional[Dict[str, str]] = None,
+        params: Optional[Dict[str, str]] = None,
+        template_parameters: Optional[Dict[str, str]] = None,
+        customization_parameters: Optional["_models.CustomizationParameters"] = None,
+        json: Optional[str] = None,
+        url: Optional[str] = None,
+        template_key: Optional[str] = None,
+        simulation_type: Optional[Union[str, "_models.SimulationType"]] = None,
+        is_microsoft_tenant: Optional[bool] = None,
+        subscription_id: Optional[str] = None,
+        resource_group_name: Optional[str] = None,
+        workspace_name: Optional[str] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class Sku(_model_base.Model):
+    """Sku information.
+
+    :ivar capacity: Sku capacity. Required.
+    :vartype capacity: int
+    :ivar family: Sku family. Required.
+    :vartype family: str
+    :ivar name: Sku name. Required.
+    :vartype name: str
+    :ivar size: Sku size. Required.
+    :vartype size: str
+    :ivar tier: Sku tier. Required.
+    :vartype tier: str
+    """
+
+    capacity: int = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Sku capacity. Required."""
+    family: str = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Sku family. Required."""
+    name: str = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Sku name. Required."""
+    size: str = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Sku size. Required."""
+    tier: str = rest_field(visibility=["read", "create", "update", "delete", "query"])
+    """Sku tier. Required."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        capacity: int,
+        family: str,
+        name: str,
+        size: str,
+        tier: str,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)
+
+
+class SystemData(_model_base.Model):
+    """Metadata pertaining to creation and last modification of the resource.
+
+    :ivar created_at: The timestamp the resource was created at.
+    :vartype created_at: ~datetime.datetime
+    :ivar created_by: The identity that created the resource.
+    :vartype created_by: str
+    :ivar created_by_type: The identity type that created the resource.
+    :vartype created_by_type: str
+    :ivar last_modified_at: The timestamp of resource last modification (UTC).
+    :vartype last_modified_at: ~datetime.datetime
+    """
+
+    created_at: Optional[datetime.datetime] = rest_field(name="createdAt", visibility=["read"], format="rfc3339")
+    """The timestamp the resource was created at."""
+    created_by: Optional[str] = rest_field(name="createdBy", visibility=["read"])
+    """The identity that created the resource."""
+    created_by_type: Optional[str] = rest_field(name="createdByType", visibility=["read"])
+    """The identity type that created the resource."""
+    last_modified_at: Optional[datetime.datetime] = rest_field(
+        name="lastModifiedAt", visibility=["read"], format="rfc3339"
+    )
+    """The timestamp of resource last modification (UTC)."""
+
+
+class TargetHarm(_model_base.Model):
+    """Target Harm.
+
+    :ivar risk_type: The risk type.
+    :vartype risk_type: str
+    :ivar risk_sub_type: The risk sub type.
+    :vartype risk_sub_type: str
+    """
+
+    risk_type: Optional[str] = rest_field(name="RiskType", visibility=["read", "create", "update", "delete", "query"])
+    """The risk type."""
+    risk_sub_type: Optional[str] = rest_field(
+        name="RiskSubType", visibility=["read", "create", "update", "delete", "query"]
+    )
+    """The risk sub type."""
+
+    @overload
+    def __init__(
+        self,
+        *,
+        risk_type: Optional[str] = None,
+        risk_sub_type: Optional[str] = None,
+    ) -> None: ...
+
+    @overload
+    def __init__(self, mapping: Mapping[str, Any]) -> None:
+        """
+        :param mapping: raw JSON to initialize the model.
+        :type mapping: Mapping[str, Any]
+        """
+
+    def __init__(self, *args: Any, **kwargs: Any) -> None:
+        super().__init__(*args, **kwargs)