utils/model.py

from pydantic import BaseModel, Field
from typing import Optional


class MultiModuleRequest(BaseModel):
    issue_key: Optional[str] = Field(
        default="", description="The unique identifier for the issue or bug ticket (e.g., JIRA key). "
    )
    summary: str = Field(..., description="The short title or summary of the bug.")
    description: Optional[str] = Field(default=None, description="A detailed description of the bug.")


class SearchSpaceRoutingRequest(BaseModel):
    issue_key: Optional[str] = Field(
        default="", description="The unique identifier for the issue or bug ticket (e.g., JIRA key). "
    )
    summary: str = Field(..., description="The short title or summary of the bug.")
    description: Optional[str] = Field(default=None, description="A detailed description of the bug.")


class SingleModuleRequest(BaseModel):
    issue_key: Optional[str] = Field(
        default="", description="The unique identifier for the issue or bug ticket (e.g., JIRA key). "
    )
    summary: str = Field(..., description="The short title or summary of the bug.")
    description: Optional[str] = Field(default=None, description="A detailed description of the bug.")
    module: str = Field(..., description="Name of the module where the bug should be localized")