diff --git a/ra_aid/agents/implementation_agent.py b/ra_aid/agents/implementation_agent.py index f0bffff..352e241 100644 --- a/ra_aid/agents/implementation_agent.py +++ b/ra_aid/agents/implementation_agent.py @@ -132,8 +132,8 @@ def run_task_implementation_agent( env_inv = get_env_inv() # Get model configuration to check for reasoning_assist_default - provider = get_config_repository().get("provider", "") - model_name = get_config_repository().get("model", "") + provider = get_config_repository().get("expert_provider", "") + model_name = get_config_repository().get("expert_model", "") logger.debug("Checking for reasoning_assist_default on %s/%s", provider, model_name) model_config = {} diff --git a/ra_aid/agents/planning_agent.py b/ra_aid/agents/planning_agent.py index 9651d26..e6dd8e8 100644 --- a/ra_aid/agents/planning_agent.py +++ b/ra_aid/agents/planning_agent.py @@ -92,8 +92,8 @@ def run_planning_agent( ) # Get model configuration - provider = get_config_repository().get("provider", "") - model_name = get_config_repository().get("model", "") + provider = get_config_repository().get("expert_provider", "") + model_name = get_config_repository().get("expert_model", "") logger.debug("Checking for reasoning_assist_default on %s/%s", provider, model_name) # Get model configuration to check for reasoning_assist_default diff --git a/ra_aid/agents/research_agent.py b/ra_aid/agents/research_agent.py index 61fbc05..b5903a7 100644 --- a/ra_aid/agents/research_agent.py +++ b/ra_aid/agents/research_agent.py @@ -144,8 +144,8 @@ def run_research_agent( ) # Get model info for reasoning assistance configuration - provider = get_config_repository().get("provider", "") - model_name = get_config_repository().get("model", "") + provider = get_config_repository().get("expert_provider", "") + model_name = get_config_repository().get("expert_model", "") # Get model configuration to check for reasoning_assist_default model_config = {}