from fleet_rlm import ModalInterpreter

with ModalInterpreter(timeout=600) as interp:
    result = interp.execute('''
import math
SUBMIT(answer=math.factorial(15))
''')
    print(result.answer)  # 1307674368000

with ModalInterpreter(timeout=600, volume_name="my-data") as interp:
    # Write data
    interp.execute('''
import json
data = {"processed": True, "count": 42}
with open('/data/results.json', 'w') as f:
    json.dump(data, f)
SUBMIT(status="saved")
''')

# Later session - data persists
with ModalInterpreter(timeout=600, volume_name="my-data") as interp:
    result = interp.execute('''
import json
with open('/data/results.json') as f:
    data = json.load(f)
SUBMIT(data=data)
''')
    print(result.data)  # {"processed": True, "count": 42}

# Read local file, execute in sandbox
code = open("scripts/analysis.py").read()
with ModalInterpreter(timeout=600) as interp:
    result = interp.execute(code)

# Basic task
uv run fleet-rlm run-basic --question "What are the first 12 Fibonacci numbers?"

# With volume
uv run fleet-rlm run-basic \
    --question "Calculate factorial of 20" \
    --volume-name my-data

with ModalInterpreter(timeout=600, volume_name="pipeline") as interp:
    # Step 1: Generate data
    interp.execute('''
import json
data = [{"id": i, "value": i**2} for i in range(100)]
with open('/data/raw.json', 'w') as f:
    json.dump(data, f)
SUBMIT(count=len(data))
''')

    # Step 2: Process (same sandbox, same volume)
    result = interp.execute('''
import json
with open('/data/raw.json') as f:
    data = json.load(f)
filtered = [d for d in data if d["value"] > 50]
SUBMIT(filtered_count=len(filtered))
''')
    print(result.filtered_count)

with ModalInterpreter(timeout=600) as interp:
    # Buffers persist across execute() calls within same sandbox
    interp.execute('add_buffer("findings", "Step 1: setup complete")')
    interp.execute('add_buffer("findings", "Step 2: data loaded")')
    result = interp.execute('''
items = get_buffer("findings")
SUBMIT(log=items)
''')
    print(result.log)  # ["Step 1: setup complete", "Step 2: data loaded"]