refactor constants out, lint

Author: alexzhang13

rlm/constants/__init__.py

@@ -7,10 +7,10 @@
 import modal
 import requests
 
-from rlm.constants.repl import APT_PACKAGES, PIP_PACKAGES
 from rlm.core.comms_utils import LMRequest, send_lm_request, send_lm_request_batched
 from rlm.core.types import REPLResult, RLMChatCompletion
 from rlm.environments.base_env import IsolatedEnv
+from rlm.environments.constants import APT_PACKAGES, PIP_PACKAGES
 
 # =============================================================================
 # Default Modal Image

rlm/constants/repl.py rlm/environments/constants.pyrlm/constants/repl.py renamed to rlm/environments/constants.py

@@ -13,17 +13,20 @@
 from typing import Any
 
 import requests
+from dotenv import load_dotenv
 from prime_sandboxes import (
     APIClient,
     BackgroundJob,
     CreateSandboxRequest,
     SandboxClient,
 )
 
-from rlm.constants.repl import APT_PACKAGES, PIP_PACKAGES
 from rlm.core.comms_utils import LMRequest, send_lm_request, send_lm_request_batched
 from rlm.core.types import REPLResult, RLMChatCompletion
 from rlm.environments.base_env import IsolatedEnv
+from rlm.environments.constants import APT_PACKAGES, PIP_PACKAGES
+
+load_dotenv()
 
 # =============================================================================
 # Broker Server Script (runs inside sandbox, handles LLM request queue)
@@ -280,12 +283,11 @@ def __init__(
         context_payload: dict | list | str | None = None,
         setup_code: str | None = None,
         network_access: bool = True,
-        api_key: str | None = None,
         persistent: bool = False,
         **kwargs: Any,
     ):
         super().__init__(persistent=persistent, **kwargs)
-        
+
         if persistent:
             raise NotImplementedError(
                 "Persistent REPLs are currently not supported for environment: PrimeREPL"