from pydantic import BaseModel, Field

class ServiceDocument(BaseModel):
    name: str = Field(..., description="Name of the service")
    port: int = Field(..., description="Port number of the service")
    category: str = Field(..., description="Category of the service (e.g., web, pwn, etc.)")