import asyncio, json, time
from rx_gs1 import MODEL, ROOT, service

async def main():
    sc = service()
    status = "errored"
    try:
        started = time.monotonic()
        try:
            tc = await sc.create_lora_training_client_async(base_model=MODEL, rank=32, seed=0)
            rec = dict(seed_kwarg=0, accepted=True, elapsed=time.monotonic()-started)
        except Exception as e:
            cause = e.__cause__
            rec = dict(seed_kwarg=0, accepted=False, error_type=type(e).__name__, error=str(e)[:500],
                       cause=(str(cause)[:500] if cause else None), elapsed=time.monotonic()-started)
        (ROOT / "seed-rejection.log").write_text(json.dumps(rec) + "\n")
        print(json.dumps(rec))
        status = "success"
    finally:
        await sc.close(status)

asyncio.run(main())