diff --git a/mev_inspect/schemas/blocks.py b/mev_inspect/schemas/blocks.py index 5ad16f2..d26e8a6 100644 --- a/mev_inspect/schemas/blocks.py +++ b/mev_inspect/schemas/blocks.py @@ -1,6 +1,8 @@ from enum import Enum from typing import Dict, List, Optional +from pydantic import BaseModel + from .utils import CamelModel, Web3Model @@ -36,3 +38,11 @@ class Block(Web3Model): def get_filtered_traces(self, hash: str) -> List[Trace]: return [trace for trace in self.traces if trace.transaction_hash == hash] + + +class NestedTrace(BaseModel): + trace: Trace + subtraces: List["NestedTrace"] + + +NestedTrace.update_forward_refs()