conversation.py

"""Conversation-level events and capabilities."""

from typing import Any

from pydantic import BaseModel, ConfigDict, Field


class UiPathConversationCapabilities(BaseModel):
    """Describes the capabilities of a conversation participant."""

    async_input_stream_emitter: bool | None = Field(
        None, alias="asyncInputStreamEmitter"
    )
    async_input_stream_handler: bool | None = Field(
        None, alias="asyncInputStreamHandler"
    )
    async_tool_call_emitter: bool | None = Field(None, alias="asyncToolCallEmitter")
    async_tool_call_handler: bool | None = Field(None, alias="asyncToolCallHandler")
    mime_types_emitted: list[str] | None = Field(None, alias="mimeTypesEmitted")
    mime_types_handled: list[str] | None = Field(None, alias="mimeTypesHandled")

    model_config = ConfigDict(
        validate_by_name=True, validate_by_alias=True, extra="allow"
    )


class UiPathConversationStartEvent(BaseModel):
    """Signals the start of a conversation event stream."""

    capabilities: UiPathConversationCapabilities | None = None
    meta_data: dict[str, Any] | None = Field(None, alias="metaData")

    model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)


class UiPathConversationStartedEvent(BaseModel):
    """Signals the acceptance of the start of a conversation."""

    capabilities: UiPathConversationCapabilities | None = None

    model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)


class UiPathConversationEndEvent(BaseModel):
    """Signals the end of a conversation event stream."""

    meta_data: dict[str, Any] | None = Field(None, alias="metaData")

    model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)