|
4 | 4 |
|
5 | 5 | # pylint: disable=protected-access |
6 | 6 |
|
7 | | -from typing import Any, List, Optional, Type, Union, Dict |
8 | 7 | import re |
| 8 | +from typing import Any, Dict, List, Optional, Type, Union |
9 | 9 |
|
10 | 10 | from azure.ai.ml._restclient.v2024_04_01_preview.models import ConnectionCategory |
11 | | -from azure.ai.ml._utils.utils import camel_to_snake |
12 | | -from azure.ai.ml._utils._experimental import experimental |
13 | | -from azure.ai.ml.constants._common import ( |
14 | | - CONNECTION_API_VERSION_KEY, |
15 | | - CONNECTION_API_TYPE_KEY, |
16 | | - CONNECTION_KIND_KEY, |
17 | | - CONNECTION_ACCOUNT_NAME_KEY, |
18 | | - CONNECTION_CONTAINER_NAME_KEY, |
19 | | - CONNECTION_RESOURCE_ID_KEY, |
20 | | - ConnectionTypes, |
21 | | - CognitiveServiceKinds, |
22 | | -) |
23 | | -from azure.ai.ml.entities._credentials import ( |
24 | | - ApiKeyConfiguration, |
25 | | - AadCredentialConfiguration, |
26 | | -) |
27 | | - |
28 | 11 | from azure.ai.ml._schema.workspace.connections.connection_subtypes import ( |
29 | | - AzureBlobStoreConnectionSchema, |
30 | | - MicrosoftOneLakeConnectionSchema, |
31 | | - AzureOpenAIConnectionSchema, |
32 | | - AzureAIServicesConnectionSchema, |
| 12 | + APIKeyConnectionSchema, |
33 | 13 | AzureAISearchConnectionSchema, |
| 14 | + AzureAIServicesConnectionSchema, |
| 15 | + AzureBlobStoreConnectionSchema, |
34 | 16 | AzureContentSafetyConnectionSchema, |
| 17 | + AzureOpenAIConnectionSchema, |
35 | 18 | AzureSpeechServicesConnectionSchema, |
36 | | - APIKeyConnectionSchema, |
| 19 | + MicrosoftOneLakeConnectionSchema, |
37 | 20 | OpenAIConnectionSchema, |
38 | 21 | SerpConnectionSchema, |
39 | 22 | ServerlessConnectionSchema, |
40 | 23 | ) |
| 24 | +from azure.ai.ml._utils._experimental import experimental |
| 25 | +from azure.ai.ml._utils.utils import camel_to_snake |
| 26 | +from azure.ai.ml.constants._common import ( |
| 27 | + CONNECTION_ACCOUNT_NAME_KEY, |
| 28 | + CONNECTION_API_TYPE_KEY, |
| 29 | + CONNECTION_API_VERSION_KEY, |
| 30 | + CONNECTION_CONTAINER_NAME_KEY, |
| 31 | + CONNECTION_KIND_KEY, |
| 32 | + CONNECTION_RESOURCE_ID_KEY, |
| 33 | + CognitiveServiceKinds, |
| 34 | + ConnectionTypes, |
| 35 | +) |
| 36 | +from azure.ai.ml.entities._credentials import AadCredentialConfiguration, ApiKeyConfiguration |
| 37 | + |
41 | 38 | from .one_lake_artifacts import OneLakeConnectionArtifact |
42 | 39 | from .workspace_connection import WorkspaceConnection |
43 | 40 |
|
44 | | - |
45 | 41 | # Dev notes: Any new classes require modifying the elif chains in the following functions in the |
46 | 42 | # WorkspaceConnection parent class: _from_rest_object, _get_entity_class_from_type, _get_schema_class_from_type |
47 | 43 |
|
48 | 44 |
|
49 | | -@experimental |
50 | 45 | class AzureBlobStoreConnection(WorkspaceConnection): |
51 | 46 | """A connection to an Azure Blob Store. |
52 | 47 |
|
@@ -153,7 +148,6 @@ def account_name(self, value: str) -> None: |
153 | 148 | # difficult for customers to find out what the target for their system ought to be. |
154 | 149 | # Due to this, we construct the target internally by composing more inputs |
155 | 150 | # that are more user-accessible. |
156 | | -@experimental |
157 | 151 | class MicrosoftOneLakeConnection(WorkspaceConnection): |
158 | 152 | """A connection to a Microsoft One Lake. Connections of this type |
159 | 153 | are further specified by their artifact class type, although |
@@ -488,7 +482,6 @@ def ai_services_resource_id(self, value: str) -> None: |
488 | 482 | self.metadata[CONNECTION_RESOURCE_ID_KEY] = value |
489 | 483 |
|
490 | 484 |
|
491 | | -@experimental |
492 | 485 | class AzureAISearchConnection(ApiOrAadConnection): |
493 | 486 | """A Connection that is specifically designed for handling connections to |
494 | 487 | Azure AI Search. |
@@ -527,7 +520,6 @@ def _get_schema_class(cls) -> Type: |
527 | 520 | return AzureAISearchConnectionSchema |
528 | 521 |
|
529 | 522 |
|
530 | | -@experimental |
531 | 523 | class AzureContentSafetyConnection(ApiOrAadConnection): |
532 | 524 | """A Connection geared towards a Azure Content Safety service. |
533 | 525 |
|
@@ -570,7 +562,6 @@ def _get_schema_class(cls) -> Type: |
570 | 562 | return AzureContentSafetyConnectionSchema |
571 | 563 |
|
572 | 564 |
|
573 | | -@experimental |
574 | 565 | class AzureSpeechServicesConnection(ApiOrAadConnection): |
575 | 566 | """A Connection geared towards an Azure Speech service. |
576 | 567 |
|
|
0 commit comments