from free_claude_code.core.anthropic.models import MessagesRequest
from free_claude_code.config.nim import NimSettings
from free_claude_code.providers.nvidia_nim.request_options import build_nim_request_body

req = MessagesRequest(
    model="nvidia/nemotron",
    messages=[{"role": "user", "content": "hi"}],
    max_tokens=10
)

nim = NimSettings()

try:
    body = build_nim_request_body(req, nim, thinking_enabled=False)
    print("Success:")
    print(body)
except Exception as e:
    import traceback
    traceback.print_exc()