# Query results by Model ID
results = client.get_model_test_results(
    modelId="model_v2",
    limit=100,
    offset=0,
    sort_by="execution_timestamp",
    sort_order="DESC"
)

# Parse results
for result in results:
    print(f"Model: {result['model_id']}")
    print(f"Monitor: {result['monitor_id']}")
    print(f"Status: {result['status']}")  # PASS, FAIL, ERROR
    print(f"Metrics: {result['metrics_json']}")
    print(f"Timestamp: {result['execution_timestamp']}")
    print("---")

# Query by stored model
stored_model_results = client.get_model_test_results_by_stored_model_id(
    storedModelId="sm_12345",
    days_back=7
)

from mcp import Server, Tool, Resource
from mcp.server.stdio import StdioServerTransport

server = Server("model-monitoring-service")

@server.tool()
def create_monitor_job(model_id: str, monitor_type: str, data_asset: str):
    """Create a new monitoring job via MCP."""
    from moc_python_sdk import HttpClient
    
    client = HttpClient(endpoint=os.getenv("MODELOP_ENDPOINT"))
    job_response = client.create_job({
        "modelTestId": f"job_{model_id}_{monitor_type}",
        "monitorSnapshotId": f"snapshot_{monitor_type}_latest",
        "dataAssetMappings": {"sample_data": data_asset}
    })
    
    return {
        "job_id": job_response['id'],
        "status": job_response['status'],
        "created_at": job_response['created_timestamp']
    }

@server.tool()
def query_recent_results(model_id: str, days_back: int = 7):
    """Query monitoring results from the past N days."""
    client = HttpClient(endpoint=os.getenv("MODELOP_ENDPOINT"))
    results = client.get_model_test_results(modelId=model_id, limit=100)
    
    # Filter by date
    from datetime import datetime, timedelta
    cutoff_date = datetime.now() - timedelta(days=days_back)
    
    filtered = [r for r in results 
                if datetime.fromisoformat(r['execution_timestamp']) > cutoff_date]
    
    return {"count": len(filtered), "results": filtered}

@server.resource()
def get_monitor_status(model_id: str):
    """Expose monitor status as a resource for agent inspection."""
    client = HttpClient(endpoint=os.getenv("MODELOP_ENDPOINT"))
    status = client.get_model_status(model_id)
    return status

# Serve the MCP protocol
async def main():
    transport = StdioServerTransport()
    await server.run(transport)

if __name__ == "__main__":
    import asyncio
    asyncio.run(main())