def _get_msg_type()

in src/dfcx_scrapi/tools/search_util.py [0:0]

• 22 lines of code
• 10 McCabe index (conditional complexity)

    def _get_msg_type(message: types.ResponseMessage):
        """Gets the response message type for a message from a fulfillment.

        Args:
          message: message structure from a fulfillment.

        Returns:
          type in {np.nan, text, custom_payload, play_audio,
            live_agent_handoff, conversation_success, output_audio_text}.
        """
        if pd.isna(message):
            value = np.nan
        elif isinstance(message, types.ResponseMessage) and (
            str(message) == ""
        ):
            value = np.nan
        elif "text" in message:
            value = "text"
        elif "payload" in message:
            value = "custom_payload"
        elif "play_audio" in message:
            value = "play_audio"
        elif "live_agent_handoff" in message:
            value = "live_agent_handoff"
        elif "conversation_success" in message:
            value = "conversation_success"
        elif "output_audio_text" in message:
            value = "output_audio_text"
        else:
            value = "unexpected value"
        return value