azure-ai-textanalytics 5.3.0b1__py3-none-any.whl → 6.0.0b1__py3-none-any.whl

azure/ai/textanalytics/__init__.py

@@ -1,203 +1,32 @@
-# ------------------------------------
-# Copyright (c) Microsoft Corporation.
-# Licensed under the MIT License.
-# ------------------------------------
+# 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=wrong-import-position
 
-from ._text_analytics_client import TextAnalyticsClient
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ._patch import *  # pylint: disable=unused-wildcard-import
+
+from ._client import TextAnalysisClient  # type: ignore
 from ._version import VERSION
-from ._base_client import TextAnalyticsApiVersion
-from ._models import (
-    DetectLanguageInput,
-    TextDocumentInput,
-    DetectedLanguage,
-    DocumentError,
-    CategorizedEntity,
-    LinkedEntity,
-    AnalyzeSentimentResult,
-    RecognizeEntitiesResult,
-    DetectLanguageResult,
-    TextAnalyticsError,
-    TextAnalyticsWarning,
-    ExtractKeyPhrasesResult,
-    RecognizeLinkedEntitiesResult,
-    TextDocumentStatistics,
-    LinkedEntityMatch,
-    TextDocumentBatchStatistics,
-    SentenceSentiment,
-    SentimentConfidenceScores,
-    MinedOpinion,
-    TargetSentiment,
-    AssessmentSentiment,
-    RecognizePiiEntitiesResult,
-    PiiEntity,
-    PiiEntityDomain,
-    AnalyzeHealthcareEntitiesResult,
-    HealthcareEntity,
-    HealthcareEntityDataSource,
-    RecognizeEntitiesAction,
-    RecognizeLinkedEntitiesAction,
-    RecognizePiiEntitiesAction,
-    ExtractKeyPhrasesAction,
-    _AnalyzeActionsType,
-    HealthcareRelation,
-    HealthcareRelationRole,
-    HealthcareEntityAssertion,
-    AnalyzeSentimentAction,
-    PiiEntityCategory,
-    HealthcareEntityRelation,
-    EntityConditionality,
-    EntityCertainty,
-    EntityAssociation,
-    HealthcareEntityCategory,
-    RecognizeCustomEntitiesAction,
-    RecognizeCustomEntitiesResult,
-    SingleLabelClassifyAction,
-    MultiLabelClassifyAction,
-    ClassifyDocumentResult,
-    ClassificationCategory,
-    AnalyzeHealthcareEntitiesAction,
-    TextAnalysisKind,
-    ExtractSummaryAction,
-    ExtractSummaryResult,
-    SummarySentence,
-    AbstractSummaryResult,
-    AbstractiveSummary,
-    SummaryContext,
-    AbstractSummaryAction,
-    DynamicClassificationResult,
-)
-from ._generated.models import (
-    HealthcareDocumentType,
-    BaseResolution,
-    ResolutionKind,
-    AgeResolution,
-    AreaResolution,
-    BooleanResolution,
-    CurrencyResolution,
-    DateTimeResolution,
-    InformationResolution,
-    LengthResolution,
-    NumberResolution,
-    NumericRangeResolution,
-    OrdinalResolution,
-    SpeedResolution,
-    TemperatureResolution,
-    TemporalSpanResolution,
-    VolumeResolution,
-    WeightResolution,
-    AgeUnit,
-    AreaUnit,
-    TemporalModifier,
-    InformationUnit,
-    LengthUnit,
-    NumberKind,
-    RangeKind,
-    RelativeTo,
-    SpeedUnit,
-    TemperatureUnit,
-    VolumeUnit,
-    WeightUnit,
-    ClassificationType,
-    DateTimeSubKind,
-)
-from ._lro import AnalyzeHealthcareEntitiesLROPoller, AnalyzeActionsLROPoller, TextAnalysisLROPoller
+
+__version__ = VERSION
+
+try:
+    from ._patch import __all__ as _patch_all
+    from ._patch import *
+except ImportError:
+    _patch_all = []
+from ._patch import patch_sdk as _patch_sdk
 
 __all__ = [
-    "TextAnalyticsApiVersion",
-    "TextAnalyticsClient",
-    "DetectLanguageInput",
-    "TextDocumentInput",
-    "DetectedLanguage",
-    "RecognizeEntitiesResult",
-    "DetectLanguageResult",
-    "CategorizedEntity",
-    "TextAnalyticsError",
-    "TextAnalyticsWarning",
-    "ExtractKeyPhrasesResult",
-    "RecognizeLinkedEntitiesResult",
-    "AnalyzeSentimentResult",
-    "TextDocumentStatistics",
-    "DocumentError",
-    "LinkedEntity",
-    "LinkedEntityMatch",
-    "TextDocumentBatchStatistics",
-    "SentenceSentiment",
-    "SentimentConfidenceScores",
-    "MinedOpinion",
-    "TargetSentiment",
-    "AssessmentSentiment",
-    "RecognizePiiEntitiesResult",
-    "PiiEntity",
-    "PiiEntityDomain",
-    "AnalyzeHealthcareEntitiesResult",
-    "HealthcareEntity",
-    "HealthcareEntityDataSource",
-    "RecognizeEntitiesAction",
-    "RecognizeLinkedEntitiesAction",
-    "RecognizePiiEntitiesAction",
-    "ExtractKeyPhrasesAction",
-    "_AnalyzeActionsType",
-    "PiiEntityCategory",
-    "HealthcareEntityRelation",
-    "HealthcareRelation",
-    "HealthcareRelationRole",
-    "HealthcareEntityAssertion",
-    "EntityConditionality",
-    "EntityCertainty",
-    "EntityAssociation",
-    "AnalyzeSentimentAction",
-    "AnalyzeHealthcareEntitiesLROPoller",
-    "AnalyzeActionsLROPoller",
-    "HealthcareEntityCategory",
-    "RecognizeCustomEntitiesAction",
-    "RecognizeCustomEntitiesResult",
-    "SingleLabelClassifyAction",
-    "MultiLabelClassifyAction",
-    "ClassifyDocumentResult",
-    "ClassificationCategory",
-    "AnalyzeHealthcareEntitiesAction",
-    "TextAnalysisLROPoller",
-    "TextAnalysisKind",
-    "ExtractSummaryAction",
-    "ExtractSummaryResult",
-    "SummarySentence",
-    "HealthcareDocumentType",
-    "BaseResolution",
-    "ResolutionKind",
-    "AgeResolution",
-    "AreaResolution",
-    "BooleanResolution",
-    "CurrencyResolution",
-    "DateTimeResolution",
-    "InformationResolution",
-    "LengthResolution",
-    "NumberResolution",
-    "NumericRangeResolution",
-    "OrdinalResolution",
-    "SpeedResolution",
-    "TemperatureResolution",
-    "TemporalSpanResolution",
-    "VolumeResolution",
-    "WeightResolution",
-    "AgeUnit",
-    "AreaUnit",
-    "TemporalModifier",
-    "InformationUnit",
-    "LengthUnit",
-    "NumberKind",
-    "RangeKind",
-    "RelativeTo",
-    "SpeedUnit",
-    "TemperatureUnit",
-    "VolumeUnit",
-    "WeightUnit",
-    "AbstractSummaryResult",
-    "AbstractiveSummary",
-    "SummaryContext",
-    "AbstractSummaryAction",
-    "ClassificationType",
-    "DynamicClassificationResult",
-    "DateTimeSubKind",
+    "TextAnalysisClient",
 ]
