ModelRed SDK

import asyncio
from modelred import ModelRed

async def main():
async with ModelRed(api_key="mr_your_api_key_here") as client: # Register an AI model
await client.register_openai_model(
model_id="my-gpt-model",
api_key="sk-your-openai-key",
model_name="gpt-3.5-turbo"
)

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

        print(f"Security Score: {result.overall_score}/10")
        print(f"Risk Level: {result.risk_level.value}")

asyncio.run(main())