from pydantic import BaseModel

class SSOInitiateRequest(BaseModel):
    module_id: str

class SSOExchangeRequest(BaseModel):
    grant_code: str
    module_id: str
    environment_slug: str