init template from debug_gym module

Author: matheper

README.md

@@ -18,7 +18,7 @@ To install the development dependencies:
 
 First, create an LLM config template by running the `debug-gym-init-llm-config` entrypoint:
 
-    debug-gym-init-llm-config ~/.config/debug_gym
+    python -m debug_gym.init_llm_config ~/.config/debug_gym
 
 > [!TIP]
 > Run `debug-gym-init-llm-config --help` for more options. By default, the template is created at `~/.config/debug_gym/llm.yaml`, but you can specify any directory.

debug_gym/entrypoints.py debug_gym/init_llm_config.py

@@ -48,3 +48,7 @@ def init_llm_config(dest_dir: str = None):
         print(f"LLM config template already exists at `{destination}`.")
 
     print("Please edit the file to configure your LLM settings.")
+
+
+if __name__ == "__main__":
+    init_llm_config()

pyproject.toml

@@ -22,9 +22,6 @@ dependencies = {file = ["requirements.txt"]}
 [tool.setuptools.packages.find]
 exclude = [".*tests.*", "exps/*", "output_*"]
 
-[project.scripts]
-debug-gym-init-llm-config = "debug_gym.entrypoints:init_llm_config"
-
 [project.optional-dependencies]
 dev = [
     "pytest",

tests/agents/test_llm_api.py

@@ -2,7 +2,6 @@
 
 import pytest
 import yaml
-from openai import RateLimitError
 
 from debug_gym.agents.llm_api import (
     LLM,

tests/test_entrypoints.py tests/test_init_llm_config.py

@@ -3,7 +3,7 @@
 import pytest
 
 from debug_gym.agents.llm_api import LLM_CONFIG_TEMPLATE
-from debug_gym.entrypoints import init_llm_config
+from debug_gym.init_llm_config import init_llm_config
 
 
 @pytest.fixture
@@ -47,23 +47,24 @@ def test_init_llm_config_with_dest_named(tmp_path, mock_argv, capsys):
 
 
 def test_init_llm_config_override(tmp_path, monkeypatch, mock_argv, capsys):
-    monkeypatch.setattr("debug_gym.entrypoints.LLM_CONFIG_TEMPLATE", "initial content")
+    llm_template_path = "debug_gym.init_llm_config.LLM_CONFIG_TEMPLATE"
+    monkeypatch.setattr(llm_template_path, "config")
 
     destination = tmp_path / "destination"
     # os.makedirs(destination, exist_ok=True)
     destination_file = destination / "llm.yaml"
 
     mock_argv(["--dest", str(destination)])
     init_llm_config()  # First copy should work
-    assert destination_file.read_text() == "initial content"
+    assert destination_file.read_text() == "config"
     assert f"LLM config template created" in capsys.readouterr().out
 
-    monkeypatch.setattr("debug_gym.entrypoints.LLM_CONFIG_TEMPLATE", "new content")
+    monkeypatch.setattr(llm_template_path, "new config")
     init_llm_config()  # No force, should not override
-    assert destination_file.read_text() == "initial content"
+    assert destination_file.read_text() == "config"
     assert f"LLM config template already exists" in capsys.readouterr().out
 
     mock_argv(["--dest", str(destination), "--force"])
     init_llm_config()  # Force override
-    assert destination_file.read_text() == "new content"
+    assert destination_file.read_text() == "new config"
     assert f"LLM config template overridden" in capsys.readouterr().out