from llama_index.core.workflow import (
    Event
)
from typing import Optional

class QueryEvent(Event):
    """Query event."""
    query: str
    tag: str
    
class VideoSearch(Event):
    """Video search event."""
    query: str
    
class ImageSearch(Event):
    """Image search event."""
    query: str
    
class GeneralSearch(Event):
    """General search event."""
    query: str
    tag: str
    ads: Optional[str] = None


class TableRetrieveEvent(Event):
    """Result of retriever."""
    table_context_str: str
    query: str

    
class Text2SQLEvent(Event):
    """Text2SQL event."""
    sql: str
    query: str

    
class RetrieveContextEvent(Event):
    """Additional context."""
    context_str: str


class FullContextEvent(Event):
    """Additional context."""
    context_str: str


class CasualChatEvent(Event):
    """Casual chat event."""
    query: str


class SearchEvent(Event):
    """General question event."""
    query: str


class VideoResourceEvent(SearchEvent):
    """General question event."""
    query: str
    
    
class ImageResourceEvent(SearchEvent):
    """General question event."""
    query: str
  
    
class GeneralSearchEvent(SearchEvent):
    """General question event."""
    query: str
    tag: str


class MilvusDBEvent(Event):
    """MilvusDB event."""
    query: str


class MilvusDBSearchEvent(Event):
    """MilvusDB search event."""
    query: str
    
class SafeStartEvent(Event):
    """Safe start event."""
    query: str
    
    
class RefuseEvent(Event):
    """Refuse event."""
    lang: str
    query: Optional[str] = None
    adult: Optional[bool] = None
    
class StatusEvent(Event):
    """Status event."""
    status: str
    
class TokenEvent(Event):
    """Token event."""
    token: str
    
class ControlEvent(Event):
    """Control event."""
    toy_id: str
    query: str
    user_id: str
    content: str