File size: 2,974 Bytes
5d1080e 79ec925 5d1080e 79ec925 5d1080e 79ec925 5d1080e 79ec925 5d1080e 79ec925 5d1080e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | import os
import sys
import time
def print_slow(str):
for letter in str:
sys.stdout.write(letter)
sys.stdout.flush()
time.sleep(0.02)
print()
def clear():
os.system('cls' if os.name == 'nt' else 'clear')
def check_runpod_balance():
# Placeholder for actual RunPod API balance check
# Normally we would use requests to hit https://api.runpod.io/v2/account/balance
return "$10.00"
def main():
clear()
print_slow("==================================================")
print_slow(" D E X T E R O S v1.1 ")
print_slow("==================================================")
print_slow("\n[SYSTEM] Initializing AI Research Swarm...")
time.sleep(0.5)
print_slow(f"[SYSTEM] RunPod Balance: {check_runpod_balance()}")
time.sleep(0.5)
print_slow("[SYSTEM] Environment: SECURE.\n")
print("Welcome, Director lyffseba.")
print("The lab is funded. The GPUs are standing by.\n")
while True:
print("----- COMMAND MENU -----")
print("1. [LAB 00] Run Local Diagnostics (Data & Tokenizer)")
print("2. [LAB 03] Compile Mojo Architecture locally (MAX Framework)")
print("3. [RUNPOD] Deploy RTX 3090 (Commence RLHF Training)")
print("4. [MODULAR] Deploy MAX Inference Endpoint")
print("5. [EXIT] Disconnect")
choice = input("\nAwaiting command (1-5): ")
if choice == '1':
print_slow("\n[EXECUTING] Launching Jupyter Lab for Lab 00...")
os.system("uv tool run --from jupyterlab jupyter-lab labs/00_getting_started.ipynb")
elif choice == '2':
print_slow("\n[EXECUTING] Compiling train.mojo via Pixi & MAX...")
os.system("cd labs/autoresearch/mojo && ~/.pixi/bin/pixi run mojo train.mojo")
print("\n")
elif choice == '3':
print_slow("\n[WARNING] This action will consume RunPod credits ($0.25/hr).")
confirm = input("Confirm deployment of 1x RTX 3090? (y/n): ")
if confirm.lower() == 'y':
print_slow("\n[DEPLOYING] Contacting RunPod API...")
# The actual runpodctl command goes here
print_slow("[SUCCESS] Pod 'autoresearch-3090' is booting up.")
print_slow("[NETWORK] Establishing SSH connection...")
print_slow("... Standing by for agent 'pi' to take over the instance.\n")
break
else:
print("\n[ABORTED] Deployment cancelled.\n")
elif choice == '4':
print_slow("\n[EXECUTING] Preparing Modular MAX serving script...")
print("Run: bash scripts/02_start_server.sh on your instance.\n")
elif choice == '5':
print_slow("\n[SYSTEM] Disconnecting... Goodbye, Director.")
break
else:
print("\n[ERROR] Invalid command.\n")
if __name__ == "__main__":
main()
|