from pydantic import BaseModel, Field
from typing import Optional, Dict


class MergeInput(BaseModel):
    first_word: str = Field(..., title="First word", description="The first word to be merged")
    second_word: str = Field(..., title="Second word", description="The second word to be merged")

class MergeOutput(BaseModel):
    status: int = Field(..., title="Status", description="The status of the response")
    response: Optional[Dict[str, str]] = Field(None, title="Response", description="The response data if any")
    error: Optional[str] = Field(None, title="Error", description="The error message if any")