17 lines
389 B
Python
17 lines
389 B
Python
from typing import Dict, List, Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from .classified_traces import Classification, Protocol
|
|
|
|
|
|
class Classifier(BaseModel):
|
|
classification: Classification
|
|
|
|
|
|
class ClassifierSpec(BaseModel):
|
|
abi_name: str
|
|
protocol: Optional[Protocol] = None
|
|
valid_contract_addresses: Optional[List[str]] = None
|
|
classifiers: Dict[str, Classifier] = {}
|