Fixing the tests and rerecording the test data

Author: Vinothini Dharmaraj

sdk/communication/azure-communication-callautomation/azure/communication/callautomation/__init__.py

@@ -8,7 +8,6 @@
 from ._version import VERSION
 from ._call_automation_client import CallAutomationClient
 from ._call_connection_client import CallConnectionClient
-from ._streaming_data_parser import StreamingDataParser
 from ._models import (
     CallConnectionProperties,
     FileSource,
@@ -32,11 +31,7 @@
     ServerCallLocator,
     GroupCallLocator,
     AzureBlobContainerRecordingStorage,
-    AzureCommunicationsRecordingStorage,
-    AudioMetadata,
-    AudioData,
-    TranscriptionMetadata,
-    TranscriptionData
+    AzureCommunicationsRecordingStorage
 )
 from ._shared.models import (
     CommunicationIdentifier,
@@ -74,9 +69,6 @@
     "CallAutomationClient",
     "CallConnectionClient",
 
-    # parser
-    "StreamingDataParser",
-
     # models for input
     "FileSource",
     "TextSource",
@@ -89,10 +81,6 @@
     "TranscriptionSubscription",
     "AzureBlobContainerRecordingStorage",
     "AzureCommunicationsRecordingStorage",
-    "AudioMetadata",
-    "AudioData",
-    "TranscriptionMetadata",
-    "TranscriptionData",
 
     # models for output
     "CallConnectionProperties",

sdk/communication/azure-communication-callautomation/azure/communication/callautomation/_models.py

@@ -6,8 +6,6 @@
 # --------------------------------------------------------------------------
 
 from typing import List, Optional, Union, TYPE_CHECKING
-from enum import Enum
-from datetime import datetime
 from typing_extensions import Literal
 from ._generated.models import (
     CallLocator,
@@ -966,188 +964,3 @@ def _from_generated(cls, cancel_add_participant_operation_result_generated: 'Can
             invitation_id=cancel_add_participant_operation_result_generated.invitation_id,
             operation_context=cancel_add_participant_operation_result_generated.operation_context
         )
-
-class TextFormat(Enum):
-    """
-    The format of transcription text.
-    """
-    DISPLAY = "display"
-
-class WordData:
-    """
-    Text in the phrase.
-    :keyword text: Text in the phrase.
-    :paramtype text: str
-    :keyword offset: The word's position within the phrase.
-    :paramtype offset: int
-    :keyword duration: Duration in ticks. 1 tick = 100 nanoseconds.
-    :paramtype duration: int
-    """
-    text:str
-    """ Text in the phrase. """
-    offset:int
-    """ The word's position within the phrase. """
-    duration:int
-    """ Duration in ticks. 1 tick = 100 nanoseconds. """
-
-    def __init__(self, text: str, offset: int, duration: int):
-        self.text = text
-        self.offset = offset
-        self.duration = duration
-
-class TranscriptionMetadata:
-    """
-    Metadata for Transcription Streaming.
-    :keyword subscription_id: Transcription Subscription Id.
-    :paramtype subscription_id: str
-    :keyword locale: The target locale in which the translated text needs to be.
-    :paramtype locale: str
-    :keyword callConnection_id: call connection Id.
-    :paramtype callConnection_id: str
-    :keyword correlation_id: correlation Id.
-    :paramtype correlation_id: str
-    """
-
-    subscription_id: str
-    """ Transcription Subscription Id. """
-    locale: str
-    """ The target locale in which the translated text needs to be. """
-    call_connection_id: str
-    """ call connection Id. """
-    correlation_id: str
-    """ correlation Id. """
-
-    def __init__(
-            self,
-            *,
-            subscription_id: str,
-            locale: str,
-            call_connection_id: str,
-            correlation_id: str):
-        self.subscription_id = subscription_id
-        self.locale = locale
-        self.call_connection_id = call_connection_id
-        self.correlation_id = correlation_id
-
-class TranscriptionData:
-    """
-    Streaming Transcription.
-    :keyword  text: The display form of the recognized word.
-    :paramtype  text: str
-    :keyword format: The format of text.
-    :paramtype format: TextFormat
-    :keyword confidence: Confidence of recognition of the whole phrase.
-    :paramtype confidence: float
-    :keyword offset: The position of this payload.
-    :paramtype offset: int
-    :keyword duration: Duration in ticks. 1 tick = 100 nanoseconds.
-    :paramtype duration: int
-    :keyword words: The result for each word of the phrase.
-    :paramtype words: List[WordData]
-    :keyword participant: The identified speaker based on participant raw ID.
-    :paramtype participant: CommunicationIdentifier
-    :keyword result_state: Status of the result of transcription.
-    :paramtype result_state: str or
-     ~azure.communication.callautomation.models.TranscriptionResultState
-    """
-    text: str
-    """ The display form of the recognized word. """
-    format: TextFormat
-    """ The format of text. """
-    confidence: float
-    """ Confidence of recognition of the whole phrase, from 0.0 (no confidence) to 1.0 (full confidence). """
-    offset: int
-    """ The position of this payload. """
-    duration: int
-    """ Duration in ticks. 1 tick = 100 nanoseconds. """
-    words: List[WordData]
-    """ The result for each word of the phrase. """
-    participant: CommunicationIdentifier
-    """ The identified speaker based on participant raw ID. """
-    result_state: 'TranscriptionResultState'
-    """ State of the result of transcription. """
-
-    def __init__(
-            self,
-            *,
-            text: str,
-            format: TextFormat,
-            confidence: float,
-            offset: int,
-            duration: int,
-            words: List[WordData],
-            participant: CommunicationIdentifier,
-            result_state: 'TranscriptionResultState'):
-        self.text = text
-        self.format = format
-        self.confidence = confidence
-        self.offset = offset
-        self.duration = duration
-        self.words = words
-        self.participant = participant
-        self.result_state = result_state
-
-class AudioMetadata:
-    """
-    Metadata for Audio Streaming.
-    :keyword  subscription_id: Audio subscription id.
-    :paramtype  subscription_id: str
-    :keyword encoding: Audio encoding.
-    :paramtype encoding: str
-    :keyword sample_rate: Audio sample rate.
-    :paramtype sample_rate: int
-    :keyword channels: Audio channels.
-    :paramtype channels: int
-    :keyword length: Audio length.
-    :paramtype length: int
-    """
-    subscription_id: str
-    """ Audio subscription id. """
-    encoding: str
-    """ Audio encoding. """
-    sample_rate: int
-    """ Audio sample rate. """
-    channels: int
-    """ Audio channels. """
-    length: int
-    """ Audio length. """
-
-    def __init__(
-            self,
-            *,
-            subscription_id: str,
-            encoding: str,
-            sample_rate: int,
-            channels: int,
-            length: int):
-        self.subscription_id = subscription_id
-        self.encoding = encoding
-        self.sample_rate = sample_rate
-        self.channels = channels
-        self.length = length
-
-class AudioData:
-    """
-    Data for Audio Streaming.
-    :keyword  data: Audio streaming data.
-    :paramtype  data: str
-    :keyword time_stamp: time stamp.
-    :paramtype time_stamp: datetime
-    :keyword is_silent: Is silent.
-    :paramtype is_silent: bool
-    """
-    data: str
-    """ Audio streaming data. """
-    time_stamp : datetime
-    """ Time stamp. """
-    is_silent : bool
-    """ Is silent. """
-    def __init__(
-           self,
-           *,
-           data: str,
-           time_stamp: str,
-           is_silent: bool):
-        self.data = data
-        self.time_stamp = time_stamp
-        self.is_silent = is_silent

sdk/communication/azure-communication-callautomation/azure/communication/callautomation/_streaming_data_parser.py

@@ -9,7 +9,7 @@
 import os
 import inspect
 from datetime import datetime, timedelta
-from typing import Dict, Any, List
+from typing import Dict, Any, List, Optional
 
 import requests
 from azure.servicebus import ServiceBusClient
@@ -170,7 +170,7 @@ def check_for_event(self, event_type: str, call_connection_id: str, wait_time: t
             time.sleep(1)
         return None
 
-    def establish_callconnection_voip(self, caller, target) -> tuple:
+    def establish_callconnection_voip(self, caller, target, *, cognitive_service_enabled: Optional[bool] = False) -> tuple:
         call_automation_client_caller = CallAutomationClient.from_connection_string(self.connection_str, source=caller) # for creating call
         call_automation_client_target = CallAutomationClient.from_connection_string(self.connection_str, source=target) # answering call, all other actions
 
@@ -189,7 +189,10 @@ def establish_callconnection_voip(self, caller, target) -> tuple:
             thread.start()
 
         # create a call
-        create_call_result = call_automation_client_caller.create_call(target_participant=target, callback_url=(self.dispatcher_callback + "?q={}".format(unique_id)))
+        create_call_result = call_automation_client_caller.create_call(
+            target_participant=target, 
+            callback_url=(self.dispatcher_callback + "?q={}".format(unique_id)),
+            cognitive_services_endpoint=self.cognitive_service_endpoint if cognitive_service_enabled else None)
 
         if create_call_result is None:
             raise ValueError("Invalid create_call_result")