From 54a3d45bbf1b8630812feef00001ec4bc52eb94c Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Tue, 16 Dec 2025 10:14:37 -0800 Subject: [PATCH] fix tests --- arch/tools/cli/utils.py | 4 ++++ arch/tools/test/test_config_generator.py | 27 ++++++++++-------------- 2 files changed, 15 insertions(+), 16 deletions(-) diff --git a/arch/tools/cli/utils.py b/arch/tools/cli/utils.py index 2f29b16e..7fd510c4 100644 --- a/arch/tools/cli/utils.py +++ b/arch/tools/cli/utils.py @@ -57,6 +57,10 @@ def convert_legacy_listeners( "timeout": "30s", } + # Handle None case + if listeners is None: + return [llm_gateway_listener], llm_gateway_listener, prompt_gateway_listener + if isinstance(listeners, dict): # legacy listeners # check if type is array or object diff --git a/arch/tools/test/test_config_generator.py b/arch/tools/test/test_config_generator.py index 7016a34f..5a19c424 100644 --- a/arch/tools/test/test_config_generator.py +++ b/arch/tools/test/test_config_generator.py @@ -94,21 +94,16 @@ def test_validate_and_render_happy_path_agent_config(monkeypatch): version: v0.3.0 agents: - - name: query_rewriter - kind: openai - endpoint: http://localhost:10500 - - name: context_builder - kind: openai - endpoint: http://localhost:10501 - - name: response_generator - kind: openai - endpoint: http://localhost:10502 - - name: research_agent - kind: openai - endpoint: http://localhost:10500 - - name: input_guard_rails - kind: openai - endpoint: http://localhost:10503 + - id: query_rewriter + url: http://localhost:10500 + - id: context_builder + url: http://localhost:10501 + - id: response_generator + url: http://localhost:10502 + - id: research_agent + url: http://localhost:10500 + - id: input_guard_rails + url: http://localhost:10503 listeners: - name: tmobile @@ -156,7 +151,7 @@ listeners: mock.mock_open().return_value, # ARCH_CONFIG_FILE_RENDERED (write) ] with mock.patch("builtins.open", m_open): - with mock.patch("config_generator.Environment"): + with mock.patch("cli.config_generator.Environment"): validate_and_render_schema()