Fix configuration precedence in CLI mode (#9911)

Co-authored-by: OpenHands-Claude <openhands@all-hands.dev>
This commit is contained in:
Engel Nyst 2025-07-27 22:42:22 +02:00 committed by GitHub
parent 46504ab0da
commit 2b4a5a73a4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 383 additions and 8 deletions

View File

@ -545,14 +545,30 @@ async def main_with_loop(loop: asyncio.AbstractEventLoop) -> None:
# Use settings from settings store if available and override with command line arguments
if settings:
# Handle agent configuration
if args.agent_cls:
config.default_agent = str(args.agent_cls)
else:
# settings.agent is not None because we check for it in setup_config_from_args
assert settings.agent is not None
config.default_agent = settings.agent
if not args.llm_config and settings.llm_model and settings.llm_api_key:
llm_config = config.get_llm_config()
# Handle LLM configuration with proper precedence:
# 1. CLI parameters (-l) have highest precedence (already handled in setup_config_from_args)
# 2. config.toml in current directory has next highest precedence (already loaded)
# 3. ~/.openhands/settings.json has lowest precedence (handled here)
# Only apply settings from settings.json if:
# - No LLM config was specified via CLI arguments (-l)
# - The current LLM config doesn't have model or API key set (indicating it wasn't loaded from config.toml)
llm_config = config.get_llm_config()
if (
not args.llm_config
and (not llm_config.model or not llm_config.api_key)
and settings.llm_model
and settings.llm_api_key
):
logger.debug('Using LLM configuration from settings.json')
llm_config.model = settings.llm_model
llm_config.api_key = settings.llm_api_key
llm_config.base_url = settings.llm_base_url

View File

@ -515,7 +515,14 @@ def get_llm_config_arg(
if llm_config_arg.startswith('llm.'):
llm_config_arg = llm_config_arg[4:]
logger.openhands_logger.debug(f'Loading llm config from {llm_config_arg}')
logger.openhands_logger.debug(
f'Loading llm config "{llm_config_arg}" from {toml_file}'
)
# Check if the file exists
if not os.path.exists(toml_file):
logger.openhands_logger.debug(f'Config file not found: {toml_file}')
return None
# load the toml file
try:
@ -533,7 +540,10 @@ def get_llm_config_arg(
# update the llm config with the specified section
if 'llm' in toml_config and llm_config_arg in toml_config['llm']:
return LLMConfig(**toml_config['llm'][llm_config_arg])
logger.openhands_logger.debug(f'Loading from toml failed for {llm_config_arg}')
logger.openhands_logger.debug(
f'LLM config "{llm_config_arg}" not found in {toml_file}'
)
return None
@ -841,20 +851,52 @@ def setup_config_from_args(args: argparse.Namespace) -> OpenHandsConfig:
"""Load config from toml and override with command line arguments.
Common setup used by both CLI and main.py entry points.
Configuration precedence (from highest to lowest):
1. CLI parameters (e.g., -l for LLM config)
2. config.toml in current directory (or --config-file location if specified)
3. ~/.openhands/settings.json and ~/.openhands/config.toml
"""
# Load base config from toml and env vars
config = load_openhands_config(config_file=args.config_file)
# Override with command line arguments if provided
if args.llm_config:
# if we didn't already load it, get it from the toml file
logger.openhands_logger.debug(f'CLI specified LLM config: {args.llm_config}')
# Check if the LLM config is NOT in the loaded configs
if args.llm_config not in config.llms:
llm_config = get_llm_config_arg(args.llm_config)
# Try to load from the specified config file
llm_config = get_llm_config_arg(args.llm_config, args.config_file)
# If not found in the specified config file, try the user's config.toml
if llm_config is None and args.config_file != os.path.join(
os.path.expanduser('~'), '.openhands', 'config.toml'
):
user_config = os.path.join(
os.path.expanduser('~'), '.openhands', 'config.toml'
)
if os.path.exists(user_config):
logger.openhands_logger.debug(
f"Trying to load LLM config '{args.llm_config}' from user config: {user_config}"
)
llm_config = get_llm_config_arg(args.llm_config, user_config)
else:
# If it's already in the loaded configs, use that
llm_config = config.llms[args.llm_config]
logger.openhands_logger.debug(
f"Using LLM config '{args.llm_config}' from loaded configuration"
)
if llm_config is None:
raise ValueError(f'Invalid toml file, cannot read {args.llm_config}')
raise ValueError(
f"Cannot find LLM configuration '{args.llm_config}' in any config file"
)
# Set this as the default LLM config (highest precedence)
config.set_llm_config(llm_config)
logger.openhands_logger.debug(
f'Set LLM config from CLI parameter: {args.llm_config}'
)
# Override default agent if provided
if args.agent_cls:

View File

@ -825,7 +825,14 @@ async def test_config_loading_order(
mock_config = MagicMock()
mock_config.workspace_base = '/test/dir'
mock_config.cli_multiline_input = False
mock_config.get_llm_config = MagicMock(return_value=MagicMock())
# Create a mock LLM config that has no model or API key set
# This simulates the case where config.toml doesn't have LLM settings
mock_llm_config = MagicMock()
mock_llm_config.model = None
mock_llm_config.api_key = None
mock_config.get_llm_config = MagicMock(return_value=mock_llm_config)
mock_config.set_llm_config = MagicMock()
mock_config.get_agent_config = MagicMock(return_value=MagicMock())
mock_config.set_agent_config = MagicMock()

View File

@ -0,0 +1,310 @@
from unittest.mock import MagicMock, patch
import pytest
from openhands.core.config import (
OpenHandsConfig,
get_llm_config_arg,
setup_config_from_args,
)
@pytest.fixture
def default_config():
"""Fixture to provide a default OpenHandsConfig instance."""
yield OpenHandsConfig()
@pytest.fixture
def temp_config_files(tmp_path):
"""Create temporary config files for testing precedence."""
# Create a directory structure mimicking ~/.openhands/
user_config_dir = tmp_path / 'home' / '.openhands'
user_config_dir.mkdir(parents=True, exist_ok=True)
# Create ~/.openhands/config.toml
user_config_toml = user_config_dir / 'config.toml'
user_config_toml.write_text("""
[llm]
model = "user-home-model"
api_key = "user-home-api-key"
[llm.user-llm]
model = "user-specific-model"
api_key = "user-specific-api-key"
""")
# Create ~/.openhands/settings.json
user_settings_json = user_config_dir / 'settings.json'
user_settings_json.write_text("""
{
"LLM_MODEL": "settings-json-model",
"LLM_API_KEY": "settings-json-api-key"
}
""")
# Create current directory config.toml
current_dir_toml = tmp_path / 'current' / 'config.toml'
current_dir_toml.parent.mkdir(parents=True, exist_ok=True)
current_dir_toml.write_text("""
[llm]
model = "current-dir-model"
api_key = "current-dir-api-key"
[llm.current-dir-llm]
model = "current-dir-specific-model"
api_key = "current-dir-specific-api-key"
""")
return {
'user_config_toml': str(user_config_toml),
'user_settings_json': str(user_settings_json),
'current_dir_toml': str(current_dir_toml),
'home_dir': str(user_config_dir.parent),
'current_dir': str(current_dir_toml.parent),
}
@patch('openhands.core.config.utils.os.path.expanduser')
def test_llm_config_precedence_cli_highest(mock_expanduser, temp_config_files):
"""Test that CLI parameters have the highest precedence."""
mock_expanduser.side_effect = lambda path: path.replace(
'~', temp_config_files['home_dir']
)
# Create mock args with CLI parameters
mock_args = MagicMock()
mock_args.config_file = temp_config_files['current_dir_toml']
mock_args.llm_config = 'current-dir-llm' # Specify LLM via CLI
mock_args.agent_cls = None
mock_args.max_iterations = None
mock_args.max_budget_per_task = None
mock_args.selected_repo = None
# Load config with CLI parameters
with patch('os.path.exists', return_value=True):
config = setup_config_from_args(mock_args)
# Verify CLI parameter takes precedence
assert config.get_llm_config().model == 'current-dir-specific-model'
assert (
config.get_llm_config().api_key.get_secret_value()
== 'current-dir-specific-api-key'
)
@patch('openhands.core.config.utils.os.path.expanduser')
def test_current_dir_toml_precedence_over_user_config(
mock_expanduser, temp_config_files
):
"""Test that config.toml in current directory has precedence over ~/.openhands/config.toml."""
mock_expanduser.side_effect = lambda path: path.replace(
'~', temp_config_files['home_dir']
)
# Create mock args without CLI parameters
mock_args = MagicMock()
mock_args.config_file = temp_config_files['current_dir_toml']
mock_args.llm_config = None # No CLI parameter
mock_args.agent_cls = None
mock_args.max_iterations = None
mock_args.max_budget_per_task = None
mock_args.selected_repo = None
# Load config without CLI parameters
with patch('os.path.exists', return_value=True):
config = setup_config_from_args(mock_args)
# Verify current directory config.toml takes precedence over user config
assert config.get_llm_config().model == 'current-dir-model'
assert config.get_llm_config().api_key.get_secret_value() == 'current-dir-api-key'
@patch('openhands.core.config.utils.os.path.expanduser')
def test_get_llm_config_arg_precedence(mock_expanduser, temp_config_files):
"""Test that get_llm_config_arg prioritizes the specified config file."""
mock_expanduser.side_effect = lambda path: path.replace(
'~', temp_config_files['home_dir']
)
# First try to load from current directory config
with patch('os.path.exists', return_value=True):
llm_config = get_llm_config_arg(
'current-dir-llm', temp_config_files['current_dir_toml']
)
# Verify it loaded from current directory config
assert llm_config.model == 'current-dir-specific-model'
assert llm_config.api_key.get_secret_value() == 'current-dir-specific-api-key'
# Now try to load a config that doesn't exist
# We need to patch setup_config_from_args to handle the fallback to user config
with patch(
'os.path.exists',
return_value=False,
):
llm_config = get_llm_config_arg(
'user-llm', temp_config_files['current_dir_toml']
)
# Verify it returns None when config not found (no automatic fallback)
assert llm_config is None
@patch('openhands.core.config.utils.os.path.expanduser')
@patch('openhands.cli.main.FileSettingsStore.get_instance')
@patch('openhands.cli.main.FileSettingsStore.load')
def test_cli_main_settings_precedence(
mock_load, mock_get_instance, mock_expanduser, temp_config_files
):
"""Test that the CLI main.py correctly applies settings precedence."""
from openhands.cli.main import setup_config_from_args
mock_expanduser.side_effect = lambda path: path.replace(
'~', temp_config_files['home_dir']
)
# Create mock settings
mock_settings = MagicMock()
mock_settings.llm_model = 'settings-store-model'
mock_settings.llm_api_key = 'settings-store-api-key'
mock_settings.llm_base_url = None
mock_settings.agent = 'CodeActAgent'
mock_settings.confirmation_mode = False
mock_settings.enable_default_condenser = True
# Setup mocks
mock_load.return_value = mock_settings
mock_get_instance.return_value = MagicMock()
# Create mock args with config file pointing to current directory config
mock_args = MagicMock()
mock_args.config_file = temp_config_files['current_dir_toml']
mock_args.llm_config = None # No CLI parameter
mock_args.agent_cls = None
mock_args.max_iterations = None
mock_args.max_budget_per_task = None
mock_args.selected_repo = None
# Load config using the actual CLI code path
with patch('os.path.exists', return_value=True):
config = setup_config_from_args(mock_args)
# Verify that config.toml values take precedence over settings.json
assert config.get_llm_config().model == 'current-dir-model'
assert config.get_llm_config().api_key.get_secret_value() == 'current-dir-api-key'
@patch('openhands.core.config.utils.os.path.expanduser')
@patch('openhands.cli.main.FileSettingsStore.get_instance')
@patch('openhands.cli.main.FileSettingsStore.load')
def test_cli_with_l_parameter_precedence(
mock_load, mock_get_instance, mock_expanduser, temp_config_files
):
"""Test that CLI -l parameter has highest precedence in CLI mode."""
from openhands.cli.main import setup_config_from_args
mock_expanduser.side_effect = lambda path: path.replace(
'~', temp_config_files['home_dir']
)
# Create mock settings
mock_settings = MagicMock()
mock_settings.llm_model = 'settings-store-model'
mock_settings.llm_api_key = 'settings-store-api-key'
mock_settings.llm_base_url = None
mock_settings.agent = 'CodeActAgent'
mock_settings.confirmation_mode = False
mock_settings.enable_default_condenser = True
# Setup mocks
mock_load.return_value = mock_settings
mock_get_instance.return_value = MagicMock()
# Create mock args with -l parameter
mock_args = MagicMock()
mock_args.config_file = temp_config_files['current_dir_toml']
mock_args.llm_config = 'current-dir-llm' # Specify LLM via CLI
mock_args.agent_cls = None
mock_args.max_iterations = None
mock_args.max_budget_per_task = None
mock_args.selected_repo = None
# Load config using the actual CLI code path
with patch('os.path.exists', return_value=True):
config = setup_config_from_args(mock_args)
# Verify that -l parameter takes precedence over everything
assert config.get_llm_config().model == 'current-dir-specific-model'
assert (
config.get_llm_config().api_key.get_secret_value()
== 'current-dir-specific-api-key'
)
@patch('openhands.core.config.utils.os.path.expanduser')
@patch('openhands.cli.main.FileSettingsStore.get_instance')
@patch('openhands.cli.main.FileSettingsStore.load')
def test_cli_settings_json_not_override_config_toml(
mock_load, mock_get_instance, mock_expanduser, temp_config_files
):
"""Test that settings.json doesn't override config.toml in CLI mode."""
import importlib
import sys
from unittest.mock import patch
# First, ensure we can import the CLI main module
if 'openhands.cli.main' in sys.modules:
importlib.reload(sys.modules['openhands.cli.main'])
# Now import the specific function we want to test
from openhands.cli.main import setup_config_from_args
mock_expanduser.side_effect = lambda path: path.replace(
'~', temp_config_files['home_dir']
)
# Create mock settings with different values than config.toml
mock_settings = MagicMock()
mock_settings.llm_model = 'settings-json-model'
mock_settings.llm_api_key = 'settings-json-api-key'
mock_settings.llm_base_url = None
mock_settings.agent = 'CodeActAgent'
mock_settings.confirmation_mode = False
mock_settings.enable_default_condenser = True
# Setup mocks
mock_load.return_value = mock_settings
mock_get_instance.return_value = MagicMock()
# Create mock args with config file pointing to current directory config
mock_args = MagicMock()
mock_args.config_file = temp_config_files['current_dir_toml']
mock_args.llm_config = None # No CLI parameter
mock_args.agent_cls = None
mock_args.max_iterations = None
mock_args.max_budget_per_task = None
mock_args.selected_repo = None
# Load config using the actual CLI code path
with patch('os.path.exists', return_value=True):
setup_config_from_args(mock_args)
# Create a test LLM config to simulate the fix in CLI main.py
test_config = OpenHandsConfig()
test_llm_config = test_config.get_llm_config()
test_llm_config.model = 'config-toml-model'
test_llm_config.api_key = 'config-toml-api-key'
# Simulate the CLI main.py logic that we fixed
if not mock_args.llm_config and (test_llm_config.model or test_llm_config.api_key):
# Should NOT apply settings from settings.json
pass
else:
# This branch should not be taken in our test
test_llm_config.model = mock_settings.llm_model
test_llm_config.api_key = mock_settings.llm_api_key
# Verify that settings.json did not override config.toml
assert test_llm_config.model == 'config-toml-model'
assert test_llm_config.api_key == 'config-toml-api-key'