Merge pull request #12 from ShaneIsley/claude/review-api-key-handling-CaUUv

refactor: leverage model registry for automatic API key handling

Author: ShaneIsley

examples/oolong_example.py

@@ -2,7 +2,6 @@
 Example: An example from the Oolong Benchmark from the RLM paper: https://arxiv.org/abs/2512.24601v1
 """
 
-import os
 import random
 import sys
 
@@ -31,10 +30,6 @@ def load_random_oolong_row() -> dict:
 
 
 def main():
-    api_key = os.environ.get("OPENAI_API_KEY")
-    if not api_key:
-        raise ValueError("OPENAI_API_KEY environment variable is not set.")
-
     # Load benchmark data
     print("Loading random row from dataset using shuffle...")
     row = load_random_oolong_row()
@@ -50,11 +45,11 @@ def main():
     logger = RLMLogger(log_dir="./logs")
 
     # Create RLM instance
+    # Note: API key is automatically loaded from OPENAI_API_KEY environment variable
     rlm = RLM(
         backend="openai",
         backend_kwargs={
             "model_name": "gpt-5-mini",
-            "api_key": api_key,
         },
         environment="subprocess",
         max_iterations=30,

examples/prime_repl_example.py

@@ -1,5 +1,3 @@
-import os
-
 from dotenv import load_dotenv
 
 from rlm import RLM
@@ -9,10 +7,10 @@
 
 logger = RLMLogger(log_dir="./logs")
 
+# Note: API key is automatically loaded from OPENAI_API_KEY environment variable
 rlm = RLM(
     backend="openai",
     backend_kwargs={
-        "api_key": os.getenv("OPENAI_API_KEY"),
         "model_name": "gpt-5-nano",
     },
     environment="prime",

examples/quickstart-sandbox-macos.py

@@ -16,8 +16,6 @@
     OPENAI_API_KEY=... python examples/quickstart-sandbox-macos.py
 """
 
-import os
-
 from dotenv import load_dotenv
 
 from rlm import RLM
@@ -27,11 +25,11 @@
 
 logger = RLMLogger(log_dir="./logs")
 
+# Note: API key is automatically loaded from OPENAI_API_KEY environment variable
 rlm = RLM(
     backend="openai",
     backend_kwargs={
         "model_name": "gpt-5-nano",
-        "api_key": os.getenv("OPENAI_API_KEY"),
     },
     environment="subprocess",  # Use SubprocessREPL instead of local
     environment_kwargs={

examples/quickstart.py

@@ -1,5 +1,3 @@
-import os
-
 from dotenv import load_dotenv
 
 from rlm import RLM
@@ -9,11 +7,11 @@
 
 logger = RLMLogger(log_dir="./logs")
 
+# Note: API key is automatically loaded from OPENAI_API_KEY environment variable
 rlm = RLM(
     backend="openai",  # or "portkey", etc.
     backend_kwargs={
         "model_name": "gpt-5-nano",
-        "api_key": os.getenv("OPENAI_API_KEY"),
     },
     environment="local",
     environment_kwargs={},