"""Represents a connection response message."""
from marshmallow import EXCLUDE, fields
from .....messaging.agent_message import AgentMessage, AgentMessageSchema
from ..message_types import CONNECTION_RESPONSE, PROTOCOL_PACKAGE
from ..models.connection_detail import ConnectionDetail, ConnectionDetailSchema
HANDLER_CLASS = (
f"{PROTOCOL_PACKAGE}.handlers."
"connection_response_handler.ConnectionResponseHandler"
)
[docs]class ConnectionResponse(AgentMessage):
"""Class representing a connection response."""
def __init__(self, *, connection: ConnectionDetail = None, **kwargs):
"""
Initialize connection response object.
Args:
connection: Connection details object
"""
super().__init__(**kwargs)
self.connection = connection
[docs]class ConnectionResponseSchema(AgentMessageSchema):
"""Connection response schema class."""
connection = fields.Nested(ConnectionDetailSchema, required=True)