modalsolve.py

import importlib
import subprocess
import sys

import modal

image = (
    modal.Image.debian_slim(python_version="3.10")
    .apt_install("git")
    .copy_local_file("/Users/robcheung/.ssh/id_rsa", "/root/.ssh/id_rsa")
    .run_commands("chmod 600 /root/.ssh/id_rsa", "ssh-keyscan -t rsa github.com >> /root/.ssh/known_hosts")
    .run_commands("git clone git@github.com:SubstrateLabs/rob-agi.git /app/rob_agi")
    .run_commands('pip install -e "/app/rob_agi"')
    .run_commands("git config --global user.email 'kousun12@gmail.com'", "git config --global user.name 'solver-rob'")
    .env({"GIT_COMMITTER_NAME": "solver-rob"})
)
app = modal.App("arc_solver", image=image)


def reload_repo():
    repo_path = "/app/rob_agi"
    try:
        subprocess.run(["git", "-C", repo_path, "pull", "origin", "master"], check=True)
        print("Git pull successful")
    except subprocess.CalledProcessError as e:
        print(f"Git pull failed: {e}")
        sys.exit(1)
    if "mymodule" in sys.modules:
        importlib.reload(sys.modules["mymodule"])


@app.function(
    secrets=[modal.Secret.from_name("llm-keys")],
    timeout=60 * 10,
)
def foo():
    reload_repo()
    from rob_agi.arc_util import load_task_set
    from rob_agi.progressive_solve import Solver

    # challenge_id = "c59eb873"  # easy
    challenge_id = "776ffc46"  # hard
    task_set = "training"
    challenges, solutions = load_task_set(task_set_name=task_set)
    c = challenges[challenge_id]
    sln = solutions.get(challenge_id)
    solver = Solver(c, sln)
    solver.run_solve(max_tries=1)

    return "hello"


@app.local_entrypoint()
def localfn():
    h = foo.remote()
    print("res", h)