from typing import Any

from camel.prompts.base import TextPrompt, TextPromptDict
from camel.types import RoleType

flake8: noqa :E501

class VideoDescriptionPromptTemplateDict(TextPromptDict):
r"""A dictionary containing :obj:TextPrompt used in the VideoDescription
task.

Attributes:
    ASSISTANT_PROMPT (TextPrompt): A prompt for the AI assistant to
        provide a shot description of the content of the current video.
"""

ASSISTANT_PROMPT = TextPrompt(
    """You are a master of video analysis.
    Please provide a shot description of the content of the current video."""
)

def __init__(self, *args: Any, **kwargs: Any) -> None:
    super().__init__(*args, **kwargs)
    self.update(
        {
            RoleType.ASSISTANT: self.ASSISTANT_PROMPT,
        }
    )