Quick Start

import asyncio
from modelred import ModelRed

async def main(): # Initialize ModelRed client
async with ModelRed(api_key="mr_your_api_key_here") as client:
print("🔐 Authenticating with ModelRed...")

        # Validate your API key
        account = await client.validate_api_key()
        print(f"✅ Connected! Plan: {account.get('plan', 'Free')}")

        # Register your AI model
        print("📝 Registering OpenAI model...")
        await client.register_openai_model(
            model_id="my-gpt-model",
            api_key="sk-your-openai-key",
            model_name="gpt-3.5-turbo",
            metadata={"environment": "testing"}
        )
        print("✅ Model registered successfully!")

        # Run a comprehensive security assessment
        print("🔍 Starting security assessment...")
        result = await client.run_assessment(
            model_id="my-gpt-model",
            test_suites=["basic_security"],
            wait_for_completion=True,
            priority="normal"
        )

        # Display results
        print("\n🎉 Assessment Complete!")
        print(f"📊 Overall Security Score: {result.overall_score}/10")
        print(f"⚠️ Risk Level: {result.risk_level.value}")
        print(f"✅ Passed Tests: {result.passed_tests}")
        print(f"❌ Failed Tests: {result.failed_tests}")
        print(f"📋 Total Tests: {result.total_tests}")

        if result.recommendations:
            print("\n💡 Security Recommendations:")
            for i, rec in enumerate(result.recommendations[:3], 1):
                print(f"   {i}. {rec}")

        if result.report_url:
            print(f"\n🔗 Full Report: {result.report_url}")

if **name** == "**main**":
asyncio.run(main())