fix demo session 2

services/inventory/app/api/internal.py

@@ -0,0 +1,55 @@
+"""
+Internal API for Inventory Service
+Handles internal service-to-service operations
+"""
+
+from fastapi import APIRouter, Depends, HTTPException, Header
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy import select, func
+from uuid import UUID
+import structlog
+
+from app.core.database import get_db
+from app.core.config import settings
+from app.models import Ingredient
+
+logger = structlog.get_logger()
+router = APIRouter(prefix="/internal", tags=["internal"])
+
+
+async def verify_internal_api_key(x_internal_api_key: str = Header(None)):
+    """Verify internal API key for service-to-service communication"""
+    required_key = settings.INTERNAL_API_KEY
+    if x_internal_api_key != required_key:
+        logger.warning("Unauthorized internal API access attempted")
+        raise HTTPException(status_code=403, detail="Invalid internal API key")
+    return True
+
+
+@router.get("/count")
+async def get_ingredient_count(
+    tenant_id: str,
+    db: AsyncSession = Depends(get_db),
+    _: bool = Depends(verify_internal_api_key)
+):
+    """
+    Get count of active ingredients for onboarding status check.
+    Internal endpoint for tenant service.
+    """
+    try:
+        count = await db.scalar(
+            select(func.count()).select_from(Ingredient)
+            .where(
+                Ingredient.tenant_id == UUID(tenant_id),
+                Ingredient.is_active == True
+            )
+        )
+
+        return {
+            "count": count or 0,
+            "tenant_id": tenant_id
+        }
+
+    except Exception as e:
+        logger.error("Failed to get ingredient count", tenant_id=tenant_id, error=str(e))
+        raise HTTPException(status_code=500, detail=f"Failed to get ingredient count: {str(e)}")

services/inventory/app/api/internal_demo.py

@@ -610,35 +610,4 @@ async def delete_demo_tenant_data(
         raise HTTPException(
             status_code=500,
             detail=f"Failed to delete demo data: {str(e)}"
-        )
-
-
-@router.get("/count")
-async def get_ingredient_count(
-    tenant_id: str,
-    db: AsyncSession = Depends(get_db),
-    _: bool = Depends(verify_internal_api_key)
-):
-    """
-    Get count of active ingredients for onboarding status check.
-    Internal endpoint for tenant service.
-    """
-    try:
-        from sqlalchemy import select, func
-
-        count = await db.scalar(
-            select(func.count()).select_from(Ingredient)
-            .where(
-                Ingredient.tenant_id == UUID(tenant_id),
-                Ingredient.is_active == True
-            )
-        )
-
-        return {
-            "count": count or 0,
-            "tenant_id": tenant_id
-        }
-
-    except Exception as e:
-        logger.error("Failed to get ingredient count", tenant_id=tenant_id, error=str(e))
-        raise HTTPException(status_code=500, detail=f"Failed to get ingredient count: {str(e)}")
+        )

services/inventory/app/main.py

@@ -33,7 +33,8 @@ from app.api import (
     sustainability,
     audit,
     ml_insights,
-    enterprise_inventory
+    enterprise_inventory,
+    internal
 )
 from app.api.internal_alert_trigger import router as internal_alert_trigger_router
 from app.api.internal_demo import router as internal_demo_router
@@ -215,6 +216,7 @@ service.add_router(dashboard.router)
 service.add_router(analytics.router)
 service.add_router(sustainability.router)
 service.add_router(internal_demo.router, tags=["internal-demo"])
+service.add_router(internal.router)
 service.add_router(ml_insights.router)  # ML insights endpoint
 service.add_router(ml_insights.internal_router)  # Internal ML insights endpoint for demo cloning
 service.add_router(internal_alert_trigger_router)  # Internal alert trigger for demo cloning