def set_output_schema(self, base_model: type[BaseModel]) -> None:
    """Sets the output schema for the request.

    Args:
      base_model: The pydantic base model to set the output schema to.
    """

    self.config.response_schema = base_model
    self.config.response_mime_type = 'application/json'
Sets the output schema for the request.

    Args:
      base_model: The pydantic base model to set the output schema to.
    
src/google/adk/models/llm_request.py