import asyncio
import os
from ag2 import Agent
from ag2.config import OpenAIConfig

api_key = os.getenv("RADIUM_API_KEY")
if not api_key:
    raise SystemExit("Set RADIUM_API_KEY before running this program.")

async def main() -> None:
    assistant = Agent(
        "radium_assistant",
        prompt="Follow the user's instruction exactly.",
        config=OpenAIConfig(
            model=os.getenv("RADIUM_MODEL", "hal-1.0"),
            api_key=api_key,
            base_url="https://api.radium.cloud/v1",
            max_tokens=512,
            temperature=0,
            timeout=90,
        ),
    )
    reply = await assistant.ask("Reply with exactly: Radium connected.")
    if not reply.body or not reply.body.strip():
        raise RuntimeError("Radium returned no visible text.")
    print(f"RADIUM_RESPONSE: {reply.body.strip()}")

asyncio.run(main())

python main.py

RADIUM_RESPONSE: Radium connected.

export RADIUM_MODEL="tycho-1.0"  # or hal-1.0 or clarke-1.0
python main.py

config = OpenAIConfig(
    model="hal-1.0",
    api_key=os.environ["RADIUM_API_KEY"],
    base_url="https://api.radium.cloud/v1",
    max_tokens=512,
)
agent = Agent("radium_assistant", config=config)