Build Autonomous Factory Inspection Agents with CrewAI and PydanticAI

                      
                     
from typing import Dict, Any
import os
import logging
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel, ValidationError

# Configuration
API_KEY = os.getenv('CREWAI_API_KEY')
if not API_KEY:
    raise EnvironmentError('Missing API key for CrewAI')

logging.basicConfig(level=logging.INFO)
app = FastAPI()

# Data model using Pydantic
class InspectionRequest(BaseModel):
    factory_id: str
    inspection_type: str

# Core logic for inspection
async def perform_inspection(request: InspectionRequest) -> Dict[str, Any]:
    try:
        # Simulated API call to CrewAI
        # Replace with actual API call logic
        logging.info(f'Performing {request.inspection_type} for factory {request.factory_id}')
        # Simulated response
        return {'status': 'success', 'data': {'factory_id': request.factory_id, 'report': 'Inspection report details'}}
    except Exception as e:
        logging.error(f'Error during inspection: {str(e)}')
        raise HTTPException(status_code=500, detail='Internal Server Error')

@app.post('/inspect/', response_model=Dict[str, Any])
async def inspect(request: InspectionRequest):
    try:
        return await perform_inspection(request)
    except ValidationError as e:
        logging.error(f'Validation error: {e}')
        raise HTTPException(status_code=422, detail=e.errors())

if __name__ == '__main__':
    import uvicorn
    uvicorn.run(app, host='0.0.0.0', port=8000)