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

azure/ai/textanalytics/_policies.py

@@ -1,66 +0,0 @@
-# ------------------------------------
-# Copyright (c) Microsoft Corporation.
-# Licensed under the MIT License.
-# ------------------------------------
-
-from azure.core.pipeline.policies import ContentDecodePolicy
-from azure.core.pipeline.policies import SansIOHTTPPolicy
-from azure.core.exceptions import HttpResponseError
-from ._models import TextDocumentBatchStatistics
-from ._lro import _FINISHED
-
-
-class TextAnalyticsResponseHookPolicy(SansIOHTTPPolicy):
-    def __init__(self, **kwargs):
-        self._response_callback = kwargs.get("raw_response_hook")
-        self._is_lro = None
-        super().__init__()
-
-    def on_request(self, request):
-        self._response_callback = request.context.options.pop(
-            "raw_response_hook", self._response_callback
-        )
-
-    def on_response(self, request, response):
-        if self._is_lro is None:
-            # determine LRO based off of initial response. If 202, we say it's an LRO
-            self._is_lro = response.http_response.status_code == 202
-        if self._response_callback:
-            data = ContentDecodePolicy.deserialize_from_http_generics(
-                response.http_response
-            )
-            if self._is_lro and (not data or data.get("status", "").lower() not in _FINISHED):
-                return
-            if response.http_response.status_code == 429:
-                return
-            if data:
-                inner = data.get("results", data)  # language API compat
-                statistics = inner.get("statistics", None)
-                model_version = inner.get("modelVersion", None)
-                batch_statistics = TextDocumentBatchStatistics._from_generated(  # pylint: disable=protected-access
-                    statistics
-                )
-                response.statistics = batch_statistics
-                response.model_version = model_version
-            response.raw_response = data
-            self._response_callback(response)
-
-
-class QuotaExceededPolicy(SansIOHTTPPolicy):
-    """Raises an exception immediately when the call quota volume has been exceeded in a F0
-    tier language resource. This is to avoid waiting the Retry-After time returned in
-    the response.
-    """
-
-    def on_response(self, request, response):
-        """Is executed after the request comes back from the policy.
-
-        :param request: Request to be modified after returning from the policy.
-        :type request: ~azure.core.pipeline.PipelineRequest
-        :param response: Pipeline response object
-        :type response: ~azure.core.pipeline.PipelineResponse
-        """
-        http_response = response.http_response
-        if http_response.status_code == 403 and \
-                "Out of call volume quota for TextAnalytics F0 pricing tier" in http_response.text():
-            raise HttpResponseError(http_response.text(), response=http_response)

azure/ai/textanalytics/_request_handlers.py

@@ -1,104 +0,0 @@
-# ------------------------------------
-# Copyright (c) Microsoft Corporation.
-# Licensed under the MIT License.
-# ------------------------------------
-
-
-from ._models import (
-    DetectLanguageInput,
-    TextDocumentInput,
-    _AnalyzeActionsType,
-)
-
-
-def _validate_input(documents, hint, whole_input_hint):
-    """Validate that batch input has either all string docs
-    or dict/DetectLanguageInput/TextDocumentInput, not a mix of both.
-    Assign country and language hints on a whole batch or per-item
-    basis.
-
-    :param list documents: The input documents.
-    :return: A list of DetectLanguageInput or TextDocumentInput
-    """
-    if not documents:
-        raise ValueError("Input documents can not be empty or None")
-
-    if isinstance(documents, str):
-        raise TypeError("Input documents cannot be a string.")
-
-    if isinstance(documents, dict):
-        raise TypeError("Input documents cannot be a dict")
-
-    if not all(isinstance(x, str) for x in documents):
-        if not all(
-            isinstance(x, (dict, TextDocumentInput, DetectLanguageInput))
-            for x in documents
-        ):
-            raise TypeError(
-                "Mixing string and dictionary/object document input unsupported."
-            )
-
-    request_batch = []
-    for idx, doc in enumerate(documents):
-        if isinstance(doc, str):
-            if hint == "country_hint" and whole_input_hint.lower() == "none":
-                whole_input_hint = ""
-            document = {"id": str(idx), hint: whole_input_hint, "text": doc}
-            request_batch.append(document)
-        if isinstance(doc, dict):
-            item_hint = doc.get(hint, None)
-            if item_hint is None:
-                doc = {
-                    "id": doc.get("id", None),
-                    hint: whole_input_hint,
-                    "text": doc.get("text", None),
-                }
-            elif item_hint.lower() == "none":
-                doc = {
-                    "id": doc.get("id", None),
-                    hint: "",
-                    "text": doc.get("text", None),
-                }
-            request_batch.append(doc)
-        if isinstance(doc, TextDocumentInput):
-            item_hint = doc.language
-            if item_hint is None:
-                doc = TextDocumentInput(
-                    id=doc.id, language=whole_input_hint, text=doc.text
-                )
-            request_batch.append(doc)
-        if isinstance(doc, DetectLanguageInput):
-            item_hint = doc.country_hint
-            if item_hint is None:
-                doc = DetectLanguageInput(
-                    id=doc.id, country_hint=whole_input_hint, text=doc.text
-                )
-            elif item_hint.lower() == "none":
-                doc = DetectLanguageInput(id=doc.id, country_hint="", text=doc.text)
-            request_batch.append(doc)
-
-    return request_batch
-
-
-def _determine_action_type(action):  # pylint: disable=too-many-return-statements
-    if action.__class__.__name__ in ["EntitiesTask", "EntitiesLROTask"]:
-        return _AnalyzeActionsType.RECOGNIZE_ENTITIES
-    if action.__class__.__name__ in ["PiiTask", "PiiLROTask"]:
-        return _AnalyzeActionsType.RECOGNIZE_PII_ENTITIES
-    if action.__class__.__name__ in ["EntityLinkingTask", "EntityLinkingLROTask"]:
-        return _AnalyzeActionsType.RECOGNIZE_LINKED_ENTITIES
-    if action.__class__.__name__ in ["SentimentAnalysisTask", "SentimentAnalysisLROTask"]:
-        return _AnalyzeActionsType.ANALYZE_SENTIMENT
-    if action.__class__.__name__ == "CustomEntitiesLROTask":
-        return _AnalyzeActionsType.RECOGNIZE_CUSTOM_ENTITIES
-    if action.__class__.__name__ == "CustomSingleLabelClassificationLROTask":
-        return _AnalyzeActionsType.SINGLE_LABEL_CLASSIFY
-    if action.__class__.__name__ == "CustomMultiLabelClassificationLROTask":
-        return _AnalyzeActionsType.MULTI_LABEL_CLASSIFY
-    if action.__class__.__name__ == "HealthcareLROTask":
-        return _AnalyzeActionsType.ANALYZE_HEALTHCARE_ENTITIES
-    if action.__class__.__name__ == "ExtractiveSummarizationLROTask":
-        return _AnalyzeActionsType.EXTRACT_SUMMARY
-    if action.__class__.__name__ == "AbstractiveSummarizationLROTask":
-        return _AnalyzeActionsType.ABSTRACT_SUMMARY
-    return _AnalyzeActionsType.EXTRACT_KEY_PHRASES