+__all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore
 
-__version__ = VERSION
+_patch_sdk()

azure/ai/textanalytics/_client.py

@@ -0,0 +1,111 @@
+# pylint: disable=line-too-long,useless-suppression
+# 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.
+# --------------------------------------------------------------------------
+
+from copy import deepcopy
+from typing import Any, TYPE_CHECKING, Union
+from typing_extensions import Self
+
+from azure.core import PipelineClient
+from azure.core.credentials import AzureKeyCredential
+from azure.core.pipeline import policies
+from azure.core.rest import HttpRequest, HttpResponse
+
+from ._configuration import TextAnalysisClientConfiguration
+from ._operations import _TextAnalysisClientOperationsMixin
+from ._utils.serialization import Deserializer, Serializer
+
+if TYPE_CHECKING:
+    from azure.core.credentials import TokenCredential
+
+
+class TextAnalysisClient(_TextAnalysisClientOperationsMixin):
+    """The language service API is a suite of natural language processing (NLP) skills built with
+    best-in-class Microsoft machine learning algorithms.  The API can be used to analyze
+    unstructured text for tasks such as sentiment analysis, key phrase extraction, language
+    detection and question answering. Further documentation can be found in <a
+    href=\\"`https://learn.microsoft.com/azure/cognitive-services/language-service/overview\\">https://learn.microsoft.com/azure/cognitive-services/language-service/overview
+    <https://learn.microsoft.com/azure/cognitive-services/language-service/overview\\">https://learn.microsoft.com/azure/cognitive-services/language-service/overview>`_</a>.0.
+
+    :param endpoint: Supported Cognitive Services endpoint (e.g.,
+     https://<resource-name>.api.cognitiveservices.azure.com). Required.
+    :type endpoint: str
+    :param credential: Credential used to authenticate requests to the service. Is either a key
+     credential type or a token credential type. Required.
+    :type credential: ~azure.core.credentials.AzureKeyCredential or
+     ~azure.core.credentials.TokenCredential
+    :keyword api_version: The API version to use for this operation. Default value is
+     "2025-05-15-preview". Note that overriding this default value may result in unsupported
+     behavior.
+    :paramtype api_version: str
+    :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
+     Retry-After header is present.
+    """
+
+    def __init__(self, endpoint: str, credential: Union[AzureKeyCredential, "TokenCredential"], **kwargs: Any) -> None:
+        _endpoint = "{Endpoint}/language"
+        self._config = TextAnalysisClientConfiguration(endpoint=endpoint, credential=credential, **kwargs)
+
+        _policies = kwargs.pop("policies", None)
+        if _policies is None:
+            _policies = [
+                policies.RequestIdPolicy(**kwargs),
+                self._config.headers_policy,
+                self._config.user_agent_policy,
+                self._config.proxy_policy,
+                policies.ContentDecodePolicy(**kwargs),
+                self._config.redirect_policy,
+                self._config.retry_policy,
+                self._config.authentication_policy,
+                self._config.custom_hook_policy,
+                self._config.logging_policy,
+                policies.DistributedTracingPolicy(**kwargs),
+                policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,
+                self._config.http_logging_policy,
+            ]
+        self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs)
+
+        self._serialize = Serializer()
+        self._deserialize = Deserializer()
+        self._serialize.client_side_validation = False
+
+    def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse:
+        """Runs the network request through the client's chained policies.
+
+        >>> from azure.core.rest import HttpRequest
+        >>> request = HttpRequest("GET", "https://www.example.org/")
+        <HttpRequest [GET], url: 'https://www.example.org/'>
+        >>> response = client.send_request(request)
+        <HttpResponse: 200 OK>
+
+        For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request
+
+        :param request: The network request you want to make. Required.
+        :type request: ~azure.core.rest.HttpRequest
+        :keyword bool stream: Whether the response payload will be streamed. Defaults to False.
+        :return: The response of your network call. Does not do error handling on your response.
+        :rtype: ~azure.core.rest.HttpResponse
+        """
+
+        request_copy = deepcopy(request)
+        path_format_arguments = {
+            "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True),
+        }
+
+        request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments)
+        return self._client.send_request(request_copy, stream=stream, **kwargs)  # type: ignore
+
+    def close(self) -> None:
+        self._client.close()
+
+    def __enter__(self) -> Self:
+        self._client.__enter__()
+        return self
+
+    def __exit__(self, *exc_details: Any) -> None:
+        self._client.__exit__(*exc_details)

