store the ground truth about frauds

migrations/version_2.sql

@@ -0,0 +1,3 @@
+-- Alter table TRANSACTION. Add 'is_real_fraud' nullable column
+ALTER TABLE TRANSACTIONS
+ADD COLUMN is_real_fraud BOOLEAN DEFAULT NULL;

src/entity/api/transaction_api.py

@@ -16,6 +16,7 @@ class TransactionApi(BaseModel):
     transaction_timestamp: int = Field(..., title="Timestamp", description="The timestamp of the transaction.")
     transaction_amount: float = Field(..., ge=0, title="Amount", description="The transaction amount.")
     transaction_category: Optional[str] = Field(None, title="Product category", description="The category of product of the transaction.")
+    transaction_is_real_fraud: Optional[bool] = Field(..., title="Ground truth", description="True if the transaction really is a fraud (ground truth).")
     merchant_name: str = Field(..., title="Name", description="The name of the merchant.")
     merchant_latitude: float = Field(..., title="Latitude", description="The latitude of the merchant.")
     merchant_longitude: float = Field(..., title="Longitude", description="The longitude of the merchant.")
@@ -94,7 +95,8 @@ class TransactionApi(BaseModel):
             customer_address_zip=self.customer_postal_code,
             customer_address_latitude=self.customer_latitude,
             customer_address_longitude=self.customer_longitude,
-            customer_address_city_population=self.customer_city_population
+            customer_address_city_population=self.customer_city_population,
+            is_real_fraud=self.transaction_is_real_fraud,
         )
 
     class Config:
@@ -119,6 +121,6 @@ class TransactionApi(BaseModel):
                 "customer_postal_code": 94105,
                 "customer_latitude": 37.7749,
                 "customer_longitude": -122.4194,
-                "customer_city_population": 870000
+                "customer_city_population": 870000,
             }
         }

src/entity/transaction.py

@@ -34,6 +34,7 @@ class Transaction(Base):
     merchant_address_latitude: Mapped[float] = mapped_column(Float, nullable=True)
     merchant_address_longitude: Mapped[float] = mapped_column(Float, nullable=True)
     is_fraud: Mapped[bool] = mapped_column(Boolean, nullable=True)
+    is_real_fraud: Mapped[bool] = mapped_column(Boolean, nullable=True)
 
     # Dependent table
     fraud_details: Mapped[FraudDetails] = relationship("FraudDetails", back_populates="transaction", cascade="all, delete-orphan", uselist=False)