Merge pull request #1 from saichandrapandraju/main

Fix Endpoint Duplication in FastAPI Swagger by Segregating Routers into Individual Tags

Author: RobGeada

src/endpoints/consumer.py renamed to src/endpoints/consumer/consumer_endpoint.py

@@ -0,0 +1,48 @@
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+from typing import Optional
+import logging
+
+router = APIRouter()
+logger = logging.getLogger(__name__)
+
+
+class ModelConfig(BaseModel):
+    target: str
+    name: str
+    version: Optional[str] = None
+
+
+class GlobalExplanationRequest(BaseModel):
+    modelConfig: ModelConfig
+
+
+@router.post("/explainers/global/lime")
+async def global_lime_explanation(request: GlobalExplanationRequest):
+    """Compute a global LIME explanation."""
+    try:
+        logger.info(
+            f"Computing global LIME explanation for model: {request.modelConfig.name}"
+        )
+        # TODO: Implement
+    except Exception as e:
+        logger.error(f"Error computing global LIME explanation: {str(e)}")
+        raise HTTPException(
+            status_code=500, detail=f"Error computing explanation: {str(e)}"
+        )
+
+
+@router.post("/explainers/global/pdp")
+async def global_pdp_explanation(request: GlobalExplanationRequest):
+    """Compute a global PDP explanation."""
+    try:
+        logger.info(
+            f"Computing global PDP explanation for model: {request.modelConfig.name}"
+        )
+        # TODO: Implement
+        return {"status": "success", "explanation": {}}
+    except Exception as e:
+        logger.error(f"Error computing global PDP explanation: {str(e)}")
+        raise HTTPException(
+            status_code=500, detail=f"Error computing explanation: {str(e)}"
+        )

src/endpoints/data_download.py renamed to src/endpoints/data/data_download.py

@@ -14,41 +14,6 @@ class ModelConfig(BaseModel):
     version: Optional[str] = None
 
 
-class GlobalExplanationRequest(BaseModel):
-    modelConfig: ModelConfig
-
-
-@router.post("/explainers/global/lime")
-async def global_lime_explanation(request: GlobalExplanationRequest):
-    """Compute a global LIME explanation."""
-    try:
-        logger.info(
-            f"Computing global LIME explanation for model: {request.modelConfig.name}"
-        )
-        # TODO: Implement
-    except Exception as e:
-        logger.error(f"Error computing global LIME explanation: {str(e)}")
-        raise HTTPException(
-            status_code=500, detail=f"Error computing explanation: {str(e)}"
-        )
-
-
-@router.post("/explainers/global/pdp")
-async def global_pdp_explanation(request: GlobalExplanationRequest):
-    """Compute a global PDP explanation."""
-    try:
-        logger.info(
-            f"Computing global PDP explanation for model: {request.modelConfig.name}"
-        )
-        # TODO: Implement
-        return {"status": "success", "explanation": {}}
-    except Exception as e:
-        logger.error(f"Error computing global PDP explanation: {str(e)}")
-        raise HTTPException(
-            status_code=500, detail=f"Error computing explanation: {str(e)}"
-        )
-
-
 class LimeExplainerConfig(BaseModel):
     n_samples: int = 300
     timeout: int = 10