move TransactionProcessingOutput to entity file

src/entity/api/transaction_api.py

@@ -1,5 +1,5 @@
-from typing import Optional, Literal
-from pydantic import BaseModel, Field
+from typing import Optional, Literal, ClassVar
+from pydantic import BaseModel, Field, ConfigDict
 from datetime import datetime
 from src.entity.transaction import Transaction
 import logging
@@ -122,4 +122,21 @@ class TransactionApi(BaseModel):
                 "customer_longitude": -122.4194,
                 "customer_city_population": 870000,
             }
-        }
+        }
+
+class TransactionProcessingOutput(BaseModel):
+    """
+    TransactionProcessingOutput is a class that represents the output of the transaction processing endpoint.
+    It contains the fraud detection result and the fraud score.
+    This class is used as the response model for the /transaction/process endpoint.
+    """
+    is_fraud: int = Field(description="The prediction result. 1 if the transaction is detected as fraudulent, 0 otherwise.")
+    fraud_score: float = Field(description="Probability of transaction being fraudulent.")
+
+    model_config: ClassVar[ConfigDict] = ConfigDict(
+        json_schema_extra = {
+            "example": {
+                "is_fraud": 1,
+                "fraud_score": 0.85
+            }
+        })

src/main.py

@@ -2,7 +2,7 @@ import os
 import logging
 import secrets
 import requests
-from typing import Annotated, Generator, ClassVar
+from typing import Annotated, Generator
 from fastapi import (
     FastAPI,
     Request,
@@ -13,7 +13,6 @@ from fastapi import (
 from fastapi.background import BackgroundTasks
 from fastapi.responses import RedirectResponse, JSONResponse
 from fastapi.security.api_key import APIKeyHeader
-from pydantic import BaseModel, Field, ConfigDict
 from psycopg.errors import UniqueViolation
 from sqlalchemy.exc import IntegrityError
 from starlette.status import (
@@ -26,7 +25,7 @@ from dotenv import load_dotenv
 from sqlalchemy import text
 from sqlalchemy.orm import Session
 
-from src.entity.api.transaction_api import TransactionApi
+from src.entity.api.transaction_api import TransactionApi, TransactionProcessingOutput
 from src.service.fraud_service import check_for_fraud_api
 from src.service.logging_service import setup_logging
 from src.service.notification_service import send_notification
@@ -81,24 +80,6 @@ def redirect_to_docs():
     Redirect to the API documentation.
     '''
     return RedirectResponse(url='/docs')
-
-
-class TransactionProcessingOutput(BaseModel):
-    """
-    TransactionProcessingOutput is a class that represents the output of the transaction processing endpoint.
-    It contains the fraud detection result and the fraud score.
-    This class is used as the response model for the /transaction/process endpoint.
-    """
-    is_fraud: int = Field(description="The prediction result. 1 if the transaction is detected as fraudulent, 0 otherwise.")
-    fraud_score: float = Field(description="Probability of transaction being fraudulent.")
-
-    model_config: ClassVar[ConfigDict] = ConfigDict(
-        json_schema_extra = {
-            "example": {
-                "is_fraud": 1,
-                "fraud_score": 0.85
-            }
-        })
     
 @app.get("/transaction/get_fraud_status/{transaction_number}",
           tags=["transaction"],