Enabled both stricter and previous object definition

mev_inspect/schemas/classified_traces.py

@@ -28,6 +28,32 @@ class ClassifiedTrace(BaseModel):
     trace_type: TraceType
     trace_address: List[int]
     classification: Classification
+    protocol: Optional[Protocol]
+    abi_name: Optional[str]
+    function_name: Optional[str]
+    function_signature: Optional[str]
+    inputs: Optional[Dict[str, Any]]
+    to_address: Optional[str]
+    from_address: Optional[str]
+    gas: Optional[int]
+    value: Optional[int]
+    gas_used: Optional[int]
+    error: Optional[str]
+
+    class Config:
+        json_encoders = {
+            # a little lazy but fine for now
+            # this is used for bytes value inputs
+            bytes: lambda b: b.hex(),
+        }
+
+
+class ClassifiedTraceStrict(BaseModel):
+    transaction_hash: str
+    block_number: int
+    trace_type: TraceType
+    trace_address: List[int]
+    classification: Classification
 
     class Config:
         json_encoders = {

tests/helpers.py

@@ -4,7 +4,6 @@ from mev_inspect.schemas.blocks import TraceType
 from mev_inspect.schemas.classified_traces import (
     Classification,
     ClassifiedTrace,
-    ClassifiedCallData,
 )
 
 
@@ -17,7 +16,7 @@ def make_transfer_trace(
     token_address: str,
     amount: int,
 ):
-    return ClassifiedCallData(
+    return ClassifiedTrace(
         transaction_hash=transaction_hash,
         block_number=block_number,
         trace_type=TraceType.call,
@@ -42,7 +41,7 @@ def make_swap_trace(
     recipient_address: str,
     recipient_input_key: str,
 ):
-    return ClassifiedCallData(
+    return ClassifiedTrace(
         transaction_hash=transaction_hash,
         block_number=block_number,
         trace_type=TraceType.call,