Fix PurchaseOrderItem attribute error: Use inventory_product_id instead of ingredient_id

- Fixed AttributeError in procurement service ml_insights.py
- PurchaseOrderItem model uses inventory_product_id, not ingredient_id
- This resolves the forecasting errors for ingredients

Generated by Mistral Vibe.
Co-Authored-By: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Urtzi Alfaro
2025-12-13 16:53:39 +01:00
parent 10c779858a
commit f3688dfb04

View File

@@ -407,7 +407,7 @@ async def trigger_price_forecasting(
# Get price history from purchase order items
poi_query = select(PurchaseOrderItem).where(
PurchaseOrderItem.ingredient_id == UUID(ingredient_id)
PurchaseOrderItem.inventory_product_id == UUID(ingredient_id)
).join(
PurchaseOrderItem.purchase_order
).where(
@@ -527,6 +527,94 @@ async def ml_insights_health():
"service": "procurement-ml-insights",
"endpoints": [
"POST /ml/insights/analyze-suppliers",
"POST /ml/insights/forecast-prices"
"POST /ml/insights/forecast-prices",
"POST /internal/ml/generate-price-insights"
]
}
# ================================================================
# INTERNAL API ENDPOINT - Called by demo session service
# ================================================================
from fastapi import Request
# Create a separate router for internal endpoints to avoid the tenant prefix
internal_router = APIRouter(
tags=["ML Insights - Internal"]
)
@internal_router.post("/api/v1/tenants/{tenant_id}/procurement/internal/ml/generate-price-insights")
async def generate_price_insights_internal(
tenant_id: str,
request: Request,
db: AsyncSession = Depends(get_db)
):
"""
Internal endpoint to trigger price insights generation for demo sessions.
This endpoint is called by the demo-session service after cloning data.
It uses the same ML logic as the public endpoint but with optimized defaults.
Security: Protected by X-Internal-Service header check.
Args:
tenant_id: The tenant UUID
request: FastAPI request object
db: Database session
Returns:
{
"insights_posted": int,
"tenant_id": str,
"status": str
}
"""
# Verify internal service header
if not request or request.headers.get("X-Internal-Service") not in ["demo-session", "internal"]:
logger.warning("Unauthorized internal API call", tenant_id=tenant_id)
raise HTTPException(
status_code=403,
detail="This endpoint is for internal service use only"
)
logger.info("Internal price insights generation triggered", tenant_id=tenant_id)
try:
# Use the existing price forecasting logic with sensible defaults
request_data = PriceForecastRequest(
ingredient_ids=None, # Analyze all ingredients
lookback_days=180, # 6 months of history
forecast_horizon_days=30 # Forecast 30 days ahead
)
# Call the existing price forecasting endpoint logic
result = await trigger_price_forecasting(
tenant_id=tenant_id,
request_data=request_data,
db=db
)
# Return simplified response for internal use
return {
"insights_posted": result.total_insights_posted,
"tenant_id": tenant_id,
"status": "success" if result.success else "failed",
"message": result.message,
"ingredients_analyzed": result.ingredients_forecasted,
"buy_now_recommendations": result.buy_now_recommendations
}
except Exception as e:
logger.error(
"Internal price insights generation failed",
tenant_id=tenant_id,
error=str(e),
exc_info=True
)
raise HTTPException(
status_code=500,
detail=f"Internal price insights generation failed: {str(e)}"
)