from typing import Dict, List, Union, Optional
from pydantic import BaseModel, Field

class JSONSchema(BaseModel):
    type: str
    format: Union[str, None] = None
    items: Union['JSONSchema', None] = None
    properties: Dict[str, 'JSONSchema'] = Field(default_factory=dict)
    additionalProperties: Union[bool, 'JSONSchema'] = True
    required: List[str] = Field(default_factory=list)

    class Config:
        arbitrary_types_allowed = True

class TaskTree(BaseModel):
    description: Optional[str]
    python_fn_signature: str
    sub_fns: List['TaskTree']

JSONSchema.update_forward_refs()
TaskTree.update_forward_refs()
#