Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 11 additions & 7 deletions neon_data_models/models/api/mq/neon.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,6 @@
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

from typing import Annotated, List, Literal, Union
from ovos_bus_client.message import Message
from pydantic import Field, TypeAdapter, model_validator

from neon_data_models.models.base import BaseModel
Expand Down Expand Up @@ -174,16 +173,21 @@ class _RequestData(SkillApiRequestData):
description="API request data including `msg_type` and call params"
)

def as_messagebus_message(self) -> Message:
def as_messagebus_message(self) -> "Message": # type: ignore
"""
Override default Message translation to account for `msg_type` being
specified in request data
"""
return Message(
msg_type=self.data.msg_type,
data={"args": self.data.args, "kwargs": self.data.kwargs},
context=self.context.model_dump()
)
try:
from ovos_bus_client.message import Message
return Message(
msg_type=self.data.msg_type,
data={"args": self.data.args, "kwargs": self.data.kwargs},
context=self.context.model_dump()
)
except ImportError:
raise RuntimeError("pip install ovos-bus-client to enable Message "
"deserialization.")

class NeonMqCallSkillApiResponse(NeonCallSkillApiResponse, MQContext):
"""
Expand Down
Loading