azure/ai/textanalytics/_configuration.py

@@ -0,0 +1,73 @@
+# 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.
+# --------------------------------------------------------------------------
+
+from typing import Any, TYPE_CHECKING, Union
+
+from azure.core.credentials import AzureKeyCredential
+from azure.core.pipeline import policies
+
+from ._version import VERSION
+
+if TYPE_CHECKING:
+    from azure.core.credentials import TokenCredential
+
+
+class TextAnalysisClientConfiguration:  # pylint: disable=too-many-instance-attributes
+    """Configuration for TextAnalysisClient.
+
+    Note that all parameters used to create this instance are saved as instance
+    attributes.
+
+    :param endpoint: Supported Cognitive Services endpoint (e.g.,
+     https://<resource-name>.api.cognitiveservices.azure.com). Required.
+    :type endpoint: str
+    :param credential: Credential used to authenticate requests to the service. Is either a key
+     credential type or a token credential type. Required.
+    :type credential: ~azure.core.credentials.AzureKeyCredential or
+     ~azure.core.credentials.TokenCredential
+    :keyword api_version: The API version to use for this operation. Default value is
+     "2025-05-15-preview". Note that overriding this default value may result in unsupported
+     behavior.
+    :paramtype api_version: str
+    """
+
+    def __init__(self, endpoint: str, credential: Union[AzureKeyCredential, "TokenCredential"], **kwargs: Any) -> None:
+        api_version: str = kwargs.pop("api_version", "2025-05-15-preview")
+
+        if endpoint is None:
+            raise ValueError("Parameter 'endpoint' must not be None.")
+        if credential is None:
+            raise ValueError("Parameter 'credential' must not be None.")
+
+        self.endpoint = endpoint
+        self.credential = credential
+        self.api_version = api_version
+        self.credential_scopes = kwargs.pop("credential_scopes", ["https://cognitiveservices.azure.com/.default"])
+        kwargs.setdefault("sdk_moniker", "ai-textanalytics/{}".format(VERSION))
+        self.polling_interval = kwargs.get("polling_interval", 30)
+        self._configure(**kwargs)
+
+    def _infer_policy(self, **kwargs):
+        if isinstance(self.credential, AzureKeyCredential):
+            return policies.AzureKeyCredentialPolicy(self.credential, "Ocp-Apim-Subscription-Key", **kwargs)
+        if hasattr(self.credential, "get_token"):
+            return policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs)
+        raise TypeError(f"Unsupported credential: {self.credential}")
+
+    def _configure(self, **kwargs: Any) -> None:
+        self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs)
+        self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs)
+        self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs)
+        self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs)
+        self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs)
+        self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs)
+        self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs)
+        self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs)
+        self.authentication_policy = kwargs.get("authentication_policy")
+        if self.credential and not self.authentication_policy:
+            self.authentication_policy = self._infer_policy(**kwargs)

azure/ai/textanalytics/{_generated/v2022_05_01/operations → _operations}/__init__.py

@@ -2,17 +2,22 @@
 # --------------------------------------------------------------------------
 # 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) AutoRest Code Generator.
+# Code generated by Microsoft (R) Python Code Generator.
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
+# pylint: disable=wrong-import-position
 
-from ._text_analytics_client_operations import TextAnalyticsClientOperationsMixin
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ._patch import *  # pylint: disable=unused-wildcard-import
+
+from ._operations import _TextAnalysisClientOperationsMixin  # type: ignore # pylint: disable=unused-import
 
 from ._patch import __all__ as _patch_all
-from ._patch import *  # type: ignore # pylint: disable=unused-wildcard-import
+from ._patch import *
 from ._patch import patch_sdk as _patch_sdk
-__all__ = [
-    'TextAnalyticsClientOperationsMixin',
-]
-__all__.extend([p for p in _patch_all if p not in __all__])
-_patch_sdk()
+
+__all__ = []
+__all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore
+_patch_sdk()