src/models.py

import os
from pydantic import BaseModel
from typing import Literal

class Config(BaseModel):
	target_namespace: str
	input_dataset: str
	input_model: str
	autotrain_project_prefix: str


class WebhookPayloadEvent(BaseModel):
	action: Literal["create", "update", "delete"]
	scope: str

class WebhookPayloadRepo(BaseModel):
	type: Literal["dataset", "model", "space"]
	name: str
	id: str
	private: bool
	headSha: str

class WebhookPayload(BaseModel):
	event: WebhookPayloadEvent
	repo: WebhookPayloadRepo


config = Config.parse_file(os.path.join(os.getcwd(), "config.json"))